!7

static double spaceWidth = 8;

// typical line dists: 6, 21
static float lineDist1 = 6, lineDist2 = 22;
static float lineHeight = lineDist2-lineDist1;

p-type {
  RGBImage img = rgbShootScreenHidingConsole(;
  Rect r = ocr_findSnippetSource(img;
  
  if (r == null) {
    print("No snippet source found.";
    ret;
  }
  print("Scanning: " + r;

  BWImage bwImage = new BWImage(img.clip(r;
  L<Rect> clips = autoSegment(bwImage;
  
  L<Rect> r2 = ocr_mergeOverlappingY(clips;
  //L<Rect> r2 = ocr_mergeOverlappingY_plusPercent(clips, 7;
  r2 = sortByField(r2, "y");
  
  /*for (Rect rx : r2) {
    print("h=" + rx.h + ". " + struct(bwImageRowBrightnesses(bwImage.clip(rx;
  }*/
  
  // Split rows
  for i over r2: {
    Rect rx = r2.get(i;
    BWImage cImg = bwImage.clip(rx);
    int h = rx.h/2;
    //float[] rows = bwImageRowBrightnesses(bwImage.clip(rx));
    //float b = rows[l(rows)/2];
    float b = (float) cImg.clip(0, h, rx.w, 1).averageBrightness();
    if (b > 0.93f) {
      Rect upper = new Rect(rx.x, rx.y, rx.w, h);
      Rect lower = new Rect(rx.x, rx.y+h, rx.w, rx.h-h);
      r2.set(i, lower);
      r2.add(i++, upper);
    }
  }
  
  // Merge rows
  for (int i = 0; i < l(r2)-1; i++) {
    Rect a = r2.get(i), b = r2.get(i+1);
    if (b.h < a.h/2 && (b.y-a.y2()) < 2) { // fix the dangling underscore
      r2.set(i, mergeRects(a, b));
      r2.remove(i+1);
    }
  }
  
  r2 = bwAutoCropClips(bwImage, r2);
  
  showImageWithSelections(bwImage.getBufferedImage(), r2;
  focusConsole();
  growToTop(consoleFrame());
  enableFullSearchInSimpleRecognizer(filledSimpleRecognizer_cached());
  L<S> lines = ocr_recognizeTextClips(bwImage, r2;
  
  lines = map(f ocr_unescape, lines);
  
  addIndents(lines, r2);
  addEmptyRows(lines, r2);
  
  printNumbered_flushed(lines;
}

svoid addIndents(L<S> lines, L<Rect> rects) {
  int n = l(lines);
  assertEquals(n, l(rects));
  int leftBorder = toInt(collectMinInt(rects, "x"));
  for i to n: {
    int dist = rects.get(i).x-leftBorder;
    print("dist=" + dist);
    int spaces = iround(dist/spaceWidth;
    if (spaces > 0)
      lines.set(i, rep(spaces, ' ') + lines.get(i));
  }
}

svoid addEmptyRows(L<S> lines, L<Rect> rects) {
  int n = l(lines);
  assertEquals(n, l(rects));
  for (int i = 0; i+1 < n; i++) {
    Rect a = rects.get(i), b = rects.get(i+1);
    if (a != null && b != null) {
      int dist = b.y-a.y2();
      float frows = (dist-lineDist1)/lineHeight;
      int rows = iround(frows);
      print("dist=" + dist + ", rows=" + frows);
      if (rows > 0) {
        lines.addAll(i+1, rep("", rows));
        rects.addAll(i+1, rep((Rect) null, rows));
        i += rows-1;
      }
    }
  }
}