mirror of
https://github.com/dragonpilot/dragonpilot.git
synced 2026-02-23 05:33:52 +08:00
35 lines
753 B
Python
35 lines
753 B
Python
import os
|
|
|
|
from cffi import FFI
|
|
from common.ffi_wrapper import suffix
|
|
|
|
mpc_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)))
|
|
libmpc_fn = os.path.join(mpc_dir, "libmpc"+suffix())
|
|
|
|
ffi = FFI()
|
|
ffi.cdef("""
|
|
const int MPC_N = 32;
|
|
|
|
typedef struct {
|
|
double x_ego, v_ego, a_ego;
|
|
} state_t;
|
|
|
|
|
|
typedef struct {
|
|
double x_ego[MPC_N+1];
|
|
double v_ego[MPC_N+1];
|
|
double a_ego[MPC_N+1];
|
|
double t[MPC_N+1];
|
|
double j_ego[MPC_N];
|
|
double cost;
|
|
} log_t;
|
|
|
|
|
|
void init(double xCost, double vCost, double aCost, double jerkCost, double constraintCost);
|
|
int run_mpc(state_t * x0, log_t * solution,
|
|
double target_x[MPC_N+1], double target_v[MPC_N+1], double target_a[MPC_N+1],
|
|
double min_a, double max_a);
|
|
""")
|
|
|
|
libmpc = ffi.dlopen(libmpc_fn)
|