Skip to content

Commit 91bc67a

Browse files
Analysis: extend histogram registry example (#5066)
1 parent 84edc8c commit 91bc67a

1 file changed

Lines changed: 15 additions & 6 deletions

File tree

Analysis/Tutorials/src/histogramRegistry.cxx

Lines changed: 15 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -162,8 +162,6 @@ struct DTask {
162162
spectra.add("before_cuts/hist1", "asdf", defaultParticleHist);
163163
spectra.add("before_cuts/hist2", "asdf", defaultParticleHist);
164164
spectra.add("before_cuts/hist3", "asdf", defaultParticleHist);
165-
spectra.add("before_cuts/hist4", "asdf", defaultParticleHist);
166-
spectra.add("before_cuts/hist5", "asdf", defaultParticleHist);
167165

168166
// clone whole category / group
169167
spectra.addClone("before_cuts/", "after_cuts/");
@@ -173,6 +171,16 @@ struct DTask {
173171
spectra.addClone("neutral/pions", "strange/funny/particles");
174172
}
175173

174+
template <bool mode, typename T>
175+
void fillHistos(const T& track)
176+
{
177+
static constexpr std::string_view subDir[] = {"before_cuts/", "after_cuts/"};
178+
179+
spectra.fill(HIST(subDir[mode]) + HIST("hist1"), track.pt(), track.eta(), 50., 0.);
180+
spectra.fill(HIST(subDir[mode]) + HIST("hist2"), track.pt(), track.eta(), 50., 0.);
181+
spectra.fill(HIST(subDir[mode]) + HIST("hist3"), track.pt(), track.eta(), 50., 0.);
182+
}
183+
176184
void process(aod::Tracks const& tracks)
177185
{
178186
using namespace aod::track;
@@ -189,10 +197,11 @@ struct DTask {
189197
spectra.fill(HIST("sigmas"), track.pt(), track.eta(), 50., 0.);
190198
spectra.fill(HIST("lambdas"), track.pt(), track.eta(), 50., 0.);
191199

192-
spectra.fill(HIST("before_cuts/hist2"), track.pt(), track.eta(), 50., 0.);
193-
spectra.fill(HIST("before_cuts/hist2"), track.pt(), track.eta(), 50., 0.);
194-
195-
spectra.fill(HIST("after_cuts/hist2"), track.pt(), track.eta(), 50., 0.);
200+
// fill histograms before and after cuts
201+
fillHistos<false>(track);
202+
if (std::rand() > (RAND_MAX / 2)) {
203+
fillHistos<true>(track);
204+
}
196205

197206
spectra.fill(HIST("cascades"), track.pt(), track.eta(), 50., 0.);
198207
spectra.fill(HIST("strange/funny/particles"), track.pt(), track.eta(), 50., 0.);

0 commit comments

Comments
 (0)