ImplicitShapeModel.hpp

Go to the documentation of this file.

#pragma once

#include <boost/shared_ptr.hpp>
#include <set>
#include <map>

#include "common_type/Pattern.hpp"
#include "common_type/VoteSpecifier.hpp"

namespace ISM
{

using std::endl;

struct ImplicitShapeModel
{
        std::map<std::string, PatternPtr> patternDefinitions;
        std::map<std::string, std::vector<VoteSpecifierPtr> > voteSpecifiersPerObject;
        std::set<std::string> objectTypes;

        ImplicitShapeModel() {};
        ImplicitShapeModel(std::map<std::string, PatternPtr> patternDefinitions,
                        std::map<std::string, std::vector<VoteSpecifierPtr> > voteSpecifiersPerObject,
                        std::set<std::string> objectTypes) : patternDefinitions(patternDefinitions),
        voteSpecifiersPerObject(voteSpecifiersPerObject),
        objectTypes(objectTypes) {}

        static bool sortVoteSpecPtrAscAlpha(const VoteSpecifierPtr& lhs, const VoteSpecifierPtr& rhs)
        {
                std::stringstream ssl;
                std::stringstream ssr;
                ssl << lhs;
                ssr << rhs;
                return ssl.str() < ssr.str();
        }

        std::string toString()
        {
                std::stringstream strm;
                strm << "PatternDefinitions:" << std::endl;
                for(auto& strPatPair : patternDefinitions)
                {
                        strm << strPatPair.first << " : " << strPatPair.second << endl;
                }
                strm << endl;

                strm << "VoteSpecifiersPerObject:" << std::endl;
                for(auto& strVsVecPair : voteSpecifiersPerObject)
                {
                        strm << strVsVecPair.first << " :" << endl;
                        std::vector<VoteSpecifierPtr> vs = strVsVecPair.second;
                        std::sort(vs.begin(), vs.end(), sortVoteSpecPtrAscAlpha);
                        for(auto& v : vs)
                        {
                                strm << v << endl;
                        }
                        strm << endl;
                }
                strm << endl;


                strm << "ObjectTypes:" << std::endl;
                for(auto& ot : objectTypes)
                {
                        strm << ot << endl;
                }
                strm << endl;

                return strm.str();
        }

}; typedef boost::shared_ptr<ImplicitShapeModel> IsmPtr;

}