forked from IQ.Lvbs/IQ.Pilot
121 lines
3.5 KiB
Python
121 lines
3.5 KiB
Python
import os
|
|
from cffi import FFI
|
|
|
|
from iqdbc.safety import LEN_TO_DLC
|
|
|
|
libsafety_dir = os.path.dirname(os.path.abspath(__file__))
|
|
libsafety_fn = os.path.join(libsafety_dir, "libsafety.so")
|
|
|
|
ffi = FFI()
|
|
|
|
ffi.cdef("""
|
|
typedef struct {
|
|
unsigned char fd : 1;
|
|
unsigned char bus : 3;
|
|
unsigned char data_len_code : 4;
|
|
unsigned char rejected : 1;
|
|
unsigned char returned : 1;
|
|
unsigned char extended : 1;
|
|
unsigned int addr : 29;
|
|
unsigned char checksum;
|
|
unsigned char data[64];
|
|
} CANPacket_t;
|
|
""", packed=True)
|
|
class CANPacket:
|
|
pass
|
|
|
|
ffi.cdef("""
|
|
bool safety_rx_hook(CANPacket_t *msg);
|
|
bool safety_tx_hook(CANPacket_t *msg);
|
|
int safety_fwd_hook(int bus_num, int addr);
|
|
int set_safety_hooks(uint16_t mode, uint16_t param);
|
|
|
|
void set_controls_allowed(bool c);
|
|
bool get_controls_allowed(void);
|
|
void set_heartbeat_engaged(bool c);
|
|
bool get_longitudinal_allowed(void);
|
|
bool get_longitudinal_gas_allowed(void);
|
|
bool get_longitudinal_brake_allowed(void);
|
|
void set_alternative_experience(int mode);
|
|
int get_alternative_experience(void);
|
|
void set_relay_malfunction(bool c);
|
|
bool get_relay_malfunction(void);
|
|
bool get_gas_pressed_prev(void);
|
|
void set_gas_pressed_prev(bool);
|
|
bool get_brake_pressed_prev(void);
|
|
bool get_regen_braking_prev(void);
|
|
bool get_steering_disengage_prev(void);
|
|
bool get_acc_main_on(void);
|
|
float get_vehicle_speed_min(void);
|
|
float get_vehicle_speed_max(void);
|
|
int get_current_safety_mode(void);
|
|
int get_current_safety_param(void);
|
|
|
|
void set_torque_meas(int min, int max);
|
|
int get_torque_meas_min(void);
|
|
int get_torque_meas_max(void);
|
|
void set_torque_driver(int min, int max);
|
|
int get_torque_driver_min(void);
|
|
int get_torque_driver_max(void);
|
|
void set_desired_torque_last(int t);
|
|
void set_rt_torque_last(int t);
|
|
void set_desired_angle_last(int t);
|
|
int get_desired_angle_last();
|
|
void set_angle_meas(int min, int max);
|
|
int get_angle_meas_min(void);
|
|
int get_angle_meas_max(void);
|
|
|
|
bool get_cruise_engaged_prev(void);
|
|
void set_cruise_engaged_prev(bool engaged);
|
|
bool get_vehicle_moving(void);
|
|
void set_timer(uint32_t t);
|
|
|
|
void safety_tick_current_safety_config();
|
|
bool safety_config_valid();
|
|
|
|
void init_tests(void);
|
|
|
|
void set_honda_fwd_brake(bool c);
|
|
bool get_honda_fwd_brake(void);
|
|
void set_honda_alt_brake_msg(bool c);
|
|
void set_honda_bosch_long(bool c);
|
|
int get_honda_hw(void);
|
|
|
|
bool get_lat_active(void);
|
|
bool get_controls_allowed_lat(void);
|
|
bool get_controls_requested_lat(void);
|
|
void set_current_safety_param_iq(uint16_t param);
|
|
uint16_t get_current_safety_param_iq(void);
|
|
bool get_enable_aol(void);
|
|
bool get_disengage_lateral_on_brake(void);
|
|
bool get_pause_lateral_on_brake(void);
|
|
void set_aol_button_press(int aol_button_press);
|
|
void set_controls_allowed_lat(bool c);
|
|
void set_controls_requested_lat(bool c);
|
|
bool get_aol_acc_main(void);
|
|
void set_acc_main_on(bool c);
|
|
int get_aol_button_press(void);
|
|
void aol_set_current_disengage_reason(int reason);
|
|
int aol_get_current_disengage_reason(void);
|
|
int get_temp_debug(void);
|
|
uint32_t get_acc_main_on_mismatches(void);
|
|
void set_aol_params(bool enable_aol, bool disengage_lateral_on_brake, bool pause_lateral_on_brake);
|
|
void set_heartbeat_engaged_aol(bool c);
|
|
void aol_heartbeat_engaged_check(void);
|
|
void set_steering_disengage(bool c);
|
|
int get_gas_interceptor_prev(void);
|
|
""")
|
|
|
|
class LibSafety:
|
|
pass
|
|
libsafety: LibSafety = ffi.dlopen(libsafety_fn)
|
|
|
|
def make_CANPacket(addr: int, bus: int, dat):
|
|
ret = ffi.new('CANPacket_t *')
|
|
ret[0].extended = 1 if addr >= 0x800 else 0
|
|
ret[0].addr = addr
|
|
ret[0].data_len_code = LEN_TO_DLC[len(dat)]
|
|
ret[0].bus = bus
|
|
ret[0].data = bytes(dat)
|
|
return ret
|