better version of signal loss function

Author: murioni

PWGLF/Tasks/Resonances/k892analysispbpb.cxx

@@ -1050,54 +1050,48 @@ struct K892analysispbpb {
     auto impactPar = mcCollision.impactParameter();
     histos.fill(HIST("QAevent/hImpactParameterGen"), impactPar);
 
-    // Signal loss estimation
-    for (const auto& mcPart : mcParticles) {
-      if (std::abs(mcPart.y()) >= 0.5 || std::abs(mcPart.pdgCode()) != 313)
-        continue;
-
-      // signal loss estimation
-      if (mcPart.pdgCode() > 0) // no cuts, purely generated
-        histos.fill(HIST("QAevent/k892genBeforeEvtSel"), mcPart.pt(), impactPar);
-      else
-        histos.fill(HIST("QAevent/k892genBeforeEvtSelAnti"), mcPart.pt(), impactPar);
-    }
-
-    if (recCollisions.size() == 0)
-      return;
-
-    auto numcontributors = -999;
     bool isSel = false;
     auto centrality = -999.;
-    for (const auto& RecCollision : recCollisions) {
-      if (!myEventSelections(RecCollision))
-        continue;
-
-      if (RecCollision.numContrib() <= numcontributors)
-        continue;
-      else
-        numcontributors = RecCollision.numContrib();
-
-      centrality = RecCollision.centFT0C();
-      isSel = true;
+    if (recCollisions.size() > 0) {
+      auto numcontributors = -999;
+      for (const auto& RecCollision : recCollisions) {
+	if (!myEventSelections(RecCollision))
+	  continue;
+
+	if (RecCollision.numContrib() <= numcontributors)
+	  continue;
+	else
+	  numcontributors = RecCollision.numContrib();
+
+	centrality = RecCollision.centFT0C();
+	isSel = true;
+      }
     }
-
-    if (!isSel)
-      return;
-
-    histos.fill(HIST("QAevent/hImpactParameterRec"), impactPar);
-    histos.fill(HIST("QAevent/hImpactParvsCentrRec"), centrality, impactPar);
-
+    
+    if (isSel) {
+      histos.fill(HIST("QAevent/hImpactParameterRec"), impactPar);
+      histos.fill(HIST("QAevent/hImpactParvsCentrRec"), centrality, impactPar);
+    }
+      
     // Generated MC
     for (const auto& mcPart : mcParticles) {
       if (std::abs(mcPart.y()) >= 0.5 || std::abs(mcPart.pdgCode()) != 313)
         continue;
 
       // signal loss estimation
       if (mcPart.pdgCode() > 0) // no cuts, purely generated
-        histos.fill(HIST("QAevent/k892genAfterEvtSel"), mcPart.pt(), impactPar);
+        histos.fill(HIST("QAevent/k892genBeforeEvtSel"), mcPart.pt(), impactPar);
       else
-        histos.fill(HIST("QAevent/k892genAfterEvtSelAnti"), mcPart.pt(), impactPar);
-
+        histos.fill(HIST("QAevent/k892genBeforeEvtSelAnti"), mcPart.pt(), impactPar);
+      
+
+      if (isSel) {
+	// signal loss estimation
+	if (mcPart.pdgCode() > 0) // no cuts, purely generated
+	  histos.fill(HIST("QAevent/k892genAfterEvtSel"), mcPart.pt(), impactPar);
+	else
+	  histos.fill(HIST("QAevent/k892genAfterEvtSelAnti"), mcPart.pt(), impactPar);
+      }
     } // end loop on gen particles
   }
   PROCESS_SWITCH(K892analysispbpb, processEvtLossSigLossMC, "Process Signal Loss, Event Loss", false);