Update source/Utils/src/FilterTracks.cc

Co-authored-by: Thomas Madlener <[email protected]>
MuonColliderSoft · Nov 27, 2024 · 7397209 · 7397209
1 parent e44c841
commit 7397209
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/source/Utils/src/FilterTracks.cc b/source/Utils/src/FilterTracks.cc
@@ -149,12 +149,9 @@ void FilterTracks::processEvent( LCEvent * evt )
     bool foundCaloState = false;
     // Check if a TrackState at the calo surface exists
     const std::vector<EVENT::TrackState*>& trackStates = trk->getTrackStates();
-    for (const auto& state : trackStates) {
-      if (state->getLocation() == EVENT::TrackState::AtCalorimeter) {
-        foundCaloState = true;
-        break;
-      }
-    }
+    const auto foundCaloState = std::find_if(trackStates.begin(), trackStates.end(), 
+                                             [](const auto ts) { return ts->getLocation() == EVENT::TrackState::AtCalorimeter) 
+                                != trackStates.end();
     if (_HasCaloState && !foundCaloState) { continue; }
 
     if(_BarrelOnly == true) {