HepMCFileReader.h

Go to the documentation of this file.

#ifndef Input_HepMCFileReader_h
#define Input_HepMCFileReader_h
 
#include <vector>
#include <map>
 
#include "FWCore/Utilities/interface/get_underlying_safe.h"
 
namespace HepMC {
  class IO_BaseClass;
  class GenEvent;
  class GenParticle;
}  // namespace HepMC
 
class HepMCFileReader {
protected:
  HepMCFileReader();
 
public:
  virtual ~HepMCFileReader();
  virtual void initialize(const std::string &filename);
  inline bool isInitialized() const;
 
  virtual bool setEvent(int event);
  virtual bool readCurrentEvent();
  virtual bool printHepMcEvent() const;
  HepMC::GenEvent *fillCurrentEventData();
  //  virtual bool fillEventData(HepMC::GenEvent *event);
  // this method prints the event information as
  // obtained by the input file in HepEvt style
  void printEvent() const;
  // get all the 'integer' properties of a particle
  // like mother, daughter, pid and status
  // 'j' is the number of the particle in the HepMc
  virtual void getStatsFromTuple(int &mo1, int &mo2, int &da1, int &da2, int &status, int &pid, int j) const;
  virtual void ReadStats();
 
  static HepMCFileReader *instance();
 
private:
  HepMC::IO_BaseClass const *input() const { return get_underlying_safe(input_); }
  HepMC::IO_BaseClass *&input() { return get_underlying_safe(input_); }
 
  // current  HepMC evt
  edm::propagate_const<HepMC::GenEvent *> evt_;
  edm::propagate_const<HepMC::IO_BaseClass *> input_;
 
  static HepMCFileReader *instance_;
 
  int rdstate() const;
  //maps to convert HepMC::GenParticle to particles # and vice versa
  // -> needed for HepEvt like output
  std::vector<HepMC::GenParticle *> index_to_particle;
  std::map<HepMC::GenParticle *, int> particle_to_index;
  // find index to HepMC::GenParticle* p in map m
  int find_in_map(const std::map<HepMC::GenParticle *, int> &m, HepMC::GenParticle *p) const;
};
 
bool HepMCFileReader::isInitialized() const { return input_ != nullptr; }
 
#endif