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

    BWIntegralImage ii = new(f);

    int w = /*256*/100;

    IBWImage scaled = virtualScaledIntegralImageBW(ii, w);

    float similarity = 0;

    FoundImg found = null;

    ImageSurface is = showImage(iBWImageToBWImage(scaled));

    Rect r = null;

    for (BWImage pat : allEyeImages) {

      FoundImg lastFound = found;

      found = or(bwImageSearch_best_virtualBig(scaled, pat, found == null ? 0f : found.sim), found);

      if (neq(found, lastFound))

        print(found + " / " + eyeImageSources.get(pat));

      if (neq(foundImgRect(found), r)) {

        r = foundImgRect(found);

        overlaySelectionsOnImageSurface(is, ll(r));

      }

    }

    print(found);

  }

}