//
// Created by Scheaven on 2020/3/23.
//

#ifndef ARI_MANAGER_H
#define ARI_MANAGER_H
#include "std_target.h"
#include "config.h"
#include "detector.h"
#include "../utils/draw_util.h"
#include "../tracker_tools/tracker.h"
#include "detecter_manager.h"
// #include "../utils/result_util.h"


using namespace std;
using namespace cv;

class AriManager{
public:
    std::vector<cv::Rect> boxes;
    int cam_id; //指定待追踪摄像机编号
    Mat frame;

    std::vector<std::shared_ptr<tracker>> CAMERAS_VCT;
    std::vector<DETECTIONS> DETECTION_VCT;

private:
    std::shared_ptr<Detector> detector;
    static void init_target(Target *t);
    void switch_SDK_TResult(int cam_id, cv::Mat img, DETECTIONS detections, TResult *t_result);
    void switch_SDK_TResult(DETECTIONS detections, TResult *t_result);
    void single_tracker(int cam_id, DETECTIONS& detections, char* img_time);

public:
    AriManager();
    ~AriManager();
    void release();
    void init_load_model();
    bool add_cam(int cam_id);
    // void single_detect_tracking(int cam_id, cv::Mat frame, FRAME_RESULT result_vec); // 检测+追踪 接口
    void single_SDK(const int cam_id, const void *img, TResult *t_result, char* img_time, const char* mode);
    void single_SDK(const int cam_id, const void *img, TResult* t_result);
    // void mul_detect_tracking(std::vector<int> cam_ids, std::vector<cv::Mat> frame_vec, std::vector<FRAME_RESULT>&  results_vec); // 多摄像头追踪
};

#endif //INC_01_CPP_SORT_TRACKER_MANAGER_H