!7

sclass DialogTree > DynModule {

  new Node root;

  transient JTree tree;

  class Node {

    LS text;

    new LinkedHashMap<LS, Node> children;

    // flag

    final static int

      funny = 1,

      correctSentence = 2,

      probablyMine = 4;

    int flags;

    S language;

    S correctIf;

    *() {}

    *(LS *text) {}

    toString {

      ret text == null ? "<ROOT>" : joinWithSpace(text);

    }

    Node getOrCreate(LS text) {

      ret pairA(getOrCreate2(text));

    }

    Pair<Node, Bool> getOrCreate2(LS text) {

      temp enter();

      Node n = children.get(text);

      bool isNew = n == null;

      if (isNew) {

        children.put(text, n = new Node(text));

        change();

      }

      ret pair(n, isNew);

    }

  }

  void start {

    addCatLogToRoot();

    onSpeechRecognized_withoutMyUtterances(voidfunc(fS s) {

      swing {

        Pair<Node, Bool> child = root.getOrCreate2(words(s));

        if (child.b) // it's a new child

          jDynamicTree_addToNodeAndScroll(tree, root, child.a);

      }

    });

  }

  JComponent visualize() {

    ret tree = jDynTree(root, func(Node n) -> L<Node> {

      temp enter();

      ret valuesList(n.children);

    });

  }

  void addCatLogToRoot {

    for (LS text : catLog_wordLists())

      root.getOrCreate(text);

  }

}