Skip to content

Commit 81e64bb

Browse files
[PWGJE] Stopped using the CPU-intensive rho_estimator in processProductionRatio (#12763)
1 parent 1bce440 commit 81e64bb

File tree

1 file changed

+25
-35
lines changed

1 file changed

+25
-35
lines changed

PWGJE/Tasks/jetShape.cxx

Lines changed: 25 additions & 35 deletions
Original file line numberDiff line numberDiff line change
@@ -260,54 +260,44 @@ struct JetShapeTask {
260260
}
261261
PROCESS_SWITCH(JetShapeTask, processJetShape, "JetShape", true);
262262

263-
void processProductionRatio(soa::Filtered<soa::Join<aod::JetCollisions, aod::BkgChargedRhos>>::iterator const& collision, soa::Join<aod::JetTracks, aod::pidTPCFullPi, aod::pidTOFFullPi, aod::pidTPCFullPr, aod::pidTOFFullPr, aod::TracksExtra, aod::TracksDCA, aod::pidTOFbeta, aod::pidTOFmass> const& tracks, soa::Join<aod::ChargedJets, aod::ChargedJetConstituents> const& jets)
263+
void processProductionRatio(soa::Filtered<aod::JetCollisions>::iterator const& collision, soa::Join<aod::JetTracks, aod::pidTPCFullPi, aod::pidTOFFullPi, aod::pidTPCFullPr, aod::pidTOFFullPr, aod::TracksExtra, aod::TracksDCA, aod::pidTOFbeta, aod::pidTOFmass> const& tracks, soa::Join<aod::ChargedJets, aod::ChargedJetConstituents> const& jets)
264264
{
265265
if (!jetderiveddatautilities::selectCollision(collision, eventSelectionBits)) {
266266
return;
267267
}
268268

269269
registry.fill(HIST("event/vertexz"), collision.posZ());
270270

271-
std::vector<typename std::decay_t<decltype(tracks)>::iterator> goodTracks;
272-
for (auto it = tracks.begin(); it != tracks.end(); ++it) {
273-
const auto& track = *it;
274-
275-
registry.fill(HIST("trackTpcNClsCrossedRows"), track.tpcNClsCrossedRows());
276-
registry.fill(HIST("trackDcaXY"), track.dcaXY());
277-
registry.fill(HIST("trackItsChi2NCl"), track.itsChi2NCl());
278-
registry.fill(HIST("trackTpcChi2NCl"), track.tpcChi2NCl());
279-
registry.fill(HIST("trackTpcNClsFound"), track.tpcNClsFound());
280-
registry.fill(HIST("trackItsNCls"), track.itsNCls());
281-
registry.fill(HIST("trackEta"), track.eta());
282-
registry.fill(HIST("trackPhi"), track.phi());
283-
284-
if (std::abs(track.eta()) > etaTrUp)
285-
continue;
286-
if (track.tpcNClsCrossedRows() < nclcrossTpcMin)
287-
continue;
288-
if (std::abs(track.dcaXY()) > dcaxyMax)
289-
continue;
290-
if (track.itsChi2NCl() > chi2ItsMax)
291-
continue;
292-
if (track.tpcChi2NCl() > chi2TpcMax)
293-
continue;
294-
if (track.tpcNClsFound() < nclTpcMin)
295-
continue;
296-
if (track.itsNCls() < nclItsMin)
297-
continue;
298-
299-
goodTracks.push_back(it);
300-
}
301-
302271
for (auto const& jet : jets) {
303272
if (!isAcceptedJet<aod::JetTracks>(jet)) {
304273
continue;
305274
}
306275

307276
// tracks conditions
308-
for (const auto& track_it : goodTracks) {
309-
310-
const auto& track = *track_it;
277+
for (const auto& track : tracks) {
278+
registry.fill(HIST("trackTpcNClsCrossedRows"), track.tpcNClsCrossedRows());
279+
registry.fill(HIST("trackDcaXY"), track.dcaXY());
280+
registry.fill(HIST("trackItsChi2NCl"), track.itsChi2NCl());
281+
registry.fill(HIST("trackTpcChi2NCl"), track.tpcChi2NCl());
282+
registry.fill(HIST("trackTpcNClsFound"), track.tpcNClsFound());
283+
registry.fill(HIST("trackItsNCls"), track.itsNCls());
284+
registry.fill(HIST("trackEta"), track.eta());
285+
registry.fill(HIST("trackPhi"), track.phi());
286+
287+
if (std::abs(track.eta()) > etaTrUp)
288+
continue;
289+
if (track.tpcNClsCrossedRows() < nclcrossTpcMin)
290+
continue;
291+
if (std::abs(track.dcaXY()) > dcaxyMax)
292+
continue;
293+
if (track.itsChi2NCl() > chi2ItsMax)
294+
continue;
295+
if (track.tpcChi2NCl() > chi2TpcMax)
296+
continue;
297+
if (track.tpcNClsFound() < nclTpcMin)
298+
continue;
299+
if (track.itsNCls() < nclItsMin)
300+
continue;
311301

312302
// PID check
313303
registry.fill(HIST("tofMass"), track.mass());

0 commit comments

Comments
 (0)