add upc selection into HF framework

Author: zhangbiao-phy

PWGHF/Tasks/taskPidStudies.cxx

@@ -197,6 +197,7 @@ struct HfTaskPidStudies {
   using CollisionsMc = soa::Join<aod::Collisions, aod::McCollisionLabels, aod::EvSels, aod::CentFT0Cs, aod::CentFT0Ms>;
   using V0sMcRec = soa::Join<aod::V0Datas, aod::V0CoreMCLabels>;
   using CascsMcRec = soa::Join<aod::CascDatas, aod::CascCoreMCLabels>;
+  using BCs = soa::Join<aod::BCsWithTimestamps, aod::BcSels, aod::Run3MatchedToBCSparse>;
 
   ctpRateFetcher rateFetcher;
   HfEventSelection hfEvSel;
@@ -336,17 +337,17 @@ struct HfTaskPidStudies {
     return Particle::NotMatched;
   }
 
-  template <typename Coll>
-  bool isCollSelected(const Coll& coll)
+  template <typename Coll, typename BCs>
+  bool isCollSelected(const Coll& coll, const BCs& bcs)
   {
-    auto bc = coll.template bc_as<aod::BCsWithTimestamps>();
+    auto bc = coll.template bc_as<BCs>();
     interactionRate = rateFetcher.fetch(ccdb.service, bc.timestamp(), bc.runNumber(), ctpFetcherSource.value) * 1.e-3; // convert to kHz
     if (interactionRate < interactionRateMin || interactionRate > interactionRateMax) {
       return false;
     }
 
     float cent{-1.f};
-    const auto rejectionMask = hfEvSel.getHfCollisionRejectionMask<true, o2::hf_centrality::CentralityEstimator::None, aod::BCsWithTimestamps>(coll, cent, ccdb, registry);
+    const auto rejectionMask = hfEvSel.getHfCollisionRejectionMask<true, o2::hf_centrality::CentralityEstimator::None, BCs>(coll, cent, ccdb, registry, bcs);
     /// monitor the satisfied event selections
     hfEvSel.fillHistograms(coll, rejectionMask, cent);
     return rejectionMask == 0;
@@ -503,10 +504,10 @@ struct HfTaskPidStudies {
                    aod::V0MCCores const&,
                    aod::McParticles const& /*particlesMc*/,
                    PidTracks const& /*tracks*/,
-                   aod::BCsWithTimestamps const&)
+                   BCs const& bcs)
   {
     for (const auto& v0 : V0s) {
-      if (applyEvSels && !isCollSelected(v0.collision_as<CollisionsMc>())) {
+      if (applyEvSels && !isCollSelected(v0.collision_as<CollisionsMc>(), bcs)) {
         continue;
       }
       if (applyTrackSels && !isTrackSelected<true>(v0)) {
@@ -524,11 +525,11 @@ struct HfTaskPidStudies {
 
   void processV0Data(aod::V0Datas const& V0s,
                      PidTracks const&,
-                     aod::BCsWithTimestamps const&,
+                     BCs const& bcs,
                      CollSels const&)
   {
     for (const auto& v0 : V0s) {
-      if (applyEvSels && !isCollSelected(v0.collision_as<CollSels>())) {
+      if (applyEvSels && !isCollSelected(v0.collision_as<CollSels>(), bcs)) {
         continue;
       }
       if (applyTrackSels && !isTrackSelected<true>(v0)) {
@@ -546,10 +547,10 @@ struct HfTaskPidStudies {
                      aod::CascMCCores const&,
                      aod::McParticles const& /*particlesMc*/,
                      PidTracks const&,
-                     aod::BCsWithTimestamps const&)
+                     BCs const& bcs)
   {
     for (const auto& casc : cascades) {
-      if (applyEvSels && !isCollSelected(casc.collision_as<CollisionsMc>())) {
+      if (applyEvSels && !isCollSelected(casc.collision_as<CollisionsMc>(), bcs)) {
         continue;
       }
       if (applyTrackSels && !isTrackSelected<false>(casc)) {
@@ -567,11 +568,11 @@ struct HfTaskPidStudies {
 
   void processCascData(aod::CascDatas const& cascades,
                        PidTracks const&,
-                       aod::BCsWithTimestamps const&,
+                       BCs const& bcs,
                        CollSels const&)
   {
     for (const auto& casc : cascades) {
-      if (applyEvSels && !isCollSelected(casc.collision_as<CollSels>())) {
+      if (applyEvSels && !isCollSelected(casc.collision_as<CollSels>(), bcs)) {
         continue;
       }
       if (applyTrackSels && !isTrackSelected<false>(casc)) {