add set_silent option

Author: vyudu

ext/MTKJuMPControlExt.jl

@@ -255,13 +255,15 @@ Solve JuMPControlProblem. Arguments:
 - prob: a JumpControlProblem
 - jump_solver: a LP solver such as HiGHS
 - ode_solver: Takes in a symbol representing the solver. Acceptable solvers may be found at https://docs.sciml.ai/DiffEqDevDocs/stable/internals/tableaus/. Note that the symbol may be different than the typical name of the solver, e.g. :Tsitouras5 rather than Tsit5.
+- silent: set the model silent (suppress model output)
 
 Returns a JuMPControlSolution, which contains both the model and the ODE solution.
 """
-function DiffEqBase.solve(prob::JuMPControlProblem, jump_solver, ode_solver::Symbol)
+function DiffEqBase.solve(prob::JuMPControlProblem, jump_solver, ode_solver::Symbol; silent = false)
     model = prob.model
     tableau_getter = Symbol(:construct, ode_solver)
     @eval tableau = $tableau_getter()
+    silent && set_silent(model)
 
     # Unregister current solver constraints
     for con in all_constraints(model)
@@ -285,6 +287,7 @@ end
 """
 function DiffEqBase.solve(prob::InfiniteOptControlProblem, jump_solver;
         derivative_method = InfiniteOpt.FiniteDifference(Backward()))
+    silent && set_silent(model)
     set_derivative_method(prob.model[:t], derivative_method)
     _solve(prob, jump_solver, derivative_method)
 end

src/ModelingToolkit.jl

@@ -347,8 +347,8 @@ export AnalysisPoint, get_sensitivity_function, get_comp_sensitivity_function,
        open_loop
 function FMIComponent end
 
-include("src/systems/optimal_control_interface.jl")
-export JuMPControlProblem, InfiniteOptControlProblem, PyomoControlProblem, CasADiControlProblem
+include("systems/optimal_control_interface.jl")
+export AbstractOptimalControlProblem, JuMPControlProblem, InfiniteOptControlProblem, PyomoControlProblem, CasADiControlProblem
 export OptimalControlSolution
 
 end # module

test/extensions/jump_control.jl

@@ -31,12 +31,12 @@ const M = ModelingToolkit
     @test jsol.sol.u ≈ osol.u
 
     # Implicit method.
-    jsol2 = @btime solve($jprob, Ipopt.Optimizer, :ImplicitEuler) # 63.031 ms, 26.49 MiB
+    jsol2 = @btime solve($jprob, Ipopt.Optimizer, :ImplicitEuler, silent = true) # 63.031 ms, 26.49 MiB
     osol2 = @btime solve($oprob, ImplicitEuler(), dt = 0.01, adaptive = false) # 129.375 μs, 61.91 KiB
     @test ≈(jsol2.sol.u, osol2.u, rtol = 0.001)
     iprob = InfiniteOptControlProblem(sys, u0map, tspan, parammap, dt = 0.01)
     isol = @btime solve(
-        $iprob, Ipopt.Optimizer, derivative_method = FiniteDifference(Backward())) # 11.540 ms, 4.00 MiB
+        $iprob, Ipopt.Optimizer, derivative_method = FiniteDifference(Backward()), silent = true) # 11.540 ms, 4.00 MiB
 
     # With a constraint
     u0map = Pair[]
@@ -46,15 +46,15 @@ const M = ModelingToolkit
 
     jprob = JuMPControlProblem(lksys, u0map, tspan, parammap; guesses = guess, dt = 0.01)
     @test num_constraints(jprob.model) == 2
-    jsol = @btime solve($jprob, Ipopt.Optimizer, :Tsitouras5) # 12.190 s, 9.68 GiB
+    jsol = @btime solve($jprob, Ipopt.Optimizer, :Tsitouras5, silent = true) # 12.190 s, 9.68 GiB
     sol = jsol.sol
     @test sol(0.6)[1] ≈ 3.5
     @test sol(0.3)[1] ≈ 7.0
 
     iprob = InfiniteOptControlProblem(
         lksys, u0map, tspan, parammap; guesses = guess, dt = 0.01)
     isol = @btime solve(
-        $iprob, Ipopt.Optimizer, derivative_method = OrthogonalCollocation(3)) # 48.564 ms, 9.58 MiB
+        $iprob, Ipopt.Optimizer, derivative_method = OrthogonalCollocation(3), silent = true) # 48.564 ms, 9.58 MiB
     sol = isol.sol
     @test sol(0.6)[1] ≈ 3.5
     @test sol(0.3)[1] ≈ 7.0