// Daniel Shiffman
// http://codingtra.in
// http://patreon.com/codingtrain
// Code for: https://youtu.be/1scFcY-xMrI
import processing.video.*;
Capture video;
color trackColor;
float threshold = 25;
float distThreshold = 50;
ArrayList<Blob> blobs = new ArrayList<Blob>();
void setup() {
size(640, 480);
String[] cameras = Capture.list();
printArray(cameras);
video = new Capture(this, cameras[1]);
video.start();
trackColor = color(152, 48, 56);
}
void captureEvent(Capture video) {
video.read();
}
void keyPressed() {
if (key == 'a') {
distThreshold+=5;
} else if (key == 'z') {
distThreshold-=5;
}
if (key == 'w') {
trackColor = color(255, 0, 0);
}
if (key == 's') {
threshold+=5;
} else if (key == 'x') {
threshold-=5;
}
println(distThreshold);
}
void draw() {
video.loadPixels();
image(video, 0, 0);
blobs.clear();
// Begin loop to walk through every pixel
for (int x = 0; x < video.width; x++ ) {
for (int y = 0; y < video.height; y++ ) {
int loc = x + y * video.width;
// What is current color
color currentColor = video.pixels[loc];
float r1 = red(currentColor);
float g1 = green(currentColor);
float b1 = blue(currentColor);
float r2 = red(trackColor);
float g2 = green(trackColor);
float b2 = blue(trackColor);
float d = distSq(r1, g1, b1, r2, g2, b2);
if (d < threshold*threshold) {
boolean found = false;
for (Blob b : blobs) {
if (b.isNear(x, y)) {
b.add(x, y);
found = true;
break;
}
}
if (!found) {
Blob b = new Blob(x, y);
blobs.add(b);
}
}
}
}
for (Blob b : blobs) {
if (b.size() > 500) {
b.show();
}
}
textAlign(RIGHT);
fill(0);
text("distance threshold: " + distThreshold, width-10, 25);
text("color threshold: " + threshold, width-10, 50);
}
// Custom distance functions w/ no square root for optimization
float distSq(float x1, float y1, float x2, float y2) {
float d = (x2-x1)*(x2-x1) + (y2-y1)*(y2-y1);
return d;
}
float distSq(float x1, float y1, float z1, float x2, float y2, float z2) {
float d = (x2-x1)*(x2-x1) + (y2-y1)*(y2-y1) +(z2-z1)*(z2-z1);
return d;
}
void mousePressed() {
// Save color where the mouse is clicked in trackColor variable
int loc = mouseX + mouseY*video.width;
trackColor = video.pixels[loc];
}
In an separate tab called Blob (the Blob object)...
// Daniel Shiffman
// http://codingtra.in
// http://patreon.com/codingtrain
// Code for: https://youtu.be/1scFcY-xMrI
class Blob {
float minx;
float miny;
float maxx;
float maxy;
ArrayList<PVector> points;
Blob(float x, float y) {
minx = x;
miny = y;
maxx = x;
maxy = y;
points = new ArrayList<PVector>();
points.add(new PVector(x, y));
}
void show() {
stroke(0);
fill(255);
strokeWeight(2);
rectMode(CORNERS);
rect(minx, miny, maxx, maxy);
for (PVector v : points) {
stroke(0, 0, 255);
point(v.x, v.y);
}
}
void add(float x, float y) {
points.add(new PVector(x, y));
minx = min(minx, x);
miny = min(miny, y);
maxx = max(maxx, x);
maxy = max(maxy, y);
}
float size() {
return (maxx-minx)*(maxy-miny);
}
boolean isNear(float x, float y) {
float d = 10000000;
for (PVector v : points) {
float tempD = distSq(x, y, v.x, v.y);
if (tempD < d) {
d = tempD;
}
}
if (d < distThreshold*distThreshold) {
return true;
} else {
return false;
}
}
}
No hay comentarios:
Publicar un comentario