// include #1005686 // PIF Classes v2

sclass CorrelateClips {

  S src1 = "#1005715", src2 = "#1005721";

  int maxTotal = 10;

  Concepts in1, in2;

  Map<Pair<Clip, Clip>, MultiSet<Pt>> correlations = new HashMap;

  L<Clip> clips1, clips2;

  *() {}

  *(S *src1, S *src2) {}

  void load() {

    in1 = new Concepts(src1).load();

    in2 = new Concepts(src2).load();

    in1.deleteAll(Found);

    in2.deleteAll(Found);

    print("Loaded from:");

    printIndent(joinLines(snippetsWithTitles(ll(src1, src2))));

    print("Total concepts: " + in1.countConcepts() + ", " + in2.countConcepts());

    clips1 = goodClips(in1);

    clips2 = goodClips(in2);

    print("Good clips: " + l(clips1) + ", " + l(clips2));

  }

  void processScreenshot(Screenshot t) {

    ImageFinder finder = new ImageFinder(t, concatLists(clips1, clips2));

    finder.maxTotal = maxTotal;

    finder.run();

    L<Found> found1 = findBackRefs(clips1, Found);

    L<Found> found2 = findBackRefs(clips2, Found);

    deleteAll(Found);

    print("#found = " + l(found1) + " / " + l(found2));

    printFound(found1);

    print();

    printFound(found2);

    print();

    for (Found a : found1)

      for (Found b : found2)

        correlate(correlations, a, b);

    for (Pair<Clip, Clip> key : correlations.keySet()) {

      print(key.a.id + " / " + key.b.id + " => " + struct(correlations.get(key)));

    }

  }

  void correlate(Map<Pair<Clip, Clip>, MultiSet<Pt>> correlations, Found a, Found b) {

    Pair key = new Pair(a.clip!, b.clip!);

    Pt p = new Pt(b.fi.r.x-a.fi.r.x, b.fi.r.y-a.fi.r.y);

    MultiSet<Pt> ms = correlations.get(key);

    if (ms == null)

      correlations.put(key, ms = new MultiSet);

    ms.add(p);

  }

  void printFound(L<Found> l) {

    for (Found f : l)

      print(f.fi.r + " " + f.clip->id);

  }

  L<Clip> goodClips(Concepts in) {

    ret [Clip c in in.list(Clip) | !swic(c.description, "bad")];

  }

}