@@ -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