Skip to content

Commit 61ed3a1

Browse files
committed
fix test consumer
1 parent c4152fd commit 61ed3a1

1 file changed

Lines changed: 41 additions & 43 deletions

File tree

ALICE3/TableProducer/OTF/testCcdbConsumer1.cxx

Lines changed: 41 additions & 43 deletions
Original file line numberDiff line numberDiff line change
@@ -58,54 +58,52 @@ struct TestCcdbConsumer1 {
5858
LOG(info) << "Initialization completed";
5959
}
6060

61-
void process(o2::aod::McCollisions const& mcCollisions, o2::aod::McParticles const& mcParticles, o2::aod::A3LookUpTables const& luts)
61+
void process(o2::aod::McCollisions::iterator const&, o2::aod::McParticles const& mcParticles, o2::aod::A3LookUpTables const& luts)
6262
{
63-
for (const auto& _ : mcCollisions) {
64-
const auto thisTable = luts.begin();
65-
smearer.viewTable(PDG_t::kElectron, thisTable.lutEl());
66-
smearer.viewTable(PDG_t::kMuonMinus, thisTable.lutMu());
67-
smearer.viewTable(PDG_t::kPiPlus, thisTable.lutPi());
68-
smearer.viewTable(PDG_t::kKPlus, thisTable.lutKa());
69-
smearer.viewTable(PDG_t::kProton, thisTable.lutPr());
70-
for (const auto& mcParticle : mcParticles) {
71-
if (!pdgsToBeHandled.count(std::abs(mcParticle.pdgCode()))) {
72-
continue;
73-
}
63+
const auto thisTable = luts.begin();
64+
smearer.viewTable(PDG_t::kElectron, thisTable.lutEl());
65+
smearer.viewTable(PDG_t::kMuonMinus, thisTable.lutMu());
66+
smearer.viewTable(PDG_t::kPiPlus, thisTable.lutPi());
67+
smearer.viewTable(PDG_t::kKPlus, thisTable.lutKa());
68+
smearer.viewTable(PDG_t::kProton, thisTable.lutPr());
69+
for (const auto& mcParticle : mcParticles) {
70+
if (!pdgsToBeHandled.count(std::abs(mcParticle.pdgCode()))) {
71+
continue;
72+
}
7473

75-
o2::track::TrackParCov trackParCov = o2::upgrade::convertMCParticleToO2Track(mcParticle, pdgDB);
76-
bool success = smearer.smearTrack(trackParCov, std::abs(mcParticle.pdgCode()), mcParticles.size());
77-
if (!success) {
78-
histos.fill(HIST("hBookkeeping"), 1.0f);
79-
continue;
80-
}
81-
histos.fill(HIST("hBookkeeping"), 2.0f);
74+
o2::track::TrackParCov trackParCov = o2::upgrade::convertMCParticleToO2Track(mcParticle, pdgDB);
75+
bool success = smearer.smearTrack(trackParCov, std::abs(mcParticle.pdgCode()), mcParticles.size());
76+
if (!success) {
77+
histos.fill(HIST("hBookkeeping"), 1.0f);
78+
continue;
79+
}
80+
histos.fill(HIST("hBookkeeping"), 2.0f);
8281

8382

84-
switch (std::abs(mcParticle.pdgCode())) {
85-
case PDG_t::kElectron:
86-
histos.fill(HIST("hDeltaPtEl"), (trackParCov.getPt() - mcParticle.pt()) / trackParCov.getPt());
87-
histos.fill(HIST("h2dDeltaPtEl"), trackParCov.getPt(), (trackParCov.getPt() - mcParticle.pt()) / trackParCov.getPt());
88-
break;
89-
case PDG_t::kMuonMinus:
90-
histos.fill(HIST("hDeltaPtMu"), (trackParCov.getPt() - mcParticle.pt()) / trackParCov.getPt());
91-
histos.fill(HIST("h2dDeltaPtMu"), trackParCov.getPt(), (trackParCov.getPt() - mcParticle.pt()) / trackParCov.getPt());
92-
break;
93-
case PDG_t::kPiPlus:
94-
histos.fill(HIST("hDeltaPtPi"), (trackParCov.getPt() - mcParticle.pt()) / trackParCov.getPt());
95-
histos.fill(HIST("h2dDeltaPtPi"), trackParCov.getPt(), (trackParCov.getPt() - mcParticle.pt()) / trackParCov.getPt());
96-
break;
97-
case PDG_t::kKPlus:
98-
histos.fill(HIST("hDeltaPtKa"), (trackParCov.getPt() - mcParticle.pt()) / trackParCov.getPt());
99-
histos.fill(HIST("h2dDeltaPtKa"), trackParCov.getPt(), (trackParCov.getPt() - mcParticle.pt()) / trackParCov.getPt());
100-
break;
101-
case PDG_t::kProton:
102-
histos.fill(HIST("hDeltaPtPr"), (trackParCov.getPt() - mcParticle.pt()) / trackParCov.getPt());
103-
histos.fill(HIST("h2dDeltaPtPr"), trackParCov.getPt(), (trackParCov.getPt() - mcParticle.pt()) / trackParCov.getPt());
104-
break;
83+
switch (std::abs(mcParticle.pdgCode())) {
84+
case PDG_t::kElectron:
85+
histos.fill(HIST("hDeltaPtEl"), (trackParCov.getPt() - mcParticle.pt()) / trackParCov.getPt());
86+
histos.fill(HIST("h2dDeltaPtEl"), trackParCov.getPt(), (trackParCov.getPt() - mcParticle.pt()) / trackParCov.getPt());
87+
break;
88+
case PDG_t::kMuonMinus:
89+
histos.fill(HIST("hDeltaPtMu"), (trackParCov.getPt() - mcParticle.pt()) / trackParCov.getPt());
90+
histos.fill(HIST("h2dDeltaPtMu"), trackParCov.getPt(), (trackParCov.getPt() - mcParticle.pt()) / trackParCov.getPt());
91+
break;
92+
case PDG_t::kPiPlus:
93+
histos.fill(HIST("hDeltaPtPi"), (trackParCov.getPt() - mcParticle.pt()) / trackParCov.getPt());
94+
histos.fill(HIST("h2dDeltaPtPi"), trackParCov.getPt(), (trackParCov.getPt() - mcParticle.pt()) / trackParCov.getPt());
95+
break;
96+
case PDG_t::kKPlus:
97+
histos.fill(HIST("hDeltaPtKa"), (trackParCov.getPt() - mcParticle.pt()) / trackParCov.getPt());
98+
histos.fill(HIST("h2dDeltaPtKa"), trackParCov.getPt(), (trackParCov.getPt() - mcParticle.pt()) / trackParCov.getPt());
99+
break;
100+
case PDG_t::kProton:
101+
histos.fill(HIST("hDeltaPtPr"), (trackParCov.getPt() - mcParticle.pt()) / trackParCov.getPt());
102+
histos.fill(HIST("h2dDeltaPtPr"), trackParCov.getPt(), (trackParCov.getPt() - mcParticle.pt()) / trackParCov.getPt());
103+
break;
105104

106-
default:
107-
break;
108-
}
105+
default:
106+
break;
109107
}
110108
}
111109
}

0 commit comments

Comments
 (0)