CbmUnigenGenerator.h

Go to the documentation of this file.

 
#ifndef CBMUNIGENGENERATOR
#define CBMUNIGENGENERATOR
 
 
#include "FairGenerator.h"
#include "TMath.h"
#include "TString.h"
#include "TVector3.h"
#include <map>
 
class TFile;
class TTree;
class UEvent;
class UParticle;
class FairIon;
class FairPrimaryGenerator;
 
 
class CbmUnigenGenerator : public FairGenerator {
 
public:
  enum EMode {
    kStandard,     
    kNoRotation,   
    kRotateFixed,  
    kReuseEvents   
  };
 
 
  CbmUnigenGenerator(const char* fileName = "", EMode mode = kStandard);
 
 
  CbmUnigenGenerator(const char* fileName, EMode mode, Double_t phi);
 
 
  virtual ~CbmUnigenGenerator();
 
 
  virtual Bool_t ReadEvent(FairPrimaryGenerator* primGen);
 
 
private:
  TString fFileName;    
  EMode fMode;          
  Double_t fPhi;        
  Bool_t fIsInit;       
  TFile* fFile;         
  TTree* fTree;         
  Int_t fCurrentEntry;  
  UEvent* fEvent;       
  Int_t fNofPrimaries;  
  Int_t fNofEvents;     
  Double_t fBetaCM;     
  Double_t fGammaCM;    
  std::map<TString, FairIon*> fIonMap;  
 
  // Constants for decimal decomposition of ion PDG.
  // For ions the PDG code is +-10LZZZAAAI, with L = number of Lambdas,
  // ZZZ = charge (number of protons), AAA = mass (sum of numbers
  // of Lambdas, protons and neutrons, I = isomer level
  static const Int_t kPdgLambda = 10000000;  
  static const Int_t kPdgCharge = 10000;     
  static const Int_t kPdgMass   = 10;        
 
 
  void AddPrimary(FairPrimaryGenerator* primGen,
                  Int_t pdgCode,
                  const TVector3& momentum);
 
 
  void CloseInput();
 
 
  Int_t GetIonCharge(Int_t pdgCode) const {
    return (pdgCode % kPdgLambda) / kPdgCharge;
  }
 
 
  Int_t GetIonLambdas(Int_t pdgCode) const {
    return (pdgCode % (10 * kPdgLambda)) / kPdgLambda;
  }
 
 
  Int_t GetIonMass(Int_t pdgCode) const {
    return (pdgCode % kPdgCharge) / kPdgMass;
  }
 
 
  Bool_t GetNextEntry();
 
 
  Bool_t Init();
 
 
  void ProcessIon(FairPrimaryGenerator* primGen,
                  Int_t pdgCode,
                  const TVector3& momentum);
 
 
  Int_t RegisterIons();
 
 
  CbmUnigenGenerator(const CbmUnigenGenerator&) = delete;
 
 
  CbmUnigenGenerator& operator=(const CbmUnigenGenerator&) = delete;
 
 
  ClassDef(CbmUnigenGenerator, 4);
};
 
#endif