!7

!include early #1033506 // Compact Module Include Gazelle V

module GazelleScreenCam {
  !include early #1025212 // +Enabled without visualization
  
  int screenNr;
  int pixelRows = 100, colors = 8;
  
  transient ImageSurface isPosterized, isRegions;
  transient new ScreenCamStream images;
  transient new BWIntegralImageStream integralImages;
  transient new SourceTriggeredStream<BWImage> scaledAndPosterizedImages;
  transient new DoubleFPSCounter fpsCounter;
  transient int fps;
  transient ScreenSelectorRadioButtons screenSelector;
  transient new RollingAverage remainingMSPerFrame;
  transient int remainingMS;

  start {
    images.directlyFeedInto(integralImages);
    
    integralImages.onNewElement(ii ->
      scaledAndPosterizedImages.newElement(
        posterizeBWImage_withMeta(colors,
          scaledBWImageFromBWIntegralImage_withMeta(pixelRows, ii))));
          
    scaledAndPosterizedImages.onNewElement(img -> {
      fpsCounter.inc();
      setField(fps := iround(fpsCounter!));
      isPosterized?.setImage(img);
      floodFill(img);
    });
    
    // 20 Hz is enough for everyone
    dm_doEvery(1000/20, r {
      if (enabled) {
        Timestamp deadline = tsNowPlusMS(1000/20);
        images.useScreen(screenNr);
        images.step();
        long remaining = deadline.minus(tsNow());
        remainingMSPerFrame.add(remaining);
        setField(remainingMS := iround(remainingMSPerFrame!));
      }
    });
  }
  
  visualize {
    screenSelector = new ScreenSelectorRadioButtons(dm_fieldLiveValue screenNr());
    screenSelector.compactLayout(true);
    screenSelector.screenLabel("");
    isPosterized = pixelatedImageSurface().setAutoZoomToDisplay(true);
    isRegions = pixelatedImageSurface().setAutoZoomToDisplay(true);
    var vis = centerAndSouth(
      withMargin(jscroll(jtabs(
        Posterized := jscroll_centered(isPosterized),
        Regions := jscroll_centered(isRegions)))),
      withMargin(jHigherScrollPane(jfullcenter(withLeftAndRightMargin(jline(
        dm_rcheckBox enabled("Watch screen"),
        screenSelector.visualize(),
        withLabelToTheRight("colors @", dm_spinner colors(2, 256)),
        withLabelToTheRight("p", dm_spinner pixelRows(1, 500)),
        dm_fieldLabelWithLabel fps(),
        dm_fieldLabelWithLabel remainingMS("Remaining ms")
      ))))),
    );
    ret vis;
  }
  
  void floodFill(BWImage img) {
    FloodFillBWImage ff = new(img);
    ff.tolerance(0);
    ff.startAt(randomPt(img));
    stepAll(ff);
    isRegions?.setImage(bitMatrixToBufferedImage(ff.visited);
  }
  
  bool useErrorHandling() { false; }
}