Skip to content

Commit 8b9df2f

Browse files
[PWGJE] Change the x-axis of dE/dx to momentum (p) (#12136)
1 parent cde96bc commit 8b9df2f

File tree

1 file changed

+35
-19
lines changed

1 file changed

+35
-19
lines changed

PWGJE/Tasks/jetShape.cxx

Lines changed: 35 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -40,22 +40,30 @@ using namespace o2::framework::expressions;
4040

4141
struct JetShapeTask {
4242

43-
Configurable<int> nBinsTpcNSigma{"nBinsTpcNSigma", 101, "Number of TPC-nsigma bins"};
44-
Configurable<float> tpcNSigmaMin{"tpcNSigmaMin", -10.1f, "Min value of Tpc-nsigma"};
45-
Configurable<float> tpcNSigmaMax{"tpcNSigmaMax", 10.1f, "Max value of Tpc-nsigma"};
46-
Configurable<int> nBinsTofNSigma{"nBinsTofNSigma", 10, "Number of TOF-nsigma bins"};
47-
Configurable<float> tofNSigmaMin{"tofNSigmaMin", -10, "Min value of Tof-nsigma"};
48-
Configurable<float> tofNSigmaMax{"tofNSigmaMax", 10, "Max value of Tof-nsigma"};
43+
Configurable<int> nBinsNSigma{"nBinsNSigma", 101, "Number of nsigma bins"};
44+
Configurable<float> nSigmaMin{"nSigmaMin", -10.1f, "Min value of nsigma"};
45+
Configurable<float> nSigmaMax{"nSigmaMax", 10.1f, "Max value of nsigma"};
46+
Configurable<int> nBinsP{"nBinsP", 700, "Number of p bins"};
47+
Configurable<int> nBinsPt{"nBinsPt", 500, "Number of pT bins"};
48+
Configurable<int> nBinsTpcDedx{"nBinsTpcDedx", 500, "Number of DEdx bins"};
49+
Configurable<int> nBinsTofBeta{"nBinsTofBeta", 350, "Number of Beta bins"};
50+
Configurable<float> pMax{"pMax", 7.0f, "Max value of p"};
51+
Configurable<float> ptMax{"ptMax", 5.0f, "Max value of pT"};
52+
Configurable<int> nBinsDistance{"nBinsDistance", 7, "Number of distance bins"};
53+
Configurable<float> distanceMax{"distanceMax", 0.7f, "Max value of distance"};
54+
Configurable<float> nSigmaTofCut{"nSigmaTofCut", 2.0f, "Number of sigma cut for TOF PID"};
4955

5056
HistogramRegistry registry{"registry",
51-
{{"tpcTofPi", "tpcTofPi", {HistType::kTHnSparseD, {{nBinsTpcNSigma, tpcNSigmaMin, tpcNSigmaMax}, {nBinsTofNSigma, tofNSigmaMin, tofNSigmaMax}, {25, 0, 5}, {14, 0, 0.7}}}},
52-
{"tpcPi", "tpcPi", {HistType::kTH2F, {{100, 0, 5}, {101, -10.1f, 10.1f}}}},
53-
{"tofPi", "tofPi", {HistType::kTH2F, {{100, 0, 5}, {101, -10.1f, 10.1f}}}},
54-
{"tpcTofPr", "tpcTofPr", {HistType::kTHnSparseD, {{nBinsTpcNSigma, tpcNSigmaMin, tpcNSigmaMax}, {nBinsTofNSigma, tofNSigmaMin, tofNSigmaMax}, {25, 0, 5}, {14, 0, 0.7}}}},
55-
{"tpcPr", "tpcPr", {HistType::kTH2F, {{50, 0, 5}, {101, -10.1f, 10.1f}}}},
56-
{"tofPr", "tofPr", {HistType::kTH2F, {{50, 0, 5}, {101, -10.1f, 10.1f}}}},
57-
{"tpcDedx", "tpcDedx", {HistType::kTHnSparseD, {{100, 0, 5}, {500, 0, 1000}, {7, 0, 0.7}}}},
58-
{"tofBeta", "tofBeta", {HistType::kTHnSparseD, {{100, 0, 5}, {350, 0.4, 1.1}, {7, 0, 0.7}}}},
57+
{{"tpcTofPi", "tpcTofPi", {HistType::kTHnSparseD, {{35, 0, pMax}, {nBinsNSigma, nSigmaMin, nSigmaMax}, {nBinsDistance, 0, distanceMax}}}},
58+
{"tpcTofPr", "tpcTofPr", {HistType::kTHnSparseD, {{35, 0, pMax}, {nBinsNSigma, nSigmaMin, nSigmaMax}, {nBinsDistance, 0, distanceMax}}}},
59+
{"tpcPi", "tpcPi", {HistType::kTH2F, {{70, 0, pMax}, {nBinsNSigma, nSigmaMin, nSigmaMax}}}},
60+
{"tofPi", "tofPi", {HistType::kTH2F, {{50, 0, ptMax}, {nBinsNSigma, nSigmaMin, nSigmaMax}}}},
61+
{"tpcPr", "tpcPr", {HistType::kTH2F, {{70, 0, pMax}, {nBinsNSigma, nSigmaMin, nSigmaMax}}}},
62+
{"tofPr", "tofPr", {HistType::kTH2F, {{50, 0, ptMax}, {nBinsNSigma, nSigmaMin, nSigmaMax}}}},
63+
{"tpcDedx", "tpcDedx", {HistType::kTHnSparseD, {{nBinsP, 0, pMax}, {nBinsTpcDedx, 0, 1000}, {nBinsDistance, 0, distanceMax}}}},
64+
{"tofBeta", "tofBeta", {HistType::kTHnSparseD, {{nBinsPt, 0, ptMax}, {nBinsTofBeta, 0.4, 1.1}, {nBinsDistance, 0, distanceMax}}}},
65+
{"pVsPtForProton", "pVsPtForProton", {HistType::kTHnSparseD, {{70, 0, pMax}, {50, 0, ptMax}, {nBinsDistance, 0, distanceMax}}}},
66+
{"pVsPtForPion", "pVsPtPion", {HistType::kTHnSparseD, {{70, 0, pMax}, {50, 0, ptMax}, {nBinsDistance, 0, distanceMax}}}},
5967
{"tofMass", "tofMass", {HistType::kTH1F, {{300, 0, 3}}}},
6068
{"jetPt", "jet pT;#it{p}_{T,jet} (GeV/#it{c});entries", {HistType::kTH1F, {{200, 0., 200.}}}},
6169
{"jetEta", "jet #eta;#eta_{jet};entries", {HistType::kTH1F, {{100, -1.0, 1.0}}}},
@@ -268,9 +276,9 @@ struct JetShapeTask {
268276
registry.fill(HIST("tofMass"), track.mass());
269277

270278
// for calculate purity
271-
registry.fill(HIST("tpcPi"), track.pt(), track.tpcNSigmaPi());
279+
registry.fill(HIST("tpcPi"), track.p(), track.tpcNSigmaPi());
272280
registry.fill(HIST("tofPi"), track.pt(), track.tofNSigmaPi());
273-
registry.fill(HIST("tpcPr"), track.pt(), track.tpcNSigmaPr());
281+
registry.fill(HIST("tpcPr"), track.p(), track.tpcNSigmaPr());
274282
registry.fill(HIST("tofPr"), track.pt(), track.tofNSigmaPr());
275283

276284
// for calculate distance
@@ -282,10 +290,18 @@ struct JetShapeTask {
282290
float distance = std::sqrt(deltaEta * deltaEta + deltaPhi1 * deltaPhi1);
283291

284292
registry.fill(HIST("distanceVsTrackpt"), distance, track.pt());
285-
registry.fill(HIST("tpcDedx"), track.pt(), track.tpcSignal(), distance);
293+
registry.fill(HIST("tpcDedx"), track.p(), track.tpcSignal(), distance);
286294
registry.fill(HIST("tofBeta"), track.pt(), track.beta(), distance);
287-
registry.fill(HIST("tpcTofPi"), track.tpcNSigmaPi(), track.tofNSigmaPi(), track.pt(), distance);
288-
registry.fill(HIST("tpcTofPr"), track.tpcNSigmaPr(), track.tofNSigmaPr(), track.pt(), distance);
295+
296+
if (std::abs(track.tofNSigmaPr()) < nSigmaTofCut) {
297+
registry.fill(HIST("pVsPtForProton"), track.p(), track.pt(), distance);
298+
registry.fill(HIST("tpcTofPr"), track.p(), track.tpcNSigmaPr(), distance);
299+
}
300+
301+
if (std::abs(track.tofNSigmaPi()) < nSigmaTofCut) {
302+
registry.fill(HIST("pVsPtForPion"), track.p(), track.pt(), distance);
303+
registry.fill(HIST("tpcTofPi"), track.p(), track.tpcNSigmaPi(), distance);
304+
}
289305
}
290306
}
291307
}

0 commit comments

Comments
 (0)