OcvMatPainter.h

#pragma once

class StatusGroup;

class OcvMatPainter
    {
    public:
        static const int OcvNoRestore       = 1 << 0;
    protected:
        struct LEVEL {
            std::string key;
            cv::Mat     mBack, mMask;
            cv::Scalar  color;
            };
        cv::Scalar maskColor;
        std::vector<LEVEL> levelVec;
        StatusGroup statusGroup;
        cv::Size size;

        int             getLevelIndex           (const std::string &key);
        bool            checkBoundings          (cv::Point &pt);
        bool            checkBoundings          (RoiRect &rect);
        virtual void    doRestore               (int idx, cv::Mat &mat);
        virtual void    doBackup                (int idx, cv::Mat &mat);
        virtual void    doPaint                 (int idx, cv::Mat &mat);
    public:
                        OcvMatPainter           ();
                        ~OcvMatPainter          ();
        virtual void    reinitialize            (cv::Size &size);
        void            restore                 (cv::Mat &mat);
        void            backup                  (cv::Mat &mat);
        void            paint                   (cv::Mat &mat);
        void            restore                 (const std::string &key, cv::Mat &mat);
        void            backup                  (const std::string &key, cv::Mat &mat);
        void            paint                   (const std::string &key, cv::Mat &mat);
        bool            addLevel                (const std::string &key, uint8_t r, uint8_t g, uint8_t b);
        void            setStatus               (const std::string &key, StatusValue status);
        void            clearStatus             (const std::string &key, StatusValue status);
        void            setRestoreOn            (const std::string &key)        { clearStatus (key, OcvNoRestore); }
        void            setRestoreOff           (const std::string &key)        { setStatus (key, OcvNoRestore); }
        void            plotRoiRect             (const std::string &key, RoiRect &rect);
        void            plotPointVector         (const std::string &key, PointVector &pointVec, IntVector &mask, int maskValue = 1);
        void            plotContours            (const std::string &key, ContourVector &contour);
        void            plotContours            (const std::string &key, ContourVector &contour, int dx, int dy, int minPts = 0, int maxPts = INT_MAX);
        void            plotContours            (const std::string &key, DoubleContourSPtrVector &contour);
        void            plotContours            (const std::string &key, DoubleContourSPtrVector &contour, int dx, int dy, int minPts = 0, int maxPts = INT_MAX);
        void            plotMarkImageData       (const std::string &key, MarkImageData *lpMarkData); 
        void            plotMarkImageDataCross  (const std::string &key, MarkImageData *lpMarkData, bool b45deg = true);
        void            plotCircle              (const std::string &key, MarkImageData *lpMarkData, bool bFilled = false);
        void            plotCircle              (const std::string &key, cv::Point &center, int radius, bool bFilled = false);
        void            plotIris                (const std::string &key, MarkImageData *lpMarkData);

        void            plotCross               (const std::string &key, DblVec center, bool b45 = false, int rad = 8);
        void            plotPoint               (const std::string &key, DblVec center, int rad = 3);
    };