diff options
Diffstat (limited to 'src/pix_opencv_blobtrack.h')
| -rw-r--r-- | src/pix_opencv_blobtrack.h | 106 |
1 files changed, 106 insertions, 0 deletions
diff --git a/src/pix_opencv_blobtrack.h b/src/pix_opencv_blobtrack.h new file mode 100644 index 0000000..2e00513 --- /dev/null +++ b/src/pix_opencv_blobtrack.h @@ -0,0 +1,106 @@ +/*----------------------------------------------------------------- +LOG + GEM - Graphics Environment for Multimedia + + Copyright (c) 1997-1999 Mark Danks. mark@danks.org + Copyright (c) Günther Geiger. geiger@epy.co.at + Copyright (c) 2001-2002 IOhannes m zmoelnig. forum::für::umläute. IEM. zmoelnig@iem.kug.ac.at + 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. + +-----------------------------------------------------------------*/ + +#ifndef INCLUDE_PIX_OPENCV_BLOBTRACK_H_ +#define INCLUDE_PIX_OPENCV_BLOBTRACK_H_ + +#ifndef _EiC +#include "opencv2/video/background_segm.hpp" +#include "opencv2/legacy/blobtrack.hpp" +#include "opencv2/legacy/legacy.hpp" +#include "opencv2/highgui/highgui.hpp" +#include <opencv2/imgproc/imgproc_c.h> +#endif + +#include "blobtrack.h" + +#include "Base/GemPixObj.h" +#include <stdio.h> +#include <RTE/MessageCallbacks.h> +/*----------------------------------------------------------------- +------------------------------------------------------------------- +CLASS + pix_opencv_blobtrack + + advanced blob tracker + +KEYWORDS + pix + +DESCRIPTION + +-----------------------------------------------------------------*/ +class GEM_EXPORT pix_opencv_blobtrack : public GemPixObj +{ + CPPEXTERN_HEADER(pix_opencv_blobtrack, GemPixObj) + + public: + + ////////// + // Constructor + pix_opencv_blobtrack(); + + protected: + + ////////// + // Destructor + virtual ~pix_opencv_blobtrack(); + + ////////// + // Do the processing + virtual void processRGBAImage(imageStruct &image); + virtual void processRGBImage(imageStruct &image); + virtual void processYUVImage(imageStruct &image); + virtual void processGrayImage(imageStruct &image); + void RunBlobTrackingAuto( IplImage* img ); + + ////////// + // Messages handling + void monitorStageMess(int arg); + void areaThresholdMess(t_float arg); + void fgTrainFramesMess(int arg); + void printParamsMess(void); + void getParamMess(t_symbol*s, int argc, t_atom*argv); // get available params or param value + void setParamMess(t_symbol*s, int argc, t_atom*argv); // set param + void getModuleMess(t_symbol*s, int argc, t_atom*argv); // list available and currently used modules + void setModuleMess(t_symbol*s, int argc, t_atom*argv); // set selected modules + + // Members + std::string m_fg_name, m_bd_name, m_bt_name, m_btgen_name, m_btpp_name, m_bta_name, m_bt_corr; + int m_FGTrainFrames; + int m_monitoring_stage; // 0 : input image, 1 : FG 3 : input with trackng info + t_float m_areaThreshold; + int x_size, y_size; + + + CvBlobTrackerAuto* m_tracker; + CvBlobTrackerAutoParam1 m_param; + + DefModule_FGDetector* m_FGModule; + DefModule_BlobDetector* m_BDModule; + DefModule_BlobTracker* m_BTModule; + DefModule_BlobTrackPostProc* m_BTPostProcModule; + DefModule_BlobTrackGen* m_BTGenModule; + DefModule_BlobTrackAnalysis* m_BTAnalysisModule; + + + private: + + void setupModules(); + void createModules(); + void releaseModules(void); + void print_params(CvVSModule* pM, const char* module); + t_outlet *m_dataout; // info outlet + +}; +#endif // for header file |