aboutsummaryrefslogtreecommitdiff
path: root/src/pix_opencv_hist_compare.cc
diff options
context:
space:
mode:
Diffstat (limited to 'src/pix_opencv_hist_compare.cc')
-rw-r--r--src/pix_opencv_hist_compare.cc279
1 files changed, 279 insertions, 0 deletions
diff --git a/src/pix_opencv_hist_compare.cc b/src/pix_opencv_hist_compare.cc
new file mode 100644
index 0000000..b81d396
--- /dev/null
+++ b/src/pix_opencv_hist_compare.cc
@@ -0,0 +1,279 @@
+////////////////////////////////////////////////////////
+//
+// GEM - Graphics Environment for Multimedia
+//
+// zmoelnig@iem.kug.ac.at
+//
+// Implementation file
+//
+// Copyright (c) 1997-2000 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_hist_compare.h"
+
+CPPEXTERN_NEW(pix_opencv_hist_compare)
+
+/////////////////////////////////////////////////////////
+//
+// pix_opencv_hist_compare
+//
+/////////////////////////////////////////////////////////
+// Constructor
+//
+/////////////////////////////////////////////////////////
+
+pix_opencv_hist_compare :: pix_opencv_hist_compare()
+{
+ comp_xsize=320;
+ comp_ysize=240;
+
+ inlet_new(this->x_obj, &this->x_obj->ob_pd, gensym("float"), gensym("save"));
+ m_dataout = outlet_new(this->x_obj, &s_anything);
+ m_measureout = outlet_new(this->x_obj, &s_anything);
+
+ save_now = 0;
+
+ rgba = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 4);
+ rgb = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 3);
+ grey = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
+ hsv = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 3 );
+
+ h_plane = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
+ s_plane = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
+ v_plane = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
+ planes[0] = h_plane;
+ planes[1] = s_plane;
+ h_saved_plane = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
+ s_saved_plane = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
+ v_saved_plane = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
+ saved_planes[0] = h_saved_plane;
+ saved_planes[1] = s_saved_plane;
+
+ int h_bins = (int)(comp_xsize/10), s_bins = (int)(comp_ysize/10);
+ {
+ int hist_size[] = { h_bins, s_bins };
+ float h_ranges[] = { 0, 180 }; // hue is [0,180]
+ float s_ranges[] = { 0, 255 };
+ float* ranges[] = { h_ranges, s_ranges };
+ hist = cvCreateHist(
+ 2,
+ hist_size,
+ CV_HIST_ARRAY,
+ ranges,
+ 1
+ );
+ int n;
+ for (n=0; n<MAX_HISTOGRAMS_TO_COMPARE; n++) {
+ saved_hist[n] = cvCreateHist(
+ 2,
+ hist_size,
+ CV_HIST_ARRAY,
+ ranges,
+ 1
+ );
+ }
+ }
+ nbsaved=0;
+}
+
+/////////////////////////////////////////////////////////
+// Destructor
+//
+/////////////////////////////////////////////////////////
+pix_opencv_hist_compare :: ~pix_opencv_hist_compare()
+{
+ //Destroy cv_images to clean memory
+ cvReleaseImage(&rgba);
+ cvReleaseImage(&rgb);
+ cvReleaseImage(&grey);
+ cvReleaseImage(&h_plane);
+ cvReleaseImage(&s_plane);
+ cvReleaseImage(&v_plane);
+ cvReleaseImage(&h_saved_plane);
+ cvReleaseImage(&s_saved_plane);
+ cvReleaseImage(&v_saved_plane);
+
+}
+
+/////////////////////////////////////////////////////////
+// processImage
+//
+/////////////////////////////////////////////////////////
+
+void pix_opencv_hist_compare :: processImage(imageStruct &image)
+{
+ int h_bins = (int)(comp_xsize/10), s_bins = (int)(comp_ysize/10);
+
+ if ((this->comp_xsize!=image.xsize)&&(this->comp_ysize!=image.ysize)) {
+
+ this->comp_xsize=image.xsize;
+ this->comp_ysize=image.ysize;
+
+ //Destroy cv_images to clean memory
+ cvReleaseImage(&rgba);
+ cvReleaseImage(&rgb);
+ cvReleaseImage(&grey);
+ cvReleaseImage(&h_plane);
+ cvReleaseImage(&s_plane);
+ cvReleaseImage(&v_plane);
+ cvReleaseImage(&h_saved_plane);
+ cvReleaseImage(&s_saved_plane);
+ cvReleaseImage(&v_saved_plane);
+
+ //Create cv_images
+ rgba = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 4);
+ rgb = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 3);
+ grey = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
+ hsv = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 3 );
+
+
+ h_plane = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
+ s_plane = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
+ v_plane = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
+ planes[0] = h_plane;
+ planes[1] = s_plane;
+ cvCvtPixToPlane( hsv, h_plane, s_plane, v_plane, 0 );
+ h_saved_plane = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
+ s_saved_plane = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
+ v_saved_plane = cvCreateImage(cvSize(comp_xsize,comp_ysize), IPL_DEPTH_8U, 1);
+ saved_planes[0] = h_saved_plane;
+ saved_planes[1] = s_saved_plane;
+
+ h_bins = (int)(comp_xsize/10);
+ s_bins = (int)(comp_ysize/10);
+ {
+ int hist_size[] = { h_bins, s_bins };
+ float h_ranges[] = { 0, 180 }; // hue is [0,180]
+ float s_ranges[] = { 0, 255 };
+ float* ranges[] = { h_ranges, s_ranges };
+ if ( hist ) {
+ cvReleaseHist( &hist );
+ hist = NULL;
+ }
+
+ hist = cvCreateHist(
+ 2,
+ hist_size,
+ CV_HIST_ARRAY,
+ ranges,
+ 1
+ );
+ int n;
+ for (n=0; n<MAX_HISTOGRAMS_TO_COMPARE; n++) {
+ if ( saved_hist[n] ) {
+ cvReleaseHist( &saved_hist[n] );
+ saved_hist[n] = NULL;
+ }
+
+ saved_hist[n] = cvCreateHist(
+ 2,
+ hist_size,
+ CV_HIST_ARRAY,
+ ranges,
+ 1
+ );
+ }
+ }
+
+ }
+
+ // no need to copy the image, just create valid header and point to image.data...
+ IplImage* imgMat = cvCreateImageHeader( cv::Size(image.xsize, image.ysize), IPL_DEPTH_8U, image.csize);
+ imgMat->imageData = (char*) image.data;
+
+ if ( image.csize == 1 ){ // gray
+ cvCvtColor(imgMat, rgb, CV_GRAY2BGR);
+ } else if ( image.csize == 4 ) { //RGBA
+ cvCvtColor(imgMat, rgb, CV_BGRA2BGR);
+ } else {
+ error("support only RGBA or GRAY image");
+ return;
+ }
+
+ // Convert to hsv
+ cvCvtColor( rgb, hsv, CV_BGR2HSV );
+ cvCvtPixToPlane( hsv, h_plane, s_plane, v_plane, 0 );
+
+ // Build the histogram and compute its contents.
+ if (save_now>=0) {
+ post("saving histogram %d\n",save_now);
+ cvCvtPixToPlane( hsv, h_saved_plane, s_saved_plane, v_saved_plane, 0 );
+ cvCalcHist( saved_planes, saved_hist[save_now], 0, 0 ); //Compute histogram
+ cvNormalizeHist( saved_hist[save_now], 1.0 ); //Normalize it
+ save_now=-1;
+ nbsaved++;
+ }
+ cvCalcHist( planes, hist, 0, 0 ); //Compute histogram
+ cvNormalizeHist( hist, 1.0 ); //Normalize it
+
+ if ( nbsaved > 0 ) {
+ double* tato = new double[nbsaved];
+ t_atom* datalist = new t_atom[nbsaved];
+ int nearest = -1;
+ double max = 0;
+
+ for ( int n=0; n<nbsaved; n++) {
+ tato[n] = cvCompareHist(hist, saved_hist[n], CV_COMP_INTERSECT);
+ SETFLOAT(&datalist[n], tato[n]);
+ if (tato[n]>max) {
+ max = tato[n];
+ nearest = n;
+ }
+ }
+ outlet_float(m_dataout, (float)nearest);
+ outlet_list( m_measureout, 0, nbsaved , datalist );
+
+ if ( tato ) delete[] tato; tato=NULL;
+ if ( datalist ) delete[] datalist; datalist=NULL;
+ } else {
+ outlet_float(m_dataout, -1.0);
+ }
+
+ // Create an image to use to visualize our histogram.
+ int scale = 10;
+ // populate our visualization with little gray squares.
+ float max_value = 0;
+ cvGetMinMaxHistValue( hist, 0, &max_value, 0, 0 );
+
+ int h = 0;
+ int s = 0;
+
+ for( h = 0; h < h_bins; h++ ) {
+ for( s = 0; s < s_bins; s++ ) {
+ float bin_val = cvQueryHistValue_2D( hist, h, s );
+ int intensity = cvRound( bin_val * 255 / max_value );
+ cvRectangle(
+ imgMat,
+ cvPoint( h*scale, s*scale ),
+ cvPoint( (h+1)*scale - 1, (s+1)*scale - 1),
+ CV_RGB(intensity,intensity,intensity), CV_FILLED, 8 , 0 );
+ }
+ }
+}
+
+/////////////////////////////////////////////////////////
+// static member function
+//
+/////////////////////////////////////////////////////////
+
+void pix_opencv_hist_compare :: obj_setupCallback(t_class *classPtr)
+{
+ class_addmethod(classPtr, (t_method)&pix_opencv_hist_compare::saveMessCallback,
+ gensym("save"), A_FLOAT, A_NULL);
+}
+
+void pix_opencv_hist_compare :: saveMess(float index)
+{
+ if (((int)index>=0)&&((int)index<MAX_HISTOGRAMS_TO_COMPARE)) save_now = (int)index;
+}
+
+void pix_opencv_hist_compare :: saveMessCallback(void *data, t_floatarg index)
+{
+ GetMyClass(data)->saveMess(index);
+}
generated by cgit v1.2.1 (git 2.18.0) at 2024-05-04 14:35:05 +0000