[PWGLF] Updated task for generated MC (#9892)

Co-authored-by: Tulika Tripathy <tripathy@clralicemac08.local>
Co-authored-by: ALICE Action Bot <alibuild@cern.ch>

Author: 3 people

PWGLF/Tasks/GlobalEventProperties/dndeta-mft-pp.cxx

@@ -774,13 +774,13 @@ struct PseudorapidityDensityMFT {
       }
       nChargedCentral++;
     }
-
-    if (nChargedCentral > 0) {
-      registry.fill(HIST("EventEfficiency"), 2.);
-      registry.fill(HIST("EventsNtrkZvtxGen_gt0t"), nCharged,
-                    mcCollision.posZ());
+    if ((mcCollision.posZ() >= cfgVzCut1) && (mcCollision.posZ() <= cfgVzCut2)) {
+      if (nChargedCentral > 0) {
+        registry.fill(HIST("EventEfficiency"), 2.);
+        registry.fill(HIST("EventsNtrkZvtxGen_gt0t"), nCharged,
+                      mcCollision.posZ());
+      }
     }
-
     //-----------
     bool atLeastOne = false;
     bool atLeastOne_gt0 = false;
@@ -799,24 +799,25 @@ struct PseudorapidityDensityMFT {
 
         auto perCollisionSampleCentral =
           midtracks.sliceBy(perColCentral, collision.globalIndex());
+        if ((collision.posZ() >= cfgVzCut1) && (collision.posZ() <= cfgVzCut2) && (mcCollision.posZ() >= cfgVzCut1) && (mcCollision.posZ() <= cfgVzCut2)) {
+          if (perCollisionSampleCentral.size() > 0) {
+            registry.fill(HIST("EventEfficiency"), 5.);
+            atLeastOne_gt0 = true;
+            registry.fill(HIST("EventsNtrkZvtxGen_gt0"),
+                          perCollisionSample.size(), collision.posZ());
+          }
 
-        if (perCollisionSampleCentral.size() > 0) {
-          registry.fill(HIST("EventEfficiency"), 5.);
-          atLeastOne_gt0 = true;
-          registry.fill(HIST("EventsNtrkZvtxGen_gt0"),
-                        perCollisionSample.size(), collision.posZ());
-        }
-
-        registry.fill(HIST("EventsZposDiff"),
-                      collision.posZ() - mcCollision.posZ());
-        if (useZDiffCut) {
-          if (std::abs(collision.posZ() - mcCollision.posZ()) > maxZDiff) {
-            continue;
+          registry.fill(HIST("EventsZposDiff"),
+                        collision.posZ() - mcCollision.posZ());
+          if (useZDiffCut) {
+            if (std::abs(collision.posZ() - mcCollision.posZ()) > maxZDiff) {
+              continue;
+            }
           }
+          registry.fill(HIST("EventsNtrkZvtxGen"), perCollisionSample.size(),
+                        collision.posZ());
+          ++moreThanOne;
         }
-        registry.fill(HIST("EventsNtrkZvtxGen"), perCollisionSample.size(),
-                      collision.posZ());
-        ++moreThanOne;
       }
     }
     if (collisions.size() == 0) {
@@ -825,37 +826,38 @@ struct PseudorapidityDensityMFT {
     if (moreThanOne > 1) {
       registry.fill(HIST("EventsSplitMult"), nCharged);
     }
+    if ((mcCollision.posZ() >= cfgVzCut1) && (mcCollision.posZ() <= cfgVzCut2)) {
+      for (auto& particle : particles) {
+        auto p = pdg->GetParticle(particle.pdgCode());
+        auto charge = 0;
+        if (p != nullptr) {
+          charge = static_cast<int>(p->Charge());
+        }
+        if (std::abs(charge) < 3.) {
+          continue;
+        }
 
-    for (auto& particle : particles) {
-      auto p = pdg->GetParticle(particle.pdgCode());
-      auto charge = 0;
-      if (p != nullptr) {
-        charge = static_cast<int>(p->Charge());
-      }
-      if (std::abs(charge) < 3.) {
-        continue;
-      }
-
-      registry.fill(HIST("TracksEtaZvtxGen_t"), particle.eta(),
-                    mcCollision.posZ());
-      if (perCollisionMCSampleCentral.size() > 0) {
-        registry.fill(HIST("TracksEtaZvtxGen_gt0t"), particle.eta(),
-                      mcCollision.posZ());
-      }
-      if (atLeastOne) {
-        registry.fill(HIST("TracksEtaZvtxGen"), particle.eta(),
+        registry.fill(HIST("TracksEtaZvtxGen_t"), particle.eta(),
                       mcCollision.posZ());
-        registry.fill(HIST("TracksPtEtaGen"), particle.pt(), particle.eta());
-        if (atLeastOne_gt0) {
-          registry.fill(HIST("TracksEtaZvtxGen_gt0"), particle.eta(),
+        if (perCollisionMCSampleCentral.size() > 0) {
+          registry.fill(HIST("TracksEtaZvtxGen_gt0t"), particle.eta(),
                         mcCollision.posZ());
         }
-      }
+        if (atLeastOne) {
+          registry.fill(HIST("TracksEtaZvtxGen"), particle.eta(),
+                        mcCollision.posZ());
+          registry.fill(HIST("TracksPtEtaGen"), particle.pt(), particle.eta());
+          if (atLeastOne_gt0) {
+            registry.fill(HIST("TracksEtaZvtxGen_gt0"), particle.eta(),
+                          mcCollision.posZ());
+          }
+        }
 
-      registry.fill(HIST("TracksPhiEtaGen"), particle.phi(), particle.eta());
-      registry.fill(HIST("TracksPhiZvtxGen"), particle.phi(),
-                    mcCollision.posZ());
-      registry.fill(HIST("TracksPtEtaGen_t"), particle.pt(), particle.eta());
+        registry.fill(HIST("TracksPhiEtaGen"), particle.phi(), particle.eta());
+        registry.fill(HIST("TracksPhiZvtxGen"), particle.phi(),
+                      mcCollision.posZ());
+        registry.fill(HIST("TracksPtEtaGen_t"), particle.pt(), particle.eta());
+      }
     }
   }