Skip to content

Commit 7430f98

Browse files
Fixing issue with 5TeV generator
1 parent 3448606 commit 7430f98

File tree

1 file changed

+26
-0
lines changed

1 file changed

+26
-0
lines changed

MC/config/PWGDQ/external/generator/GeneratorPromptCharmonia.C

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -810,6 +810,32 @@ FairGenerator* GeneratorCocktailPromptCharmoniaToMuonEvtGen_pp13TeV()
810810
return genCocktailEvtGen;
811811
}
812812

813+
FairGenerator* GeneratorCocktailPromptCharmoniaToMuonEvtGen_pp5TeV()
814+
{
815+
816+
auto genCocktailEvtGen = new o2::eventgen::GeneratorEvtGen<GeneratorCocktail>();
817+
818+
auto genJpsi = new o2::eventgen::O2_GeneratorParamJpsiFwdY;
819+
genJpsi->SetNSignalPerEvent(1); // 1 J/psi generated per event by GeneratorParam
820+
auto genPsi = new o2::eventgen::O2_GeneratorParamPsiFwdY;
821+
genPsi->SetNSignalPerEvent(1); // 1 Psi(2S) generated per event by GeneratorParam
822+
genCocktailEvtGen->AddGenerator(genJpsi, 1); // add J/psi generator
823+
genCocktailEvtGen->AddGenerator(genPsi, 1); // add Psi(2S) generator
824+
825+
TString pdgs = "443;100443";
826+
std::string spdg;
827+
TObjArray* obj = pdgs.Tokenize(";");
828+
genCocktailEvtGen->SetSizePdg(obj->GetEntriesFast());
829+
for (int i = 0; i < obj->GetEntriesFast(); i++) {
830+
spdg = obj->At(i)->GetName();
831+
genCocktailEvtGen->AddPdg(std::stoi(spdg), i);
832+
printf("PDG %d \n", std::stoi(spdg));
833+
}
834+
genCocktailEvtGen->SetForceDecay(kEvtDiMuon);
835+
836+
return genCocktailEvtGen;
837+
}
838+
813839

814840
FairGenerator*
815841
GeneratorParamPromptPsiToJpsiPiPiEvtGen_pp13TeV(TString pdgs = "100443")

0 commit comments

Comments
 (0)