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