new function: get_duration

kestrelquantum · May 10, 2024 · 78b0e89 · 78b0e89
1 parent 6bd6ac9
commit 78b0e89
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/methods_named_trajectory.jl b/src/methods_named_trajectory.jl
@@ -8,6 +8,7 @@ export update!
 export update_bound!
 export get_times
 export get_timesteps
+export get_duration
 
 using OrderedCollections
 
@@ -307,6 +308,15 @@ function get_timesteps(traj::NamedTrajectory)
     end
 end
 
+"""
+    get_duration(::NamedTrajectory)
+
+Returns the duration of a trajectory.
+"""
+function get_duration(traj::NamedTrajectory)
+    return get_times(traj)[end]
+end
+
 """
     size(traj::NamedTrajectory) = (dim = traj.dim, T = traj.T)
 """