AbstractTrainer.h

Go to the documentation of this file.

#pragma once

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

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

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

#include "trainer/source/AbstractSource.h"
#include "trainer/source/ObjectSetList.h"

#include "trainer/generator/AbstractGraphGenerator.h"

namespace SceneModel {

  class AbstractTrainer
  {
  public:
    AbstractTrainer();
    
    ~AbstractTrainer();
    
    void loadTrajectoriesAndBuildTree();
    
    void loadTrajectoriesAndBuildTree(std::string pType);
    
    boost::shared_ptr<TreeNode> getTree();
    
  protected:
    
    boost::shared_ptr<TreeNode> root;
    
    boost::shared_ptr<AbstractSource> source;
    
    boost::shared_ptr<AbstractGraphGenerator> generator;
  };
}