MappedProbabilityTable.h

Go to the documentation of this file.

#pragma once

// Global includes
#include <map>
#include <string>
#include <vector>

// Package includes
#include <boost/foreach.hpp>
#include <boost/property_tree/ptree.hpp>

// Local includes
#include "helper/ProbabilityTable.h"
#include "helper/SerializationHelper.h"

namespace ProbabilisticSceneRecognition {
  
  class MappedProbabilityTable {
  public:

    MappedProbabilityTable();
    
    MappedProbabilityTable(boost::property_tree::ptree& pPt);
    
    ~MappedProbabilityTable();
    
    void load(boost::property_tree::ptree& pPt);
    
    void save(boost::property_tree::ptree& pPt); 
    
    void initializeTable(unsigned int pRows);
    
    double getProbability(unsigned int pRow, std::string pType);
    
    void add(std::string pType);
    
    void add(unsigned int pRow, std::string pType);
    
    void setDefaultClassCounter(unsigned int pRow, double pCount);
    
    void normalize();
    
    unsigned int getNumberOfColumns();
    
    unsigned int getNumberOfRows();
    
  private:
    
    unsigned int getIndex(std::string pType);
    
    boost::shared_ptr<ProbabilityTable> mTable;
    
    std::map<std::string, unsigned int> mMappingTypeToIndice;
  };
}