TrkrClusterContainerv1.cc

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

#include "TrkrClusterContainerv1.h"
#include "TrkrCluster.h"

#include <cstdlib>

void TrkrClusterContainerv1::Reset()
{
  while (m_clusmap.begin() != m_clusmap.end())
  {
    delete m_clusmap.begin()->second;
    m_clusmap.erase(m_clusmap.begin());
  }
  return;
}

void TrkrClusterContainerv1::identify(std::ostream& os) const
{
  os << "-----TrkrClusterContainerv1-----" << std::endl;
  ConstIterator iter;
  os << "Number of clusters: " << size() << std::endl;
  for (iter = m_clusmap.begin(); iter != m_clusmap.end(); ++iter)
  {
    int layer = TrkrDefs::getLayer(iter->first);
    os << "clus key " << iter->first << " layer " << layer << std::endl;
    (iter->second)->identify();
  }
  os << "------------------------------" << std::endl;
  return;
}

void TrkrClusterContainerv1::addClusterSpecifyKey(const TrkrDefs::cluskey key, TrkrCluster* newclus)
{
  const auto [iter, success] = m_clusmap.insert(std::make_pair(key, newclus));
  if (!success)
  {
    std::cout << "TrkrClusterContainerv1::AddClusterSpecifyKey: duplicate key: " << key << " exiting now" << std::endl;
    exit(1);
  }
}

void TrkrClusterContainerv1::removeCluster(TrkrDefs::cluskey key)
{
  m_clusmap.erase(key);
}

TrkrClusterContainer::ConstRange
TrkrClusterContainerv1::getClusters() const
{
  return std::make_pair(m_clusmap.cbegin(), m_clusmap.cend());
}

TrkrCluster*
TrkrClusterContainerv1::findCluster(TrkrDefs::cluskey key) const
{
  auto it = m_clusmap.find(key);
  return it == m_clusmap.end() ? nullptr : it->second;
}

unsigned int TrkrClusterContainerv1::size() const
{
  return m_clusmap.size();
}