add plot()

Author: ocots

Project.toml

@@ -10,6 +10,7 @@ Interpolations = "a98d9a8b-a2ab-59e6-89dd-64a1c18fca59"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 MLStyle = "d8e11817-5142-5d16-987a-aa16d5891078"
 MacroTools = "1914dd2f-81c6-5fcd-8719-6d5c9610ff09"
+NamedTupleTools = "d9ec5142-1e00-5aa0-9d6a-321866360f50"
 OrderedCollections = "bac558e1-5e72-5ebc-8fee-abe8a469f55d"
 Parameters = "d96e819e-fc66-5662-9728-84c9c7592b0a"
 PrettyTables = "08abe8d2-0d0c-5749-adfa-8a2ac140af0d"
@@ -34,6 +35,7 @@ JSON3 = "1.14"
 LinearAlgebra = "1"
 MLStyle = "0.4"
 MacroTools = "0.5"
+NamedTupleTools = "0.14.3"
 OrderedCollections = "1.8.0"
 Parameters = "0.12"
 Plots = "1.40"

ext/plot.jl

@@ -1112,6 +1112,47 @@ function Plots.plot!(
     kwargs...,
 )
 
+    model = CTModels.model(sol)
+
+    # check if the plot is empty
+    if isempty(p.series_list)
+        
+        attr = NamedTuple((Symbol(key),value) for (key,value) in p.attr if key != :layout)
+
+        pnew = __initial_plot(
+            sol,
+            description...;
+            layout=layout,
+            control=control,
+            model=model,
+            size=__size_plot(
+                sol,
+                model,
+                control,
+                layout,
+                description...;
+                state_style=state_style,
+                control_style=control_style,
+                costate_style=costate_style,
+                path_style=path_style,
+                dual_style=dual_style,
+            ),
+            state_style=state_style,
+            control_style=control_style,
+            costate_style=costate_style,
+            path_style=path_style,
+            dual_style=dual_style,
+            attr...,
+            kwargs...,
+        )
+
+        # replace p by pnew, must have a side effect
+        for k ∈ fieldnames(typeof(p))
+           setfield!(p, k, getfield(pnew, k))
+        end
+        
+    end
+
     # plot the solution with infos from the model
     return __plot!(
         p,
@@ -1124,7 +1165,7 @@ function Plots.plot!(
         state_style=state_style,
         control_style=control_style,
         costate_style=costate_style,
-        model=CTModels.model(sol),
+        model=model,
         state_bounds_style=state_bounds_style,
         control_bounds_style=control_bounds_style,
         time_style=time_style,

test/extras/plot_manual.jl

@@ -1,3 +1,6 @@
+using Pkg
+Pkg.activate(".")
+
 using CTModels
 using Plots
 
@@ -37,6 +40,8 @@ function get_solution()
     end
     CTModels.definition!(pre_ocp, definition) # does not correspond to the solution
 
+    CTModels.time_dependence!(pre_ocp; autonomous=false)
+
     # build model
     ocp = CTModels.build_model(pre_ocp)
 
@@ -127,7 +132,15 @@ end;
 
 sol = get_solution();
 
+#
+plt = plot(size=(800,800))
+pp = plot!(plt, sol)
+
+pp
+plt
+
 # layout = :group
+
 plot(sol; layout=:group, control=:components)
 plot(sol; layout=:group, control=:norm)
 plot(sol; layout=:group, control=:all)