!7

cmodule WebCamFaceFinder > DynImageSurfaceAndEnabled {
  transient BufferedImage lastImage;
  FaceFinderByPictures1 finder;
  switchable bool overlay;
  switchable bool useFullFinder;
  switchable bool withNoFace = true;
  transient ReliableSingleThread rstUpdateImage = dm_rst(this, r updateImage);
  transient long duration = -1; // ms per recognition
  transient JLabel lblInfo;

  start-thread {
    dm_onFieldChange useFullFinder(r makeFinder);
    dm_onFieldChange withNoFace(r makeFinder);
    dm_runInQAndWait(r makeFinder);
    doEvery(100, rstUpdateImage);
  }

  void makeFinder {
    FaceFinderByPictures1 finder;
    File f = newFile(withNoFace ? ai_faceFinderDir("with-no-face") : ai_faceFinderDir(),
      useFullFinder
        ? "full-finder.struct"
        : "smallest-finder-over-95.struct");
    finder = (FaceFinderByPictures1) unstructFileOpt(f);
    if (finder != null)
      print("Made finder: " + f);
    else {
      finder = defaultFaceFinderByPictures();
      print("Using default finder");
    }
    setField(+finder);
  }
  
  afterVisualize {
    imageSurface_pixelated(imageSurface);
    containerAddFirst(buttons,
      lblInfo = jlabel(),
      dm_checkBox('overlay),
      dm_checkBox('useFullFinder),
      dm_checkBox('withNoFace));
  }
  
  void updateImage enter {
    if (imageSurface == null || !enabled) ret;
    final BufferedImage img = dm_webCamImage();
    temp tempAfterwards(r { lastImage = img });
    if (img == null || img == lastImage) ret;
    RGBImage rgb = RGBImage(img);
    long time = sysNow(); 
    FoundImgPlus<RGBImage> found = finder.getPlus(rgb);
    setField(duration := sysNow()-time);
    setText(lblInfo, duration + " ms. Score: " + formatIntPercent_99(toIntPercent(foundImgScore(found))));
    setSelection(foundImgToRect(found));
    if (overlay) {
      BufferedImage img2 = cloneBufferedImage(img);
      if (found != null)
        //fillRect(img2, found.r, Color.white);
        drawScaledPixelatedImage(img2, toBufferedImage(found.data), found.r);
      setImage(img2);
    } else
      setImage(img);
  }
}