!752

concept GrabbedImage {

  long lastScreenshotTime;

  Rect lastRect;

  BWImage image;

  S md5;

  int count;

  BWImage visualizedSplit;

  new L<Rect> rects;

  new L<BWImage> characters;

  new L<S> characterMD5s;

  S recognition;

}

concept RecogInfo {

  new L<S> lines;

}

static SimpleRecognizer sr;

static Concepts in;

static JTable table;

p {

  in = new Concepts("#1005951").load();

  conceptsAndBot();

  sr = new SimpleRecognizer;

  sr.load(joinLines(uniq(RecogInfo).lines));

  update();

}

svoid update {

  try {

    if (table != null) {

      frameIcon(table, "#1005541");

      setFrameTitle(table, "Calculating - Recognitions");

    }

    new L<L> data;

    for (GrabbedImage gi : sortByFieldDesc(in.list(GrabbedImage), "lastScreenshotTime")) {

      if (gi.visualizedSplit == null) {

        L<Rect> rects = horizontalAutoSplit2ThenAutoCrop(gi.image);

        cset(gi, visualizedSplit := new BWImage(mergeImagePartsHorizontally(gi.image.toRGB(), rects)));

        for (Rect r : rects) {

          BWImage cImg = gi.image.clip(r);

          gi.rects.add(r);

          gi.characters.add(cImg);

          gi.characterMD5s.add(md5OfBWImage(cImg));

        }

      }

      S rec = sr.recognizeCheat(gi.image);

      if (rec != null)

        gi.recognition = "! " + rec;

      else

        gi.recognition = sr.recognize(gi.image);

      data.add(ll(gi.md5, gi.image.getBufferedImage(), gi.recognition, gi.visualizedSplit.getBufferedImage(), join(" ", gi.characterMD5s)));

    }

    bool first = table == null;

    table = showTableWithImages("Recognitions", splitAtSlash("MD5/Image/Recognition/Split/Character MD5s"), data, table);

    if (!first) ret;

    awt {

      table.setRowHeight(25);

      tablePopupMenu(table, voidfunc(JPopupMenu menu, final int row) {

        addMenuItem(menu, "Teach...", r {

          S md5 = cast getTableCell(table, row, 0);

          final GrabbedImage gi = findConcept(in, GrabbedImage, +md5);

          final new JTextField tf;

          final JCheckBox grouped = jCheckBox("grouped", true);

          final JButton btnSubmit = jbutton("Submit", r {

            S s = trim(tf.getText());

            bool group = isChecked(grouped);

            L<Rect> rects = gi.rects;

            if (group) {

              rects = ocr_groupRects(rects, s);

              s = ocr_unmerge(s);

            }

            L<S> characterMD5s = md5OfBWImageParts(gi.image, rects);

            S line = "The images " + quote(join(" ", characterMD5s)) + " are the " + (group ? "grouped " : "") + "characters " + quote(s) + ".";

            print(">> " + line);

            RecogInfo ri = uniq(RecogInfo);

            if (setAdd(ri.lines, line)) {

              sr.load(line);

              ri.change();

            }

            disposeFrame(tf);

          });

          onUpdateAndNow(ll(tf, grouped), r {

            btnSubmit.setEnabled(ocr_charCountDropSpaces(trim(tf.getText()), isChecked(grouped)) == l(gi.characters));

          });

          showFormTitled("Teach",

            "Image:" , new ImageSurface(gi.image.getBufferedImage()),

            "Split:" , new ImageSurface(gi.visualizedSplit.getBufferedImage()),

            "Meaning:", tf,

            "", grouped,

            "", btnSubmit);

        });

        addMenuItem(menu, "Mark on screen", r-thread {

          Rect r;

          pointArrowsToArea(r = parseRect(getTableLine(table, row).get(3)));

          print("Marking: " + r);

        });

      });

      gcAndPrintMemoryInfo();

      awtCalcOnConceptChanges(table, 1000, 1000, func update, false);

    }

  } finally {

    frameIcon(table, null);

    setFrameTitle(table, "Recognitions");

  }

}