HeuristicalTreeGenerator.h

Go to the documentation of this file.

#pragma once

// Global includes
#include <queue>

#include <boost/foreach.hpp>
#include <boost/shared_ptr.hpp>

// Local includes
#include "trainer/TreeNode.h"

#include "trainer/source/Object.h"
#include "trainer/source/ObjectSet.h"
#include "trainer/source/ObjectSetList.h"

#include "trainer/generator/AbstractGraphGenerator.h"
#include "trainer/generator/heuristic/AbstractHeuristic.h"
#include "trainer/generator/heuristic/DirectionRelationHeuristic.h"
#include <ISM/common_type/ObjectSet.hpp>


namespace SceneModel {
  
  class HeuristicalTreeGenerator : public AbstractGraphGenerator {
  public:

    HeuristicalTreeGenerator();
    
    ~HeuristicalTreeGenerator();
    
    void buildTree(ObjectSetList pObjectSets, boost::shared_ptr<TreeNode>& pRoot);
    
    void buildTree(ObjectSetList pTrajectories, boost::shared_ptr<TreeNode>& pRoot, std::string pType);
    
    void addHeuristic(boost::shared_ptr<AbstractHeuristic> pHeuristic);
    
    void evaluateHeuristics(std::vector<boost::shared_ptr<TreeNode> > pClusters);
    
    void evaluateHeuristics(std::vector<boost::shared_ptr<TreeNode> > pClusters,
                            boost::shared_ptr<TreeNode> pChild);
  private:
    
    void deleteEmptyObjectSets(ObjectSetList& pObjectSets);
    
    std::vector<boost::shared_ptr<AbstractHeuristic> > mHeuristics;
  };
}