Fixes AstrodynamicalSolvers doc examples

JuliaAstro · Nov 11, 2024 · d682c7f · d682c7f
1 parent aae090b
commit d682c7f
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 19 deletions.
diff --git a/docs/make.jl b/docs/make.jl
@@ -5,7 +5,7 @@ using DifferentialEquations
 using Plots
 
 makedocs(
-    format = Documenter.HTML(),
+    format = Documenter.HTML(size_threshold = nothing),
     sitename = "GeneralAstrodynamics.jl",
     authors = "Joey Carpinelli",
     pages = [

diff --git a/lib/AstrodynamicalSolvers/docs/src/index.md b/lib/AstrodynamicalSolvers/docs/src/index.md
@@ -27,17 +27,17 @@ using Plots
 μ = 0.012150584395829193
 
 planar = let
-    u, T = halo(μ, 1) # lyapunov (planar) orbit
-    u = [u.x, 0, 0, 0, u.ẏ, 0]
-    problem = ODEProblem(CR3BFunction(), u, (0, T), (μ,))
+    ic = halo(μ, 1) # lyapunov (planar) orbit
+    u = CartesianState(ic)
+    problem = ODEProblem(CR3BFunction(), u, (0, ic.Δt), (μ,))
     solution = solve(problem, Vern9(), reltol=1e-14, abstol=1e-14)
     plot(solution, idxs=(:x,:y,:z), title = "Lyapunov Orbit", label=:none, size=(1600,900), dpi=400, aspect_ratio=1)
 end
 
 extraplanar = let
-    u, T = halo(μ, 2; amplitude=0.01) # halo (non-planar) orbit
-    u = [u.x, 0, u.z, 0, u.ẏ, 0]
-    problem = ODEProblem(CR3BFunction(), u, (0, T), (μ,))
+    ic = halo(μ, 2; amplitude=0.01) # halo (non-planar) orbit
+    u = CartesianState(ic)
+    problem = ODEProblem(CR3BFunction(), u, (0, ic.Δt), (μ,))
     solution = solve(problem, Vern9(), reltol=1e-14, abstol=1e-14)
     plot(solution, idxs=(:x,:y,:z), title = "Halo Orbit", label=:none, size=(1600,900), dpi=400, aspect_ratio=1)
 end
@@ -61,14 +61,14 @@ using Plots
 μ = 0.012150584395829193
 
 unstable = let
-    u, T = halo(μ, 1; amplitude=0.005)
+    ic = halo(μ, 1; amplitude=0.005)
 
-    u = [u.x, 0, u.z, 0, u.ẏ, 0]
-    Φ = monodromy(u, μ, T)
+    u = CartesianState(ic)
+    Φ = monodromy(u, μ, ic.Δt)
 
     ics = let
-        problem = ODEProblem(CR3BFunction(stm=true), vcat(u, vec(I(6))), (0, T), (μ,))
-        solution = solve(problem, Vern9(), reltol=1e-12, abstol=1e-12, saveat=(T / 10))
+        problem = ODEProblem(CR3BFunction(stm=true), vcat(u, vec(I(6))), (0, ic.Δt), (μ,))
+        solution = solve(problem, Vern9(), reltol=1e-12, abstol=1e-12, saveat=(ic.Δt / 10))
 
         solution.u
     end
@@ -79,22 +79,22 @@ unstable = let
     ]
 
     problem = EnsembleProblem(
-        ODEProblem(CR3BFunction(), u, (0.0, 2T), (μ,)),
+        ODEProblem(CR3BFunction(), u, (0.0, 2 * ic.Δt), (μ,)),
         prob_func=(prob, i, repeat) -> remake(prob; u0=perturbations[i]),
     )
 
     solution = solve(problem, Vern9(), trajectories=length(perturbations), reltol=1e-14, abstol=1e-14)
 end
 
 stable = let
-    u, T = halo(μ, 2; amplitude=0.005)
+    ic = halo(μ, 2; amplitude=0.005)
 
-    u = [u.x, 0, u.z, 0, u.ẏ, 0]
-    Φ = monodromy(u, μ, T)
+    u = CartesianState(ic)
+    Φ = monodromy(u, μ, ic.Δt)
 
     ics = let
-        problem = ODEProblem(CR3BFunction(stm=true), vcat(u, vec(I(6))), (0, T), (μ,))
-        solution = solve(problem, Vern9(), reltol=1e-12, abstol=1e-12, saveat=(T / 10))
+        problem = ODEProblem(CR3BFunction(stm=true), vcat(u, vec(I(6))), (0, ic.Δt), (μ,))
+        solution = solve(problem, Vern9(), reltol=1e-12, abstol=1e-12, saveat=(ic.Δt / 10))
 
         solution.u
     end
@@ -105,7 +105,7 @@ stable = let
     ]
 
     problem = EnsembleProblem(
-        ODEProblem(CR3BFunction(), u, (0.0, -2.1T), (μ,)),
+        ODEProblem(CR3BFunction(), u, (0.0, -2.1 * ic.Δt), (μ,)),
         prob_func=(prob, i, repeat) -> remake(prob; u0=perturbations[i]),
     )