!752

!include #1004250 // Eye

static S imageID = "#1004248";

static JTable table;
static BufferedImage image;
static RGBImage rgbImage;
static BWImage bwImage;
static ImageSurface imageSurface;
static new L<Map> data;
static JFrame imageFrame;
static new Map<S, S> comments;

static ImageReader recognizer; // just for compiling

p {
  imageID = or2(get(args, 0), imageID);
  
  image = loadImage2(imageID);
  imageSurface = showImage(image, "The Image [" + imageID + "]");
  imageSurface.setZoom(2);
  
  /*
  ri = new XRecognizableImage;
  //ri.imageInfo = ii;
  ri.setImage(image);
  ri.imageSurface.setZoom(2);
  ri.setDrawMarkLines(true);
  ri.setDrawConfidenceBoxes(true);
  imageFrame = showFrame("The Image [" + imageID + "]", ri);
  frameIcon(imageFrame, "#1004230"); 
  moveToTopRightCorner(imageFrame);
  */

  // hints
  
  hint("There are 6 letters");
  
  // compute stuff
  
  addLine("Image Size", "Image size is " + iW() + " * " + iH() + " px");
  addLine("Flexible Segmenter", func {
    L<Segment> segments = new FlexibleSegmenter().segment(iBW());
    ret segmentsToString(segments);
  });

  rebuildTable();
  tableColumnMaxWidth(table, 0, 200);
  
  onDoubleClickOrEnter(table, voidfunc(int row) {
    Map map = getTableLineAsMap(table, row);
    if (map == null) ret;
    S text = getString(map, "Text");
    showBoxes(findBoxes(text));
  });
}

svoid addLine(S who, O gen) {
  S line;
  if (gen instanceof S)
    line = (S) gen;
  else try {
    line = str(callF(gen));
  } catch e {
    who = "[Error] " + who;
    e.printStackTrace();
    line = exceptionToStringShort(e);
  }
  data.add(litorderedmap("Who", who, "Text", line));
}

svoid hint(S text) {
  addLine("Hint", text);
}

static int iW() {
  ret iB().getWidth();
}

static int iH() {
  ret iB().getHeight();
}

static BufferedImage iB() {
  ret image;
}

static RGBImage iRGB() {
  if (rgbImage == null)
    rgbImage = new RGBImage(image);
  ret rgbImage;
}

static BWImage iBW() {
  if (bwImage == null)
    bwImage = new BWImage(image);
  ret bwImage;
}

static S segmentsToString(L<Segment> l) {
  ret join(" ", map(func(Segment s) { rectToString(s.boundingBox) }, l));
}

static S rectToString(Rectangle r) {
  ret "[" + r.x + " " + r.y + " " + r.width + " " + r.height + "]";
}

svoid showBoxes(L<Rectangle> l) {
  print("Showing boxes: " + struct(l));
  /*
  new L<DebugItem> di;
  int n = 0;
  for (Rectangle r : l) {
    ++n;
    Subrecognition sr = new Subrecognition(r, null, null, "Box " + n);
    di.add(new DebugItem("bla", sr));
  }
  */
  BufferedImage img = copyImage(image);
  drawBoxes(img, l, Color.red, 0.5f);
  imageSurface.setImage(img);
  //ri.setDebugInfo(di);
  //ri.repaintImageSurface();
}

static L<Rectangle> findBoxes(S s) {
  L<S> tok = javaTok(s);
  new L<Rectangle> boxes;
  for (int i = 1; i < l(tok); i += 2) {
    if (eq(tok.get(i), "[")) try {
      int x = parseInt(get(tok, i+2));
      int y = parseInt(get(tok, i+4));
      int w = parseInt(get(tok, i+6));
      int h = parseInt(get(tok, i+8));
      boxes.add(new Rectangle(x, y, w, h));
    } catch {}
  }
  ret boxes;
}

svoid addComment(S text, S comment) {
  comments.put(text, comment);
  for (Map m : data)
    if (eq(text, get(m, "Text")))
      m.put("Comment", comment);
  rebuildTable();
}

svoid rebuildTable() {
  table = showTable(table, data, "Image Analysis");
}