Fix ML selection for D0 and D0bar on data and MCRec

samrangy · samrangy · commit f039176c528c · 2025-11-13T20:23:06.000+01:00
diff --git a/PWGHF/HFC/Tasks/taskCorrelationD0Hadrons.cxx b/PWGHF/HFC/Tasks/taskCorrelationD0Hadrons.cxx
@@ -317,8 +317,8 @@ struct HfTaskCorrelationD0Hadrons {
       registry.fill(HIST("hBdtScorePromptD0bar"), bdtScorePromptD0bar);
       registry.fill(HIST("hBdtScoreBkgD0bar"), bdtScoreBkgD0bar);
 
-      if (bdtScorePromptD0 < mlOutputPromptD0->at(effBinD) || bdtScoreBkgD0 > mlOutputBkgD0->at(effBinD) ||
-          bdtScorePromptD0bar < mlOutputPromptD0bar->at(effBinD) || bdtScoreBkgD0bar > mlOutputBkgD0bar->at(effBinD)) {
+      if ((bdtScorePromptD0 < mlOutputPromptD0->at(effBinD) || bdtScoreBkgD0 > mlOutputBkgD0->at(effBinD)) &&
+          (bdtScorePromptD0bar < mlOutputPromptD0bar->at(effBinD) || bdtScoreBkgD0bar > mlOutputBkgD0bar->at(effBinD))) {
         continue;
       }
     }
@@ -347,8 +347,8 @@ struct HfTaskCorrelationD0Hadrons {
       if (ptBinD < 0 || effBinD < 0) {
         continue;
       }
-      if (bdtScorePromptD0 < mlOutputPromptD0->at(ptBinD) || bdtScoreBkgD0 > mlOutputBkgD0->at(ptBinD) ||
-          bdtScorePromptD0bar < mlOutputPromptD0bar->at(ptBinD) || bdtScoreBkgD0bar > mlOutputBkgD0bar->at(ptBinD)) {
+      if ((bdtScorePromptD0 < mlOutputPromptD0->at(ptBinD) || bdtScoreBkgD0 > mlOutputBkgD0->at(ptBinD)) &&
+          (bdtScorePromptD0bar < mlOutputPromptD0bar->at(ptBinD) || bdtScoreBkgD0bar > mlOutputBkgD0bar->at(ptBinD))) {
         continue;
       }
       if (trackDcaXY > dcaXYTrackMax || trackDcaZ > dcaZTrackMax || trackTpcCrossedRows < nTpcCrossedRaws) {
@@ -476,8 +476,8 @@ struct HfTaskCorrelationD0Hadrons {
       registry.fill(HIST("hBdtScorePromptD0bar"), bdtScorePromptD0bar);
       registry.fill(HIST("hBdtScoreBkgD0bar"), bdtScoreBkgD0bar);
 
-      if (bdtScorePromptD0 < mlOutputPromptD0->at(effBinD) || bdtScoreBkgD0 > mlOutputBkgD0->at(effBinD) ||
-          bdtScorePromptD0bar < mlOutputPromptD0bar->at(effBinD) || bdtScoreBkgD0bar > mlOutputBkgD0bar->at(effBinD)) {
+      if ((bdtScorePromptD0 < mlOutputPromptD0->at(effBinD) || bdtScoreBkgD0 > mlOutputBkgD0->at(effBinD)) &&
+          (bdtScorePromptD0bar < mlOutputPromptD0bar->at(effBinD) || bdtScoreBkgD0bar > mlOutputBkgD0bar->at(effBinD))) {
         continue;
       }
     }
@@ -505,8 +505,8 @@ struct HfTaskCorrelationD0Hadrons {
       bool const isD0Prompt = pairEntry.isPrompt();
       int const statusPromptHadron = pairEntry.trackOrigin();
 
-      if (bdtScorePromptD0 < mlOutputPromptD0->at(ptBinD) || bdtScoreBkgD0 > mlOutputBkgD0->at(ptBinD) ||
-          bdtScorePromptD0bar < mlOutputPromptD0bar->at(ptBinD) || bdtScoreBkgD0bar > mlOutputBkgD0bar->at(ptBinD)) {
+      if ((bdtScorePromptD0 < mlOutputPromptD0->at(ptBinD) || bdtScoreBkgD0 > mlOutputBkgD0->at(ptBinD)) &&
+          (bdtScorePromptD0bar < mlOutputPromptD0bar->at(ptBinD) || bdtScoreBkgD0bar > mlOutputBkgD0bar->at(ptBinD))) {
         continue;
       }
       if (trackDcaXY > dcaXYTrackMax || trackDcaZ > dcaZTrackMax || trackTpcCrossedRows < nTpcCrossedRaws) {

Original file line number	Diff line number	Diff line change
`@@ -317,8 +317,8 @@ struct HfTaskCorrelationD0Hadrons {`
`317`	`317`	`registry.fill(HIST("hBdtScorePromptD0bar"), bdtScorePromptD0bar);`
`318`	`318`	`registry.fill(HIST("hBdtScoreBkgD0bar"), bdtScoreBkgD0bar);`
`319`	`319`
`320`		`- if (bdtScorePromptD0 < mlOutputPromptD0->at(effBinD) \|\| bdtScoreBkgD0 > mlOutputBkgD0->at(effBinD) \|\|`
`321`		`- bdtScorePromptD0bar < mlOutputPromptD0bar->at(effBinD) \|\| bdtScoreBkgD0bar > mlOutputBkgD0bar->at(effBinD)) {`
	`320`	`+ if ((bdtScorePromptD0 < mlOutputPromptD0->at(effBinD) \|\| bdtScoreBkgD0 > mlOutputBkgD0->at(effBinD)) &&`
	`321`	`+ (bdtScorePromptD0bar < mlOutputPromptD0bar->at(effBinD) \|\| bdtScoreBkgD0bar > mlOutputBkgD0bar->at(effBinD))) {`
`322`	`322`	`continue;`
`323`	`323`	`}`
`324`	`324`	`}`
`@@ -347,8 +347,8 @@ struct HfTaskCorrelationD0Hadrons {`
`347`	`347`	`if (ptBinD < 0 \|\| effBinD < 0) {`
`348`	`348`	`continue;`
`349`	`349`	`}`
`350`		`- if (bdtScorePromptD0 < mlOutputPromptD0->at(ptBinD) \|\| bdtScoreBkgD0 > mlOutputBkgD0->at(ptBinD) \|\|`
`351`		`- bdtScorePromptD0bar < mlOutputPromptD0bar->at(ptBinD) \|\| bdtScoreBkgD0bar > mlOutputBkgD0bar->at(ptBinD)) {`
	`350`	`+ if ((bdtScorePromptD0 < mlOutputPromptD0->at(ptBinD) \|\| bdtScoreBkgD0 > mlOutputBkgD0->at(ptBinD)) &&`
	`351`	`+ (bdtScorePromptD0bar < mlOutputPromptD0bar->at(ptBinD) \|\| bdtScoreBkgD0bar > mlOutputBkgD0bar->at(ptBinD))) {`
`352`	`352`	`continue;`
`353`	`353`	`}`
`354`	`354`	`if (trackDcaXY > dcaXYTrackMax \|\| trackDcaZ > dcaZTrackMax \|\| trackTpcCrossedRows < nTpcCrossedRaws) {`
`@@ -476,8 +476,8 @@ struct HfTaskCorrelationD0Hadrons {`
`476`	`476`	`registry.fill(HIST("hBdtScorePromptD0bar"), bdtScorePromptD0bar);`
`477`	`477`	`registry.fill(HIST("hBdtScoreBkgD0bar"), bdtScoreBkgD0bar);`
`478`	`478`
`479`		`- if (bdtScorePromptD0 < mlOutputPromptD0->at(effBinD) \|\| bdtScoreBkgD0 > mlOutputBkgD0->at(effBinD) \|\|`
`480`		`- bdtScorePromptD0bar < mlOutputPromptD0bar->at(effBinD) \|\| bdtScoreBkgD0bar > mlOutputBkgD0bar->at(effBinD)) {`
	`479`	`+ if ((bdtScorePromptD0 < mlOutputPromptD0->at(effBinD) \|\| bdtScoreBkgD0 > mlOutputBkgD0->at(effBinD)) &&`
	`480`	`+ (bdtScorePromptD0bar < mlOutputPromptD0bar->at(effBinD) \|\| bdtScoreBkgD0bar > mlOutputBkgD0bar->at(effBinD))) {`
`481`	`481`	`continue;`
`482`	`482`	`}`
`483`	`483`	`}`
`@@ -505,8 +505,8 @@ struct HfTaskCorrelationD0Hadrons {`
`505`	`505`	`bool const isD0Prompt = pairEntry.isPrompt();`
`506`	`506`	`int const statusPromptHadron = pairEntry.trackOrigin();`
`507`	`507`
`508`		`- if (bdtScorePromptD0 < mlOutputPromptD0->at(ptBinD) \|\| bdtScoreBkgD0 > mlOutputBkgD0->at(ptBinD) \|\|`
`509`		`- bdtScorePromptD0bar < mlOutputPromptD0bar->at(ptBinD) \|\| bdtScoreBkgD0bar > mlOutputBkgD0bar->at(ptBinD)) {`
	`508`	`+ if ((bdtScorePromptD0 < mlOutputPromptD0->at(ptBinD) \|\| bdtScoreBkgD0 > mlOutputBkgD0->at(ptBinD)) &&`
	`509`	`+ (bdtScorePromptD0bar < mlOutputPromptD0bar->at(ptBinD) \|\| bdtScoreBkgD0bar > mlOutputBkgD0bar->at(ptBinD))) {`
`510`	`510`	`continue;`
`511`	`511`	`}`
`512`	`512`	`if (trackDcaXY > dcaXYTrackMax \|\| trackDcaZ > dcaZTrackMax \|\| trackTpcCrossedRows < nTpcCrossedRaws) {`