#pragma once

#include <opencv2/freetype.hpp>

#include "../vp_node.h"


namespace vp_nodes {
    // on screen display(short as osd) node.
    // used for displaying vehicle plate on frame, draw rectangle according to plate color
    class vp_plate_osd_node: public vp_node
    {
    private:
        // support chinese font
        cv::Ptr<cv::freetype::FreeType2> ft2;
        float mask_threshold = 0.3;

        // draw color on frame
        std::map<std::string, cv::Scalar> draw_colors {{"blue", cv::Scalar(255, 0, 0)},
                                                {"green", cv::Scalar(0, 255, 0)},
                                                {"yellow", cv::Scalar(0, 255, 255)},
                                                {"white", cv::Scalar(255, 255, 255)}};

        // map to Chinese
        std::map<std::string, std::string> text_colors {{"blue", "蓝"},
                                                {"green", "绿"},
                                                {"yellow", "黄"},
                                                {"white", "白"}};
        
        // history plates at the bottom of screen 
        std::vector<cv::Mat> plates_his;
        int height_his = 100;
        int gap_his = 10;
        bool display_his = true;

    protected:
        virtual std::shared_ptr<vp_objects::vp_meta> handle_frame_meta(std::shared_ptr<vp_objects::vp_frame_meta> meta) override;
    public:
        vp_plate_osd_node(std::string node_name, std::string font = "", bool display_his = true);
        ~vp_plate_osd_node();
    };
}