KFParticle_Container.h

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

#ifndef KFPARTICLESPHENIX_KFPARTICLECONTAINER_H
#define KFPARTICLESPHENIX_KFPARTICLECONTAINER_H

#include <phool/PHObject.h>

#include <cstddef>   // for size_t
#include <iostream>  // for cout, ostream
#include <iterator>
#include <map>

class KFParticle;

// using ConstIter = Map::const_iterator;
// typedef std::map<int, KFParticle*>::const_iterator ConstIter;

class KFParticle_Container : public PHObject
{
 public:
  using Map = std::map<unsigned int, KFParticle*>;
  using ConstIter = Map::const_iterator;
  using Iter = Map::iterator;

  KFParticle_Container();
  KFParticle_Container(const KFParticle_Container& kfparticlemap);
  KFParticle_Container& operator=(const KFParticle_Container& kfparticlemap);
  ~KFParticle_Container() override;

  void identify(std::ostream& os = std::cout) const override;
  // cppcheck-suppress [virtualCallInConstructor]
  void Reset() override;
  int isValid() const override { return 1; }
  PHObject* CloneMe() const override { return new KFParticle_Container(*this); }

  bool empty() const { return m_kfpmap.empty(); }
  size_t size() const { return m_kfpmap.size(); }
  size_t count(unsigned int key) const { return m_kfpmap.count(key); }
  void clear() { Reset(); }

  const KFParticle* get(unsigned int key) const;
  KFParticle* get(unsigned int key);

  ConstIter begin() const { return m_kfpmap.begin(); }
  ConstIter find(unsigned int key) const { return m_kfpmap.find(key); }
  ConstIter end() const { return m_kfpmap.end(); }

  Iter begin() { return m_kfpmap.begin(); }
  Iter find(unsigned int key) { return m_kfpmap.find(key); }
  Iter end() { return m_kfpmap.end(); }

  KFParticle* insert(const KFParticle* particle);
  ConstIter addParticle(KFParticle* particle);
  ConstIter addParticleSpecifyKey(unsigned int key, KFParticle* particle);

  Map returnParticlesByPDGid(int PDGid);

  size_t erase(unsigned int key);

 private:
  Map m_kfpmap;

  ClassDefOverride(KFParticle_Container, 1)
};

#endif  // KFPARTICLESPHENIX_KFPARTICLECONTAINER_H