aboutsummaryrefslogtreecommitdiff
path: root/src/pix_opencv_hu_compare.cc
diff options
context:
space:
mode:
Diffstat (limited to 'src/pix_opencv_hu_compare.cc')
-rw-r--r--src/pix_opencv_hu_compare.cc468
1 files changed, 468 insertions, 0 deletions
diff --git a/src/pix_opencv_hu_compare.cc b/src/pix_opencv_hu_compare.cc
new file mode 100644
index 0000000..101c448
--- /dev/null
+++ b/src/pix_opencv_hu_compare.cc
@@ -0,0 +1,468 @@
+////////////////////////////////////////////////////////
+//
+// GEM - Graphics Environment for Multimedia
+//
+// zmoelnig@iem.kug.ac.at
+//
+// Implementation file
+//
+// Copyright (c) 1997-1998 Mark Danks.
+// Copyright (c) Günther Geiger.
+// Copyright (c) 2001-2002 IOhannes m zmoelnig. forum::für::umläute. IEM
+// Copyright (c) 2002 James Tittle & Chris Clepper
+// For information on usage and redistribution, and for a DISCLAIMER OF ALL
+// WARRANTIES, see the file, "GEM.LICENSE.TERMS" in this distribution.
+//
+/////////////////////////////////////////////////////////
+
+#include "pix_opencv_hu_compare.h"
+
+CPPEXTERN_NEW_WITH_GIMME(pix_opencv_hu_compare)
+
+/////////////////////////////////////////////////////////
+//
+// pix_opencv_hu_compare
+//
+/////////////////////////////////////////////////////////
+// Constructor
+//
+/////////////////////////////////////////////////////////
+pix_opencv_hu_compare :: pix_opencv_hu_compare(int argc, t_atom*argv)
+{
+ m_dataout = outlet_new(this->x_obj, &s_anything);
+ m_posout = outlet_new(this->x_obj, &s_anything);
+
+ comp_xsize=320;
+ comp_ysize=240;
+
+ rgba = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 4);
+ rgb = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 3);
+ gray = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
+ rgbar = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 4);
+ rgbr = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 3);
+ grayr = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
+
+ x_method = CV_CONTOURS_MATCH_I1;
+ x_storage = cvCreateMemStorage(0);
+
+ x_bcontourr = NULL;
+ x_minsize = 10*10;
+ x_cdistance = 0.1;
+}
+
+/////////////////////////////////////////////////////////
+// Destructor
+//
+/////////////////////////////////////////////////////////
+pix_opencv_hu_compare :: ~pix_opencv_hu_compare()
+{
+ cvReleaseImage(&rgba);
+ cvReleaseImage(&rgb);
+ cvReleaseImage(&gray);
+ cvReleaseImage(&rgbar);
+ cvReleaseImage(&rgbr);
+ cvReleaseImage(&grayr);
+}
+
+/////////////////////////////////////////////////////////
+// processDualImage
+//
+/////////////////////////////////////////////////////////
+void pix_opencv_hu_compare :: processRGBA_RGBA(imageStruct &left, imageStruct &right)
+{
+ double dist = 100.0, ndist;
+ int i = 0; // Indicator of cycles.
+ CvSeq *contourl=NULL, *contourlp;
+ CvRect rect;
+ CvMemStorage *mstorage;
+ CvSeq *contourr = NULL;
+ int size;
+
+ if ((left.xsize!=right.xsize) || (left.ysize!=right.ysize) )
+ {
+ post( "pix_opencv_hu_compare : left and right image are not of the same size" );
+ return;
+ }
+
+ if ((this->comp_xsize!=left.xsize)&&(this->comp_ysize!=left.ysize))
+ {
+ this->comp_xsize=left.xsize;
+ this->comp_ysize=left.ysize;
+
+ cvReleaseImage(&rgba);
+ cvReleaseImage(&rgb);
+ cvReleaseImage(&gray);
+ cvReleaseImage(&rgbar);
+ cvReleaseImage(&rgbr);
+ cvReleaseImage(&grayr);
+
+ rgba = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 4);
+ rgb = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 3);
+ gray = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
+ rgbar = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 4);
+ rgbr = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 3);
+ grayr = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
+ }
+
+ memcpy( rgbar->imageData, right.data, right.xsize*right.ysize*4 );
+ cvCvtColor(rgbar, grayr, CV_BGRA2GRAY);
+
+ // calculate the biggest contour
+ try {
+ cvFindContours( grayr, x_storage, &contourr, sizeof(CvContour), CV_RETR_LIST, CV_CHAIN_APPROX_SIMPLE, cvPoint(0,0) );
+ }
+ catch(...) {
+ post( "pix_opencv_hu_compare : error calculating contours" );
+ return;
+ }
+
+ if ( contourr )
+ {
+ size=0;
+ for( ; contourr != 0; contourr = contourr->h_next )
+ {
+ rect = cvContourBoundingRect( contourr, 1);
+ if ( rect.width*rect.height > size && rect.width*rect.height < (comp_xsize-2)*(comp_ysize-2))
+ {
+ x_bcontourr = contourr;
+ size = rect.width*rect.height;
+ }
+ }
+ }
+
+ memcpy( rgba->imageData, left.data, left.xsize*left.ysize*4 );
+ cvCvtColor(rgba, gray, CV_BGRA2GRAY);
+
+ mstorage = cvCreateMemStorage(0);
+
+ cvFindContours( gray, mstorage, &contourl, sizeof(CvContour), CV_RETR_LIST, CV_CHAIN_APPROX_SIMPLE, cvPoint(0,0) );
+
+ i=0;
+ if ( contourl && x_bcontourr )
+ {
+ contourlp=contourl;
+ for( ; contourlp != 0; contourlp = contourlp->h_next )
+ {
+ rect = cvContourBoundingRect( contourlp, 1);
+ if ( rect.width*rect.height > x_minsize && rect.width*rect.height < (comp_xsize-2)*(comp_ysize-2))
+ {
+ ndist = cvMatchShapes( x_bcontourr, contourlp, x_method, 0 );
+ if ( ndist < dist ) dist = ndist;
+ if ( ndist < x_cdistance )
+ {
+ cvRectangle( gray, cvPoint(rect.x,rect.y), cvPoint(rect.x+rect.width,rect.y+rect.height), CV_RGB(255,255,255), 2, 8 , 0 );
+ cvDrawContours( gray, contourlp, CV_RGB(255,255,255), CV_RGB(255,255,255), 0, 1, 8, cvPoint(0,0) );
+ SETFLOAT(&rlist[0], i++);
+ SETFLOAT(&rlist[1], rect.x);
+ SETFLOAT(&rlist[2], rect.y);
+ SETFLOAT(&rlist[3], rect.width);
+ SETFLOAT(&rlist[4], rect.height);
+ outlet_list( m_posout, 0, 5, rlist );
+ }
+ else
+ {
+ cvRectangle( gray, cvPoint(rect.x,rect.y), cvPoint(rect.x+rect.width,rect.y+rect.height), CV_RGB(128,128,128), 2, 8 , 0 );
+ cvDrawContours( gray, contourlp, CV_RGB(128,128,128), CV_RGB(128,128,128), 0, 1, 8, cvPoint(0,0) );
+ }
+ }
+ }
+ }
+
+ if ( dist < 100.00 ) outlet_float( m_dataout, dist );
+
+ cvReleaseMemStorage(&mstorage);
+
+ cvCvtColor(gray, rgba, CV_GRAY2BGR);
+ memcpy( left.data, rgba->imageData, left.xsize*left.ysize*4 );
+
+}
+
+/////////////////////////////////////////////////////////
+// processDualImage
+//
+/////////////////////////////////////////////////////////
+void pix_opencv_hu_compare :: processRGB_RGB(imageStruct &left, imageStruct &right)
+{
+ double dist = 100.0, ndist;
+ int i = 0; // Indicator of cycles.
+ CvSeq *contourl=NULL, *contourlp;
+ CvRect rect;
+ CvMemStorage *mstorage;
+ CvSeq *contourr = NULL;
+ int size;
+
+ if ((left.xsize!=right.xsize) || (left.ysize!=right.ysize) )
+ {
+ post( "pix_opencv_hu_compare : left and right image are not of the same size" );
+ return;
+ }
+
+ if ((this->comp_xsize!=left.xsize)&&(this->comp_ysize!=left.ysize))
+ {
+ this->comp_xsize=left.xsize;
+ this->comp_ysize=left.ysize;
+
+ cvReleaseImage(&rgba);
+ cvReleaseImage(&rgb);
+ cvReleaseImage(&gray);
+ cvReleaseImage(&rgbar);
+ cvReleaseImage(&rgbr);
+ cvReleaseImage(&grayr);
+
+ rgba = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 4);
+ rgb = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 3);
+ gray = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
+ rgbar = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 4);
+ rgbr = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 3);
+ grayr = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
+ }
+
+ memcpy( rgbr->imageData, right.data, right.xsize*right.ysize*3 );
+ cvCvtColor(rgbr, grayr, CV_BGRA2GRAY);
+
+ // calculate the biggest contour
+ try {
+ cvFindContours( grayr, x_storage, &contourr, sizeof(CvContour), CV_RETR_LIST, CV_CHAIN_APPROX_SIMPLE, cvPoint(0,0) );
+ }
+ catch(...) {
+ post( "pix_opencv_hu_compare : error calculating contours" );
+ return;
+ }
+
+ if ( contourr )
+ {
+ size=0;
+ for( ; contourr != 0; contourr = contourr->h_next )
+ {
+ rect = cvContourBoundingRect( contourr, 1);
+ if ( rect.width*rect.height > size && rect.width*rect.height < (comp_xsize-2)*(comp_ysize-2))
+ {
+ x_bcontourr = contourr;
+ size = rect.width*rect.height;
+ }
+ }
+ }
+
+ memcpy( rgb->imageData, left.data, left.xsize*left.ysize*3 );
+ cvCvtColor(rgb, gray, CV_BGRA2GRAY);
+
+ mstorage = cvCreateMemStorage(0);
+
+ cvFindContours( gray, mstorage, &contourl, sizeof(CvContour), CV_RETR_LIST, CV_CHAIN_APPROX_SIMPLE, cvPoint(0,0) );
+
+ if ( contourl && x_bcontourr )
+ {
+ contourlp=contourl;
+ for( ; contourlp != 0; contourlp = contourlp->h_next )
+ {
+ rect = cvContourBoundingRect( contourlp, 1);
+ if ( rect.width*rect.height > x_minsize && rect.width*rect.height < (comp_xsize-2)*(comp_ysize-2))
+ {
+ ndist = cvMatchShapes( x_bcontourr, contourlp, x_method, 0 );
+ if ( ndist < dist ) dist = ndist;
+ if ( ndist < x_cdistance )
+ {
+ cvRectangle( gray, cvPoint(rect.x,rect.y), cvPoint(rect.x+rect.width,rect.y+rect.height), CV_RGB(255,255,255), 2, 8 , 0 );
+ cvDrawContours( gray, contourlp, CV_RGB(255,255,255), CV_RGB(255,255,255), 0, 1, 8, cvPoint(0,0) );
+ }
+ else
+ {
+ cvRectangle( gray, cvPoint(rect.x,rect.y), cvPoint(rect.x+rect.width,rect.y+rect.height), CV_RGB(128,128,128), 2, 8 , 0 );
+ cvDrawContours( gray, contourlp, CV_RGB(128,128,128), CV_RGB(128,128,128), 0, 1, 8, cvPoint(0,0) );
+ }
+ }
+ }
+ }
+
+ if ( dist < 100.00 ) outlet_float( m_dataout, dist );
+
+ cvReleaseMemStorage(&mstorage);
+
+ cvCvtColor(gray, rgb, CV_GRAY2BGR);
+ memcpy( left.data, rgb->imageData, left.xsize*left.ysize*3 );
+
+}
+
+/////////////////////////////////////////////////////////
+// processDualImage
+//
+/////////////////////////////////////////////////////////
+void pix_opencv_hu_compare :: processGray_Gray(imageStruct &left, imageStruct &right)
+{
+ double dist = 100.0, ndist;
+ int i = 0; // Indicator of cycles.
+ CvSeq *contourl=NULL, *contourlp;
+ CvRect rect;
+ CvMemStorage *mstorage;
+ CvSeq *contourr = NULL;
+ int size;
+
+ if ((left.xsize!=right.xsize) || (left.ysize!=right.ysize) )
+ {
+ post( "pix_opencv_hu_compare : left and right image are not of the same size" );
+ return;
+ }
+
+ if ((this->comp_xsize!=left.xsize)&&(this->comp_ysize!=left.ysize))
+ {
+ this->comp_xsize=left.xsize;
+ this->comp_ysize=left.ysize;
+
+ cvReleaseImage(&rgba);
+ cvReleaseImage(&rgb);
+ cvReleaseImage(&gray);
+ cvReleaseImage(&rgbar);
+ cvReleaseImage(&rgbr);
+ cvReleaseImage(&grayr);
+
+ rgba = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 4);
+ rgb = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 3);
+ gray = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
+ rgbar = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 4);
+ rgbr = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 3);
+ grayr = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
+ }
+
+ memcpy( grayr->imageData, right.data, right.xsize*right.ysize );
+
+ // calculate the biggest contour
+ try {
+ cvFindContours( grayr, x_storage, &contourr, sizeof(CvContour), CV_RETR_LIST, CV_CHAIN_APPROX_SIMPLE, cvPoint(0,0) );
+ }
+ catch(...) {
+ post( "pix_opencv_hu_compare : error calculating contours" );
+ return;
+ }
+
+ if ( contourr )
+ {
+ size=0;
+ for( ; contourr != 0; contourr = contourr->h_next )
+ {
+ rect = cvContourBoundingRect( contourr, 1);
+ if ( rect.width*rect.height > size && rect.width*rect.height < (comp_xsize-2)*(comp_ysize-2))
+ {
+ x_bcontourr = contourr;
+ size = rect.width*rect.height;
+ }
+ }
+ }
+
+ memcpy( gray->imageData, left.data, left.xsize*left.ysize );
+
+ mstorage = cvCreateMemStorage(0);
+
+ cvFindContours( gray, mstorage, &contourl, sizeof(CvContour), CV_RETR_LIST, CV_CHAIN_APPROX_SIMPLE, cvPoint(0,0) );
+
+ if ( contourl && x_bcontourr )
+ {
+ contourlp=contourl;
+ for( ; contourlp != 0; contourlp = contourlp->h_next )
+ {
+ rect = cvContourBoundingRect( contourlp, 1);
+ if ( rect.width*rect.height > x_minsize && rect.width*rect.height < (comp_xsize-2)*(comp_ysize-2))
+ {
+ ndist = cvMatchShapes( x_bcontourr, contourlp, x_method, 0 );
+ if ( ndist < dist ) dist = ndist;
+ if ( ndist < x_cdistance )
+ {
+ cvRectangle( gray, cvPoint(rect.x,rect.y), cvPoint(rect.x+rect.width,rect.y+rect.height), CV_RGB(255,255,255), 2, 8 , 0 );
+ cvDrawContours( gray, contourlp, CV_RGB(255,255,255), CV_RGB(255,255,255), 0, 1, 8, cvPoint(0,0) );
+ }
+ else
+ {
+ cvRectangle( gray, cvPoint(rect.x,rect.y), cvPoint(rect.x+rect.width,rect.y+rect.height), CV_RGB(128,128,128), 2, 8 , 0 );
+ cvDrawContours( gray, contourlp, CV_RGB(128,128,128), CV_RGB(128,128,128), 0, 1, 8, cvPoint(0,0) );
+ }
+ }
+ }
+ }
+
+ if ( dist < 100.00 ) outlet_float( m_dataout, dist );
+
+ cvReleaseMemStorage(&mstorage);
+
+ memcpy( left.data, gray->imageData, left.xsize*left.ysize );
+
+}
+
+void pix_opencv_hu_compare :: processYUV_YUV(imageStruct &left, imageStruct &right)
+{
+ post( "pix_opencv_hu_compare : YUV colorspace not supported" );
+}
+
+/////////////////////////////////////////////////////////
+// static member function
+//
+/////////////////////////////////////////////////////////
+void pix_opencv_hu_compare :: obj_setupCallback(t_class *classPtr)
+{
+ class_addmethod(classPtr, (t_method)&pix_opencv_hu_compare::floatMethodMessCallback,
+ gensym("method"), A_FLOAT, A_NULL);
+ class_addmethod(classPtr, (t_method)&pix_opencv_hu_compare::floatMinSizeMessCallback,
+ gensym("minsize"), A_FLOAT, A_NULL);
+ class_addmethod(classPtr, (t_method)&pix_opencv_hu_compare::clearMessCallback,
+ gensym("clear"), A_NULL);
+ class_addmethod(classPtr, (t_method)&pix_opencv_hu_compare::floatCriteriaMessCallback,
+ gensym("criteria"), A_FLOAT, A_NULL);
+}
+
+void pix_opencv_hu_compare :: floatMethodMessCallback(void *data, t_floatarg method)
+{
+ GetMyClass(data)->floatMethodMess((float)method);
+}
+
+void pix_opencv_hu_compare :: floatMinSizeMessCallback(void *data, t_floatarg minsize)
+{
+ GetMyClass(data)->floatMinSizeMess((float)minsize);
+}
+
+void pix_opencv_hu_compare :: clearMessCallback(void *data)
+{
+ GetMyClass(data)->clearMess();
+}
+
+void pix_opencv_hu_compare :: floatCriteriaMessCallback(void *data, t_floatarg criteria)
+{
+ GetMyClass(data)->floatCriteriaMess((float)criteria);
+}
+
+void pix_opencv_hu_compare :: floatMethodMess(float method)
+{
+ if ((int)method==CV_CONTOURS_MATCH_I1)
+ {
+ post( "pix_opencv_hu_compare : method set to CV_CONTOURS_MATCH_I1" );
+ x_method = (int)method;
+ }
+ if ((int)method==CV_CONTOURS_MATCH_I2)
+ {
+ post( "pix_opencv_hu_compare : method set to CV_CONTOURS_MATCH_I2" );
+ x_method = (int)method;
+ }
+ if ((int)method==CV_CONTOURS_MATCH_I3)
+ {
+ post( "pix_opencv_hu_compare : method set to CV_CONTOURS_MATCH_I3" );
+ x_method = (int)method;
+ }
+}
+
+void pix_opencv_hu_compare :: floatMinSizeMess(float minsize)
+{
+ if ( (int)minsize > 0 )
+ {
+ x_minsize = (int)minsize;
+ }
+}
+
+void pix_opencv_hu_compare :: clearMess(void)
+{
+ x_bcontourr = NULL;
+}
+
+void pix_opencv_hu_compare :: floatCriteriaMess(float criteria)
+{
+ if ( criteria > 0.0 )
+ {
+ x_cdistance = criteria;
+ }
+}
generated by cgit v1.2.1 (git 2.18.0) at 2024-05-04 09:24:38 +0000