ManuallyDefPseudoHeuristic.cpp

Go to the documentation of this file.

#include "ManuallyDefPseudoHeuristic.hpp"
#include <boost/lexical_cast.hpp>
namespace ISM
{

  ManuallyDefPseudoHeuristic::ManuallyDefPseudoHeuristic(std::vector<std::pair<std::vector<ClusterObject>, uint16_t>>
                  clusters) : Heuristic("ManuallyDefPseudoHeuristic"), mClusters(clusters)
  {

  }
  ManuallyDefPseudoHeuristic::~ManuallyDefPseudoHeuristic()
  {
  }
  void ManuallyDefPseudoHeuristic::applyHeuristic(const TracksPtr& tracks)
  {
          std::vector<TrackPtr> result;
      for(std::pair<std::vector<ClusterObject>, uint16_t>& clusterObjectsToCluster : mClusters)
      {
          for(ClusterObject& clusterObject : clusterObjectsToCluster.first)
          {
                  bool found = false;
              for(TrackPtr& track : tracks->tracks)
              {
                  if(clusterObject.isCluster)
                  {
                          if(track->type.find("_sub") != std::string::npos)
                          {
                                  std::string temp = track->type;
                                  temp.erase(temp.begin(), temp.begin() + temp.find_first_of("_sub") + 4);
                                  if(boost::get<uint16_t>(clusterObject.mObjectOrCluster) ==
                                                  boost::lexical_cast<uint16_t>(temp))
                                  {
                                          found = true;
                                          result.push_back(track);
                                  }
                          }
                  } else if(track->type == boost::get<std::string>(clusterObject.mObjectOrCluster))
                  {
                          found = true;
                          result.push_back(track);
                  }
              }
              if(!found)
              {
                  result.clear();
                  break;
              }
          }
          if(result.size() == clusterObjectsToCluster.first.size())
          {
              cluster = TracksPtr(new Tracks(result));
                  confidence = 1.0;
                  break;
          }
      }
  }
}