Adding DCA distributions of global tracks

Author: KotliarovAr

PWGJE/Tasks/ChJetTriggerQATask.cxx

@@ -41,8 +41,8 @@
 #include "Framework/HistogramRegistry.h"
 
 using namespace o2;
-using namespace o2::framework;
-using namespace o2::framework::expressions;
+using namespace framework;
+using namespace framework::expressions;
 
 // What this task should do
 // Event by event fill
@@ -57,15 +57,10 @@ using namespace o2::framework::expressions;
 struct ChJetTriggerQATask {
 
   Configurable<std::string> evSel{"evSel", "sel8", "choose event selection"};
-  Configurable<float> cfgVertexCut{"cfgVertexCut", 10.0,
-                                   "Accepted z-vertex range"};
-  Configurable<float> cfgTPCVolume{"cfgTPCVolume", 0.9,
-                                   "Full eta range"}; // eta range of TPC
-  Configurable<float> cfgJetR{"cfgJetR", 0.4,
-                              "jet resolution parameter"}; // jet cone radius
-  Configurable<float> cfgJetPtMin{
-    "cfgJetPtMin", 0.15,
-    "minimum jet pT constituent cut"}; // minimum jet constituent pT
+  Configurable<float> cfgVertexCut{"cfgVertexCut", 10.0, "Accepted z-vertex range"};
+  Configurable<float> cfgTPCVolume{"cfgTPCVolume", 0.9, "Full eta range"}; // eta range of TPC
+  Configurable<float> cfgJetR{"cfgJetR", 0.4, "jet resolution parameter"}; // jet cone radius
+  Configurable<float> cfgJetPtMin{"cfgJetPtMin", 0.15, "minimum jet pT constituent cut"}; // minimum jet constituent pT
 
   Configurable<float> cfgTrackPhiMinCut{"cfgTrackPhiMinCut", -999, "track min phi cut"};
   Configurable<float> cfgTrackPhiMaxCut{"cfgTrackPhiMaxCut", 999, "track max phi cut"};
@@ -75,19 +70,21 @@ struct ChJetTriggerQATask {
   Configurable<bool> bHighPtTrigger{"bHighPtTrigger", false, "charged jet high pT trigger selection"};
   Configurable<bool> bTrackLowPtTrigger{"bTrackLowPtTrigger", false, "track low pT trigger selection"};
   Configurable<bool> bTrackHighPtTrigger{"bTrackHighPtTrigger", false, "track high pT trigger selection"};
-
   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"};
 
+  ConfigurableAxis dcaXY_Binning{"dcaXY_Binning", {100, -5., 5.}, ""}; 
+  ConfigurableAxis dcaZ_Binning{"dcaZ_Binning", {100, -5., 5.}, ""}; 
+
   float fiducialVolume; // 0.9 - jetR
 
   HistogramRegistry spectra;
 
   int eventSelection = -1;
   int trackSelection = -1;
-  void init(o2::framework::InitContext&)
-  {
+
+  void init(InitContext&) {
     fiducialVolume = static_cast<float>(cfgTPCVolume) - static_cast<float>(cfgJetR);
     eventSelection = jetderiveddatautilities::initialiseEventSelection(static_cast<std::string>(evSel));
     trackSelection = jetderiveddatautilities::initialiseTrackSelection(static_cast<std::string>(trackSelections));
@@ -113,6 +110,11 @@ struct ChJetTriggerQATask {
     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.}}});
     spectra.add("global1overP_tpcglobalPDiff_phirestrict", "difference of 1/p global and TPC inner momentum vs global momentum with selection applied restricted phi", {HistType::kTH2F, {{100, 0., +100.}, {500, -8., +8.}}});
 
+    spectra.add("DCAx_track_Phi_pT", "track DCAx vs phi & pT of global tracks w. nITSClusters #ge 4", kTH3F, {dcaXY_Binning, {60, 0., TMath::TwoPi()}, {100, 0., 100.}});
+    spectra.add("DCAy_track_Phi_pT", "track DCAy vs phi & pT of global tracks w. nITSClusters #ge 4", kTH3F, {dcaXY_Binning, {60, 0., TMath::TwoPi()}, {100, 0., 100.}});
+    spectra.add("DCAz_track_Phi_pT", "track DCAz vs phi & pT of global tracks w. nITSClusters #ge 4", kTH3F, {dcaZ_Binning, {60, 0., TMath::TwoPi()}, {100, 0., 100.}});
+    spectra.add("nITSClusters_TrackPt", "Number of ITS hits vs phi & pT of global tracks", kTH3F, {{7, 1., 8.}, {60, 0., TMath::TwoPi()}, {100, 0., 100.}});
+
     // Supplementary plots
     if (bAddSupplementHistosToOutput) {
       spectra.add("ptJetChInclFullVol", "inclusive charged jet pT in full volume", {HistType::kTH1F, {{200, 0., +200.}}});
@@ -137,14 +139,14 @@ struct ChJetTriggerQATask {
   Filter trackFilter = (nabs(aod::jtrack::eta) < static_cast<float>(cfgTPCVolume)) && (aod::jtrack::phi > static_cast<float>(cfgTrackPhiMinCut)) && (aod::jtrack::phi < static_cast<float>(cfgTrackPhiMaxCut)) && (aod::jtrack::pt > static_cast<float>(cfgJetPtMin));
 
   // declare filters on jets
-  Filter jetRadiusSelection = (o2::aod::jet::r == nround(cfgJetR.node() * 100.0f));
-
-  using filteredJets = o2::soa::Filtered<o2::aod::ChargedJets>;
-
-  void
-    process(soa::Filtered<soa::Join<JetCollisions,
-                                    aod::JChTrigSels, aod::EvSels>>::iterator const& collision,
-            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&)
+  Filter jetRadiusSelection = (aod::jet::r == nround(cfgJetR.node() * 100.0f));
+    
+  using filteredColl = soa::Filtered<soa::Join<JetCollisions, aod::JChTrigSels, aod::EvSels>>::iterator;
+  using filteredJTracks = soa::Filtered<soa::Join<aod::JTracks, aod::JTrackPIs>>;
+  // using Tracks = soa::Join<aod::Tracks, aod::TracksExtra>; // could be deleted? it is not iterator
+  using filteredJets = soa::Filtered<soa::Join<aod::ChargedJets, aod::ChargedJetConstituents>>;
+
+  void process(filteredColl const& collision, filteredJTracks const& tracks, filteredJets const& jets/*, Tracks const&*/)
   {
 
     if (!collision.selection_bit(aod::evsel::kNoTimeFrameBorder)) {
@@ -179,9 +181,10 @@ struct ChJetTriggerQATask {
       spectra.fill(HIST("vertexZ"),
                    collision.posZ()); // Inclusive Track Cross TPC Rows
 
-      for (auto& trk : tracks) { // loop over filtered tracks in full TPC volume having pT > 100 MeV
+      for (auto const& 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, aod::TracksDCA>>();
 
-        auto const& originalTrack = trk.track_as<soa::Join<aod::Tracks, aod::TracksExtra>>();
         spectra.fill(HIST("globalP_tpcglobalPDiff_withoutcuts"), trk.p(), trk.p() - originalTrack.tpcInnerParam());
 
         if (TMath::Abs(trk.phi() - TMath::Pi()) < phiAngleRestriction) {
@@ -192,6 +195,16 @@ struct ChJetTriggerQATask {
           continue;
         }
 
+        if(originalTrack.itsNCls() >= 4){ // correspond to number of track hits in ITS layers 
+          auto dcaX = originalTrack.x() - collision.posX();
+          auto dcaY = originalTrack.y() - collision.posY();
+          spectra.fill(HIST("DCAx_track_Phi_pT"), dcaX, trk.phi(), trk.pt());
+          spectra.fill(HIST("DCAy_track_Phi_pT"), dcaY, trk.phi(), trk.pt());
+          spectra.fill(HIST("DCAz_track_Phi_pT"), originalTrack.dcaZ(), trk.phi(), trk.pt());
+        }
+
+        spectra.fill(HIST("nITSClusters_TrackPt"), originalTrack.itsNCls(), trk.phi(), trk.pt());
+
         spectra.fill(HIST("globalP_tpcglobalPDiff"), trk.p(), trk.p() - originalTrack.tpcInnerParam());
         if (trk.p() > 0 && originalTrack.tpcInnerParam() > 0) {
           spectra.fill(HIST("global1overP_tpcglobalPDiff"), trk.p(), 1. / trk.p() - 1. / originalTrack.tpcInnerParam());
@@ -308,4 +321,4 @@ WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)
 
   return WorkflowSpec{adaptAnalysisTask<ChJetTriggerQATask>(
     cfgc, TaskName{"jet-charged-trigger-qa"})};
-}
+}