allow print_summary to be set in solve_ocp

murrayrm · murrayrm · commit 54de2a3c33b8 · 2022-11-19T08:04:49.000-08:00
diff --git a/control/optimal.py b/control/optimal.py
@@ -897,7 +897,8 @@ def __init__(
 def solve_ocp(
         sys, horizon, X0, cost, trajectory_constraints=None, terminal_cost=None,
         terminal_constraints=[], initial_guess=None, basis=None, squeeze=None,
-        transpose=None, return_states=True, log=False, **kwargs):
+        transpose=None, return_states=True, print_summary=True, log=False,
+        **kwargs):
 
     """Compute the solution to an optimal control problem
 
@@ -951,6 +952,9 @@ def solve_ocp(
     log : bool, optional
         If `True`, turn on logging messages (using Python logging module).
 
+    print_summary : bool, optional
+        If `True` (default), print a short summary of the computation.
+
     return_states : bool, optional
         If True, return the values of the state at each time (default = True).
 
@@ -1017,7 +1021,8 @@ def solve_ocp(
 
     # Solve for the optimal input from the current state
     return ocp.compute_trajectory(
-        X0, squeeze=squeeze, transpose=transpose, return_states=return_states)
+        X0, squeeze=squeeze, transpose=transpose, print_summary=print_summary,
+        return_states=return_states)
 
 
 # Create a model predictive controller for an optimal control problem
