!7

cmodule MarkedImages > DynSCP {
  File dir;

  transient DynamicHStack buttons;
  transient ButtonImageLoader imageLoader;
  
  O[] menuItems() {
    ret litobjarray("Select directory...", r select);
  }
  
  start {
    if (dir == null) dir = picturesDir();
    ownResource(vmBus_onDirChange(dir, r updateButtons));
  }

  JComponent visualize2() {
    if (dir == null)
      ret jHandleFileDrop(jcenteredbutton("Select directory...", rThread select), voidfunc(File f) { setDir(f) });
    setDir(dir);
    ret getComponent();
  }
  
  void select enter {
    selectDir(voidfunc(File dir) { setDir(dir) });
  }
  
  void setDir(File dir) enter {
    setField(+dir);

    buttons = makeNiceButtons();
    imageLoader = new ButtonImageLoader(buttons) {
      void setAButtonsImage(JButton b, File f) {
        BufferedImage img = loadBufferedImage(f);
        BufferedImage preview = scaleBufferedImageToHeight(imageHeight, img);
        LPairS pairs = fileInfoFileAsPairs(f);
        fOr (PairS p : pairs) {
          O o = unstruct(p.b); // Note: unsafe
          if (o instanceof Scored) o = o/Scored!;
          Rect r = optCast Rect(o);
          if (r != null)
            drawRect(preview, rescaleRect_imageToImage(r, img, preview), Color.white);
        }
        setButtonImage(b, preview);
      }
    };
    updateButtons();
    setComponent(jHigherScrollPane(jfullcenter(buttons)));
  }
  
  void updateButtons {
    if (dir == null || buttons == null) ret;
    temp enter();
    L<File> imageFiles = sortFilesByName(listImageFiles(dir));
    L<TextImageAction> l = filesForNiceButtons(latestFilesFirst(imageFiles), f showImage);
    for (final TextImageAction tia : l)
      tia.popupMenuMaker = imageFilesPopupMenuFunction(tia);
    replaceNiceButtons(buttons, l);
    setModuleName(nImages(l) + " - " + f2s(dir));
    if (imageLoader != null) imageLoader.update();
  }
}