MpdUrqmdGenerator.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                MpdUrqmdGenerator header file                  -----
// -----          Created 11/06/04  by V. Friese / D.Bertini           -----
// -------------------------------------------------------------------------
 
#define GZIP_SUPPORT   // version with gz support
 
#ifndef MPDURQMDGENERATOR_H
#define MPDURQMDGENERATOR_H
 
#include "FairGenerator.h"
 
#include <fstream>
#include <map>
 
#ifdef GZIP_SUPPORT
#ifndef __CLING__
#include <zlib.h>
#endif
#endif
 
class TVirtualMCStack;
class FairPrimaryGenerator;
 
class MpdUrqmdGenerator : public FairGenerator
{
  public:
    MpdUrqmdGenerator();
 
    MpdUrqmdGenerator(const char* fileName);
 
    ~MpdUrqmdGenerator();
 
    Bool_t ReadEvent(FairPrimaryGenerator* primGen);
 
    // Skip some events in file
    Bool_t SkipEvents(Int_t count);
 
    void SetEventPlane(Double_t phiMin, Double_t phiMax);
 
    void SetXYZ(Double32_t x = 0., Double32_t y = 0., Double32_t z = 0.)
    {
        fX = x;
        fY = y;
        fZ = z;
    }
 
  private:
#ifdef GZIP_SUPPORT
#ifndef __CLING__
    gzFile fInputFile;   
#endif
#else
    FILE* fInputFile;   
#endif
 
    std::map<Int_t, Int_t> fParticleTable;   
 
    Double32_t fPhiMin, fPhiMax;   // Limits of event plane angle
    Bool_t fEventPlaneSet;         // Flag whether event plane angle is used
 
    const Char_t* fFileName;   
 
    Double32_t fX, fY, fZ;   // Point vertex coordinates [cm]
 
    void ReadConversionTable();
 
    MpdUrqmdGenerator(const MpdUrqmdGenerator&) = delete;
    MpdUrqmdGenerator& operator=(const MpdUrqmdGenerator&) = delete;
 
    ClassDef(MpdUrqmdGenerator, 1);
};
 
#endif