Topology.h

Go to the documentation of this file.

#pragma once

#include <boost/shared_ptr.hpp>

#include "topology_creator/Relation.h"
#include "trainer/TreeNode.h"

namespace SceneModel
{

// similar to lib_ism:
class Topology
{
public:
    Topology();

    ~Topology();

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

    std::string mIdentifier;

    bool mUsedInOptimization;

    void setEvaluationResult(double pAverageRecognitionRuntime, double pFalsePositives, double pFalseNegatives);

    double getAverageRecognitionRuntime() const;

    double getFalsePositives() const;

    double getFalseNegatives() const;

    bool isEvaluated() const;

    void setCost(double pCost);

    double getCost() const;

    bool isCostValid() const;

    void setTree(boost::shared_ptr<TreeNode> pTree);

    boost::shared_ptr<TreeNode> getTree();

    bool isTreeValid() const;

private:
    bool mEvaluated;

    unsigned int mFalsePositives;

    double mAverageRecognitionRuntime;

    unsigned int mFalseNegatives;

    bool mCostValid;

    double mCost;

    boost::shared_ptr<TreeNode> mTree;

};

}