!752

!include #1003283 // class C

sclass OccTree {
  int count;
  new HashMap<C, OccTree> followUp;
  
  void add(L<C> script) {
    OccTree node = this;
    ++count;
    for (C c : script)
      (node = node.followUp(c)).count++;
  }
  
  OccTree followUp(C c) {
    OccTree node = followUp.get(c);
    if (node == null)
      followUp.put(c, node = new OccTree);
    ret node;
  }
  
  C randomChoice() {
    MultiSet<C> ms = new MultiSet(false);
    for (C c : keys(followUp))
      ms.add(c, followUp.get(c).count);
    ret msOneOf(ms);
  }
}

p {
  L<C> script = poemToC("#1003260");
  new OccTree tree;
  tree.add(script);
  tree.add(poemToC("Usr: hello"));
  tree.add(poemToC("Usr: hallo"));
  //printStructure(tree);
  print(asciiHeading("Random"));
  print(unparsePoem(poemFromC(randomScript(tree))));
}

static L<C> randomScript(OccTree tree) {
  C c;
  new L<C> script;
  while ((c = tree.randomChoice()) != null) {
    script.add(c);
    tree = tree.followUp(c);
  }
  ret script;
}