#ifndef TRACK_H
|
#define TRACK_H
|
|
#include "dataType.h"
|
|
#include "kalmfilter.h"
|
#include "model.h"
|
#include "time.h"
|
#include "stdio.h"
|
#include "stdlib.h"
|
#include <iostream>
|
#include<sys/time.h>
|
#include "../utils/geometry_util.h"
|
#include <queue>
|
#include "../utils/queue_util.hpp"
|
#include "../utils/log_util.h"
|
#include "../utils/time_util.h"
|
|
using namespace std;
|
|
class Track
|
{
|
|
enum TrackState {Tentative = 1, Confirmed, Deleted}; // 轨迹的状态，Tentative 是刚开始的时候，Confirmed是确认了是人的轨迹，Deleted是删除的轨迹
|
|
public:
|
bool isCurrent;
|
DETECTBOX xywh;
|
Track();
|
~Track();
|
Track(KAL_MEAN& mean, KAL_COVA& covariance, uint64_t track_id,
|
int n_init, int max_age, const FEATURE& feature, char* img_time);
|
void predit(KalmFilter *kf);
|
void update(KalmFilter * const kf, const DETECTION_ROW &detection, char* img_time);
|
bool mark_missed(char* img_time);
|
bool is_confirmed();
|
bool is_deleted();
|
bool is_tentative();
|
DETECTBOX to_tlwh();
|
DETECTBOX to_xywh();
|
int time_since_update; //累积检测不到的次数
|
uint64_t track_id;
|
FEATURESS features;
|
float confidence;
|
KAL_MEAN mean;
|
KAL_COVA covariance;
|
|
int hits;
|
int age;
|
int _n_init;
|
int _max_age;
|
TrackState state;
|
int time_substract(struct timeval *result, struct timeval *begin,struct timeval *end);
|
|
public:
|
// 附加的其他算法
|
// 逗留
|
// struct timeval last_s_time, last_e_time, last_diff_time; // 逗留的开始时间、结束时间，和持续时间
|
int last_s_time, last_e_time, last_diff_time; // 逗留的开始时间、结束时间，和持续时间
|
int last_time; //持续逗留时间
|
bool isWander;
|
|
//跌倒
|
float rate; //人体身高变化率
|
bool isFall;
|
bool first_Fall;
|
int rateScale; //判断是否逐步跌倒(短时间内)
|
MinQueue<float> rate_queue;
|
// struct timeval fall_s_time, fall_e_time, fall_diff_time; // 跌倒的持续时间 ,字段fall_diff_time代表了两个含义。距离最小比例的时间+累积跌倒时间
|
// struct timeval min_rate_time; //最大比例的定位时间
|
|
int fall_s_time, fall_e_time, fall_diff_time; // 跌倒的持续时间 ,字段fall_diff_time代表了两个含义。距离最小比例的时间+累积跌倒时间
|
int min_rate_time; //最大比例的定位时间
|
|
|
int fall_total_time;
|
float minRate; // 队列中的最大比例
|
float hisMinRate;
|
SPoint center_point, min_center_point; //重心的高度
|
float human_h;
|
float human_w;
|
|
|
|
//奔跑
|
bool is_runing_t;
|
bool is_runing_p;
|
bool isRuning;
|
SPoint last_point;
|
// struct timeval single_s_time, single_diff_time; // 单帧的结束时间可以用逗留里的时间代替
|
int single_s_time, single_diff_time; // 单帧的结束时间可以用逗留里的时间代替
|
int single_time; // 单帧所花销的时长(ms)
|
queue<float> velocity_queue;
|
float sum_velocity;
|
double run_rate;
|
double last_rate;
|
double rate_area;
|
|
// 帽子、口罩、抽烟
|
bool is_hat;
|
int hatScore;
|
int helmetScore;
|
int headScore;
|
|
bool is_mask;
|
int maskScore;
|
|
bool is_smoke;
|
int smokeScore;
|
|
private:
|
// struct timeval start,stop,diff;
|
int start,stop,diff;
|
double sum_timer;
|
bool init_t;
|
void init_track_time();
|
void featuresAppendOne(const FEATURE& f);
|
|
};
|
|
#endif // TRACK_H
|
