hide and plot

Author: ocots

Project.toml

@@ -9,11 +9,15 @@ CTDirect = "790bbbee-bee9-49ee-8912-a9de031322d5"
 CTFlows = "1c39547c-7794-42f7-af83-d98194f657c2"
 CommonSolve = "38540f10-b2f7-11e9-35d8-d573e4eb0ff2"
 DocStringExtensions = "ffbed154-4ef7-542d-bbb7-c09d3a79fcae"
+Plots = "91a5bcdd-55d7-5caf-9e0b-520d859cae80"
+Printf = "de0858da-6303-5e67-8744-51eddeeeb8d7"
 
 [compat]
 CTBase = "0.13"
 CTDirect = "0.12"
 CTFlows = "0.6"
 CommonSolve = "0.2"
 DocStringExtensions = "0.9"
+Plots = "1.40.9"
+Printf = "1.11.0"
 julia = "1.10"

docs/src/oc_mam.md

@@ -21,6 +21,7 @@ The system's deterministic dynamics are given by:
 # Define the vector field
 f(u, v) = [u - u^3 - 10*u*v^2,  -(1 - u^2)*v]
 f(x) = f(x...)
+nothing # hide
 ```
 
 ## Optimal Control Formulation
@@ -41,6 +42,7 @@ function ocp(T)
   end
   return action
 end
+nothing # hide
 ```
 
 ## Initial Guess
@@ -61,6 +63,7 @@ u(t) = f(x(t))
 
 # Initial guess
 init = (state=x, control=u)
+nothing # hide
 ```
 
 ## Solving the Problem
@@ -105,7 +108,7 @@ To find the maximum likelihood path, we also need to minimize the transient time
 ```@example oc_mam
 objectives = []
 Ts = range(1,100,100)
-sol = solve(ocp(Ts[1]); init=init, grid_size=50)
+sol = solve(ocp(Ts[1]); display=false, init=init, grid_size=50)
 println(" Time   Objective     Iterations")
 for T=Ts
     global sol = solve(ocp(T); display=false, init=sol, grid_size=1000)
@@ -115,8 +118,10 @@ end
 ```
 
 ```@example oc_mam
-@show Ts[argmin(objectives)]
-plt1 = scatter(Ts, log10.(objectives))
-plt2 = scatter(Ts[20:100], log10.(objectives[20:100]))
-plot(plt1,plt2)
+T_min = Ts[argmin(objectives)]
+plt1 = scatter(Ts, log10.(objectives), xlabel="Time", label="Objective (log10)")
+vline!(plt1, [T_min], label="Minimum", z_order=:back)
+plt2 = scatter(Ts[20:100], log10.(objectives[20:100]), xlabel="Time", label="Objective (log10)")
+vline!(plt2, [T_min], label="Minimum", z_order=:back)
+plot(plt1, plt2, layout=(2,1), size=(800,800))
 ```