!759

static S theme =

  "Let's examine your input";

static O preprocessor = func(S in) {

  S s = findLastQuoted(in);

  ret s == null ? null : parsePoemLine(s).type();

};

//include #1003412 // LooseBot v4

!include #1003413 // LooseBot v5

!include #1003407 // OccTree2

static OccTree2 tree, botTree;

static O makeBot;

sbool walkTree_debug;

sbool useAdapter = true, useGrabber = true;

p {

  L<Dialog> dialogs = loadDialogs(theme);

  tree = goodDialogs2occTree2(dialogs);

  botTree = goodDialogs2occTree2(dialogs);

  printOccTree2(botTree);

  print("Tree size: " + tree.size() + ", bot tree size: " + botTree.size());

  makeBot = func {

    LooseBot bot = new LooseBot(botTree, func { new WordAdapter });

    bot.debug = true;

    if (useGrabber) pcall {

      bot.grabber = makeGrabber(botTree, preprocessor);

      if (bot.grabber != null)

        print("Grabber made!");

    }

    ret bot;

  };

  //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));

        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) {

  //assertEquals(e, output);

  if (neq(e, output)) {

    print("Bot fail: " + e + " vs " + output);

    fail();

  }

}

static bool isInput(E e) {

  ret e.q()

    //|| (e.state() && !matchStart("bot", e.state));

    ;

}