Merge pull request #29 from dpad/dev/STM-convenience-functions

Add some minor convenience functions for extracting STMs from Dual Trajectories.
dpad · May 5, 2021 · 4f1cf33 · 4f1cf33
2 parents 1ce6e96 + 978c90f
commit 4f1cf33
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/dynamics/sensitivities.jl b/src/dynamics/sensitivities.jl
@@ -65,6 +65,15 @@ end
 end
 
 # TODO: Better functions for extracting STMs and stability index from Dual numbers
+function extract_STMs(sol::Trajectory, t)
+    extract_STMs(sol(t))
+end
+function extract_STMs(sol::Trajectory)
+    extract_STMs(sol.u)
+end
+function extract_STMs(state::State)
+    extract_STMs([state.u0])[1]
+end
 function extract_STMs(u)
     [hcat([Array(t.partials) for t in ut]...)' for ut in u]
 end