!7

!include #1007303 // CirclesAndLines

static CirclesAndLines cal;
static Circle a, b, c, d;
static Canvas canvas;
static JButton btnThink, btnStartOver;
volatile sbool thinking;

extend Base {
  new L<S> traits;
  bool hasTrait(S t) { ret containsIC(traits(), t); }
  L<S> traits() { if (text != null && neq(first(traits), text)) traits.add(0, text); ret traits; }
  
  S textForRender() {
    L<S> traits = traits();
    if (l(traits) <= 1) ret text;
    ret text + " [" + join(", ", dropFirst(traits)) + "]";
  }
}

p-subst {
  cal = new CirclesAndLines;
  a = cal.addCircle(#1007291, 1/3.0, 0.25, "DONALD");
  b = cal.addCircle(#1007292, 2/3.0, 0.25, "BOMBS");
  c = cal.addCircle(#1007314, 1/3.0, 0.75, "GOOD");
  d = cal.addCircle(#1007319, 2/3.0, 0.75, "BAD");
  addArrows();
  canvas = cal.showAsFrame(800, 600);
  swing {
    addToWindow(canvas, jcenteredline(
      btnThink = jbutton("Think", f action),
      btnStartOver = jbutton("Start over", f startOver)));
  }
  
  repeat {
    sleepSeconds(2);
    if (thinking) pcall {
      print("Thinking");
      bool t;
      time { t = think(); }
      if (!t) {
        thinking(false);
        print("Done");
      } else {
        print("Updating");
        canvas.update();
      }
    }
  }
}

svoid startOver {
  cal.lock.lock();
  try {
    for (Circle c : cal.circles) c.traits.clear();
    cal.lines.clear();
    addArrows();
  } finally {
    cal.lock.unlock();
  }
  canvas.update();
}

svoid addArrows {
  cal.addArrow(a, b, "MAKES");
  cal.addArrow(b, d, "ARE");
}

svoid thinking(bool b) {
  thinking = b;
  setText(btnThink, b ? "Stop thinking" : "Think");
}

svoid action {
  thinking(!thinking);
}

// for (X is Y): addTrait(X, Y);
// for (X makes bad): addLink(X is bad);
sbool think() {
  cal.lock.lock();
  try {
    bool change = false;
    for (Line l : cloneList(cal.lines)) {
      if (l.hasTrait("is") || l.hasTrait("are")) {
        int n = l(l.a.traits());
        // l.a.traits().addAll(l.b.traits()); // funny version
        setAddAll(l.a.traits(), l.b.traits());
        change |= l(l.a.traits()) > n;
      }
      if (eqic(l.text, "makes") && l.b.hasTrait("bad")) {
        change |= addLink(l.a, "IS", cal.findCircle("bad"));
      }
    }
    ret change;
  } finally {
    cal.lock.unlock();
  }
}

sbool addLink(Circle a, S link, Circle b) {
  Arrow arrow = cal.findArrow(a, b);
  if (arrow == null) {
    arrow = cal.addArrow(a, b, link);
    true;
  } else {
    if (arrow.hasTrait(link)) false;
    arrow.traits().add(link);
    true;
  }
}