Homec4science
Diffusion visu (master) Browse

R8913/BlobDetection.pde
74d987430babmaster

Last Change
Raw File

BlobDetection.pde
View Options

import java.util.ArrayList;
import java.util.List;
import java.util.TreeSet;
class BlobDetection {
PImage findConnectedComponents(PImage input, boolean onlyBiggest) {
PImage img = input.copy();
img.loadPixels();
int w = img.width;
int h = img.height;
int[] sets = new int[200];
int[] count = new int[200];
sets[1] = 1;
int setsNum = 0;
for (int x = 0; x < w; x++) {
img.pixels[x] = Integer.MAX_VALUE;
}
for (int y = 0; y < h; y++) {
img.pixels[y*w] = Integer.MAX_VALUE;
}
int next_set = 1;
for (int y = 1; y < h; y++) {
for (int x = 1; x < w; x++) {
if (img.pixels[y*w + x] == #FFFFFF) {
int W = img.pixels[y*w +x-1];
int nw = img.pixels[(y-1)*w +x-1];
int n = img.pixels[(y-1)*w +x];
int ne = img.pixels[(y-1)*w +x+1];
int lowest = minimum(W,nw,n,ne);
if (lowest == Integer.MAX_VALUE) {
img.pixels[y*w + x] = next_set;
//new set made
if (sets.length == setsNum + 1) {
//if not enough room realloc the array
int[] newsets = new int[sets.length*2];
int[] newcount = new int[sets.length*2];
for (int i = 0; i < sets.length; i++) {
newsets[i] = sets[i];
newcount[i] = count[i];
}
sets = newsets;
count = newcount;
}
setsNum += 1;
sets[next_set] = next_set;
count[next_set] = 1;
next_set++;
} else {
img.pixels[y*w + x] = lowest;
if (nw != lowest && nw != Integer.MAX_VALUE) sets[nw] = lowest;
if (W != lowest && W != Integer.MAX_VALUE) sets[W] = lowest;
if (n != lowest && n != Integer.MAX_VALUE) sets[n] = lowest;
if (ne != lowest && ne != Integer.MAX_VALUE) sets[ne] = lowest;
count[lowest]++; //increase the count for set with label lowest
}
} else {
img.pixels[y*w + x] = Integer.MAX_VALUE;
}
}
}
int classMaxCount = 1;
int maxcount = 0;
for (int i = 1; i < setsNum; i++) {
while (sets[sets[i]] != sets[i]) {
count[sets[i]] += count[i];
sets[i] = sets[sets[i]];
if (count[sets[i]] > maxcount) {
classMaxCount = sets[i];
maxcount = count[sets[i]];
}
}
}
for (int i = 0; i < img.pixels.length; i++) {
if (img.pixels[i] != Integer.MAX_VALUE) {
img.pixels[i] = sets[img.pixels[i]];
}
}
if (onlyBiggest) {
for (int i = 0; i<img.pixels.length; i++) {
if (img.pixels[i] == Integer.MAX_VALUE || img.pixels[i] != classMaxCount) img.pixels[i] = color(0);
else img.pixels[i] = color(152,175,114);
}
} else {
for (int i = 0; i<img.pixels.length; i++) {
if (img.pixels[i] == Integer.MAX_VALUE) img.pixels[i] = 0;
else {
int clas = img.pixels[i];
img.pixels[i] = color(clas*139%255, clas*173%255, clas*83%255);
}
}
}
img.updatePixels();
return img;
}
int minimum(int a, int b, int c, int d) {
return min(min(a,b), min(c,d));
}
}
c4science · Help