mstracker.h

基于均值漂移理论的该进算法

//Added by xdb 2007-03-19
#ifndef MSTRACKER_H
#define MSTRACKER_H

#pragma  once

#include "colourModel.h"
//#include "CImg.h"
#include <string>
//using namespace cimg_library;
using namespace std;

class CMsTracker
{
public:
	CMsTracker();
	~CMsTracker();
	BOOL InitTracker(CRect trackRect, BYTE *pImage, UINT bytesPerLine, UINT imageWidth,
								UINT imageHeight);
	BOOL UpdateTracker( int loopMax, BYTE *pImage, UINT bytesPerLine, int tolerWidth );	
	void ReleaseTracker();

protected:	

private:
	
	int			m_centerx;
	int			m_centery;
	int			m_sizex;
	int			m_sizey;
	int			m_halfsizex;
	int			m_halfsizey;
	double		**weight_array;
	double		**kernel_array;
	int			**kernelDeriv_array;	

	colourModel Qu;
	colourModel Pu;
	int			dx, dy, pdx, pdy;
	int			m_imageWidth, m_imageHeight;




	double kernel(int x,int y,int half_x,int half_y);
	void evalKernel (double*** kArray,int half_x,int half_y);
	void evalKernelDeriv (int*** kArray,int half_x,int half_y);
	//void writeFrame(CImg<unsigned char>* frame,int centre_x,int centre_y,int half_size_x,int half_size_y,string name);
	void DrawFrame( BYTE *pImage, int centre_x,int centre_y,int half_size_x,int half_size_y, UINT bytesPerLine );
	void updateWeights(double*** weights,colourModel *Pu,colourModel *Qu,unsigned int x_size,unsigned int y_size);
	void computeDisplacement(double*** weights,int*** kArray,unsigned int centre_x,unsigned int centre_y,int half_x,int half_y,int *dx,int *dy);
};

#endif