TopologyTreeTrainer.h

Go to the documentation of this file.

#pragma once

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

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

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

#include "trainer/generator/AbstractGraphGenerator.h"
#include "trainer/generator/topology_tree/TopologyTreeGenerator.h"

namespace SceneModel {

class TopologyTreeTrainer: public AbstractTrainer
{
public:

  TopologyTreeTrainer(std::vector<boost::shared_ptr<Relation>> pRelations);

  ~TopologyTreeTrainer();

  void addSceneGraphMessages(std::vector<ISM::ObjectSetPtr> pMessages);

private:

  boost::shared_ptr<PbdSceneGraphSource> pbdSource;
};

}