[PWGEM/Dilepton] add DileptonSVMC.h (#16649)

Author: dsekihat

PWGEM/Dilepton/Core/Dilepton.h

@@ -107,7 +107,7 @@ using FilteredMyMuon = FilteredMyMuons::iterator;
 using MyEMH_electron = o2::aod::pwgem::dilepton::utils::EventMixingHandler<std::tuple<int, int, int, int>, std::pair<int, int>, o2::aod::pwgem::dilepton::utils::EMTrack>;
 using MyEMH_muon = o2::aod::pwgem::dilepton::utils::EventMixingHandler<std::tuple<int, int, int, int>, std::pair<int, int>, o2::aod::pwgem::dilepton::utils::EMFwdTrack>;
 
-template <o2::aod::pwgem::dilepton::utils::pairutil::DileptonPairType pairtype, bool withSCT, typename TEMH, typename... Types>
+template <o2::aod::pwgem::dilepton::utils::pairutil::DileptonPairType pairtype, typename TEMH, typename... Types>
 struct Dilepton {
 
   // Configurables
@@ -849,26 +849,6 @@ struct Dilepton {
     fDimuonCut.EnableTTCA(dimuoncuts.enableTTCA);
   }
 
-  template <typename TTrack>
-  bool foundHFSV(TTrack const& track)
-  {
-    int ptbin = lower_bound(dielectroncuts.binsMLSCT.value.begin(), dielectroncuts.binsMLSCT.value.end(), track.pt()) - dielectroncuts.binsMLSCT.value.begin() - 1;
-    if (ptbin < 0) {
-      ptbin = 0;
-    } else if (static_cast<int>(dielectroncuts.binsMLSCT.value.size()) - 2 < ptbin) {
-      ptbin = static_cast<int>(dielectroncuts.binsMLSCT.value.size()) - 2;
-    }
-
-    for (int i = 0; i < static_cast<int>(track.nSV()); i++) {
-      auto probaSCT = track.probaSCT(i);
-      // LOGF(info, "track.globalIndex() = %d, pt = %f, i = %d, probaSCT[0] = %f, probaSCT[1] = %f, probaSCT[2] = %f, probaSCT[3] = %f", track.globalIndex(), track.pt(), i, probaSCT[0], probaSCT[1], probaSCT[2], probaSCT[3]);
-      if (probaSCT[1] > dielectroncuts.cutsMLSCTeT_prompt_hc.value[ptbin] || probaSCT[2] > dielectroncuts.cutsMLSCTeT_nonprompt_hc.value[ptbin] || probaSCT[3] > dielectroncuts.cutsMLSCTeT_hb.value[ptbin]) {
-        return true;
-      }
-    }
-    return false;
-  }
-
   template <typename TQvectors>
   bool isGoodQvector(TQvectors const& qvectors)
   {
@@ -921,11 +901,6 @@ struct Dilepton {
             return false;
           }
         }
-        if constexpr (withSCT) {
-          if (foundHFSV(t1) || foundHFSV(t2)) {
-            return false;
-          }
-        }
       } else if constexpr (pairtype == o2::aod::pwgem::dilepton::utils::pairutil::DileptonPairType::kDimuon) {
         if (!cut.template IsSelectedTrack<false>(t1) || !cut.template IsSelectedTrack<false>(t2)) {
           return false;
@@ -1587,27 +1562,13 @@ struct Dilepton {
           return false;
         }
       }
-
-      if constexpr (withSCT) {
-        if (foundHFSV(t1) || foundHFSV(t2)) {
-          return false;
-        }
-      }
-
     } else if constexpr (pairtype == o2::aod::pwgem::dilepton::utils::pairutil::DileptonPairType::kDimuon) {
       if (!cut.IsSelectedTrack(t1) || !cut.IsSelectedTrack(t2)) {
         return false;
       }
       if (!map_best_match_globalmuon[t1.globalIndex()] || !map_best_match_globalmuon[t2.globalIndex()]) {
         return false;
       }
-
-      // if (!o2::aod::pwgem::dilepton::utils::emtrackutil::isBestMatch(t1, cut, tracks)) {
-      //   return false;
-      // }
-      // if (!o2::aod::pwgem::dilepton::utils::emtrackutil::isBestMatch(t2, cut, tracks)) {
-      //   return false;
-      // }
     }
 
     if constexpr (pairtype == o2::aod::pwgem::dilepton::utils::pairutil::DileptonPairType::kDielectron) {

PWGEM/Dilepton/Core/DileptonMC.h

@@ -438,9 +438,6 @@ struct DileptonMC {
     fRegistry.addClone("Generated/sm/PromptPi0/", "Generated/sm/NonPromptJPsi/");
     fRegistry.addClone("Generated/sm/PromptPi0/", "Generated/sm/PromptPsi2S/");
     fRegistry.addClone("Generated/sm/PromptPi0/", "Generated/sm/NonPromptPsi2S/");
-    // fRegistry.addClone("Generated/sm/PromptPi0/", "Generated/sm/Upsilon1S/");
-    // fRegistry.addClone("Generated/sm/PromptPi0/", "Generated/sm/Upsilon2S/");
-    // fRegistry.addClone("Generated/sm/PromptPi0/", "Generated/sm/Upsilon3S/");
 
     fRegistry.add("Generated/ccbar/c2l_c2l/uls/hs", "generated dilepton", o2::framework::HistType::kTHnSparseD, {axis_mass, axis_pt, axis_y, axis_dphi_ee, axis_deta_ee, axis_cos_theta_pol, axis_phi_pol, axis_quadmom, axis_aco, axis_asym_pt, axis_dphi_e_ee}, true);
     fRegistry.addClone("Generated/ccbar/c2l_c2l/uls/", "Generated/ccbar/c2l_c2l/lspp/");

PWGEM/Dilepton/Core/DileptonSV.h

@@ -1106,20 +1106,6 @@ struct DileptonSV {
       candidate.phi2 = t2.phi();
     }
 
-    // if constexpr (pairtype == o2::aod::pwgem::dilepton::utils::pairutil::DileptonPairType::kDielectron) {
-    //   auto trackParCov1 = getTrackParCov(t1);
-    //   auto trackParCov2 = getTrackParCov(t2);
-    //   if(!findSV(collision, trackParCov1, trackParCov2, candidate)){
-    //     return false;
-    //   }
-    // } else if constexpr (pairtype == o2::aod::pwgem::dilepton::utils::pairutil::DileptonPairType::kDimuon) {
-    //   auto trackParCov1 = o2::aod::fwdtrackutils::getTrackParCovFwd(t1, t1);
-    //   auto trackParCov2 = o2::aod::fwdtrackutils::getTrackParCovFwd(t2, t2);
-    //   if(!findSVFwd(collision, trackParCov1, trackParCov2, candidate)){
-    //     return false;
-    //   }
-    // }
-
     // if constexpr (pairtype == o2::aod::pwgem::dilepton::utils::pairutil::DileptonPairType::kDielectron) {
     //   if (!cut.IsSelectedPair(t1, t2)) {
     //     return false;