CombinatorialGraphGenerator.h

Go to the documentation of this file.

#pragma once

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

#include <trainer/TreeNode.h>
#include <trainer/AbstractTrainer.h>

#include <trainer/source/AbstractSource.h>
#include <trainer/source/PbdSceneGraphSource.h>

#include <trainer/generator/AbstractGraphGenerator.h>

// Local includes
#include "learner/foreground/ocm/combinatorial_optimization/CombinatorialOptimizer.h"

namespace ProbabilisticSceneRecognition {

class CombinatorialGraphGenerator: public SceneModel::AbstractGraphGenerator
{
public:
    CombinatorialGraphGenerator(std::vector<boost::shared_ptr<SceneObjectLearner>> pLearners,
                                std::vector<std::string> pObjectTypes);

    ~CombinatorialGraphGenerator();

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

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

private:

    std::vector<boost::shared_ptr<SceneObjectLearner>> mLearners;

    std::vector<std::string> mObjectTypes;
};

}