Skip to content

Commit f51c77c

Browse files
committed
Fix error messages relevant to LorentzVectors
1 parent cfb2a73 commit f51c77c

File tree

1 file changed

+20
-12
lines changed

1 file changed

+20
-12
lines changed

PWGUD/Tasks/ProcessMCDPMJetSGv3.cxx

Lines changed: 20 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,10 @@ struct ProcessMCDPMJetSGv3 {
5252
using TCs = soa::Join<aod::UDTracks, aod::UDTracksPID, aod::UDTracksExtra, aod::UDTracksFlags, aod::UDTracksDCA, aod::UDMcTrackLabels>;
5353
// using TCs = soa::Join<aod::UDTracks, aod::UDTracksExtra, aod::UDTracksFlags, aod::UDTracksPID, aod::UDMcTrackLabels>;
5454
using TC = TCs::iterator;
55+
using LorentzVectorM = ROOT::Math::LorentzVector<ROOT::Math::PxPyPzM4D<double>>;
56+
57+
58+
5559

5660
double massPion = 0.;
5761
double massKaon = 0.;
@@ -168,28 +172,35 @@ struct ProcessMCDPMJetSGv3 {
168172
continue;
169173
counterMC += 1;
170174
// if(mcParticle.isPhysicalPrimary()) counterMC += 1;
171-
TLorentzVector protoMC;
172-
protoMC.SetXYZM(mcParticle.px(), mcParticle.py(), mcParticle.pz(), massPion);
175+
LorentzVectorM protoMC(
176+
mcParticle.px(),
177+
mcParticle.py(),
178+
mcParticle.pz(),
179+
massPion
180+
);
173181
if (std::fabs(protoMC.Eta()) < 0.8 && protoMC.Pt() > 0.1) {
174182
counter += 1;
175183
}
176184
if (!mcParticle.isPhysicalPrimary())
177185
continue;
178186
// if(mcParticle.isPhysicalPrimary() && fabs(mcParticle.eta())<0.9){ // do this in the context of the MC loop ! (context matters!!!)
179-
TLorentzVector pMC;
187+
// LorentzVectorM pMC;
188+
LorentzVectorM pMC(mcParticle.px(), mcParticle.py(), mcParticle.pz(), massPion);
180189
if (std::abs(mcParticle.pdgCode()) == codePion) {
181190
// histos.fill(HIST("ptGeneratedPion"), mcParticle.pt());
182-
pMC.SetXYZM(mcParticle.px(), mcParticle.py(), mcParticle.pz(), massPion);
191+
// LorentzVectorM pMC(mcParticle.px(), mcParticle.py(), mcParticle.pz(), massPion);
183192
histos.fill(HIST("ptGeneratedPion"), pMC.Pt());
184193
}
185194
if (std::abs(mcParticle.pdgCode()) == codeKaon) {
186195
// histos.fill(HIST("ptGenerateKaon"), mcParticle.pt());
187-
pMC.SetXYZM(mcParticle.px(), mcParticle.py(), mcParticle.pz(), massKaon);
196+
// LorentzVectorM pMC(mcParticle.px(), mcParticle.py(), mcParticle.pz(), massKaon);
197+
pMC.SetM(massKaon);
188198
histos.fill(HIST("ptGeneratedKaon"), pMC.Pt());
189199
}
190200
if (std::abs(mcParticle.pdgCode()) == codeProton) {
191201
// histos.fill(HIST("ptGeneratedProton"), mcParticle.pt());
192-
pMC.SetXYZM(mcParticle.px(), mcParticle.py(), mcParticle.pz(), massProton);
202+
// LorentzVectorM pMC(mcParticle.px(), mcParticle.py(), mcParticle.pz(), massProton);
203+
pMC.SetM(massProton);
193204
histos.fill(HIST("ptGeneratedProton"), pMC.Pt());
194205
}
195206
if (std::abs(pMC.Rapidity()) < 0.8) {
@@ -258,12 +269,9 @@ struct ProcessMCDPMJetSGv3 {
258269
double dEdx = track.tpcSignal();
259270
histos.fill(HIST("hdEdx"), momentum, dEdx);
260271

261-
TLorentzVector pion;
262-
pion.SetXYZM(track.px(), track.py(), track.pz(), o2::constants::physics::MassPionCharged);
263-
TLorentzVector kaon;
264-
kaon.SetXYZM(track.px(), track.py(), track.pz(), o2::constants::physics::MassKaonCharged);
265-
TLorentzVector proton;
266-
proton.SetXYZM(track.px(), track.py(), track.pz(), o2::constants::physics::MassProton);
272+
LorentzVectorM pion(track.px(), track.py(), track.pz(), o2::constants::physics::MassPionCharged);
273+
LorentzVectorM kaon(track.px(), track.py(), track.pz(), o2::constants::physics::MassKaonCharged);
274+
LorentzVectorM proton(track.px(), track.py(), track.pz(), o2::constants::physics::MassProton);
267275
auto nSigmaPi = -999.;
268276
auto nSigmaKa = -999.;
269277
auto nSigmaPr = -999.;

0 commit comments

Comments
 (0)