KFParticle_Container.cc

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

/*****************/
/* Cameron Dean  */
/*   LANL 2020   */
/* cdean@bnl.gov */
/*****************/

// Ideas taken from SvtxTrackMap_v1 & TrkrClusterContainer

#include "KFParticle_Container.h"

#include <KFParticle.h>

#include <cstdlib>
#include <iterator>  // for reverse_iterator
#include <map>       // for _Rb_tree_const_iterator, _Rb_tree_iterator
#include <ostream>   // for operator<<, endl, ostream, basic_ostream, bas...
#include <utility>   // for pair, make_pair

KFParticle_Container::KFParticle_Container()
  : m_kfpmap()
{
}

KFParticle_Container::KFParticle_Container(const KFParticle_Container& kfparticlemap)
  : m_kfpmap()
{
  for (auto& iter : kfparticlemap)
  {
    KFParticle* particle = dynamic_cast<KFParticle*>(iter.second->Clone());
    m_kfpmap.insert(std::make_pair(particle->Id(), particle));
  }
}

KFParticle_Container& KFParticle_Container::operator=(const KFParticle_Container& kfparticlemap)
{
  Reset();
  for (auto& iter : kfparticlemap)
  {
    KFParticle* particle = dynamic_cast<KFParticle*>(iter.second->Clone());
    m_kfpmap.insert(std::make_pair(particle->Id(), particle));
  }
  return *this;
}

KFParticle_Container::~KFParticle_Container()
{
  Reset();
}

void KFParticle_Container::Reset()
{
  for (auto& iter : m_kfpmap)
  {
    KFParticle* particle = iter.second;
    delete particle;
  }
  m_kfpmap.clear();
}

void KFParticle_Container::identify(std::ostream& os) const
{
  os << "KFParticle_Container: size = " << m_kfpmap.size() << std::endl;
  return;
}

const KFParticle* KFParticle_Container::get(unsigned int id) const
{
  ConstIter iter = m_kfpmap.find(id);
  if (iter == m_kfpmap.end())
  {
    return nullptr;
  }
  return iter->second;
}

KFParticle* KFParticle_Container::get(unsigned int id)
{
  Iter iter = m_kfpmap.find(id);
  if (iter == m_kfpmap.end())
  {
    return nullptr;
  }
  return iter->second;
}

KFParticle* KFParticle_Container::insert(const KFParticle* particle)
{
  unsigned int index = 0;
  if (!m_kfpmap.empty())
  {
    index = m_kfpmap.rbegin()->first + 1;
  }
  m_kfpmap.insert(std::make_pair(index, dynamic_cast<KFParticle*>(particle->Clone())));
  // m_kfpmap[index]->SetId(index);
  return m_kfpmap[index];
}

KFParticle_Container::ConstIter
KFParticle_Container::addParticle(KFParticle* particle)
{
  return addParticleSpecifyKey(particle->Id(), particle);
}

KFParticle_Container::ConstIter
KFParticle_Container::addParticleSpecifyKey(unsigned int id, KFParticle* particle)
{
  auto ret = m_kfpmap.insert(std::make_pair(id, particle));
  if (!ret.second)
  {
    std::cout << "KFParticle_Container::AddParticleSpecifyKey: duplicate id: " << id << " exiting now" << std::endl;
    exit(1);
  }
  else
  {
    return ret.first;
  }
}

KFParticle_Container::Map
KFParticle_Container::returnParticlesByPDGid(int PDGid)
{
  Map requiredParticles;

  for (auto& iter : m_kfpmap)
  {
    if (iter.second->GetPDG() == PDGid)
    {
      requiredParticles.insert(std::make_pair(iter.first, iter.second));
    }
  }

  return requiredParticles;
}

size_t KFParticle_Container::erase(unsigned int key)
{
  delete m_kfpmap[key];
  return m_kfpmap.erase(key);
}