length call

kestrelquantum · Jul 31, 2024 · 60084d5 · 60084d5
1 parent b26231d
commit 60084d5
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/methods_named_trajectory.jl b/src/methods_named_trajectory.jl
@@ -36,6 +36,16 @@ function Base.vec(Z::NamedTrajectory)
 end
 
 
+"""
+    length(::NamedTrajectory)
+
+Returns the length of all variables of the trajectory.
+"""
+function Base.length(Z::NamedTrajectory)
+    return Z.dim * Z.T + Z.global_dim
+end
+
+
 """
     copy(::NamedTrajectory)