initial commit

1 files changed, 176 insertions, 0 deletions

diff --git a/gaunt01/src/testApp.cpp b/gaunt01/src/testApp.cpp
new file mode 100644
index 0000000..fdff756
--- /dev/null
+++ b/gaunt01/src/testApp.cpp
@@ -0,0 +1,176 @@
+#include "testApp.h"
+
+//--------------------------------------------------------------
+void testApp::setup(){
+
+	#ifdef _USE_LIVE_VIDEO
+        vidGrabber.setVerbose(true);
+        vidGrabber.initGrabber(320,240);
+	#else
+        vidPlayer.loadMovie("camoutput.mov");
+        vidPlayer.play();
+	#endif
+
+    colorImg.allocate(320,240);
+	grayImage.allocate(320,240);
+	grayBg.allocate(320,240);
+	grayDiff.allocate(320,240);
+
+	bLearnBakground = true;
+	threshold = 80;
+}
+
+//--------------------------------------------------------------
+void testApp::update(){
+	ofBackground(100,100,100);
+
+    bool bNewFrame = false;
+
+	#ifdef _USE_LIVE_VIDEO
+       vidGrabber.grabFrame();
+	   bNewFrame = vidGrabber.isFrameNew();
+    #else
+        vidPlayer.idleMovie();
+        bNewFrame = vidPlayer.isFrameNew();
+	#endif
+
+	if (bNewFrame){
+
+		#ifdef _USE_LIVE_VIDEO
+            colorImg.setFromPixels(vidGrabber.getPixels(), 320,240);
+	    #else
+            colorImg.setFromPixels(vidPlayer.getPixels(), 320,240);
+        #endif
+
+        grayImage = colorImg;
+		if (bLearnBakground == true){
+			grayBg = grayImage;		// the = sign copys the pixels from grayImage into grayBg (operator overloading)
+			bLearnBakground = false;
+		}
+
+		// take the abs value of the difference between background and incoming and then threshold:
+		grayDiff.absDiff(grayBg, grayImage);
+		grayDiff.threshold(threshold);
+
+		// find contours which are between the size of 20 pixels and 1/3 the w*h pixels.
+		// also, find holes is set to true so we will get interior contours as well....
+		contourFinder.findContours(grayDiff, 20, (340*240)/3, 10, true);	// find holes
+	}
+
+}
+
+//--------------------------------------------------------------
+void testApp::draw(){
+
+	// draw the incoming, the grayscale, the bg and the thresholded difference
+	ofSetHexColor(0xffffff);
+	colorImg.draw(20,20);
+	grayImage.draw(360,20);
+	grayBg.draw(20,280);
+	grayDiff.draw(360,280);
+
+	// then draw the contours:
+
+	ofFill();
+	ofSetHexColor(0x333333);
+	ofRect(360,540,320,240);
+	ofSetHexColor(0xffffff);
+
+	// we could draw the whole contour finder
+	//contourFinder.draw(360,540);
+
+	// or, instead we can draw each blob individually,
+	// this is how to get access to them:
+    for (int i = 0; i < contourFinder.nBlobs; i++){
+        contourFinder.blobs[i].draw(360,540);
+    }
+
+	// finally, a report:
+
+	ofSetHexColor(0xffffff);
+	char reportStr[1024];
+	sprintf(reportStr, "bg subtraction and blob detection\npress ' ' to capture bg\nthreshold %i (press: +/-)\nnum blobs found %i, fps: %f", threshold, contourFinder.nBlobs, ofGetFrameRate());
+	ofDrawBitmapString(reportStr, 20, 600);
+
+	float gap=ofGetHeight()-ofGetWidth();
+
+	ofTranslate(0,ofGetHeight(),0);
+	ofRotate(cam_angle,1,0,0);
+	ofTranslate(0,-ofGetHeight(),0);
+
+	for (float i=0;i<ofGetWidth();i+=ofGetWidth()/10) {
+        glBegin(GL_LINES);
+            glVertex3f(i,gap,0);
+            glVertex3f(i,ofGetHeight(),0);
+        glEnd();
+        glBegin(GL_LINES);
+            glVertex3f(0,i+gap,0);
+            glVertex3f(ofGetWidth(),i+gap,0);
+        glEnd();
+	}
+
+}
+
+//--------------------------------------------------------------
+void testApp::keyPressed(int key){
+
+	switch (key){
+		case ' ':
+			bLearnBakground = true;
+			break;
+		case '+':
+			threshold ++;
+			if (threshold > 255) threshold = 255;
+			break;
+		case '-':
+			threshold --;
+			if (threshold < 0) threshold = 0;
+			break;
+        case 'a':
+			cam_angle+=1;
+			break;
+        case 'z':
+			cam_angle-=1;
+			break;
+	}
+}
+
+//--------------------------------------------------------------
+void testApp::keyReleased(int key){
+
+}
+
+//--------------------------------------------------------------
+void testApp::mouseMoved(int x, int y ){
+
+}
+
+//--------------------------------------------------------------
+void testApp::mouseDragged(int x, int y, int button){
+
+}
+
+//--------------------------------------------------------------
+void testApp::mousePressed(int x, int y, int button){
+
+}
+
+//--------------------------------------------------------------
+void testApp::mouseReleased(int x, int y, int button){
+
+}
+
+//--------------------------------------------------------------
+void testApp::windowResized(int w, int h){
+
+}
+
+//--------------------------------------------------------------
+void testApp::gotMessage(ofMessage msg){
+
+}
+
+//--------------------------------------------------------------
+void testApp::dragEvent(ofDragInfo dragInfo){
+
+}