Rapid Optimal Control Kit
Project description
rockit
Description
Rockit (Rapid Optimal Control kit) is a software framework to quickly prototype optimal control problems (aka dynamic optimization) that may arise in engineering: e.g. iterative learning (ILC), model predictive control (NMPC), system identification, and motion planning.
Notably, the software allows free end-time problems and multi-stage optimal problems. The software is currently focused on direct methods and relies heavily on CasADi. The software is developed by the KU Leuven MECO research team.
Installation
Install using pip: pip install rockit-meco
Hello world
(Taken from the example gallery)
You may try it live in your browser: .
Import the project:
from rockit import *
Start an optimal control environment with a time horizon of 10 seconds starting from t0=0s. (free-time problems can be configured with `FreeTime(initial_guess))
ocp = Ocp(t0=0, T=10)
Define two scalar states (vectors and matrices also supported)
x1 = ocp.state()
x2 = ocp.state()
Define one piecewise constant control input
(use order=1
for piecewise linear)
u = ocp.control()
Compose time-dependent expressions a.k.a. signals
(explicit time-dependence is supported with ocp.t
)
e = 1 - x2**2
Specify differential equations for states
(DAEs also supported with ocp.algebraic
and add_alg
)
ocp.set_der(x1, e * x1 - x2 + u)
ocp.set_der(x2, x1)
Lagrange objective term: signals in an integrand
ocp.add_objective(ocp.integral(x1**2 + x2**2 + u**2))
Mayer objective term: signals evaluated at t_f = t0_+T
ocp.add_objective(ocp.at_tf(x1**2))
Path constraints
(must be valid on the whole time domain running from t0
to tf
,
grid options available such as grid='integrator'
or grid='inf'
)
ocp.subject_to(x1 >= -0.25)
ocp.subject_to(-1 <= (u <= 1 ))
Boundary constraints
ocp.subject_to(ocp.at_t0(x1) == 0)
ocp.subject_to(ocp.at_t0(x2) == 1)
Pick an NLP solver backend
(CasADi nlpsol
plugin)
ocp.solver('ipopt')
Pick a solution method
such as SingleShooting
, MultipleShooting
, DirectCollocation
with arguments:
- N -- number of control intervals
- M -- number of integration steps per control interval
- grid -- could specify e.g. UniformGrid() or GeometricGrid(4)
method = MultipleShooting(N=10, intg='rk')
ocp.method(method)
Solve:
sol = ocp.solve()
Show structure:
ocp.spy()
Post-processing:
tsa, x1a = sol.sample(x1, grid='control')
tsb, x1b = sol.sample(x1, grid='integrator')
tsc, x1c = sol.sample(x1, grid='integrator', refine=100)
plot(tsa, x1a, '-')
plot(tsb, x1b, 'o')
plot(tsc, x1c, '.')
Matlab interface
Rockit comes with a (almost) feature-complete interface to Matlab. Installation steps:
- Check which Python versions your Matlab installation supports, e.g.
Python 3.6
- Open up a compatible Python environment in a terminal (if you don't have one, consider miniconda and create an environment by performing commands
conda create --name myspace python=3.6
andconda activate myspace
inside the Anaconda Prompt). - Perform
pip install "rockit-meco>=0.1.11" "casadi>=3.5.5"
in that teminal - Launch Matlab from that same terminal (Type the full path+name of the Matlab executable. In Windows you may find the Matlab executable by right-clicking the icon from the start menu)
- Install CasADi for Matlab from https://github.com/casadi/casadi/releases/tag/3.5.5: pick the latest applicable matlab archive, unzip it, and add it to the Matlab path (without subdirectories)
- Make sure you remove any other CasADi version from the Matlab path.
- Only for Matlab >=2019b: make sure you do have in-process ExecutionMode for speed
pyenv('ExecutionMode','InProcess')
- Add rockit to the matlab path:
addpath(char(py.rockit.matlab_path))
- Run the
hello_world
example from the example directory (cd(char(py.rockit.example_dir))
)
Debugging:
- Check if the correct CasADi Python is found: py.imp.find_module('casadi')
- Check if the correct CasADi Matlab is found:
edit casadi.SerializerBase
, should have a method called 'connect' - Matlab error "Conversion to double from py.numpy.ndarray is not possible." -> Consult your Matlab release notes to verify that your Python version is supported
- Matlab error "Python Error: RuntimeError: .../casadi/core/serializing_stream.hpp:171: Assertion "false" failed:" -> May occur on Linux for some configurations. Consult rockit authors
Presentations
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.