!752

static S imageID = "#1004061";

!include #1004250 // Eye

!include #1004067 // XRecognizableImage

static File infoFile;

static InputImage inputImage;

static ImageInfo ii;

static XRecognizableImage ri;

static ImageReader recognizer;

static volatile L<DebugItem> debugItems;

static JRecognizedText textArea;

sbool triggerRecog;

volatile sbool recognizing;

p {

  RGBImage img = new RGBImage(loadImg(imageID));

  // load imageInfo

  ii = new ImageInfo;

  infoFile = getProgramFile("imageInfo.tree");

  S iitext = loadTextFile(infoFile);

  if (nempty(iitext))

    ii.getTree().set(Tree.parse(iitext));

  ri = new XRecognizableImage;

  ri.imageInfo = ii;

  ri.setImage(img);

  ri.setDrawMarkLines(true);

  ri.setCorrections(ii.getCorrections());

  inputImage = new InputImage(img.toBW());

  textArea = new JRecognizedText;

  loadRecognizer();

  if (recognizer == null)

    recognizer = new SaR(new LineFinder, new FlexibleSegmenter, new CompareImages);

  recog(recognizer);  

  learnCorrections();

  recog(recognizer);

  JFrame frame = showFrame(centerAndSouth(vgrid(ri, textArea),

    null /*jbutton("Recognize again", r { recog(); })*/));

  swingTimer(frame, 100, r{

    if (triggerRecog && !recognizing) { triggerRecog = false; recogInBackground(); }

  });

}

svoid saveImageInfo() {

  saveTextFile(infoFile, ii.getTree().toString());

  ri.setCorrections(ii.getCorrections());

  ri.repaintImageSurface();

}

svoid recognizerModified() {

  S s = recognizer.getDescription();

  saveTextFile(getProgramFile("recognizer.tree"), s);

  print("Saved recognizer (" + l(s) + " chars)");

  triggerRecog = true;

}

svoid loadRecognizer() {

  S s = loadTextFile(getProgramFile("recognizer.tree"));

  if (nempty(s))

    recognizer = OCRUtil.makeImageReader(s);

}

svoid learnCorrections() {

  S desc = recognizer.getDescription();

  CharacterLearner characterLearner = recognizer.getCharacterLearner();

  if (characterLearner != null)

    for (Correction c : ii.getCorrections()) {

      Rectangle r = c.getRectangle();

      Point p = new Point(r.x+r.width/2, r.y+r.height/2);

      Subrecognition subrecognition = findSubrecognition(p);

      continue if subrecognition == null;

      ImageWithMarkLines imageWithMarkLines =

        new ImageWithMarkLines(subrecognition.image,

          subrecognition.topLine, subrecognition.baseLine);

      characterLearner.learnCharacter(imageWithMarkLines, c.getText());

    }

  if (neq(recognizer.getDescription(), desc))

    recognizerModified();

}

static Subrecognition findSubrecognition(Point p) {

  for (DebugItem item : debugItems)

    if (item.data instanceof Subrecognition) {

      Subrecognition s = (Subrecognition) item.data;

      Rectangle r = new Rectangle(s.clip);

      r.grow(1, 1);

      if (r.contains(p))

        return s;

    }

  null;

}

svoid recog(ImageReader rec) {

  long t = now();

  // do it!

  rec.setCollectDebugInfo(true);

  RecognizedText rt;

  time {

    rt = rec.extendedReadImage(inputImage);

  }

  debugItems = rec.getDebugInfo();

  done(t, "recog");

  recogDone(rt);

}

static ImageReader cloneRecognizer(ImageReader recognizer) {

  ret OCRUtil.makeImageReader(recognizer.getDescription());

}

svoid recogDone(final RecognizedText rt) {

  awtIfNecessary {

    print("Debug items: " + l(debugItems));

    ri.setDebugInfo(debugItems);

    S text = rt != null ? rt.text : "";

    textArea.setText(text, debugItems);

    ri.repaintImageSurface();

  }

}

svoid recogInBackground() {

  recognizing = true;

  final ImageReader rec = cloneRecognizer(recognizer);

  thread "Recognizing" {

    try {

      recog(rec);

    } finally {

      recognizing = false;

    }

  }

}