@@ -163,7 +163,7 @@ struct HfCorrelatorDplusHadrons {
163163 Produces<aod::Hadron> entryHadron;
164164
165165 Configurable<int > selectionFlagDplus{" selectionFlagDplus" , 7 , " Selection Flag for Dplus" }; // 7 corresponds to topo+PID cuts
166- Configurable<bool > isPromptAnalysis{" isPromptAnalysis" , true , " Flag for prompt D+-hadron correlations" };
166+ Configurable<bool > isPromptAnalysis{" isPromptAnalysis" , true , " Flag for prompt D+-hadron correlations" };
167167 Configurable<int > numberEventsMixed{" numberEventsMixed" , 5 , " Number of events mixed in ME process" };
168168 Configurable<bool > applyEfficiency{" applyEfficiency" , true , " Flag for applying D-meson efficiency weights" };
169169 Configurable<bool > removeDaughters{" removeDaughters" , true , " Flag for removing D-meson daughters from correlations" };
@@ -345,8 +345,10 @@ struct HfCorrelatorDplusHadrons {
345345 for (unsigned int iclass = 0 ; iclass < classMl->size (); iclass++) {
346346 outputMl[iclass] = candidate.mlProbDplusToPiKPi ()[classMl->at (iclass)];
347347 }
348- if (isPromptAnalysis) entryDplusCandRecoInfo (hfHelper.invMassDplusToPiKPi (candidate), candidate.pt (), outputMl[0 ], outputMl[1 ]); // 0: BkgBDTScore, 1:PromptBDTScore
349- else entryDplusCandRecoInfo (hfHelper.invMassDplusToPiKPi (candidate), candidate.pt (), outputMl[0 ], outputMl[2 ]); // 0: BkgBDTScore, 2:FDBDTScore
348+ if (isPromptAnalysis)
349+ entryDplusCandRecoInfo (hfHelper.invMassDplusToPiKPi (candidate), candidate.pt (), outputMl[0 ], outputMl[1 ]); // 0: BkgBDTScore, 1:PromptBDTScore
350+ else
351+ entryDplusCandRecoInfo (hfHelper.invMassDplusToPiKPi (candidate), candidate.pt (), outputMl[0 ], outputMl[2 ]); // 0: BkgBDTScore, 2:FDBDTScore
350352 entryDplus (candidate.phi (), candidate.eta (), candidate.pt (), hfHelper.invMassDplusToPiKPi (candidate), poolBin, gCollisionId , timeStamp);
351353
352354 // Dplus-Hadron correlation dedicated section
@@ -368,8 +370,10 @@ struct HfCorrelatorDplusHadrons {
368370 track.pt (), poolBin);
369371 entryDplusHadronRecoInfo (hfHelper.invMassDplusToPiKPi (candidate), false );
370372 entryDplusHadronGenInfo (false , false , 0 );
371- if (isPromptAnalysis) entryDplusHadronMlInfo (outputMl[0 ], outputMl[1 ]);
372- else entryDplusHadronMlInfo (outputMl[0 ], outputMl[2 ]);
373+ if (isPromptAnalysis)
374+ entryDplusHadronMlInfo (outputMl[0 ], outputMl[1 ]);
375+ else
376+ entryDplusHadronMlInfo (outputMl[0 ], outputMl[2 ]);
373377 entryTrackRecoInfo (track.dcaXY (), track.dcaZ (), track.tpcNClsCrossedRows ());
374378 if (cntDplus == 0 ) {
375379 entryHadron (track.phi (), track.eta (), track.pt (), poolBin, gCollisionId , timeStamp);
@@ -461,8 +465,10 @@ struct HfCorrelatorDplusHadrons {
461465 outputMl[iclass] = candidate.mlProbDplusToPiKPi ()[classMl->at (iclass)];
462466 }
463467 registry.fill (HIST (" hMassDplusMcRecSig" ), hfHelper.invMassDplusToPiKPi (candidate), candidate.pt (), efficiencyWeightD);
464- if (isPromptAnalysis) entryDplusCandRecoInfo (hfHelper.invMassDplusToPiKPi (candidate), candidate.pt (), outputMl[0 ], outputMl[1 ]);
465- else entryDplusCandRecoInfo (hfHelper.invMassDplusToPiKPi (candidate), candidate.pt (), outputMl[0 ], outputMl[2 ]);
468+ if (isPromptAnalysis)
469+ entryDplusCandRecoInfo (hfHelper.invMassDplusToPiKPi (candidate), candidate.pt (), outputMl[0 ], outputMl[1 ]);
470+ else
471+ entryDplusCandRecoInfo (hfHelper.invMassDplusToPiKPi (candidate), candidate.pt (), outputMl[0 ], outputMl[2 ]);
466472 entryDplusCandGenInfo (isDplusPrompt);
467473 } else {
468474 registry.fill (HIST (" hPtCandMcRecBkg" ), candidate.pt ());
@@ -491,8 +497,10 @@ struct HfCorrelatorDplusHadrons {
491497 candidate.pt (),
492498 track.pt (), poolBin);
493499 entryDplusHadronRecoInfo (hfHelper.invMassDplusToPiKPi (candidate), isDplusSignal);
494- if (isPromptAnalysis) entryDplusHadronMlInfo (outputMl[0 ], outputMl[1 ]);
495- else entryDplusHadronMlInfo (outputMl[0 ], outputMl[2 ]);
500+ if (isPromptAnalysis)
501+ entryDplusHadronMlInfo (outputMl[0 ], outputMl[1 ]);
502+ else
503+ entryDplusHadronMlInfo (outputMl[0 ], outputMl[2 ]);
496504 if (track.has_mcParticle ()) {
497505 auto mcParticle = track.template mcParticle_as <aod::McParticles>();
498506 isPhysicalPrimary = mcParticle.isPhysicalPrimary ();
@@ -700,8 +708,10 @@ struct HfCorrelatorDplusHadrons {
700708 for (unsigned int iclass = 0 ; iclass < classMl->size (); iclass++) {
701709 outputMl[iclass] = candidate.mlProbDplusToPiKPi ()[classMl->at (iclass)];
702710 }
703- if (isPromptAnalysis) entryDplusHadronMlInfo (outputMl[0 ], outputMl[1 ]);
704- else entryDplusHadronMlInfo (outputMl[0 ], outputMl[2 ]);
711+ if (isPromptAnalysis)
712+ entryDplusHadronMlInfo (outputMl[0 ], outputMl[1 ]);
713+ else
714+ entryDplusHadronMlInfo (outputMl[0 ], outputMl[2 ]);
705715 entryTrackRecoInfo (pAssoc.dcaXY (), pAssoc.dcaZ (), pAssoc.tpcNClsCrossedRows ());
706716 }
707717 }
0 commit comments