Skip to content

Commit

Permalink
Adding plots to ITSTPC matching QC
Browse files Browse the repository at this point in the history
  • Loading branch information
@chiarazampolli
chiarazampolli committed Sep 29, 2023
1 parent 0804624 commit c9df921
Show file tree
Hide file tree
Showing 2 changed files with 627 additions and 270 deletions.
171 changes: 114 additions & 57 deletions Detectors/GlobalTracking/include/GlobalTracking/MatchITSTPCQC.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,7 @@
#include "Steer/MCKinematicsReader.h"
#include <unordered_map>
#include <vector>
#include <array>

namespace o2
{
Expand All @@ -46,6 +47,10 @@ struct LblInfo {
class MatchITSTPCQC
{
public:
enum matchType : int { TPC = 0,
ITS,
SIZE };

MatchITSTPCQC() = default;
~MatchITSTPCQC();

Expand All @@ -56,37 +61,41 @@ class MatchITSTPCQC
void finalize();
void reset();

TH1D* getHistoPt() const { return mPt; }
TH1D* getHistoPtTPC() const { return mPtTPC; }
TEfficiency* getFractionITSTPCmatch() const { return mFractionITSTPCmatch; }
TH1D* getHistoPtNum(matchType m) const { return mPtNum[m]; }
TH1D* getHistoPtDen(matchType m) const { return mPtDen[m]; }
TEfficiency* getFractionITSTPCmatch(matchType m) const { return mFractionITSTPCmatch[m]; }

TH1D* getHistoPtNumNoEta0(matchType m) const { return mPtNum_noEta0[m]; }
TH1D* getHistoPtDenNoEta0(matchType m) const { return mPtDen_noEta0[m]; }
TEfficiency* getFractionITSTPCmatchNoEta0(matchType m) const { return mFractionITSTPCmatch_noEta0[m]; }

TH1F* getHistoPhi() const { return mPhi; }
TH1F* getHistoPhiTPC() const { return mPhiTPC; }
TEfficiency* getFractionITSTPCmatchPhi() const { return mFractionITSTPCmatchPhi; }
TH1F* getHistoPhiNum(matchType m) const { return mPhiNum[m]; }
TH1F* getHistoPhiDen(matchType m) const { return mPhiDen[m]; }
TEfficiency* getFractionITSTPCmatchPhi(matchType m) const { return mFractionITSTPCmatchPhi[m]; }

TH2F* getHistoPhiVsPt() const { return mPhiVsPt; }
TH2F* getHistoPhiVsPtTPC() const { return mPhiVsPtTPC; }
TEfficiency* getFractionITSTPCmatchPhiVsPt() const { return mFractionITSTPCmatchPhiVsPt; }
TH2F* getHistoPhiVsPtNum(matchType m) const { return mPhiVsPtNum[m]; }
TH2F* getHistoPhiVsPtDen(matchType m) const { return mPhiVsPtDen[m]; }
TEfficiency* getFractionITSTPCmatchPhiVsPt(matchType m) const { return mFractionITSTPCmatchPhiVsPt[m]; }

TH1F* getHistoEta() const { return mEta; }
TH1F* getHistoEtaTPC() const { return mEtaTPC; }
TEfficiency* getFractionITSTPCmatchEta() const { return mFractionITSTPCmatchEta; }
TH1F* getHistoEtaNum(matchType m) const { return mEtaNum[m]; }
TH1F* getHistoEtaDen(matchType m) const { return mEtaDen[m]; }
TEfficiency* getFractionITSTPCmatchEta(matchType m) const { return mFractionITSTPCmatchEta[m]; }

TH2F* getHistoEtaVsPt() const { return mEtaVsPt; }
TH2F* getHistoEtaVsPtTPC() const { return mEtaVsPtTPC; }
TEfficiency* getFractionITSTPCmatchEtaVsPt() const { return mFractionITSTPCmatchEtaVsPt; }
TH2F* getHistoEtaVsPtNum(matchType m) const { return mEtaVsPtNum[m]; }
TH2F* getHistoEtaVsPtDen(matchType m) const { return mEtaVsPtDen[m]; }
TEfficiency* getFractionITSTPCmatchEtaVsPt(matchType m) const { return mFractionITSTPCmatchEtaVsPt[m]; }

TH1F* getHistoPtPhysPrim() const { return mPtPhysPrim; }
TH1F* getHistoPtTPCPhysPrim() const { return mPtTPCPhysPrim; }
TEfficiency* getFractionITSTPCmatchPhysPrim() const { return mFractionITSTPCmatchPhysPrim; }
TH1F* getHistoPtPhysPrimNum(matchType m) const { return mPtPhysPrimNum[m]; }
TH1F* getHistoPtPhysPrimDen(matchType m) const { return mPtPhysPrimDen[m]; }
TEfficiency* getFractionITSTPCmatchPhysPrim(matchType m) const { return mFractionITSTPCmatchPhysPrim[m]; }

TH1F* getHistoPhiPhysPrim() const { return mPhiPhysPrim; }
TH1F* getHistoPhiTPCPhysPrim() const { return mPhiTPCPhysPrim; }
TEfficiency* getFractionITSTPCmatchPhiPhysPrim() const { return mFractionITSTPCmatchPhiPhysPrim; }
TH1F* getHistoPhiPhysPrimNum(matchType m) const { return mPhiPhysPrimNum[m]; }
TH1F* getHistoPhiPhysPrimDen(matchType m) const { return mPhiPhysPrimDen[m]; }
TEfficiency* getFractionITSTPCmatchPhiPhysPrim(matchType m) const { return mFractionITSTPCmatchPhiPhysPrim[m]; }

TH1F* getHistoEtaPhysPrim() const { return mEtaPhysPrim; }
TH1F* getHistoEtaTPCPhysPrim() const { return mEtaTPCPhysPrim; }
TEfficiency* getFractionITSTPCmatchEtaPhysPrim() const { return mFractionITSTPCmatchEtaPhysPrim; }
TH1F* getHistoEtaPhysPrimNum(matchType m) const { return mEtaPhysPrimNum[m]; }
TH1F* getHistoEtaPhysPrimDen(matchType m) const { return mEtaPhysPrimDen[m]; }
TEfficiency* getFractionITSTPCmatchEtaPhysPrim(matchType m) const { return mFractionITSTPCmatchEtaPhysPrim[m]; }

TH2F* getHistoResidualPt() const { return mResidualPt; }
TH2F* getHistoResidualPhi() const { return mResidualPhi; }
Expand All @@ -95,7 +104,18 @@ class MatchITSTPCQC
TH1F* getHistoChi2Matching() const { return mChi2Matching; }
TH1F* getHistoChi2Refit() const { return mChi2Refit; }
TH2F* getHistoTimeResVsPt() const { return mTimeResVsPt; }
TH1F* getHistoDCAr() const { return mDCAr; }

TH1D* getHisto1OverPtNum(matchType m) const { return m1OverPtNum[m]; }
TH1D* getHisto1OverPtDen(matchType m) const { return m1OverPtDen[m]; }
TEfficiency* getFractionITSTPCmatch1OverPt(matchType m) const { return mFractionITSTPCmatch1OverPt[m]; }

TH1D* getHisto1OverPtPhysPrimNum(matchType m) const { return m1OverPtPhysPrimNum[m]; }
TH1D* getHisto1OverPtPhysPrimDen(matchType m) const { return m1OverPtPhysPrimDen[m]; }
TEfficiency* getFractionITSTPCmatchPhysPrim1OverPt(matchType m) const { return mFractionITSTPCmatchPhysPrim1OverPt[m]; }

void getHistos(TObjArray& objar);

void setSources(GID::mask_t src) { mSrc = src; }
void setUseMC(bool b) { mUseMC = b; }
bool getUseMC() const { return mUseMC; }
Expand All @@ -115,52 +135,80 @@ class MatchITSTPCQC
void setMinDCAtoBeamPipeDistanceCut(float v) { mDCACut = v; }
void setMinDCAtoBeamPipeYCut(float v) { mDCACutY = v; }

// to remove after merging QC PR
TH1D* getHistoPt() const { return nullptr; } // old
TH1D* getHistoPtTPC() const { return nullptr; } // old
TEfficiency* getFractionITSTPCmatch() const { return nullptr; } // old
TH1F* getHistoPhi() const { return nullptr; } // old
TH1F* getHistoPhiTPC() const { return nullptr; } // old
TEfficiency* getFractionITSTPCmatchPhi() const { return nullptr; } // old
TH2F* getHistoPhiVsPt() const { return nullptr; } // old
TEfficiency* getFractionITSTPCmatchPhiVsPt() const { return nullptr; } // old
TH1F* getHistoEta() const { return nullptr; } // old
TEfficiency* getFractionITSTPCmatchEta() const { return nullptr; } // old
TH2F* getHistoEtaVsPt() const { return nullptr; } // old
TH2F* getHistoEtaVsPtTPC() const { return nullptr; } // old
TEfficiency* getFractionITSTPCmatchEtaVsPt() const { return nullptr; } // old
TH1F* getHistoPtPhysPrim() const { return nullptr; } // old
TH1F* getHistoPtTPCPhysPrim() const { return nullptr; } // old
TEfficiency* getFractionITSTPCmatchPhysPrim() const { return nullptr; } // old
TH1F* getHistoPhiPhysPrim() const { return nullptr; } // old
TEfficiency* getFractionITSTPCmatchPhiPhysPrim() const { return nullptr; } // old
TH1F* getHistoEtaPhysPrim() const { return nullptr; } // old
TH1F* getHistoEtaTPCPhysPrim() const { return nullptr; } // old
TEfficiency* getFractionITSTPCmatchEtaPhysPrim() const { return nullptr; } // old

private:
std::shared_ptr<o2::globaltracking::DataRequest> mDataRequest;
o2::globaltracking::RecoContainer mRecoCont;
GID::mask_t mSrc = GID::getSourcesMask("TPC,ITS-TPC");
GID::mask_t mAllowedSources = GID::getSourcesMask("TPC,ITS-TPC");
GID::mask_t mSrc = GID::getSourcesMask("ITS,TPC,ITS-TPC");
GID::mask_t mAllowedSources = GID::getSourcesMask("ITS,TPC,ITS-TPC");
// TPC
gsl::span<const o2::tpc::TrackTPC> mTPCTracks;
// ITS
gsl::span<const o2::its::TrackITS> mITSTracks;
// ITS-TPC
gsl::span<const o2::dataformats::TrackTPCITS> mITSTPCTracks;
bool mUseMC = false;
float mBz = 0; ///< nominal Bz
std::unordered_map<o2::MCCompLabel, LblInfo> mMapLabels; // map with labels that have been found for the matched ITSTPC tracks; key is the label,
// value is the LbLinfo with the id of the track with the highest pT found with that label so far,
// and the flag to say if it is a physical primary or not
std::unordered_map<o2::MCCompLabel, LblInfo> mMapTPCLabels; // map with labels that have been found for the unmatched TPC tracks; key is the label,
// value is the LblInfo with the id of the track with the highest number of TPC clusters found
// with that label so far, and the flag to say if it is a physical primary or not
std::array<std::unordered_map<o2::MCCompLabel, LblInfo>, matchType::SIZE> mMapLabels; // map with labels that have been found for the matched ITSTPC tracks; key is the label,
// value is the LbLinfo with the id of the track with the highest pT found with that label so far,
// and the flag to say if it is a physical primary or not
std::array<std::unordered_map<o2::MCCompLabel, LblInfo>, matchType::SIZE> mMapRefLabels; // map with labels that have been found for the unmatched TPC tracks; key is the label,
// value is the LblInfo with the id of the track with the highest number of TPC clusters found
// with that label so far, and the flag to say if it is a physical primary or not
o2::steer::MCKinematicsReader mcReader; // reader of MC information

// Pt
TH1D* mPt = nullptr;
TH1D* mPtTPC = nullptr;
TEfficiency* mFractionITSTPCmatch = nullptr;
TH1F* mPtPhysPrim = nullptr;
TH1F* mPtTPCPhysPrim = nullptr;
TEfficiency* mFractionITSTPCmatchPhysPrim = nullptr;
TH1D* mPtNum[matchType::SIZE] = {};
TH1D* mPtDen[matchType::SIZE] = {};
TEfficiency* mFractionITSTPCmatch[matchType::SIZE] = {};
TH1D* mPtNum_noEta0[matchType::SIZE] = {};
TH1D* mPtDen_noEta0[matchType::SIZE] = {};
TEfficiency* mFractionITSTPCmatch_noEta0[matchType::SIZE] = {};
TH1F* mPtPhysPrimNum[matchType::SIZE] = {};
TH1F* mPtPhysPrimDen[matchType::SIZE] = {};
TEfficiency* mFractionITSTPCmatchPhysPrim[matchType::SIZE] = {};
// Phi
TH1F* mPhi = nullptr;
TH1F* mPhiTPC = nullptr;
TEfficiency* mFractionITSTPCmatchPhi = nullptr;
TH1F* mPhiPhysPrim = nullptr;
TH1F* mPhiTPCPhysPrim = nullptr;
TEfficiency* mFractionITSTPCmatchPhiPhysPrim = nullptr;
TH2F* mPhiVsPt = nullptr;
TH2F* mPhiVsPtTPC = nullptr;
TEfficiency* mFractionITSTPCmatchPhiVsPt = nullptr;
TH1F* mPhiNum[matchType::SIZE] = {};
TH1F* mPhiDen[matchType::SIZE] = {};
TEfficiency* mFractionITSTPCmatchPhi[matchType::SIZE] = {};
TH1F* mPhiPhysPrimNum[matchType::SIZE] = {};
TH1F* mPhiPhysPrimDen[matchType::SIZE] = {};
TEfficiency* mFractionITSTPCmatchPhiPhysPrim[matchType::SIZE] = {};
TH2F* mPhiVsPtNum[matchType::SIZE] = {};
TH2F* mPhiVsPtDen[matchType::SIZE] = {};
TEfficiency* mFractionITSTPCmatchPhiVsPt[matchType::SIZE] = {};
// Eta
TH1F* mEta = nullptr;
TH1F* mEtaTPC = nullptr;
TEfficiency* mFractionITSTPCmatchEta = nullptr;
TH1F* mEtaPhysPrim = nullptr;
TH1F* mEtaTPCPhysPrim = nullptr;
TEfficiency* mFractionITSTPCmatchEtaPhysPrim = nullptr;
TH2F* mEtaVsPt = nullptr;
TH2F* mEtaVsPtTPC = nullptr;
TEfficiency* mFractionITSTPCmatchEtaVsPt = nullptr;
TH1F* mEtaNum[matchType::SIZE] = {};
TH1F* mEtaDen[matchType::SIZE] = {};
TEfficiency* mFractionITSTPCmatchEta[matchType::SIZE] = {};
TH1F* mEtaPhysPrimNum[matchType::SIZE] = {};
TH1F* mEtaPhysPrimDen[matchType::SIZE] = {};
TEfficiency* mFractionITSTPCmatchEtaPhysPrim[matchType::SIZE] = {};
TH2F* mEtaVsPtNum[matchType::SIZE] = {};
TH2F* mEtaVsPtDen[matchType::SIZE] = {};
TEfficiency* mFractionITSTPCmatchEtaVsPt[matchType::SIZE] = {};
// Residuals
TH2F* mResidualPt = nullptr;
TH2F* mResidualPhi = nullptr;
Expand All @@ -169,11 +217,20 @@ class MatchITSTPCQC
TH1F* mChi2Matching = nullptr;
TH1F* mChi2Refit = nullptr;
TH2F* mTimeResVsPt = nullptr;
TH1F* mDCAr = nullptr;
// 1/Pt
TH1D* m1OverPtNum[matchType::SIZE] = {};
TH1D* m1OverPtDen[matchType::SIZE] = {};
TEfficiency* mFractionITSTPCmatch1OverPt[matchType::SIZE] = {};
TH1D* m1OverPtPhysPrimNum[matchType::SIZE] = {};
TH1D* m1OverPtPhysPrimDen[matchType::SIZE] = {};
TEfficiency* mFractionITSTPCmatchPhysPrim1OverPt[matchType::SIZE] = {};

void setEfficiency(TEfficiency* eff, TH1* hnum, TH1* hden, bool is2D = false);

int mNTPCSelectedTracks = 0;
int mNITSTPCSelectedTracks = 0;
int mNITSSelectedTracks = 0;
int mNITSTPCSelectedTracks[matchType::SIZE] = {0, 0};

// cut values
float mPtCut = 0.1f;
Expand Down
Loading

0 comments on commit c9df921

Please sign in to comment.