CMat.h

// Confusion matrix class   
// Written by Sergey G. Kosov in 2013 for Project X
#pragma once

#include "types.h"

namespace DirectGraphicalModels
{
    class CPriorEdge;

    // ================================ Confusion Matrix Class ================================
    class CCMat
    {
    public:
        DllExport CCMat(byte nStates);
        DllExport virtual ~CCMat(void);

        DllExport void  reset(void);
        DllExport void  save(const std::string &path, const std::string &name = std::string(), short idx = -1) const;
        DllExport void  load(const std::string &path, const std::string &name = std::string(), short idx = -1);
        DllExport void  estimate(const Mat &gt, const Mat &solution, const Mat &mask = Mat());
        DllExport void  estimate(byte gt, byte solution);
        
        DllExport Mat   getConfusionMatrix(void) const;
        DllExport float getAccuracy(void) const;    

//      Old version
//      DllExport void  print(char *fileName, int shiftBase);
//      DllExport void  getAccuracy(int shiftBase, double *base, double *occlusion, double *overall);


    private:
        CPriorEdge  * m_pConfusionMatrix;       

    //  DllExport void  saveFile(FILE *pFile) const {CPriorEdge::saveFile(pFile);} 
    //  DllExport void  loadFile(FILE *pFile) {CPriorEdge::loadFile(pFile);}        
    };
}