//
|
// Created by basic on 18-8-24.
|
//
|
|
#ifndef JUDGINGRETROGRADE_JUDGMENTRETROGRADETOOL_H
|
#define JUDGINGRETROGRADE_JUDGMENTRETROGRADETOOL_H
|
|
#include <iostream>
|
#include <map>
|
#include <list>
|
#include <opencv2/opencv.hpp>
|
#include <basic/pipe/TimerElement.h>
|
#include "YoloRpcElement.h"
|
#include <QSharedMemory>
|
#include <QString>
|
#include <QJsonArray>
|
#include <QJsonObject>
|
#include <QJsonDocument>
|
#include <QtGui/QPolygon>
|
|
struct VectorPoint{
|
cv::Point2f start_Point;
|
cv::Point2f end_Point;
|
};
|
|
typedef VectorPoint BaseLine;
|
namespace JRTOOL{
|
//以pt1为基准
|
//计算两个向量的夹角,0~90之间为正常,90~180为逆行
|
static float getAngelOfTwoVector(VectorPoint &pt1, VectorPoint &pt2){
|
float theta = atan2(pt1.end_Point.x - pt1.start_Point.x, pt1.end_Point.y - pt1.start_Point.y)
|
- atan2(pt2.end_Point.x - pt2.start_Point.x, pt2.end_Point.y - pt2.start_Point.y);
|
if (theta > CV_PI)
|
theta -= 2 * CV_PI;
|
if (theta < -CV_PI)
|
theta += 2 * CV_PI;
|
theta = theta * 180.0 / CV_PI;
|
return abs(theta);
|
}
|
|
//滤波
|
static float low_filter(float a, float b){
|
float sample_value;
|
float X=0.1;
|
sample_value=(1-X)*b+X*a;
|
return(sample_value);
|
}
|
}
|
|
|
class JudgmentRetrogradeTool :public TimerElement {
|
public:
|
JudgmentRetrogradeTool();
|
|
virtual ~JudgmentRetrogradeTool();
|
|
bool init(QString area,QString line);
|
|
void setYoloObjects(std::vector<ScoredRect> value);
|
|
bool getPerRet(const long&);
|
|
bool getTriggerState() const;
|
|
private:
|
void setPerPoint(const long&,cv::Point2f);
|
void setPerRect(const long&,cv::Rect rect);
|
|
//设置上行的运动基准线
|
void setUpBaseline(cv::Point start,cv::Point end);
|
|
//设置上行检测区域
|
void setUpDetectionArea();
|
|
//设置下行的运动基准线
|
void setDownBaseline(cv::Point start,cv::Point end);
|
|
//设置下行检测区域
|
void setDownDetectionArea();
|
|
|
void func();
|
|
float getTheta(BaseLine&,VectorPoint&);
|
|
QJsonArray getJsonArrayFromQString(const QString strJson);
|
|
|
private:
|
// virtual void threadInitial() override;
|
virtual void timerFunc()override;
|
// virtual void threadClosing()override;
|
|
private:
|
|
//基准线向量
|
BaseLine m_UpBaseLine;
|
BaseLine m_DownBaseLine;
|
|
//人员运动轨迹
|
//跟踪id,运动中心点
|
std::map<long,std::vector<cv::Point2f>> m_mapPerPoint;
|
|
//跟踪id,与基准线的夹角列表
|
std::map<long,std::list<float>> m_mapPerDirection;
|
|
//跟踪id,逆行状态
|
std::map<long,bool> m_mapPerRet;
|
|
QPolygon m_polygon;
|
|
TriggerElement m_triggerElement;
|
|
std::vector<ScoredRect> mObjs;
|
};
|
|
|
#endif //JUDGINGRETROGRADE_JUDGMENTRETROGRADETOOL_H
|
