@@ -53,6 +53,12 @@ enum PairTypeOfSelMassSel {
5353 DDbar, // This is a D0-D0bar pair
5454 DbarD // This is a D0bar-D0 pair
5555};
56+
57+ enum Origin {
58+ Unassigned = 0 ,
59+ Prompt,
60+ NonPrompt
61+ };
5662} // namespace
5763
5864using McParticlesPlus2Prong = soa::Join<aod::McParticles, aod::HfCand2ProngMcGen>;
@@ -804,10 +810,10 @@ struct HfCorrelatorDMesonPairs {
804810 registry.fill (HIST (" hMass" ), hfHelper.invMassD0ToPiK (candidate1), candidate1.pt ());
805811 registry.fill (HIST (" hPtVsYVsNContribMcRec" ), candidate1.pt (), hfHelper.yD0 (candidate1), collision.numContrib ());
806812 registry.fill (HIST (" hNContribMcRec" ), collision.numContrib ());
807- if (originRec1 == 1 ) {
813+ if (originRec1 == Prompt ) {
808814 registry.fill (HIST (" hMassMcRecPrompt" ), hfHelper.invMassD0ToPiK (candidate1), candidate1.pt ());
809815 registry.fill (HIST (" hPtVsYVsNContribMcRecPrompt" ), candidate1.pt (), hfHelper.yD0 (candidate1), collision.numContrib ());
810- } else if (originRec1 == 2 ) {
816+ } else if (originRec1 == NonPrompt ) {
811817 registry.fill (HIST (" hMassMcRecNonPrompt" ), hfHelper.invMassD0ToPiK (candidate1), candidate1.pt ());
812818 registry.fill (HIST (" hPtVsYVsNContribMcRecNonPrompt" ), candidate1.pt (), hfHelper.yD0 (candidate1), collision.numContrib ());
813819 }
@@ -825,9 +831,9 @@ struct HfCorrelatorDMesonPairs {
825831 registry.fill (HIST (" hMass" ), hfHelper.invMassD0barToKPi (candidate1), candidate1.pt ());
826832 registry.fill (HIST (" hPtVsYVsNContribMcRec" ), candidate1.pt (), hfHelper.yD0 (candidate1), collision.numContrib ());
827833 registry.fill (HIST (" hNContribMcRec" ), collision.numContrib ());
828- if (originRec1 == 1 ) {
834+ if (originRec1 == Prompt ) {
829835 registry.fill (HIST (" hMassMcRecPrompt" ), hfHelper.invMassD0barToKPi (candidate1), candidate1.pt ());
830- } else if (originRec1 == 2 ) {
836+ } else if (originRec1 == NonPrompt ) {
831837 registry.fill (HIST (" hMassMcRecNonPrompt" ), hfHelper.invMassD0barToKPi (candidate1), candidate1.pt ());
832838 }
833839 } else if (isTrueDCand1) {
@@ -1012,10 +1018,10 @@ struct HfCorrelatorDMesonPairs {
10121018 }
10131019
10141020 registry.fill (HIST (" hPtVsYVsNContribMcGen" ), particle1.pt (), particle1.y (), numPvContributorsGen);
1015- if (originGen1 == 1 ) {
1021+ if (originGen1 == Prompt ) {
10161022 registry.fill (HIST (" hPtVsYVsNContribMcGenPrompt" ), particle1.pt (), particle1.y (), numPvContributorsGen);
10171023 }
1018- if (originGen1 == 2 ) {
1024+ if (originGen1 == NonPrompt ) {
10191025 registry.fill (HIST (" hPtVsYVsNContribMcGenNonPrompt" ), particle1.pt (), particle1.y (), numPvContributorsGen);
10201026 }
10211027 registry.fill (HIST (" hNContribMcGen" ), numPvContributorsGen);
0 commit comments