#ifndef FACEEXTRACTELEMENT_H
#define FACEEXTRACTELEMENT_H

#include "SaveVideoRpc.h"
#include <FaceServer.h>
#include <FaceSearchServer.h>
#include <basic/pipe/TimerElement.h>
#include <basic/util/BASE64/Base64.h>
#include "TrackingTrigger.h"
#include <basic/rpc/IceRpc.hpp>
#include <basic/util/fastdfs/FastFds.hpp>
#include <opencv2/opencv.hpp>
#include <queue>
#include <mutex>
#include <jsoncpp/json/json.h>
#include "DBStruct.h"
#define VECTOR_MAX 50

class QSharedMemory;

struct FaceToExtract {
    string uuid;
    string time;
    cv::Mat image;
    FaceDetect::Faces facesPos;
    std::vector<ScoredRect> scoredRects;
    std::vector<cv::Mat> faceImages;
};

class FaceExtractElement : public TimerElement {
public:

    FaceExtractElement(std::string shareMemoryName,const SdkRule& rule);
    //FaceExtractElement(std::string shareMemoryName,const SdkRule& rule,SaveVideoRpcClient_t& rpcClient);

    ~FaceExtractElement();

    void setFdfs(FastFdsWithLock *p_fdfsClient) {
        fdfsClient = p_fdfsClient;
    }

    void submitFaceExtract(FaceToExtract faceToExtract);

//    AlarmServerInterface::TableNames getTableNames() const;
    std::vector<::FaceDetect::Data> getFeatures() const;

private:

    virtual void timerFunc() override;

    IceRpcClient<FaceDetect::FaceExtractServerPrx> faceExtractRpcClient;
    IceRpcClient<FaceSearch::FaceFeatureSearchServerPrx> faceSearchRpcClient;
    FastFdsWithLock *fdfsClient;
    std::vector<FaceToExtract> faceExtractQueue;
    std::mutex imageQueueMutex;
    QSharedMemory *sharedMemory;

//    AlarmServerInterface::TableNames tableNames;
    std::vector<::FaceDetect::Data> features;

    Base64 base64;
    SdkRule m_sdkRule;
    SaveVideoRpcClient_t m_rpcClient;
};


#endif // FACEEXTRACTELEMENT_H