srecord noeq G22RegionThinner_v2(IImageRegion originalRegion) is Steppable {
Rect bounds;
IImageRegion region;
bool phase1done;
settable bool debug;
settable bool doFinalStep = true;
settable int lastPhaseThreshold1 = 2;
settable int lastPhaseThreshold = 5;
// 0 = outside of region
// 1 = inner pixel
// 2 = border pixel
// index is in bounds coordinates
byte[] pixels;
int idx(int x, int y) {
ret (y-bounds.y)*bounds.w+x-bounds.x;
}
int idx(Pt p) { ret idx(p.x, p.y); }
Pt idxToPt(int idx) {
ret pt(bounds.x+(idx % bounds.w), bounds.y+idx/bounds.w);
}
byte getPixel(Pt p) {
ret !containsPt(bounds, p) ? 0 : pixels[idx(p)];
}
byte getPixel(int x, int y) {
ret !containsPt(bounds, x, y) ? 0 : pixels[idx(x, y)];
}
bool contains(int x, int y) {
ret getPixel(x, y) != 0;
}
bool clearPixel(int x, int y) {
if (!region.contains(x, y)) false;
pixels[idx(x, y)] = 0;
true;
}
void init {
if (bounds != null) ret;
bounds = originalRegion.bounds();
pixels = new byte[area(bounds)];
for (Pt p : originalRegion.pixelIterator())
pixels[idx(p.x, p.y)] = 1;
region = new ThinnedRegion;
}
class ThinnedRegion is IImageRegion {
public Img image() { ret originalRegion.image(); }
public Rect bounds() { ret bounds; }
public bool contains(int x, int y) {
ret containsPt(bounds, x, y) && pixels[idx(x, y)] > 0;
}
public ItIt pixelIterator() {
ret iff_null(new IF0 {
int idx = 0;
public Pt get() {
for (; idx < pixels.length; idx++)
if (pixels[idx] > 0)
ret idxToPt(idx++);
null;
}
});
}
}
public bool step() {
init();
if (phase1done)
ret finalStep();
L traces = g22_allBorderTraces_withDiagonals(region);
for (points : traces)
for (p : points)
pixels[idx(p)] = 2;
new PtBuffer toDelete;
for (points : traces) {
int nPoints = l(points);
BitSet deletable = emptyBitSet(nPoints);
for i to nPoints: {
ping();
if (deletableBorderPoint(points, i))
deletable.set(i);
}
// handle special cases
for i to nPoints:
if (cyclicGet(deletable, nPoints, i-1)
&& !deletable.get(i)
&& cyclicGet(deletable, nPoints, i+1)) {
Pt p = points.get(i);
Pt prev = ptMinus(cyclicGet(points, i-1), p);
Pt next = ptMinus(cyclicGet(points, i+1), p);
int dir1 = onePathLookupDirection(prev);
int dir2 = onePathLookupDirection(next);
int diff = mod(dir2-dir1, 8);
if (debug) printVars("special case", +p, +prev, +next, +dir1, +dir2, +diff);
if (diff == 1 || diff == 7)
deletable.set(i);
}
for i to nPoints:
if (deletable.get(i))
toDelete.add(points.get(i));
}
for (p : toDelete)
pixels[idx(p)] = 0;
if (empty(toDelete))
set phase1done;
true;
}
bool deletableBorderPoint(PtBuffer points, int i) {
L range = cyclicSubList_incl(points, i-3, i+3);
Pt p = points.get(i);
// special case
if (eq(range.get(1), p) || eq(range.get(5), p)) false;
int surroundingBorderPixels = 0, surroundingInnerPixels = 0;
for (int dir = 1; dir <= 8; dir++) {
Pt p2 = ptPlus(p, onePathDirection(dir));
byte value = getPixel(p2);
if (value == 2 && !range.contains(p2))
surroundingBorderPixels++;
else if (value == 1)
surroundingInnerPixels++;
}
bool deletable = surroundingInnerPixels > 0 && surroundingBorderPixels == 0;
printVars ifdef G22RegionThinner_debug(+p, +surroundingInnerPixels, +surroundingBorderPixels, +deletable, +range);
ret deletable;
}
IImageRegion region aka get() { ret or(region, originalRegion); }
// go from 2 pixels wide to 1 pixel wide (TODO)
bool finalStep() {
if (!doFinalStep) false;
bool change;
int x1 = bounds.x, x2 = bounds.x2();
int y1 = bounds.y, y2 = bounds.y2();
printVars ifdef G22RegionThinner_lastPhase_debug("finalStep", +x1, +y1, +x2, +y2);
for (int x = x1; x < x2; x++)
for (int y = y1; y < y2; y++) {
// need a set pixel
if (!contains(x, y)) continue;
// check if this pixel is essential to hold the structure
// together by doing a floodfill in the 3x3 neighborhood
// (simulating the pixel being cleared already)
reMutable x; reMutable y;
var bwImage = bwImageFromFunction(3, (xx, yy) ->
(xx != 1 || yy != 1) && contains(x+xx-1, y+yy-1) ? 0 : 1);
FastRegions_BWImage regionMaker = new(bwImage);
regionMaker.withDiagonals(true);
regionMaker.run();
// get all the black regions out of the 3x3 image
var regions = regionMaker.regions();
regions = filter(regions, r -> r.brightness() == 0);
bool delete = false;
int pixels = 0;
// no regions? it's a lonely pixel - keep it
if (empty(regions)) {}
else if (l(regions) == 1) {
// one region
pixels = first(regions).numberOfPixels();
// if it's only one or two pixels, we are at the end of a line - keep
if (pixels <= lastPhaseThreshold1)
delete = false;
else {
// delete pixel if the region is small, but not too small
// (lastPhaseThreshold is 5 by default)
if (pixels < lastPhaseThreshold)
delete = true;
else if (pixels == lastPhaseThreshold) {
// if it's exactly 5 pixels, check out how far
// the region "surrounds" the pixel (does it touch 2 or 3 sides?)
// delete pixel if it is not surrounded
int sides = 4-(int)
(bwImage.getFloatPixel(1, 0)
+ bwImage.getFloatPixel(0, 1)
+ bwImage.getFloatPixel(2, 1)
+ bwImage.getFloatPixel(1, 2));
delete = sides < 3;
}
}
} else
// we have more than one region - pixel was
// structurally important, keep it
{}
printVars ifdef G22RegionThinner_lastPhase_debug(
+x, +y, bwImage := escapeNewLines(bwImageToString(bwImage)),
regions := l(regions), +pixels, +delete);
if (delete)
change |= clearPixel(x, y);
}
ret change;
}
}