PHHepMCGenEventMap.h

Go to the documentation of this file.

#ifndef __PHHEPMCGENEVENTMAP_H__
#define __PHHEPMCGENEVENTMAP_H__

#include "PHHepMCGenEvent.h"

#include <phool/PHObject.h>
#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();
  PHHepMCGenEventMap(const PHHepMCGenEventMap& eventmap);
  PHHepMCGenEventMap& operator=(const PHHepMCGenEventMap& eventmap);

  virtual ~PHHepMCGenEventMap();

  void identify(std::ostream& os = std::cout) const;
  void Reset();
  int isValid() const { return 1; }
  PHHepMCGenEventMap* Clone(const char* newname = "") const { 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;

  ClassDef(PHHepMCGenEventMap, 4);
};

#endif