!7

static double zoom = 1.5;

static volatile BufferedImage img, recognizedImage;

static L<RecognizedText> texts;

static ImageSurface is, isWord;

static volatile int score;

static volatile L<RecognizedText> posTexts, negTexts;

static Set<S> dontPrint;

sbool scoreBox;

static int coresToUse = 1;

p-subst-autorestart {

  run(#1015409);

  ocr_recognizeMultiLine_scored_maxPixels = 20000;

  hideConsoleOnFirstFrame();

  dontPrint = mechSetCI("Movie Review Cat | Negative/positive words don't print list");

  thread "The Loop" { theLoop(); }

  noMainDone("Cat!");

}

svoid theLoop {

  repeat with sleep 0 {

    temp tempSetCoresToUse(coresToUse);

    img = shootScreen2();

    long recTime = sysNow();

    assertEquals("Cores to use", coresToUse, coresToUse_fixed());

    texts = ocr_recognizeMultiLine_scored(img);

    print(filledSimpleRecognizer_cached().cacheStats());

    recTime = sysNow()-recTime;

    time "Post" {

      MultiSet<S> multiSet = asCIMultiSet(words(lines(mapMethod('text, texts))));

      Set<S> positiveWords = mechSetCI("Movie Review Cat | Positive words");

      Set<S> negativeWords = mechSetCI("Movie Review Cat | Negative words");

      final MultiSet<S> plus = multiSetSetIntersection(multiSet, positiveWords);

      final MultiSet<S> minus = multiSetSetIntersection(multiSet, negativeWords);

      score = l(plus)-l(minus);

      posTexts = [RecognizedText r : texts | containsOneOfTheWords(r.text(), setMinusSet(keys(plus), dontPrint))];

      negTexts = [RecognizedText r : texts | containsOneOfTheWords(r.text(), setMinusSet(keys(minus), dontPrint))];

      L<S> popPlus = mostPopularEntriesFromMultiMap(3, plus);

      L<S> popMinus = mostPopularEntriesFromMultiMap(3, minus);

      S text = "Score: " + score;

      if (scoreBox) {

        if (score >= 0 && nempty(popPlus)) text += "\n" + joinWithComma(popPlus);

        if (score <= 0 && nempty(popMinus)) text += "\n" + joinWithComma(popMinus);

        flatInfoBox(recTime + " ms | " + escapeNewLines(text));

      }

      S imageID = minusZeroPlus(score, #1101254, #1101256, #1101255);

      recognizedImage = img;

    }

    bool first = is == null;

    is = showImage_hcentered(is, "Movie Review Cat!", loadImage2_cached(imageID));

    if (first) {

      setDoubleBuffered(is, true);

      frameTitlePopupMenu(is, voidfunc(JPopupMenu menu) {

        menu.add(boolFieldMenuItem(mc(), 'scoreBox));

        menu.add(intFieldMenuItem(mc(), 'coresToUse));

      });

      alwaysOnTopFrame(is);

      addToWindowWithMargin(is, jfullcenter(isWord = jImageSurface()));

      enlargeFrameLeftAndBottom(is, 50);

      isWord.setZoom(2);

      awtEvery(isWord, 333, f cycleWord);

    }

  }

}

svoid cycleWord {

  if (recognizedImage == null) ret;

  RecognizedText hilite = random(score > 0 ? posTexts : score < 0 ? negTexts : null);

  if (hilite != null) {

    BufferedImage clip = clipBufferedImage(recognizedImage, hilite.r);

    isWord.setImage(clip);

    showZoomedScreenClipInPlace(hilite.r, clip, zoom, 0.5);

  } else

    isWord.setImage(newBufferedImage(1, 1, Color.white));

}