detectors/hyp/hypTools/PndHypBupGenerator.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                PndHypBupGenerator header file                 -----
// -----                Created  by Alicia Sanchez                 -----
// -------------------------------------------------------------------------

#ifndef PNDHYPBUPGENERATOR_H
#define PNDHYPBUPGENERATOR_H 1

#include "../hyp/THParticle.h"


#include "FairGenerator.h"

class TClonesArray;
class TFile;
class TTree;
class TVirtualMCStack;

class PndPrimaryGenerator;
//class CbmParticle;



class PndHypBupGenerator : public FairGenerator  
{

 public: 

  PndHypBupGenerator();


  PndHypBupGenerator(const char* fileName);


  virtual ~PndHypBupGenerator();

  void SetPhiRange     (Double32_t phimin=0  , Double32_t phimax=360)
    {fPhiMin=phimin; fPhiMax=phimax;};

  void SetThetaRange   (Double32_t thetamin=0, Double32_t thetamax=90)
  {fThetaMin=thetamin; fThetaMax=thetamax; fThetaRangeIsSet=kTRUE;};

  void SetPRange(Double32_t pmin=0   , Double32_t pmax=10)
  {fPMin=pmin; fPMax=pmax; fPRangeIsSet=kTRUE;}

  void GammaEmissPar(Bool_t On=kFALSE){fGamOn=On;}
        
  virtual Bool_t ReadEvent(FairPrimaryGenerator* primGen);



 private:

  Int_t iEvent;      
  const Char_t* fFileName;   
  TFile* fInputFile;        
  TTree* fInputTree;        
  TClonesArray* fParticles;  

  Double32_t fPMin,  fPMax;      
  Double32_t fPhiMin, fPhiMax;     
  Double32_t fThetaMin, fThetaMax; 
  Bool_t     fThetaRangeIsSet;     // True if theta range is set
  Bool_t     fPRangeIsSet;         // True if abs.momentum range is set
  Bool_t     fGamOn;               // True if. gamma emission is selected

  void CloseInput();


  //Int_t RegisterIons();



  //std::map<TString,CbmParticle*> fIonMap;       //!
        
        
  ClassDef(PndHypBupGenerator,2);

};

#endif