!752

concepts.

concept Example {

  BWImage image;

  S imageMD5;

  S text;

  // new fields in this program:

  L<Rect> rects;

  BWImage visualizedSplit;

  S guessedText;

  //bool countMatch;

}

static MultiMap<S, S> learnedChars;

p {

  loadConceptsFrom("#1006005");

  Concepts in = mainConcepts;

  for (Example e : in.list(Example))

    e.rects = horizontalAutoSplit2ThenAutoCrop(e.image);

  showConceptsTable_dropFields.set(ll("imageMD5"));

  final JTable table = showConceptsTable(Example);

  setFrameTitle(table, "Recognizing texts");

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

    final Example e = (Example) getConcept((long) getTableCell(table, row, 0));

    addMenuItem(menu, "Copy single character to clipboard...", r {

      final JTextField tf = jTextField("1");

      showForm("Character index (1 to " + l(e.rects) + ")",  tf, r {

        copyBWImageToClipboard(e.image.clip(e.rects.get(parseInt(trim(tf.getText()))-1)));

      });

    });

  });

  learnedChars = getLearnedChars();

  for (Example e : in.list(Example))

    guess(e);

  for (Example e : in.list(Example)) {

    // first, auto-split

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

    new StringBuilder buf;

    for (Rect r : e.rects) {

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

      S md5 = md5OfBWImage(cImg);

      L<S> l = learnedChars.get(md5);

      S c = or(first(l), "?");

      buf.append(c);

    }

    cset(e, guessedText := str(buf));

  }

}

svoid guess(Example e) {

  if (nempty(e.text))

    learnedChars.setPut(e.imageMD5, e.text);

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

  S text = e.text.replace(" ", "");

  bool countMatch = l(rects) == l(text);

  if (!countMatch) ret;

  print("Guessing " + text);

  for i over rects: {

    Rect r = rects.get(i);

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

    S md5 = md5OfBWImage(cImg);

    learnedChars.setPut(md5, substring(text, i, 1));

  }

}