#ifndef IMAGEDRAWELEMENT_H
#define IMAGEDRAWELEMENT_H

#include <YoloServer.h>
//#include <FaceServer.h>
#include <basic/pipe_element/ImageFactoryElement.h>
#include <basic/core/DoubleBufferedData.h>
#include "../StructureApp/TrackingTrigger.h"

class ImageDrawElement : public ImageFactoryElement {
public:
    ImageDrawElement();

    // ImageFactoryElement interface
//    void setFaces(std::vector<ScoredRect> value);

    void setYoloObjects(std::vector<ScoredRect> value);

private:
    virtual void processImage(cv::Mat &) override;

    void darwProperty(cv::Mat &image, string key, string value, int x, int y);

private:
//    DoubleBufferedData<std::vector<ScoredRect>> faces;
    DoubleBufferedData<std::vector<ScoredRect>> yoloObjects;
};


#endif // IMAGEDRAWELEMENT_H