!759

static S theme =
  "What's the date";
  
!include #1003501 // Engine

static OccTree2 tree, botTree;
sbool walkTree_debug;
sbool useAdapter = true;

p {
  L<Dialog> dialogs = loadDialogs(theme);
  expandDialogs(dialogs);
  tree = makeOccTree(dialogs);
  botTree = makeOccTree(dialogs);
  printOccTree2(botTree);
  print("Tree size: " + tree.size() + ", bot tree size: " + botTree.size());
  //walkTree_debug = true;
  try {
    walkTree();
  } catch e {
    print("Engine not working, abort.");
    printStackTrace(e);
    ret;
  }
  print("Tree walked! " + inputFed + " -> " + outputGot);
  
  int origSize = botTree.size()-1;
  print("Engine works! Trying to shrink the tree.");
  
  while (canShrinkTree()) {
    print("Tree shrunk, trying it again.");
  }
  
  int newSize = botTree.size()-1;
  printOccTree2(botTree);
  if (newSize < origSize) {
    long percent = newSize == 0 ? 9999 : round(origSize*100.0/newSize-100);
    print("Done shrinking tree (" + origSize + " -> " + newSize + " entries - " + percent + "% shrinkage!)");
  } else
    print("Tree unshrinkable.");
}

static int inputFed, outputGot;

sbool canShrinkTree() {
  L<OccTree2> leaves = botTree.allLeaves();
  print("Leaves found: " + l(leaves));
  
  bool anyChange = false;
  for (OccTree2 leaf : leaves) {
    print("Trying to remove leaf: " + leaf.e + " (level " + leaf.level() + ")");
    Reattach reattach = removeLeaf(leaf);
    //printOccTree2(botTree);
    if (canWalkTree()) {
      print("Leaf removed successfully!");
      anyChange = true;
    } else
      reattach.reattach();
  }
  
  ret anyChange;
}

sclass Reattach {
  OccTree2 node, leaf;
  int idx;

  void reattach() {
    leaf.parent = node;
    node.next.add(idx, leaf);
  }
}

static Reattach removeLeaf(OccTree2 leaf) {
  new Reattach r;
  r.leaf = leaf;
  r.node = leaf.parent;
  r.idx = leaf.parent.next.indexOf(leaf);
  leaf.parent.next.remove(r.idx);
  leaf.parent = null;
  ret r;
}

sbool canWalkTree() {
  ret canWalkTree(tree, call(makeBot));
}

sbool canWalkTree(OccTree2 tree, O bot) {
  try { walkTree(tree, bot); true; } catch e {
    if (walkTree_debug)
      printStackTrace(e);
    ret false;
  }
}

svoid walkTree() {
  walkTree(tree, call(makeBot));
}

svoid walkTree(OccTree2 tree, O bot) {
  if (tree.isLeaf())
    assertNull(getOutput(bot));
  else
    for (E e : tree.followUpKeys()) {
      if (!contains(ll("good", "bad"), e.state)) {
        bool input = isInput(e);
        if (walkTree_debug)
          print((input ? "FEED " : "CHECK ") + structure(e));
        if (input) {
          assertNull(getOutput(bot));
          feed(bot, e);
        } else
          checkOutput(e, getOutput(bot), bot);
        walkTree(tree.followUp(e), bot);
        if (walkTree_debug)
          print("Rewind " + structure(e));
        rewind(bot);
      }
    }
}

svoid rewind(O bot) {
  call(bot, "rewind"); // rewind one step
}

svoid feed(O bot, E e) {
  call(bot, "take", e);
  ++inputFed;
}

static E getOutput(O bot) {
  E e = cast call(bot, "getSingleOutput");
  if (e != null) ++outputGot;
  ret e;
}

svoid checkOutput(E e, E output, O bot) {
  //if (e.state() && matchStart("result is", e.text())) ret;
  //assertEquals(e, output);
  if (neq(e, output)) {
    print(callOpt(bot, "thoughts"));
    fail("Bot fail: " + e + " vs " + output);
  }
}

// should match isCommand() in LooseBot
static bool isInput(E e) {
  ret e.q()
    || e.state() && matchStart("result is ", e.text());
}

Began life as a copy of #1003485