| 1 |
#ifndef Input_HepMCFileReader_h
|
| 2 |
#define Input_HepMCFileReader_h
|
| 3 |
|
| 4 |
// $Id: HepMCFileReader.h,v 1.1 2012/06/07 19:10:11 antoniov Exp $
|
| 5 |
|
| 6 |
/** \class HepMCFileReader
|
| 7 |
*
|
| 8 |
* This class is used by the implementation of DaqEventFactory present
|
| 9 |
* in this package to read in the full event raw data from a flat
|
| 10 |
* binary file.
|
| 11 |
* WARNING: If you want to use this class for other purposes you must
|
| 12 |
* always invoke the method initialize before starting using the interface
|
| 13 |
* it exposes.
|
| 14 |
*
|
| 15 |
* $Date: 2012/06/07 19:10:11 $
|
| 16 |
* $Revision: 1.1 $
|
| 17 |
* \author G. Bruno - CERN, EP Division
|
| 18 |
*/
|
| 19 |
|
| 20 |
#include <vector>
|
| 21 |
#include <map>
|
| 22 |
|
| 23 |
|
| 24 |
namespace HepMC {
|
| 25 |
class IO_BaseClass;
|
| 26 |
class GenEvent;
|
| 27 |
class GenParticle;
|
| 28 |
}
|
| 29 |
|
| 30 |
|
| 31 |
class HepMCFileReader {
|
| 32 |
|
| 33 |
protected:
|
| 34 |
HepMCFileReader();
|
| 35 |
|
| 36 |
public:
|
| 37 |
virtual ~HepMCFileReader();
|
| 38 |
virtual void initialize(const std::string &filename);
|
| 39 |
inline bool isInitialized() const;
|
| 40 |
|
| 41 |
virtual bool setEvent(int event);
|
| 42 |
virtual bool readCurrentEvent();
|
| 43 |
virtual bool printHepMcEvent() const;
|
| 44 |
HepMC::GenEvent *fillCurrentEventData();
|
| 45 |
// virtual bool fillEventData(HepMC::GenEvent *event);
|
| 46 |
// this method prints the event information as
|
| 47 |
// obtained by the input file in HepEvt style
|
| 48 |
void printEvent() const;
|
| 49 |
// get all the 'integer' properties of a particle
|
| 50 |
// like mother, daughter, pid and status
|
| 51 |
// 'j' is the number of the particle in the HepMc
|
| 52 |
virtual void getStatsFromTuple(int &mo1, int &mo2, int &da1, int &da2,
|
| 53 |
int &status, int &pid, int j) const;
|
| 54 |
virtual void ReadStats();
|
| 55 |
|
| 56 |
static HepMCFileReader *instance();
|
| 57 |
|
| 58 |
private:
|
| 59 |
// current HepMC evt
|
| 60 |
HepMC::GenEvent *evt_;
|
| 61 |
HepMC::IO_BaseClass *input_;
|
| 62 |
|
| 63 |
static HepMCFileReader *instance_;
|
| 64 |
|
| 65 |
int rdstate() const;
|
| 66 |
//maps to convert HepMC::GenParticle to particles # and vice versa
|
| 67 |
// -> needed for HepEvt like output
|
| 68 |
std::vector<HepMC::GenParticle*> index_to_particle;
|
| 69 |
std::map<HepMC::GenParticle*,int> particle_to_index;
|
| 70 |
// find index to HepMC::GenParticle* p in map m
|
| 71 |
int find_in_map(const std::map<HepMC::GenParticle*,int>& m,
|
| 72 |
HepMC::GenParticle *p) const;
|
| 73 |
};
|
| 74 |
|
| 75 |
|
| 76 |
bool HepMCFileReader::isInitialized() const
|
| 77 |
{
|
| 78 |
return input_ != 0;
|
| 79 |
}
|
| 80 |
|
| 81 |
#endif
|
