[PWGLF] add eta and primary cut to mc closure

PWGLF/Tasks/Strangeness/cascadecorrelations.cxx

@@ -288,7 +288,7 @@
           casc.v0cosPA(pvx, pvy, pvz) < v0setting_cospa ||
           casc.casccosPA(pvx, pvy, pvz) < cascadesetting_cospa ||
           casc.dcav0topv(pvx, pvy, pvz) < cascadesetting_mindcav0topv ||
           TMath::Abs(casc.mLambda() - 1.115683) > cascadesetting_v0masswindow) {
         // It failed at least one topo selection
         cascflags(0);
         continue;
@@ -296,13 +296,13 @@
       registry.fill(HIST("hSelectionStatus"), 3); // passes topo
       // registry.fill(HIST("hMassXi3"), casc.mXi(), casc.pt());
 
       if (TMath::Abs(posTrack.eta()) > etaTracks || TMath::Abs(negTrack.eta()) > etaTracks || TMath::Abs(bachTrack.eta()) > etaTracks) {
         cascflags(0);
         continue;
       }
       registry.fill(HIST("hSelectionStatus"), 4); // passes track eta
 
       if (TMath::Abs(casc.eta()) > etaCascades) {
         cascflags(0);
         continue;
       }
@@ -314,23 +314,23 @@
       // Lambda check
       if (casc.sign() < 0) {
         // Proton check:
         if (TMath::Abs(posTrack.tpcNSigmaPr()) > tpcNsigmaProton) {
           cascflags(0);
           continue;
         }
         // Pion check:
         if (TMath::Abs(negTrack.tpcNSigmaPi()) > tpcNsigmaPion) {
           cascflags(0);
           continue;
         }
       } else {
         // Proton check:
         if (TMath::Abs(negTrack.tpcNSigmaPr()) > tpcNsigmaProton) {
           cascflags(0);
           continue;
         }
         // Pion check:
         if (TMath::Abs(posTrack.tpcNSigmaPi()) > tpcNsigmaPion) {
           cascflags(0);
           continue;
         }
@@ -339,8 +339,8 @@
       // registry.fill(HIST("hMassXi4"), casc.mXi(), casc.pt());
 
       // Bachelor check
       if (TMath::Abs(bachTrack.tpcNSigmaPi()) < tpcNsigmaBachelor) {
         if (TMath::Abs(bachTrack.tpcNSigmaKa()) < tpcNsigmaBachelor) {
           // consistent with both!
           cascflags(2);
           registry.fill(HIST("hSelectionStatus"), 7); // passes bach PID
@@ -363,7 +363,7 @@
           registry.fill(HIST("hMassXiPlus"), casc.mXi(), casc.pt(), casc.yXi());
         }
         continue;
       } else if (TMath::Abs(bachTrack.tpcNSigmaKa()) < tpcNsigmaBachelor) {
         cascflags(3);
         registry.fill(HIST("hSelectionStatus"), 7); // passes bach PID
         if (casc.sign() < 0) {
@@ -521,6 +521,11 @@
       {"MC/hGenMultNoReco", "hGenMultNoReco", {HistType::kTH1I, {{100, 0, 100, "Number of generated charged primaries"}}}},
       {"MC/hGenMultOneReco", "hGenMultOneReco", {HistType::kTH1I, {{100, 0, 100, "Number of generated charged primaries"}}}},
       {"MC/hSplitEvents", "hSplitEvents", {HistType::kTH1I, {{10, 0, 10, "Number of rec. events per gen event"}}}},
+
+      // debug
+      {"MC/hPhi", "hPhi", {HistType::kTH1F, {{180, 0, TwoPI}}}},
+      {"MC/hEta", "hEta", {HistType::kTH1F, {{100, -2, 2}}}},
+      {"MC/hRapidity", "hRapidity", {HistType::kTH1F, {{100, -2, 2}}}},
     },
   };
 
@@ -924,7 +929,8 @@
     } // collisions
   } // process mixed events
 
-  Filter genCascadesFilter = nabs(aod::mcparticle::pdgCode) == 3312;
+  Configurable<float> etaGenCascades{"etaGenCascades", 0.8, "min/max of eta for generated cascades"};
+  Filter genCascadesFilter = (nabs(aod::mcparticle::pdgCode) == 3312 && nabs(aod::mcparticle::eta) < etaGenCascades);
 
   void processMC(aod::McCollision const&, soa::SmallGroups<soa::Join<aod::McCollisionLabels, MyCollisionsMult>> const& collisions, soa::Filtered<aod::McParticles> const& genCascades, aod::McParticles const& mcParticles)
   {
@@ -947,6 +953,15 @@
       return;
     }
 
+    // QA
+    for (auto& casc : genCascades) {
+      if (!casc.isPhysicalPrimary())
+        continue;
+      registry.fill(HIST("MC/hPhi"), casc.phi());
+      registry.fill(HIST("MC/hEta"), casc.eta());
+      registry.fill(HIST("MC/hRapidity"), casc.y());
+    }
+
     for (auto& [c0, c1] : combinations(genCascades, genCascades)) { // combinations automatically applies strictly upper in case of 2 identical tables
       // Define the trigger as the particle with the highest pT. As we can't swap the cascade tables themselves, we swap the addresses and later dereference them
       auto* triggerAddress = &c0;
@@ -959,6 +974,9 @@
 
       double dphi = RecoDecay::constrainAngle(trigger.phi() - assoc.phi(), -PIHalf);
 
+      if (!trigger.isPhysicalPrimary() || !assoc.isPhysicalPrimary())
+        continue; // require the cascades to be primaries
+
       if (trigger.pdgCode() < 0) { // anti-trigg --> Plus
         if (assoc.pdgCode() < 0) { // anti-assoc --> Plus
           registry.fill(HIST("MC/hMCPlusPlus"), dphi, trigger.y() - assoc.y(), trigger.pt(), assoc.pt(), vtxz, FT0mult);