changing reflection and corr bkg handling

Author: MattOckleton

PWGJE/Tasks/jetCorrelationD0.cxx

@@ -87,8 +87,7 @@ DECLARE_SOA_COLUMN(D0MD, d0MD, float);
 DECLARE_SOA_COLUMN(D0PtD, d0PtD, float);
 DECLARE_SOA_COLUMN(D0EtaD, d0EtaD, float);
 DECLARE_SOA_COLUMN(D0PhiD, d0PhiD, float);
-DECLARE_SOA_COLUMN(D0MatchedFrom, d0MatchedFrom, int);
-DECLARE_SOA_COLUMN(D0SelectedAs, d0SelectedAs, int);
+DECLARE_SOA_COLUMN(D0Category, d0Category, int);
 DECLARE_SOA_COLUMN(D0DecayChannel, d0DecayChannel, int8_t);
 } // namespace d0Info
 
@@ -115,9 +114,7 @@ DECLARE_SOA_TABLE(D0McDTables, "AOD", "D0MCDTABLE",
                   d0Info::D0Eta,
                   d0Info::D0Phi,
                   d0Info::D0Y,
-                  d0Info::D0MatchedFrom,
-                  d0Info::D0SelectedAs,
-                  d0Info::D0DecayChannel);
+                  d0Info::D0Category);
 
 DECLARE_SOA_TABLE(D0McPTables, "AOD", "D0MCPTABLE",
                   o2::soa::Index<>,
@@ -358,6 +355,7 @@ struct JetCorrelationD0 {
 
       int matchedFrom = 0;
       int selectedAs = 0;
+      int category; // 0 signal, 1 reflection, 2-5 correlated backgrounds
 
       if (d0DecayChannel > 0) { // matched to a D0 on truth level (any channel)
         matchedFrom = 1;
@@ -369,6 +367,28 @@ struct JetCorrelationD0 {
       } else if (d0Candidate.candidateSelFlag() & BIT(1)) { // CandidateSelFlag == BIT(1) -> selected as D0bar
         selectedAs = -1;
       }
+      if ((d0DecayChannel == 1) && (selectedAs == matchedFrom)) {
+        category = 0; // signal -> D0 or D0bar, π+ K− π+
+      }
+      if ((d0DecayChannel == 1) && (selectedAs == -1*matchedFrom)) {
+        category = 1; // reflection
+      }
+      if (d0DecayChannel == 2) {
+        category = 2; // corr bkg: π+ K− π0
+      }
+
+      if (d0DecayChannel == 3) {
+        category = 3; // corr bkg: π+ π−
+      }
+      if (d0DecayChannel == 4) {
+        category = 4; // corr bkg: π+ π− π0
+      }
+      if (d0DecayChannel == 5) {
+        category = 5; // corr bkg: K+ K−
+      }
+
+
+
 
       tableD0McDetector(tableCollision.lastIndex(), // might want to add some more detector level D0 quantities like prompt or non prompt info
                         scores[2],
@@ -379,9 +399,7 @@ struct JetCorrelationD0 {
                         d0Candidate.eta(),
                         d0Candidate.phi(),
                         d0Candidate.y(),
-                        matchedFrom,
-                        selectedAs,
-                        d0DecayChannel);
+                        category);
       for (const auto& jet : jets) {
         if (jet.pt() < jetPtCutMin) {
           continue;