!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;

//static int coresToUse() { ret numberOfCores()/2; }

p-subst-autorestart {

  ocr_recognizeMultiLine_scored_maxPixels = 20000;

  hideConsoleOnFirstFrame();

  //mL_setCacheTime(60.0);

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

  thread "The Loop" { theLoop(); }

  noMainDone("Cat!");

}

svoid theLoop {

  repeat with sleep 0 {

    //print("Shooting");

    //img = shootScreenHidingConsole();

    img = shootScreen2();

    //print("Recognizing");

    long recTime = sysNow();

    texts = ocr_recognizeMultiLine_scored(img);

    print(filledSimpleRecognizer_cached().cacheStats());

    recTime = sysNow()-recTime;

    time "Post" {

      L<S> lines = mapMethod('text, texts);

      MultiSet<S> multiSet = asCIMultiSet(onlyStringsLongerThan(1, words(lines(lines))));

      //pnl(multiSetAsMap_popularFirst(multiSet));

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

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

      Set<S> set = asCISet(keys(multiSet));

      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 (score >= 0 && nempty(popPlus))

        text += "\n" + joinWithComma(popPlus);

      if (score <= 0 && nempty(popMinus))

        text += "\n" + joinWithComma(popMinus);

      //infoBox(text);

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

      S imageID = score == 0 ? #1101256

        : score > 0 ? #1101255

        : #1101254;

      recognizedImage = img;

    }

    bool first = is == null;

    is = showImage_hcentered(is, loadImage2_cached(imageID));

    if (first) {

      alwaysOnTopFrame(is);

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

      enlargeFrameLeftAndBottom(is, 50);

      isWord.setZoom(2);

      awtEvery(isWord, 333, f cycleWord);

    }

    //packFrame/*InTopRightCorner*/(is);

  }

}

svoid cycleWord {

  if (recognizedImage == null) ret;

  RecognizedText hilite = null;

  if (score > 0) hilite = random(posTexts);

  else if (score < 0) hilite = random(negTexts);

  if (hilite != null) {

    final RecognizedText _hilite = hilite;

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

    isWord.setImage(clip);

    swing {

      BufferedImage scaledClip = scaleBufferedImage(clip, zoom);

      JWindow win = makeWindow(jImageLabel(scaledClip));

      Pt p = centerOfRect(_hilite.r);

      int w = scaledClip.getWidth(), h = scaledClip.getHeight();

      win.setBounds(p.x-w/2, p.y-h/2, w, h);

      win.setAlwaysOnTop(true);

      win.setVisible(true);

      disposeWindowAfter(win, 0.5);

    }

  } else

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

}