[PWGJE] efficiency 3d (#16021)

Author: joachimckh

PWGJE/Tasks/jetSpectraEseTask.cxx

@@ -65,6 +65,7 @@ using namespace o2::framework::expressions;
 
 struct JetSpectraEseTask {
   Configurable<std::string> cfgEfficiency{"cfgEfficiency", "", "CCDB path to efficiency"};
+  Configurable<std::string> cfgEfficiency3D{"cfgEfficiency3D", "", "CCDB path to 3D efficiency"};
   Configurable<float> jetPtMin{"jetPtMin", 5.0, "minimum jet pT cut"};
   Configurable<float> jetR{"jetR", 0.2, "jet resolution parameter"};
   Configurable<float> randomConeR{"randomConeR", 0.4, "size of random Cone for estimating background fluctuations"};
@@ -147,8 +148,10 @@ struct JetSpectraEseTask {
 
   Service<ccdb::BasicCCDBManager> ccdb;
   struct Efficiency {
-    TH1D* hEff = nullptr;
+    TH1F* hEff = nullptr;
+    TH3F* h3Eff = nullptr;
     bool isLoaded = false;
+    bool is3D = false;
   } cfg;
 
   Filter trackCuts = (aod::jtrack::pt >= trackPtMin && aod::jtrack::pt < trackPtMax && aod::jtrack::eta > trackEtaMin && aod::jtrack::eta < trackEtaMax);
@@ -414,12 +417,14 @@ struct JetSpectraEseTask {
     }
     if (doprocessMCGenTrack) {
       LOGF(info, "JetSpectraEseTask::init() - MCGen track");
+      registry.add("mcgen/h3TrackPtGen", ";#it{p}_{T,track}; vtxZ; #eta", {HistType::kTH3F, {{assocTrackPt}, {vertexZAxis}, {etaAxis}}});
       registry.add("mcgen/hTrackPtGen", "", {HistType::kTH1F, {{assocTrackPt}}});
       registry.add("mcgen/hTrackEtaGen", "", {HistType::kTH1F, {{etaAxis}}});
       registry.add("mcgen/hTrackPhiGen", "", {HistType::kTH1F, {{phiAxis}}});
     }
     if (doprocessMCRecoTrack) {
       LOGF(info, "JetSpectraEseTask::init() - MCRec track");
+      registry.add("mcrec/h3TrackPtReco", ";#it{p}_{T,track}; vtxZ; #eta", {HistType::kTH3F, {{assocTrackPt}, {vertexZAxis}, {etaAxis}}});
       registry.add("mcrec/hTrackPtReco", "", {HistType::kTH1F, {{assocTrackPt}}});
       registry.add("mcrec/hTrackEtaReco", "", {HistType::kTH1F, {{etaAxis}}});
       registry.add("mcrec/hTrackPhiReco", "", {HistType::kTH1F, {{phiAxis}}});
@@ -434,22 +439,37 @@ struct JetSpectraEseTask {
       return;
     }
     if (!cfgEfficiency.value.empty()) {
-      cfg.hEff = ccdb->getForTimeStamp<TH1D>(cfgEfficiency, timestamp);
+      cfg.hEff = ccdb->getForTimeStamp<TH1F>(cfgEfficiency, timestamp);
       if (cfg.hEff == nullptr) {
         LOGF(fatal, "Could not load track efficiency from %s", cfgEfficiency.value.c_str());
       }
       LOGF(info, "Loaded tracking efficiency from %s (%p)", cfgEfficiency.value.c_str(), (void*)cfg.hEff);
     }
+    if (!cfgEfficiency3D.value.empty()) {
+      cfg.h3Eff = ccdb->getForTimeStamp<TH3F>(cfgEfficiency3D, timestamp);
+      if (cfg.h3Eff == nullptr) {
+        LOGF(fatal, "Could not load track efficiency from %s", cfgEfficiency3D.value.c_str());
+      }
+      LOGF(info, "Loaded 3D tracking efficiency from %s (%p)", cfgEfficiency3D.value.c_str(), (void*)cfg.h3Eff);
+      cfg.is3D = true;
+    }
     cfg.isLoaded = true;
     return;
   }
 
   template <typename TTrack>
-  double getEfficiency(TTrack track)
+  double getEfficiency(TTrack track, auto vtxZ)
   {
     double eff{1.0};
-    if (cfg.hEff)
-      eff = cfg.hEff->GetBinContent(cfg.hEff->FindBin(track.pt()));
+    if (cfg.is3D) {
+      if (cfg.h3Eff) {
+        eff = cfg.h3Eff->GetBinContent(cfg.h3Eff->FindBin(track.pt(), track.eta(), vtxZ));
+      }
+    } else {
+      if (cfg.hEff) {
+        eff = cfg.hEff->GetBinContent(cfg.hEff->FindBin(track.pt()));
+      }
+    }
     if (eff == 0)
       return -1.;
     else
@@ -528,7 +548,7 @@ struct JetSpectraEseTask {
       for (const auto& track : tracks) {
         if (!jetderiveddatautilities::selectTrack(track, trackSelection))
           continue;
-        double weff = getEfficiency(track);
+        double weff = getEfficiency(track, collision.posZ());
         if (weff < 0)
           continue;
         auto deta = track.eta() - jet.eta();
@@ -538,7 +558,7 @@ struct JetSpectraEseTask {
       }
     }
     for (const auto& track : tracks) {
-      double weff = getEfficiency(track);
+      double weff = getEfficiency(track, collision.posZ());
       if (weff < 0)
         continue;
       registry.fill(HIST("trackQA/before/hTrackPt"), centrality, track.pt(), weff);
@@ -626,7 +646,7 @@ struct JetSpectraEseTask {
         for (const auto& track : tracks2) {
           if (!jetderiveddatautilities::selectTrack(track, trackSelection))
             continue;
-          double weff = getEfficiency(track);
+          double weff = getEfficiency(track, c2.posZ());
           if (weff < 0)
             continue;
           auto deta = track.eta() - jet.eta();
@@ -636,7 +656,7 @@ struct JetSpectraEseTask {
       }
 
       for (const auto& track : tracks2) {
-        double weff = getEfficiency(track);
+        double weff = getEfficiency(track, c2.posZ());
         if (weff < 0)
           continue;
         registry.fill(HIST("trackQA/before/hTrackPtMixed"), centrality, track.pt(), weff);
@@ -907,6 +927,7 @@ struct JetSpectraEseTask {
         continue;
       }
 
+      registry.fill(HIST("mcgen/h3TrackPtGen"), particle.pt(), collision.posZ(), particle.eta());
       registry.fill(HIST("mcgen/hTrackPtGen"), particle.pt());
       registry.fill(HIST("mcgen/hTrackEtaGen"), particle.eta());
       registry.fill(HIST("mcgen/hTrackPhiGen"), particle.phi());
@@ -972,6 +993,7 @@ struct JetSpectraEseTask {
         }
         seenMcParticles.push_back(particle.globalIndex());
 
+        registry.fill(HIST("mcrec/h3TrackPtReco"), track.pt(), collision.posZ(), track.eta());
         registry.fill(HIST("mcrec/hTrackPtReco"), track.pt());
         registry.fill(HIST("mcrec/hTrackEtaReco"), track.eta());
         registry.fill(HIST("mcrec/hTrackPhiReco"), track.phi());