/*
 * @Author: yangzilong
 * @Date: 2021-12-17 20:27:57
 * @Last Modified by: yangzilong
 * @Email: yangzilong@objecteye.com
 * @Description:
 */
#pragma once

#include "../ai_platform/det_obj_header.h"
#include <vector>
#include <string>
#include <map>
#include <bitset>
#include "../ai_platform/task_param_manager.h"
#include "../ai_platform/header.h"
#include "../util/vpc_util.h"

// #define EXPANSION_PROPORTION 0.1
#define EXPANSION_PROPORTION 0.25 
#define FACE_EXPANSION_PROPORTION 0.5
#define IMG_CHANNELS 3

using namespace std;

class DeviceMemory;

typedef struct multi_obj_data_t{
	long long id;
	DeviceMemory *memPtr;
	std::vector<video_object_info> objs;
	string task_id;
} multi_obj_data_t;


struct OBJ_VALUE {
	bool finishTracker;  //轨迹结束可以保存了
	int frameCount;
	bool isupdate;
	int lost;
	vpc_img_info snapShot;
	vpc_img_info snapShotLittle;
	sy_rect obj_pos;
	double confidence;	   		//	置信度
	float last_area;
	float max_area;
	OBJ_INDEX index;

	bitset< EDGESIZE >  flags;   //标志位，标记快照框应该如何判断，left:0  top:1  right:2  bottom:3
	sy_point landmark_point[25];//人脸关键点位置信息
	sy_rect position;  			//检测实际目标框
	//-added by zsh 220719 人脸姿态角-
	float roll = 0.0; 
	float yaw = 0.0;												
	float pitch = 0.0;
	//-------------------------------
};

struct OBJ_VALUES {
	OBJ_VALUE snapShots[3]; //缓存三张抓拍图 0-轨迹首张 1-轨迹最佳 2-轨迹末张
	bool exists[3] = {false, false, false}; //标识对应抓拍图是否存在
};

class snapshot_reprocessing
{
public:
	//更新最优快照
	snapshot_reprocessing(int devId);
	~snapshot_reprocessing();

	vector<multi_obj_data_t> get_vehicle_snapshot(vector<DeviceMemory*> vec_devMem, vector<onelevel_det_result>& ol_det_result, int skip_frame);

	void screen_effective_snapshot(vector<onelevel_det_result> &_onelevel_det_result);
	void filter_vehicle(vector<onelevel_det_result> &_onelevel_det_result);

	void update_bestsnapshot(vector<DeviceMemory*> vec_devMem, vector<onelevel_det_result> &ol_det_result, vector<vector<int>>& delete_object_id);
	map<OBJ_KEY, OBJ_VALUE> get_total_snapshot_info();

	int update_face_bestsnapshot(vector<DeviceMemory*> vec_devMem, vector<onelevel_det_result> &ol_det_result, vector<vector<int>>& delete_object_id);
	map<OBJ_KEY, OBJ_VALUE> get_total_face_snapshot_info();

	void update_village_bestsnapshot(vector<DeviceMemory*> vec_devMem, vector<onelevel_det_result> &ol_det_result, vector<vector<int>>& delete_object_id);
	map<OBJ_KEY, OBJ_VALUES> get_total_village_snapshot_info();

	void release_finished_locus_snapshot(const string taskid, const int objid = -1, bool bRelease = true);   //-1为删除该路所有任务的快照图
	void release_village_finished_locus_snapshot(const string taskid, const int objid = -1, bool bRelease = true);
	void release_finished_face_locus_snapshot(const string taskid, const int objid = -1, bool bRelease = true);

private:
	bool best_snapshot_judge_algor(const OBJ_KEY& obj_key, const OBJ_VALUE& obj_value, int left, int top, int width, int height, int image_width, int image_height);
	bool best_face_snapshot_judge_algor_v2(const OBJ_KEY& obj_key, const OBJ_VALUE& obj_value, int left, int top, int width, int height, int image_width, int image_height, float roll, float yaw, float pitch);

private:
	map<OBJ_KEY, OBJ_VALUE> total_snapshot_info;
	map<OBJ_KEY, OBJ_VALUE> total_face_snapshot_info;
	map<OBJ_KEY, OBJ_VALUES> total_village_snapshot_info;

	map<string, set<int>> algor_index_table;
	task_param_manager *m_task_param_manager;
};