!7

static JNavigation nav;
static File dataDir;

p-subst {
  autoRestart();
  nav = jnavigation();
  frameIcon(#1101244, showMaximizedFrame(nav));
  dataDir = javaxDataDir("Screen shots for text recognition");
  
  for (final File f : sortFilesByDateDesc(listPNGFiles(dataDir)))
    recognizeFile(f, false);
    
  onNewFileInDirWithExtension(dataDir, ".png", voidfunc(File f) {
    recognizeFile(f, true)
  });
}

svoid recognizeFile(final File f, bool onTop) {
  final BufferedImage img = loadBufferedImage(f);
  print("Recognizing: " + f.getName() + " (" + bufferedImageSizeWithStar(img) + ")");
  long time = sysNow();
  final L<RecognizedText> lines = ocr_recognizeMultiLine_scored(img);
  print("  " + (sysNow()-time) + "  ms");
  F0<JComponent> makeComponent = func -> JComponent {
    final new ImageSurface is;
    is.specialPurposed = true;
    showImageWithRecognizedTexts(img, is, lines);
    
    onImageSurfaceHighlightChange(is, voidfunc(Rect r) {
      RecognizedText rec = findByField(lines, +r);
      setToolTip(is, rec == null ? null : str(rec.text));
    });
    
    imageSurfacePopupMenuWhenHighlighted_top(is, voidfunc(JPopupMenu menu, final Rect r) {
      addPopupMenuItem(menu, "OK", r {
        RecognizedText rec = findByField(lines, +r);
        appendToTextFile_uniq(changeExtension(f, ".expected"),
          struct(ll(rec.text!, rec.r)));
      });
      addSeparator(menu);
    });
    
    ret jtabs(
      "Picture", jscroll(is),
      "Raw", jtextarea(pnlStructToString(lines))
    );
  };
  
  addToNavPossiblyOnTop(onTop, nav, f.getName(), makeComponent);
}