DotHelper.hpp

Go to the documentation of this file.

#pragma once

#include <boost/filesystem/path.hpp>
#include <map>

//ISM includes
#include "common_type/VoteSpecifier.hpp"
#include "common_type/Tracks.hpp"
#include "combinatorial_trainer/ObjectRelation.hpp"

namespace ISM {

using boost::filesystem::path;

class DotHelper {

        public:
                DotHelper(std::map<std::string, ISM::TracksPtr>& tracksPerPattern) :
                        mTracksPerPattern(tracksPerPattern) {}

                void storeISMToDot(const path outputPath, const std::string& patternName,
                                const std::string& topologyIdentifier,
                                const std::map<std::string, std::vector<ISM::VoteSpecifierPtr>>& objectDefinitons);
                void storeRelationsToDot(const path outputPath, const std::string& patternName,
                                const std::string& topologyIdentifier,
                                const ISM::ObjectRelations& relations,
                                ISM::ObjectRelations referenceRelations = ObjectRelations());

        private:
                std::map<std::string, ISM::TracksPtr> mTracksPerPattern;
                const double NODE_DISTANCE = 2.5;

}; typedef boost::shared_ptr<DotHelper> DotHelperPtr;

}