MutliSourceVideoProcess.h
4.59 KB
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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
//#pragma once
#ifndef __MutliSourceVideoProcess_H__
#define __MutliSourceVideoProcess_H__
#include <stdlib.h>
#include <iostream>
#include "nvml.h"
#include "time.h"
#include "opencv2/opencv.hpp"
#include "opencv2/highgui/highgui.hpp"
#include "VPTProcess.h"
#include <queue>
#include <set>
#include "common.h"
#include "../FFNvDecoder/ImageSaveGPU.h"
#include "mvpt_process_assist.h"
#include "snapshot_helper.h"
#include <atomic>
#ifdef _MSC_VER
#ifdef _DEBUG
#pragma comment(lib, "opencv_world310d.lib")
#else
#pragma comment(lib, "opencv_world310.lib")
#endif
#endif
using namespace cv;
using namespace std;
#ifndef _MSC_VER
#ifndef TRUE
#define TRUE 1
#endif
#ifndef FALSE
#define FALSE 0
#endif
#define Sleep(a) usleep((a)*1000)
//typedef int BOOL;
#define BOOL bool
typedef unsigned int DWORD;
typedef void* LPVOID;
#endif
#ifdef _DEBUG
#define DEBUG_MSG(msg,...) \
{ \
printf("%s %s [%d]: ", __FILE__, __FUNCTION__, __LINE__); \
printf(msg,##__VA_ARGS__); \
printf("\n"); \
}
#else
#define DEBUG_MSG(msg,...)
#endif
#define MAXLENGTH 416
#define MINLENGTH 224
#define THREAD_COUNT 30
#define SNAPSHOTFRAME 15
#define LOSTMAXFRAMECCOUNT 4
#define MAX_BATCH 10
struct VideoObjectSS
{
unsigned char * obj_person;
unsigned char * obj_bike;
unsigned char * obj_car;
VideoObjectSS() : obj_person(NULL), obj_bike(NULL), obj_car(NULL){}
};
enum TaskOperator
{
ADDTASK,
PAUSETASK,
RESTARTTASK,
FINISHTASK
};
struct Operator{
int changeTaskID;
char videoFileName[256];
char resultFolderLittleName[256];
char resultFolderName[256];
char result_folder_face[256];
bool on_image_display;
int jpeg_quality;
int decMode;
TaskOperator changeTaskOperator;
VIDEO_OBJECT_SNAPSHOT_CALLBACK taskObjCallbackFunc;
VIDEO_REALTIME_CALLBACK taskRealTimeCallbackFunc;
sy_rect minBoxsize[DETECTTYPE];
};
class CMutliSourceVideoProcess
{
public:
CMutliSourceVideoProcess();
~CMutliSourceVideoProcess();
int InitAlgorthim(mvpt_param vptParam, VIDEO_OBJECT_INFO_CALLBACK tObjInfoCallbackFunc, VIDEO_FINISH_CALLBACK tFinishCallbackFunc);
int get_task_progress(int taskid, double & progress);
void OperatorTask();
bool HasNewTask() {
return !TaskOperatorQ.empty();
}
void AddOperator(int taskID, int taskOper);
int AddOperator(task_param tparam);
void callTaskObjInfoCallbackFunc(int objCount, VPT_ObjInfo *obj, int taskFrameCount, int taskId);
bool AddTask(task_param tparam);
void PauseTask(const int taskID);
void RestartTask(const int taskID);
void FinishTask(const int taskID);
void ViewTask(const int taskID);
void FinishViewTask();
int FinishProcessThread();
void FinishDecode(const int taskID);
int GetRuningNb();
void algorthim_process();
private:
std::thread ProcessThread;
std::mutex _tx_add_task;
deque<Operator> TaskOperatorQ;
int capacity;
double gpu_total_memory;
std::thread thrd;
void* authority_handle;
public: /*��������Ӧ����public�� �������̺߳����л��õ����µ����� ÿ����дһ��get����̫������*/
void * VPT_Handle;
int section_batch_size;
int licence_status;
int thrd_status;
int mgpuid;
int skip_frame_ {5}; // 控制跳帧参数
vector<Task> tasks;
int AddTaskSucFlag; //0:��ʼ��״̬ 1����������ɹ� -1����������ʧ��
int TaskinPlay;
int TotalTask;
set<int> TaskinPlayID;
int viewTaskID;
FINISH_CALLBACK taskFinishCallbackFunc;
OBJECT_INFO_CALLBACK taskObjInfoCallbackFunc;
std::mutex taskMutex;
std::condition_variable taskCondVar;
//�㷨���ò���
sy_command m_hp_analysis_config; //�Ƿ���������ʶ��
sy_command m_hcp_analysis_config; //�Ƿ��������Ƽ��ʶ��
sy_command m_vehicle_analysis_config; //�Ƿ�����������ɫʶ��
sy_command m_vehicle_recg_config; //�Ƿ�������������ʶ��
sy_command m_vehicle_plate_det_recg_config; //�Ƿ�������Υ����Ϊ���
sy_command m_hf_recg_config; //�Ƿ�����������ʶ��
sy_command m_hcf_recg_config; //�Ƿ������ﳵ����ʶ��
sy_command m_vcf_recg_config; //�Ƿ�����������ʶ��
sy_command m_face_det_config; //�Ƿ����������
snapshot_helper m_snaphot_helper;
private:
bool m_bExit{false};
bool m_bProcessExit{false};
VPTProcess m_vptProcess;
};
#endif