Daniel@0: load ('def.lefty');
Daniel@0: definit ();
Daniel@0: #
Daniel@0: # initialize window data
Daniel@0: #
Daniel@0: canvas = defcanvas;
Daniel@0: wrect = [0 = ['x' = -5; 'y' = 0;]; 1 = ['x' = 410; 'y' = 500;];];
Daniel@0: setwidgetattr (canvas, ['window' = wrect;]);
Daniel@0: #
Daniel@0: # data structures
Daniel@0: #
Daniel@0: nodearray = [];
Daniel@0: nodenum = 0;
Daniel@0: dist = ['x' = 40; 'y' = 40;];
Daniel@0: defsize = ['x' = 10; 'y' = 10;];
Daniel@0: fontname = 'fixed';
Daniel@0: fontsize = 18;
Daniel@0: tree = null;
Daniel@0: 
Daniel@0: # drawing functions
Daniel@0: #
Daniel@0: boxnode = function (node) {
Daniel@0:     local center;
Daniel@0:     box (canvas, node, node.rect, ['color' = 0; 'fill' = 'on';]);
Daniel@0:     box (canvas, node, node.rect);
Daniel@0:     center = [
Daniel@0:         'x' = (node.rect[0].x + node.rect[1].x) / 2;
Daniel@0:         'y' = (node.rect[0].y + node.rect[1].y) / 2;
Daniel@0:     ];
Daniel@0:     if (node.name)
Daniel@0:         text (canvas, node, center, node.name, fontname, fontsize, 'cc');
Daniel@0: };
Daniel@0: circlenode = function (node) {
Daniel@0:     local center, radius;
Daniel@0:     center = [
Daniel@0:         'x' = (node.rect[0].x + node.rect[1].x) / 2;
Daniel@0:         'y' = (node.rect[0].y + node.rect[1].y) / 2;
Daniel@0:     ];
Daniel@0:     radius = [
Daniel@0:         'x' = center.x - node.rect[0].x;
Daniel@0:         'y' = center.y - node.rect[0].y;
Daniel@0:     ];
Daniel@0:     arc (canvas, node, center, radius, ['color' = 0; 'fill' = 'on';]);
Daniel@0:     arc (canvas, node, center, radius);
Daniel@0:     if (node.name)
Daniel@0:         text (canvas, node, center, node.name, fontname, fontsize, 'cc');
Daniel@0: };
Daniel@0: drawnode = boxnode;
Daniel@0: drawedge = function (node1, node2) {
Daniel@0:     line (canvas, null,
Daniel@0:             [
Daniel@0:                 'x' = (node1.rect[1].x + node1.rect[0].x) / 2;
Daniel@0:                 'y' = node1.rect[0].y;
Daniel@0:             ], [
Daniel@0:                 'x' = (node2.rect[1].x + node2.rect[0].x) / 2;
Daniel@0:                 'y' = node2.rect[1].y;
Daniel@0:             ]);
Daniel@0: };
Daniel@0: drawtree = function (node) {
Daniel@0:     local i;
Daniel@0:     for (i in nodearray)
Daniel@0:         drawnode (nodearray[i]);
Daniel@0:     drawtreerec (node);
Daniel@0: };
Daniel@0: drawtreerec = function (node) {
Daniel@0:     local i, n;
Daniel@0:     if ((n = tablesize (node.ch)) > 0) {
Daniel@0:         for (i = 0; i < n; i = i + 1) {
Daniel@0:             drawedge (node, node.ch[i]);
Daniel@0:             drawtreerec (node.ch[i]);
Daniel@0:         }
Daniel@0:     }
Daniel@0: };
Daniel@0: redraw = function (c) {
Daniel@0:     if (tree)
Daniel@0:         drawtree (tree);
Daniel@0: };
Daniel@0: 
Daniel@0: # layout functions
Daniel@0: #
Daniel@0: complayout = function () {
Daniel@0:     leafx = 0;
Daniel@0:     leafrank = 0;
Daniel@0:     dolayout (tree, wrect[1].y - 10);
Daniel@0:     remove ('leafx');
Daniel@0:     remove ('leafrank');
Daniel@0: };
Daniel@0: dolayout = function (node, pary) {
Daniel@0:     local r, n, i, size, lchp, rchp;
Daniel@0:     size = nodesize (node);
Daniel@0:     if (node.chn > 0) {
Daniel@0:         for (i = 0; i < node.chn; i = i + 1)
Daniel@0:             dolayout (node.ch[i], pary - size.y - dist.y);
Daniel@0:         node.rank = (node.ch[0].rank + node.ch[node.chn - 1].rank) / 2;
Daniel@0:         lchp = node.ch[0].rect;
Daniel@0:         rchp = node.ch[node.chn - 1].rect;
Daniel@0:         r[0].x = lchp[0].x + ((rchp[1].x - lchp[0].x) - size.x) / 2;
Daniel@0:         r[0].y = pary - size.y;
Daniel@0:         r[1].x = r[0].x + size.x;
Daniel@0:         r[1].y = pary;
Daniel@0:         node.rect = r;
Daniel@0:     } else {
Daniel@0:         node.rank = leafrank;
Daniel@0:         r[0].x = leafx;
Daniel@0:         r[0].y = pary - size.y;
Daniel@0:         r[1].x = r[0].x + size.x;
Daniel@0:         r[1].y = pary;
Daniel@0:         leafrank = leafrank + 1;
Daniel@0:         leafx = r[1].x + dist.x;
Daniel@0:         node.rect = r;
Daniel@0:     }
Daniel@0: };
Daniel@0: 
Daniel@0: # editing functions
Daniel@0: #
Daniel@0: inode = function (point, name) {
Daniel@0:     local i, nnum, size;
Daniel@0:     nnum = nodenum;
Daniel@0:     if (~name)
Daniel@0:         name = ask ('give name of node:');
Daniel@0:     nodearray[nnum].ch = [];
Daniel@0:     nodearray[nnum].chn = 0;
Daniel@0:     nodearray[nnum].name = name;
Daniel@0:     size = nodesize (nodearray[nnum]);
Daniel@0:     nodearray[nnum].rect[0] = point;
Daniel@0:     nodearray[nnum].rect[1] = ['x' = point.x + size.x; 'y' = point.y + size.y;];
Daniel@0:     nodenum = nodenum + 1;
Daniel@0:     if (~tree) {
Daniel@0:         tree = nodearray[nnum];
Daniel@0:         tree.depth = 0;
Daniel@0:         complayout ();
Daniel@0:         drawtree (tree);
Daniel@0:     } else
Daniel@0:         drawtree (nodearray[nnum]);
Daniel@0:     return nodearray[nnum];
Daniel@0: };
Daniel@0: iedge = function (node1, node2) {
Daniel@0:     node1.ch[node1.chn] = node2;
Daniel@0:     node1.chn = node1.chn + 1;
Daniel@0:     node2.depth = node1.depth + 1;
Daniel@0:     complayout ();
Daniel@0:     clear (canvas);
Daniel@0:     drawtree (tree);
Daniel@0: };
Daniel@0: fix = function (node, op, np) {
Daniel@0:     if (node.depth ~= 0)
Daniel@0:         dist.y = dist.y + (op.y - np.y) / node.depth;
Daniel@0:     if (node.rank ~= 0)
Daniel@0:         dist.x = dist.x + (np.x - op.x) / node.rank;
Daniel@0:     complayout ();
Daniel@0:     clear (canvas);
Daniel@0:     drawtree (tree);
Daniel@0: };
Daniel@0: nodesize = function (node) {
Daniel@0:     local siz;
Daniel@0:     if (~(siz = textsize (canvas, node.name, fontname, fontsize)))
Daniel@0:         siz = defsize;
Daniel@0:     else {
Daniel@0:         siz.x = siz.x + 8;
Daniel@0:         siz.y = siz.y + 8;
Daniel@0:     }
Daniel@0:     return siz;
Daniel@0: };
Daniel@0: changenode = function (nodestyle) {
Daniel@0:     drawnode = nodestyle;
Daniel@0:     clear (canvas);
Daniel@0:     drawtree (tree);
Daniel@0: };
Daniel@0: 
Daniel@0: # user interface functions
Daniel@0: #
Daniel@0: leftdown = function (data) {
Daniel@0:     if (~data.obj)
Daniel@0:         inode (data.pos, null);
Daniel@0: };
Daniel@0: leftup = function (data) {
Daniel@0:     if (data.pobj)
Daniel@0:         fix (data.pobj, data.ppos, data.pos);
Daniel@0: };
Daniel@0: middleup = function (data) {
Daniel@0:     if (data.pobj & data.obj)
Daniel@0:         iedge (data.pobj, data.obj);
Daniel@0: };
Daniel@0: dops = function () {
Daniel@0:     local s;
Daniel@0: 
Daniel@0:     s = ['x' = 8 * 300; 'y' = 10.5 * 300;];
Daniel@0:     fontname = 'Times-Roman';
Daniel@0:     canvas = createwidget (-1, ['type' = 'ps'; 'size' = s;]);
Daniel@0:     setwidgetattr (canvas, ['window' = wrect;]);
Daniel@0:     drawtree (tree);
Daniel@0:     destroywidget (canvas);
Daniel@0:     canvas=defcanvas;
Daniel@0:     fontname = 'fixed';
Daniel@0: };