TopologyTreeGenerator.h

Go to the documentation of this file.

#pragma once

#include "topology_creator/Relation.h"
#include "trainer/TreeNode.h"
#include "trainer/generator/AbstractGraphGenerator.h"
#include "trainer/source/ObjectSetList.h"

namespace SceneModel {

class TopologyTreeGenerator: public AbstractGraphGenerator
{

public:
    TopologyTreeGenerator();

    ~TopologyTreeGenerator();

    void buildTree(ObjectSetList pObjectSets, boost::shared_ptr<TreeNode>& pRoot);

    void buildTree(ObjectSetList pTrajectories, boost::shared_ptr<TreeNode>& pRoot, std::string pType);

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

private:
    std::vector<boost::shared_ptr<Relation>> mRelations;

};

}