//
// Created by basic on 19-8-14.
//

#ifndef FACEDETECT_DETECT_H
#define FACEDETECT_DETECT_H

#include <cstdio>
#include <map>
#include <vector>

#include "opencv2/opencv.hpp"
#include "THFaceTracking_i.h"
#include "common.h"

#include <QObject>

class faceTracking : public QObject
{
    Q_OBJECT

public:
    faceTracking(int devId, int chan = 1, int maxFaceNum = 3,
                 int minWidth = 50, int yaw = 45, int pitch = 45, int roll = 30,
                 int confidence = 60, int sampSize = 640, int interval = 5);

    ~faceTracking();

    int detect(int chan, BYTE *imgData, THFT_FaceInfo *outObjs);
    void run(int chan);

    void setThresVal(int confidence = 60, int width = 50, int yaw = 45, int pitch = 45, int roll = 30);

    int resetWH(int w, int h, int chan=0);

    void setnSampleSize(int size){
        nSampleSize = size;
    }

private:
    int initTrack(void);
    int initDecode(int devId);
    bool sendFacesToExtract(int chan, THFT_FaceInfo *faces, int num, BYTE *imgData, int width, int height);
    void showImage(cv::Mat &frame);
private:
    int nChannel = 1;
    int nMaxFaceNum = 10;
    int nSampleSize = 640;
    int treshYaw = 45;//angle of yaw,from -90 to +90,left is negative,right is postive
    int treshPitch = 45;//angle of pitch,from -90 to +90,up is negative,down is postive
    int treshRoll = 30;//angle of roll,from -90 to +90,left is negative,right is postive
    float treshConfidence = 0.8;//confidence of face pose(from 0 to 1,0.6 is suggested threshold)
    int treshMinWidth = 50;//RECT.width threshold
    int nDetectionIntervalFrame = 5;//interval frame number of face detection for face tracking
    float confidenceInc = 0.05;

    double interval = 3000;//500ms  send

    cv::VideoCapture vc;
    int nWidth;
    int nHeight;

    std::map<int, std::map<long, float>> recChanFaces;

    std::map<long, double> mIDTime;

signals:

    void drawImage(ImgToShow image);

    void signalTips(std::string str);

    void signalAdvertise(bool value);
private:
    double screenSaveTime_ms = 3000;
    void drawFaceRect(cv::Mat& frame, RECT rcFace);
};



#endif //FACEDETECT_DETECT_H