Merge pull request #64 from control-toolbox/extension-plot

up ext plot

Author: ocots

Project.toml

@@ -1,7 +1,7 @@
 name = "CTModels"
 uuid = "34c4fa32-2049-4079-8329-de33c2a22e2d"
 authors = ["Olivier Cots <olivier.cots@toulouse-inp.fr>"]
-version = "0.2.1"
+version = "0.2.2"
 
 [deps]
 CTBase = "54762871-cc72-4466-b8e8-f6c8b58076cd"
@@ -12,6 +12,7 @@ MLStyle = "d8e11817-5142-5d16-987a-aa16d5891078"
 MacroTools = "1914dd2f-81c6-5fcd-8719-6d5c9610ff09"
 Parameters = "d96e819e-fc66-5662-9728-84c9c7592b0a"
 PrettyTables = "08abe8d2-0d0c-5749-adfa-8a2ac140af0d"
+RecipesBase = "3cdcf5f2-1ef4-517c-9805-6587b60abb01"
 
 [weakdeps]
 JLD2 = "033835bb-8acc-5ee8-8aae-3f567f8a3819"
@@ -34,4 +35,5 @@ MacroTools = "0.5"
 Parameters = "0.12"
 Plots = "1.40"
 PrettyTables = "2.4"
+RecipesBase = "1.3"
 julia = "1.10"

src/CTModels.jl

@@ -9,6 +9,7 @@ using MLStyle
 using Parameters # @with_kw: to have default values in struct
 using MacroTools: striplines
 using PrettyTables # To print a table
+import RecipesBase: plot, plot!, RecipesBase
 
 # aliases
 const Dimension = Int
@@ -29,6 +30,16 @@ import_ocp_solution(args...; kwargs...) = throw(CTBase.ExtensionError(:JLD2, :JS
 include("default.jl")
 include("utils.jl")
 include("types.jl")
+
+# to be extended
+function RecipesBase.plot(sol::AbstractSolution; kwargs...)
+    throw(CTBase.ExtensionError(:Plots))
+end
+function RecipesBase.plot!(p::RecipesBase.AbstractPlot, sol::AbstractSolution; kwargs...)
+    throw(CTBase.ExtensionError(:Plots))
+end
+
+#
 include("init.jl")
 include("dual_model.jl")
 include("state.jl")
@@ -42,4 +53,7 @@ include("print.jl")
 include("model.jl")
 include("solution.jl")
 
+#
+export plot, plot!
+
 end

test/extras/plot_manual.jl

@@ -173,4 +173,4 @@ plot!(plt, sol; time=:default, state_style=style, costate_style=style, control_s
 plt = plot(sol; layout=:group, control=:components)
 plot!(plt, sol; layout=:group, control=:components)
 plot!(plt, sol; layout=:group, control=:norm)
-plot!(plt, sol; layout=:group, control=:all)
+#plot!(plt, sol; layout=:group, control=:all)