From 8e10c57c7e053d8789747cf1e2c5fa78f2f65cc7 Mon Sep 17 00:00:00 2001
From: Scheaven <xuepengqiang>
Date: 星期三, 11 八月 2021 20:17:55 +0800
Subject: [PATCH] add t
---
src/h_interface.cpp | 64 +++++++++++++++++++++++++++++--
1 files changed, 59 insertions(+), 5 deletions(-)
diff --git a/src/h_interface.cpp b/src/h_interface.cpp
index 1d34297..98e2595 100644
--- a/src/h_interface.cpp
+++ b/src/h_interface.cpp
@@ -9,23 +9,68 @@
string m_staticStruct::model_path = "path";
int m_staticStruct::type = 1;
+bool m_staticStruct::isTrack= true;
+int m_staticStruct::max_cam_num = 0;
+int m_staticStruct::wander_time = 0;
+float m_staticStruct::mv_velocity = 0;
+float m_staticStruct::fall_rate = 0;
-// 鍒涘缓
+
API void* create(const char *conf, int *max_chan)
{
+ // CLog::Initialize("/opt/vasystem/bin/models/baseDetector/log.properties");
CLog::Initialize("../config/log4cplus.properties");
ReadJsonFromFile(conf);
AriManager *handle = new AriManager();
+ *max_chan = m_staticStruct::max_cam_num;
+ for (int i = 1; i <= m_staticStruct::max_cam_num; ++i)
+ {
+ handle->add_cam(i);
+ }
+ Timer::getInstance()->reset();
return handle;
}
-// 鍘熸煡鎵剧粨鏋滄帴鍙�
API void* get_result(void *handle, const void *img, const int chan)
{
AriManager *h = (AriManager*)handle;
TResult *t_result = (TResult*)malloc(sizeof(TResult));
init_TResult(t_result);
- h->single_SDK(chan, img, *t_result);
+
+ Timer::getInstance()->out("eveTime before yolo");
+ // h->single_SDK(chan, img, *t_result);
+ Timer::getInstance()->out("eveTime runing yolo");
+ return t_result;
+}
+
+
+API void* get_result2(void *handle, const void *img, const int chan, const char* timestamp, const char* mode)
+{
+ DEBUG("-----------------------begin------------ ");
+ AriManager *h = (AriManager*)handle;
+ TResult *t_result = (TResult*)malloc(sizeof(TResult));
+ init_TResult(t_result);
+
+ if(!m_staticStruct::isTrack)
+ {
+ Timer::getInstance()->out("get_result2 before yolo");
+ h->single_SDK(chan, img, t_result);
+ Timer::getInstance()->out("get_result2 runing yolo");
+ return t_result;
+ }
+
+ h->single_SDK(chan, img, t_result, const_cast<char*>(timestamp), mode);
+ DEBUG("--cam id:" + to_string(chan) + " image human_count:" + to_string(t_result->count));
+ for (int i = 0; i < t_result->count; ++i)
+ {
+ if(mode == "video"){
+ DEBUG("--human_id:" + to_string(t_result->targets[i].id)+" human_confidence:" + to_string(t_result->targets[i].confidence) + " human_attribute:" + std::string(t_result->targets[i].attribute) + " human_top:" + to_string(t_result->targets[i].rect.top)+" human_left:" + to_string(t_result->targets[i].rect.left) + " human_right:" + to_string(t_result->targets[i].rect.right)+ " human_bottom:" + to_string(t_result->targets[i].rect.bottom));
+ }
+ else{
+ DEBUG("human_confidence:" + to_string(t_result->targets[i].confidence) + " human_top:" + to_string(t_result->targets[i].rect.top)+" human_left:" + to_string(t_result->targets[i].rect.left) + " human_right:" + to_string(t_result->targets[i].rect.right)+ " human_bottom:" + to_string(t_result->targets[i].rect.bottom));
+ }
+ }
+ DEBUG("---------------------end over------------------------\n\n");
return t_result;
}
@@ -51,7 +96,6 @@
h = NULL;
}
-// 璇诲彇Json鏂囦欢
void ReadJsonFromFile(const char* filename)
{
Json::Reader reader;
@@ -72,8 +116,19 @@
{
std::string model_path = root["param"]["model_path"].asString();
int type = root["param"]["type"].asInt();
+ bool isTrack = root["param"]["isTrack"].asBool();
+ int max_cam_num = root["param"]["max_cam_num"].asInt();
+ int wander_time = root["param"]["wander_time"].asInt();
+ int mv_velocity = root["param"]["mv_velocity"].asFloat();
+ int fall_rate = root["param"]["fall_rate"].asFloat();
m_staticStruct::model_path = model_path;
m_staticStruct::type = type;
+ m_staticStruct::isTrack = isTrack;
+ m_staticStruct::max_cam_num = max_cam_num;
+ m_staticStruct::wander_time = wander_time;
+ m_staticStruct::mv_velocity = mv_velocity;
+ m_staticStruct::fall_rate = fall_rate;
+
}
in.close();
}
@@ -83,4 +138,3 @@
t->count = 0;
t->targets = nullptr;
}
-
--
Gitblit v1.8.0