OcmModel.h

Go to the documentation of this file.

#pragma once

// Global includes
#include <vector>

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

#include <trainer/TreeNode.h>
#include <trainer/source/ObjectSet.h>

#include <visualization/psm/ProbabilisticPrimarySceneObjectVisualization.h>

// Local includes
#include "learner/foreground/ocm/ocm/OcmTree.h"

#include "learner/foreground/ocm/ocm/shape/GaussianMixtureModel.h"

#include "helper/ProbabilityTable.h"
#include "helper/MappedProbabilityTable.h"

namespace ProbabilisticSceneRecognition {
  
  class OcmModel {
  public:
    
    OcmModel(const boost::shared_ptr<SceneModel::TreeNode> pRoot);
    
    ~OcmModel();
    
    void initializeVisualizer(boost::shared_ptr<Visualization::ProbabilisticPrimarySceneObjectVisualization> mSuperior);
    
    void save(double pWorkspaceVolume, boost::property_tree::ptree& pPt);
    
    void saveShape(double pWorkspaceVolume, boost::property_tree::ptree& pPt);
    
    void saveAppearance(boost::property_tree::ptree& pPt);
    
    void saveOcclusion(boost::property_tree::ptree& pPt);
    
    unsigned int getNumberOfSlots();
    
  public:
    
    boost::shared_ptr<OcmTree> mRoot;
    
    boost::shared_ptr<MappedProbabilityTable> mAppearanceTable;
    
    boost::shared_ptr<MappedProbabilityTable> mHypothesisTable;
    
    boost::shared_ptr<ProbabilityTable> mOcclusionTable;
  };
}