PndEvtGenGenerator.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                 PndEvtGenGenerator header file                  -----
// -----               Created 05/02/07  by S.Spataro                 -----
// -------------------------------------------------------------------------

#ifndef PND_EVTGENGENERATOR_H
#define PND_EVTGENGENERATOR_H

#include "FairGenerator.h"
#include "TString.h"
#include <fstream>

class TClonesArray;
class TFile;
class TTree;
class TVirtualMCStack;
class FairPrimaryGenerator;
class TF1;

class PndEvtGenGenerator : public FairGenerator {

 public:
  PndEvtGenGenerator();

  PndEvtGenGenerator(const Char_t *fileName);
  PndEvtGenGenerator(const Char_t *fileName, Double_t Rsigma, TF1 *DensityFunction);

  virtual ~PndEvtGenGenerator();

  virtual Bool_t ReadEvent(FairPrimaryGenerator *primGen);

 protected:
  Bool_t ReadAsciiEvent(FairPrimaryGenerator *primGen);
  Bool_t ReadRootEvent(FairPrimaryGenerator *primGen);
  Bool_t Init();

 private:
  Bool_t SetBranchAddresses();

  Int_t iEvent;          
  TString fFileName;     
  TFile *fInputRootFile; 
  TTree *fInputTree;     
  FILE *fInputAsciiFile;

  Double_t *fRPx, *fRPy, *fRPz; 
  Double_t *fRVx, *fRVy, *fRVz; 
  Int_t *fRPdg;                 
  Int_t *fRDF, *fRDL;           
  Int_t fRNTrk;                 

  Bool_t fFileType; 

  int fGasmode;   
  double fRsigma; 

  TF1 *fDensityFunction; 

  void CloseInput();

  ClassDef(PndEvtGenGenerator, 1);
};

#endif