MpdLAQGSMGeneratorExt.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                MpdLAQGSMGeneratorExt header file                 -----
// -------------------------------------------------------------------------
 
#ifndef MPDLAQGSMGENERATOREXT_H
#define MPDLAQGSMGENERATOREX_H 1
 
#include "MpdLAQGSMGenerator.h"
 
#include <BmnFairRunSim.h>
#include <TClonesArray.h>
#include <TVector3.h>
#include <fstream>
#include <map>
#include <vector>
 
#ifndef __CLING__
#include <zlib.h>
#endif
 
using namespace std;
 
class MpdLAQGSMGeneratorExt : public MpdLAQGSMGenerator
{
  public:
    MpdLAQGSMGeneratorExt();
 
    MpdLAQGSMGeneratorExt(const vector<TString>& fileNames,
                          const Bool_t use_collider_system = kTRUE,
                          Int_t QGSM_format_ID = 0,
                          Int_t Max_Event_Number = 0,
                          Int_t pdg = 3122,
                          Bool_t extract_from_dir = kTRUE);
 
    virtual ~MpdLAQGSMGeneratorExt();
 
    //  static la_tab_t la_tab[78];    //!  list of light particles known for MpdLAQGSMGeneratorExt
 
    Bool_t ReadEvent(FairPrimaryGenerator* primGen);
    Bool_t (MpdLAQGSMGeneratorExt::*ReadEventImpl)(FairPrimaryGenerator* primGen);
    Bool_t ReadEventFromTxt(FairPrimaryGenerator* primGen);
    Bool_t ExtractEventFromTxt(FairPrimaryGenerator* primGen);
    Bool_t ReadEventFromFiltered(FairPrimaryGenerator* primGen);
    Bool_t SkipEvents(Int_t nSkip);   // AZ
 
    BmnFairRunSim* GetRunSimInst() { return fRunSimInst; }
    void SetRunSimInst(BmnFairRunSim* v) { fRunSimInst = v; }
 
  protected:
    Bool_t fExtract_from_dir;
    Int_t iFile;
    Int_t fSelPDG;              
    vector<TString> fFileVec;   
    TClonesArray* fMom;         
    TString fMomBranchName;
    BmnFairRunSim* fRunSimInst;   
 
    TTree* fMTree;        
    TFile* fSourceFile;   
    Long64_t fNEvents;
    Long64_t fIEvent;
 
    Bool_t OpenNext();
 
    ClassDef(MpdLAQGSMGeneratorExt, 1);
};
 
#endif