From b82d4e13d5edf4e76c6cb7e80ffcc6e4780d72d0 Mon Sep 17 00:00:00 2001
From: "se.gorbunov" <se.gorbunov@gsi.de>
Date: Tue, 11 Oct 2022 07:55:07 +0000
Subject: [PATCH] L1: trd2dQa: P resolution plots vs N sts hits
---
reco/qa/CbmTrackingTrdQa.cxx | 51 ++++++++++++++++++++++++++++++++++--
reco/qa/CbmTrackingTrdQa.h | 25 +++++++++++++-----
2 files changed, 67 insertions(+), 9 deletions(-)
diff --git a/reco/qa/CbmTrackingTrdQa.cxx b/reco/qa/CbmTrackingTrdQa.cxx
index 078e179bde..1cb1fb8863 100644
--- a/reco/qa/CbmTrackingTrdQa.cxx
+++ b/reco/qa/CbmTrackingTrdQa.cxx
@@ -31,6 +31,7 @@
#include "CbmMCDataArray.h"
#include "CbmMCDataManager.h"
#include "CbmMCTrack.h"
+#include "CbmStsTrack.h"
#include "CbmTimeSlice.h"
#include "CbmTrackMatchNew.h"
#include "CbmTrdHit.h"
@@ -148,6 +149,7 @@ void CbmTrackingTrdQa::Exec(Option_t* /*opt*/)
// check consistency
//assert(fGlobalTracks->GetEntriesFast() == fGlobalTrackMatches->GetEntriesFast());
assert(fTrdTracks->GetEntriesFast() == fTrdTrackMatches->GetEntriesFast());
+ assert(fStsTracks->GetEntriesFast() == fStsTrackMatches->GetEntriesFast());
std::vector<CbmLink> events = fTimeSlice->GetMatch().GetLinks();
std::sort(events.begin(), events.end());
@@ -198,10 +200,11 @@ void CbmTrackingTrdQa::Exec(Option_t* /*opt*/)
// Get momentum
mcTrack->GetMomentum(momentum);
+ info.fP = momentum.Mag();
info.fPt = momentum.Pt();
info.fPdg = mcTrack->GetPdgCode();
info.fY = mcTrack->GetRapidity() - fYCM;
- info.fIsFast = (info.fPt > 0.1);
+ info.fIsFast = (info.fP > 0.1);
// Continue only for reconstructible tracks
@@ -276,9 +279,19 @@ void CbmTrackingTrdQa::Exec(Option_t* /*opt*/)
Int_t nAllHits = trdTrack->GetNofHits();
assert(nTrue + nWrong + nFake == nAllHits);
+ int nStsHits = 0;
+ {
+ Int_t stsTrackId = info.fStsTrackMatch;
+ if (stsTrackId >= 0) {
+ CbmStsTrack* stsTrack = (CbmStsTrack*) fStsTracks->At(stsTrackId);
+ assert(stsTrack);
+ nStsHits = stsTrack->GetNofHits();
+ }
+ }
+
double qp = globalTrack->GetParamFirst()->GetQp();
//double q = (qp >= 1.) ? 1. : -1.;
- double p = (fabs(qp) > 1. / 100.) ? 1. / fabs(qp) : 100.;
+ double p = (fabs(qp) > 1. / 1000.) ? 1. / fabs(qp) : 1000.;
double tx = globalTrack->GetParamFirst()->GetTx();
double ty = globalTrack->GetParamFirst()->GetTy();
double pt = sqrt((tx * tx + ty * ty) / (1. + tx * tx + ty * ty)) * p;
@@ -298,6 +311,14 @@ void CbmTrackingTrdQa::Exec(Option_t* /*opt*/)
fhPidPtY["prmE"][Pdg2Idx(info.fPdg)]->Fill(info.fY, info.fPt);
fhNpRecPrim->Fill(Double_t(nAllHits));
if (info.fPt > 0.001) { fhPtResPrim->Fill((pt / info.fPt - 1.)); }
+ if (info.fP > 0.001) {
+ double dp = p / info.fP - 1.;
+ fhPResPrim->Fill(dp);
+ if (nStsHits == 0) { fhPResPrimSts0->Fill(dp); }
+ if (nStsHits == 1) { fhPResPrimSts1->Fill(dp); }
+ if (nStsHits == 2) { fhPResPrimSts2->Fill(dp); }
+ if (nStsHits >= 3) { fhPResPrimSts3->Fill(dp); }
+ }
}
else {
nRecSec++;
@@ -445,6 +466,14 @@ InitStatus CbmTrackingTrdQa::Init()
fTrdTrackMatches = (TClonesArray*) fManager->GetObject("TrdTrackMatch");
assert(fTrdTrackMatches);
+ // Get StsTrack array
+ fStsTracks = (TClonesArray*) fManager->GetObject("StsTrack");
+ assert(fStsTracks);
+
+ // Get StsTrackMatch array
+ fStsTrackMatches = (TClonesArray*) fManager->GetObject("StsTrackMatch");
+ assert(fStsTrackMatches);
+
// Create histograms
CreateHistos();
@@ -729,6 +758,21 @@ void CbmTrackingTrdQa::CreateHistos()
fhPtResPrim = new TH1F("hPtPrim", "Resolution Pt Primaries [100%]", 100, -1., 1.);
fHistoList->Add(fhPtResPrim);
+ fhPResPrim = new TH1F("hPPrim", "Resolution P Primaries [100%]", 100, -1., 1.);
+ fHistoList->Add(fhPResPrim);
+
+ fhPResPrimSts0 = new TH1F("hPPrimSts0", "Resolution P Primaries [100%], No Sts hits", 100, -1., 1.);
+ fHistoList->Add(fhPResPrimSts0);
+
+ fhPResPrimSts1 = new TH1F("hPPrimSts1", "Resolution P Primaries [100%], 1 Sts hit", 100, -1., 1.);
+ fHistoList->Add(fhPResPrimSts1);
+
+ fhPResPrimSts2 = new TH1F("hPPrimSts2", "Resolution P Primaries [100%], 2 Sts hits", 100, -1., 1.);
+ fHistoList->Add(fhPResPrimSts2);
+
+ fhPResPrimSts3 = new TH1F("hPPrimSts3", "Resolution P Primaries [100%], >=3 Sts hits", 100, -1., 1.);
+ fHistoList->Add(fhPResPrimSts3);
+
TIter next(fHistoList);
while (TH1* histo = ((TH1*) next())) {
fOutFolder.Add(histo);
@@ -831,6 +875,8 @@ void CbmTrackingTrdQa::FillTrackMatchMap(Int_t& nRec, Int_t& nGhosts, Int_t& nCl
assert(trdTrack);
nRec++;
+ int iStsTrack = globalTrack->GetStsTrackIndex();
+
// --- TrackMatch
assert(iTrdTrack >= 0 && iTrdTrack < fTrdTrackMatches->GetEntriesFast());
@@ -875,6 +921,7 @@ void CbmTrackingTrdQa::FillTrackMatchMap(Int_t& nRec, Int_t& nGhosts, Int_t& nCl
}
info.fGlobalTrackMatch = iGlobalTrack;
info.fTrdTrackMatch = iTrdTrack;
+ info.fStsTrackMatch = iStsTrack;
info.fQuali = quali;
info.fMatchedNHitsAll = nHits;
info.fMatchedNHitsTrue = nTrue;
diff --git a/reco/qa/CbmTrackingTrdQa.h b/reco/qa/CbmTrackingTrdQa.h
index f2d1414f58..15de64179b 100644
--- a/reco/qa/CbmTrackingTrdQa.h
+++ b/reco/qa/CbmTrackingTrdQa.h
@@ -116,6 +116,7 @@ private:
std::map<Int_t, Int_t> fHitMap = {}; /// Trd station -> number of attached hits
Int_t fGlobalTrackMatch = -1; /// matched GlobalTrack index
Int_t fTrdTrackMatch = -1; /// matched TrdTrack index
+ Int_t fStsTrackMatch = -1; /// matched StsTrack index
Double_t fQuali = 0.; /// percentage of matched hits
Int_t fMatchedNHitsAll = 0; /// number of matched hits
Int_t fMatchedNHitsTrue = 0; /// number of matched hits
@@ -125,6 +126,7 @@ private:
Bool_t fIsFast {0};
Bool_t fIsLong {0};
Double_t fPt {0.};
+ Double_t fP {0.};
Int_t fPdg = 0;
Double_t fY = 0.;
};
@@ -146,14 +148,16 @@ private:
CbmMCDataArray* fMCTracks = nullptr; //! MCtrack
/// Trd
- Int_t fTrdNstations = 0; // Number of Trd stations
- CbmMCDataArray* fTrdPoints = nullptr; //! TrdPoints
- TClonesArray* fTrdHits = nullptr; //! TrdHits
- TClonesArray* fTrdHitMatch = nullptr; //! TrdHitMatch
- TClonesArray* fGlobalTracks = nullptr; //! GlobalTrack
+ Int_t fTrdNstations = 0; // Number of Trd stations
+ CbmMCDataArray* fTrdPoints = nullptr; //! TrdPoints
+ TClonesArray* fTrdHits = nullptr; //! TrdHits
+ TClonesArray* fTrdHitMatch = nullptr; //! TrdHitMatch
+ TClonesArray* fGlobalTracks = nullptr; //! GlobalTrack
//TClonesArray* fGlobalTrackMatches = nullptr; //! GlobalTrackMatch
- TClonesArray* fTrdTracks = nullptr; //! TrdTrack
- TClonesArray* fTrdTrackMatches = nullptr; //! TrdTrackMatch
+ TClonesArray* fTrdTracks = nullptr; //! TrdTrack
+ TClonesArray* fTrdTrackMatches = nullptr; //! TrdTrackMatch
+ TClonesArray* fStsTracks = nullptr; //! StsTrack
+ TClonesArray* fStsTrackMatches = nullptr; //! StsTrackMatch
/** Geometry parameters **/
TVector3 fTargetPos = {0., 0., 0.}; // Target centre position
@@ -217,6 +221,13 @@ private:
// Pt resolution
TH1F* fhPtResPrim = nullptr;
+ // P resolution
+ TH1F* fhPResPrim = nullptr;
+ TH1F* fhPResPrimSts0 = nullptr;
+ TH1F* fhPResPrimSts1 = nullptr;
+ TH1F* fhPResPrimSts2 = nullptr;
+ TH1F* fhPResPrimSts3 = nullptr;
+
/** List of histograms **/
TList* fHistoList = nullptr;
--
GitLab