GLOQC: Add PV contrib vs ITS tracks

Signed-off-by: Felix Schlepper <[email protected]>

Author: f3sch

Detectors/GLOQC/include/GLOQC/MatchITSTPCQC.h

@@ -74,6 +74,7 @@ class MatchITSTPCQC
   bool processV0(int iv, o2::globaltracking::RecoContainer& recoData, std::vector<float>& mTBinClOcc, float pvTime);
   bool refitV0(const o2::dataformats::V0Index& id, o2::dataformats::V0& v0, o2::globaltracking::RecoContainer& recoData);
 
+  // returning raw pointer but this task keeps ownership!
   TH1D* getHistoPtNum(matchType m) const { return mPtNum[m]; }
   TH1D* getHistoPtDen(matchType m) const { return mPtDen[m]; }
   TEfficiency* getFractionITSTPCmatch(matchType m) const { return mFractionITSTPCmatch[m]; }
@@ -144,6 +145,9 @@ class MatchITSTPCQC
   TH3F* getHistoK0MassVsPtVsOccpp() const { return mK0MassVsPtVsOccpp; }
   TH3F* getHistoK0MassVsPtVsOccPbPb() const { return mK0MassVsPtVsOccPbPb; }
 
+  auto getHistoPVNContVsITSTracks() const { return mPVNContVsITSTracks; }
+  auto getHistoPVNContVsITSTracksPbPb() const { return mPVNContVsITSTracksPbPb; }
+
   void getHistos(TObjArray& objar);
 
   /// \brief Publishes the histograms to the publisher e.g. the one provided by the QC task
@@ -247,6 +251,9 @@ class MatchITSTPCQC
       publisher->startPublishing(mK0MassVsPtVsOccpp);
       publisher->startPublishing(mK0MassVsPtVsOccPbPb);
     }
+
+    publisher->startPublishing(mPVNContVsITSTracks);
+    publisher->startPublishing(mPVNContVsITSTracksPbPb);
   }
 
   void setTrkSources(GID::mask_t src) { mSrc = src; }
@@ -315,6 +322,8 @@ class MatchITSTPCQC
   gsl::span<const o2::its::TrackITS> mITSTracks;
   // ITS-TPC
   gsl::span<const o2::dataformats::TrackTPCITS> mITSTPCTracks;
+  // PVs
+  gsl::span<const o2::dataformats::PrimaryVertex> mPVs;
   bool mUseMC = false;                                                                     // Usage of the MC information
   bool mUseTrkPID = false;                                                                 // Usage of the PID hypothesis in tracking
   float mBz = 0;                                                                           ///< nominal Bz
@@ -457,7 +466,11 @@ class MatchITSTPCQC
   float mK0MaxDCA = 0.01;      // max DCA to select the K0
   float mK0MinCosPA = 0.995;   // min cosPA to select the K0
 
-  ClassDefNV(MatchITSTPCQC, 4);
+  // for Mutliplicty PV
+  TH2F* mPVNContVsITSTracks{nullptr};
+  TH2F* mPVNContVsITSTracksPbPb{nullptr};
+
+  ClassDefNV(MatchITSTPCQC, 5);
 };
 } // namespace gloqc
 } // namespace o2

Detectors/GLOQC/src/MatchITSTPCQC.cxx

@@ -13,6 +13,7 @@
 
 #include "GLOQC/MatchITSTPCQC.h"
 #include "ReconstructionDataFormats/TrackTPCITS.h"
+#include "ReconstructionDataFormats/VtxTrackRef.h"
 #include "DataFormatsTPC/TrackTPC.h"
 #include "DetectorsBase/Propagator.h"
 #include "SimulationDataFormat/MCUtils.h"
@@ -133,6 +134,10 @@ void MatchITSTPCQC::deleteHistograms()
   // K0
   delete mK0MassVsPtVsOccpp;
   delete mK0MassVsPtVsOccPbPb;
+
+  // PV
+  delete mPVNContVsITSTracks;
+  delete mPVNContVsITSTracksPbPb;
 }
 
 //__________________________________________________________
@@ -220,6 +225,9 @@ void MatchITSTPCQC::reset()
     mK0MassVsPtVsOccpp->Reset();
     mK0MassVsPtVsOccPbPb->Reset();
   }
+
+  mPVNContVsITSTracks->Reset();
+  mPVNContVsITSTracksPbPb->Reset();
 }
 
 //__________________________________________________________
@@ -440,6 +448,29 @@ bool MatchITSTPCQC::init()
     mK0MassVsPtVsOccPbPb = new TH3F("mK0MassVsPtVsOccPbPb", "K0 invariant mass vs Pt vs TPC occupancy; Pt [GeV/c]; K0s mass [GeV/c^2]; TPC occ", nbinsPtK0, xbinsPtK0, nbinsMassK0, ybinsMassK0, nbinsMultK0PbPb, zbinsMultK0PbPb);
   }
 
