#ifndef TRACKINGTRIGGER_H
#define TRACKINGTRIGGER_H
#include <vector>
#include <opencv2/opencv.hpp>
#include <functional>
#include <atomic>
using namespace std;
struct ScoredRect{
    //ScoredRect():id(-1){}
    float score;
    cv::Rect rect;
    long id;
    map<string,string> properties;
};

class TrackingTrigger
{
public:
    TrackingTrigger(float threshold)://threshold：触发阈值，满足要求阈值的人脸才会触发
            threshold(threshold),faceTrackingId(0)
    {}

    bool triggerOnce(ScoredRect& rect){//检测一次人脸后，for循环内每个人脸封装成ScoredRect结构体进行调用，返回值是该人脸是否被触发
        bool found = false;
        for(auto lastRect: lastScoreRects){
            if(lastRect.id>=0&&(rect.rect&lastRect.rect).area()>lastRect.rect.area()*0.4){
                found = true;
                rect.id = lastRect.id;
                rect.properties = lastRect.properties;
                tempScoreRects.push_back(rect);
                break;
            }
        }
        if(!found){
            if(rect.score<threshold){
                tempScoreRects.push_back(rect);
                return false;
            }else{
                rect.id = faceTrackingId++;
                tempScoreRects.push_back(rect);
                return true;
            }
        }
        return false;
    }
    void triggerLine(){//for 循环遍历后调用，结束一组触发判断
        lastScoreRects = tempScoreRects;
        tempScoreRects.clear();
    }

    ScoredRect& getLastRect(){
        return tempScoreRects[tempScoreRects.size() -1];
    }

    std::vector<ScoredRect> getLastScoreRects() const
    {
        return lastScoreRects;
    }

private:
    float threshold;
    std::vector<ScoredRect> lastScoreRects;
    std::vector<ScoredRect> tempScoreRects;
    std::atomic<long> faceTrackingId;
};

#endif // TRACKINGTRIGGER_H