MarkerHistogram.h

// Histogram Marker Class 
// Written by Sergey Kosov in 2015 for Project X
#pragma once

#include "Marker.h"

namespace DirectGraphicalModels 
{ 
    class CTrainNode;
    namespace vis
    {
    
    // ================================ Histogram Marker Class ================================
    class CMarkerHistogram : public CMarker
    {
    public:
        DllExport CMarkerHistogram(const CTrainNode& nodeTrainer, default_pallete palette = DEF_PALETTE_12, vec_string_t vFeatureNames = vec_string_t())
            : CMarker(palette)
            , m_nodeTrainer(nodeTrainer)
            , m_vFeatureNames(vFeatureNames) 
        {}
        DllExport CMarkerHistogram(const CTrainNode& nodeTrainer, const vec_nColor_t& vPalette, vec_string_t vFeatureNames = vec_string_t())
            : CMarker(vPalette)
            , m_nodeTrainer(nodeTrainer)
            , m_vFeatureNames(vFeatureNames) 
        {}
        DllExport virtual ~CMarkerHistogram(void) = default;

        DllExport Mat   drawHistogram(void) const { return drawHistogram(CV_RGB(0, 0, 0)); }
        DllExport Mat   drawHistogram2D(void) const { return drawHistogram2D(CV_RGB(0, 0, 0)); }
        DllExport Mat   drawClassificationMap2D(float Z) const;
        DllExport void  showHistogram(void);
        DllExport void  close(void) const;


    private:
        DllExport Mat   drawHistogram(Scalar color) const;
        DllExport Mat   drawHistogram2D(Scalar color) const;
        int             getActiveState(Scalar color) const;
        Mat             drawFeatureHistogram(word f, int activeState = -1) const;
        Mat             drawFeatureHistogram2D(word f, int activeState = -1) const;
        Mat             drawLegend(int maxHeight, int activeState = -1) const;
    

    protected:
        const CTrainNode&           m_nodeTrainer;      


    private:
        static const Size           margin;
        static const byte           bkgIntencity;
        static const double         frgWeight;
        static const std::string    wndName;

        vec_string_t                m_vFeatureNames;
    };

} }