DPPythia8Generator.h

Go to the documentation of this file.

// SPDX-License-Identifier: LGPL-3.0-or-later
// SPDX-FileCopyrightText: Copyright CERN for the benefit of the SHiP
// Collaboration
 
#ifndef SHIPGEN_DPPYTHIA8GENERATOR_H_
#define SHIPGEN_DPPYTHIA8GENERATOR_H_
#define DPP8GENERATOR_H
// Avoid the inclusion of dlfcn.h by Pyhtia.h that CINT is not able to process
// #ifdef __CINT__
// #define _DLFCN_H_
// #define _DLFCN_H
// #endif
 
#include "FairLogger.h"  // for FairLogger, MESSAGE_ORIGIN
#include "Generator.h"
#include "HNLPythia8Generator.h"
#include "Pythia8/Pythia.h"
#include "TFile.h"
#include "TH2F.h"
#include "TROOT.h"
#include "TRandom1.h"
#include "TRandom3.h"
#include "TString.h"
#include "TTree.h"
 
class FairPrimaryGenerator;
 
class DPPythia8Generator : public SHiP::Generator {
 public:
  DPPythia8Generator();
 
  ~DPPythia8Generator() override;
 
  Bool_t ReadEvent(FairPrimaryGenerator*) override;
  void SetParameters(char*);
  void Print() { fPythia->settings.listAll(); };
  void List(int id) { fPythia->particleData.list(id); };
 
  // void SetDecayToHadrons(){
  // std::cout << " INFO: Adding decay to hadrons." << std::endl;
  // fHadDecay = true;
  // };
 
  using SHiP::Generator::Init;
  Bool_t Init() override;
  Bool_t Init(const char* inFile) override { return Init(inFile, 0); };
 
  Bool_t Init(const char* inFile, int startEvent) override {
    LOG(warning) << "Init with files not implemented for DPPythia8Generator. "
                    "Using default Init() instead";
    return Init();
  };
 
  void SetMom(Double_t mom) { fMom = mom; };
  Double_t GetMom() { return fMom; };
  void SetId(Double_t id) { fId = id; };
  void SetDPId(Int_t id) { fDP = id; };
  Int_t GetDPId() { return fDP; };
  void SetLmin(Double_t z) { fLmin = z * 10; };
  void SetLmax(Double_t z) { fLmax = z * 10; };
  void SetSmearBeam(Double_t sb) { fsmearBeam = sb; };
  void SetfFDs(Double_t z) { fFDs = z; };
  void UseRandom1() {
    fUseRandom1 = kTRUE;
    fUseRandom3 = kFALSE;
  };
  void UseRandom3() {
    fUseRandom1 = kFALSE;
    fUseRandom3 = kTRUE;
  };
 
  void SetPbrem(TH2F* pdf) {
    fpbrem = kTRUE;
    fpbremPDF = pdf;
  };
  Bool_t IsPbrem() { return fpbrem; };
  void SetDY() { fdy = kTRUE; };
 
  Double_t MinDPMass() { return fDPminM; };
  void SetMinDPMass(Double_t m) { fDPminM = m; };
 
  void UseDeepCopy() { fDeepCopy = kTRUE; };
  Int_t nrOfRetries() { return fnRetries; };
  Int_t nrOfDP() { return fnDPtot; };
  Pythia8::Pythia* getPythiaInstance() { return fPythia; };
  Pythia8::Pythia* fPythia;  
  // Pythia8::Pythia* fPythiaHadDecay;             //!
 private:
  std::shared_ptr<Pythia8::RndmEngine> fRandomEngine;  
 
 protected:
  // Bool_t fHadDecay;    //select hadronic decay
  Double_t fMom;       // proton energy
  Int_t fDP;           // DP ID
  Int_t fId;           // target type
  Bool_t fUseRandom1;  // flag to use TRandom1
  Bool_t fUseRandom3;  // flag to use TRandom3 (default)
  Bool_t
      fpbrem;  // flag to do proton bremstrahlung production (default is false)
  TH2F* fpbremPDF;   // pointer to TH2 containing PDF(p,theta) to have a dark
                     // photon with momentum p and angle theta to be produced by
                     // pbrem.
  Bool_t fdy;        // flag to do Drell-Yan QCD production
  Double_t fDPminM;  // Minimum mass, in GeV, for the DP produced in ffbar to DP
                     // QCD production.
  Double_t fLmin;    // m minimum  decay position z
  Double_t fLmax;    // m maximum decay position z
  Int_t fnRetries;   // number of events without any DP
  Int_t fnDPtot;  // total number of DP from multiple mesons in single collision
  Double_t fctau;  // dark photon lifetime
  Double_t fFDs;   // correction for Pythia6 to match measured Ds production
  Double_t fsmearBeam;  // finite beam size
  TFile* fInputFile;    
  TTree* fTree;         
  Int_t fNevents, fn, fShipEventNr;
  Float_t hpx[1], hpy[1], hpz[1], hE[1], hM[1], mpx[1], mpy[1], mpz[1], mE[1],
      hid[1], mid[1];
  Bool_t fDeepCopy;     // not used
  FairLogger* fLogger;  
};
 
#endif  // SHIPGEN_DPPYTHIA8GENERATOR_H_