Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
import processing.video.*;
import processing.serial.*;
Serial myPort;
Capture video;
final int numLed = 24;
int ledNum =0;
// you must have these global varables to use the PxPGetPixelDark()
int rDark, gDark, bDark, aDark;
int rLed, gLed, bLed, aLed;
int rOrg, gOrg, bOrg, aOrg;
int rTemp, gTemp, bTemp, aTemp;
PImage ourImage;
int runNumber = 0;
int acceptableError = 3;
int[] done;
int[] numPixelsInLed;
long[] ledIntensity;
int[] ledPower;
long targetIntensity = 99999999;
void setup() {
done = new int[numLed];
numPixelsInLed = new int[numLed];
ledIntensity = new long[numLed];
ledPower = new int[numLed];
for (int i=0; i<numLed; i++) {
ledPower[i] = 255;
}
printArray(Serial.list());
String portName = Serial.list()[31];
myPort = new Serial(this, portName, 9600);
size(640, 480);
video = new Capture(this, width, height);
video.start();
noStroke();
smooth();
delay(1000); // Wait for serial port to open
}
void draw() {
if (video.available()) {
if (done[ledNum] == 0) {
clearDisplay();
delay(1000);
video.read();
image(video, 0, 0, width, height); // Draw the webcam video onto the screen
saveFrame("data/no_leds.jpg");
if (runNumber != 0) {
if ((ledIntensity[ledNum] - targetIntensity)*100/targetIntensity > acceptableError) {
ledPower[ledNum] -= pow(0.75, runNumber)*100+1;
}
if ((targetIntensity - ledIntensity[ledNum])*100/targetIntensity > acceptableError) {
ledPower[ledNum] += pow(0.75, runNumber)*100+1;
}
if (abs(targetIntensity - ledIntensity[ledNum])*100/targetIntensity <= acceptableError) {
done[ledNum] = 1;
print("Led ");
print(ledNum);
print(" done");
}
if (ledPower[ledNum] > 255) {
ledPower[ledNum] = 255;
}
if (ledPower[ledNum] < 0) {
ledPower[ledNum]= 0;
}
}
setLedPower(ledNum, ledPower[ledNum]);
delay(1000);
video.read();
image(video, 0, 0, width, height); // Draw the webcam video onto the screen
delay(10);
while (myPort.available() > 0) {
int inByte = myPort.read();
//print(char(inByte));
}
String imageName = "data/";
imageName+= str(ledNum);
imageName += "_led.jpg";
saveFrame(imageName);
String originalImageName = "data/org";
originalImageName+= str(ledNum);
originalImageName += ".jpg";
if (runNumber == 0) {
saveFrame(originalImageName);
}
PImage noLedImg = loadImage("data/no_leds.jpg");
PImage ledImg = loadImage(imageName);
PImage originalImg = loadImage(originalImageName);
noLedImg.loadPixels();
ledImg.loadPixels();
originalImg.loadPixels();
background (0);
loadPixels();
ledIntensity[ledNum] = 0;
numPixelsInLed[ledNum] = 0;
for (int x = 0; x<width; x++) {
for (int y = 0; y<height; y++) {
PxPGetPixelDark(x, y, noLedImg.pixels, width);
PxPGetPixelLed(x, y, ledImg.pixels, width);
PxPGetPixelOrg(x, y, originalImg.pixels, width);
if ((rOrg+gOrg/2+bOrg/3)-(rDark+gDark/2+bDark/3) > 75) {
ledIntensity[ledNum] = ledIntensity[ledNum] +(rLed+gLed/2+bLed/3) -(rDark+gDark/2+bDark/3);
rTemp=255;
gTemp=255;
bTemp=255;
numPixelsInLed[ledNum]++;
} else {
rTemp= 0;
gTemp=0;
bTemp=0;
}
PxPSetPixel(x, y, rTemp, gTemp, bTemp, 255, pixels, width);
}
}
ledIntensity[ledNum] /= numPixelsInLed[ledNum];
if (targetIntensity > ledIntensity[ledNum] && runNumber == 0) {
targetIntensity = ledIntensity[ledNum];
}
updatePixels();
}
print(ledNum);
print(',');
print(ledPower[ledNum]);
print(',');
println(ledIntensity[ledNum]);
ledNum++;
if (ledNum == numLed) {
int donezo = 0;
for (int i=0; i<numLed; i++) {
donezo += done[i];
}
if (donezo == numLed) {
println("DONE");
for (int i=0; i<numLed; i++) {
print(i);
print(" \t ");
println(ledPower[i]);
}
print("int level[");
print(ledNum);
print("] = {");
for (int i=0; i<numLed-1; i++) {
print(ledPower[i]);
print(',');
}
print(ledPower[numLed - 1]);
println("};");
lightUpEven();
while (true);
}
print("Target intensity: ");
if (runNumber == 0) {
targetIntensity -= 1;
}
println(targetIntensity);
ledNum = 0;
runNumber++;
}
}
}
void PxPGetPixelOrg(int x, int y, int[] pixelArray, int pixelsWidth) {
int thisPixel=pixelArray[x+y*pixelsWidth]; // getting the colors as an int from the pixels[]
aOrg = (thisPixel >> 24) & 0xFF; // we need to shift and mask to get each component alone
rOrg = (thisPixel >> 16) & 0xFF; // this is faster than calling red(), green() , blue()
gOrg = (thisPixel >> 8) & 0xFF;
bOrg = thisPixel & 0xFF;
}
void PxPGetPixelDark(int x, int y, int[] pixelArray, int pixelsWidth) {
int thisPixel=pixelArray[x+y*pixelsWidth]; // getting the colors as an int from the pixels[]
aDark = (thisPixel >> 24) & 0xFF; // we need to shift and mask to get each component alone
rDark = (thisPixel >> 16) & 0xFF; // this is faster than calling red(), green() , blue()
gDark = (thisPixel >> 8) & 0xFF;
bDark = thisPixel & 0xFF;
}
void PxPGetPixelLed(int x, int y, int[] pixelArray, int pixelsWidth) {
int thisPixel=pixelArray[x+y*pixelsWidth]; // getting the colors as an int from the pixels[]
aLed = (thisPixel >> 24) & 0xFF; // we need to shift and mask to get each component alone
rLed = (thisPixel >> 16) & 0xFF; // this is faster than calling red(), green() , blue()
gLed = (thisPixel >> 8) & 0xFF;
bLed = thisPixel & 0xFF;
}
void PxPSetPixel(int x, int y, int r, int g, int b, int a, int[] pixelArray, int pixelsWidth) {
a =(a << 24);
r = r << 16; // We are packing all 4 composents into one int
g = g << 8; // so we need to shift them to their places
color argb = a | r | g | b; // binary "or" operation adds them all into one int
pixelArray[x+y*pixelsWidth]= argb; // finaly we set the int with te colors into the pixels[]
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.