!include once #1015415 // GrabbedImage

sclass OCRRecognizeAndTeach {

  BufferedImage img;

  ImageSurface is; // can pre-set this to reuse image surface

  L<Rect> rects; // use these if not null, otherwise auto-segment

  L<RecognizedText> texts;

  ReliableSingleThread rst = new(r { reRecognize() });

  *() {}

  *(ImageSurface *is) { go(is.getImage()); }

  *(BufferedImage img) { go(img); }

  void go(BufferedImage img, L<Rect> rects) {

    this.rects = rects;

    go(img);

  }

  void go(BufferedImage img) {

    print("Recognizing");

    this.img = img;

    texts = rec();

    is = showImageWithRecognizedTexts(img, is, texts);

    showImageWithRecognizedTexts_makeTeachable2(img, is, texts, simpleRecognizer(), rTrigger(rst));

  }

  L<RecognizedText> rec() {

    ret time("OCR", () ->

      rects != null

        ? ocr_recognizeMultiLine_scored(new BWImage(img), simpleRecognizer(), rects)

        : ocr_recognizeMultiLine_scored(img));

  }

  void reRecognize {

    print("Re-recognizing...");

    temp tempDisposeWindow(smallLoadingAnim());

    replaceCollection(texts, rec());

    is.repaint();

    print("Done");

  }

  void cleanMeUp {

    if (rst != null) rst.cancel();

    disposeWindow(is);

    is = null;

  }

}