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;
}


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