!7

cmodule LearnDogsVsCats > DynSingleFunctionWithPrintLog {
  /*switchable*/ transient File dir = userDir("dev/dogs-vs-cats/train");
  LPair<RGBImage, Rect> scaledImages;
  
  transient int eyeWidth = /*32*/10;
  transient double eyeSurround = 0.5;
  
  start {
    if (!dm_osBooting())
      thread { doIt(); }
  }

  void doIt {
    // load scaled down test images
    
    L<File> allImageFiles = listImageFiles(dir);
    Map<File, LPair<Rect, S>> markingsMap = new Map;
    for (File f : allImageFiles)
      mapPutIfNemptyValue(markingsMap, f, loadImageFileMarkings(f));
    pnlStruct(markingsMap);
    if (empty(markingsMap)) ret with print("No marked images found in " + dir);
    
    new Map<File, BWIntegralImage> iImages;
    for (File f : keys(markingsMap))
      iImages.put(f, BWIntegralImage(f));
    print("Have " + n2(iImages, "integral image") + ", total size: " + str_toM(deepObjectSize(iImages)));
    
    new L<BWImage> allEyeImages;
    new Map<BWImage, File> eyeImageSources;
    for (File f : keys(markingsMap))
      if (fileNameContains(f, "cat"))
        for (Pair<Rect, S> mark : markingsMap.get(f))
          if (eqic(mark.b, "eye")) {
            BWImage img = bwResizeSmooth(iBWImageToBWImage(safeClipIBWImage(virtualScaledIntegralImageBW(iImages.get(f)),
              growRectProportionally(eyeSurround, mark.a)
            )), eyeWidth);
            allEyeImages.add(img);
            eyeImageSources.put(img, f);
          }
    //showImage(mergeBWImagesVertically(allEyeImages));
    
    //showImage(scaleDownUsingIntegralImageBW(random(values(iImages)), 256));
    //showImage(iBWImageToBWImage(virtualScaledIntegralImageBW(random(values(iImages)), 256)));

    //File f = random(allImageFiles);
    File f = firstFileWithName(allImageFiles, "cat.2516.jpg");
    print(f);
    
    BWII_MultiPatternFinder finder = new(allEyeImages, BWIntegralImage(f));
    
    ImageSurface is = null;
    for (FoundImg found : finder) {
      if (is == null)
        is = showImage(iBWImageToBWImage(finder.scaled));
      print(found + " / " + eyeImageSources.get(pat));
      overlaySelectionsOnImageSurface(is, ll(foundImgRect(found)));
    }
  }
}