!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;

}