AbstractHeuristic.h

Go to the documentation of this file.

#pragma once

// Global includes
#include <boost/shared_ptr.hpp>

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

#include "trainer/source/ObjectSetList.h"

namespace SceneModel {
  
  class AbstractHeuristic {
  public:

    AbstractHeuristic(std::string pDescription);
    
    virtual ~AbstractHeuristic();

    virtual void apply(std::vector<boost::shared_ptr<TreeNode> > pNodes);
    
    virtual void apply(std::vector<boost::shared_ptr<TreeNode> > pNodes, boost::shared_ptr<TreeNode> pChild);
    
    virtual boost::shared_ptr<TreeNode> getBestCluster();
    
    virtual boost::shared_ptr<TreeNode> getBestParentNode();
    
    void reset();
    
    bool operator < (const AbstractHeuristic& heuristic) const;
    
  private:
    
    std::string mDescription;
    
  protected:
    double score;
    
    std::vector<boost::shared_ptr<TreeNode> > candidates;
  };
}