CbmUnigenGenerator.h

Go to the documentation of this file.

/* Copyright (C) 2006-2019 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Dmytro Kresan [committer], Volker Friese */
 
#ifndef CBMUNIGENGENERATOR
#define CBMUNIGENGENERATOR
 
 
#include "FairGenerator.h"
 
#include "TMath.h"
#include "TString.h"
#include "TVector3.h"
 
#include <map>
 
#include "UEvent.h"
 
class TFile;
class TTree;
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);
 
  Int_t GetNumAvailableEvents() { return fAvailableEvents; }
 
private:
  TString fFileName      = "";             
  EMode fMode            = kStandard;      
  Double_t fPhi          = 0.;             
  Bool_t fIsInit         = kFALSE;         
  TFile* fFile           = nullptr;        
  TTree* fTree           = nullptr;        
  Int_t fCurrentEntry    = -1;             
  UEvent* fEvent         = new UEvent();   
  Int_t fNofPrimaries    = 0;              
  Int_t fNofEvents       = 0;              
  Double_t fBetaCM       = 0;              
  Double_t fGammaCM      = 0;              
  Int_t fAvailableEvents = 0;              
  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, 5);
};
 
#endif