PHHepMCGenEventMap.h

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

#ifndef PHHEPMC_PHHEPMCGENEVENTMAP_H
#define PHHEPMC_PHHEPMCGENEVENTMAP_H

#include "PHHepMCGenEvent.h"

#include <phool/PHObject.h>

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

class PHHepMCGenEventMap : public PHObject
{
 public:
  typedef std::map<int, PHHepMCGenEvent*> HepMCGenEventMap;
  typedef std::map<int, PHHepMCGenEvent*>::const_iterator ConstIter;
  typedef std::map<int, PHHepMCGenEvent*>::iterator Iter;
  typedef std::map<int, PHHepMCGenEvent*>::const_reverse_iterator ConstReverseIter;
  typedef std::map<int, PHHepMCGenEvent*>::reverse_iterator ReverseIter;

  PHHepMCGenEventMap() = default;
  PHHepMCGenEventMap(const PHHepMCGenEventMap& eventmap);
  PHHepMCGenEventMap& operator=(const PHHepMCGenEventMap& eventmap);

  ~PHHepMCGenEventMap() override;

  void identify(std::ostream& os = std::cout) const override;
// cppcheck-suppress virtualCallInConstructor
  void Reset() override;
  int isValid() const override { return 1; }
  PHHepMCGenEventMap* CloneMe() const override { return new PHHepMCGenEventMap(*this); }
  bool empty() const { return _map.empty(); }
  size_t size() const { return _map.size(); }
  size_t count(int idkey) const { return _map.count(idkey); }
  void clear() { Reset(); }
  const PHHepMCGenEvent* get(int idkey) const;
  PHHepMCGenEvent* get(int idkey);

  PHHepMCGenEvent* insert(const PHHepMCGenEvent* event) { return insert_active_event(event); }
  PHHepMCGenEvent* insert_active_event(const PHHepMCGenEvent* event = nullptr);
  PHHepMCGenEvent* insert_background_event(const PHHepMCGenEvent* event = nullptr);
  PHHepMCGenEvent* insert_event(const int embedding_id, const PHHepMCGenEvent* event = nullptr);

  size_t erase(int idkey)
  {
    delete _map[idkey];
    return _map.erase(idkey);
  }

  ConstIter find(unsigned int idkey) const { return _map.find(idkey); }
  Iter find(int idkey) { return _map.find(idkey); }
  ConstIter begin() const { return _map.begin(); }
  ConstIter end() const { return _map.end(); }
  Iter begin() { return _map.begin(); }
  Iter end() { return _map.end(); }
  ConstReverseIter rbegin() const { return _map.rbegin(); }
  ConstReverseIter rend() const { return _map.rend(); }
  ReverseIter rbegin() { return _map.rbegin(); }
  ReverseIter rend() { return _map.rend(); }
  //
  const HepMCGenEventMap& get_map() const { return _map; }
  HepMCGenEventMap& get_map() { return _map; }

 private:
  HepMCGenEventMap _map;

  ClassDefOverride(PHHepMCGenEventMap, 4);
};

#endif