TrkrClusterHitAssocv2.cc

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

#include "TrkrClusterHitAssocv2.h"
#include "TrkrDefs.h"

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

namespace
{
  TrkrClusterHitAssocv2::Map dummy_map;
}

//_________________________________________________________________________
void TrkrClusterHitAssocv2::Reset()
{
  for(unsigned int layer = 0;layer < max_layer; layer++)
    for(unsigned int phi_segment = 0;phi_segment < max_phisegment;phi_segment++)
    for(unsigned int z_segment = 0; z_segment < max_zsegment; z_segment++)
  { m_map[layer][phi_segment][z_segment].clear(); }
}

//_________________________________________________________________________
void TrkrClusterHitAssocv2::identify(std::ostream &os) const
{
  std::multimap<TrkrDefs::cluskey, TrkrDefs::hitkey>::const_iterator iter;
  os << "-----TrkrClusterHitAssocv2-----" << std::endl;
  os << "Number of associations: " << size() << std::endl;
  for(unsigned int layer = 0;layer < max_layer; layer++)
    for(unsigned int phi_segment = 0;phi_segment < max_phisegment;phi_segment++)
    for(unsigned int z_segment = 0; z_segment < max_zsegment; z_segment++)
  {
    for( const auto& pair:m_map[layer][phi_segment][z_segment] )
    {
      int layer = TrkrDefs::getLayer(pair.first);
      os << "clus key " << pair.first << std::dec
        << " layer " << layer
        << " hit key: " << pair.second << std::endl;
    }
  }
  os << "------------------------------" << std::endl;

  return;

}

//_________________________________________________________________________
void TrkrClusterHitAssocv2::addAssoc(TrkrDefs::cluskey ckey, unsigned int hidx)
{
  unsigned int layer = TrkrDefs::getLayer(ckey);
  unsigned int sector = TrkrDefs::getPhiElement(ckey);
  unsigned int side = TrkrDefs::getZElement(ckey);

  // bound check
  if( layer < max_layer && sector < max_phisegment && side < max_zsegment )
  {
    m_map[layer][sector][side].insert(std::make_pair(ckey, hidx));
  } else {
    std::cout
      << "TrkrClusterHitAssocv2::addAssoc - out of range access."
      << " layer: " << layer
      << " sector: " << sector
      << " side: " << side
      << std::endl;
  }
}

//_________________________________________________________________________
TrkrClusterHitAssocv2::Map* TrkrClusterHitAssocv2::getClusterMap(TrkrDefs::hitsetkey hitsetkey)
{
  unsigned int layer = TrkrDefs::getLayer(hitsetkey);
  unsigned int sector = TrkrDefs::getPhiElement(hitsetkey);
  unsigned int side = TrkrDefs::getZElement(hitsetkey);

  // bound check
  if( layer < max_layer && sector < max_phisegment && side < max_zsegment )
  {
     return &m_map[layer][sector][side];
  } else {
    std::cout
      << "TrkrClusterHitAssocv2::getClusterMap - out of range access."
      << " layer: " << layer
      << " sector: " << sector
      << " side: " << side
      << std::endl;
    return nullptr;
  }
}

//_________________________________________________________________________
TrkrClusterHitAssocv2::ConstRange TrkrClusterHitAssocv2::getHits(TrkrDefs::cluskey ckey)
{
  unsigned int layer = TrkrDefs::getLayer(ckey);
  unsigned int sector = TrkrDefs::getPhiElement(ckey);
  unsigned int side = TrkrDefs::getZElement(ckey);

  // bound check
  if( layer < max_layer && sector < max_phisegment && side < max_zsegment )
  {
    return std::make_pair( m_map[layer][sector][side].lower_bound(ckey), m_map[layer][sector][side].upper_bound(ckey) );
  } else {
    std::cout
      << "TrkrClusterHitAssocv2::getHits - out of range access."
      << " layer: " << layer
      << " sector: " << sector
      << " side: " << side
      << std::endl;
    return std::make_pair(dummy_map.cbegin(), dummy_map.cend() );
  }
}

//_________________________________________________________________________
unsigned int TrkrClusterHitAssocv2::size() const
{
  unsigned int size = 0;
  for(unsigned layer = 0;layer < max_layer; layer++)
    for(unsigned phi_segment = 0;phi_segment < max_phisegment;phi_segment++)
    for(unsigned z_segment = 0; z_segment < max_zsegment; z_segment++)
  { size += m_map[layer][phi_segment][z_segment].size(); }

  return size;
}