+  auto makeBins = [](int out) {
+    std::array<Double_t, 101> xbins{0};
+    Double_t xlogmin = TMath::Log10(1);
+    Double_t xlogmax = TMath::Log10(out);
+    Double_t dlogx = (xlogmax - xlogmin) / 100;
+    for (int i = 0; i <= 100; i++) {
+      Double_t xlog = xlogmin + (i * dlogx);
+      xbins[i] = TMath::Exp(TMath::Log(10) * xlog);
+    }
+    return xbins;
+  };
+
+  {
+    auto xbinsPV = makeBins(250);
+    auto xbinsITS = makeBins(500);
+    mPVNContVsITSTracks = new TH2F("mPVNContVsITSTracks", "PV NCont. vs ITS Tracks;PV NCont.;ITS Tracks", 100, xbinsPV.data(), 100, xbinsITS.data());
+  }
+  {
+    auto xbinsPV = makeBins(10'000);
+    auto xbinsITS = makeBins(20'000);
+    mPVNContVsITSTracksPbPb = new TH2F("mPVNContVsITSTracksPbPb", "PV NCont. vs ITS Tracks;PV NCont.;ITS Tracks", 100, xbinsPV.data(), 100, xbinsITS.data());
+  }
+
   LOG(info) << "Printing configuration cuts";
   printParams();
 
@@ -516,10 +547,19 @@ void MatchITSTPCQC::run(o2::framework::ProcessingContext& ctx)
   mTPCTracks = mRecoCont.getTPCTracks();
   mITSTracks = mRecoCont.getITSTracks();
   mITSTPCTracks = mRecoCont.getTPCITSTracks();
+  mPVs = mRecoCont.getPrimaryVertices();
 
   LOG(info) << "****** Number of found ITSTPC tracks = " << mITSTPCTracks.size();
   LOG(info) << "****** Number of found TPC    tracks = " << mTPCTracks.size();
   LOG(info) << "****** Number of found ITS    tracks = " << mITSTracks.size();
+  LOG(info) << "****** Number of PVs                 = " << mPVs.size();
+
+  const auto& pvm = mRecoCont.getPrimaryVertexMatchedTrackRefs();
+  for (int i{0}; i < mPVs.size(); ++i) {
+    const auto& pv = mPVs[i];
+    const auto& m = pvm[i];
+    ((mIsHI) ? mPVNContVsITSTracksPbPb : mPVNContVsITSTracks)->Fill(pv.getNContributors(), mITSTracks.size());
+  }
 
   // cache selection for TPC and ITS tracks
   std::vector<bool> isTPCTrackSelectedEntry(mTPCTracks.size(), false);
@@ -991,11 +1031,9 @@ void MatchITSTPCQC::run(o2::framework::ProcessingContext& ctx)
     }
   }
 
-  if (mDoK0QC && mRecoCont.getPrimaryVertices().size() > 0) {
+  if (mDoK0QC && mPVs.size() > 0) {
     // now doing K0S
     mFitterV0.setBz(mBz);
-    const auto pvertices = mRecoCont.getPrimaryVertices();
-    LOG(info) << "****** Number of PVs                 = " << pvertices.size();
 
     // getting occupancy estimator
     mNHBPerTF = o2::base::GRPGeomHelper::instance().getGRPECS()->getNHBFPerTF();
@@ -1421,6 +1459,10 @@ void MatchITSTPCQC::getHistos(TObjArray& objar)
   // V0
   objar.Add(mK0MassVsPtVsOccpp);
   objar.Add(mK0MassVsPtVsOccPbPb);
+
+  // PV Multiplicity
+  objar.Add(mPVNContVsITSTracks);
+  objar.Add(mPVNContVsITSTracksPbPb);
 }
 
 void MatchITSTPCQC::printParams() const

Detectors/GlobalTrackingWorkflow/qc/src/ITSTPCMatchingQCSpec.cxx

@@ -37,6 +37,7 @@ void ITSTPCMatchingQCDevice::init(InitContext& /*ic*/)
   mMatchITSTPCQC = std::make_unique<o2::gloqc::MatchITSTPCQC>();
   mMatchITSTPCQC->setDataRequest(mDataRequest);
   mMatchITSTPCQC->setTrkSources(o2::dataformats::GlobalTrackID::getSourcesMask(mTrkSources));
+  mMatchITSTPCQC->setNBinsPt(params.nBinsPt);
   mMatchITSTPCQC->setMinPtITSCut(params.minPtITSCut);
   mMatchITSTPCQC->setEtaITSCut(params.etaITSCut);
   mMatchITSTPCQC->setMinNClustersITS(params.minNITSClustersCut);