Cluster.hpp

Go to the documentation of this file.

#pragma once

#include "heuristic_trainer/DataCollector.hpp"
#include "ObjectRelation.hpp"
#include <boost/shared_ptr.hpp>

namespace ISM
{
class Cluster
{
        public:
                Cluster(const std::pair<TrackPtr, boost::shared_ptr<Cluster> >& referenceWithParent,
                                ObjectRelations& topology, const std::vector<std::pair<int, TrackPtr> >& mostCommonObjects,
                                std::map<TrackPtr, int>& objectOccurences);

                TracksPtr getChildren()
                {
                        return children;
                }

                TrackPtr getReference()
                {
                        return reference;
                }

                std::vector<VoteSpecifierPtr> getVotersByVotersTypeAndObservedId(const std::string& type, const std::string& observedId) const;
                void recommendLevel(unsigned level);
                unsigned getRecommendedLevel();

        private:
                TrackPtr reference;
                TracksPtr children;
                std::map<std::string, std::map<std::string, std::vector<VoteSpecifierPtr> > > votesByVotersTypeAndObservedId;
                unsigned recommendedLevel;

}; typedef boost::shared_ptr<Cluster> ClusterPtr;
}