add ptHatMaxFraction cut

Author: aimeric-landou

PWGJE/Tasks/trackEfficiency.cxx

@@ -82,6 +82,7 @@ struct TrackEfficiency {
   Configurable<float> ptHatMin{"ptHatMin", 5, "min pT hat of collisions"};
   Configurable<float> ptHatMax{"ptHatMax", 300, "max pT hat of collisions"};
   Configurable<float> pTHatExponent{"pTHatExponent", 6.0, "exponent of the event weight for the calculation of pTHat"};
+  Configurable<float> pTHatMaxFractionMCD{"pTHatMaxFractionMCD", 999.0, "maximum fraction of hard scattering for reconstructed track acceptance in MC"};
 
   Configurable<int> useTrueTrackWeight{"useTrueTrackWeight", 1, "test configurable, to be removed"};
 
@@ -112,6 +113,13 @@ struct TrackEfficiency {
       if (!(jetderiveddatautilities::selectTrack(track, trackSelection) && jetderiveddatautilities::selectTrackDcaZ(track, trackDcaZmax))) {
         continue;
       }
+
+      float simPtRef = 10.;
+      float pTHat = simPtRef / (std::pow(weight, 1.0 / pTHatExponent));
+      if (track.pt() > pTHatMaxFractionMCD * pTHat) {
+        continue;
+      }
+
       registry.fill(HIST("h2_centrality_track_pt"), collision.centrality(), track.pt(), weight);
       registry.fill(HIST("h2_centrality_track_eta"), collision.centrality(), track.eta(), weight);
       registry.fill(HIST("h2_centrality_track_phi"), collision.centrality(), track.phi(), weight);
@@ -166,8 +174,17 @@ struct TrackEfficiency {
       registry.get<TH1>(HIST("hTrackCutsCounts"))->GetXaxis()->SetBinLabel(1, "allTracksInSelColl");
       registry.get<TH1>(HIST("hTrackCutsCounts"))->GetXaxis()->SetBinLabel(2, "trackSel");
       registry.get<TH1>(HIST("hTrackCutsCounts"))->GetXaxis()->SetBinLabel(3, "hasMcParticle");
-      registry.get<TH1>(HIST("hTrackCutsCounts"))->GetXaxis()->SetBinLabel(4, "mcPartIsPrimary");
-      registry.get<TH1>(HIST("hTrackCutsCounts"))->GetXaxis()->SetBinLabel(5, "etaAcc"); // not actually applied here but it will give an idea of what will be done in the post processing
+
+      if (doprocessEFficiencyPurity) {
+        registry.get<TH1>(HIST("hTrackCutsCounts"))->GetXaxis()->SetBinLabel(4, "mcPartIsPrimary");
+        registry.get<TH1>(HIST("hTrackCutsCounts"))->GetXaxis()->SetBinLabel(5, "etaAcc"); // not actually applied here but it will give an idea of what will be done in the post processing
+      }
+      if (doprocessEFficiencyPurityWeighted) {
+        registry.get<TH1>(HIST("hTrackCutsCounts"))->GetXaxis()->SetBinLabel(4, "ptHatMaxFraction");
+        registry.get<TH1>(HIST("hTrackCutsCounts"))->GetXaxis()->SetBinLabel(5, "mcPartIsPrimary");
+        registry.get<TH1>(HIST("hTrackCutsCounts"))->GetXaxis()->SetBinLabel(6, "etaAcc"); // not actually applied here but it will give an idea of what will be done in the post processing
+
+      }
 
       AxisSpec ptAxisEff = {nBinsLowPt, 0., 10., "#it{p}_{T} (GeV/#it{c})"};
       AxisSpec ptAxisHighEff = {18, 10., 100., "#it{p}_{T} (GeV/#it{c})"};
@@ -480,8 +497,9 @@ struct TrackEfficiency {
     }
     registry.fill(HIST("hMcCollCutsCounts"), 5.5); // at least one of the reconstructed collisions associated with this mcCollision is selected with regard to centrality
 
+    float simPtRef = 10.;
     float mcCollEventWeight = mcCollision.weight();
-    float pTHat = 10. / (std::pow(mcCollEventWeight, 1.0 / pTHatExponent));
+    float pTHat = simPtRef / (std::pow(mcCollEventWeight, 1.0 / pTHatExponent));
     if (pTHat < ptHatMin || pTHat > ptHatMax) { // only allows mcCollisions with weight in between min and max
       return;
     }
@@ -541,6 +559,12 @@ struct TrackEfficiency {
         }
         registry.fill(HIST("hTrackCutsCounts"), 2.5);
 
+        float pTHat = 10. / (std::pow(weight, 1.0 / pTHatExponent));
+        if (jet.pt() > pTHatMaxMCD * pTHat || pTHat < pTHatAbsoluteMin) {
+          continue;
+        }
+        registry.fill(HIST("hTrackCutsCounts"), 3.5);
+
         auto mcParticle = track.mcParticle_as<JetParticlesWithOriginal>();
         auto trueTrackMcCollision = mcParticle.mcCollision_as<aod::JetMcCollisions>();
         float trueTrackCollEventWeight = useTrueTrackWeight ? trueTrackMcCollision.weight() : mcCollEventWeight; // test1
@@ -566,7 +590,7 @@ struct TrackEfficiency {
           continue;
         }
 
-        registry.fill(HIST("hTrackCutsCounts"), 3.5);
+        registry.fill(HIST("hTrackCutsCounts"), 4.5);
 
         registry.fill(HIST("h3_track_pt_track_eta_track_phi_associatedtrack_primary"), track.pt(), track.eta(), track.phi(), trueTrackCollEventWeight);
         registry.fill(HIST("h3_particle_pt_particle_eta_particle_phi_associatedtrack_primary"), jMcParticleFromTrack.pt(), jMcParticleFromTrack.eta(), jMcParticleFromTrack.phi(), trueTrackCollEventWeight);
@@ -587,7 +611,7 @@ struct TrackEfficiency {
         }
 
         if (std::abs(jMcParticleFromTrack.eta()) < trackEtaAcceptanceCountQA) { // not actually applied here but it will give an idea of what will be done in the post processing
-          registry.fill(HIST("hTrackCutsCounts"), 4.5);
+          registry.fill(HIST("hTrackCutsCounts"), 5.5);
         }
       }
     }