Marker.h

// Marker Class 
// Written by Sergey Kosov in 2008 - 2010 for MPII
// Adopted by Sergey Kosov in 2012 for Project X
// Modified by Sergey Kosov in 2015 for Project X
#pragma once

#include "types.h"

namespace DirectGraphicalModels { namespace vis
{
    // Default paletes
    const static Scalar colors24[] =  
    { 
        CV_RGB(255, 0,   0),        // Red
        CV_RGB(255, 128, 0),        // Orange
        CV_RGB(255, 255, 0),        // Yellow
        CV_RGB(128, 255, 0),        // Chartreuse
        CV_RGB(0,   255, 0),        // Green
        CV_RGB(0,   255, 128),      // Spring green
        CV_RGB(0,   255, 255),      // Cyan
        CV_RGB(0,   128, 255),      // 
        CV_RGB(0,   0,   255),      // Blue
        CV_RGB(128, 0,   255),      // 
        CV_RGB(255, 0,   255),      // Purple
        CV_RGB(255, 0,   128),      // 

        CV_RGB(128, 0,   0),        // Dark Red
        CV_RGB(128, 64,  0),        // Dark Orange
        CV_RGB(128, 128, 0),        // Dark Yellow
        CV_RGB(64,  128, 0),        // Dark Chartreuse
        CV_RGB(0,   128, 0),        // Dark Green
        CV_RGB(0,   128, 64),       // Dark Spring green
        CV_RGB(0,   128, 128),      // Dark Cyan
        CV_RGB(0,   64,  128),      // 
        CV_RGB(0,   0,   128),      // Dark Blue
        CV_RGB(64,  0,   128),      // 
        CV_RGB(128, 0,   128),      // Dark Purple
        CV_RGB(128, 0,   64)        // 
    }; 

    enum default_pallete {
        DEF_PALETTE_3,      
        DEF_PALETTE_3_INV,  
        DEF_PALETTE_6,      
        DEF_PALETTE_6_INV,  
        DEF_PALETTE_12,     
        DEF_PALETTE_12_INV, 
        DEF_PALETTE_24,     
        DEF_PALETTE_24_INV, 
        DEF_PALETTE_24_M,   
        DEF_PALETTE_36,     
        DEF_PALETTE_36_INV, 
        DEF_PALETTE_72,     
        DEF_PALETTE_72_INV  
    };
    enum mark_flags {
        MARK_GRID       = 1,    
        MARK_OVER       = 2,    
        MARK_NO_ZERO    = 4,    
        MARK_BW         = 8,    
        MARK_PERCLASS   = 16,   
        MARK_PERCENT    = 128   
    };
    

    // ================================ Marker Class ================================
    class CMarker
    {
        enum textProp
        {
            TP_CENTER   = 0,
            TP_LEFT     = 1,
            TP_RIGHT    = 2,
            TP_TOP      = 4,
            TP_BOTTOM   = 8,

            TP_PERCENT  = 128       // This flag may be mixed with the <mark_flags>
        };
    
    public:
        DllExport CMarker(default_pallete palette = DEF_PALETTE_12);
        DllExport CMarker(const vec_nColor_t &vPalette);
        DllExport virtual ~CMarker(void);
    
        DllExport void          markClasses(Mat &base, const Mat &classes, byte flag = 0) const;                // Does nothing on error
        DllExport Mat           drawPotentials(const Mat &potential, byte flag = 0) const;
        DllExport Mat           drawConfusionMatrix(const Mat &confusionMat, byte flag = 0) const;
    
    
    protected:
        vec_nColor_t            m_vPalette;                     


    private:
        Mat drawVector(const Mat &potential, byte flag) const;
        Mat drawMatrix(const Mat &potential, byte flag) const;
        Mat drawVoxel (const Mat &potential, byte flag) const;  // WARNING: not implemented

        template<typename T> 
        void drawSquare(Mat &img, byte x, byte y, const Scalar &color, T val, double fontScale = 1.0, byte textProp = 0) const;
        void drawRectangle(Mat &img, Point pt1, Point pt2, const Scalar &color, float val, double fontScale = 1.0, byte textProp = TP_CENTER) const;
        void drawRectangle(Mat &img, Point pt1, Point pt2, const Scalar &color, const std::string &str = std::string(), double fontScale = 1.0, byte textProp = TP_CENTER) const;


    private:
        static const byte   bkgIntencity;
        static const byte   frgIntensity;
        static const int    ds;                             // size of the cell
            

    private:
        // Copy semantics are disabled
        CMarker(const CMarker&) = delete;
        const CMarker& operator=(const CMarker&) = delete;
    };



    DllExport Mat drawDictionary(const Mat &dictionary, double m = 1);

} }