TrkrClusterHitAssocv3.cc

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file TrkrClusterHitAssocv3.cc

#include "TrkrClusterHitAssocv3.h"
#include "TrkrDefs.h"

#include <ostream>  // for operator<<, endl, basic_ostream, ostream, basic_o...

namespace
{
  TrkrClusterHitAssocv3::Map dummy_map;
}

//_________________________________________________________________________
void TrkrClusterHitAssocv3::Reset()
{
  std::map<TrkrDefs::hitsetkey, Map> empty_map;
  m_map.swap(empty_map);
}
//_________________________________________________________________________
void TrkrClusterHitAssocv3::identify(std::ostream& os) const
{
  std::multimap<TrkrDefs::cluskey, TrkrDefs::hitkey>::const_iterator iter;
  os << "-----TrkrClusterHitAssocv3-----" << std::endl;
  os << "Number of associations: " << size() << std::endl;
  for (const auto& map_pair : m_map)
  {
    for (const auto& pair : map_pair.second)
    {
      os << "clus key " << pair.first << std::dec
         << " layer " << (unsigned int) TrkrDefs::getLayer(pair.first)
         << " hit key: " << pair.second << std::endl;
    }
  }
  os << "------------------------------" << std::endl;

  return;
}

//_________________________________________________________________________
void TrkrClusterHitAssocv3::addAssoc(TrkrDefs::cluskey ckey, unsigned int hidx)
{
  // get hitset key from cluster
  const TrkrDefs::hitsetkey hitsetkey = TrkrDefs::getHitSetKeyFromClusKey(ckey);

  // find relevant association map, create one if not found
  Map& clusterMap = m_map[hitsetkey];

  // insert association
  clusterMap.insert(std::make_pair(ckey, hidx));
}

//_________________________________________________________________________
TrkrClusterHitAssocv3::Map* TrkrClusterHitAssocv3::getClusterMap(TrkrDefs::hitsetkey hitsetkey)
{
  // find relevant association map, create one if not found, and return adress
  return &m_map[hitsetkey];
}

//_________________________________________________________________________
TrkrClusterHitAssocv3::ConstRange TrkrClusterHitAssocv3::getHits(TrkrDefs::cluskey ckey)
{
  // get hitset key from cluster
  const TrkrDefs::hitsetkey hitsetkey = TrkrDefs::getHitSetKeyFromClusKey(ckey);

  // find relevant association map, create one if not found
  const auto iter = m_map.find(hitsetkey);
  if (iter != m_map.end())
  {
    return std::make_pair(iter->second.lower_bound(ckey), iter->second.upper_bound(ckey));
  }
  else
  {
    return std::make_pair(dummy_map.cbegin(), dummy_map.cend());
  }
}

//_________________________________________________________________________
unsigned int TrkrClusterHitAssocv3::size() const
{
  unsigned int size = 0;
  for (const auto& map_pair : m_map)
  {
    size += map_pair.second.size();
  }

  return size;
}