!7

static volatile BufferedImage img, recognizedImage;
static L<RecognizedText> texts;
static ImageSurface is, isWord;
static volatile int score;
static volatile L<RecognizedText> posTexts, negTexts;

p-subst-autorestart {
  repeat with sleep 0 {
    //print("Shooting");
    //img = shootScreenHidingConsole();
    img = shootScreen2();
    //print("Recognizing");
    
    time {
      L<RecognizedText> texts = ocr_recognizeMultiLine_scored(img);
      L<S> lines = mapMethod('text, texts);
      MultiSet<S> multiSet = asCIMultiSet(onlyStringsLongerThan(1, words(lines(lines))));
      //pnl(multiSetAsMap_popularFirst(multiSet));
      Set<S> positiveWords = mechSetCI("Positive words");
      Set<S> negativeWords = mechSetCI("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(), keys(plus))];
      negTexts = [RecognizedText r : texts | containsOneOfTheWords(r.text(), keys(minus))];
        
      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(escapeNewLines(text));
      S imageID = score == 0 ? #1101256
        : score > 0 ? #1101255
        : #1101254;
      recognizedImage = img;

      bool first = quickShowImage_isFirst();
      ImageSurface is = quickShowImage(loadImage2(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)
    isWord.setImage(clipBufferedImage(recognizedImage, hilite.r));
  else
    isWord.setImage(newBufferedImage(1, 1, Color.white));
}