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 "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