ObjectTracking/ObjectTracker.h at master · ccchang1023/ObjectTracking · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
#include <opencv2/opencv.hpp>
#include <opencv2/opencv_lib.hpp>

#if !defined(OBEJCTTRACKER_H_INCLUDED_)
#define OBEJCTTRACKER_H_INCLUDED_


#define  HISTOGRAM_LENGTH	16384
#define HISTOGRAM_QUANTIFIZE 32

//#define  HISTOGRAM_LENGTH			8192
//#define HISTOGRAM_QUANTIFIZE 16

 //#define  HISTOGRAM_LENGTH		512
 //#define HISTOGRAM_QUANTIFIZE 64

//#define  HISTOGRAM_LENGTH		2048
//#define HISTOGRAM_QUANTIFIZE 32


#define  MAX_OBJECT_TRACK_NUMBER	2

using namespace cv;

typedef unsigned char   UBYTE8;
typedef char            BYTE8;
typedef short           SINT16;
typedef unsigned int    UINT32;
typedef int             INT32;
typedef unsigned long   ULONG_32;
typedef long            LONG_32;
typedef float           FLOAT32;
typedef double          DOUBLE64;
typedef long double     LDOUBLE80;
typedef bool		    BBOOL;
enum IMAGE_TYPE {MD_RGB, MD_RGBA, MD_YUV, MD_WAVELET};


class CObjectTracker
{
private:

	struct OBJECT_SPECS
	{
		BBOOL   Status;
		BBOOL   assignedAnObject;
		BBOOL   applyKalman;
		SINT16  X;
		SINT16  Y;
		FLOAT32 KALMAX_X;
		FLOAT32 KALMAX_Y;
		SINT16  W;
		SINT16  H;
		SINT16  vectorX;
		SINT16  vectorY;
		FLOAT32  initHistogram[HISTOGRAM_LENGTH];
	};

	OBJECT_SPECS m_sTrackingObjectTable[MAX_OBJECT_TRACK_NUMBER+1];
	INT32	 m_nImageWidth;
	INT32	 m_nImageHeight;
	IMAGE_TYPE m_eIMAGE_TYPE;
	UBYTE8  m_cSkipValue;
	SINT16	 m_nFrameCtr;
	UBYTE8  m_cActiveObject;

	ULONG_32 m_uTotalTime;
	SINT16   m_nMaxEstimationTime;


	void FindHistogram(UBYTE8 *frame,UBYTE8 activeObjectNo,FLOAT32 (*histogram));
	void DrawObjectBox(UBYTE8 *frame,UBYTE8 activeObjectNo);

	void FindWightsAndCOM(UBYTE8 *frame,UBYTE8 activeObjectNo,FLOAT32 (*histogram));
	void applyKalmanFilter(UBYTE8 activeObjectNo);

	FLOAT32 FindDistance(UBYTE8 (*histogram),UBYTE8 activeObjectNo);
	UBYTE8 CheckEdgeExistance(UBYTE8 *frame,SINT16 _x,SINT16 _y);
	void UpdateInitialHistogram(UBYTE8 (*histogram),UBYTE8 activeObjectNo);
	void FindNextLocation(UBYTE8 *frame,UBYTE8 activeObjectNo);
	ULONG_32 GetPixelValues(UBYTE8 *frame,SINT16 x,SINT16 y);
	void SetPixelValues(UBYTE8 *frame,ULONG_32 pixelValues,SINT16 x,SINT16 y);
	ULONG_32 GetBoxColor(UBYTE8 activeObjectNo);
	FLOAT32 CompareHistogram(UBYTE8 (*histogram),UBYTE8 activeObjectNo);

public:

	CObjectTracker(INT32 imW,INT32 imH,IMAGE_TYPE eImageType);
	virtual ~CObjectTracker() ;
	void ObjeckTrackerHandlerByUser(UBYTE8 *frame);
	CvPoint getTrackerCenter(UBYTE8 activeObjectNo);
	void ObjectTrackerInitObjectParameters(SINT16 x,SINT16 y,SINT16 Width,SINT16 Height);
	void enableKalmanFilter(UBYTE8 activeObjectNo);
	void disableKalmanFilter(UBYTE8 activeObjectNo);

};//end of trackobject class
//---------------------------------------------------------------------------
#endif