From 7b555924a46e83b194e5fab92fc6d6e564d6a5ab Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?llu=C3=ADs=20g=C3=B3mez=20i=20bigord=C3=A0?= Date: Sun, 25 May 2008 21:51:21 +0000 Subject: svn path=/trunk/externals/pix_opencv/; revision=9913 --- pix_opencv_threshold.cpp | 266 +++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 266 insertions(+) create mode 100644 pix_opencv_threshold.cpp (limited to 'pix_opencv_threshold.cpp') diff --git a/pix_opencv_threshold.cpp b/pix_opencv_threshold.cpp new file mode 100644 index 0000000..b102c0d --- /dev/null +++ b/pix_opencv_threshold.cpp @@ -0,0 +1,266 @@ +//////////////////////////////////////////////////////// +// +// 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_threshold.h" + +CPPEXTERN_NEW(pix_opencv_threshold) + +///////////////////////////////////////////////////////// +// +// pix_opencv_threshold +// +///////////////////////////////////////////////////////// +// Constructor +// +///////////////////////////////////////////////////////// +pix_opencv_threshold :: pix_opencv_threshold() +{ + inlet_new(this->x_obj, &this->x_obj->ob_pd, gensym("float"), gensym("ft1")); + inlet_new(this->x_obj, &this->x_obj->ob_pd, gensym("float"), gensym("ft2")); + threshold_value = 1; + max_value = 255; + threshold_mode = 0; + comp_xsize = 0; + comp_ysize = 0; + orig = NULL; + gray = NULL; +} + +///////////////////////////////////////////////////////// +// Destructor +// +///////////////////////////////////////////////////////// +pix_opencv_threshold :: ~pix_opencv_threshold() +{ + //Destroy cv_images to clean memory + cvReleaseImage(&orig); + cvReleaseImage(&gray); + cvReleaseImage(&edge); + cvReleaseImage(&cedge); + cvReleaseImage(&cedgergb); + cvReleaseImage(&rgb); +} + +///////////////////////////////////////////////////////// +// processImage +// +///////////////////////////////////////////////////////// +void pix_opencv_threshold :: processRGBAImage(imageStruct &image) +{ + unsigned char *pixels = image.data; + + if ((this->comp_xsize!=image.xsize)||(this->comp_ysize!=image.ysize)||(!orig)) { + + this->comp_xsize = image.xsize; + this->comp_ysize = image.ysize; + + //Destroy cv_images to clean memory + cvReleaseImage(&orig); + cvReleaseImage(&gray); + cvReleaseImage(&edge); + cvReleaseImage(&cedge); + cvReleaseImage(&cedgergb); + cvReleaseImage(&rgb); + + //create the orig image with new size + orig = cvCreateImage(cvSize(image.xsize,image.ysize), IPL_DEPTH_8U, 4); + + // Create the output images with new sizes + gray = cvCreateImage(cvSize(orig->width,orig->height), IPL_DEPTH_8U, 1); + } + // Here we make a copy of the pixel data from image to orig->imageData + // orig is a IplImage struct, the default image type in openCV, take a look on the IplImage data structure here + // http://www.cs.iit.edu/~agam/cs512/lect-notes/opencv-intro/opencv-intro.html + memcpy( orig->imageData, image.data, image.xsize*image.ysize*4 ); + + // Convert to grayscale + cvCvtColor(orig, gray, CV_BGRA2GRAY); + + // Applies fixed-level thresholding to single-channel array. + switch(this->threshold_mode) { + case 0: + cvThreshold(gray, gray, (float)this->threshold_value, (float)this->max_value, CV_THRESH_BINARY); + break; + case 1: + cvThreshold(gray, gray, (float)this->threshold_value, (float)this->max_value, CV_THRESH_BINARY_INV); + break; + case 2: + cvThreshold(gray, gray, (float)this->threshold_value, (float)this->max_value, CV_THRESH_TRUNC); + break; + case 3: + cvThreshold(gray, gray, (float)this->threshold_value, (float)this->max_value, CV_THRESH_TOZERO); + break; + case 4: + cvThreshold(gray, gray, (float)this->threshold_value, (float)this->max_value, CV_THRESH_TOZERO_INV); + break; + } + + cvCvtColor(gray, orig, CV_GRAY2BGRA); + + //copy back the processed frame to image + memcpy( image.data, orig->imageData, image.xsize*image.ysize*4 ); +} + +void pix_opencv_threshold :: processRGBImage(imageStruct &image) +{ + unsigned char *pixels = image.data; + + if ((this->comp_xsize!=image.xsize)||(this->comp_ysize!=image.ysize)||(!rgb)) { + + this->comp_xsize = image.xsize; + this->comp_ysize = image.ysize; + + //Destroy cv_images to clean memory + cvReleaseImage(&orig); + cvReleaseImage(&gray); + cvReleaseImage(&edge); + cvReleaseImage(&cedge); + cvReleaseImage(&cedgergb); + cvReleaseImage(&rgb); + + //create the orig image with new size + rgb = cvCreateImage(cvSize(image.xsize,image.ysize), IPL_DEPTH_8U, 3); + + // Create the output images with new sizes + gray = cvCreateImage(cvSize(rgb->width,rgb->height), IPL_DEPTH_8U, 1); + + } + // FEM UNA COPIA DEL PACKET A image->imageData ... http://www.cs.iit.edu/~agam/cs512/lect-notes/opencv-intro/opencv-intro.html aqui veiem la estructura de IplImage + memcpy( rgb->imageData, image.data, image.xsize*image.ysize*3 ); + + // Convert to grayscale + cvCvtColor(rgb, gray, CV_RGB2GRAY); + + // Applies fixed-level thresholding to single-channel array. + switch(this->threshold_mode) { + case 0: + cvThreshold(gray, gray, (float)this->threshold_value, (float)this->max_value, CV_THRESH_BINARY); + break; + case 1: + cvThreshold(gray, gray, (float)this->threshold_value, (float)this->max_value, CV_THRESH_BINARY_INV); + break; + case 2: + cvThreshold(gray, gray, (float)this->threshold_value, (float)this->max_value, CV_THRESH_TRUNC); + break; + case 3: + cvThreshold(gray, gray, (float)this->threshold_value, (float)this->max_value, CV_THRESH_TOZERO); + break; + case 4: + cvThreshold(gray, gray, (float)this->threshold_value, (float)this->max_value, CV_THRESH_TOZERO_INV); + break; + } + + cvCvtColor(gray, rgb, CV_GRAY2BGR); + + //cvShowImage(wndname, cedge); + memcpy( image.data, rgb->imageData, image.xsize*image.ysize*3 ); +} + +void pix_opencv_threshold :: processYUVImage(imageStruct &image) +{ +} + +void pix_opencv_threshold :: processGrayImage(imageStruct &image) +{ + unsigned char *pixels = image.data; + + if ((this->comp_xsize!=image.xsize)||(this->comp_ysize!=image.ysize)||(!rgb)) { + + this->comp_xsize = image.xsize; + this->comp_ysize = image.ysize; + + //Destroy cv_images to clean memory + cvReleaseImage(&orig); + cvReleaseImage(&gray); + cvReleaseImage(&edge); + cvReleaseImage(&cedge); + cvReleaseImage(&cedgergb); + cvReleaseImage(&rgb); + + //create the orig image with new size + rgb = cvCreateImage(cvSize(image.xsize,image.ysize), IPL_DEPTH_8U, 3); + + // Create the output images with new sizes + cedgergb = cvCreateImage(cvSize(rgb->width,rgb->height), IPL_DEPTH_8U, 3); + + gray = cvCreateImage(cvSize(rgb->width,rgb->height), IPL_DEPTH_8U, 1); + edge = cvCreateImage(cvSize(rgb->width,rgb->height), IPL_DEPTH_8U, 1); + + } + // FEM UNA COPIA DEL PACKET A image->imageData ... http://www.cs.iit.edu/~agam/cs512/lect-notes/opencv-intro/opencv-intro.html aqui veiem la estructura de IplImage + memcpy( gray->imageData, image.data, image.xsize*image.ysize ); + + // Convert to RGB + cvCvtColor( gray, rgb, CV_GRAY2RGB); + + cvSmooth( gray, edge, CV_BLUR, 3, 3, 0, 0 ); + cvNot( gray, edge ); + + // Run the edge detector on grayscale + cvCanny(gray, edge, (float)this->edge_thresh, (float)this->edge_thresh*3, 3); + + cvZero( cedgergb ); + // copy edge points + cvCopy( rgb, cedgergb, edge ); + + cvCvtColor( cedgergb, gray, CV_RGB2GRAY); + //cvShowImage(wndname, cedge); + memcpy( image.data, gray->imageData, image.xsize*image.ysize ); +} + +///////////////////////////////////////////////////////// +// floatThreshMess +// +///////////////////////////////////////////////////////// +void pix_opencv_threshold :: floatThreshMess (float edge_thresh) +{ + this->edge_thresh = (int)edge_thresh; +} +void pix_opencv_threshold :: floatMaxMess (float thresh_max) +{ + if ((thresh_max>=0)&&(thresh_max<256)) this->max_value = (int)thresh_max; +} +void pix_opencv_threshold :: floatModeMess (float mode) +{ + f ((mode>=0)&&(mode<7)) this->threshold_mode = (int)edge_thresh; +} + +///////////////////////////////////////////////////////// +// static member function +// +///////////////////////////////////////////////////////// +void pix_opencv_threshold :: obj_setupCallback(t_class *classPtr) +{ + class_addmethod(classPtr, (t_method)&pix_opencv_threshold::floatModeMessCallback, + gensym("mode"), A_FLOAT, A_NULL); + class_addmethod(classPtr, (t_method)&pix_opencv_threshold::floatMaxMessCallback, + gensym("ft1"), A_FLOAT, A_NULL); + class_addmethod(classPtr, (t_method)&pix_opencv_threshold::floatTreshMessCallback, + gensym("ft2"), A_FLOAT, A_NULL); +} +void pix_opencv_threshold :: floatTreshMessCallback(void *data, t_floatarg edge_thresh) +{ + GetMyClass(data)->floatThreshMess((float)edge_thresh); +} +void pix_opencv_threshold :: floatMaxMessCallback(void *data, t_floatarg edge_thresh) +{ + GetMyClass(data)->floatMaxMess((float)edge_thresh); +} +void pix_opencv_threshold :: floatModeMessCallback(void *data, t_floatarg edge_thresh) +{ + GetMyClass(data)->floatModeMess((float)edge_thresh); +} -- cgit v1.2.1