Skip to content

Commit 4d1b7a2

Browse files
committed
PWGEM/Dilepton: try to reduce memory of ML PID
1 parent 3d95ef4 commit 4d1b7a2

File tree

1 file changed

+26
-7
lines changed

1 file changed

+26
-7
lines changed

PWGEM/Dilepton/TableProducer/skimmerPrimaryElectron.cxx

Lines changed: 26 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -588,7 +588,7 @@ struct skimmerPrimaryElectron {
588588

589589
Preslice<aod::TrackAssoc> trackIndicesPerCollision = aod::track_association::collisionId;
590590
std::vector<std::pair<int, int>> stored_trackIds;
591-
Filter trackFilter = o2::aod::track::pt > minpt&& nabs(o2::aod::track::eta) < maxeta&& o2::aod::track::itsChi2NCl < maxchi2its&& ncheckbit(aod::track::v001::detectorMap, (uint8_t)o2::aod::track::ITS) == true;
591+
Filter trackFilter = o2::aod::track::itsChi2NCl < maxchi2its && ncheckbit(aod::track::v001::detectorMap, (uint8_t)o2::aod::track::ITS) == true;
592592
using MyFilteredTracks = soa::Filtered<MyTracks>;
593593

594594
Partition<MyFilteredTracks> posTracks = o2::aod::track::signed1Pt > 0.f;
@@ -611,9 +611,13 @@ struct skimmerPrimaryElectron {
611611
auto tracks_per_coll = tracks.sliceBy(perCol, collision.globalIndex());
612612
for (const auto& track : tracks_per_coll) {
613613
float probaEl = 1.0;
614-
if (!checkTrack<false>(collision, track) || !isElectron(collision, track, probaEl)) {
614+
if (!checkTrack<false>(collision, track)) {
615+
continue;
616+
}
617+
if (!isElectron(collision, track, probaEl)) {
615618
continue;
616619
}
620+
617621
fillTrackTable<false>(collision, track, probaEl);
618622
}
619623

@@ -641,7 +645,10 @@ struct skimmerPrimaryElectron {
641645
for (const auto& trackId : trackIdsThisCollision) {
642646
auto track = trackId.template track_as<MyTracks>();
643647
float probaEl = 1.0;
644-
if (!checkTrack<false>(collision, track) || !isElectron(collision, track, probaEl)) {
648+
if (!checkTrack<false>(collision, track)) {
649+
continue;
650+
}
651+
if (!isElectron(collision, track, probaEl)) {
645652
continue;
646653
}
647654
fillTrackTable<false>(collision, track, probaEl);
@@ -672,7 +679,10 @@ struct skimmerPrimaryElectron {
672679
auto tracks_per_coll = tracks.sliceBy(perCol, collision.globalIndex());
673680
for (const auto& track : tracks_per_coll) {
674681
float probaEl = 1.0;
675-
if (!checkTrack<false>(collision, track) || !isElectron(collision, track, probaEl)) {
682+
if (!checkTrack<false>(collision, track)) {
683+
continue;
684+
}
685+
if (!isElectron(collision, track, probaEl)) {
676686
continue;
677687
}
678688
fillTrackTable<false>(collision, track, probaEl);
@@ -705,7 +715,10 @@ struct skimmerPrimaryElectron {
705715
for (const auto& trackId : trackIdsThisCollision) {
706716
auto track = trackId.template track_as<MyTracks>();
707717
float probaEl = 1.0;
708-
if (!checkTrack<false>(collision, track) || !isElectron(collision, track, probaEl)) {
718+
if (!checkTrack<false>(collision, track)) {
719+
continue;
720+
}
721+
if (!isElectron(collision, track, probaEl)) {
709722
continue;
710723
}
711724
fillTrackTable<false>(collision, track, probaEl);
@@ -740,7 +753,10 @@ struct skimmerPrimaryElectron {
740753
auto tracks_per_coll = tracks.sliceBy(perCol, collision.globalIndex());
741754
for (const auto& track : tracks_per_coll) {
742755
float probaEl = 1.0;
743-
if (!checkTrack<true>(collision, track) || !isElectron(collision, track, probaEl)) {
756+
if (!checkTrack<true>(collision, track)) {
757+
continue;
758+
}
759+
if (!isElectron(collision, track, probaEl)) {
744760
continue;
745761
}
746762
fillTrackTable<true>(collision, track, probaEl);
@@ -772,7 +788,10 @@ struct skimmerPrimaryElectron {
772788
for (const auto& trackId : trackIdsThisCollision) {
773789
auto track = trackId.template track_as<MyTracksMC>();
774790
float probaEl = 1.0;
775-
if (!checkTrack<true>(collision, track) || !isElectron(collision, track, probaEl)) {
791+
if (!checkTrack<true>(collision, track)) {
792+
continue;
793+
}
794+
if (!isElectron(collision, track, probaEl)) {
776795
continue;
777796
}
778797
fillTrackTable<true>(collision, track, probaEl);

0 commit comments

Comments
 (0)