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 <map>
#include <utility>  // for pair

class KFParticle;

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

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

  void identify(std::ostream& os = std::cout) const override;
  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