PWGJE: Adding histograms to trigger task to study correlation of global and TPC inner momentum (#6875)

Author: nzardosh

PWGJE/Tasks/ChJetTriggerQATask.cxx

@@ -78,6 +78,8 @@ struct ChJetTriggerQATask {
 
   Configurable<bool> bAddSupplementHistosToOutput{"bAddAdditionalHistosToOutput", false, "add supplementary histos to the output"};
 
+  Configurable<float> phiAngleRestriction{"phiAngleRestriction", 0.3, "angle to restrict track phi for plotting tpc momentum"};
+
   float fiducialVolume; // 0.9 - jetR
 
   HistogramRegistry spectra;
@@ -103,6 +105,11 @@ struct ChJetTriggerQATask {
     spectra.add("ptetaLeadingJetFullVol", "pT vs eta leading jet", {HistType::kTH2F, {{100, 0., +100.}, {80, -1., 1.}}});
     spectra.add("ptphiLeadingJetFullVol", "pT vs phi leading jet", {HistType::kTH2F, {{100, 0., +100.}, {60, 0, TMath::TwoPi()}}});
 
+    spectra.add("globalP_tpcglobalPDiff_withoutcuts", "difference of global and TPC inner momentum vs global momentum without any selection applied", {HistType::kTH2F, {{100, 0., +100.}, {200, -100., +100.}}});
+    spectra.add("globalP_tpcglobalPDiff", "difference of global and TPC inner momentum vs global momentum with selection applied", {HistType::kTH2F, {{100, 0., +100.}, {200, -100., +100.}}});
+    spectra.add("globalP_tpcglobalPDiff_withoutcuts_phirestrict", "difference of global and TPC inner momentum vs global momentum without any selection applied in a restricted phi", {HistType::kTH2F, {{100, 0., +100.}, {200, -100., +100.}}});
+    spectra.add("globalP_tpcglobalPDiff_phirestrict", "difference of global and TPC inner momentum vs global momentum with selection applied restricted phi", {HistType::kTH2F, {{100, 0., +100.}, {200, -100., +100.}}});
+
     // Supplementary plots
     if (bAddSupplementHistosToOutput) {
       spectra.add("ptJetChInclFullVol", "inclusive charged jet pT in full volume", {HistType::kTH1F, {{200, 0., +200.}}});
@@ -134,7 +141,7 @@ struct ChJetTriggerQATask {
   void
     process(soa::Filtered<soa::Join<JetCollisions,
                                     aod::JChTrigSels, aod::EvSels>>::iterator const& collision,
-            soa::Filtered<JetTracks> const& tracks, o2::soa::Filtered<soa::Join<o2::aod::ChargedJets, aod::ChargedJetConstituents>> const& jets)
+            soa::Filtered<soa::Join<JetTracks, aod::JTrackPIs>> const& tracks, o2::soa::Filtered<soa::Join<o2::aod::ChargedJets, aod::ChargedJetConstituents>> const& jets, soa::Join<aod::Tracks, aod::TracksExtra> const&)
   {
 
     if (!collision.selection_bit(aod::evsel::kNoTimeFrameBorder)) {
@@ -171,10 +178,22 @@ struct ChJetTriggerQATask {
 
       for (auto& trk : tracks) { // loop over filtered tracks in full TPC volume having pT > 100 MeV
 
+        auto const& originalTrack = trk.track_as<soa::Join<aod::Tracks, aod::TracksExtra>>();
+        spectra.fill(HIST("globalP_tpcglobalPDiff_withoutcuts"), trk.p() - originalTrack.tpcInnerParam(), trk.p());
+
+        if (TMath::Abs(trk.phi() - TMath::Pi()) < phiAngleRestriction) {
+          spectra.fill(HIST("globalP_tpcglobalPDiff_withoutcuts_phirestrict"), trk.p() - originalTrack.tpcInnerParam(), trk.p());
+        }
+
         if (!jetderiveddatautilities::selectTrack(trk, trackSelection)) {
           continue;
         }
 
+        spectra.fill(HIST("globalP_tpcglobalPDiff"), trk.p() - originalTrack.tpcInnerParam(), trk.p());
+        if (TMath::Abs(trk.phi() - TMath::Pi()) < phiAngleRestriction) {
+          spectra.fill(HIST("globalP_tpcglobalPDiff_phirestrict"), trk.p() - originalTrack.tpcInnerParam(), trk.p());
+        }
+
         spectra.fill(
           HIST("ptphiTrackInclGood"), trk.pt(),
           trk.phi()); // Inclusive Track pT vs phi spectrum in TPC volume