add makeTrackState function to MTJ

paulgessinger · Dec 12, 2023 · ebc9779 · ebc9779
1 parent 27cc858
commit ebc9779
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/Core/include/Acts/EventData/MultiTrajectory.hpp b/Core/include/Acts/EventData/MultiTrajectory.hpp
@@ -246,6 +246,15 @@ class MultiTrajectory {
     return self().addTrackState_impl(mask, iprevious);
   }
 
+  /// Add a track state to the container and return a track state proxy to it
+  /// This effectively calls @c addTrackState and @c getTrackState
+  /// @note Only available if the track state container is not read-only
+  /// @return a track state proxy to the newly added track state
+  template <bool RO = ReadOnly, typename = std::enable_if_t<!RO>>
+  TrackStateProxy makeTrackState(IndexType istate) {
+    return getTrackState(addTrackState(istate));
+  }
+
   /// Visit all previous states starting at a given endpoint.
   ///
   /// @param iendpoint  index of the last state