[PWGEM/Dilepton] add chi2 cut in testPV.cxx

Author: dsekihat

PWGEM/Dilepton/Tasks/testPV.cxx

@@ -41,6 +41,7 @@ struct testPV {
   struct : ConfigurableGroup {
     std::string prefix = "eventCut";
     Configurable<int> cfgEventGeneratorId{"cfgEventGeneratorId", -1, "event generator index. e.g. select gap/signal events"};
+    Configurable<float> cfgChi2PerNcontribMax{"cfgChi2PerNcontribMax", 1e+10, "max. chi2/Ncontrib of PV"};
     Configurable<float> cfgZvtxMin{"cfgZvtxMin", -10.f, "min. Zvtx"};
     Configurable<float> cfgZvtxMax{"cfgZvtxMax", 10.f, "max. Zvtx"};
     Configurable<bool> cfgRequireFT0AND{"cfgRequireFT0AND", true, "require FT0AND"};
@@ -136,6 +137,9 @@ struct testPV {
     if (eventCut.cfgRequireVertexTOFmatched && !collision.selection_bit(o2::aod::evsel::kIsVertexTOFmatched)) {
       return false;
     }
+    if (collision.chi2() / collision.numContrib() > eventCut.cfgChi2PerNcontribMax) {
+      return false;
+    }
     return true;
   }
 
@@ -190,19 +194,21 @@ struct testPV {
     } // end of collision loop
   }
 
-  using MyCollisions = soa::Join<aod::Collisions, aod::EvSels, aod::McCollisionLabels>;
+  using MyCollisions = soa::Join<aod::Collisions, aod::EvSels>;
+  using MyCollisionsMC = soa::Join<MyCollisions, aod::McCollisionLabels>;
   using MyBCs = soa::Join<aod::BCsWithTimestamps, aod::BcSels>;
 
   Filter collisionFilter_evsel = eventCut.cfgZvtxMin < o2::aod::collision::posZ && o2::aod::collision::posZ < eventCut.cfgZvtxMax;
   using FilteredMyCollisions = soa::Filtered<MyCollisions>;
+  using FilteredMyCollisionsMC = soa::Filtered<MyCollisionsMC>;
 
   void processData(FilteredMyCollisions const& collisions, MyBCs const& bcs)
   {
     run<false>(bcs, collisions, nullptr, nullptr);
   }
   PROCESS_SWITCH(testPV, processData, "processData", true);
 
-  void processMC(FilteredMyCollisions const& collisions, MyBCs const& bcs, aod::McCollisions const& mcCollisions, aod::McParticles const& mcParticles)
+  void processMC(FilteredMyCollisionsMC const& collisions, MyBCs const& bcs, aod::McCollisions const& mcCollisions, aod::McParticles const& mcParticles)
   {
     run<true>(bcs, collisions, mcCollisions, mcParticles);
   }