TopologyAdapter.h

Go to the documentation of this file.

#pragma once

#include <topology_creator/Topology.h>
#include <ISM/combinatorial_trainer/Topology.hpp>
#include <ISM/combinatorial_optimization/CostFunction.hpp>

namespace ProbabilisticSceneRecognition
{

class TopologyAdapter
{
private:
    unsigned int mPSMTopologyIndex;
    unsigned int mISMTopologyIndex;
    std::vector<SceneModel::Relation> mRelations;
    std::string mSceneId;

public:
    class CostFunction: public ISM::CostFunction<ISM::TopologyPtr> {
    public:
        double calculateCost(ISM::TopologyPtr instance);
    };

    TopologyAdapter(std::vector<std::string> pObjectTypes, std::string pSceneId);

    ~TopologyAdapter();

    ISM::TopologyPtr psmToIsm(boost::shared_ptr<SceneModel::Topology> pPsmTopology);

    boost::shared_ptr<SceneModel::Topology> ismToPsm(ISM::TopologyPtr pIsmTopology);
};

}