Skip to content

Commit c093253

Browse files
authored
[Common] Further changes to QA task
1 parent 3117517 commit c093253

File tree

1 file changed

+20
-14
lines changed

1 file changed

+20
-14
lines changed

Common/Tasks/flowTest.cxx

Lines changed: 20 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -54,10 +54,13 @@ struct flowTest {
5454

5555
void init(InitContext&)
5656
{
57-
// pT histograms
57+
// QA and detailed studies
5858
histos.add<TH1>("hImpactParameter", "hImpactParameter", HistType::kTH1D, {axisB});
5959
histos.add<TH2>("hNchVsImpactParameter", "hNchVsImpactParameter", HistType::kTH2D, {axisB, axisNch});
6060
histos.add<TH1>("hEventPlaneAngle", "hEventPlaneAngle", HistType::kTH1D, {axisPhi});
61+
histos.add<TH2>("hTrackPhiVsEventPlaneAngle", "hTrackPhiVsEventPlaneAngle", HistType::kTH2D, {axisPhi, axisPhi});
62+
63+
// analysis
6164
histos.add<TH2>("hPtVsPhiGenerated", "hPtVsPhiGenerated", HistType::kTH2D, {axisPhi, axisPt});
6265
histos.add<TH2>("hPtVsPhiGlobal", "hPtVsPhiGlobal", HistType::kTH2D, {axisPhi, axisPt});
6366
histos.add<TH3>("hBVsPtVsPhiGenerated", "hBVsPtVsPhiGenerated", HistType::kTH3D, {axisB, axisPhi, axisPt});
@@ -83,9 +86,10 @@ struct flowTest {
8386
histos.add<TH3>("hBVsPtVsPhiGlobalOmega", "hBVsPtVsPhiGlobalOmega", HistType::kTH3D, {axisB, axisPhi, axisPt});
8487
}
8588

86-
using recoTracks = soa::Join<aod::TracksIU, aod::TracksExtra, aod::McTrackLabels>;
89+
using recoTracks = soa::Join<aod::TracksIU, aod::TracksExtra>;
90+
using recoTracksWithLabels = soa::Join<aod::TracksIU, aod::TracksExtra, aod::McTrackLabels>;
8791

88-
void process(aod::McCollision const& mcCollision, soa::Join<aod::McParticles, aod::ParticlesToTracks> const& mcParticles, recoTracks const&)
92+
void process(aod::McCollision const& mcCollision, soa::Join<aod::McParticles, aod::ParticlesToTracks> const& mcParticles, recoTracksWithLabels const&)
8993
{
9094

9195
float imp = mcCollision.impactParameter();
@@ -118,6 +122,8 @@ struct flowTest {
118122
deltaPhi += 2. * TMath::Pi();
119123
if (deltaPhi > 2. * TMath::Pi())
120124
deltaPhi -= 2. * TMath::Pi();
125+
126+
histos.fill(HIST("hTrackPhiVsEventPlaneAngle"), evPhi, mcParticle.phi());
121127
histos.fill(HIST("hPtVsPhiGenerated"), deltaPhi, mcParticle.pt());
122128
histos.fill(HIST("hBVsPtVsPhiGenerated"), imp, deltaPhi, mcParticle.pt());
123129

@@ -131,28 +137,28 @@ struct flowTest {
131137
bool validITSTrackFake = false;
132138
bool validITSABTrack = false;
133139
bool validITSABTrackFake = false;
134-
bool validAnalysisTrack = false;
140+
bool validAnalysisTrack = false;
135141
bool validAnalysisTrackFake = false;
136142
if (mcParticle.has_tracks()) {
137-
auto const& tracks = mcParticle.tracks_as<recoTracks>();
143+
auto const& tracks = mcParticle.tracks_as<recoTracksWithLabels>();
138144
for (auto const& track : tracks) {
139145
bool isITSFake = false;
140146

141-
for (int bit = 0; bit < 7; bit++) {
142-
if (bitcheck(track.mcMask(), bit)) {
147+
for(int bit = 0; bit < 7; bit++){
148+
if(bitcheck(track.mcMask(), bit)){
143149
isITSFake = true;
144150
}
145151
}
146152

147-
if (track.tpcNClsFound() >= analysisMinimumTPCClusters && track.itsNCls() >= analysisMinimumITSClusters) {
153+
if (track.tpcNClsFound() > analysisMinimumTPCClusters && track.itsNCls() > analysisMinimumITSClusters){
148154
validAnalysisTrack = true;
149-
if (isITSFake) {
155+
if(isITSFake){
150156
validAnalysisTrackFake = true;
151157
}
152158
}
153159
if (track.hasTPC() && track.hasITS()) {
154160
validGlobal = true;
155-
if (isITSFake) {
161+
if(isITSFake){
156162
validGlobalFake = true;
157163
}
158164
}
@@ -164,13 +170,13 @@ struct flowTest {
164170
}
165171
if (track.hasITS() && track.itsChi2NCl() > -1e-6) {
166172
validITSTrack = true;
167-
if (isITSFake) {
173+
if(isITSFake){
168174
validITSTrackFake = true;
169175
}
170176
}
171177
if (track.hasITS() && track.itsChi2NCl() < -1e-6) {
172178
validITSABTrack = true;
173-
if (isITSFake) {
179+
if(isITSFake){
174180
validITSABTrackFake = true;
175181
}
176182
}
@@ -182,13 +188,13 @@ struct flowTest {
182188
histos.fill(HIST("hPtVsPhiGlobal"), deltaPhi, mcParticle.pt());
183189
histos.fill(HIST("hBVsPtVsPhiGlobal"), imp, deltaPhi, mcParticle.pt());
184190
}
185-
if (validGlobalFake) {
191+
if (validGlobalFake){
186192
histos.fill(HIST("hBVsPtVsPhiGlobalFake"), imp, deltaPhi, mcParticle.pt());
187193
}
188194
if (validAnalysisTrack) {
189195
histos.fill(HIST("hBVsPtVsPhiAnalysis"), imp, deltaPhi, mcParticle.pt());
190196
}
191-
if (validAnalysisTrackFake) {
197+
if (validAnalysisTrackFake){
192198
histos.fill(HIST("hBVsPtVsPhiAnalysisFake"), imp, deltaPhi, mcParticle.pt());
193199
}
194200
// if any track present, fill

0 commit comments

Comments
 (0)