diff options
Diffstat (limited to 'src/pix_opencv_backgroundsubtractor.h')
-rw-r--r-- | src/pix_opencv_backgroundsubtractor.h | 96 |
1 files changed, 96 insertions, 0 deletions
diff --git a/src/pix_opencv_backgroundsubtractor.h b/src/pix_opencv_backgroundsubtractor.h new file mode 100644 index 0000000..cffa9b2 --- /dev/null +++ b/src/pix_opencv_backgroundsubtractor.h @@ -0,0 +1,96 @@ +/*----------------------------------------------------------------- +LOG + GEM - Graphics Environment for Multimedia + + Threshold filter + + 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_BACKGROUNDSUBTRACTOR_H_ +#define INCLUDE_PIX_OPENCV_BACKGROUNDSUBTRACTOR_H_ + +#include "opencv2/opencv.hpp" +#if HAVE_LIBOPENCV_CL +#include "opencv2/ocl/ocl.hpp" +#endif /* HAVE_LIBOPENCV_CL */ + +#include "Base/GemPixObj.h" +#include <RTE/MessageCallbacks.h> +#include "Gem/Exception.h" + +#include <iostream> + +/*----------------------------------------------------------------- +------------------------------------------------------------------- +CLASS + pix_opencv_backgroundsubtractor + + generic dynamic background subtractor, should implement severals algo +KEYWORDS + pix + +DESCRIPTION + +-----------------------------------------------------------------*/ +class GEM_EXPORT pix_opencv_backgroundsubtractor: public GemPixObj +{ + CPPEXTERN_HEADER(pix_opencv_backgroundsubtractor, GemPixObj) + + public: + + ////////// + // Constructor + pix_opencv_backgroundsubtractor(); + + protected: + + // Message handling + void enumParamsMess(); + void setParamMess(t_symbol *s, int argc, t_atom* argv); + void getParamMess(t_symbol *paramName); + void paramHelpMess(); + void algoMess(t_symbol *s, int argc, t_atom* argv); + void cpuModeMess(int val); + + + ////////// + // Destructor + virtual ~pix_opencv_backgroundsubtractor(); + + ////////// + // Do the processing + virtual void processImage(imageStruct &image); + virtual void startRendering(); + virtual void stopRendering(); + + private: + cv::Ptr<cv::BackgroundSubtractor> m_fgbgMOG; + cv::Ptr<cv::BackgroundSubtractorGMG> m_fgbgGMG; + +#if HAVE_LIBOPENCV_CL + cv::ocl::MOG m_oclMOG; + cv::ocl::MOG2 m_oclMOG2; + //~ cv::ocl::GMG m_oclGMG; + cv::ocl::oclMat d_input, d_fgmask; +#endif /* HAVE_LIBOPENCV_CL */ + + cv::Mat m_fgmask, m_segm; + + std::string m_algoName; + + t_outlet *m_dataout; + + t_float m_threshold, m_initFrames; + bool m_rendering, m_gpuMode, m_forceCPU; + + std::vector<std::string> m_bgsub_algos; + +}; +#endif // for header file |