PndFlukaGenerator.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                 PndFlukaGenerator header file                 -----
// -----                                                               -----
// -------------------------------------------------------------------------

#ifndef PND_FLUKAGENERATOR_H
#define PND_FLUKAGENERATOR_H

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

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

class PndFlukaGenerator : public FairGenerator {

 public:
  PndFlukaGenerator();

  PndFlukaGenerator(const Char_t *fileName);

  virtual ~PndFlukaGenerator();

  virtual Bool_t ReadEvent(FairPrimaryGenerator *primGen);

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

 private:
  Bool_t SetBranchAddresses();

  Int_t iEvent;          
  Int_t iEntry;          
  TString fFileName;     
  TFile *fInputRootFile; 
  TTree *fInputTree;     

  Double_t fpx, fpy, fpz; 
  Double_t fvx, fvy, fvz; 
  Int_t fpid;             
  Int_t fnpart;           

  void CloseInput();

  ClassDef(PndFlukaGenerator, 2);
};

#endif