camshift.h

tracciatore di mani con webcam

#if !defined(___HV_CAMSHIFT_INCLUDED__)
#define ___HV_CAMSHIFT_INCLUDED__

#include <cv.h>
#include <cvaux.h>
#include "Rect.h"
#include "ProbDistrProvider.h"

class CamShift {
public:
  CamShift();

public:
  void Initialize(int width, int height);
  void PrepareTracking(IplImage* image, const ProbDistrProvider* pProbProv, const CRect& area);
  void Track(IplImage* image);
  void DrawOverlay(IplImage* image, int overlay_level) const;
  void GetArea(CRect& area) const;

protected:
  void DoHSV(IplImage* image, bool initialize);
  void DoExternal(IplImage* image);
  void CheckBackProject(IplImage* image, int view) const;
  void DrawCross(IplImage* image) const;

protected:
  bool                      m_tracking;
  int                       m_img_width, m_img_height;

  // for external probability image
  const ProbDistrProvider*  m_pProbProv;
  IplImage*                 m_prob_map;
  CvBox2D                   m_box;
  CvConnectedComp           m_comp;

  // for internal probability map based on HSV histogram
  CvCamShiftTracker         m_cCamShift;
  CvRect                    m_object;

  // HSV histogram parameters
  int                       m_Smin, m_Vmin, m_Vmax;
  int                       m_threshold;
  int                       m_bins;
};


#endif // ___HV_CAMSHIFT_INCLUDED__