GLOQC: Add PV contrib vs ITS tracks

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,8 @@ class MatchITSTPCQC
   TH3F* getHistoK0MassVsPtVsOccpp() const { return mK0MassVsPtVsOccpp; }
   TH3F* getHistoK0MassVsPtVsOccPbPb() const { return mK0MassVsPtVsOccPbPb; }
 
+  auto getHistoPVNContVsITSTracks() const { return mPVNContVsITSTracks; }
+
   void getHistos(TObjArray& objar);
 
   /// \brief Publishes the histograms to the publisher e.g. the one provided by the QC task
@@ -247,6 +250,8 @@ class MatchITSTPCQC
       publisher->startPublishing(mK0MassVsPtVsOccpp);
       publisher->startPublishing(mK0MassVsPtVsOccPbPb);
     }
+
+    publisher->startPublishing(mPVNContVsITSTracks);
   }
 
   void setTrkSources(GID::mask_t src) { mSrc = src; }
@@ -315,6 +320,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 +464,10 @@ 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};
+
+  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,9 @@ void MatchITSTPCQC::deleteHistograms()
   // K0
   delete mK0MassVsPtVsOccpp;
   delete mK0MassVsPtVsOccPbPb;
+
+  // PV
+  delete mPVNContVsITSTracks;
 }
 
 //__________________________________________________________
@@ -220,6 +224,8 @@ void MatchITSTPCQC::reset()
     mK0MassVsPtVsOccpp->Reset();
     mK0MassVsPtVsOccPbPb->Reset();
   }
+
+  mPVNContVsITSTracks->Reset();
 }
 
 //__________________________________________________________
@@ -440,6 +446,24 @@ 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);
   }
 
+  Double_t* xbinsPV = new Double_t[100 + 1];
+  Double_t xlogminPV = TMath::Log10(1);
+  Double_t xlogmaxPV = TMath::Log10(10000);
+  Double_t dlogxPV = (xlogmaxPV - xlogminPV) / 100;
+  for (int i = 0; i <= 100; i++) {
+    Double_t xlogPV = xlogminPV + i * dlogxPV;
+    xbinsPV[i] = TMath::Exp(TMath::Log(10) * xlogPV);
+  }
+  Double_t* xbinsITS = new Double_t[100 + 1];
+  Double_t xlogminITS = TMath::Log10(1);
+  Double_t xlogmaxITS = TMath::Log10(15000);
+  Double_t dlogxITS = (xlogmaxITS - xlogminITS) / 100;
+  for (int i = 0; i <= 100; i++) {
+    Double_t xlogITS = xlogminITS + i * dlogxITS;
+    xbinsITS[i] = TMath::Exp(TMath::Log(10) * xlogITS);
+  }
+  mPVNContVsITSTracks = new TH2F("mPVNContVsITSTracks", "PV NCont. vs ITS Tracks;PV NCont.;ITS Tracks", 100, xbinsPV, 100, xbinsITS);
+
   LOG(info) << "Printing configuration cuts";
   printParams();
 
@@ -448,6 +472,8 @@ bool MatchITSTPCQC::init()
   delete[] ybinsMassK0;
   delete[] zbinsMultK0pp;
   delete[] zbinsMultK0PbPb;
+  delete[] xbinsPV;
+  delete[] xbinsITS;
 
   return true;
 }
@@ -516,10 +542,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];
+    mPVNContVsITSTracks->Fill(pv.getNContributors(), m.getEntriesOfSource(dataformats::GlobalTrackID::ITS));
+  }
 
   // cache selection for TPC and ITS tracks
   std::vector<bool> isTPCTrackSelectedEntry(mTPCTracks.size(), false);
@@ -991,11 +1026,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 +1454,9 @@ void MatchITSTPCQC::getHistos(TObjArray& objar)
   // V0
   objar.Add(mK0MassVsPtVsOccpp);
   objar.Add(mK0MassVsPtVsOccPbPb);
+
+  // PV Multiplicity
+  objar.Add(mPVNContVsITSTracks);
 }
 
 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);