!7

cmodule LearnDogsVsCats > DynSingleFunctionWithPrintLog {
  /*switchable*/ transient File dir = userDir("dev/dogs-vs-cats/train");
  transient int eyeWidth = /*32*/10;
  transient double eyeSurround = 0.5;
  
  transient L<File> allImageFiles;
  transient L<BWImage> allEyeImages;
  transient Map<BWImage, File> eyeImageSources;
  transient ImageSurface is, isFound;
   
  start {
    if (!dm_osBooting())
      thread { doIt(); }
  }
  
  JComponent makeButtons() {
    ret hstackWithSpacing(super.makeButtons(),
      jbutton("Random Image", rThread randomImage));
  }
  
  void randomImage {
    recognizeImage(random(allImageFiles));
  }

  void doIt {
    // load scaled down test images
    
    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)));
    
    allEyeImages = new L;
    eyeImageSources = new Map;
    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");
    
    recognizeImage(f);
  }
  
  void recognizeImage(File f) {
    print(f);
    
    BWII_MultiPatternFinder finder = new(allEyeImages, BWIntegralImage(f));
    disposePIFrame(is);
    is = null;
    
    for (FoundImg found : finder) {
      if (is == null || is.getImage().getWidth() != finder.currentScaleWidth)
        is = showImage(is, iBWImageToBWImage(finder.scaled));
      print(found + " / " + eyeImageSources.get(finder.pattern));
      overlaySelectionsOnImageSurface(is, ll(foundImgRect(found)));
      isFound = showZoomedImage(isFound, "Found", finder.pattern);
    }
  }
}