MpdLAQGSMGenerator.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                MpdLAQGSMGenerator header file                 -----
// -------------------------------------------------------------------------
 
#ifndef MPDLAQGSMGENERATOR_H
#define MPDLAQGSMGENERATOR_H 1
 
#include "FairGenerator.h"
 
#include <fstream>
#include <map>
#include <vector>
 
#ifndef __CLING__
#include <zlib.h>
#endif
 
class TDatabasePDG;
class FairPrimaryGenerator;
class FairIon;
 
class MpdLAQGSMGenerator : public FairGenerator
{
 
    struct la_tab_t
    {
        Int_t pdg;
        Int_t Z;
        Int_t strange;
        Int_t lepton;
        Int_t A;
        Float_t mass;
        char name[10];
    };
 
  public:
    MpdLAQGSMGenerator();
 
    MpdLAQGSMGenerator(const char* fileName,
                       const Bool_t use_collider_system = kTRUE,
                       Int_t QGSM_format_ID = 0,
                       Int_t Max_Event_Number = 0);
 
    virtual ~MpdLAQGSMGenerator();
 
    //  static la_tab_t la_tab[78];    //!  list of light particles known for MpdLAQGSMGenerator
 
    virtual Bool_t Init()
    {
        InitGenerator(nullptr);
        return kTRUE;
    }
    void InitGenerator(const char* light_particles_filename);   // fill list of known light particles
 
    virtual Bool_t ReadEvent(FairPrimaryGenerator* primGen);
 
    virtual Bool_t GetEventHeader(char* ss);
 
    Bool_t SkipEvents(Int_t nSkip);   // AZ
 
    Bool_t general_fgets(char* ss, Int_t nn = 250, FILE* p = 0);
    Bool_t general_feof(void* p);
 
    void SetXYZ(Double_t x = 0., Double_t y = 0., Double_t z = 0.)
    {
        fX = x;
        fY = y;
        fZ = z;
    }
 
  protected:
    FILE* fInputFile;   
#ifndef __CLING__
    gzFile fGZInputFile;   
#endif
    const Char_t* fFileName;     
    TDatabasePDG* fPDG;          
    Int_t fQGSM_format_ID;       //   Reflect format changes
    Bool_t fUseColliderSystem;   //   kTRUE- for NICA/MPD, kFALSE - for lab system (CBM)
    // la_tab_t la_tab[84];                //!  list of light particles known for MpdLAQGSMGenerator
    std::vector<la_tab_t*> fLa_tab;   
    Bool_t fGZ_input;                 
 
    Double_t fX, fY, fZ;   // Point vertex coordinates [cm]
 
    void CloseInput();
 
    Int_t RegisterIons(Int_t Max_Event_Number = 0);
    Int_t RegisterIons1();   // AZ
 
    Int_t CreatePdgCode(Int_t Z, Int_t A, Int_t Strange, Int_t user = 0);
    Bool_t FindParticle(Int_t Z, Int_t strange, Int_t lepton, Int_t A, Float_t mass, Int_t& PDG, char name[11]);
    Bool_t CreateNucleus(Int_t Z, Float_t mass, Int_t pdgCode, char pdgName[11]);
 
    std::map<TString, FairIon*> fIonMap;   
 
    ClassDef(MpdLAQGSMGenerator, 1);
};
 
#endif