sclass FloodFillBWImage_v2 extends FloodFillBWImage {
  Matrix<Region> matrix;
  bool verbose;
  new LinkedHashSet<Region> regions;
  
  class Region {
    new PointSetBitMatrix points;
    Rect boundingBox;
    
    void add aka addPoint(Pt p) {
      points.set(p, true);
      boundingBox = rectUnionWithPt(boundingBox, p);
    }
    
    int size() { ret points.nSetBits(); }
    
    Iterable<Pt> points() { ret points.points; }
    
    toString { ret "Region with " + nPixels(size()) + " within " + boundingBox; }
    
    IBWImage asImage() {
      ret iBWImageFromFunction((x, y) -> points.get(pt(x, y)) ? 0 : 1, w, h);
    }
  }
  
  void addedPoint(Pt p) {
    new Region r;
    r.add(p);
    matrix.put(p, r);
    print("addedPoint " + p + ": " + r);
    regions.add(r);
  }
  
  void visited(Pt p) {
    if (verbose) print("Visited point: " + p);
  }

  *(IBWImage *img) {
    super(img); 
    matrix = new ArrayMatrix(w, h);
  }
  
  bool isConnectable(Pt a, Pt b) {
    float brightnessA = img.getFloatPixel(a);
    float brightnessB = img.getFloatPixel(b);
    ret abs(brightnessA-brightnessB) <= tolerance;
  }
  
  void createBridge(Pt a, Pt b) {
    var regionA = matrix.get(a);
    var regionB = matrix.get(b);
    
    if (regionB == null) fail("regionB = null: " + b);
    
    for (p : regionB.points()) {
      matrix.set(p, regionA);
      regionA.addPoint(p);
    }
    
    regions.remove(regionB);
  }
  
  void dropOnePixelRegions {
    regions = filterSet(regions, regions.size() > 1);
  }
}