!7

sS text = [[
  One node "A".
  One node "B".
  B to the right of A.
]];

static new TreeSet<S> nodes;
static new LinkedHashSet<Relation> relations;
static new LinkedHashSet<Relation> maybes;
static new TreeMap<S, Pt> coordinates;
static Canvas canvas;
static JCheckBox cbMaybe;
static JTextArea ta;
static CirclesAndLines cal;

p-subst-noconsole { load('text); doIt(); }

svoid doIt {
  clearAll(nodes, relations, maybes, coordinates);
  if (ta != null) { text = getText(ta); save('text); }
  for (S s : sentences(text)) {
    new Matches m;
    if "One node *"
      nodes.add($1);
    else if "* to the right of *"
      relations.add(Relation("to the right of", $1, $2));
    else if "An arrow from * to *"
      relations.add(Relation("An arrow from * to *", $1, $2));
  }
  print("Got nodes: " + sfu(nodes));
  print("Got relations: " + sfu(relations));
  makeCoordinates();
  print("Got coordinates: " + sfu(coordinates));
  bool first = canvas == null;
  CirclesAndLines newCAL = calFromCoordinates(coordinates, relations);
  canvas = showCAL_2(cal, newCAL, canvas);
  if (cal == null) cal = newCAL;
  if (first) swing {
    increaseFrameWidth(canvas, 300);
    addToFrame(canvas, withTopMargin(jcenteredline(
      cbMaybe = onChange(jcheckbox(cbMaybe, "Strict interpretation"), f doIt), jbutton("Render again", f doIt))));
    setSplitPaneLater(addToWindowSplitRight_aggressive(canvas, ta = typeWriterTextArea(text)), 0.6f);
    onUpdate(ta, f doIt);
  }
}

svoid makeCoordinates {
  for (S node : nodes) coordinates.put(node, pt(random(100), random(100)));
  processRules();
  if (isChecked(cbMaybe)) {
    migrateList(maybes, relations);
    processRules();
  }
}

svoid processRules {
  for (Relation r : cloneList(relations)) {
    if (r.is("to the right of")) {
      maybes.add(Relation("on the same height as", r.a(), r.b()));
      Pt pa = coordinates.get(r.a()), pb = coordinates.get(r.b());
      if (pa != null && pb != null)
        pa.x = max(pa.x, pb.x+10);
    }
    if (r.is("on the same height as")) {
      Pt pa = coordinates.get(r.a()), pb = coordinates.get(r.b());
      if (pa != null && pb != null)
        pa.y = pb.y;
    }
  }
}