aboutsummaryrefslogtreecommitdiffstats
path: root/src/layout.js
blob: 95237c5b14860197bc600c113bb53a47665f72ad (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
import Viz from './viz.es';

const graphviz = new Viz({ workerURL: 'lib/lite.render.js' });

export const init = () => {
  return 'digraph { node [shape="rectangle"];\n';
}

export const addNode = (dot, node) => {
  const id = node.data.id;
  dot += `  "${id}" [fixedsize=true; width=${node.dom.offsetWidth / 72}; height=${node.dom.offsetHeight / 72}];\n`;

  for (const replyId of node.data.replies || [])
    dot += `  "${id}" -> "${replyId}";\n`;

  return dot;
};

const toNumber = (i) => Number(i);

export const render = async (dot, canvas, setPosition, setSize) => {
  dot += '}';

  const info = await graphviz.renderJSONObject(dot, { format: 'json0' });
  window.info = info;

  const [_x, _y, width, height] = info.bb.split(',').map(toNumber);

  setSize(width, height);

  for (const object of info.objects) {
    const [x, y] = object.pos.split(',').map(toNumber);
    setPosition(object.name, x, height - y);
  }

  const ctx = canvas.getContext('2d');
  ctx.lineWidth = 4;
  ctx.lineCap = 'round';
  ctx.lineJoin = 'round';
  ctx.strokeStyle = '#696969';
  ctx.fillStyle = '#696969';

  for (const edge of info.edges) {
    const pos = edge.pos.slice(2);
    const points = pos.split(' ').map(p => {
      const [x, y] = p.split(',');
      return [toNumber(x), height - toNumber(y)];
    });

    ctx.beginPath();
    ctx.moveTo(...points[1]);
    for (let i = 4; i < points.length; i += 3) {
      const [ax, ay] = points[i - 2];
      const [bx, by] = points[i - 1];
      const [x, y] = points[i];
      ctx.bezierCurveTo(ax, ay, bx, by, x, y);
    }
    ctx.stroke();

    const [lx, ly] = points[points.length - 1];
    const [tx, ty] = points[0];
    const [dx, dy] = [tx - lx, ty - ly];
    const [hx, hy] = [dy*0.7, dx*-0.7];

    ctx.beginPath();
    ctx.moveTo(lx + dx*0.8, ly + dy*0.8);
    ctx.lineTo(lx - dx*0.5 + hx, ly - dy*0.5 + hy);
    ctx.lineTo(lx - dx*0.5 - hx, ly - dy*0.5 - hy);
    ctx.fill();
  }
}