Systemy Multimedialne

Systemy Multimedialne [ Opis \| Projekt 1 \| Projekt 2 \| Projekt 3 ]

Dokumenty
Sprawozdanie - pobierz plik

Kod źródłowy
#ifdef _CH_ #pragma package <opencv> #endif #ifndef _EiC #include <cv.h> #include <highgui.h> #include <stdio.h> #include <ctype.h> #endif #include <iostream> using namespace std; //************************************************************************************************** CvCapture* capture = 0; IplImage frame=0, imageShow=0, imageSave=0; IplImage backgroundBase = 0; IplImage* background = 0; //************************************************************************************************** CvVideoWriter* outputVideo; //************************************************************************************************ CvRect selection; CvPoint origin; int select_object = 0; //********************************************************************************************** int type; int make = 0; int show = 1; int save = 0; int mask = 0; int color[3] = {0,0,0}; int epsilon = 50; int c = 'i'; //************************************************************************************************ void on_mouse( int event, int x, int y, int flags, void* param ) { //================================================================================================ if( !frame ) return; //================================================================================================ if( select_object ) { selection.x = MIN(x,origin.x); selection.y = MIN(y,origin.y); selection.width = selection.x + CV_IABS(x - origin.x); selection.height = selection.y + CV_IABS(y - origin.y); selection.x = MAX( selection.x, 0 ); selection.y = MAX( selection.y, 0 ); selection.width = MIN( selection.width, frame->width ); selection.height = MIN( selection.height, frame->height ); selection.width -= selection.x; selection.height -= selection.y; } //================================================================================================ switch( event ) { case CV_EVENT_RBUTTONDOWN: color[2] = color[1] = color[0] = 0; make = 0; break; case CV_EVENT_LBUTTONDOWN: origin = cvPoint(x,y); selection = cvRect(x,y,0,0); select_object = 1; break; case CV_EVENT_LBUTTONUP: select_object = 0; make = 1; if( selection.width > 0 && selection.height > 0 ) { color[2] = color[1] = color[0] = 0; for(int x=selection.x; x<selection.x+selection.width; x++) { for(int y=selection.y; y<selection.y+selection.height; y++) { color[2] += (int)((uchar)(frame->imageData + frame->widthStepy))[3x+2]; color[1] += (int)((uchar)(frame->imageData + frame->widthStepy))[3x+1]; color[0] += (int)((uchar)(frame->imageData + frame->widthStepy))[3x+0]; } } color[2] /= selection.widthselection.height; color[1] /= selection.widthselection.height; color[0] /= selection.widthselection.height; } break; } //================================================================================================ } //************************************************************************************************ int main( int argc, char argv ) { //================================================================================================ int x,y; //================================================================================================ if( argc == 1 \|\| (argc == 2 && strlen(argv[1]) == 1 && isdigit(argv[1][0]))) { capture = cvCaptureFromCAM( argc == 2 ? argv[1][0] - '0' : 0 ); type = ( argc == 1 ) ? 'c' : 'f'; } else if( argc == 2 ) { capture = cvCaptureFromAVI( argv[1] ); type = 'f'; } //================================================================================================ if( !capture ) { fprintf(stderr,"Could not initialize capturing...\n"); return -1; } //================================================================================================ printf("POMOC\n" "\t lewy klawisz myszy - wybranie punktu, ktorego kolor ma byc zamieniany\n" "\t prawy klawisz myszy - kasowanie powyzszego wyboru\n" "\t klawisze 1 do 9 - wybranie obrazka tla\n" "\t klawisz r - ustawienie wartosci zakresu rozpoznawania koloru (eplsilon)\n" "\t klawisz s - uruchomienie / zatrzymanie rejestrowania obrazu do pliku\n" "\t klawisz d - zatrzymanie / wznowienie wyswietlania filmu w okienku\n" "\t klawisz a - uruchomienie / zatrzymanie maski\n" "\t klawisz q lub ESC - wyjscie z programu\n" ""); //================================================================================================ for(;;) { //---------------------------------------------------------------------------------------------- switch((int)c) { //-------------------------------------------------------------------------------------------- default: c = cvWaitKey(10); break; //-------------------------------------------------------------------------------------------- case 'i': cvNamedWindow( "BlueBox", 1 ); cvSetMouseCallback( "BlueBox", on_mouse, 0 ); c = '1'; break; //-------------------------------------------------------------------------------------------- case 'r': cvDestroyWindow("BlueBox"); cout << "Podaj epsilon: "; cin >> epsilon; c = 'i'; break; //-------------------------------------------------------------------------------------------- case 'a': if( mask == 1 ) // stop { cout << "Info: wylaczenie maski" << endl; mask = 0; } else // start { cout << "Info: wlaczenie maski" << endl; mask = 1; } c = cvWaitKey(10); break; //-------------------------------------------------------------------------------------------- case 'd': if( show == 1 ) // stop { cout << "Info: wstrzyanie wyswietlania" << endl; show = 0; } else // start { cout << "Info: wznowienie wyswietlania" << endl; show = 1; } c = cvWaitKey(10); break; //-------------------------------------------------------------------------------------------- case 's': if( save == 1 ) // stop { cout << "Info: nagrywanie zakonczone" << endl; cvReleaseVideoWriter( &outputVideo ); save = 0; } else // start { cout << "Info: nagrywanie rozpoczete" << endl; //outputVideo = cvCreateVideoWriter( "video.avi", CV_FOURCC('I', '4', '2', '0'), 15.0, cvGetSize(frame), 1 ); outputVideo = cvCreateVideoWriter( "video.avi", 0, 15.0, cvGetSize(frame), 1 ); save = 1; } c = cvWaitKey(10); break; //-------------------------------------------------------------------------------------------- } //---------------------------------------------------------------------------------------------- if((int)c == '1') backgroundBase = cvLoadImage("foto/background1.jpg", -1); if((int)c == '2') backgroundBase = cvLoadImage("foto/background2.jpg", -1); if((int)c == '3') backgroundBase = cvLoadImage("foto/background3.jpg", -1); if((int)c == '4') backgroundBase = cvLoadImage("foto/background4.jpg", -1); if((int)c == '5') backgroundBase = cvLoadImage("foto/background5.jpg", -1); if((int)c == '6') backgroundBase = cvLoadImage("foto/background6.jpg", -1); if((int)c == '7') backgroundBase = cvLoadImage("foto/background7.jpg", -1); if((int)c == '8') backgroundBase = cvLoadImage("foto/background8.jpg", -1); if((int)c == '9') backgroundBase = cvLoadImage("foto/background9.jpg", -1); if((int)c == 'q' \|\| (int)c == 27) break; //---------------------------------------------------------------------------------------------- frame = cvQueryFrame( capture ); if( !frame ) break; if( !imageShow ) { imageShow = cvCreateImage( cvGetSize(frame), 8, 3 ); imageShow->origin = frame->origin; imageSave = cvCreateImage( cvGetSize(frame), 8, 3 ); imageSave->origin = frame->origin; } //---------------------------------------------------------------------------------------------- background = cvCreateImage( cvSize( imageShow->width & -2, imageShow->height & -2 ), 8, 3 ); cvResize(backgroundBase, background); //---------------------------------------------------------------------------------------------- // nie obracaj X if(type == 'f') cvCopy( frame, imageShow, 0 ); //---------------------------------------------------------------------------------------------- // obr�� X if(type == 'c') { for(x=0; x<frame->width; x++) { for(y=0; y<frame->height; y++) { ((uchar)(imageShow->imageData + imageShow->widthStepy))[3x+2] = ((uchar)(frame->imageData + frame->widthStepy))[3(frame->width-x-1)+2]; ((uchar)(imageShow->imageData + imageShow->widthStepy))[3x+1] = ((uchar)(frame->imageData + frame->widthStepy))[3(frame->width-x-1)+1]; ((uchar)(imageShow->imageData + imageShow->widthStepy))[3x+0] = ((uchar)(frame->imageData + frame->widthStepy))[3(frame->width-x-1)+0]; } } } //---------------------------------------------------------------------------------------------- // dodaj t�o if( make == 1 ) { for(x=0; x<frame->width; x++) { for(y=0; y<frame->height; y++) { if( ( abs( ((uchar)(imageShow->imageData + imageShow->widthStepy))[3x+2] - color[2] ) < epsilon) && ( abs( ((uchar)(imageShow->imageData + imageShow->widthStepy))[3x+1] - color[1] ) < epsilon) && ( abs( ((uchar)(imageShow->imageData + imageShow->widthStepy))[3x+0] - color[0] ) < epsilon) ) { ((uchar)(imageShow->imageData + imageShow->widthStepy))[3x+2] = ((uchar)(background->imageData + background->widthStepy))[3x+2]; ((uchar)(imageShow->imageData + imageShow->widthStepy))[3x+1] = ((uchar)(background->imageData + background->widthStepy))[3x+1]; ((uchar)(imageShow->imageData + imageShow->widthStepy))[3x+0] = ((uchar)(background->imageData + background->widthStepy))[3x+0]; } } } } //---------------------------------------------------------------------------------------------- // dodaj mask� if( mask == 1 ) { for(x=0; x<frame->width; x++) { for(y=0; y<frame->height; y++) { ((uchar)(imageShow->imageData + imageShow->widthStepy))[3x+2] = ( ((uchar)(background->imageData + background->widthStepy))[3x+2] + ((uchar)(imageShow->imageData + imageShow->widthStepy))[3x+2] ) / 2; ((uchar)(imageShow->imageData + imageShow->widthStepy))[3x+1] = ( ((uchar)(background->imageData + background->widthStepy))[3x+1] + ((uchar)(imageShow->imageData + imageShow->widthStepy))[3x+1] ) / 2; ((uchar)(imageShow->imageData + imageShow->widthStepy))[3x+0] = ( ((uchar)(background->imageData + background->widthStepy))[3x+0] + ((uchar)(imageShow->imageData + imageShow->widthStepy))[3x+0] ) / 2; } } } //---------------------------------------------------------------------------------------------- // obr�� Y for(x=0; x<frame->width; x++) { for(y=0; y<frame->height; y++) { ((uchar)(imageSave->imageData + imageSave->widthStepy))[3x+2] = ((uchar)(imageShow->imageData + imageShow->widthStep(frame->height-y-1)))[3x+2]; ((uchar)(imageSave->imageData + imageSave->widthStepy))[3x+1] = ((uchar)(imageShow->imageData + imageShow->widthStep(frame->height-y-1)))[3x+1]; ((uchar)(imageSave->imageData + imageSave->widthStepy))[3x+0] = ((uchar)(imageShow->imageData + imageShow->widthStep(frame->height-y-1)))[3x+0]; } } //---------------------------------------------------------------------------------------------- if( select_object && selection.width > 0 && selection.height > 0 ) { cvSetImageROI( imageShow, selection ); cvXorS( imageShow, cvScalarAll(255), imageShow, 0 ); cvResetImageROI( imageShow ); } //---------------------------------------------------------------------------------------------- if( show == 1 ) cvShowImage( "BlueBox", imageShow ); if( save == 1 ) cvWriteFrame( outputVideo, imageSave ); //---------------------------------------------------------------------------------------------- cvReleaseImage( &imageShow ); cvReleaseImage( &imageSave ); cvReleaseImage( &background ); //---------------------------------------------------------------------------------------------- } //================================================================================================ cvReleaseCapture( &capture ); cvDestroyWindow("BlueBox"); return 0; //================================================================================================ } //************************************************************************************************** #ifdef _EiC main(1,"camera.cpp"); #endif
Wyniki działania
Film źródłowy	Film wyeksportowany
Please update to Flash 9.0.115 to enjoy high quality video !	Please update to Flash 9.0.115 to enjoy high quality video !

Dokumenty

Kod źródłowy

Wyniki działania