InvSmokeDetector/InvFireDetector.h at master · captainjtx/InvSmokeDetector · 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
#include "stdlib.h"
#include <stdio.h>
#include <cv.h>
#include <cxcore.h>
#include <highgui.h>
#include <cvaux.h>//±ØÐëÒý´ËÍ·ÎÄ¼þ
#include <iostream>
#include <fstream>
#include "SwaveletFilter.h"
#include "FrGrdExtract.h"
#include "ColorFilter.h"
#include "constants.h"

using namespace std;
using namespace cv;

class InvFireDetector
{
public:
	bool      isPause;
	IplImage* pSrc;
	IplImage* pFeedBack;
	IplImage* pEnergy;
	CvRect    pRec[MAX_OF_BLOCK];
	int       blockNum;
	int       Mark[MAX_OF_BLOCK];

	int       frameNum;
	bool      invAlarm;
	bool      fireAlarm;
	int       fireBlockNum;

	int       countFrame;
	int       countFireBlock;
	InvFireDetector();
	void init(IplImage* pFrame,const char*wavename="haar");
	void update(IplImage* pFrame);
	void show();
	void release();
	void keyMenu();

	int  blockRowNum;
	int  blockColumnNum;
	CvRect* blockSeq;
	int  *blockMark;

private:
  CvVideoWriter* resultWriter;
	CvVideoWriter* energyWriter;
	CvVideoWriter* foreWriter;
	CvVideoWriter* backWriter;

	FrGrdExtract *mFrGrdExtract;
	ColorFilter *mColorFilter;
	SwaveletFilter *mSwaveletFilter;

	CvMat* pBackGround;
	CvMat* pBackMat;
	const char* wname;

	int count;
};