sequence_face_landmarks.h

#ifndef __SFL_SEQUENCE_FACE_LANDMARKS__
#define __SFL_SEQUENCE_FACE_LANDMARKS__

// std
#include <string>
#include <list>
#include <memory>

// OpenCV
#include <opencv2/core.hpp>

namespace sfl
{
    struct Face
    {
        int id;                             
        cv::Rect bbox;                      
        std::vector<cv::Point> landmarks;   
    };

    struct Frame
    {
        int id;                                 
        int width;                              
        int height;                             
        std::list<std::unique_ptr<Face>> faces; 

        const Face* getFace(int id) const;
    };

    enum FaceTrackingType
    {
        TRACKING_NONE = 0,
        TRACKING_BRISK = 1,
        TRACKING_LBP = 2
    };

    class SequenceFaceLandmarks
    {
    public:

        virtual const Frame& addFrame(const cv::Mat& frame, int id = -1) = 0;

        virtual const std::list<std::unique_ptr<Frame>>& getSequence() const = 0;

        virtual std::list<std::unique_ptr<Frame>>& getSequenceMutable() = 0;

        virtual void clear() = 0;

        virtual std::shared_ptr<SequenceFaceLandmarks> clone() = 0;

        virtual const std::string& getModel() const = 0;

        virtual float getFrameScale() const = 0;

        virtual const std::string& getInputPath() const = 0;

        virtual FaceTrackingType getTracking() const = 0;

        virtual void load(const std::string& filePath) = 0;

        virtual void save(const std::string& filePath) const = 0;

        virtual void setFrameScale(float frame_scale) = 0;

        virtual void setModel(const std::string& modelPath) = 0;

        virtual void setInputPath(const std::string& inputPath) = 0;

        virtual void setTracking(FaceTrackingType tracking) = 0;
        
        virtual size_t size() const = 0;

        static std::shared_ptr<SequenceFaceLandmarks> create(
            const std::string& landmarks_path, float frame_scale = 1.0f,
            FaceTrackingType tracking = TRACKING_NONE);

        static std::shared_ptr<SequenceFaceLandmarks> create(
            float frame_scale = 1.0f, FaceTrackingType tracking = TRACKING_NONE);
    };

}   // namespace sfl

#endif  // __SFL_SEQUENCE_FACE_LANDMARKS__