// VideoTool.cpp : Defines the entry point for the console application. #include #include #include #include #include #include #include #include #include #include // TODO: reference additional headers your program requires here #include "cv.h" #include "highgui.h" bool alreadyHaveIt(int j,int imageIndices[12]); // ARGH! why is this necessary ?? #define _NO_HMC_WINSOCK_ #include "SmallSocket.h" #undef _NO_HMC_WINSOCK_ using namespace std; // Our Globals! double Hue_Min = -.1; double Hue_Max = .5; double Sat_Min = .12; double Sat_Max = .99; double Val_Min = .04; double Val_Max = .99; double Val_Dist_Max = 2; double Val_Dist_Min = .5; int R_Min = 20; // these are the two functions you'll need to write void RGBtoHSV( double r, double g, double b, double *h, double *s, double *v ); void processRed( IplImage* frame, IplImage* f_p ); string processCard( IplImage* frame, IplImage* f_p ); void findSet(string ids[]); void setGlobals(); // the code appears at the very bottom of this file char wndname[] = "VideoTool"; char wndname2[] = "Undistorted image"; char wndname3[] = "Painted image"; char *setwnd[] = {"set0","set1","set2","set3","set4","set5","set6","set7","set8","set9","set10","set11"}; // here, static refers to the fact that these variables // are local to this file (they're not available in other files, // which is safer because it allows reuse of their names) static CvSize imageSize; static CvPoint old_click_pt; static CvPoint new_click_pt; static double framenum; // global points for drawing boxes, defining loops, etc. static CvPoint pt1; static CvPoint pt2; static CvPoint pt3; static CvPoint pt4; static CvPoint pt5; static CvPoint pt6; static CvPoint pt7; // global calibration data static float* M; static float* D; // more global variables // these are simply recording all the different states you // can put the program in via the keyboard static bool bmpRecording = false; static bool capturing = true; static bool getImageFromFile = false; static bool getImageAgainFromFile = false; static bool getImage0FromVideoBMPs = false; static bool saveNextImage = false; static bool quittingTime = false; static bool detectRed = false; static bool showing_savedImage = false; static bool saveOnlyOneImageToFile = true; static bool firstTimeForSet = true; static char* filename = new char[150]; static char* filename2 = new char[150]; static int frameNumber = 0; static int keyCode = -1; // pointers to images in memory static IplImage *frame; static IplImage* undis_frame; static IplImage* undis_frame_painted; static IplImage** setimages; static CvArr* undisMap; // globals for tuning parameters by keyboard... static int blueValue = 255; // handle mouse clicks here void mouse_callback(int event, int x, int y, int flags) { if (event == CV_EVENT_LBUTTONDOWN) { // cout << "(x,y) = (" << x << "," << y << ")" << endl; // reset old_click_pt old_click_pt.x = new_click_pt.x; old_click_pt.y = new_click_pt.y; // get new click point -- note the coordinate change in y new_click_pt.x = x; // coming in from the window system new_click_pt.y = imageSize.height-y; // window system and images have different y axes } } // print pixel information void pixelInformation(IplImage* f) { if (f == NULL) return; // see if things have changed... if (new_click_pt.x != old_click_pt.x || new_click_pt.y != old_click_pt.y) // has it changed? { if (new_click_pt.x < 0 || new_click_pt.x > 319 || new_click_pt.y < 0 || new_click_pt.y > 239) return; // draw a line from old to new // cvLine(frame,old_click_pt,new_click_pt,CV_RGB(255,0,0),/*thickness*/1,/*connectivity*/8); // set the old = the new old_click_pt.x = new_click_pt.x; old_click_pt.y = new_click_pt.y; // print the RGB values and coordinates of the newly clicked point: // note that blue is #0, green is #1, and red is #2 !! /* note that this uses the undistorted frame right here... */ // this is insane!!! // see the processRed function for an explanation // of why we need "realy" int realy = new_click_pt.y; if (getImageAgainFromFile || getImageFromFile) { realy = imageSize.height - realy; } uchar* pixel = cvPtr2D(f,realy,new_click_pt.x); cout << "(x,y) = (" << new_click_pt.x << "," << new_click_pt.y << ") with (r,g,b) = (" << (int)(pixel[2]) << "," << (int)(pixel[1]) << "," << (int)(pixel[0]) << ")"; double hue,sat,val; RGBtoHSV((double)pixel[2],(double)pixel[1],(double)pixel[0],&hue,&sat,&val); cout << " and (h,s,v) = (" << hue << "," << sat << "," << val << ")" << endl; } } // the thread for the server DWORD WINAPI ThreadFunc( LPVOID lpParam ) { char recvbuf[80]; char sendbuf[80]; string t("HI"); SmallSocket SS(SmallSocket::SERVER); sprintf( recvbuf, "Server Started", *(DWORD*)lpParam ); //MessageBox( NULL, szMsg, "ThreadFunc", MB_OK ); while (true) { SS.receive(recvbuf,80); cout << "Received: " << recvbuf << endl; if (recvbuf[0] == 'q') { SS.sendout("q",2); // set a variable that tells it to quit... quittingTime = true; // cvReleaseCapture( &capture ); // cvDestroyWindow("result"); break; } if (recvbuf[0] == 'e') { // just an example of passing in more information // and using sscanf to extract it // sscanf(recvbuf,"q %lf %lf %lf ",&savedXforLog,&savedYforLog,&savedTforLog); // //the connection requires that something come back... // note that we're sending two zeros every time here... sprintf(sendbuf, "%d %d", 0, 0); SS.sendout(sendbuf,80); } // pause for a bit... Sleep(10); } return 0; } int main( int argc, char** argv ) { // in case you want to log things... //ofstream logfile("logfile_out.txt"); //ifstream loginfile("logfile_in.txt"); // set up the globals... setGlobals(); // start the server's thread // make sure that all imageprocessing is in main's thread DWORD dwThreadId, dwThrdParam = 1; HANDLE hThread; char szMsg[80]; hThread = CreateThread( NULL, // default security attributes 0, // use default stack size ThreadFunc, // thread function &dwThrdParam, // argument to thread function 0, // use default creation flags &dwThreadId); // returns the thread identifier // Check the return value for success. if (hThread == NULL) { sprintf( szMsg, "CreateThread failed." ); MessageBox( NULL, szMsg, "main", MB_OK ); } /* * set up the video windows */ for (int i=0 ; i<12 ; ++i) cvNamedWindow( setwnd[i], CV_WINDOW_AUTOSIZE ); cvNamedWindow( wndname, CV_WINDOW_AUTOSIZE ); cvNamedWindow( wndname3, CV_WINDOW_AUTOSIZE ); cvNamedWindow( wndname2, CV_WINDOW_AUTOSIZE ); /* * set up the mouse input */ //cvSetMouseCallback( wndname, mouse_callback ); cvSetMouseCallback( wndname2, mouse_callback ); /* * more camera-specific stuff */ CvCapture* capture = 0; capture = cvCaptureFromCAM(0); // this is allocating the undis_frame once (it will get written each time) // get one frame and then copy it... if (cvGrabFrame(capture)) { frame = cvRetrieveFrame( capture ); undis_frame = cvCloneImage(frame); undis_frame_painted = cvCloneImage(frame); // now let's set up the undistortion parameters // note that the docs say 3*wider, but that did not work // 3*wider _and_ 3*higher worked... undisMap = cvCreateMat( imageSize.width*3, imageSize.height*3, CV_32SC3 ); // need to release this at the end of the program... cvUnDistortInit( frame, undisMap, M, D, 1 ); } else { cout << "No camera??" << endl; capturing = false; } // main loop while (!quittingTime) { if (capturing) { if( !cvGrabFrame( capture )) break; frame = cvRetrieveFrame( capture ); if( !frame ) break; // cvUnDistortOnce( frame, undis_frame, M, D, 1 ); // might take some extra time... cvUnDistort( frame, undis_frame, undisMap, 1 ); cvReleaseImage(&undis_frame_painted); undis_frame_painted = cvCloneImage(undis_frame); } else if (getImageFromFile || getImageAgainFromFile) { if (getImageFromFile) { // ask for a frame filename int i; // should factor this path out!! if (getImage0FromVideoBMPs) { i = 0; getImage0FromVideoBMPs = false; } else { cout << "What image number (in C:\\Program Files\\ERSP\\sample_code\\driver\\VideoTool\\VideoBMPs): "; cin >> i; } sprintf(filename2,"C:\\Program Files\\ERSP\\sample_code\\driver\\VideoTool\\VideoBMPs\\image%05d.bmp",i); // we probably need to release the images returned by cvLoadImage somewhere... //frame = cvLoadImage(filename2); //undis_frame = cvCloneImage(frame); cvReleaseImage(&undis_frame); undis_frame = cvLoadImage(filename2); cvReleaseImage(&undis_frame_painted); undis_frame_painted = cvLoadImage(filename2); if (undis_frame == NULL) { cout << "Could not find that image.\n"; continue; } getImageFromFile = false; getImageAgainFromFile = true; } else if (getImageAgainFromFile) { cvReleaseImage(&undis_frame_painted); undis_frame_painted = cvCloneImage(undis_frame); } } else { ; } // click only on undis_image pixelInformation(undis_frame_painted); // save the images to a file, if desired if (bmpRecording == true && capturing == true) { sprintf(filename,"C:\\Program Files\\ERSP\\sample_code\\driver\\VideoTool\\BMPs\\image%05d.bmp",frameNumber); cout << "Captured single image (dist): " << filename << endl; cvSaveImage(filename,frame); sprintf(filename2,"C:\\Program Files\\ERSP\\sample_code\\driver\\VideoTool\\BMPsUnd\\image%05d.bmp",frameNumber); cout << "Captured single image (undist): " << filename2 << endl; cvSaveImage(filename2,undis_frame); ++frameNumber; //saveNextImage = false; //if (getImageFromFile) { // bmpRecording = false; //} if (saveOnlyOneImageToFile == true) { bmpRecording = false; } } // here we can add graphics to each frame // in order that they show up on the screen // we also decide what to display in this routine if (detectRed == true) { processCard(undis_frame,undis_frame_painted); } /* * display everything in the windows */ cvShowImage(wndname, frame); cvShowImage(wndname2, undis_frame); cvShowImage(wndname3, undis_frame_painted); //handle key presses /* * quits */ if( (keyCode = cvWaitKey( 10 )) == ((int)'q') ) { break; // quit everything } /* * we use = and - to increase and decrease blueValue */ else if(keyCode == (int)'c') processCard(undis_frame,undis_frame_painted); else if (keyCode == (int)'=') { if (blueValue < 251) blueValue += 5; } else if (keyCode == (int)'-') { if (blueValue > 4) blueValue -= 5; } else if (keyCode == (int)'1') { if(Hue_Max - .01 >= -.5) Hue_Max -= .01; } else if (keyCode == (int)'2') { if(Hue_Max + .01 <= 1.0) Hue_Max += .01; } else if (keyCode == (int)'3') { if(Sat_Max - .01 >= 0) Sat_Max -= .01; } else if (keyCode == (int)'4') { if(Sat_Max + .01 <= 1.0) Sat_Max += .01; } else if (keyCode == (int)'5') { if(Val_Max - .01 >= 0) Val_Max -= .01; } else if (keyCode == (int)'6') { if(Val_Max + .01 <= 1.0) Val_Max += .01; } else if (keyCode == (int)'!') { if(Hue_Min - .01 >= -.5) Hue_Min -= .01; } else if (keyCode == (int)'@') { if(Hue_Min + .01 <= 1.0) Hue_Min += .01; } else if (keyCode == (int)'#') { if(Sat_Min - .01 >= 0) Sat_Min -= .01; } else if (keyCode == (int)'$') { if(Sat_Min + .01 <= 1.0) Sat_Min += .01; } else if (keyCode == (int)'%') { if(Val_Min - .01 >= 0) Val_Min -= .01; } else if (keyCode == (int)'^') { if(Val_Min + .01 <= 1.0) Val_Min += .01; } else if (keyCode == (int)'0'){ if(Val_Dist_Max +.1 <=3.0) Val_Dist_Max +=.1; } else if (keyCode == (int)'9'){ if(Val_Dist_Max -.1 >=0.0) Val_Dist_Max -=.1; } else if (keyCode == (int)')'){ if(Val_Dist_Min +.1 <=1.0) Val_Dist_Min +=.1; } else if (keyCode == (int)'('){ if(Val_Dist_Min -.1 >=0.0) Val_Dist_Min -=.1; } else if (keyCode == (int)'7'){ if(R_Min - 5 >= 0) R_Min -=5; } else if (keyCode == (int)'8'){ if(R_Min + 5 <= 255) R_Min +=5; } else if (keyCode == (int)'p') { cout << "Hue_Max: " << Hue_Max << " Hue_Min: " << Hue_Min << endl; cout << "Sat_Max: " << Sat_Max << " Sat_Min: " << Sat_Min << endl; cout << "Val_Max: " << Val_Max << " Val_Min: " << Val_Min << endl; cout << "Val_Dist_Max: " << Val_Dist_Max << "Val_Dist_Min" << Val_Dist_Min << endl; cout << "R_Min: " << R_Min << endl; } /* * plays the game of Set */ else if (keyCode == (int)'S') { capturing = false; getImageFromFile = false; char filenameset[500]; // for filename... string setID[12]; /* is this the first play? -- we need to open 12 windows... */ if (firstTimeForSet == true) { srand( (unsigned)time( NULL ) ); //cout << "Setting up set!" << endl; setimages = new IplImage*[12]; // should delete this somewhere... for (int i=0 ; i<12 ; ++i) { sprintf(filenameset,"C:\\Program Files\\ERSP\\sample_code\\driver\\VideoTool\\VideoBMPs\\image%05d.bmp",i); //cvReleaseImage(&undis_frame); setimages[i] = cvLoadImage(filenameset); // load image0 12 times. //cout << "Setting up window " << i << " named " << setwnd[i] << endl; cvShowImage(setwnd[i], setimages[i]); setID[i] = processCard(cvLoadImage(filenameset),cvLoadImage(filenameset)); //cout << setID[i] << endl; } findSet(setID); firstTimeForSet = false; } else { //cout << "Reading in random images..." << endl; int imageIndices[12]; for (int i=0 ; i<12 ; ++i) { imageIndices[i] = -1; } for (int i=0 ; i<12 ; ++i) { //cout << "i is " << i << endl; while (1) { int randImage = (int)(81.0*rand()/RAND_MAX); if (randImage < 0 || randImage > 80) continue; //cout << " randImage is " << randImage << endl; bool b = alreadyHaveIt(randImage,imageIndices); if (!b) { imageIndices[i] = randImage; } else { //cout << "Have it "; //for (int i=0 ; i<12 ; ++i) // cout << imageIndices[i] << " "; //cout << endl; continue; } break; } sprintf(filenameset,"C:\\Program Files\\ERSP\\sample_code\\driver\\VideoTool\\VideoBMPs\\image%05d.bmp",imageIndices[i]); //cvReleaseImage(&(setimages[i])); setimages[i] = cvLoadImage(filenameset); // load image0 12 times. //cout << "Setting up window " << i << " named " << setwnd[i] << " with image #" << imageIndices[i] << endl; cvShowImage(setwnd[i], setimages[i]); setID[i] = processCard(cvLoadImage(filenameset),cvLoadImage(filenameset)); } findSet(setID); } } /* * records video to bitmaps (files) */ else if (keyCode == (int)'f') { if (bmpRecording) { cout << "Stopping the writing to bitmaps\n"; bmpRecording = false; } else { cout << "Starting to write to bitmaps\n"; bmpRecording = true; } showing_savedImage = false; } /* * gets image from a file # in ./VideoBMPs */ else if (keyCode == (int)'g') { cout << "Getting image from file.\n"; capturing = false; getImageFromFile = true; getImageAgainFromFile = false; bmpRecording = false; saveNextImage = false; showing_savedImage = true; } /* * toggle continuous video */ else if (keyCode == (int)'s') { if (capturing) { cout << "Stopping continuous capturing.\n"; capturing = false; getImageFromFile = false; } else { cout << "Starting continuous capturing.\n"; capturing = true; getImageFromFile = false; getImageAgainFromFile = false; /* reset undis_image */ // note this assumes that everything went well // initially with the first captured frame... !! cvReleaseImage(&undis_frame); undis_frame = cvCloneImage(frame); cvReleaseImage(&undis_frame_painted); undis_frame_painted = cvCloneImage(frame); // not to mention that these transition images have to be released! } showing_savedImage = false; } else if (keyCode == (int)'e') { // emulate server behavior saveNextImage = true; showing_savedImage = false; } /* * toggle finding red (and displaying) */ else if (keyCode == (int)'r') { detectRed = !detectRed; } /* * the spacebar advances the image in the VideoBMPs directory */ else if (keyCode == (int)' ') { if (showing_savedImage == true) { int i; sscanf(filename2,"C:\\Program Files\\ERSP\\sample_code\\driver\\VideoTool\\VideoBMPs\\image%05d.bmp",&i); sprintf(filename2,"C:\\Program Files\\ERSP\\sample_code\\driver\\VideoTool\\VideoBMPs\\image%05d.bmp",++i); cvReleaseImage(&undis_frame); undis_frame = cvLoadImage(filename2); cvReleaseImage(&undis_frame_painted); undis_frame_painted = cvLoadImage(filename2); // wrap around! if (undis_frame == NULL) { i = 0; sprintf(filename2,"C:\\Program Files\\ERSP\\sample_code\\driver\\VideoTool\\VideoBMPs\\image%05d.bmp",i); cvReleaseImage(&undis_frame); undis_frame = cvLoadImage(filename2); cvReleaseImage(&undis_frame_painted); undis_frame_painted = cvLoadImage(filename2); continue; } cout << "Getting image " << i << " from VideoBMPs.\n"; getImageAgainFromFile = true; getImageFromFile = false; } /* * initial hit of the space bar... */ else { cout << "No image from file is currently showing..." << endl; cout << "Getting image 0 from VideoBMPs.\n"; capturing = false; getImageFromFile = true; getImageAgainFromFile = false; getImage0FromVideoBMPs = true; bmpRecording = false; saveNextImage = false; showing_savedImage = true; } } } // release things before quitting cvReleaseCapture( &capture ); cvDestroyWindow("result"); } // end of main bool alreadyHaveIt(int j,int imageIndices[12]) { for (int i=0 ; i<12 ; ++i) { if (imageIndices[i] == j) return true; } return false; } // just in case you'd like to use these... inline double mymin(double a, double b) { return ab?a:b; } // read about HSV at http://en.wikipedia.org/wiki/HSV_color_space // and http://www.cs.rit.edu/~ncs/color/t_convert.html#RGB%20to%20HSV%20&%20HSV%20to%20RGB // // feel free to grab code from above or elsewhere online, if you'd like // be sure you understand what RANGE _your_ code for HSV is "returning" // in the h, s, and v pointers... void RGBtoHSV( double r, double g, double b, double *h, double *s, double *v ) { double maxRGB; double minRGB; double delta; // get max and min if(r >= g && r >= b) { maxRGB = r; if(g >= b) minRGB = b; else minRGB = g; } else if(g >= b) { maxRGB = g; if(r >= b) minRGB = b; else minRGB = r; } else { maxRGB = b; if(r >= g) minRGB = g; else minRGB = r; } delta = maxRGB - minRGB; *v = maxRGB/255.0; if (maxRGB <= 0.0) { *h = -1; *s = 0; return; } *s = delta/maxRGB; // better scaling than 255! if(delta == 0) { *h = -1; return; } if(r == maxRGB) *h = (g - b)/delta; else if (g == maxRGB) *h = (1.0/3.0) + (b-r)/delta; else *h = (2.0/3.0) + (r-g)/delta; } // here is where we seek out the red molding and indicate what we've found // the input named "frame" is the source location of our pixels // the "output" named f_p is the destination for drawing string processCard(IplImage* frame, IplImage* f_p) { // some useful local variables... int r,g,b; double hue,sat,val; // pixels are handled as uchar*'s uchar* color; // a pixel from the input, "frame" uchar* painted; // a pixel from the output, "painted" uchar* painted2; // a pixel from the output, "painted" // this draws a rectangle from pt4 to pt5 with the color // specified by the three RGB values in the CV_RGB macro // the R, G, and B channels are all 0 to 255... // see the "this is insane!" caution below in order to // use graphics on the live video stream... cvRectangle(f_p, pt4, pt5, CV_RGB(0,0,255), 1); // 1 == thickness cvCircle( f_p, pt4, 10, CV_RGB(255,255,0), /*thickness*/ 1 ); // This loops through the pixels in the rectangle specified // by pt4 and pt5 (see setGlobals for their corners) int xlow = (int)mymin(pt4.x,pt5.x); int xhi = (int)mymax(pt4.x,pt5.x); int ylow = (int)mymin(pt4.y,pt5.y); int yhi = (int)mymax(pt4.y,pt5.y); int matrix[250][250]; assert(xhi - xlow < 250 && yhi - ylow < 250); int realy; int numRed = 0; int numBlue = 0; int numGreen = 0; int numColored[250]; for(int x = xlow; x Hue_Min && sat < Sat_Max && sat > Sat_Min && val < Val_Max && val > Val_Min) // set hue, and make saturation allowable within a range of the value. if(r>g && r>b && (g==0 || b==0)) //red { painted[0] = 0; // sets blue to 0 painted[1] = 0; // sets green to 0 painted[2] = 255; // sets red to 255 matrix[y-ylow][x-xlow]=1; numRed++; numColored[x-xlow]++; } else if(b>g && b>r && (r==0 || g==0)) //blue { painted[0] = 255; // sets blue to 0 painted[1] = 0; // sets green to 0 painted[2] = 0; // sets red to 255 matrix[y-ylow][x-xlow]=1; numBlue++; numColored[x-xlow]++; } else if(g>b && g>r && (r==0 || b==0)) //green { painted[0] = 0; // sets blue to 0 painted[1] = 255; // sets green to 0 painted[2] = 0; // sets red to 255 matrix[y-ylow][x-xlow]=1; numGreen++; numColored[x-xlow]++; } else matrix[y-ylow][x-xlow]=0; } } // what color is the card? string rgb = ""; if(numRed>numGreen && numRed>numBlue) rgb = "r"; else if(numGreen>numRed && numGreen>numBlue) rgb = "g"; else if(numBlue>numRed && numBlue>numGreen) rgb = "b"; int numColsColored = 0; bool firstColored = true; bool firstUncolored = true; int seenBackground = 0; int totColored = 0; int numObjects = 0; int unColored; string shape = ""; int totPixelsCol = 0; // can we find the shape?? for (int x = xlow; x30) { shape = "o"; } numColsColored++; } firstColored = false; firstUncolored = true; } } else if(firstUncolored) { firstUncolored = false; seenBackground++; } else { unColored++; if(unColored>5) firstColored = true; } } if(totColored/numObjects>68 && shape=="") shape = "d"; else if(shape=="") shape = "s"; // cout << shape << endl << numObjects << endl /*<< totColored << endl*/ << endl; //cout << endl << "Num Objects " << numObjects << endl; // cout << "Number of Columns Colored " << numColsColored << endl; string num = "1"; if(numObjects == 2) num = "2"; else if(numObjects == 3) num = "3"; string texture = ""; int strip = 0; if(numObjects == 2) strip = numColored[(xhi+xlow)/2+xlow]; else strip = numColored[(xhi+xlow)/2 - xlow]; if(strip > 90) texture = "f"; else if((totPixelsCol/numObjects < 870 && strip <10) || totPixelsCol/numObjects < 860) texture = "s"; else texture = "o"; /* // shows offset line for strip checking static CvPoint ptLine1; static CvPoint ptLine2; ptLine1.y = pt4.y; ptLine1.x = (xhi+xlow)/2+2*xlow; ptLine2.y = pt5.y; ptLine2.x = (xhi+xlow)/2+2*xlow; cvRectangle(f_p, ptLine1, ptLine2, CV_RGB(0,255,255), 1); // 1 == thickness */ string id = shape; id += rgb; id += num; id += texture; //cout << "End processCard " << /*shape<