!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);

    }

  }

}