TrkrClusterContainerv2.cc

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

#include "TrkrClusterContainerv2.h"
#include "TrkrCluster.h"
#include "TrkrDefs.h"

#include <cstdlib>

namespace
{
  TrkrClusterContainer::Map dummy_map;
}

//_________________________________________________________________
void TrkrClusterContainerv2::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++)
      {
        for (const auto& pair : m_clusmap[layer][phi_segment][z_segment])
        {
          delete pair.second;
        }

        m_clusmap[layer][phi_segment][z_segment].clear();
      }
    }
  }
}

//_________________________________________________________________
void TrkrClusterContainerv2::identify(std::ostream& os) const
{
  os << "-----TrkrClusterContainerv2-----" << std::endl;
  os << "Number of clusters: " << 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++)
      {
        const auto iter = m_clusmap[layer][phi_segment][z_segment].begin();
        const int flayer = TrkrDefs::getLayer(iter->first);
        const unsigned int fsector = TrkrDefs::getPhiElement(iter->first);
        const unsigned int fside = TrkrDefs::getZElement(iter->first);
        std::cout << "layer: " << layer << " | " << flayer
                  << " phi_seg: " << phi_segment << " | " << fsector
                  << " z_seg: " << z_segment << " | " << fside
                  << " nclu: " << m_clusmap[layer][phi_segment][z_segment].size()
                  << std::endl;

        for (const auto& pair : m_clusmap[layer][phi_segment][z_segment])
        {
          os << "clus key " << pair.first << " layer " << TrkrDefs::getLayer(pair.first) << std::endl;
          (pair.second)->identify();
        }
      }
    }
  }

  os << "------------------------------" << std::endl;
}

//_________________________________________________________________
void TrkrClusterContainerv2::removeCluster(TrkrDefs::cluskey key)
{
  unsigned int layer = TrkrDefs::getLayer(key);
  unsigned int sector = TrkrDefs::getPhiElement(key);
  unsigned int side = TrkrDefs::getZElement(key);

  // bound check
  if (layer < max_layer && sector < max_phisegment && side < max_zsegment)
  {
    m_clusmap[layer][sector][side].erase(key);
  }
  else
  {
    std::cout
        << "TrkrClusterContainerv2::removeCluster - out of range access."
        << " layer: " << layer
        << " sector: " << sector
        << " side: " << side
        << std::endl;
  }
}

//_________________________________________________________________
void TrkrClusterContainerv2::addClusterSpecifyKey(const TrkrDefs::cluskey key, TrkrCluster* newclus)
{
  unsigned int layer = TrkrDefs::getLayer(key);
  unsigned int sector = TrkrDefs::getPhiElement(key);
  unsigned int side = TrkrDefs::getZElement(key);

  // bound check
  if (layer < max_layer && sector < max_phisegment && side < max_zsegment)
  {
    const auto [iter, success] = m_clusmap[layer][sector][side].insert(std::make_pair(key, newclus));
    if (!success)
    {
      std::cout << "TrkrClusterContainerv2::AddClusterSpecifyKey: duplicate key: " << key << " exiting now" << std::endl;
      exit(1);
    }
  }
  else
  {
    std::cout
        << "TrkrClusterContainerv2::addClusterSpecifyKey - out of range access."
        << " layer: " << layer
        << " sector: " << sector
        << " side: " << side
        << std::endl;
  }
}

//_________________________________________________________________
TrkrClusterContainerv2::ConstRange
TrkrClusterContainerv2::getClusters(TrkrDefs::hitsetkey hitsetkey)
{
  const unsigned int layer = TrkrDefs::getLayer(hitsetkey);
  const unsigned int sector = TrkrDefs::getPhiElement(hitsetkey);
  const unsigned int side = TrkrDefs::getZElement(hitsetkey);

  // bound check
  if (layer < max_layer && sector < max_phisegment && side < max_zsegment)
  {
    return std::make_pair(m_clusmap[layer][sector][side].cbegin(), m_clusmap[layer][sector][side].cend());
  }
  else
  {
    std::cout
        << "TrkrClusterContainerv2::getClusters - out of range access."
        << " layer: " << layer
        << " sector: " << sector
        << " side: " << side
        << std::endl;

    return std::make_pair(dummy_map.cbegin(), dummy_map.cend());
  }
}

//_________________________________________________________________
TrkrCluster* TrkrClusterContainerv2::findCluster(TrkrDefs::cluskey key) const
{
  const unsigned int layer = TrkrDefs::getLayer(key);
  const unsigned int sector = TrkrDefs::getPhiElement(key);
  const unsigned int side = TrkrDefs::getZElement(key);

  // bound check
  if (layer < max_layer && sector < max_phisegment && side < max_zsegment)
  {
    auto it = m_clusmap[layer][sector][side].find(key);
    if (it != m_clusmap[layer][sector][side].end())
    {
      return it->second;
    }
    else
    {
      return nullptr;
    }
  }
  else
  {
    std::cout
        << "TrkrClusterContainerv2::findOrAddCluster - out of range access."
        << " layer: " << layer
        << " sector: " << sector
        << " side: " << side
        << std::endl;
    return nullptr;
  }
}

//_________________________________________________________________
unsigned int TrkrClusterContainerv2::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_clusmap[layer][phi_segment][z_segment].size();
      }
    }
  }

  return size;
}