bldc/lispBM/lispif_vesc_extensions.c

4474 lines
123 KiB
C
Raw Normal View History

2022-01-14 11:54:24 -08:00
/*
Copyright 2022 Benjamin Vedder benjamin@vedder.se
Copyright 2022 Joel Svensson svenssonjoel@yahoo.se
This file is part of the VESC firmware.
The VESC firmware is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
The VESC firmware is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include "lispif.h"
#include "lispbm.h"
2022-02-21 17:37:23 -08:00
#include "extensions/array_extensions.h"
2022-01-14 11:54:24 -08:00
#include "commands.h"
#include "mc_interface.h"
#include "timeout.h"
#include "servo_dec.h"
#include "servo_simple.h"
#include "encoder/encoder.h"
2022-01-14 16:25:11 -08:00
#include "comm_can.h"
#include "bms.h"
2022-03-16 10:40:51 -07:00
#include "utils_math.h"
#include "utils_sys.h"
2022-01-15 15:14:55 -08:00
#include "hw.h"
2022-02-06 14:12:05 -08:00
#include "mcpwm_foc.h"
#include "imu.h"
2022-02-21 17:37:23 -08:00
#include "mempools.h"
#include "app.h"
2022-02-23 16:19:09 -08:00
#include "spi_bb.h"
#include "i2c.h"
2022-03-31 10:37:36 -07:00
#include "confgenerator.h"
#include "worker.h"
#include "app.h"
#include "canard_driver.h"
#include "firmware_metadata.h"
#include "log.h"
#include "buffer.h"
2022-11-04 11:46:36 -07:00
#include "rb.h"
2022-01-14 11:54:24 -08:00
#include <math.h>
2022-03-07 01:26:51 -08:00
#include <ctype.h>
#include <stdarg.h>
2022-01-14 11:54:24 -08:00
2022-03-31 06:18:59 -07:00
typedef struct {
// BMS
lbm_uint v_tot;
lbm_uint v_charge;
lbm_uint i_in;
lbm_uint i_in_ic;
lbm_uint ah_cnt;
lbm_uint wh_cnt;
lbm_uint cell_num;
lbm_uint v_cell;
lbm_uint bal_state;
lbm_uint temp_adc_num;
lbm_uint temps_adc;
lbm_uint temp_ic;
lbm_uint temp_hum;
lbm_uint hum;
lbm_uint temp_max_cell;
lbm_uint soc;
lbm_uint soh;
lbm_uint can_id;
lbm_uint ah_cnt_chg_total;
lbm_uint wh_cnt_chg_total;
lbm_uint ah_cnt_dis_total;
lbm_uint wh_cnt_dis_total;
lbm_uint msg_age;
// GPIO
lbm_uint pin_mode_out;
lbm_uint pin_mode_od;
2022-05-13 01:14:37 -07:00
lbm_uint pin_mode_od_pu;
lbm_uint pin_mode_od_pd;
2022-03-31 06:18:59 -07:00
lbm_uint pin_mode_in;
lbm_uint pin_mode_in_pu;
lbm_uint pin_mode_in_pd;
2022-05-13 01:14:37 -07:00
lbm_uint pin_mode_analog;
2022-03-31 06:18:59 -07:00
lbm_uint pin_rx;
lbm_uint pin_tx;
lbm_uint pin_swdio;
lbm_uint pin_swclk;
lbm_uint pin_hall1;
lbm_uint pin_hall2;
lbm_uint pin_hall3;
lbm_uint pin_adc1;
lbm_uint pin_adc2;
2022-09-27 14:07:42 -07:00
lbm_uint pin_ppm;
2022-03-31 10:37:36 -07:00
// Settings
lbm_uint l_current_min;
lbm_uint l_current_max;
lbm_uint l_current_min_scale;
lbm_uint l_current_max_scale;
lbm_uint l_in_current_min;
lbm_uint l_in_current_max;
2022-05-06 05:37:08 -07:00
lbm_uint l_abs_current_max;
2022-03-31 10:37:36 -07:00
lbm_uint l_min_erpm;
lbm_uint l_max_erpm;
lbm_uint l_min_vin;
lbm_uint l_max_vin;
lbm_uint l_min_duty;
lbm_uint l_max_duty;
2022-03-31 10:37:36 -07:00
lbm_uint l_watt_min;
lbm_uint l_watt_max;
2022-07-31 04:30:10 -07:00
lbm_uint motor_type;
2022-05-06 05:37:08 -07:00
lbm_uint foc_sensor_mode;
2022-03-31 10:37:36 -07:00
lbm_uint foc_current_kp;
lbm_uint foc_current_ki;
lbm_uint foc_motor_l;
lbm_uint foc_motor_ld_lq_diff;
lbm_uint foc_motor_r;
lbm_uint foc_motor_flux_linkage;
lbm_uint foc_observer_gain;
2022-05-06 05:37:08 -07:00
lbm_uint foc_hfi_voltage_start;
lbm_uint foc_hfi_voltage_run;
lbm_uint foc_hfi_voltage_max;
lbm_uint foc_sl_erpm_hfi;
lbm_uint m_invert_direction;
lbm_uint m_out_aux_mode;
2022-03-31 10:37:36 -07:00
lbm_uint min_speed;
lbm_uint max_speed;
lbm_uint controller_id;
lbm_uint app_to_use;
2022-05-06 05:37:08 -07:00
lbm_uint ppm_ctrl_type;
lbm_uint ppm_pulse_start;
lbm_uint ppm_pulse_end;
lbm_uint ppm_pulse_center;
lbm_uint ppm_ramp_time_pos;
lbm_uint ppm_ramp_time_neg;
2022-04-10 12:21:35 -07:00
// Sysinfo
lbm_uint hw_name;
lbm_uint fw_ver;
lbm_uint has_phase_filters;
lbm_uint uuid;
lbm_uint runtime;
lbm_uint git_branch;
lbm_uint git_hash;
lbm_uint compiler;
2022-04-19 08:21:19 -07:00
// Rates
lbm_uint rate_100k;
lbm_uint rate_200k;
lbm_uint rate_400k;
lbm_uint rate_700k;
2022-04-28 04:42:14 -07:00
// Other
lbm_uint half_duplex;
2022-03-31 06:18:59 -07:00
} vesc_syms;
static vesc_syms syms_vesc = {0};
static void(*ext_callback)(void) = 0;
2022-11-09 07:00:43 -08:00
static char print_val_buffer[256];
2022-03-31 06:18:59 -07:00
2022-03-31 10:37:36 -07:00
static bool get_add_symbol(char *name, lbm_uint* id) {
if (!lbm_get_symbol_by_name(name, id)) {
if (!lbm_add_symbol_const(name, id)) {
return false;
}
}
return true;
}
static bool compare_symbol(lbm_uint sym, lbm_uint *comp) {
if (*comp == 0) {
if (comp == &syms_vesc.v_tot) {
get_add_symbol("bms-v-tot", comp);
} else if (comp == &syms_vesc.v_charge) {
get_add_symbol("bms-v-charge", comp);
} else if (comp == &syms_vesc.i_in) {
get_add_symbol("bms-i-in", comp);
} else if (comp == &syms_vesc.i_in_ic) {
get_add_symbol("bms-i-in-ic", comp);
} else if (comp == &syms_vesc.ah_cnt) {
get_add_symbol("bms-ah-cnt", comp);
} else if (comp == &syms_vesc.wh_cnt) {
get_add_symbol("bms-wh-cnt", comp);
} else if (comp == &syms_vesc.cell_num) {
get_add_symbol("bms-cell-num", comp);
} else if (comp == &syms_vesc.v_cell) {
get_add_symbol("bms-v-cell", comp);
} else if (comp == &syms_vesc.bal_state) {
get_add_symbol("bms-bal-state", comp);
} else if (comp == &syms_vesc.temp_adc_num) {
get_add_symbol("bms-temp-adc-num", comp);
} else if (comp == &syms_vesc.temps_adc) {
get_add_symbol("bms-temps-adc", comp);
} else if (comp == &syms_vesc.temp_ic) {
get_add_symbol("bms-temp-ic", comp);
} else if (comp == &syms_vesc.temp_hum) {
get_add_symbol("bms-temp-hum", comp);
} else if (comp == &syms_vesc.hum) {
get_add_symbol("bms-hum", comp);
} else if (comp == &syms_vesc.temp_max_cell) {
get_add_symbol("bms-temp-cell-max", comp);
} else if (comp == &syms_vesc.soc) {
get_add_symbol("bms-soc", comp);
} else if (comp == &syms_vesc.soh) {
get_add_symbol("bms-soh", comp);
} else if (comp == &syms_vesc.can_id) {
get_add_symbol("bms-can-id", comp);
} else if (comp == &syms_vesc.ah_cnt_chg_total) {
get_add_symbol("bms-ah-cnt-chg-total", comp);
} else if (comp == &syms_vesc.wh_cnt_chg_total) {
get_add_symbol("bms-wh-cnt-chg-total", comp);
} else if (comp == &syms_vesc.ah_cnt_dis_total) {
get_add_symbol("bms-ah-cnt-dis-total", comp);
} else if (comp == &syms_vesc.wh_cnt_dis_total) {
get_add_symbol("bms-wh-cnt-dis-total", comp);
} else if (comp == &syms_vesc.msg_age) {
get_add_symbol("bms-msg-age", comp);
}
else if (comp == &syms_vesc.pin_mode_out) {
get_add_symbol("pin-mode-out", comp);
} else if (comp == &syms_vesc.pin_mode_od) {
get_add_symbol("pin-mode-od", comp);
2022-05-13 01:14:37 -07:00
} else if (comp == &syms_vesc.pin_mode_od_pu) {
get_add_symbol("pin-mode-od-pu", comp);
} else if (comp == &syms_vesc.pin_mode_od_pd) {
get_add_symbol("pin-mode-od-pd", comp);
2022-03-31 10:37:36 -07:00
} else if (comp == &syms_vesc.pin_mode_in) {
get_add_symbol("pin-mode-in", comp);
} else if (comp == &syms_vesc.pin_mode_in_pu) {
get_add_symbol("pin-mode-in-pu", comp);
} else if (comp == &syms_vesc.pin_mode_in_pd) {
get_add_symbol("pin-mode-in-pd", comp);
2022-05-13 01:14:37 -07:00
} else if (comp == &syms_vesc.pin_mode_analog) {
get_add_symbol("pin-mode-analog", comp);
2022-03-31 10:37:36 -07:00
} else if (comp == &syms_vesc.pin_rx) {
get_add_symbol("pin-rx", comp);
} else if (comp == &syms_vesc.pin_tx) {
get_add_symbol("pin-tx", comp);
} else if (comp == &syms_vesc.pin_swdio) {
get_add_symbol("pin-swdio", comp);
} else if (comp == &syms_vesc.pin_swclk) {
get_add_symbol("pin-swclk", comp);
} else if (comp == &syms_vesc.pin_hall1) {
get_add_symbol("pin-hall1", comp);
} else if (comp == &syms_vesc.pin_hall2) {
get_add_symbol("pin-hall2", comp);
} else if (comp == &syms_vesc.pin_hall3) {
get_add_symbol("pin-hall3", comp);
} else if (comp == &syms_vesc.pin_adc1) {
get_add_symbol("pin-adc1", comp);
} else if (comp == &syms_vesc.pin_adc2) {
get_add_symbol("pin-adc2", comp);
2022-09-27 14:07:42 -07:00
} else if (comp == &syms_vesc.pin_ppm) {
get_add_symbol("pin-ppm", comp);
2022-03-31 10:37:36 -07:00
}
else if (comp == &syms_vesc.l_current_min) {
get_add_symbol("l-current-min", comp);
} else if (comp == &syms_vesc.l_current_max) {
get_add_symbol("l-current-max", comp);
} else if (comp == &syms_vesc.l_current_min_scale) {
get_add_symbol("l-current-min-scale", comp);
} else if (comp == &syms_vesc.l_current_max_scale) {
get_add_symbol("l-current-max-scale", comp);
} else if (comp == &syms_vesc.l_in_current_min) {
get_add_symbol("l-in-current-min", comp);
} else if (comp == &syms_vesc.l_in_current_max) {
get_add_symbol("l-in-current-max", comp);
2022-05-06 05:37:08 -07:00
} else if (comp == &syms_vesc.l_abs_current_max) {
get_add_symbol("l-abs-current-max", comp);
2022-03-31 10:37:36 -07:00
} else if (comp == &syms_vesc.l_min_erpm) {
get_add_symbol("l-min-erpm", comp);
} else if (comp == &syms_vesc.l_max_erpm) {
get_add_symbol("l-max-erpm", comp);
} else if (comp == &syms_vesc.l_min_vin) {
get_add_symbol("l-min-vin", comp);
} else if (comp == &syms_vesc.l_max_vin) {
get_add_symbol("l-max-vin", comp);
} else if (comp == &syms_vesc.l_min_duty) {
get_add_symbol("l-min-duty", comp);
} else if (comp == &syms_vesc.l_max_duty) {
get_add_symbol("l-max-duty", comp);
2022-03-31 10:37:36 -07:00
} else if (comp == &syms_vesc.l_watt_min) {
get_add_symbol("l-watt-min", comp);
} else if (comp == &syms_vesc.l_watt_max) {
get_add_symbol("l-watt-max", comp);
2022-07-31 04:30:10 -07:00
} else if (comp == &syms_vesc.motor_type) {
get_add_symbol("motor-type", comp);
2022-05-06 05:37:08 -07:00
} else if (comp == &syms_vesc.foc_sensor_mode) {
get_add_symbol("foc-sensor-mode", comp);
2022-03-31 10:37:36 -07:00
} else if (comp == &syms_vesc.foc_current_kp) {
get_add_symbol("foc-current-kp", comp);
} else if (comp == &syms_vesc.foc_current_ki) {
get_add_symbol("foc-current-ki", comp);
} else if (comp == &syms_vesc.foc_motor_l) {
get_add_symbol("foc-motor-l", comp);
} else if (comp == &syms_vesc.foc_motor_ld_lq_diff) {
get_add_symbol("foc-motor-ld-lq-diff", comp);
} else if (comp == &syms_vesc.foc_motor_r) {
get_add_symbol("foc-motor-r", comp);
} else if (comp == &syms_vesc.foc_motor_flux_linkage) {
get_add_symbol("foc-motor-flux-linkage", comp);
} else if (comp == &syms_vesc.foc_observer_gain) {
get_add_symbol("foc-observer-gain", comp);
2022-05-06 05:37:08 -07:00
} else if (comp == &syms_vesc.foc_hfi_voltage_start) {
get_add_symbol("foc-hfi-voltage-start", comp);
} else if (comp == &syms_vesc.foc_hfi_voltage_run) {
get_add_symbol("foc-hfi-voltage-run", comp);
} else if (comp == &syms_vesc.foc_hfi_voltage_max) {
get_add_symbol("foc-hfi-voltage-max", comp);
} else if (comp == &syms_vesc.foc_sl_erpm_hfi) {
get_add_symbol("foc-sl-erpm-hfi", comp);
} else if (comp == &syms_vesc.m_invert_direction) {
get_add_symbol("m-invert-direction", comp);
} else if (comp == &syms_vesc.m_out_aux_mode) {
get_add_symbol("m-out-aux-mode", comp);
2022-03-31 10:37:36 -07:00
} else if (comp == &syms_vesc.min_speed) {
get_add_symbol("min-speed", comp);
} else if (comp == &syms_vesc.max_speed) {
get_add_symbol("max-speed", comp);
} else if (comp == &syms_vesc.controller_id) {
get_add_symbol("controller-id", comp);
} else if (comp == &syms_vesc.app_to_use) {
get_add_symbol("app-to-use", comp);
2022-05-06 05:37:08 -07:00
} else if (comp == &syms_vesc.ppm_ctrl_type) {
get_add_symbol("ppm-ctrl-type", comp);
} else if (comp == &syms_vesc.ppm_pulse_start) {
get_add_symbol("ppm-pulse-start", comp);
} else if (comp == &syms_vesc.ppm_pulse_end) {
get_add_symbol("ppm-pulse-end", comp);
} else if (comp == &syms_vesc.ppm_pulse_center) {
get_add_symbol("ppm-pulse-center", comp);
} else if (comp == &syms_vesc.ppm_ramp_time_pos) {
get_add_symbol("ppm-ramp-time-pos", comp);
} else if (comp == &syms_vesc.ppm_ramp_time_neg) {
get_add_symbol("ppm-ramp-time-neg", comp);
2022-03-31 10:37:36 -07:00
}
2022-04-10 12:21:35 -07:00
else if (comp == &syms_vesc.hw_name) {
get_add_symbol("hw-name", comp);
} else if (comp == &syms_vesc.fw_ver) {
get_add_symbol("fw-ver", comp);
} else if (comp == &syms_vesc.has_phase_filters) {
get_add_symbol("has-phase-filters", comp);
} else if (comp == &syms_vesc.uuid) {
get_add_symbol("uuid", comp);
} else if (comp == &syms_vesc.runtime) {
get_add_symbol("runtime", comp);
} else if (comp == &syms_vesc.git_branch) {
get_add_symbol("git-branch", comp);
} else if (comp == &syms_vesc.git_hash) {
get_add_symbol("git-hash", comp);
} else if (comp == &syms_vesc.compiler) {
get_add_symbol("compiler", comp);
2022-04-10 12:21:35 -07:00
}
2022-04-19 08:21:19 -07:00
else if (comp == &syms_vesc.rate_100k) {
get_add_symbol("rate-100k", comp);
} else if (comp == &syms_vesc.rate_200k) {
get_add_symbol("rate-200k", comp);
} else if (comp == &syms_vesc.rate_400k) {
get_add_symbol("rate-400k", comp);
} else if (comp == &syms_vesc.rate_700k) {
get_add_symbol("rate-700k", comp);
}
2022-04-28 04:42:14 -07:00
else if (comp == &syms_vesc.half_duplex) {
get_add_symbol("half-duplex", comp);
}
2022-03-31 10:37:36 -07:00
}
return *comp == sym;
}
2022-01-14 11:54:24 -08:00
// Helpers
2022-01-21 01:52:56 -08:00
static bool is_number_all(lbm_value *args, lbm_uint argn) {
for (lbm_uint i = 0;i < argn;i++) {
if (!lbm_is_number(args[i])) {
2022-01-16 07:21:18 -08:00
return false;
}
2022-01-14 16:25:11 -08:00
}
return true;
}
static bool is_symbol_true_false(lbm_value v) {
bool res = lbm_is_symbol_true(v) || lbm_is_symbol_nil(v);
lbm_set_error_reason("Argument must be t or nil (true or false)");
return res;
}
static char *error_reason_no_number = "Argument(s) must be a number.";
static char *error_reason_argn = "Invalid number of arguments.";
static bool check_number_all(lbm_value *args, lbm_uint argn) {
if (!is_number_all(args, argn)) {
lbm_set_error_reason(error_reason_no_number);
return false;
}
2022-01-16 07:21:18 -08:00
return true;
2022-01-14 16:25:11 -08:00
}
static bool check_argn(lbm_uint argn, lbm_uint n) {
if (argn != n) {
lbm_set_error_reason(error_reason_argn);
return false;
} else {
return true;
}
}
static bool check_argn_number(lbm_value *args, lbm_uint argn, lbm_uint n) {
if (!is_number_all(args, argn)) {
lbm_set_error_reason(error_reason_no_number);
return false;
} else if (argn != n) {
lbm_set_error_reason(error_reason_argn);
return false;
} else {
return true;
}
}
#define CHECK_NUMBER_ALL() if (!check_number_all(args, argn)) {return ENC_SYM_EERROR;}
#define CHECK_ARGN(n) if (!check_argn(argn, n)) {return ENC_SYM_EERROR;}
#define CHECK_ARGN_NUMBER(n) if (!check_argn_number(args, argn, n)) {return ENC_SYM_EERROR;}
2022-01-14 11:54:24 -08:00
// Various commands
2022-01-21 01:52:56 -08:00
static lbm_value ext_print(lbm_value *args, lbm_uint argn) {
for (lbm_uint i = 0; i < argn; i ++) {
lbm_value t = args[i];
2022-01-14 11:54:24 -08:00
if (lbm_is_ptr(t) && lbm_type_of(t) == LBM_TYPE_ARRAY) {
2022-01-21 01:52:56 -08:00
lbm_array_header_t *array = (lbm_array_header_t *)lbm_car(t);
2022-01-14 11:54:24 -08:00
switch (array->elt_type){
case LBM_TYPE_CHAR:
2022-02-04 16:45:03 -08:00
commands_printf_lisp("%s", (char*)array->data);
2022-01-14 11:54:24 -08:00
break;
default:
return ENC_SYM_NIL;
2022-01-14 11:54:24 -08:00
break;
}
} else if (lbm_type_of(t) == LBM_TYPE_CHAR) {
2022-01-21 01:52:56 -08:00
if (lbm_dec_char(t) =='\n') {
commands_printf_lisp(" ");
2022-01-14 11:54:24 -08:00
} else {
commands_printf_lisp("%c", lbm_dec_char(t));
2022-01-14 11:54:24 -08:00
}
} else {
2022-11-09 07:00:43 -08:00
lbm_print_value(print_val_buffer, 256, t);
commands_printf_lisp("%s", print_val_buffer);
2022-01-14 11:54:24 -08:00
}
}
return ENC_SYM_TRUE;
2022-01-14 11:54:24 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_set_servo(lbm_value *args, lbm_uint argn) {
2022-01-16 07:21:18 -08:00
CHECK_ARGN_NUMBER(1);
servo_simple_set_output(lbm_dec_as_float(args[0]));
return ENC_SYM_TRUE;
2022-01-14 11:54:24 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_reset_timeout(lbm_value *args, lbm_uint argn) {
2022-01-14 11:54:24 -08:00
(void)args; (void)argn;
timeout_reset();
return ENC_SYM_TRUE;
2022-01-14 11:54:24 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_get_ppm(lbm_value *args, lbm_uint argn) {
2022-01-14 11:54:24 -08:00
(void)args; (void)argn;
return lbm_enc_float(servodec_get_servo(0));
2022-01-14 11:54:24 -08:00
}
static lbm_value ext_get_ppm_age(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
return lbm_enc_float((float)servodec_get_time_since_update() / 1000.0);
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_get_encoder(lbm_value *args, lbm_uint argn) {
2022-01-14 11:54:24 -08:00
(void)args; (void)argn;
return lbm_enc_float(encoder_read_deg());
2022-01-14 11:54:24 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_get_vin(lbm_value *args, lbm_uint argn) {
2022-01-14 11:54:24 -08:00
(void)args; (void)argn;
return lbm_enc_float(mc_interface_get_input_voltage_filtered());
2022-01-14 11:54:24 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_select_motor(lbm_value *args, lbm_uint argn) {
2022-01-16 07:21:18 -08:00
CHECK_ARGN_NUMBER(1);
int i = lbm_dec_as_i32(args[0]);
2022-01-14 11:54:24 -08:00
if (i != 0 && i != 1 && i != 2) {
return ENC_SYM_EERROR;
2022-01-14 11:54:24 -08:00
}
mc_interface_select_motor_thread(i);
return ENC_SYM_TRUE;
2022-01-14 11:54:24 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_get_selected_motor(lbm_value *args, lbm_uint argn) {
2022-01-14 16:25:11 -08:00
(void)args; (void)argn;
2022-01-21 01:52:56 -08:00
return lbm_enc_i(mc_interface_motor_now());
2022-01-14 16:25:11 -08:00
}
2022-11-09 12:53:17 -08:00
static lbm_value get_or_set_float(bool set, float *val, lbm_value *lbm_val) {
if (set) {
*val = lbm_dec_as_float(*lbm_val);
return ENC_SYM_TRUE;
} else {
return lbm_enc_float(*val);
}
}
static lbm_value get_or_set_i(bool set, int *val, lbm_value *lbm_val) {
if (set) {
*val = lbm_dec_as_i32(*lbm_val);
return ENC_SYM_TRUE;
} else {
return lbm_enc_i(*val);
}
}
static lbm_value get_or_set_bool(bool set, bool *val, lbm_value *lbm_val) {
if (set) {
*val = lbm_dec_as_i32(*lbm_val);
return ENC_SYM_TRUE;
} else {
return lbm_enc_i(*val);
}
}
static lbm_value get_set_bms_val(bool set, lbm_value *args, lbm_uint argn) {
lbm_value res = ENC_SYM_EERROR;
2022-01-14 16:25:11 -08:00
2022-11-09 12:53:17 -08:00
lbm_value set_arg = 0;
if (set && argn >= 1) {
set_arg = args[argn - 1];
argn--;
2022-11-09 13:16:12 -08:00
if (!lbm_is_number(set_arg)) {
lbm_set_error_reason(error_reason_no_number);
return ENC_SYM_EERROR;
}
2022-11-09 12:53:17 -08:00
}
2022-01-14 16:25:11 -08:00
if (argn != 1 && argn != 2) {
2022-03-31 10:37:36 -07:00
return res;
2022-01-14 16:25:11 -08:00
}
if (lbm_type_of(args[0]) != LBM_TYPE_SYMBOL) {
2022-03-31 10:37:36 -07:00
return res;
2022-01-14 16:25:11 -08:00
}
lbm_uint name = lbm_dec_sym(args[0]);
2022-11-09 12:53:17 -08:00
bms_values *val = (bms_values*)bms_get_values();
2022-01-14 16:25:11 -08:00
2022-03-31 06:18:59 -07:00
if (compare_symbol(name, &syms_vesc.v_tot)) {
2022-11-09 12:53:17 -08:00
res = get_or_set_float(set, &val->v_tot, &set_arg);
2022-03-31 06:18:59 -07:00
} else if (compare_symbol(name, &syms_vesc.v_charge)) {
2022-11-09 12:53:17 -08:00
res = get_or_set_float(set, &val->v_charge, &res);
2022-03-31 06:18:59 -07:00
} else if (compare_symbol(name, &syms_vesc.i_in)) {
2022-11-09 12:53:17 -08:00
res = get_or_set_float(set, &val->i_in, &set_arg);
2022-03-31 06:18:59 -07:00
} else if (compare_symbol(name, &syms_vesc.i_in_ic)) {
2022-11-09 12:53:17 -08:00
res = get_or_set_float(set, &val->i_in_ic, &set_arg);
2022-03-31 06:18:59 -07:00
} else if (compare_symbol(name, &syms_vesc.ah_cnt)) {
2022-11-09 12:53:17 -08:00
res = get_or_set_float(set, &val->ah_cnt, &set_arg);
2022-03-31 06:18:59 -07:00
} else if (compare_symbol(name, &syms_vesc.wh_cnt)) {
2022-11-09 12:53:17 -08:00
res = get_or_set_float(set, &val->wh_cnt, &set_arg);
2022-03-31 06:18:59 -07:00
} else if (compare_symbol(name, &syms_vesc.cell_num)) {
2022-11-09 12:53:17 -08:00
res = get_or_set_i(set, &val->cell_num, &set_arg);
2022-03-31 06:18:59 -07:00
} else if (compare_symbol(name, &syms_vesc.v_cell)) {
2022-01-21 01:52:56 -08:00
if (argn != 2 || !lbm_is_number(args[1])) {
return ENC_SYM_EERROR;
2022-01-14 16:25:11 -08:00
}
int c = lbm_dec_as_i32(args[1]);
2022-01-14 16:25:11 -08:00
if (c < 0 || c >= val->cell_num) {
return ENC_SYM_EERROR;
2022-01-14 16:25:11 -08:00
}
2022-11-09 12:53:17 -08:00
res = get_or_set_float(set, &val->v_cell[c], &set_arg);
2022-03-31 06:18:59 -07:00
} else if (compare_symbol(name, &syms_vesc.bal_state)) {
2022-01-21 01:52:56 -08:00
if (argn != 2 || !lbm_is_number(args[1])) {
return ENC_SYM_EERROR;
2022-01-14 16:25:11 -08:00
}
int c = lbm_dec_as_i32(args[1]);
2022-01-14 16:25:11 -08:00
if (c < 0 || c >= val->cell_num) {
return ENC_SYM_EERROR;
2022-01-14 16:25:11 -08:00
}
2022-11-09 12:53:17 -08:00
res = get_or_set_bool(set, &val->bal_state[c], &set_arg);
2022-03-31 06:18:59 -07:00
} else if (compare_symbol(name, &syms_vesc.temp_adc_num)) {
2022-11-09 12:53:17 -08:00
res = get_or_set_i(set, &val->temp_adc_num, &set_arg);
2022-03-31 06:18:59 -07:00
} else if (compare_symbol(name, &syms_vesc.temps_adc)) {
2022-01-21 01:52:56 -08:00
if (argn != 2 || !lbm_is_number(args[1])) {
return ENC_SYM_EERROR;
2022-01-14 16:25:11 -08:00
}
int c = lbm_dec_as_i32(args[1]);
2022-01-14 16:25:11 -08:00
if (c < 0 || c >= val->temp_adc_num) {
return ENC_SYM_EERROR;
2022-01-14 16:25:11 -08:00
}
2022-11-09 12:53:17 -08:00
res = get_or_set_float(set, &val->temps_adc[c], &set_arg);
2022-03-31 06:18:59 -07:00
} else if (compare_symbol(name, &syms_vesc.temp_ic)) {
2022-11-09 12:53:17 -08:00
res = get_or_set_float(set, &val->temp_ic, &set_arg);
2022-03-31 06:18:59 -07:00
} else if (compare_symbol(name, &syms_vesc.temp_hum)) {
2022-11-09 12:53:17 -08:00
res = get_or_set_float(set, &val->temp_hum, &set_arg);
2022-03-31 06:18:59 -07:00
} else if (compare_symbol(name, &syms_vesc.hum)) {
2022-11-09 12:53:17 -08:00
res = get_or_set_float(set, &val->hum, &set_arg);
2022-03-31 06:18:59 -07:00
} else if (compare_symbol(name, &syms_vesc.temp_max_cell)) {
2022-11-09 12:53:17 -08:00
res = get_or_set_float(set, &val->temp_max_cell, &set_arg);
2022-03-31 06:18:59 -07:00
} else if (compare_symbol(name, &syms_vesc.soc)) {
2022-11-09 12:53:17 -08:00
res = get_or_set_float(set, &val->soc, &set_arg);
2022-03-31 06:18:59 -07:00
} else if (compare_symbol(name, &syms_vesc.soh)) {
2022-11-09 12:53:17 -08:00
res = get_or_set_float(set, &val->soh, &set_arg);
2022-03-31 06:18:59 -07:00
} else if (compare_symbol(name, &syms_vesc.can_id)) {
2022-11-09 12:53:17 -08:00
res = get_or_set_i(set, &val->can_id, &set_arg);
2022-03-31 06:18:59 -07:00
} else if (compare_symbol(name, &syms_vesc.ah_cnt_chg_total)) {
2022-11-09 12:53:17 -08:00
res = get_or_set_float(set, &val->ah_cnt_chg_total, &set_arg);
2022-03-31 06:18:59 -07:00
} else if (compare_symbol(name, &syms_vesc.wh_cnt_chg_total)) {
2022-11-09 12:53:17 -08:00
res = get_or_set_float(set, &val->wh_cnt_chg_total, &set_arg);
2022-03-31 06:18:59 -07:00
} else if (compare_symbol(name, &syms_vesc.ah_cnt_dis_total)) {
2022-11-09 12:53:17 -08:00
res = get_or_set_float(set, &val->ah_cnt_dis_total, &set_arg);
2022-03-31 06:18:59 -07:00
} else if (compare_symbol(name, &syms_vesc.wh_cnt_dis_total)) {
2022-11-09 12:53:17 -08:00
res = get_or_set_float(set, &val->wh_cnt_dis_total, &set_arg);
2022-03-31 06:18:59 -07:00
} else if (compare_symbol(name, &syms_vesc.msg_age)) {
res = lbm_enc_float(UTILS_AGE_S(val->update_time));
2022-01-14 16:25:11 -08:00
}
2022-11-09 12:53:17 -08:00
if (res != ENC_SYM_EERROR && set) {
val->update_time = chVTGetSystemTimeX();
}
2022-01-14 16:25:11 -08:00
return res;
}
2022-11-09 12:53:17 -08:00
static lbm_value ext_get_bms_val(lbm_value *args, lbm_uint argn) {
return get_set_bms_val(false, args, argn);
}
static lbm_value ext_set_bms_val(lbm_value *args, lbm_uint argn) {
return get_set_bms_val(true, args, argn);
}
2022-11-10 06:24:34 -08:00
static lbm_value ext_send_bms_can(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
bms_send_status_can();
return ENC_SYM_TRUE;
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_get_adc(lbm_value *args, lbm_uint argn) {
2022-01-16 07:21:18 -08:00
CHECK_NUMBER_ALL();
2022-01-15 15:14:55 -08:00
if (argn == 0) {
return lbm_enc_float(ADC_VOLTS(ADC_IND_EXT));
2022-01-15 15:14:55 -08:00
} else if (argn == 1) {
lbm_int channel = lbm_dec_as_i32(args[0]);
2022-01-16 07:21:18 -08:00
if (channel == 0) {
return lbm_enc_float(ADC_VOLTS(ADC_IND_EXT));
2022-01-16 07:21:18 -08:00
} else if (channel == 1) {
return lbm_enc_float(ADC_VOLTS(ADC_IND_EXT2));
} else if (channel == 2) {
return lbm_enc_float(ADC_VOLTS(ADC_IND_EXT3));
2022-01-15 15:14:55 -08:00
} else {
return ENC_SYM_EERROR;
2022-01-15 15:14:55 -08:00
}
} else {
return ENC_SYM_EERROR;
2022-01-15 15:14:55 -08:00
}
}
static lbm_value ext_get_adc_decoded(lbm_value *args, lbm_uint argn) {
CHECK_NUMBER_ALL();
if (argn == 0) {
return lbm_enc_float(app_adc_get_decoded_level());
} else if (argn == 1) {
lbm_int channel = lbm_dec_as_i32(args[0]);
if (channel == 0) {
return lbm_enc_float(app_adc_get_decoded_level());
} else if (channel == 1) {
return lbm_enc_float(app_adc_get_decoded_level2());
} else {
return ENC_SYM_EERROR;
}
} else {
return ENC_SYM_EERROR;
}
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_systime(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
return lbm_enc_i32(chVTGetSystemTimeX());
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_secs_since(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(1);
return lbm_enc_float(UTILS_AGE_S(lbm_dec_as_u32(args[0])));
}
2022-01-25 11:59:15 -08:00
static lbm_value ext_set_aux(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(2);
int port = lbm_dec_as_u32(args[0]);
bool on = lbm_dec_as_u32(args[1]);
2022-01-25 11:59:15 -08:00
if (port == 1) {
if (on) {
AUX_ON();
} else {
AUX_OFF();
}
return ENC_SYM_TRUE;
2022-01-25 11:59:15 -08:00
} else if (port == 2) {
if (on) {
AUX2_ON();
} else {
AUX2_OFF();
}
return ENC_SYM_TRUE;
2022-01-25 11:59:15 -08:00
}
return ENC_SYM_EERROR;
2022-01-25 11:59:15 -08:00
}
2022-02-06 14:12:05 -08:00
static lbm_value ext_get_imu_rpy(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
float rpy[3];
imu_get_rpy(rpy);
lbm_value imu_data = ENC_SYM_NIL;
imu_data = lbm_cons(lbm_enc_float(rpy[2]), imu_data);
imu_data = lbm_cons(lbm_enc_float(rpy[1]), imu_data);
imu_data = lbm_cons(lbm_enc_float(rpy[0]), imu_data);
2022-02-06 14:12:05 -08:00
return imu_data;
}
static lbm_value ext_get_imu_quat(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
float q[4];
imu_get_quaternions(q);
lbm_value imu_data = ENC_SYM_NIL;
imu_data = lbm_cons(lbm_enc_float(q[3]), imu_data);
imu_data = lbm_cons(lbm_enc_float(q[2]), imu_data);
imu_data = lbm_cons(lbm_enc_float(q[1]), imu_data);
imu_data = lbm_cons(lbm_enc_float(q[0]), imu_data);
2022-02-06 14:12:05 -08:00
return imu_data;
}
static lbm_value ext_get_imu_acc(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
float acc[3];
imu_get_accel(acc);
lbm_value imu_data = ENC_SYM_NIL;
imu_data = lbm_cons(lbm_enc_float(acc[2]), imu_data);
imu_data = lbm_cons(lbm_enc_float(acc[1]), imu_data);
imu_data = lbm_cons(lbm_enc_float(acc[0]), imu_data);
2022-02-06 14:12:05 -08:00
return imu_data;
}
static lbm_value ext_get_imu_gyro(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
float gyro[3];
imu_get_gyro(gyro);
lbm_value imu_data = ENC_SYM_NIL;
imu_data = lbm_cons(lbm_enc_float(gyro[2]), imu_data);
imu_data = lbm_cons(lbm_enc_float(gyro[1]), imu_data);
imu_data = lbm_cons(lbm_enc_float(gyro[0]), imu_data);
2022-02-06 14:12:05 -08:00
return imu_data;
}
static lbm_value ext_get_imu_mag(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
float mag[3];
imu_get_mag(mag);
lbm_value imu_data = ENC_SYM_NIL;
imu_data = lbm_cons(lbm_enc_float(mag[2]), imu_data);
imu_data = lbm_cons(lbm_enc_float(mag[1]), imu_data);
imu_data = lbm_cons(lbm_enc_float(mag[0]), imu_data);
2022-02-06 14:12:05 -08:00
return imu_data;
}
static lbm_value ext_get_imu_acc_derot(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
float acc[3];
imu_get_accel_derotated(acc);
lbm_value imu_data = ENC_SYM_NIL;
imu_data = lbm_cons(lbm_enc_float(acc[2]), imu_data);
imu_data = lbm_cons(lbm_enc_float(acc[1]), imu_data);
imu_data = lbm_cons(lbm_enc_float(acc[0]), imu_data);
return imu_data;
}
static lbm_value ext_get_imu_gyro_derot(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
float gyro[3];
imu_get_gyro_derotated(gyro);
lbm_value imu_data = ENC_SYM_NIL;
imu_data = lbm_cons(lbm_enc_float(gyro[2]), imu_data);
imu_data = lbm_cons(lbm_enc_float(gyro[1]), imu_data);
imu_data = lbm_cons(lbm_enc_float(gyro[0]), imu_data);
return imu_data;
}
2022-02-06 14:12:05 -08:00
static lbm_value ext_send_data(lbm_value *args, lbm_uint argn) {
if (argn != 1 || (!lbm_is_cons(args[0]) && !lbm_is_array(args[0]))) {
return ENC_SYM_EERROR;
2022-02-06 14:12:05 -08:00
}
lbm_value curr = args[0];
const int max_len = 50;
2022-02-06 14:12:05 -08:00
uint8_t to_send[max_len];
2022-02-21 17:37:23 -08:00
uint8_t *to_send_ptr = to_send;
2022-02-06 14:12:05 -08:00
int ind = 0;
if (lbm_type_of(args[0]) == LBM_TYPE_ARRAY) {
2022-02-21 17:37:23 -08:00
lbm_array_header_t *array = (lbm_array_header_t *)lbm_car(args[0]);
if (array->elt_type != LBM_TYPE_BYTE) {
return ENC_SYM_EERROR;
2022-02-06 14:12:05 -08:00
}
2022-02-21 17:37:23 -08:00
to_send_ptr = (uint8_t*)array->data;
ind = array->size;
} else {
while (lbm_is_cons(curr)) {
2022-02-21 17:37:23 -08:00
lbm_value arg = lbm_car(curr);
2022-02-06 14:12:05 -08:00
2022-02-21 17:37:23 -08:00
if (lbm_is_number(arg)) {
to_send[ind++] = lbm_dec_as_u32(arg);
2022-02-21 17:37:23 -08:00
} else {
return ENC_SYM_EERROR;
2022-02-21 17:37:23 -08:00
}
if (ind == max_len) {
break;
}
curr = lbm_cdr(curr);
}
2022-02-06 14:12:05 -08:00
}
2022-02-21 17:37:23 -08:00
commands_send_app_data(to_send_ptr, ind);
2022-02-06 14:12:05 -08:00
return ENC_SYM_TRUE;
2022-02-06 14:12:05 -08:00
}
2022-03-13 01:15:24 -08:00
static lbm_value ext_get_remote_state(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
float gyro[3];
imu_get_gyro_derotated(gyro);
lbm_value state = ENC_SYM_NIL;
2022-03-13 01:15:24 -08:00
state = lbm_cons(lbm_enc_i(app_nunchuk_get_is_rev()), state);
state = lbm_cons(lbm_enc_i(app_nunchuk_get_bt_z()), state);
state = lbm_cons(lbm_enc_i(app_nunchuk_get_bt_c()), state);
state = lbm_cons(lbm_enc_float(app_nunchuk_get_decoded_x()), state);
state = lbm_cons(lbm_enc_float(app_nunchuk_get_decoded_y()), state);
2022-03-13 01:15:24 -08:00
return state;
}
2022-03-13 11:14:33 -07:00
static bool check_eeprom_addr(int addr) {
if (addr < 0 || addr > 127) {
lbm_set_error_reason("Address must be 0 to 127");
2022-03-13 11:14:33 -07:00
return false;
}
return true;
}
static lbm_value ext_eeprom_store_f(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(2);
int addr = lbm_dec_as_i32(args[0]);
2022-03-13 11:14:33 -07:00
if (!check_eeprom_addr(addr)) {
return ENC_SYM_EERROR;
2022-03-13 11:14:33 -07:00
}
eeprom_var v;
v.as_float = lbm_dec_as_float(args[1]);
2022-03-13 11:14:33 -07:00
return lbm_enc_i(conf_general_store_eeprom_var_custom(&v, addr));
}
static lbm_value ext_eeprom_read_f(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(1);
int addr = lbm_dec_as_i32(args[0]);
2022-03-13 11:14:33 -07:00
if (!check_eeprom_addr(addr)) {
return ENC_SYM_EERROR;
2022-03-13 11:14:33 -07:00
}
eeprom_var v;
bool res = conf_general_read_eeprom_var_custom(&v, addr);
return res ? lbm_enc_float(v.as_float) : ENC_SYM_NIL;
2022-03-13 11:14:33 -07:00
}
static lbm_value ext_eeprom_store_i(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(2);
int addr = lbm_dec_as_i32(args[0]);
2022-03-13 11:14:33 -07:00
if (!check_eeprom_addr(addr)) {
return ENC_SYM_EERROR;
2022-03-13 11:14:33 -07:00
}
eeprom_var v;
v.as_i32 = lbm_dec_as_float(args[1]);
2022-03-13 11:14:33 -07:00
return lbm_enc_i(conf_general_store_eeprom_var_custom(&v, addr));
}
static lbm_value ext_eeprom_read_i(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(1);
int addr = lbm_dec_as_i32(args[0]);
2022-03-13 11:14:33 -07:00
if (!check_eeprom_addr(addr)) {
return ENC_SYM_EERROR;
2022-03-13 11:14:33 -07:00
}
eeprom_var v;
bool res = conf_general_read_eeprom_var_custom(&v, addr);
return res ? lbm_enc_i32(v.as_i32) : ENC_SYM_NIL;
2022-03-13 11:14:33 -07:00
}
2022-04-10 12:21:35 -07:00
static lbm_value ext_sysinfo(lbm_value *args, lbm_uint argn) {
lbm_value res = ENC_SYM_EERROR;
2022-04-10 12:21:35 -07:00
if (argn != 1) {
return res;
}
if (lbm_type_of(args[0]) != LBM_TYPE_SYMBOL) {
return res;
}
lbm_uint name = lbm_dec_sym(args[0]);
if (compare_symbol(name, &syms_vesc.hw_name)) {
lbm_value lbm_res;
2022-04-28 09:00:07 -07:00
if (lbm_create_array(&lbm_res, LBM_TYPE_CHAR, strlen(HW_NAME) + 1)) {
2022-04-10 12:21:35 -07:00
lbm_array_header_t *arr = (lbm_array_header_t*)lbm_car(lbm_res);
strcpy((char*)arr->data, HW_NAME);
res = lbm_res;
} else {
res = ENC_SYM_MERROR;
2022-04-10 12:21:35 -07:00
}
} else if (compare_symbol(name, &syms_vesc.fw_ver)) {
res = ENC_SYM_NIL;
2022-04-10 12:21:35 -07:00
res = lbm_cons(lbm_enc_i(FW_TEST_VERSION_NUMBER), res);
res = lbm_cons(lbm_enc_i(FW_VERSION_MINOR), res);
res = lbm_cons(lbm_enc_i(FW_VERSION_MAJOR), res);
} else if (compare_symbol(name, &syms_vesc.has_phase_filters)) {
#ifdef HW_HAS_PHASE_FILTERS
res = ENC_SYM_TRUE;
2022-04-10 12:21:35 -07:00
#else
res = ENC_SYM_NIL;
2022-04-10 12:21:35 -07:00
#endif
} else if (compare_symbol(name, &syms_vesc.uuid)) {
res = ENC_SYM_NIL;
2022-04-10 12:21:35 -07:00
res = lbm_cons(lbm_enc_i(STM32_UUID_8[11]), res);
res = lbm_cons(lbm_enc_i(STM32_UUID_8[10]), res);
res = lbm_cons(lbm_enc_i(STM32_UUID_8[9]), res);
res = lbm_cons(lbm_enc_i(STM32_UUID_8[8]), res);
res = lbm_cons(lbm_enc_i(STM32_UUID_8[7]), res);
res = lbm_cons(lbm_enc_i(STM32_UUID_8[6]), res);
res = lbm_cons(lbm_enc_i(STM32_UUID_8[5]), res);
res = lbm_cons(lbm_enc_i(STM32_UUID_8[4]), res);
res = lbm_cons(lbm_enc_i(STM32_UUID_8[3]), res);
res = lbm_cons(lbm_enc_i(STM32_UUID_8[2]), res);
res = lbm_cons(lbm_enc_i(STM32_UUID_8[1]), res);
res = lbm_cons(lbm_enc_i(STM32_UUID_8[0]), res);
} else if (compare_symbol(name, &syms_vesc.runtime)) {
res = lbm_enc_u64(g_backup.runtime);
} else if (compare_symbol(name, &syms_vesc.git_branch)) {
lbm_value lbm_res;
if (lbm_create_array(&lbm_res, LBM_TYPE_CHAR, strlen(GIT_BRANCH_NAME) + 1)) {
lbm_array_header_t *arr = (lbm_array_header_t*)lbm_car(lbm_res);
strcpy((char*)arr->data, GIT_BRANCH_NAME);
res = lbm_res;
} else {
res = ENC_SYM_MERROR;
}
} else if (compare_symbol(name, &syms_vesc.git_hash)) {
lbm_value lbm_res;
if (lbm_create_array(&lbm_res, LBM_TYPE_CHAR, strlen(GIT_COMMIT_HASH) + 1)) {
lbm_array_header_t *arr = (lbm_array_header_t*)lbm_car(lbm_res);
strcpy((char*)arr->data, GIT_COMMIT_HASH);
res = lbm_res;
} else {
res = ENC_SYM_MERROR;
}
} else if (compare_symbol(name, &syms_vesc.compiler)) {
lbm_value lbm_res;
if (lbm_create_array(&lbm_res, LBM_TYPE_CHAR, strlen(ARM_GCC_VERSION) + 1)) {
lbm_array_header_t *arr = (lbm_array_header_t*)lbm_car(lbm_res);
strcpy((char*)arr->data, ARM_GCC_VERSION);
res = lbm_res;
} else {
res = ENC_SYM_MERROR;
}
2022-04-10 12:21:35 -07:00
}
return res;
}
// App set commands
static lbm_value ext_app_adc_detach(lbm_value *args, lbm_uint argn) {
if (argn == 1){
if(lbm_dec_as_u32(args[0]) != 0){
return ENC_SYM_EERROR;
}
}else{
CHECK_ARGN_NUMBER(2);
}
uint32_t mode = lbm_dec_as_u32(args[0]);
bool detach = lbm_dec_as_char(args[1]) > 0 ? true : false;
switch (mode){
case 0:
app_adc_detach_adc(false);
app_adc_detach_buttons(false);
break;
case 1:
app_adc_detach_adc(detach);
break;
case 2:
app_adc_detach_buttons(detach);
break;
case 3:
app_adc_detach_adc(detach);
app_adc_detach_buttons(detach);
break;
default:
return ENC_SYM_EERROR;
}
return ENC_SYM_TRUE;
}
static lbm_value ext_app_adc_override(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(2);
uint32_t target = lbm_dec_as_u32(args[0]);
float val = lbm_dec_as_float(args[1]);
bool state = val > 0.0 ? true : false;
switch (target){
case 0:
app_adc_adc1_override(val);
break;
case 1:
app_adc_adc2_override(val);
break;
case 2:
app_adc_rev_override(state);
break;
case 3:
app_adc_cc_override(state);
break;
default:
return ENC_SYM_EERROR;
}
return ENC_SYM_TRUE;
}
2022-10-07 07:07:32 -07:00
static lbm_value ext_app_ppm_detach(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(1);
app_ppm_detach(lbm_dec_as_u32(args[0]) > 0);
return ENC_SYM_TRUE;
}
static lbm_value ext_app_ppm_override(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(1);
app_ppm_override(lbm_dec_as_float(args[0]));
return ENC_SYM_TRUE;
}
2022-10-07 10:36:52 -07:00
static lbm_value ext_set_remote_state(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(5);
chuck_data chuk = {0};
float js_y = (lbm_dec_as_float(args[0]) + 1.0) * 128.0;
utils_truncate_number(&js_y, 0.0, 255.0);
chuk.js_y = (int)js_y;
float js_x = (lbm_dec_as_float(args[1]) + 1.0) * 128.0;
utils_truncate_number(&js_x, 0.0, 255.0);
chuk.js_x = (int)js_x;
chuk.bt_c = lbm_dec_as_u32(args[2]) > 0;
chuk.bt_z = lbm_dec_as_u32(args[3]) > 0;
chuk.is_rev = lbm_dec_as_u32(args[4]) > 0;
chuk.rev_has_state = true;
app_nunchuk_update_output(&chuk);
return ENC_SYM_TRUE;
}
static lbm_value ext_app_disable_output(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(1);
app_disable_output(lbm_dec_as_i32(args[0]));
return ENC_SYM_TRUE;
}
2022-11-09 02:56:59 -08:00
static lbm_value ext_app_pas_get_rpm(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
return lbm_enc_float(app_pas_get_pedal_rpm());
}
2022-01-14 11:54:24 -08:00
// Motor set commands
2022-01-21 01:52:56 -08:00
static lbm_value ext_set_current(lbm_value *args, lbm_uint argn) {
2022-07-05 04:41:24 -07:00
CHECK_NUMBER_ALL();
timeout_reset();
2022-07-05 04:41:24 -07:00
if (argn == 1) {
mc_interface_set_current(lbm_dec_as_float(args[0]));
} else if (argn == 2) {
mc_interface_set_current_off_delay(lbm_dec_as_float(args[1]));
mc_interface_set_current(lbm_dec_as_float(args[0]));
} else {
return ENC_SYM_EERROR;
2022-07-05 04:41:24 -07:00
}
return ENC_SYM_TRUE;
2022-01-14 11:54:24 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_set_current_rel(lbm_value *args, lbm_uint argn) {
2022-07-05 04:41:24 -07:00
CHECK_NUMBER_ALL();
timeout_reset();
2022-07-05 04:41:24 -07:00
if (argn == 1) {
mc_interface_set_current_rel(lbm_dec_as_float(args[0]));
} else if (argn == 2) {
mc_interface_set_current_off_delay(lbm_dec_as_float(args[1]));
mc_interface_set_current_rel(lbm_dec_as_float(args[0]));
} else {
return ENC_SYM_EERROR;
2022-07-05 04:41:24 -07:00
}
return ENC_SYM_TRUE;
2022-01-14 11:54:24 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_set_duty(lbm_value *args, lbm_uint argn) {
2022-01-16 07:21:18 -08:00
CHECK_ARGN_NUMBER(1);
2022-03-03 01:39:28 -08:00
timeout_reset();
mc_interface_set_duty(lbm_dec_as_float(args[0]));
return ENC_SYM_TRUE;
2022-01-14 11:54:24 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_set_brake(lbm_value *args, lbm_uint argn) {
2022-01-16 07:21:18 -08:00
CHECK_ARGN_NUMBER(1);
2022-03-03 01:39:28 -08:00
timeout_reset();
mc_interface_set_brake_current(lbm_dec_as_float(args[0]));
return ENC_SYM_TRUE;
2022-01-14 11:54:24 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_set_brake_rel(lbm_value *args, lbm_uint argn) {
2022-01-16 07:21:18 -08:00
CHECK_ARGN_NUMBER(1);
2022-03-03 01:39:28 -08:00
timeout_reset();
mc_interface_set_brake_current_rel(lbm_dec_as_float(args[0]));
return ENC_SYM_TRUE;
2022-01-14 11:54:24 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_set_handbrake(lbm_value *args, lbm_uint argn) {
2022-01-16 07:21:18 -08:00
CHECK_ARGN_NUMBER(1);
2022-03-03 01:39:28 -08:00
timeout_reset();
mc_interface_set_handbrake(lbm_dec_as_float(args[0]));
return ENC_SYM_TRUE;
2022-01-14 11:54:24 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_set_handbrake_rel(lbm_value *args, lbm_uint argn) {
2022-01-16 07:21:18 -08:00
CHECK_ARGN_NUMBER(1);
2022-03-03 01:39:28 -08:00
timeout_reset();
mc_interface_set_handbrake_rel(lbm_dec_as_float(args[0]));
return ENC_SYM_TRUE;
2022-01-14 11:54:24 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_set_rpm(lbm_value *args, lbm_uint argn) {
2022-01-16 07:21:18 -08:00
CHECK_ARGN_NUMBER(1);
2022-03-03 01:39:28 -08:00
timeout_reset();
mc_interface_set_pid_speed(lbm_dec_as_float(args[0]));
return ENC_SYM_TRUE;
2022-01-14 11:54:24 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_set_pos(lbm_value *args, lbm_uint argn) {
2022-01-16 07:21:18 -08:00
CHECK_ARGN_NUMBER(1);
2022-03-03 01:39:28 -08:00
timeout_reset();
mc_interface_set_pid_pos(lbm_dec_as_float(args[0]));
return ENC_SYM_TRUE;
2022-01-14 11:54:24 -08:00
}
2022-04-14 12:17:45 -07:00
static lbm_value ext_foc_openloop(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(2);
timeout_reset();
2022-11-08 03:46:08 -08:00
mc_interface_set_openloop_current(lbm_dec_as_float(args[0]), lbm_dec_as_float(args[1]));
return ENC_SYM_TRUE;
2022-04-14 12:17:45 -07:00
}
2022-06-17 01:43:19 -07:00
static lbm_value ext_foc_beep(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(3);
timeout_reset();
mcpwm_foc_beep(lbm_dec_as_float(args[0]), lbm_dec_as_float(args[1]), lbm_dec_as_float(args[2]));
return ENC_SYM_TRUE;
2022-06-17 01:43:19 -07:00
}
2022-01-14 11:54:24 -08:00
// Motor get commands
2022-01-21 01:52:56 -08:00
static lbm_value ext_get_current(lbm_value *args, lbm_uint argn) {
2022-01-14 11:54:24 -08:00
(void)args; (void)argn;
return lbm_enc_float(mc_interface_get_tot_current_filtered());
2022-01-14 11:54:24 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_get_current_dir(lbm_value *args, lbm_uint argn) {
2022-01-14 11:54:24 -08:00
(void)args; (void)argn;
return lbm_enc_float(mc_interface_get_tot_current_directional_filtered());
2022-01-14 11:54:24 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_get_current_in(lbm_value *args, lbm_uint argn) {
2022-01-14 11:54:24 -08:00
(void)args; (void)argn;
return lbm_enc_float(mc_interface_get_tot_current_in_filtered());
2022-01-14 11:54:24 -08:00
}
static lbm_value ext_get_id(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
return lbm_enc_float(mcpwm_foc_get_id_filter());
}
static lbm_value ext_get_iq(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
return lbm_enc_float(mcpwm_foc_get_iq_filter());
}
static lbm_value ext_get_vd(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
return lbm_enc_float(mcpwm_foc_get_vd());
}
static lbm_value ext_get_vq(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
return lbm_enc_float(mcpwm_foc_get_vq());
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_get_duty(lbm_value *args, lbm_uint argn) {
2022-01-14 11:54:24 -08:00
(void)args; (void)argn;
return lbm_enc_float(mc_interface_get_duty_cycle_now());
2022-01-14 11:54:24 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_get_rpm(lbm_value *args, lbm_uint argn) {
2022-01-14 11:54:24 -08:00
(void)args; (void)argn;
return lbm_enc_float(mc_interface_get_rpm());
2022-01-14 11:54:24 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_get_temp_fet(lbm_value *args, lbm_uint argn) {
2022-01-14 11:54:24 -08:00
(void)args; (void)argn;
return lbm_enc_float(mc_interface_temp_fet_filtered());
2022-01-14 11:54:24 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_get_temp_mot(lbm_value *args, lbm_uint argn) {
2022-01-14 11:54:24 -08:00
(void)args; (void)argn;
return lbm_enc_float(mc_interface_temp_motor_filtered());
2022-01-14 11:54:24 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_get_speed(lbm_value *args, lbm_uint argn) {
2022-01-14 11:54:24 -08:00
(void)args; (void)argn;
return lbm_enc_float(mc_interface_get_speed());
2022-01-14 11:54:24 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_get_dist(lbm_value *args, lbm_uint argn) {
2022-01-14 11:54:24 -08:00
(void)args; (void)argn;
return lbm_enc_float(mc_interface_get_distance());
2022-01-14 11:54:24 -08:00
}
2022-07-20 02:04:57 -07:00
static lbm_value ext_get_dist_abs(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
return lbm_enc_float(mc_interface_get_distance_abs());
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_get_batt(lbm_value *args, lbm_uint argn) {
2022-01-14 11:54:24 -08:00
(void)args; (void)argn;
return lbm_enc_float(mc_interface_get_battery_level(0));
2022-01-14 11:54:24 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_get_fault(lbm_value *args, lbm_uint argn) {
2022-01-14 11:54:24 -08:00
(void)args; (void)argn;
2022-01-21 01:52:56 -08:00
return lbm_enc_i(mc_interface_get_fault());
2022-01-14 11:54:24 -08:00
}
2022-01-16 07:21:18 -08:00
// CAN-commands
2022-01-21 01:52:56 -08:00
static lbm_value ext_can_current(lbm_value *args, lbm_uint argn) {
2022-01-16 07:21:18 -08:00
CHECK_NUMBER_ALL();
if (argn == 2) {
comm_can_set_current(lbm_dec_as_i32(args[0]), lbm_dec_as_float(args[1]));
2022-01-16 07:21:18 -08:00
} else if (argn == 3) {
comm_can_set_current_off_delay(lbm_dec_as_i32(args[0]), lbm_dec_as_float(args[1]), lbm_dec_as_float(args[2]));
2022-01-16 07:21:18 -08:00
} else {
return ENC_SYM_EERROR;
2022-01-16 07:21:18 -08:00
}
return ENC_SYM_TRUE;
2022-01-16 07:21:18 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_can_current_rel(lbm_value *args, lbm_uint argn) {
2022-01-16 07:21:18 -08:00
CHECK_NUMBER_ALL();
if (argn == 2) {
comm_can_set_current_rel(lbm_dec_as_i32(args[0]), lbm_dec_as_float(args[1]));
2022-01-16 07:21:18 -08:00
} else if (argn == 3) {
comm_can_set_current_rel_off_delay(lbm_dec_as_i32(args[0]), lbm_dec_as_float(args[1]), lbm_dec_as_float(args[2]));
2022-01-16 07:21:18 -08:00
} else {
return ENC_SYM_EERROR;
2022-01-16 07:21:18 -08:00
}
return ENC_SYM_TRUE;
2022-01-16 07:21:18 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_can_duty(lbm_value *args, lbm_uint argn) {
2022-01-16 07:21:18 -08:00
CHECK_ARGN_NUMBER(2);
comm_can_set_duty(lbm_dec_as_i32(args[0]), lbm_dec_as_float(args[1]));
return ENC_SYM_TRUE;
2022-01-16 07:21:18 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_can_brake(lbm_value *args, lbm_uint argn) {
2022-01-16 07:21:18 -08:00
CHECK_ARGN_NUMBER(2);
comm_can_set_current_brake(lbm_dec_as_i32(args[0]), lbm_dec_as_float(args[1]));
return ENC_SYM_TRUE;
2022-01-16 07:21:18 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_can_brake_rel(lbm_value *args, lbm_uint argn) {
2022-01-16 07:21:18 -08:00
CHECK_ARGN_NUMBER(2);
comm_can_set_current_brake_rel(lbm_dec_as_i32(args[0]), lbm_dec_as_float(args[1]));
return ENC_SYM_TRUE;
2022-01-16 07:21:18 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_can_rpm(lbm_value *args, lbm_uint argn) {
2022-01-16 07:21:18 -08:00
CHECK_ARGN_NUMBER(2);
comm_can_set_rpm(lbm_dec_as_i32(args[0]), lbm_dec_as_float(args[1]));
return ENC_SYM_TRUE;
2022-01-16 07:21:18 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_can_pos(lbm_value *args, lbm_uint argn) {
2022-01-16 07:21:18 -08:00
CHECK_ARGN_NUMBER(2);
comm_can_set_pos(lbm_dec_as_i32(args[0]), lbm_dec_as_float(args[1]));
return ENC_SYM_TRUE;
2022-01-21 01:52:56 -08:00
}
static lbm_value ext_can_get_current(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(1);
can_status_msg *stat0 = comm_can_get_status_msg_id(lbm_dec_as_i32(args[0]));
2022-01-21 01:52:56 -08:00
if (stat0) {
return lbm_enc_float(stat0->current);
2022-01-21 01:52:56 -08:00
} else {
return lbm_enc_float(0.0);
2022-01-21 01:52:56 -08:00
}
}
static lbm_value ext_can_get_current_dir(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(1);
can_status_msg *stat0 = comm_can_get_status_msg_id(lbm_dec_as_i32(args[0]));
2022-01-21 01:52:56 -08:00
if (stat0) {
return lbm_enc_float(stat0->current * SIGN(stat0->duty));
2022-01-21 01:52:56 -08:00
} else {
return lbm_enc_float(0.0);
2022-01-21 01:52:56 -08:00
}
2022-01-16 07:21:18 -08:00
}
2022-02-13 10:20:54 -08:00
static lbm_value ext_can_get_current_in(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(1);
can_status_msg_4 *stat4 = comm_can_get_status_msg_4_id(lbm_dec_as_i32(args[0]));
2022-02-13 10:20:54 -08:00
if (stat4) {
return lbm_enc_float((float)stat4->current_in);
2022-02-13 10:20:54 -08:00
} else {
return lbm_enc_float(0.0);
2022-02-13 10:20:54 -08:00
}
}
static lbm_value ext_can_get_duty(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(1);
can_status_msg *stat0 = comm_can_get_status_msg_id(lbm_dec_as_i32(args[0]));
2022-02-13 10:20:54 -08:00
if (stat0) {
return lbm_enc_float(stat0->duty);
2022-02-13 10:20:54 -08:00
} else {
return lbm_enc_float(0.0);
2022-02-13 10:20:54 -08:00
}
}
static lbm_value ext_can_get_rpm(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(1);
can_status_msg *stat0 = comm_can_get_status_msg_id(lbm_dec_as_i32(args[0]));
2022-02-13 10:20:54 -08:00
if (stat0) {
return lbm_enc_float(stat0->rpm);
2022-02-13 10:20:54 -08:00
} else {
return lbm_enc_float(0.0);
2022-02-13 10:20:54 -08:00
}
}
static lbm_value ext_can_get_temp_fet(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(1);
can_status_msg_4 *stat4 = comm_can_get_status_msg_4_id(lbm_dec_as_i32(args[0]));
2022-02-13 10:20:54 -08:00
if (stat4) {
return lbm_enc_float((float)stat4->temp_fet);
2022-02-13 10:20:54 -08:00
} else {
return lbm_enc_float(0.0);
2022-02-13 10:20:54 -08:00
}
}
static lbm_value ext_can_get_temp_motor(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(1);
can_status_msg_4 *stat4 = comm_can_get_status_msg_4_id(lbm_dec_as_i32(args[0]));
2022-02-13 10:20:54 -08:00
if (stat4) {
return lbm_enc_float((float)stat4->temp_motor);
2022-02-13 10:20:54 -08:00
} else {
return lbm_enc_float(0.0);
2022-02-13 10:20:54 -08:00
}
}
static lbm_value ext_can_get_speed(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(1);
can_status_msg *stat0 = comm_can_get_status_msg_id(lbm_dec_as_i32(args[0]));
2022-02-13 10:20:54 -08:00
if (stat0) {
const volatile mc_configuration *conf = mc_interface_get_configuration();
const float rpm = stat0->rpm / (conf->si_motor_poles / 2.0);
return lbm_enc_float((rpm / 60.0) * conf->si_wheel_diameter * M_PI / conf->si_gear_ratio);
2022-02-13 10:20:54 -08:00
} else {
return lbm_enc_float(0.0);
2022-02-13 10:20:54 -08:00
}
}
static lbm_value ext_can_get_dist(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(1);
can_status_msg_5 *stat5 = comm_can_get_status_msg_5_id(lbm_dec_as_i32(args[0]));
2022-02-13 10:20:54 -08:00
if (stat5) {
const volatile mc_configuration *conf = mc_interface_get_configuration();
const float tacho_scale = (conf->si_wheel_diameter * M_PI) / (3.0 * conf->si_motor_poles * conf->si_gear_ratio);
return lbm_enc_float((float)stat5->tacho_value * tacho_scale);
2022-02-13 10:20:54 -08:00
} else {
return lbm_enc_float(0.0);
2022-02-13 10:20:54 -08:00
}
}
static lbm_value ext_can_get_ppm(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(1);
can_status_msg_6 *stat6 = comm_can_get_status_msg_6_id(lbm_dec_as_i32(args[0]));
if (stat6) {
return lbm_enc_float((float)stat6->ppm);
} else {
return lbm_enc_float(0.0);
}
}
static lbm_value ext_can_get_adc(lbm_value *args, lbm_uint argn) {
if (argn != 1 && argn != 2) {
return ENC_SYM_EERROR;
}
CHECK_NUMBER_ALL();
lbm_int channel = 0;
if (argn == 2) {
channel = lbm_dec_as_i32(args[1]);
}
can_status_msg_6 *stat6 = comm_can_get_status_msg_6_id(lbm_dec_as_i32(args[0]));
if (stat6) {
if (channel == 0) {
return lbm_enc_float(stat6->adc_1);
} else if (channel == 1) {
return lbm_enc_float(stat6->adc_2);
} else if (channel == 2) {
return lbm_enc_float(stat6->adc_3);
} else {
return ENC_SYM_EERROR;
}
} else {
return lbm_enc_float(-1.0);
}
}
2022-02-13 10:20:54 -08:00
static int cmp_int (const void * a, const void * b) {
return ( *(int*)a - *(int*)b );
}
static lbm_value ext_can_list_devs(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
int dev_num = 0;
2022-02-13 12:49:26 -08:00
can_status_msg *msg = comm_can_get_status_msg_index(dev_num);
while (msg && msg->id >= 0) {
2022-02-13 10:20:54 -08:00
dev_num++;
2022-02-13 12:49:26 -08:00
msg = comm_can_get_status_msg_index(dev_num);
2022-02-13 10:20:54 -08:00
}
int devs[dev_num];
for (int i = 0;i < dev_num;i++) {
2022-02-13 12:49:26 -08:00
msg = comm_can_get_status_msg_index(i);
2022-02-13 10:20:54 -08:00
if (msg) {
devs[i] = msg->id;
} else {
devs[i] = -1;
}
}
qsort(devs, dev_num, sizeof(int), cmp_int);
lbm_value dev_list = ENC_SYM_NIL;
2022-02-13 10:20:54 -08:00
2022-02-13 12:49:26 -08:00
for (int i = (dev_num - 1);i >= 0;i--) {
2022-02-13 10:20:54 -08:00
if (devs[i] >= 0) {
dev_list = lbm_cons(lbm_enc_i(devs[i]), dev_list);
} else {
break;
}
}
return dev_list;
}
static lbm_value ext_can_scan(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
lbm_value dev_list = ENC_SYM_NIL;
2022-02-13 10:20:54 -08:00
2022-02-13 12:49:26 -08:00
for (int i = 253;i >= 0;i--) {
2022-02-13 10:20:54 -08:00
if (comm_can_ping(i, 0)) {
dev_list = lbm_cons(lbm_enc_i(i), dev_list);
}
}
return dev_list;
}
static lbm_value ext_can_send(lbm_value *args, lbm_uint argn, bool is_eid) {
2022-02-21 17:37:23 -08:00
if (argn != 2 || !lbm_is_number(args[0])) {
return ENC_SYM_EERROR;
}
lbm_value curr = args[1];
uint8_t to_send[8];
int ind = 0;
if (lbm_type_of(args[0]) == LBM_TYPE_ARRAY) {
2022-02-21 17:37:23 -08:00
lbm_array_header_t *array = (lbm_array_header_t *)lbm_car(args[0]);
if (array->elt_type != LBM_TYPE_BYTE) {
return ENC_SYM_EERROR;
}
2022-02-21 17:37:23 -08:00
ind = array->size;
if (ind > 8) {
ind = 0;
}
2022-02-21 17:37:23 -08:00
memcpy(to_send, array->data, ind);
} else {
while (lbm_is_cons(curr)) {
2022-02-21 17:37:23 -08:00
lbm_value arg = lbm_car(curr);
if (lbm_is_number(arg)) {
to_send[ind++] = lbm_dec_as_u32(arg);
2022-02-21 17:37:23 -08:00
} else {
return ENC_SYM_EERROR;
2022-02-21 17:37:23 -08:00
}
if (ind == 8) {
break;
}
curr = lbm_cdr(curr);
}
}
if (is_eid) {
comm_can_transmit_eid(lbm_dec_as_u32(args[0]), to_send, ind);
} else {
comm_can_transmit_sid(lbm_dec_as_u32(args[0]), to_send, ind);
}
return ENC_SYM_TRUE;
}
static lbm_value ext_can_send_sid(lbm_value *args, lbm_uint argn) {
return ext_can_send(args, argn, false);
}
static lbm_value ext_can_send_eid(lbm_value *args, lbm_uint argn) {
return ext_can_send(args, argn, true);
}
2022-01-16 07:21:18 -08:00
// Math
2022-01-21 01:52:56 -08:00
static lbm_value ext_sin(lbm_value *args, lbm_uint argn) {
2022-01-16 07:21:18 -08:00
CHECK_ARGN_NUMBER(1)
return lbm_enc_float(sinf(lbm_dec_as_float(args[0])));
2022-01-16 07:21:18 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_cos(lbm_value *args, lbm_uint argn) {
2022-01-16 07:21:18 -08:00
CHECK_ARGN_NUMBER(1)
return lbm_enc_float(cosf(lbm_dec_as_float(args[0])));
2022-01-16 07:21:18 -08:00
}
2022-02-14 10:19:48 -08:00
static lbm_value ext_tan(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(1)
return lbm_enc_float(tanf(lbm_dec_as_float(args[0])));
2022-02-14 10:19:48 -08:00
}
static lbm_value ext_asin(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(1)
return lbm_enc_float(asinf(lbm_dec_as_float(args[0])));
2022-02-14 10:19:48 -08:00
}
static lbm_value ext_acos(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(1)
return lbm_enc_float(acosf(lbm_dec_as_float(args[0])));
2022-02-14 10:19:48 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_atan(lbm_value *args, lbm_uint argn) {
2022-01-16 07:21:18 -08:00
CHECK_ARGN_NUMBER(1)
return lbm_enc_float(atanf(lbm_dec_as_float(args[0])));
2022-01-16 07:21:18 -08:00
}
2022-02-05 15:23:23 -08:00
static lbm_value ext_atan2(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(2)
return lbm_enc_float(atan2f(lbm_dec_as_float(args[0]), lbm_dec_as_float(args[1])));
2022-02-05 15:23:23 -08:00
}
2022-01-21 01:52:56 -08:00
static lbm_value ext_pow(lbm_value *args, lbm_uint argn) {
2022-01-16 07:21:18 -08:00
CHECK_ARGN_NUMBER(2)
return lbm_enc_float(powf(lbm_dec_as_float(args[0]), lbm_dec_as_float(args[1])));
2022-01-16 07:21:18 -08:00
}
2022-02-06 14:12:05 -08:00
static lbm_value ext_sqrt(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(1)
return lbm_enc_float(sqrtf(lbm_dec_as_float(args[0])));
2022-02-06 14:12:05 -08:00
}
2022-02-14 10:19:48 -08:00
static lbm_value ext_log(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(1)
return lbm_enc_float(logf(lbm_dec_as_float(args[0])));
2022-02-14 10:19:48 -08:00
}
static lbm_value ext_log10(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(1)
return lbm_enc_float(log10f(lbm_dec_as_float(args[0])));
2022-02-14 10:19:48 -08:00
}
static lbm_value ext_deg2rad(lbm_value *args, lbm_uint argn) {
CHECK_NUMBER_ALL();
if (argn == 1) {
return lbm_enc_float(DEG2RAD_f(lbm_dec_as_float(args[0])));
} else {
lbm_value out_list = ENC_SYM_NIL;
for (int i = (argn - 1);i >= 0;i--) {
out_list = lbm_cons(lbm_enc_float(DEG2RAD_f(lbm_dec_as_float(args[i]))), out_list);
}
return out_list;
}
}
static lbm_value ext_rad2deg(lbm_value *args, lbm_uint argn) {
CHECK_NUMBER_ALL();
if (argn == 1) {
return lbm_enc_float(RAD2DEG_f(lbm_dec_as_float(args[0])));
} else {
lbm_value out_list = ENC_SYM_NIL;
for (int i = (argn - 1);i >= 0;i--) {
out_list = lbm_cons(lbm_enc_float(RAD2DEG_f(lbm_dec_as_float(args[i]))), out_list);
}
return out_list;
}
}
static lbm_value ext_vec3_rot(lbm_value *args, lbm_uint argn) {
CHECK_NUMBER_ALL();
if (argn != 6 && argn != 7) {
return ENC_SYM_EERROR;
}
float input[] = {lbm_dec_as_float(args[0]), lbm_dec_as_float(args[1]), lbm_dec_as_float(args[2])};
float rotation[] = {lbm_dec_as_float(args[3]), lbm_dec_as_float(args[4]), lbm_dec_as_float(args[5])};
float output[3];
bool reverse = false;
if (argn == 7) {
reverse = lbm_dec_as_i32(args[6]);
}
utils_rotate_vector3(input, rotation, output, reverse);
lbm_value out_list = ENC_SYM_NIL;
out_list = lbm_cons(lbm_enc_float(output[2]), out_list);
out_list = lbm_cons(lbm_enc_float(output[1]), out_list);
out_list = lbm_cons(lbm_enc_float(output[0]), out_list);
return out_list;
}
static lbm_value ext_throttle_curve(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(4);
return lbm_enc_float(utils_throttle_curve(
lbm_dec_as_float(args[0]),
lbm_dec_as_float(args[1]),
lbm_dec_as_float(args[2]),
lbm_dec_as_i32(args[3])));
}
// Bit operations
/*
* args[0]: Initial value
* args[1]: Offset in initial value to modify
* args[2]: Value to modify with
* args[3]: Size in bits of value to modify with
*/
static lbm_value ext_bits_enc_int(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(4)
2022-09-24 06:29:31 -07:00
uint32_t initial = lbm_dec_as_u32(args[0]);
uint32_t offset = lbm_dec_as_u32(args[1]);
uint32_t number = lbm_dec_as_u32(args[2]);
uint32_t bits = lbm_dec_as_u32(args[3]);
initial &= ~((0xFFFFFFFF >> (32 - bits)) << offset);
initial |= (number << (32 - bits)) >> (32 - bits - offset);
if (initial > ((1 << 27) - 1)) {
return lbm_enc_i32(initial);
} else {
return lbm_enc_i(initial);
}
}
/*
* args[0]: Value
* args[1]: Offset in initial value to get
* args[2]: Size in bits of value to get
*/
static lbm_value ext_bits_dec_int(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(3)
2022-09-24 06:29:31 -07:00
uint32_t val = lbm_dec_as_u32(args[0]);
uint32_t offset = lbm_dec_as_u32(args[1]);
uint32_t bits = lbm_dec_as_u32(args[2]);
val >>= offset;
val &= 0xFFFFFFFF >> (32 - bits);
if (val > ((1 << 27) - 1)) {
return lbm_enc_i32(val);
} else {
return lbm_enc_i(val);
}
}
// Events that will be sent to lisp if a handler is registered
static volatile bool event_handler_registered = false;
static volatile bool event_can_sid_en = false;
static volatile bool event_can_eid_en = false;
2022-02-06 14:12:05 -08:00
static volatile bool event_data_rx_en = false;
static volatile bool event_shutdown_en = false;
static lbm_uint event_handler_pid;
static lbm_uint sym_event_can_sid;
static lbm_uint sym_event_can_eid;
static lbm_uint sym_event_data_rx;
static lbm_uint sym_event_shutdown;
static lbm_value ext_enable_event(lbm_value *args, lbm_uint argn) {
if (argn != 1 && argn != 2) {
return ENC_SYM_EERROR;
}
if (argn == 2 && !lbm_is_number(args[1])) {
return ENC_SYM_EERROR;
}
bool en = true;
if (argn == 2 && !lbm_dec_as_i32(args[1])) {
en = false;
}
lbm_uint name = lbm_dec_sym(args[0]);
if (name == sym_event_can_sid) {
event_can_sid_en = en;
} else if (name == sym_event_can_eid) {
event_can_eid_en = en;
} else if (name == sym_event_data_rx) {
2022-02-06 14:12:05 -08:00
event_data_rx_en = en;
} else if (name == sym_event_shutdown) {
event_shutdown_en = en;
} else {
return ENC_SYM_EERROR;
}
return ENC_SYM_TRUE;
}
static lbm_value ext_register_event_handler(lbm_value *args, lbm_uint argn) {
if (argn != 1 || !lbm_is_number(args[0])) {
return ENC_SYM_EERROR;
}
event_handler_pid = lbm_dec_i(args[0]);
event_handler_registered = true;
return ENC_SYM_TRUE;
}
2022-02-06 14:12:05 -08:00
/*
* args[0]: Motor, 1 or 2
* args[1]: Phase, 1, 2 or 3
* args[2]: Use raw ADC values. Optional argument.
*/
static lbm_value ext_raw_adc_current(lbm_value *args, lbm_uint argn) {
CHECK_NUMBER_ALL();
if (argn != 2 && argn != 3) {
return ENC_SYM_EERROR;
2022-02-06 14:12:05 -08:00
}
uint32_t motor = lbm_dec_as_i32(args[0]);
uint32_t phase = lbm_dec_as_i32(args[1]);
2022-02-06 14:12:05 -08:00
volatile float ofs1, ofs2, ofs3;
mcpwm_foc_get_current_offsets(&ofs1, &ofs2, &ofs3, motor == 2);
float scale = FAC_CURRENT;
if (argn == 3 && lbm_dec_as_i32(args[2]) != 0) {
2022-02-06 14:12:05 -08:00
scale = 1.0;
ofs1 = 0.0; ofs2 = 0.0; ofs3 = 0.0;
}
if (motor == 1) {
switch(phase) {
case 1: return lbm_enc_float(((float)GET_CURRENT1() - ofs1) * scale);
case 2: return lbm_enc_float(((float)GET_CURRENT2() - ofs2) * scale);
case 3: return lbm_enc_float(((float)GET_CURRENT3() - ofs3) * scale);
default: return ENC_SYM_EERROR;
2022-02-06 14:12:05 -08:00
}
} else if (motor == 2) {
#ifdef HW_HAS_DUAL_MOTORS
switch(phase) {
case 1: return lbm_enc_float(((float)GET_CURRENT1_M2() - ofs1) * scale);
case 2: return lbm_enc_float(((float)GET_CURRENT2_M2() - ofs2) * scale);
case 3: return lbm_enc_float(((float)GET_CURRENT3_M2() - ofs3) * scale);
default: return ENC_SYM_EERROR;
}
2022-02-06 14:12:05 -08:00
#else
return ENC_SYM_EERROR;
2022-02-06 14:12:05 -08:00
#endif
} else {
return ENC_SYM_EERROR;
2022-02-06 14:12:05 -08:00
}
}
/*
* args[0]: Motor, 1 or 2
* args[1]: Phase, 1, 2 or 3
* args[2]: Use raw ADC values. Optional argument.
*/
static lbm_value ext_raw_adc_voltage(lbm_value *args, lbm_uint argn) {
CHECK_NUMBER_ALL();
if (argn != 2 && argn != 3) {
return ENC_SYM_EERROR;
2022-02-06 14:12:05 -08:00
}
uint32_t motor = lbm_dec_as_i32(args[0]);
uint32_t phase = lbm_dec_as_i32(args[1]);
2022-02-06 14:12:05 -08:00
float ofs1, ofs2, ofs3;
mcpwm_foc_get_voltage_offsets(&ofs1, &ofs2, &ofs3, motor == 2);
float scale = ((VIN_R1 + VIN_R2) / VIN_R2) * ADC_VOLTS_PH_FACTOR;
if (argn == 3 && lbm_dec_as_i32(args[2]) != 0) {
2022-02-06 14:12:05 -08:00
scale = 4095.0 / V_REG;
ofs1 = 0.0; ofs2 = 0.0; ofs3 = 0.0;
}
float Va = 0.0, Vb = 0.0, Vc = 0.0;
if (motor == 2) {
#ifdef HW_HAS_DUAL_MOTORS
Va = (ADC_VOLTS(ADC_IND_SENS4) - ofs1) * scale;
Vb = (ADC_VOLTS(ADC_IND_SENS5) - ofs2) * scale;
Vc = (ADC_VOLTS(ADC_IND_SENS6) - ofs3) * scale;
#else
return ENC_SYM_EERROR;
2022-02-06 14:12:05 -08:00
#endif
} else if (motor == 1) {
Va = (ADC_VOLTS(ADC_IND_SENS1) - ofs1) * scale;
Vb = (ADC_VOLTS(ADC_IND_SENS2) - ofs2) * scale;
Vc = (ADC_VOLTS(ADC_IND_SENS3) - ofs3) * scale;
} else {
return ENC_SYM_EERROR;
2022-02-06 14:12:05 -08:00
}
switch(phase) {
case 1: return lbm_enc_float(Va);
case 2: return lbm_enc_float(Vb);
case 3: return lbm_enc_float(Vc);
default: return ENC_SYM_EERROR;
2022-02-06 14:12:05 -08:00
}
}
static lbm_value ext_raw_mod_alpha(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
return lbm_enc_float(mcpwm_foc_get_mod_alpha_raw());
2022-02-06 14:12:05 -08:00
}
static lbm_value ext_raw_mod_beta(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
return lbm_enc_float(mcpwm_foc_get_mod_beta_raw());
2022-02-06 14:12:05 -08:00
}
static lbm_value ext_raw_mod_alpha_measured(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
return lbm_enc_float(mcpwm_foc_get_mod_alpha_measured());
2022-02-06 14:12:05 -08:00
}
static lbm_value ext_raw_mod_beta_measured(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
return lbm_enc_float(mcpwm_foc_get_mod_beta_measured());
2022-02-06 14:12:05 -08:00
}
static lbm_value ext_raw_hall(lbm_value *args, lbm_uint argn) {
CHECK_NUMBER_ALL();
if (argn != 1 && argn != 2) {
return ENC_SYM_EERROR;
2022-02-06 14:12:05 -08:00
}
int motor = lbm_dec_i(args[0]);
int samples = mc_interface_get_configuration()->m_hall_extra_samples;
if (argn == 2) {
2022-03-31 12:16:44 -07:00
samples = lbm_dec_i(args[1]);
2022-02-06 14:12:05 -08:00
}
if ((motor != 1 && motor != 2) || samples < 0 || samples > 20) {
return ENC_SYM_EERROR;
2022-02-06 14:12:05 -08:00
}
int hall = utils_read_hall(motor == 2, samples);
lbm_value hall_list = ENC_SYM_NIL;
2022-02-06 14:12:05 -08:00
hall_list = lbm_cons(lbm_enc_i((hall >> 2) & 1), hall_list);
hall_list = lbm_cons(lbm_enc_i((hall >> 1) & 1), hall_list);
hall_list = lbm_cons(lbm_enc_i((hall >> 0) & 1), hall_list);
return hall_list;
}
2022-02-21 17:37:23 -08:00
// UART
static SerialConfig uart_cfg = {
2022-04-28 04:42:14 -07:00
2500000, 0, 0, 0
2022-02-21 17:37:23 -08:00
};
2022-04-28 04:42:14 -07:00
2022-02-21 17:37:23 -08:00
static bool uart_started = false;
static lbm_value ext_uart_start(lbm_value *args, lbm_uint argn) {
2022-04-28 09:18:48 -07:00
if ((argn != 1 && argn != 2) || !lbm_is_number(args[0])) {
return ENC_SYM_EERROR;
2022-04-28 04:42:14 -07:00
}
2022-02-21 17:37:23 -08:00
int baud = lbm_dec_as_i32(args[0]);
2022-02-21 17:37:23 -08:00
if (baud < 10 || baud > 10000000) {
return ENC_SYM_EERROR;
2022-02-21 17:37:23 -08:00
}
2022-04-28 04:42:14 -07:00
bool half_duplex = false;
if (argn == 2) {
if (lbm_is_symbol(args[1])) {
if (compare_symbol(lbm_dec_sym(args[1]), &syms_vesc.half_duplex)) {
half_duplex = true;
} else {
return ENC_SYM_EERROR;
2022-04-28 04:42:14 -07:00
}
} else {
return ENC_SYM_TERROR;
2022-04-28 04:42:14 -07:00
}
}
2022-02-21 17:37:23 -08:00
app_configuration *appconf = mempools_alloc_appconf();
conf_general_read_app_configuration(appconf);
if (appconf->app_to_use == APP_UART ||
appconf->app_to_use == APP_PPM_UART ||
appconf->app_to_use == APP_ADC_UART) {
appconf->app_to_use = APP_NONE;
conf_general_store_app_configuration(appconf);
app_set_configuration(appconf);
}
mempools_free_appconf(appconf);
uart_cfg.speed = baud;
2022-04-28 04:42:14 -07:00
uart_cfg.cr3 = half_duplex ? USART_CR3_HDSEL : 0;
2022-02-21 17:37:23 -08:00
sdStop(&HW_UART_DEV);
sdStart(&HW_UART_DEV, &uart_cfg);
2022-04-28 04:42:14 -07:00
2022-02-21 17:37:23 -08:00
palSetPadMode(HW_UART_TX_PORT, HW_UART_TX_PIN, PAL_MODE_ALTERNATE(HW_UART_GPIO_AF));
2022-04-28 04:42:14 -07:00
if (!half_duplex) {
palSetPadMode(HW_UART_RX_PORT, HW_UART_RX_PIN, PAL_MODE_ALTERNATE(HW_UART_GPIO_AF));
}
2022-02-21 17:37:23 -08:00
uart_started = true;
return ENC_SYM_TRUE;
2022-02-21 17:37:23 -08:00
}
static void wait_uart_tx_task(void *arg) {
(void)arg;
while(!chOQIsEmptyI(&HW_UART_DEV.oqueue)){
chThdSleepMilliseconds(1);
}
chThdSleepMilliseconds(1);
HW_UART_DEV.usart->CR1 |= USART_CR1_RE;
}
2022-02-21 17:37:23 -08:00
static lbm_value ext_uart_write(lbm_value *args, lbm_uint argn) {
if (argn != 1 || (!lbm_is_cons(args[0]) && !lbm_is_array(args[0]))) {
return ENC_SYM_EERROR;
2022-02-21 17:37:23 -08:00
}
if (!uart_started) {
return ENC_SYM_EERROR;
2022-02-21 17:37:23 -08:00
}
const int max_len = 20;
uint8_t to_send[max_len];
uint8_t *to_send_ptr = to_send;
int ind = 0;
if (lbm_type_of(args[0]) == LBM_TYPE_ARRAY) {
2022-02-21 17:37:23 -08:00
lbm_array_header_t *array = (lbm_array_header_t *)lbm_car(args[0]);
if (array->elt_type != LBM_TYPE_BYTE) {
return ENC_SYM_EERROR;
2022-02-21 17:37:23 -08:00
}
to_send_ptr = (uint8_t*)array->data;
ind = array->size;
} else {
2022-02-23 16:19:09 -08:00
lbm_value curr = args[0];
while (lbm_is_cons(curr)) {
2022-02-21 17:37:23 -08:00
lbm_value arg = lbm_car(curr);
if (lbm_is_number(arg)) {
to_send[ind++] = lbm_dec_as_u32(arg);
2022-02-21 17:37:23 -08:00
} else {
return ENC_SYM_EERROR;
2022-02-21 17:37:23 -08:00
}
if (ind == max_len) {
break;
}
curr = lbm_cdr(curr);
}
}
if (uart_cfg.cr3 & USART_CR3_HDSEL) {
HW_UART_DEV.usart->CR1 &= ~USART_CR1_RE;
sdWrite(&HW_UART_DEV, to_send_ptr, ind);
worker_execute(wait_uart_tx_task, 0);
} else{
sdWrite(&HW_UART_DEV, to_send_ptr, ind);
}
2022-02-21 17:37:23 -08:00
return ENC_SYM_TRUE;
2022-02-21 17:37:23 -08:00
}
static lbm_value ext_uart_read(lbm_value *args, lbm_uint argn) {
if ((argn != 2 && argn != 3 && argn != 4) ||
lbm_type_of(args[0]) != LBM_TYPE_ARRAY || !lbm_is_number(args[1])) {
return ENC_SYM_EERROR;
2022-02-21 17:37:23 -08:00
}
unsigned int num = lbm_dec_as_u32(args[1]);
2022-02-21 17:37:23 -08:00
if (num > 512) {
return ENC_SYM_EERROR;
2022-02-21 17:37:23 -08:00
}
if (num == 0 || !uart_started) {
2022-02-21 17:37:23 -08:00
return lbm_enc_i(0);
}
unsigned int offset = 0;
if (argn >= 3) {
if (!lbm_is_number(args[2])) {
return ENC_SYM_EERROR;
2022-02-21 17:37:23 -08:00
}
offset = lbm_dec_as_u32(args[2]);
2022-02-21 17:37:23 -08:00
}
int stop_at = -1;
if (argn >= 4) {
if (!lbm_is_number(args[3])) {
return ENC_SYM_EERROR;
2022-02-21 17:37:23 -08:00
}
stop_at = lbm_dec_as_u32(args[3]);
2022-02-21 17:37:23 -08:00
}
lbm_array_header_t *array = (lbm_array_header_t *)lbm_car(args[0]);
if (array->elt_type != LBM_TYPE_BYTE || array->size < (num + offset)) {
return ENC_SYM_EERROR;
2022-02-21 17:37:23 -08:00
}
unsigned int count = 0;
msg_t res = sdGetTimeout(&HW_UART_DEV, TIME_IMMEDIATE);
while (res != MSG_TIMEOUT) {
((uint8_t*)array->data)[offset + count] = (uint8_t)res;
count++;
if (res == stop_at || count >= num) {
break;
}
res = sdGetTimeout(&HW_UART_DEV, TIME_IMMEDIATE);
}
return lbm_enc_i(count);
}
2022-02-23 16:19:09 -08:00
static i2c_bb_state i2c_cfg = {
HW_UART_RX_PORT, HW_UART_RX_PIN,
HW_UART_TX_PORT, HW_UART_TX_PIN,
2022-04-19 08:21:19 -07:00
I2C_BB_RATE_400K,
2022-02-23 16:19:09 -08:00
0,
0,
{{NULL, NULL}, NULL, NULL}
};
static bool i2c_started = false;
static lbm_value ext_i2c_start(lbm_value *args, lbm_uint argn) {
2022-04-19 14:44:43 -07:00
if (argn > 3) {
return ENC_SYM_EERROR;
2022-04-19 08:21:19 -07:00
}
i2c_cfg.rate = I2C_BB_RATE_200K;
2022-04-19 14:44:43 -07:00
if (argn >= 1) {
2022-04-19 08:21:19 -07:00
if (!lbm_is_symbol(args[0])) {
return ENC_SYM_EERROR;
2022-04-19 08:21:19 -07:00
}
if (compare_symbol(lbm_dec_sym(args[0]), &syms_vesc.rate_100k)) {
i2c_cfg.rate = I2C_BB_RATE_100K;
} else if (compare_symbol(lbm_dec_sym(args[0]), &syms_vesc.rate_200k)) {
i2c_cfg.rate = I2C_BB_RATE_200K;
} else if (compare_symbol(lbm_dec_sym(args[0]), &syms_vesc.rate_400k)) {
i2c_cfg.rate = I2C_BB_RATE_400K;
} else if (compare_symbol(lbm_dec_sym(args[0]), &syms_vesc.rate_700k)) {
i2c_cfg.rate = I2C_BB_RATE_700K;
} else {
return ENC_SYM_EERROR;
2022-04-19 08:21:19 -07:00
}
}
2022-02-23 16:19:09 -08:00
2022-04-19 14:44:43 -07:00
stm32_gpio_t *sda_gpio = HW_UART_RX_PORT;
2022-11-10 11:36:46 -08:00
uint32_t sda_pin = HW_UART_RX_PIN;
2022-04-19 14:44:43 -07:00
if (argn >= 2) {
if (!lbm_is_symbol(args[1]) ||
2022-11-10 11:36:46 -08:00
!lispif_symbol_to_io(lbm_dec_sym(args[1]), &sda_gpio, &sda_pin)) {
return ENC_SYM_EERROR;
2022-04-19 14:44:43 -07:00
}
}
i2c_cfg.sda_gpio = sda_gpio;
i2c_cfg.sda_pin = sda_pin;
stm32_gpio_t *scl_gpio = HW_UART_TX_PORT;
2022-11-10 11:36:46 -08:00
uint32_t scl_pin = HW_UART_TX_PIN;
2022-04-19 14:44:43 -07:00
if (argn >= 3) {
if (!lbm_is_symbol(args[2]) ||
2022-11-10 11:36:46 -08:00
!lispif_symbol_to_io(lbm_dec_sym(args[2]), &scl_gpio, &scl_pin)) {
return ENC_SYM_EERROR;
2022-04-19 14:44:43 -07:00
}
}
i2c_cfg.scl_gpio = scl_gpio;
i2c_cfg.scl_pin = scl_pin;
bool is_using_uart_pins =
(sda_gpio == HW_UART_TX_PORT && sda_pin == HW_UART_TX_PIN) ||
(scl_gpio == HW_UART_TX_PORT && scl_pin == HW_UART_TX_PIN) ||
(sda_gpio == HW_UART_RX_PORT && sda_pin == HW_UART_RX_PIN) ||
(scl_gpio == HW_UART_RX_PORT && scl_pin == HW_UART_RX_PIN);
2022-05-12 07:14:16 -07:00
if (is_using_uart_pins) {
app_configuration *appconf = mempools_alloc_appconf();
conf_general_read_app_configuration(appconf);
if (appconf->app_to_use == APP_UART ||
appconf->app_to_use == APP_PPM_UART ||
appconf->app_to_use == APP_ADC_UART) {
appconf->app_to_use = APP_NONE;
conf_general_store_app_configuration(appconf);
app_set_configuration(appconf);
}
mempools_free_appconf(appconf);
2022-02-23 16:19:09 -08:00
}
i2c_bb_init(&i2c_cfg);
i2c_started = true;
return ENC_SYM_TRUE;
2022-02-23 16:19:09 -08:00
}
static lbm_value ext_i2c_tx_rx(lbm_value *args, lbm_uint argn) {
if (argn != 2 && argn != 3) {
return ENC_SYM_EERROR;
2022-02-23 16:19:09 -08:00
}
if (!i2c_started) {
return lbm_enc_i(0);
}
uint16_t addr = 0;
size_t txlen = 0;
size_t rxlen = 0;
uint8_t *txbuf = 0;
uint8_t *rxbuf = 0;
const unsigned int max_len = 20;
uint8_t to_send[max_len];
if (!lbm_is_number(args[0])) {
return ENC_SYM_EERROR;
2022-02-23 16:19:09 -08:00
}
addr = lbm_dec_as_u32(args[0]);
2022-02-23 16:19:09 -08:00
if (lbm_type_of(args[1]) == LBM_TYPE_ARRAY) {
2022-02-23 16:19:09 -08:00
lbm_array_header_t *array = (lbm_array_header_t *)lbm_car(args[1]);
if (array->elt_type != LBM_TYPE_BYTE) {
return ENC_SYM_EERROR;
2022-02-23 16:19:09 -08:00
}
txbuf = (uint8_t*)array->data;
txlen = array->size;
} else {
lbm_value curr = args[1];
while (lbm_is_cons(curr)) {
2022-02-23 16:19:09 -08:00
lbm_value arg = lbm_car(curr);
if (lbm_is_number(arg)) {
to_send[txlen++] = lbm_dec_as_u32(arg);
2022-02-23 16:19:09 -08:00
} else {
return ENC_SYM_EERROR;
2022-02-23 16:19:09 -08:00
}
if (txlen == max_len) {
break;
}
curr = lbm_cdr(curr);
}
if (txlen > 0) {
txbuf = to_send;
}
}
if (argn >= 3 && lbm_type_of(args[2]) == LBM_TYPE_ARRAY) {
2022-02-23 16:19:09 -08:00
lbm_array_header_t *array = (lbm_array_header_t *)lbm_car(args[2]);
if (array->elt_type != LBM_TYPE_BYTE) {
return ENC_SYM_EERROR;
2022-02-23 16:19:09 -08:00
}
rxbuf = (uint8_t*)array->data;
rxlen = array->size;
}
return lbm_enc_i(i2c_bb_tx_rx(&i2c_cfg, addr, txbuf, txlen, rxbuf, rxlen) ? 1 : 0);
}
static lbm_value ext_i2c_restore(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
if (!i2c_started) {
return lbm_enc_i(0);
}
i2c_bb_restore_bus(&i2c_cfg);
return lbm_enc_i(1);
}
2022-03-13 14:53:25 -07:00
static lbm_value ext_gpio_configure(lbm_value *args, lbm_uint argn) {
CHECK_ARGN(2);
if (!lbm_is_symbol(args[0]) || !lbm_is_symbol(args[1])) {
return ENC_SYM_EERROR;
2022-03-13 14:53:25 -07:00
}
lbm_uint name = lbm_dec_sym(args[1]);
iomode_t mode = PAL_MODE_OUTPUT_PUSHPULL;
2022-03-31 06:18:59 -07:00
if (compare_symbol(name, &syms_vesc.pin_mode_out)) {
2022-03-13 14:53:25 -07:00
mode = PAL_MODE_OUTPUT_PUSHPULL;
2022-03-31 06:18:59 -07:00
} else if (compare_symbol(name, &syms_vesc.pin_mode_od)) {
2022-03-13 14:53:25 -07:00
mode = PAL_MODE_OUTPUT_OPENDRAIN;
2022-05-13 01:14:37 -07:00
} else if (compare_symbol(name, &syms_vesc.pin_mode_od_pu)) {
mode = PAL_MODE_OUTPUT_OPENDRAIN | PAL_STM32_PUDR_PULLUP;
} else if (compare_symbol(name, &syms_vesc.pin_mode_od_pd)) {
mode = PAL_MODE_OUTPUT_OPENDRAIN | PAL_STM32_PUDR_PULLDOWN;
2022-03-31 06:18:59 -07:00
} else if (compare_symbol(name, &syms_vesc.pin_mode_in)) {
2022-03-13 14:53:25 -07:00
mode = PAL_MODE_INPUT;
2022-03-31 06:18:59 -07:00
} else if (compare_symbol(name, &syms_vesc.pin_mode_in_pu)) {
2022-03-13 14:53:25 -07:00
mode = PAL_MODE_INPUT_PULLUP;
2022-03-31 06:18:59 -07:00
} else if (compare_symbol(name, &syms_vesc.pin_mode_in_pd)) {
2022-03-13 14:53:25 -07:00
mode = PAL_MODE_INPUT_PULLDOWN;
2022-05-13 01:14:37 -07:00
} else if (compare_symbol(name, &syms_vesc.pin_mode_analog)) {
mode = PAL_STM32_MODE_ANALOG;
2022-03-13 14:53:25 -07:00
} else {
return ENC_SYM_EERROR;
2022-03-13 14:53:25 -07:00
}
2022-11-10 11:36:46 -08:00
stm32_gpio_t *port; uint32_t pin;
if (lispif_symbol_to_io(lbm_dec_sym(args[0]), &port, &pin)) {
2022-03-13 14:53:25 -07:00
palSetPadMode(port, pin, mode);
} else {
return ENC_SYM_EERROR;
2022-03-13 14:53:25 -07:00
}
return ENC_SYM_TRUE;
2022-03-13 14:53:25 -07:00
}
static lbm_value ext_gpio_write(lbm_value *args, lbm_uint argn) {
CHECK_ARGN(2);
if (!lbm_is_symbol(args[0]) || !lbm_is_number(args[1])) {
return ENC_SYM_EERROR;
2022-03-13 14:53:25 -07:00
}
2022-11-10 11:36:46 -08:00
stm32_gpio_t *port; uint32_t pin;
if (lispif_symbol_to_io(lbm_dec_sym(args[0]), &port, &pin)) {
palWritePad(port, pin, lbm_dec_as_i32(args[1]));
2022-03-13 14:53:25 -07:00
} else {
return ENC_SYM_EERROR;
2022-03-13 14:53:25 -07:00
}
return ENC_SYM_TRUE;
2022-03-13 14:53:25 -07:00
}
static lbm_value ext_gpio_read(lbm_value *args, lbm_uint argn) {
CHECK_ARGN(1);
if (!lbm_is_symbol(args[0])) {
return ENC_SYM_EERROR;
2022-03-13 14:53:25 -07:00
}
2022-11-10 11:36:46 -08:00
stm32_gpio_t *port; uint32_t pin;
if (lispif_symbol_to_io(lbm_dec_sym(args[0]), &port, &pin)) {
2022-03-13 14:53:25 -07:00
return lbm_enc_i(palReadPad(port, pin));
} else {
return ENC_SYM_EERROR;
2022-03-13 14:53:25 -07:00
}
}
2022-03-07 01:26:51 -08:00
static lbm_value ext_str_from_n(lbm_value *args, lbm_uint argn) {
if ((argn != 1 && argn != 2) || !lbm_is_number(args[0])) {
return ENC_SYM_EERROR;
2022-03-07 01:26:51 -08:00
}
if (argn == 2 && lbm_type_of(args[1]) != LBM_TYPE_ARRAY) {
return ENC_SYM_EERROR;
2022-03-07 01:26:51 -08:00
}
char *format = 0;
if (argn == 2) {
format = lbm_dec_str(args[1]);
}
char buffer[100];
size_t len = 0;
switch (lbm_type_of(args[0])) {
case LBM_TYPE_FLOAT:
2022-03-07 01:26:51 -08:00
if (!format) {
format = "%g";
2022-03-07 01:26:51 -08:00
}
len = snprintf(buffer, sizeof(buffer), format, (double)lbm_dec_as_float(args[0]));
2022-03-07 01:26:51 -08:00
break;
default:
if (!format) {
format = "%d";
}
len = snprintf(buffer, sizeof(buffer), format, lbm_dec_as_i32(args[0]));
2022-03-07 01:26:51 -08:00
break;
}
if (len > sizeof(buffer)) {
len = sizeof(buffer);
}
lbm_value res;
if (lbm_create_array(&res, LBM_TYPE_CHAR, len + 1)) {
2022-03-07 01:26:51 -08:00
lbm_array_header_t *arr = (lbm_array_header_t*)lbm_car(res);
memcpy(arr->data, buffer, len);
((char*)(arr->data))[len] = '\0';
return res;
} else {
return ENC_SYM_MERROR;
2022-03-07 01:26:51 -08:00
}
}
static lbm_value ext_str_merge(lbm_value *args, lbm_uint argn) {
int len_tot = 0;
for (unsigned int i = 0;i < argn;i++) {
char *str = lbm_dec_str(args[i]);
if (str) {
len_tot += strlen(str);
} else {
return ENC_SYM_EERROR;
2022-03-07 01:26:51 -08:00
}
}
lbm_value res;
if (lbm_create_array(&res, LBM_TYPE_CHAR, len_tot + 1)) {
2022-03-07 01:26:51 -08:00
lbm_array_header_t *arr = (lbm_array_header_t*)lbm_car(res);
unsigned int offset = 0;
for (unsigned int i = 0;i < argn;i++) {
offset += sprintf((char*)arr->data + offset, "%s", lbm_dec_str(args[i]));
}
((char*)(arr->data))[len_tot] = '\0';
return res;
} else {
return ENC_SYM_MERROR;
2022-03-07 01:26:51 -08:00
}
}
static lbm_value ext_str_to_i(lbm_value *args, lbm_uint argn) {
if (argn != 1 && argn != 2) {
return ENC_SYM_EERROR;
2022-03-07 01:26:51 -08:00
}
char *str = lbm_dec_str(args[0]);
if (!str) {
return ENC_SYM_EERROR;
2022-03-07 01:26:51 -08:00
}
int base = 0;
if (argn == 2) {
if (!lbm_is_number(args[1])) {
return ENC_SYM_EERROR;
2022-03-07 01:26:51 -08:00
}
base = lbm_dec_as_u32(args[1]);
2022-03-07 01:26:51 -08:00
}
return lbm_enc_i32(strtol(str, NULL, base));
2022-03-07 01:26:51 -08:00
}
static lbm_value ext_str_to_f(lbm_value *args, lbm_uint argn) {
if (argn != 1) {
return ENC_SYM_EERROR;
2022-03-07 01:26:51 -08:00
}
char *str = lbm_dec_str(args[0]);
if (!str) {
return ENC_SYM_EERROR;
2022-03-07 01:26:51 -08:00
}
return lbm_enc_float(strtof(str, NULL));
2022-03-07 01:26:51 -08:00
}
static lbm_value ext_str_part(lbm_value *args, lbm_uint argn) {
if ((argn != 2 && argn != 3) || !lbm_is_number(args[1])) {
return ENC_SYM_EERROR;
2022-03-07 01:26:51 -08:00
}
char *str = lbm_dec_str(args[0]);
if (!str) {
return ENC_SYM_EERROR;
2022-03-07 01:26:51 -08:00
}
size_t len = strlen(str);
unsigned int start = lbm_dec_as_u32(args[1]);
2022-03-07 01:26:51 -08:00
if (start >= len) {
return ENC_SYM_EERROR;
2022-03-07 01:26:51 -08:00
}
unsigned int n = len - start;
if (argn == 3) {
if (!lbm_is_number(args[2])) {
return ENC_SYM_EERROR;
2022-03-07 01:26:51 -08:00
}
n = MIN(lbm_dec_as_u32(args[2]), n);
2022-03-07 01:26:51 -08:00
}
lbm_value res;
if (lbm_create_array(&res, LBM_TYPE_CHAR, n + 1)) {
2022-03-07 01:26:51 -08:00
lbm_array_header_t *arr = (lbm_array_header_t*)lbm_car(res);
memcpy(arr->data, str + start, n);
((char*)(arr->data))[n] = '\0';
return res;
} else {
return ENC_SYM_MERROR;
2022-03-07 01:26:51 -08:00
}
}
static lbm_value ext_str_split(lbm_value *args, lbm_uint argn) {
if (argn != 2) {
return ENC_SYM_EERROR;
2022-03-07 01:26:51 -08:00
}
char *str = lbm_dec_str(args[0]);
if (!str) {
return ENC_SYM_EERROR;
2022-03-07 01:26:51 -08:00
}
char *split = lbm_dec_str(args[1]);
int step = 0;
if (!split) {
if (lbm_is_number(args[1])) {
step = MAX(lbm_dec_as_i32(args[1]), 1);
2022-03-07 01:26:51 -08:00
} else {
return ENC_SYM_EERROR;
2022-03-07 01:26:51 -08:00
}
}
if (step > 0) {
lbm_value res = ENC_SYM_NIL;
2022-03-07 01:26:51 -08:00
int len = strlen(str);
for (int i = len / step;i >= 0;i--) {
int ind_now = i * step;
if (ind_now >= len) {
continue;
}
int step_now = step;
while ((ind_now + step_now) > len) {
step_now--;
}
lbm_value tok;
if (lbm_create_array(&tok, LBM_TYPE_CHAR, step_now + 1)) {
2022-03-07 01:26:51 -08:00
lbm_array_header_t *arr = (lbm_array_header_t*)lbm_car(tok);
memcpy(arr->data, str + ind_now, step_now);
((char*)(arr->data))[step_now] = '\0';
res = lbm_cons(tok, res);
} else {
return ENC_SYM_MERROR;
2022-03-07 01:26:51 -08:00
}
}
return res;
} else {
lbm_value res = ENC_SYM_NIL;
const char *s = str;
while (*(s += strspn(s, split)) != '\0') {
size_t len = strcspn(s, split);
2022-03-07 01:26:51 -08:00
lbm_value tok;
if (lbm_create_array(&tok, LBM_TYPE_CHAR, len + 1)) {
2022-03-07 01:26:51 -08:00
lbm_array_header_t *arr = (lbm_array_header_t*)lbm_car(tok);
memcpy(arr->data, s, len);
2022-03-07 01:26:51 -08:00
((char*)(arr->data))[len] = '\0';
res = lbm_cons(tok, res);
} else {
return ENC_SYM_MERROR;
2022-03-07 01:26:51 -08:00
}
s += len;
2022-03-07 01:26:51 -08:00
}
return lbm_list_destructive_reverse(res);
2022-03-07 01:26:51 -08:00
}
}
static lbm_value ext_str_replace(lbm_value *args, lbm_uint argn) {
if (argn != 2 && argn != 3) {
return ENC_SYM_EERROR;
2022-03-07 01:26:51 -08:00
}
char *orig = lbm_dec_str(args[0]);
if (!orig) {
return ENC_SYM_TERROR;
2022-03-07 01:26:51 -08:00
}
char *rep = lbm_dec_str(args[1]);
if (!rep) {
return ENC_SYM_TERROR;
2022-03-07 01:26:51 -08:00
}
char *with = "";
if (argn == 3) {
with = lbm_dec_str(args[2]);
if (!with) {
return ENC_SYM_TERROR;
2022-03-07 01:26:51 -08:00
}
}
// See https://stackoverflow.com/questions/779875/what-function-is-to-replace-a-substring-from-a-string-in-c
char *result; // the return string
char *ins; // the next insert point
char *tmp; // varies
int len_rep; // length of rep (the string to remove)
int len_with; // length of with (the string to replace rep with)
int len_front; // distance between rep and end of last rep
int count; // number of replacements
len_rep = strlen(rep);
if (len_rep == 0) {
return args[0]; // empty rep causes infinite loop during count
}
len_with = strlen(with);
// count the number of replacements needed
ins = orig;
for (count = 0; (tmp = strstr(ins, rep)); ++count) {
ins = tmp + len_rep;
}
size_t len_res = strlen(orig) + (len_with - len_rep) * count + 1;
lbm_value lbm_res;
if (lbm_create_array(&lbm_res, LBM_TYPE_CHAR, len_res)) {
2022-03-07 01:26:51 -08:00
lbm_array_header_t *arr = (lbm_array_header_t*)lbm_car(lbm_res);
tmp = result = (char*)arr->data;
} else {
return ENC_SYM_MERROR;
2022-03-07 01:26:51 -08:00
}
// first time through the loop, all the variable are set correctly
// from here on,
// tmp points to the end of the result string
// ins points to the next occurrence of rep in orig
// orig points to the remainder of orig after "end of rep"
while (count--) {
ins = strstr(orig, rep);
len_front = ins - orig;
tmp = strncpy(tmp, orig, len_front) + len_front;
tmp = strcpy(tmp, with) + len_with;
orig += len_front + len_rep; // move to next "end of rep"
}
strcpy(tmp, orig);
return lbm_res;
}
static lbm_value ext_str_to_lower(lbm_value *args, lbm_uint argn) {
if (argn != 1) {
return ENC_SYM_EERROR;
2022-03-07 01:26:51 -08:00
}
char *orig = lbm_dec_str(args[0]);
if (!orig) {
return ENC_SYM_TERROR;
2022-03-07 01:26:51 -08:00
}
int len = strlen(orig);
lbm_value lbm_res;
if (lbm_create_array(&lbm_res, LBM_TYPE_CHAR, len + 1)) {
2022-03-07 01:26:51 -08:00
lbm_array_header_t *arr = (lbm_array_header_t*)lbm_car(lbm_res);
for (int i = 0;i < len;i++) {
((char*)(arr->data))[i] = tolower(orig[i]);
}
((char*)(arr->data))[len] = '\0';
return lbm_res;
} else {
return ENC_SYM_MERROR;
2022-03-07 01:26:51 -08:00
}
}
static lbm_value ext_str_to_upper(lbm_value *args, lbm_uint argn) {
if (argn != 1) {
return ENC_SYM_EERROR;
2022-03-07 01:26:51 -08:00
}
char *orig = lbm_dec_str(args[0]);
if (!orig) {
return ENC_SYM_TERROR;
2022-03-07 01:26:51 -08:00
}
int len = strlen(orig);
lbm_value lbm_res;
if (lbm_create_array(&lbm_res, LBM_TYPE_CHAR, len + 1)) {
2022-03-07 01:26:51 -08:00
lbm_array_header_t *arr = (lbm_array_header_t*)lbm_car(lbm_res);
for (int i = 0;i < len;i++) {
((char*)(arr->data))[i] = toupper(orig[i]);
}
((char*)(arr->data))[len] = '\0';
return lbm_res;
} else {
return ENC_SYM_MERROR;
2022-03-07 01:26:51 -08:00
}
}
static lbm_value ext_str_cmp(lbm_value *args, lbm_uint argn) {
if (argn != 2) {
return ENC_SYM_EERROR;
2022-03-07 01:26:51 -08:00
}
char *str1 = lbm_dec_str(args[0]);
if (!str1) {
return ENC_SYM_EERROR;
2022-03-07 01:26:51 -08:00
}
char *str2 = lbm_dec_str(args[1]);
if (!str2) {
return ENC_SYM_EERROR;
2022-03-07 01:26:51 -08:00
}
return lbm_enc_i(strcmp(str1, str2));
}
2022-11-09 07:00:43 -08:00
// TODO: This is very similar to ext-print. Maybe they can share code.
2022-11-10 04:01:13 -08:00
static lbm_value to_str(char *delimiter, lbm_value *args, lbm_uint argn) {
2022-11-09 07:00:43 -08:00
const int str_len = 300;
char *str = lispif_malloc(str_len);
if (!str) {
return ENC_SYM_MERROR;
}
int str_ofs = 0;
for (lbm_uint i = 0; i < argn; i ++) {
lbm_value t = args[i];
int max = str_len - str_ofs - 1;
if (lbm_is_ptr(t) && lbm_type_of(t) == LBM_TYPE_ARRAY) {
lbm_array_header_t *array = (lbm_array_header_t *)lbm_car(t);
switch (array->elt_type){
case LBM_TYPE_CHAR: {
int chars = 0;
if (str_ofs == 0) {
chars = snprintf(str + str_ofs, max, "%s", (char*)array->data);
} else {
2022-11-10 04:01:13 -08:00
chars = snprintf(str + str_ofs, max, "%s%s", delimiter, (char*)array->data);
2022-11-09 07:00:43 -08:00
}
if (chars >= max) {
str_ofs += max;
} else {
str_ofs += chars;
}
} break;
default:
return ENC_SYM_NIL;
break;
}
} else if (lbm_type_of(t) == LBM_TYPE_CHAR) {
int chars = 0;
if (str_ofs == 0) {
chars = snprintf(str + str_ofs, max, "%c", lbm_dec_char(t));
} else {
2022-11-10 04:01:13 -08:00
chars = snprintf(str + str_ofs, max, "%s%c", delimiter, lbm_dec_char(t));
2022-11-09 07:00:43 -08:00
}
if (chars >= max) {
str_ofs += max;
} else {
str_ofs += chars;
}
} else {
lbm_print_value(print_val_buffer, 256, t);
int chars = 0;
if (str_ofs == 0) {
chars = snprintf(str + str_ofs, max, "%s", print_val_buffer);
} else {
2022-11-10 04:01:13 -08:00
chars = snprintf(str + str_ofs, max, "%s%s", delimiter, print_val_buffer);
2022-11-09 07:00:43 -08:00
}
if (chars >= max) {
str_ofs += max;
} else {
str_ofs += chars;
}
}
}
lbm_value res;
if (lbm_create_array(&res, LBM_TYPE_CHAR, str_ofs + 1)) {
lbm_array_header_t *arr = (lbm_array_header_t*)lbm_car(res);
strncpy((char*)arr->data, str, str_ofs + 1);
return res;
} else {
lispif_free(str);
return ENC_SYM_MERROR;
}
}
2022-11-10 04:01:13 -08:00
static lbm_value ext_to_str(lbm_value *args, lbm_uint argn) {
return to_str(" ", args, argn);
}
static lbm_value ext_to_str_delim(lbm_value *args, lbm_uint argn) {
if (argn < 1) {
return ENC_SYM_EERROR;
}
char *delim = lbm_dec_str(args[0]);
if (!delim) {
return ENC_SYM_EERROR;
}
return to_str(delim, args + 1, argn - 1);
}
static lbm_value ext_str_len(lbm_value *args, lbm_uint argn) {
CHECK_ARGN(1);
char *str = lbm_dec_str(args[0]);
if (!str) {
return ENC_SYM_EERROR;
}
lbm_array_header_t *array = (lbm_array_header_t *)lbm_car(args[0]);
return lbm_enc_i(strnlen(str, array->size));
}
2022-03-31 10:37:36 -07:00
// Configuration
static lbm_value ext_conf_set(lbm_value *args, lbm_uint argn) {
lbm_value res = ENC_SYM_EERROR;
2022-03-31 10:37:36 -07:00
if (argn != 2) {
return res;
}
if (!lbm_is_symbol(args[0])) {
2022-03-31 10:37:36 -07:00
return res;
}
if (!lbm_is_number(args[1])) {
return res;
}
lbm_uint name = lbm_dec_sym(args[0]);
2022-04-12 05:09:16 -07:00
int changed_mc = 0;
int changed_app = 0;
2022-03-31 10:37:36 -07:00
2022-04-12 05:09:16 -07:00
mc_configuration *mcconf = (mc_configuration*)mc_interface_get_configuration();
app_configuration *appconf = (app_configuration*)app_get_configuration();
2022-03-31 10:37:36 -07:00
const float speed_fact = ((mcconf->si_motor_poles / 2.0) * 60.0 *
mcconf->si_gear_ratio) / (mcconf->si_wheel_diameter * M_PI);
2022-04-12 05:09:16 -07:00
// Safe changes that can be done instantly on the pointer. It is not that good to do
// it this way, but it is much faster.
// TODO: Check regularly and make sure that these stay safe.
2022-03-31 10:37:36 -07:00
if (compare_symbol(name, &syms_vesc.l_current_min)) {
mcconf->l_current_min = -fabsf(lbm_dec_as_float(args[1]));
2022-04-12 05:09:16 -07:00
changed_mc = 1;
2022-03-31 10:37:36 -07:00
} else if (compare_symbol(name, &syms_vesc.l_current_max)) {
mcconf->l_current_max = lbm_dec_as_float(args[1]);
2022-04-12 05:09:16 -07:00
changed_mc = 1;
2022-03-31 10:37:36 -07:00
} else if (compare_symbol(name, &syms_vesc.l_current_min_scale)) {
mcconf->l_current_min_scale = lbm_dec_as_float(args[1]);
2022-04-12 05:09:16 -07:00
changed_mc = 1;
2022-03-31 10:37:36 -07:00
} else if (compare_symbol(name, &syms_vesc.l_current_max_scale)) {
mcconf->l_current_max_scale = lbm_dec_as_float(args[1]);
2022-04-12 05:09:16 -07:00
changed_mc = 1;
2022-03-31 10:37:36 -07:00
} else if (compare_symbol(name, &syms_vesc.l_in_current_min)) {
mcconf->l_in_current_min = -fabsf(lbm_dec_as_float(args[1]));
2022-04-12 05:09:16 -07:00
changed_mc = 1;
2022-03-31 10:37:36 -07:00
} else if (compare_symbol(name, &syms_vesc.l_in_current_max)) {
mcconf->l_in_current_max = lbm_dec_as_float(args[1]);
2022-04-12 05:09:16 -07:00
changed_mc = 1;
2022-05-06 05:37:08 -07:00
} else if (compare_symbol(name, &syms_vesc.l_abs_current_max)) {
mcconf->l_abs_current_max = lbm_dec_as_float(args[1]);
changed_mc = 1;
2022-03-31 10:37:36 -07:00
} else if (compare_symbol(name, &syms_vesc.l_min_erpm)) {
mcconf->l_min_erpm = -fabsf(lbm_dec_as_float(args[1]));
2022-04-12 05:09:16 -07:00
changed_mc = 1;
2022-03-31 10:37:36 -07:00
} else if (compare_symbol(name, &syms_vesc.l_max_erpm)) {
mcconf->l_max_erpm = lbm_dec_as_float(args[1]);
2022-04-12 05:09:16 -07:00
changed_mc = 1;
} else if (compare_symbol(name, &syms_vesc.l_min_vin)) {
mcconf->l_min_vin = lbm_dec_as_float(args[1]);
changed_mc = 1;
} else if (compare_symbol(name, &syms_vesc.l_max_vin)) {
mcconf->l_max_vin = lbm_dec_as_float(args[1]);
changed_mc = 1;
} else if (compare_symbol(name, &syms_vesc.l_min_duty)) {
mcconf->l_min_duty = lbm_dec_as_float(args[1]);
changed_mc = 1;
} else if (compare_symbol(name, &syms_vesc.l_max_duty)) {
mcconf->l_max_duty = lbm_dec_as_float(args[1]);
changed_mc = 1;
2022-03-31 10:37:36 -07:00
} else if (compare_symbol(name, &syms_vesc.min_speed)) {
mcconf->l_min_erpm = -fabsf(lbm_dec_as_float(args[1])) * speed_fact;
2022-04-12 05:09:16 -07:00
changed_mc = 1;
2022-03-31 10:37:36 -07:00
} else if (compare_symbol(name, &syms_vesc.max_speed)) {
mcconf->l_max_erpm = lbm_dec_as_float(args[1]) * speed_fact;
2022-04-12 05:09:16 -07:00
changed_mc = 1;
} else if (compare_symbol(name, &syms_vesc.l_watt_min)) {
mcconf->l_watt_min = -fabsf(lbm_dec_as_float(args[1]));
changed_mc = 1;
} else if (compare_symbol(name, &syms_vesc.l_watt_max)) {
mcconf->l_watt_max = lbm_dec_as_float(args[1]);
changed_mc = 1;
2022-05-06 05:37:08 -07:00
} else if (compare_symbol(name, &syms_vesc.m_invert_direction)) {
mcconf->m_invert_direction = lbm_dec_as_i32(args[1]);
changed_mc = 1;
} else if (compare_symbol(name, &syms_vesc.m_out_aux_mode)) {
mcconf->m_out_aux_mode = lbm_dec_as_i32(args[1]);
changed_mc = 1;
2022-03-31 10:37:36 -07:00
} else if (compare_symbol(name, &syms_vesc.controller_id)) {
appconf->controller_id = lbm_dec_as_i32(args[1]);
2022-04-12 05:09:16 -07:00
changed_app = 1;
2022-03-31 10:37:36 -07:00
}
2022-04-12 05:09:16 -07:00
// Unsafe changes that require reconfiguration.
if (changed_mc == 0 && changed_app == 0) {
mcconf = mempools_alloc_mcconf();
*mcconf = *mc_interface_get_configuration();
appconf = mempools_alloc_appconf();
*appconf = *app_get_configuration();
2022-07-31 04:30:10 -07:00
if (compare_symbol(name, &syms_vesc.motor_type)) {
mcconf->motor_type = lbm_dec_as_i32(args[1]);
changed_mc = 2;
} else if (compare_symbol(name, &syms_vesc.foc_sensor_mode)) {
2022-05-06 05:37:08 -07:00
mcconf->foc_sensor_mode = lbm_dec_as_i32(args[1]);
changed_mc = 2;
} else if (compare_symbol(name, &syms_vesc.foc_current_kp)) {
2022-04-12 05:09:16 -07:00
mcconf->foc_current_kp = lbm_dec_as_float(args[1]);
changed_mc = 2;
} else if (compare_symbol(name, &syms_vesc.foc_current_ki)) {
mcconf->foc_current_ki = lbm_dec_as_float(args[1]);
changed_mc = 2;
} else if (compare_symbol(name, &syms_vesc.foc_motor_l)) {
mcconf->foc_motor_l = lbm_dec_as_float(args[1]) * 1e-6;
changed_mc = 2;
} else if (compare_symbol(name, &syms_vesc.foc_motor_ld_lq_diff)) {
mcconf->foc_motor_ld_lq_diff = lbm_dec_as_float(args[1]) * 1e-6;
changed_mc = 2;
} else if (compare_symbol(name, &syms_vesc.foc_motor_r)) {
mcconf->foc_motor_r = lbm_dec_as_float(args[1]) * 1e-3;
changed_mc = 2;
} else if (compare_symbol(name, &syms_vesc.foc_motor_flux_linkage)) {
mcconf->foc_motor_flux_linkage = lbm_dec_as_float(args[1]) * 1e-3;
changed_mc = 2;
} else if (compare_symbol(name, &syms_vesc.foc_observer_gain)) {
mcconf->foc_observer_gain = lbm_dec_as_float(args[1]) * 1e6;
changed_mc = 2;
2022-05-06 05:37:08 -07:00
} else if (compare_symbol(name, &syms_vesc.foc_hfi_voltage_start)) {
mcconf->foc_hfi_voltage_start = lbm_dec_as_float(args[1]);
changed_mc = 2;
} else if (compare_symbol(name, &syms_vesc.foc_hfi_voltage_run)) {
mcconf->foc_hfi_voltage_run = lbm_dec_as_float(args[1]);
changed_mc = 2;
} else if (compare_symbol(name, &syms_vesc.foc_hfi_voltage_max)) {
mcconf->foc_hfi_voltage_max = lbm_dec_as_float(args[1]);
changed_mc = 2;
} else if (compare_symbol(name, &syms_vesc.foc_sl_erpm_hfi)) {
mcconf->foc_sl_erpm_hfi = lbm_dec_as_float(args[1]);
changed_mc = 2;
} else if (compare_symbol(name, &syms_vesc.app_to_use)) {
appconf->app_to_use = lbm_dec_as_i32(args[1]);
changed_app = 2;
2022-05-06 05:37:08 -07:00
} else if (compare_symbol(name, &syms_vesc.ppm_ctrl_type)) {
appconf->app_ppm_conf.ctrl_type = lbm_dec_as_i32(args[1]);
changed_app = 2;
} else if (compare_symbol(name, &syms_vesc.ppm_pulse_start)) {
appconf->app_ppm_conf.pulse_start = lbm_dec_as_float(args[1]);
changed_app = 2;
} else if (compare_symbol(name, &syms_vesc.ppm_pulse_end)) {
appconf->app_ppm_conf.pulse_end = lbm_dec_as_float(args[1]);
changed_app = 2;
} else if (compare_symbol(name, &syms_vesc.ppm_pulse_center)) {
appconf->app_ppm_conf.pulse_center = lbm_dec_as_float(args[1]);
changed_app = 2;
} else if (compare_symbol(name, &syms_vesc.ppm_ramp_time_pos)) {
appconf->app_ppm_conf.ramp_time_pos = lbm_dec_as_float(args[1]);
changed_app = 2;
} else if (compare_symbol(name, &syms_vesc.ppm_ramp_time_neg)) {
appconf->app_ppm_conf.ramp_time_neg = lbm_dec_as_float(args[1]);
changed_app = 2;
2022-04-12 05:09:16 -07:00
}
}
if (changed_mc > 0) {
2022-04-12 06:01:57 -07:00
commands_apply_mcconf_hw_limits(mcconf);
2022-04-12 05:09:16 -07:00
if (changed_mc == 2) {
mc_interface_set_configuration(mcconf);
}
res = ENC_SYM_TRUE;
2022-04-12 05:09:16 -07:00
} else if (changed_app > 0) {
if (changed_app == 2) {
app_set_configuration(appconf);
}
res = ENC_SYM_TRUE;
2022-03-31 10:37:36 -07:00
} else {
lbm_set_error_reason("Parameter not recognized");
}
2022-04-12 05:09:16 -07:00
if (changed_mc == 2 || changed_app == 2) {
mempools_free_mcconf(mcconf);
mempools_free_appconf(appconf);
}
2022-03-31 10:37:36 -07:00
return res;
}
2022-05-06 09:41:50 -07:00
static inline float lim_max(float min, float max) { (void)min; return max; }
static inline float lim_min(float min, float max) { (void)max; return min; }
2022-03-31 10:37:36 -07:00
static lbm_value ext_conf_get(lbm_value *args, lbm_uint argn) {
lbm_value res = ENC_SYM_EERROR;
2022-03-31 10:37:36 -07:00
if (argn != 1 && argn != 2) {
return res;
}
if (lbm_type_of(args[0]) != LBM_TYPE_SYMBOL) {
return res;
}
if (argn == 2 && !lbm_is_number(args[1])) {
return res;
}
2022-05-06 09:41:50 -07:00
int defaultcfg = 0;
2022-03-31 10:37:36 -07:00
if (argn == 2) {
2022-05-06 09:41:50 -07:00
defaultcfg = lbm_dec_as_i32(args[1]);
2022-03-31 10:37:36 -07:00
}
lbm_uint name = lbm_dec_sym(args[0]);
2022-04-12 05:09:16 -07:00
mc_configuration *mcconf;
app_configuration *appconf;
2022-03-31 10:37:36 -07:00
if (defaultcfg) {
2022-04-12 05:09:16 -07:00
mcconf = mempools_alloc_mcconf();
appconf = mempools_alloc_appconf();
2022-03-31 10:37:36 -07:00
confgenerator_set_defaults_mcconf(mcconf);
confgenerator_set_defaults_appconf(appconf);
2022-05-06 09:41:50 -07:00
if (defaultcfg == 2) {
#ifdef HW_LIM_CURRENT
mcconf->l_current_max = lim_max(HW_LIM_CURRENT);
mcconf->l_current_min = lim_min(HW_LIM_CURRENT);
#else
mcconf->l_current_max = 500.0;
mcconf->l_current_min = -500.0;
#endif
#ifdef HW_LIM_CURRENT_IN
mcconf->l_in_current_max = lim_max(HW_LIM_CURRENT_IN);
mcconf->l_in_current_min = lim_min(HW_LIM_CURRENT_IN);
#else
mcconf->l_in_current_max = 500.0;
mcconf->l_in_current_min = -500.0;
#endif
#ifdef HW_LIM_CURRENT_ABS
mcconf->l_abs_current_max = lim_max(HW_LIM_CURRENT_ABS);
#else
mcconf->l_abs_current_max = 500.0;
#endif
#ifdef HW_LIM_VIN
mcconf->l_max_vin = lim_max(HW_LIM_CURRENT_ABS);
mcconf->l_min_vin = lim_min(HW_LIM_CURRENT_ABS);
#else
mcconf->l_max_vin = 100.0;
mcconf->l_min_vin = 3.0;
#endif
#ifdef HW_LIM_ERPM
mcconf->l_max_erpm = lim_max(HW_LIM_ERPM);
mcconf->l_min_erpm = lim_min(HW_LIM_ERPM);
#else
mcconf->l_max_erpm = 500000.0;
mcconf->l_min_erpm = -500000.0;
#endif
#ifdef HW_LIM_DUTY_MIN
mcconf->l_min_duty = lim_max(HW_LIM_DUTY_MIN);
#else
mcconf->l_min_duty = 0.1;
#endif
#ifdef HW_LIM_DUTY_MAX
mcconf->l_max_duty = lim_max(HW_LIM_DUTY_MAX);
#else
mcconf->l_max_duty = 0.98;
#endif
#ifdef HW_LIM_TEMP_FET
mcconf->l_temp_fet_start = lim_max(HW_LIM_TEMP_FET);
mcconf->l_temp_fet_end = lim_max(HW_LIM_TEMP_FET);
#else
mcconf->l_temp_fet_start = 120.0;
mcconf->l_temp_fet_end = 120.0;
#endif
}
2022-03-31 10:37:36 -07:00
} else {
2022-04-12 05:09:16 -07:00
mcconf = (mc_configuration*)mc_interface_get_configuration();
appconf = (app_configuration*)app_get_configuration();
2022-03-31 10:37:36 -07:00
}
const float speed_fact = ((mcconf->si_motor_poles / 2.0) * 60.0 *
mcconf->si_gear_ratio) / (mcconf->si_wheel_diameter * M_PI);
if (compare_symbol(name, &syms_vesc.l_current_min)) {
res = lbm_enc_float(mcconf->l_current_min);
} else if (compare_symbol(name, &syms_vesc.l_current_max)) {
res = lbm_enc_float(mcconf->l_current_max);
} else if (compare_symbol(name, &syms_vesc.l_current_min_scale)) {
res = lbm_enc_float(mcconf->l_current_min_scale);
} else if (compare_symbol(name, &syms_vesc.l_current_max_scale)) {
res = lbm_enc_float(mcconf->l_current_max_scale);
} else if (compare_symbol(name, &syms_vesc.l_in_current_min)) {
res = lbm_enc_float(mcconf->l_in_current_min);
} else if (compare_symbol(name, &syms_vesc.l_in_current_max)) {
res = lbm_enc_float(mcconf->l_in_current_max);
2022-05-06 05:37:08 -07:00
} else if (compare_symbol(name, &syms_vesc.l_abs_current_max)) {
res = lbm_enc_float(mcconf->l_abs_current_max);
2022-03-31 10:37:36 -07:00
} else if (compare_symbol(name, &syms_vesc.l_min_erpm)) {
res = lbm_enc_float(mcconf->l_min_erpm);
} else if (compare_symbol(name, &syms_vesc.l_max_erpm)) {
res = lbm_enc_float(mcconf->l_max_erpm);
} else if (compare_symbol(name, &syms_vesc.l_min_vin)) {
res = lbm_enc_float(mcconf->l_min_vin);
} else if (compare_symbol(name, &syms_vesc.l_max_vin)) {
res = lbm_enc_float(mcconf->l_max_vin);
} else if (compare_symbol(name, &syms_vesc.l_min_duty)) {
res = lbm_enc_float(mcconf->l_min_duty);
} else if (compare_symbol(name, &syms_vesc.l_max_duty)) {
res = lbm_enc_float(mcconf->l_max_duty);
2022-03-31 10:37:36 -07:00
} else if (compare_symbol(name, &syms_vesc.l_watt_min)) {
res = lbm_enc_float(mcconf->l_watt_min);
} else if (compare_symbol(name, &syms_vesc.l_watt_max)) {
res = lbm_enc_float(mcconf->l_watt_max);
2022-07-31 04:30:10 -07:00
} else if (compare_symbol(name, &syms_vesc.motor_type)) {
res = lbm_enc_i(mcconf->motor_type);
2022-05-06 05:37:08 -07:00
} else if (compare_symbol(name, &syms_vesc.foc_sensor_mode)) {
res = lbm_enc_i(mcconf->foc_sensor_mode);
2022-03-31 10:37:36 -07:00
} else if (compare_symbol(name, &syms_vesc.foc_current_kp)) {
res = lbm_enc_float(mcconf->foc_current_kp);
} else if (compare_symbol(name, &syms_vesc.foc_current_ki)) {
res = lbm_enc_float(mcconf->foc_current_ki);
} else if (compare_symbol(name, &syms_vesc.foc_motor_l)) {
res = lbm_enc_float(mcconf->foc_motor_l * 1e6);
} else if (compare_symbol(name, &syms_vesc.foc_motor_ld_lq_diff)) {
res = lbm_enc_float(mcconf->foc_motor_ld_lq_diff * 1e6);
} else if (compare_symbol(name, &syms_vesc.foc_motor_r)) {
res = lbm_enc_float(mcconf->foc_motor_r * 1e3);
} else if (compare_symbol(name, &syms_vesc.foc_motor_flux_linkage)) {
res = lbm_enc_float(mcconf->foc_motor_flux_linkage * 1e3);
} else if (compare_symbol(name, &syms_vesc.foc_observer_gain)) {
res = lbm_enc_float(mcconf->foc_observer_gain * 1e-6);
2022-05-06 05:37:08 -07:00
} else if (compare_symbol(name, &syms_vesc.foc_hfi_voltage_start)) {
res = lbm_enc_float(mcconf->foc_hfi_voltage_start);
} else if (compare_symbol(name, &syms_vesc.foc_hfi_voltage_run)) {
res = lbm_enc_float(mcconf->foc_hfi_voltage_run);
} else if (compare_symbol(name, &syms_vesc.foc_hfi_voltage_max)) {
res = lbm_enc_float(mcconf->foc_hfi_voltage_max);
} else if (compare_symbol(name, &syms_vesc.foc_sl_erpm_hfi)) {
res = lbm_enc_float(mcconf->foc_sl_erpm_hfi);
} else if (compare_symbol(name, &syms_vesc.m_invert_direction)) {
res = lbm_enc_i(mcconf->m_invert_direction);
} else if (compare_symbol(name, &syms_vesc.m_out_aux_mode)) {
res = lbm_enc_i(mcconf->m_out_aux_mode);
2022-03-31 10:37:36 -07:00
} else if (compare_symbol(name, &syms_vesc.min_speed)) {
res = lbm_enc_float(mcconf->l_min_erpm / speed_fact);
} else if (compare_symbol(name, &syms_vesc.max_speed)) {
res = lbm_enc_float(mcconf->l_max_erpm / speed_fact);
} else if (compare_symbol(name, &syms_vesc.controller_id)) {
res = lbm_enc_i(appconf->controller_id);
} else if (compare_symbol(name, &syms_vesc.app_to_use)) {
res = lbm_enc_i(appconf->app_to_use);
2022-05-06 05:37:08 -07:00
} else if (compare_symbol(name, &syms_vesc.ppm_ctrl_type)) {
res = lbm_enc_i(appconf->app_ppm_conf.ctrl_type);
} else if (compare_symbol(name, &syms_vesc.ppm_pulse_start)) {
res = lbm_enc_float(appconf->app_ppm_conf.pulse_start);
} else if (compare_symbol(name, &syms_vesc.ppm_pulse_end)) {
res = lbm_enc_float(appconf->app_ppm_conf.pulse_end);
} else if (compare_symbol(name, &syms_vesc.ppm_pulse_center)) {
res = lbm_enc_float(appconf->app_ppm_conf.pulse_center);
} else if (compare_symbol(name, &syms_vesc.ppm_ramp_time_pos)) {
res = lbm_enc_float(appconf->app_ppm_conf.ramp_time_pos);
} else if (compare_symbol(name, &syms_vesc.ppm_ramp_time_neg)) {
res = lbm_enc_float(appconf->app_ppm_conf.ramp_time_neg);
2022-03-31 10:37:36 -07:00
}
2022-04-12 05:09:16 -07:00
if (defaultcfg) {
mempools_free_mcconf(mcconf);
mempools_free_appconf(appconf);
}
2022-03-31 10:37:36 -07:00
if (lbm_type_of(res) == LBM_TYPE_SYMBOL && lbm_dec_sym(res) == SYM_EERROR) {
lbm_set_error_reason("Parameter not recognized");
}
return res;
}
static lbm_value ext_conf_store(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
mc_configuration *mcconf = mempools_alloc_mcconf();
*mcconf = *mc_interface_get_configuration();
bool res_mc = conf_general_store_mc_configuration(mcconf, mc_interface_get_motor_thread() == 2);
mempools_free_mcconf(mcconf);
app_configuration *appconf = mempools_alloc_appconf();
*appconf = *app_get_configuration();
bool res_app = conf_general_store_app_configuration(appconf);
mempools_free_appconf(appconf);
return lbm_enc_sym((res_mc && res_app) ? SYM_TRUE : SYM_NIL);
}
2022-05-06 10:49:58 -07:00
typedef struct {
bool detect_can;
float max_power_loss;
float min_current_in;
float max_current_in;
float openloop_rpm;
float sl_erpm;
lbm_cid id;
} detect_args;
static void detect_task(void *arg) {
detect_args *a = (detect_args*)arg;
int res = conf_general_detect_apply_all_foc_can(a->detect_can, a->max_power_loss,
a->min_current_in, a->max_current_in, a->openloop_rpm, a->sl_erpm);
lbm_unblock_ctx(a->id, lbm_enc_i(res));
}
static lbm_value ext_conf_detect_foc(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(6);
static detect_args a;
a.detect_can = lbm_dec_as_i32(args[0]);
a.max_power_loss = lbm_dec_as_float(args[1]);
a.min_current_in = lbm_dec_as_float(args[2]);
a.max_current_in = lbm_dec_as_float(args[3]);
a.openloop_rpm = lbm_dec_as_float(args[4]);
a.sl_erpm = lbm_dec_as_float(args[5]);
a.id = lbm_get_current_cid();
worker_execute(detect_task, &a);
lbm_block_ctx_from_extension();
return ENC_SYM_TRUE;
2022-05-06 10:49:58 -07:00
}
static lbm_value ext_conf_set_pid_offset(lbm_value *args, lbm_uint argn) {
if (argn != 1 && argn != 2) {
lbm_set_error_reason(error_reason_argn);
return ENC_SYM_EERROR;
}
if (!lbm_is_number(args[0]) || (argn == 2 && !is_symbol_true_false(args[1]))) {
return ENC_SYM_EERROR;
}
float angle = lbm_dec_as_float(args[0]);
if (angle < -360.0 || angle > 360.0) {
lbm_set_error_reason("Invalid angle. Range should be -360 to 360.");
return ENC_SYM_EERROR;
}
bool store = false;
if (argn == 2) {
store = lbm_is_symbol_true(args[1]);
}
mc_interface_update_pid_pos_offset(angle, store);
return ENC_SYM_TRUE;
}
typedef struct {
float current;
int samples;
lbm_cid id;
} measure_res_args;
static void measure_res_task(void *arg) {
measure_res_args *a = (measure_res_args*)arg;
2022-11-08 09:21:06 -08:00
float res = -1.0;
mcpwm_foc_measure_resistance(a->current, a->samples, true, &res);
lbm_unblock_ctx(a->id, lbm_enc_float(res));
}
static lbm_value ext_conf_measure_res(lbm_value *args, lbm_uint argn) {
if (argn != 1 && argn != 2) {
lbm_set_error_reason(error_reason_argn);
return ENC_SYM_EERROR;
}
CHECK_NUMBER_ALL();
if (mc_interface_get_configuration()->motor_type != MOTOR_TYPE_FOC) {
lbm_set_error_reason("Motor type must be FOC");
return ENC_SYM_EERROR;
}
static measure_res_args a;
a.current = lbm_dec_as_float(args[0]);
a.samples = 100;
if (argn == 2) {
a.samples = lbm_dec_as_u32(args[1]);
}
a.id = lbm_get_current_cid();
worker_execute(measure_res_task, &a);
lbm_block_ctx_from_extension();
return ENC_SYM_TRUE;
}
2022-04-14 04:45:36 -07:00
// Extra array extensions
static lbm_value ext_bufclear(lbm_value *args, lbm_uint argn) {
lbm_value res = ENC_SYM_EERROR;
2022-04-14 04:45:36 -07:00
if ((argn != 1 && argn != 2 && argn != 3 && argn != 4) || !lbm_is_array(args[0])) {
return res;
}
lbm_array_header_t *array = (lbm_array_header_t *)lbm_car(args[0]);
if (array->elt_type != LBM_TYPE_BYTE) {
return res;
}
uint8_t clear_byte = 0;
if (argn >= 2) {
if (!lbm_is_number(args[1])) {
return res;
}
clear_byte = lbm_dec_as_u32(args[1]);
}
unsigned int start = 0;
if (argn >= 3) {
if (!lbm_is_number(args[2])) {
return res;
}
unsigned int start_new = lbm_dec_as_u32(args[2]);
if (start_new < array->size) {
start = start_new;
} else {
return res;
}
}
unsigned int len = array->size - start;
if (argn >= 4) {
if (!lbm_is_number(args[3])) {
return res;
}
unsigned int len_new = lbm_dec_as_u32(args[3]);
if (len_new <= len) {
len = len_new;
}
}
2022-04-14 08:44:59 -07:00
memset((char*)array->data + start, clear_byte, len);
res = ENC_SYM_TRUE;
2022-04-14 04:45:36 -07:00
return res;
}
2022-04-18 09:06:21 -07:00
static lbm_value ext_bufcpy(lbm_value *args, lbm_uint argn) {
lbm_value res = ENC_SYM_EERROR;
2022-04-18 09:06:21 -07:00
if (argn != 5 || !lbm_is_array(args[0]) || !lbm_is_number(args[1])
|| !lbm_is_array(args[2]) || !lbm_is_number(args[3]) || !lbm_is_number(args[4])) {
return res;
}
lbm_array_header_t *array1 = (lbm_array_header_t *)lbm_car(args[0]);
if (array1->elt_type != LBM_TYPE_BYTE) {
return res;
}
unsigned int start1 = lbm_dec_as_u32(args[1]);
lbm_array_header_t *array2 = (lbm_array_header_t *)lbm_car(args[2]);
if (array2->elt_type != LBM_TYPE_BYTE) {
return res;
}
unsigned int start2 = lbm_dec_as_u32(args[3]);
unsigned int len = lbm_dec_as_u32(args[4]);
if (start1 < array1->size && start2 < array2->size) {
if (len > (array1->size - start1)) {
len = (array1->size - start1);
}
if (len > (array2->size - start2)) {
len = (array2->size - start2);
}
memcpy((char*)array1->data + start1, (char*)array2->data + start2, len);
}
res = ENC_SYM_TRUE;
2022-04-18 09:06:21 -07:00
return res;
}
static lbm_value ext_bufset_bit(lbm_value *args, lbm_uint argn) {
lbm_value res = ENC_SYM_EERROR;
2022-04-18 09:06:21 -07:00
if (argn != 3 || !lbm_is_array(args[0]) ||
!lbm_is_number(args[1]) || !lbm_is_number(args[2])) {
return res;
}
lbm_array_header_t *array = (lbm_array_header_t *)lbm_car(args[0]);
if (array->elt_type != LBM_TYPE_BYTE) {
return res;
}
unsigned int pos = lbm_dec_as_u32(args[1]);
unsigned int bit = lbm_dec_as_u32(args[2]) ? 1 : 0;
unsigned int bytepos = pos / 8;
unsigned int bitpos = pos % 8;
if (bytepos < array->size) {
((uint8_t*)array->data)[bytepos] &= ~(1 << bitpos);
((uint8_t*)array->data)[bytepos] |= (bit << bitpos);
}
2022-07-25 05:26:21 -07:00
res = ENC_SYM_TRUE;
2022-04-18 09:06:21 -07:00
return res;
}
static lbm_value make_list(int num, ...) {
va_list arguments;
va_start (arguments, num);
lbm_value res = ENC_SYM_NIL;
for (int i = 0; i < num; i++) {
res = lbm_cons(va_arg(arguments, lbm_value), res);
}
va_end (arguments);
return lbm_list_destructive_reverse(res);
}
static lbm_uint sym_res;
static lbm_uint sym_loop;
static lbm_uint sym_break;
static lbm_uint sym_brk;
static lbm_uint sym_rst;
static lbm_value ext_me_defun(lbm_value *argsi, lbm_uint argn) {
if (argn != 3) {
return ENC_SYM_EERROR;
}
lbm_value name = argsi[0];
lbm_value args = argsi[1];
lbm_value body = argsi[2];
// (define name (lambda args body))
return make_list(3,
lbm_enc_sym(SYM_DEFINE),
name,
make_list(3,
lbm_enc_sym(SYM_LAMBDA),
args,
body));
}
static lbm_value ext_me_loopfor(lbm_value *args, lbm_uint argn) {
if (argn != 5) {
return ENC_SYM_EERROR;
}
lbm_value it = args[0];
lbm_value start = args[1];
lbm_value cond = args[2];
lbm_value update = args[3];
lbm_value body = args[4];
// (let ((loop (lambda (it res break) (if cond (loop update body break) res)))) (call-cc (lambda (brk) (loop start nil brk))))
return make_list(3,
lbm_enc_sym(SYM_LET),
make_list(1,
make_list(2,
lbm_enc_sym(sym_loop),
make_list(3,
lbm_enc_sym(SYM_LAMBDA),
make_list(3, it, lbm_enc_sym(sym_res), lbm_enc_sym(sym_break)),
make_list(4,
lbm_enc_sym(SYM_IF),
cond,
make_list(4, lbm_enc_sym(sym_loop), update, body, lbm_enc_sym(sym_break)),
lbm_enc_sym(sym_res))))),
make_list(2,
lbm_enc_sym(SYM_CALLCC),
make_list(3,
lbm_enc_sym(SYM_LAMBDA),
make_list(1, lbm_enc_sym(sym_brk)),
make_list(4, lbm_enc_sym(sym_loop), start, ENC_SYM_NIL, lbm_enc_sym(sym_brk)))));
}
static lbm_value ext_me_loopwhile(lbm_value *args, lbm_uint argn) {
if (argn != 2) {
return ENC_SYM_EERROR;
}
lbm_value cond = args[0];
lbm_value body = args[1];
// (let ((loop (lambda (res break) (if cond (loop body break) res)))) (call-cc (lambda (brk) (loop nil brk))))
return make_list(3,
lbm_enc_sym(SYM_LET),
make_list(1,
make_list(2,
lbm_enc_sym(sym_loop),
make_list(3,
lbm_enc_sym(SYM_LAMBDA),
make_list(2, lbm_enc_sym(sym_res), lbm_enc_sym(sym_break)),
make_list(4,
lbm_enc_sym(SYM_IF),
cond,
make_list(3, lbm_enc_sym(sym_loop), body, lbm_enc_sym(sym_break)),
lbm_enc_sym(sym_res))))),
make_list(2,
lbm_enc_sym(SYM_CALLCC),
make_list(3,
lbm_enc_sym(SYM_LAMBDA),
make_list(1, lbm_enc_sym(sym_brk)),
make_list(3, lbm_enc_sym(sym_loop), ENC_SYM_NIL, lbm_enc_sym(sym_brk)))));
}
static lbm_value ext_me_looprange(lbm_value *args, lbm_uint argn) {
if (argn != 4) {
return ENC_SYM_EERROR;
}
lbm_value it = args[0];
lbm_value start = args[1];
lbm_value end = args[2];
lbm_value body = args[3];
// (let ((loop (lambda (it res break) (if (< it end) (loop (+ it 1) body break) res)))) (call-cc (lambda (brk) (loop start nil brk))))
return make_list(3,
lbm_enc_sym(SYM_LET),
make_list(1,
make_list(2,
lbm_enc_sym(sym_loop),
make_list(3,
lbm_enc_sym(SYM_LAMBDA),
make_list(3, it, lbm_enc_sym(sym_res), lbm_enc_sym(sym_break)),
make_list(4,
lbm_enc_sym(SYM_IF),
make_list(3, lbm_enc_sym(SYM_LT), it, end),
make_list(4, lbm_enc_sym(sym_loop), make_list(3, lbm_enc_sym(SYM_ADD), it, lbm_enc_i(1)), body, lbm_enc_sym(sym_break)),
lbm_enc_sym(sym_res))))),
make_list(2,
lbm_enc_sym(SYM_CALLCC),
make_list(3,
lbm_enc_sym(SYM_LAMBDA),
make_list(1, lbm_enc_sym(sym_brk)),
make_list(4, lbm_enc_sym(sym_loop), start, ENC_SYM_NIL, lbm_enc_sym(sym_brk)))));
}
static lbm_value ext_me_loopforeach(lbm_value *args, lbm_uint argn) {
if (argn != 3) {
return ENC_SYM_EERROR;
}
lbm_value it = args[0];
lbm_value lst = args[1];
lbm_value body = args[2];
// (let ((loop (lambda (it rst res break) (if (eq it nil) res (loop (car rst) (cdr rst) body break))))) (call-cc (lambda (brk) (loop (car lst) (cdr lst) nil brk))))
return make_list(3,
lbm_enc_sym(SYM_LET),
make_list(1,
make_list(2,
lbm_enc_sym(sym_loop),
make_list(3,
lbm_enc_sym(SYM_LAMBDA),
make_list(4, it, lbm_enc_sym(sym_rst), lbm_enc_sym(sym_res), lbm_enc_sym(sym_break)),
make_list(4,
lbm_enc_sym(SYM_IF),
make_list(3, lbm_enc_sym(SYM_EQ), it, ENC_SYM_NIL),
lbm_enc_sym(sym_res),
make_list(5,
lbm_enc_sym(sym_loop),
make_list(2, lbm_enc_sym(SYM_CAR), lbm_enc_sym(sym_rst)),
make_list(2, lbm_enc_sym(SYM_CDR), lbm_enc_sym(sym_rst)),
body,
lbm_enc_sym(sym_break))
)))),
make_list(2,
lbm_enc_sym(SYM_CALLCC),
make_list(3,
lbm_enc_sym(SYM_LAMBDA),
make_list(1, lbm_enc_sym(sym_brk)),
make_list(5,
lbm_enc_sym(sym_loop),
make_list(2, lbm_enc_sym(SYM_CAR), lst),
make_list(2, lbm_enc_sym(SYM_CDR), lst),
ENC_SYM_NIL,
lbm_enc_sym(sym_brk)))));
}
static lbm_value ext_uavcan_last_rawcmd(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(1);
int can_if = lbm_dec_as_i32(args[0]);
uavcan_cmd_info info = canard_driver_last_rawcmd(can_if);
lbm_value out_list = ENC_SYM_NIL;
out_list = lbm_cons(lbm_enc_float(info.age), out_list);
out_list = lbm_cons(lbm_enc_float(info.value), out_list);
return out_list;
}
static lbm_value ext_uavcan_last_rpmcmd(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(1);
int can_if = lbm_dec_as_i32(args[0]);
uavcan_cmd_info info = canard_driver_last_rpmcmd(can_if);
lbm_value out_list = ENC_SYM_NIL;
out_list = lbm_cons(lbm_enc_float(info.age), out_list);
out_list = lbm_cons(lbm_enc_float(info.value), out_list);
return out_list;
}
2022-09-04 13:18:15 -07:00
static lbm_value ext_lbm_set_quota(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(1);
uint32_t q = lbm_dec_as_u32(args[0]);
if (q < 1) {
return ENC_SYM_EERROR;
}
lbm_set_eval_step_quota(q);
return ENC_SYM_TRUE;
}
2022-09-26 04:32:41 -07:00
static lbm_value ext_plot_init(lbm_value *args, lbm_uint argn) {
if (argn != 2) {
return ENC_SYM_EERROR;
2022-09-26 04:32:41 -07:00
}
char *namex = lbm_dec_str(args[0]);
if (!namex) {
return ENC_SYM_EERROR;
2022-09-26 04:32:41 -07:00
}
char *namey = lbm_dec_str(args[0]);
if (!namey) {
return ENC_SYM_EERROR;
2022-09-26 04:32:41 -07:00
}
commands_init_plot(namex, namey);
return ENC_SYM_TRUE;
}
static lbm_value ext_plot_add_graph(lbm_value *args, lbm_uint argn) {
if (argn != 1) {
return ENC_SYM_EERROR;
2022-09-26 04:32:41 -07:00
}
char *name = lbm_dec_str(args[0]);
if (!name) {
return ENC_SYM_EERROR;
2022-09-26 04:32:41 -07:00
}
commands_plot_add_graph(name);
return ENC_SYM_TRUE;
}
static lbm_value ext_plot_set_graph(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(1);
commands_plot_set_graph(lbm_dec_as_i32(args[0]));
return ENC_SYM_TRUE;
}
static lbm_value ext_plot_send_points(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(2);
commands_send_plot_points(
lbm_dec_as_float(args[0]),
lbm_dec_as_float(args[1]));
return ENC_SYM_TRUE;
}
2022-10-14 04:48:17 -07:00
// IO-boards
static lbm_value ext_ioboard_get_adc(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(2);
int id = lbm_dec_as_i32(args[0]);
int channel = lbm_dec_as_i32(args[1]);
if (channel < 1 || channel > 8) {
lbm_set_error_reason("Channel must be 1 - 8");
return ENC_SYM_EERROR;
}
io_board_adc_values *val = 0;
if (channel >= 5) {
val = comm_can_get_io_board_adc_5_8_id(id);
channel -= 4;
} else {
val = comm_can_get_io_board_adc_1_4_id(id);
}
if (val) {
return lbm_enc_float(val->adc_voltages[channel - 1]);
} else {
return lbm_enc_float(-1.0);
}
}
static lbm_value ext_ioboard_get_digital(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(2);
int id = lbm_dec_as_i32(args[0]);
int channel = lbm_dec_as_i32(args[1]);
if (channel < 1 || channel > 64) {
lbm_set_error_reason("Channel must be 1 - 64");
return ENC_SYM_EERROR;
}
io_board_digial_inputs *val = comm_can_get_io_board_digital_in_id(id);
if (val) {
return lbm_enc_i(val->inputs >> (channel - 1));
} else {
return lbm_enc_i(-1);
}
}
static lbm_value ext_ioboard_set_digital(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(3);
int id = lbm_dec_as_i32(args[0]);
int channel = lbm_dec_as_i32(args[1]);
bool on = lbm_dec_as_i32(args[2]);
comm_can_io_board_set_output_digital(id, channel, on);
return ENC_SYM_TRUE;
}
static lbm_value ext_ioboard_set_pwm(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(3);
int id = lbm_dec_as_i32(args[0]);
int channel = lbm_dec_as_i32(args[1]);
float duty = lbm_dec_as_float(args[2]);
comm_can_io_board_set_output_pwm(id, channel, duty);
return ENC_SYM_TRUE;
}
// Logging
static lbm_value ext_log_start(lbm_value *args, lbm_uint argn) {
CHECK_ARGN(5);
if (!lbm_is_number(args[0]) ||
!lbm_is_number(args[1]) ||
!lbm_is_number(args[2]) ||
!is_symbol_true_false(args[3]) ||
!is_symbol_true_false(args[4])) {
return ENC_SYM_EERROR;
}
log_start(
lbm_dec_as_i32(args[0]),
lbm_dec_as_i32(args[1]),
lbm_dec_as_float(args[2]),
lbm_is_symbol_true(args[3]),
lbm_is_symbol_true(args[4]),
lbm_is_symbol_true(args[4]));
return ENC_SYM_TRUE;
}
static lbm_value ext_log_stop(lbm_value *args, lbm_uint argn) {
CHECK_ARGN_NUMBER(1);
log_stop(lbm_dec_as_i32(args[0]));
return ENC_SYM_TRUE;
}
static lbm_value ext_log_config_field(lbm_value *args, lbm_uint argn) {
if (argn != 8) {
lbm_set_error_reason(error_reason_argn);
return ENC_SYM_EERROR;
}
int arg_now = 0;
int can_id = -1;
if (lbm_is_number(args[arg_now])) {
can_id = lbm_dec_as_i32(args[arg_now++]);
} else {
return ENC_SYM_EERROR;
}
int field_ind = -1;
if (lbm_is_number(args[arg_now])) {
field_ind = lbm_dec_as_i32(args[arg_now++]);
} else {
return ENC_SYM_EERROR;
}
char *key = lbm_dec_str(args[arg_now++]);
if (key == NULL) {
return ENC_SYM_EERROR;
}
char *name = lbm_dec_str(args[arg_now++]);
if (name == NULL) {
return ENC_SYM_EERROR;
}
char *unit = lbm_dec_str(args[arg_now++]);
if (unit == NULL) {
return ENC_SYM_EERROR;
}
int precision = -1;
if (lbm_is_number(args[arg_now])) {
precision = lbm_dec_as_i32(args[arg_now++]);
} else {
return ENC_SYM_EERROR;
}
bool is_relative = false;
if (is_symbol_true_false(args[arg_now])) {
is_relative = lbm_is_symbol_true(args[arg_now++]);
} else {
return ENC_SYM_EERROR;
}
bool is_timestamp = false;
if (is_symbol_true_false(args[arg_now])) {
is_timestamp = lbm_is_symbol_true(args[arg_now++]);
} else {
return ENC_SYM_EERROR;
}
log_config_field(can_id, field_ind, key, name, unit, precision, is_relative, is_timestamp);
return ENC_SYM_TRUE;
}
static lbm_value ext_log_send_f32(lbm_value *args, lbm_uint argn) {
unsigned int arg_now = 0;
int can_id = -1;
if (lbm_is_number(args[arg_now])) {
can_id = lbm_dec_as_i32(args[arg_now++]);
} else {
return ENC_SYM_EERROR;
}
int field_start = -1;
if (lbm_is_number(args[arg_now])) {
field_start = lbm_dec_as_i32(args[arg_now++]);
} else {
return ENC_SYM_EERROR;
}
int32_t ind = 0;
uint8_t *buffer = mempools_get_packet_buffer();
buffer[ind++] = COMM_LOG_DATA_F32;
buffer_append_int16(buffer, field_start, &ind);
int append_cnt = 0;
while (arg_now < argn) {
if (lbm_is_number(args[arg_now])) {
buffer_append_float32_auto(buffer, lbm_dec_as_float(args[arg_now]), &ind);
append_cnt++;
if (append_cnt >= 100) {
mempools_free_packet_buffer(buffer);
return ENC_SYM_EERROR;
}
} else if (lbm_is_cons(args[arg_now])) {
lbm_value curr = args[arg_now];
while (lbm_is_cons(curr)) {
lbm_value val = lbm_car(curr);
if (lbm_is_number(val)) {
buffer_append_float32_auto(buffer, lbm_dec_as_float(val), &ind);
append_cnt++;
if (append_cnt >= 100) {
mempools_free_packet_buffer(buffer);
return ENC_SYM_EERROR;
}
} else {
mempools_free_packet_buffer(buffer);
return ENC_SYM_EERROR;
}
curr = lbm_cdr(curr);
}
} else {
mempools_free_packet_buffer(buffer);
return ENC_SYM_EERROR;
}
arg_now++;
}
if (can_id >= 0 && can_id < 255) {
comm_can_send_buffer(can_id, buffer, ind, 0);
} else {
commands_send_packet(buffer, ind);
}
mempools_free_packet_buffer(buffer);
return ENC_SYM_TRUE;
}
2022-11-09 08:10:07 -08:00
static lbm_value ext_gnss_lat_lon(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
volatile gnss_data *g = mc_interface_gnss();
lbm_value lat_lon = ENC_SYM_NIL;
lat_lon = lbm_cons(lbm_enc_double(g->lon), lat_lon);
lat_lon = lbm_cons(lbm_enc_double(g->lat), lat_lon);
return lat_lon;
}
static lbm_value ext_gnss_height(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
return lbm_enc_float(mc_interface_gnss()->height);
}
static lbm_value ext_gnss_speed(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
return lbm_enc_float(mc_interface_gnss()->speed);
}
static lbm_value ext_gnss_hdop(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
return lbm_enc_float(mc_interface_gnss()->hdop);
}
static lbm_value ext_gnss_date_time(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
volatile gnss_data *g = mc_interface_gnss();
int ms = g->ms_today % 1000;
int s = (g->ms_today / 1000) % 60;
int m = ((g->ms_today / 1000) / 60) % 60;
int h = ((g->ms_today / 1000) / (60 * 60)) % 24;
lbm_value lat_lon = ENC_SYM_NIL;
lat_lon = lbm_cons(lbm_enc_i(ms), lat_lon);
lat_lon = lbm_cons(lbm_enc_i(s), lat_lon);
lat_lon = lbm_cons(lbm_enc_i(m), lat_lon);
lat_lon = lbm_cons(lbm_enc_i(h), lat_lon);
lat_lon = lbm_cons(lbm_enc_i(g->dd), lat_lon);
lat_lon = lbm_cons(lbm_enc_i(g->mo), lat_lon);
lat_lon = lbm_cons(lbm_enc_i(g->yy), lat_lon);
return lat_lon;
}
static lbm_value ext_gnss_age(lbm_value *args, lbm_uint argn) {
(void)args; (void)argn;
return lbm_enc_float(UTILS_AGE_S(mc_interface_gnss()->last_update));
}
2022-07-25 05:26:21 -07:00
static lbm_value ext_empty(lbm_value *args, lbm_uint argn) {
(void)args;(void)argn;
return ENC_SYM_TRUE;
}
// Declare native lib extension
lbm_value ext_load_native_lib(lbm_value *args, lbm_uint argn);
lbm_value ext_unload_native_lib(lbm_value *args, lbm_uint argn);
2022-11-04 11:46:36 -07:00
// Event ringbuffer
static bool pause_gc(uint32_t num_free, int timeout_cnt) {
lbm_pause_eval_with_gc(num_free);
while (lbm_get_eval_state() != EVAL_CPS_STATE_PAUSED && timeout_cnt > 0) {
chThdSleep(1);
timeout_cnt--;
}
return timeout_cnt > 0;
}
typedef enum {
EXT_EVENT_SYM = 0,
EXT_EVENT_SYM_ARRAY,
EXT_EVENT_SYM_INT_ARRAY
} EXT_EVENT_t;
typedef struct {
EXT_EVENT_t type;
lbm_uint sym;
int32_t i;
char *array;
int32_t array_len;
} ext_event;
#define EVENT_BUFFER_LEN 20
static thread_t *event_tp = NULL;
static ext_event event_buffer[EVENT_BUFFER_LEN] = {0};
static rb_t rb_events;
static THD_WORKING_AREA(event_thread_wa, 256);
static THD_FUNCTION(event_thread, arg) {
(void)arg;
event_tp = chThdGetSelfX();
chRegSetThreadName("Lisp Events");
for (;;) {
chEvtWaitAnyTimeout((eventmask_t)1, 20);
if (!event_handler_registered) {
rb_flush(&rb_events);
continue;
}
int event_cnt = rb_get_item_count(&rb_events);
if (event_cnt > 0) {
if (lbm_get_eval_state() == EVAL_CPS_STATE_PAUSED) {
continue;
}
2022-11-05 08:04:32 -07:00
lispif_lock_lbm();
2022-11-04 11:46:36 -07:00
if (!pause_gc(10 * event_cnt, 1000)) {
lbm_continue_eval();
lispif_unlock_lbm();
continue;
}
} else {
continue;
}
while (!rb_is_empty(&rb_events) && event_handler_registered) {
ext_event e;
rb_pop(&rb_events, &e);
if (e.type == EXT_EVENT_SYM) {
lbm_send_message(event_handler_pid, lbm_enc_sym(e.sym));
} else if (e.type == EXT_EVENT_SYM_ARRAY) {
lbm_value val;
if (lbm_share_array(&val, e.array, LBM_TYPE_BYTE, e.array_len)) {
lbm_value msg = lbm_cons(lbm_enc_sym(e.sym), val);
if (lbm_is_ptr(msg)) {
lbm_send_message(event_handler_pid, msg);
} else {
lispif_free(e.array);
}
} else {
lispif_free(e.array);
}
} else if (e.type == EXT_EVENT_SYM_INT_ARRAY) {
lbm_value val;
if (lbm_share_array(&val, e.array, LBM_TYPE_BYTE, e.array_len)) {
lbm_value msg_data = lbm_cons(lbm_enc_i32(e.i), val);
lbm_value msg = lbm_cons(lbm_enc_sym(e.sym), msg_data);
if (lbm_is_ptr(msg_data) && lbm_is_ptr(msg)) {
lbm_send_message(event_handler_pid, msg);
} else {
lispif_free(e.array);
}
} else {
lispif_free(e.array);
}
}
}
lbm_continue_eval();
lispif_unlock_lbm();
}
}
static void event_add(ext_event e, uint8_t *opt_array, int opt_array_len) {
if (!event_handler_registered) {
return;
}
if (!rb_is_full(&rb_events)) {
if (opt_array != NULL) {
e.array = lispif_malloc(opt_array_len);
2022-11-07 01:51:24 -08:00
e.array_len = opt_array_len;
2022-11-04 11:46:36 -07:00
if (e.array == NULL) {
return;
}
memcpy(e.array, opt_array, opt_array_len);
}
if (!rb_insert(&rb_events, &e) && e.array != NULL) {
lispif_free(e.array);
}
}
chEvtSignal(event_tp, (eventmask_t) 1);
}
2022-01-14 11:54:24 -08:00
void lispif_load_vesc_extensions(void) {
2022-05-10 10:05:43 -07:00
lispif_stop_lib();
2022-11-04 11:46:36 -07:00
if (event_tp == NULL) {
rb_init(&rb_events, event_buffer, sizeof(ext_event), EVENT_BUFFER_LEN);
chThdCreateStatic(event_thread_wa, sizeof(event_thread_wa), NORMALPRIO - 2, event_thread, NULL);
}
rb_flush(&rb_events);
#ifdef HW_ADC_EXT_GPIO
palSetPadMode(HW_ADC_EXT_GPIO, HW_ADC_EXT_PIN, PAL_MODE_INPUT_ANALOG);
#endif
#ifdef HW_ADC_EXT2_GPIO
palSetPadMode(HW_ADC_EXT2_GPIO, HW_ADC_EXT2_PIN, PAL_MODE_INPUT_ANALOG);
#endif
lbm_add_symbol_const("event-can-sid", &sym_event_can_sid);
lbm_add_symbol_const("event-can-eid", &sym_event_can_eid);
lbm_add_symbol_const("event-data-rx", &sym_event_data_rx);
lbm_add_symbol_const("event-shutdown", &sym_event_shutdown);
lbm_add_symbol_const("a01", &sym_res);
lbm_add_symbol_const("a02", &sym_loop);
lbm_add_symbol_const("break", &sym_break);
lbm_add_symbol_const("a03", &sym_brk);
lbm_add_symbol_const("a04", &sym_rst);
2022-03-31 06:18:59 -07:00
memset(&syms_vesc, 0, sizeof(syms_vesc));
2022-01-14 11:54:24 -08:00
// Various commands
2022-01-21 01:52:56 -08:00
lbm_add_extension("print", ext_print);
lbm_add_extension("timeout-reset", ext_reset_timeout);
lbm_add_extension("get-ppm", ext_get_ppm);
lbm_add_extension("get-ppm-age", ext_get_ppm_age);
2022-01-21 01:52:56 -08:00
lbm_add_extension("get-encoder", ext_get_encoder);
lbm_add_extension("set-servo", ext_set_servo);
lbm_add_extension("get-vin", ext_get_vin);
lbm_add_extension("select-motor", ext_select_motor);
lbm_add_extension("get-selected-motor", ext_get_selected_motor);
lbm_add_extension("get-bms-val", ext_get_bms_val);
2022-11-09 12:53:17 -08:00
lbm_add_extension("set-bms-val", ext_set_bms_val);
2022-11-10 06:24:34 -08:00
lbm_add_extension("send-bms-can", ext_send_bms_can);
2022-01-21 01:52:56 -08:00
lbm_add_extension("get-adc", ext_get_adc);
lbm_add_extension("get-adc-decoded", ext_get_adc_decoded);
2022-01-21 01:52:56 -08:00
lbm_add_extension("systime", ext_systime);
lbm_add_extension("secs-since", ext_secs_since);
2022-01-25 11:59:15 -08:00
lbm_add_extension("set-aux", ext_set_aux);
lbm_add_extension("event-register-handler", ext_register_event_handler);
lbm_add_extension("event-enable", ext_enable_event);
2022-02-06 14:12:05 -08:00
lbm_add_extension("get-imu-rpy", ext_get_imu_rpy);
lbm_add_extension("get-imu-quat", ext_get_imu_quat);
lbm_add_extension("get-imu-acc", ext_get_imu_acc);
lbm_add_extension("get-imu-gyro", ext_get_imu_gyro);
lbm_add_extension("get-imu-mag", ext_get_imu_mag);
lbm_add_extension("get-imu-acc-derot", ext_get_imu_acc_derot);
lbm_add_extension("get-imu-gyro-derot", ext_get_imu_gyro_derot);
2022-02-06 14:12:05 -08:00
lbm_add_extension("send-data", ext_send_data);
2022-03-13 01:15:24 -08:00
lbm_add_extension("get-remote-state", ext_get_remote_state);
2022-03-13 11:14:33 -07:00
lbm_add_extension("eeprom-store-f", ext_eeprom_store_f);
lbm_add_extension("eeprom-read-f", ext_eeprom_read_f);
lbm_add_extension("eeprom-store-i", ext_eeprom_store_i);
lbm_add_extension("eeprom-read-i", ext_eeprom_read_i);
2022-04-10 12:21:35 -07:00
lbm_add_extension("sysinfo", ext_sysinfo);
2022-07-25 05:26:21 -07:00
lbm_add_extension("import", ext_empty);
2022-01-14 11:54:24 -08:00
2022-06-17 01:43:19 -07:00
// APP commands
lbm_add_extension("app-adc-detach", ext_app_adc_detach);
lbm_add_extension("app-adc-override", ext_app_adc_override);
2022-10-07 07:07:32 -07:00
lbm_add_extension("app-ppm-detach", ext_app_ppm_detach);
lbm_add_extension("app-ppm-override", ext_app_ppm_override);
2022-10-07 10:36:52 -07:00
lbm_add_extension("set-remote-state", ext_set_remote_state);
lbm_add_extension("app-disable-output", ext_app_disable_output);
2022-11-09 02:56:59 -08:00
lbm_add_extension("app-pas-get-rpm", ext_app_pas_get_rpm);
2022-01-14 11:54:24 -08:00
// Motor set commands
2022-01-21 01:52:56 -08:00
lbm_add_extension("set-current", ext_set_current);
lbm_add_extension("set-current-rel", ext_set_current_rel);
lbm_add_extension("set-duty", ext_set_duty);
lbm_add_extension("set-brake", ext_set_brake);
lbm_add_extension("set-brake-rel", ext_set_brake_rel);
lbm_add_extension("set-handbrake", ext_set_handbrake);
lbm_add_extension("set-handbrake-rel", ext_set_handbrake_rel);
lbm_add_extension("set-rpm", ext_set_rpm);
lbm_add_extension("set-pos", ext_set_pos);
2022-04-14 12:17:45 -07:00
lbm_add_extension("foc-openloop", ext_foc_openloop);
2022-06-17 01:43:19 -07:00
lbm_add_extension("foc-beep", ext_foc_beep);
2022-01-14 11:54:24 -08:00
// Motor get commands
2022-01-21 01:52:56 -08:00
lbm_add_extension("get-current", ext_get_current);
lbm_add_extension("get-current-dir", ext_get_current_dir);
lbm_add_extension("get-current-in", ext_get_current_in);
lbm_add_extension("get-id", ext_get_id);
lbm_add_extension("get-iq", ext_get_iq);
lbm_add_extension("get-vd", ext_get_vd);
lbm_add_extension("get-vq", ext_get_vq);
2022-01-21 01:52:56 -08:00
lbm_add_extension("get-duty", ext_get_duty);
lbm_add_extension("get-rpm", ext_get_rpm);
lbm_add_extension("get-temp-fet", ext_get_temp_fet);
lbm_add_extension("get-temp-mot", ext_get_temp_mot);
lbm_add_extension("get-speed", ext_get_speed);
lbm_add_extension("get-dist", ext_get_dist);
2022-07-20 02:04:57 -07:00
lbm_add_extension("get-dist-abs", ext_get_dist_abs);
2022-01-21 01:52:56 -08:00
lbm_add_extension("get-batt", ext_get_batt);
lbm_add_extension("get-fault", ext_get_fault);
2022-01-16 07:21:18 -08:00
// CAN-comands
2022-01-21 01:52:56 -08:00
lbm_add_extension("canset-current", ext_can_current);
lbm_add_extension("canset-current-rel", ext_can_current_rel);
lbm_add_extension("canset-duty", ext_can_duty);
lbm_add_extension("canset-brake", ext_can_brake);
lbm_add_extension("canset-brake-rel", ext_can_brake_rel);
lbm_add_extension("canset-rpm", ext_can_rpm);
lbm_add_extension("canset-pos", ext_can_pos);
lbm_add_extension("canget-current", ext_can_get_current);
lbm_add_extension("canget-current-dir", ext_can_get_current_dir);
2022-02-13 10:20:54 -08:00
lbm_add_extension("canget-current-in", ext_can_get_current_in);
lbm_add_extension("canget-duty", ext_can_get_duty);
lbm_add_extension("canget-rpm", ext_can_get_rpm);
lbm_add_extension("canget-temp-fet", ext_can_get_temp_fet);
lbm_add_extension("canget-temp-motor", ext_can_get_temp_motor);
lbm_add_extension("canget-speed", ext_can_get_speed);
lbm_add_extension("canget-dist", ext_can_get_dist);
lbm_add_extension("canget-ppm", ext_can_get_ppm);
lbm_add_extension("canget-adc", ext_can_get_adc);
2022-02-13 10:20:54 -08:00
lbm_add_extension("can-list-devs", ext_can_list_devs);
lbm_add_extension("can-scan", ext_can_scan);
lbm_add_extension("can-send-sid", ext_can_send_sid);
lbm_add_extension("can-send-eid", ext_can_send_eid);
2022-01-16 07:21:18 -08:00
// Math
2022-01-21 01:52:56 -08:00
lbm_add_extension("sin", ext_sin);
lbm_add_extension("cos", ext_cos);
2022-02-14 10:19:48 -08:00
lbm_add_extension("tan", ext_tan);
lbm_add_extension("asin", ext_asin);
lbm_add_extension("acos", ext_acos);
2022-01-21 01:52:56 -08:00
lbm_add_extension("atan", ext_atan);
2022-02-05 15:23:23 -08:00
lbm_add_extension("atan2", ext_atan2);
2022-01-21 01:52:56 -08:00
lbm_add_extension("pow", ext_pow);
2022-02-06 14:12:05 -08:00
lbm_add_extension("sqrt", ext_sqrt);
2022-02-14 10:19:48 -08:00
lbm_add_extension("log", ext_log);
lbm_add_extension("log10", ext_log10);
lbm_add_extension("deg2rad", ext_deg2rad);
lbm_add_extension("rad2deg", ext_rad2deg);
lbm_add_extension("vec3-rot", ext_vec3_rot);
lbm_add_extension("throttle-curve", ext_throttle_curve);
// Bit operations
lbm_add_extension("bits-enc-int", ext_bits_enc_int);
lbm_add_extension("bits-dec-int", ext_bits_dec_int);
2022-02-06 14:12:05 -08:00
// Raw readings
lbm_add_extension("raw-adc-current", ext_raw_adc_current);
lbm_add_extension("raw-adc-voltage", ext_raw_adc_voltage);
lbm_add_extension("raw-mod-alpha", ext_raw_mod_alpha);
lbm_add_extension("raw-mod-beta", ext_raw_mod_beta);
lbm_add_extension("raw-mod-alpha-measured", ext_raw_mod_alpha_measured);
lbm_add_extension("raw-mod-beta-measured", ext_raw_mod_beta_measured);
lbm_add_extension("raw-hall", ext_raw_hall);
2022-02-21 17:37:23 -08:00
// UART
uart_started = false;
lbm_add_extension("uart-start", ext_uart_start);
lbm_add_extension("uart-write", ext_uart_write);
lbm_add_extension("uart-read", ext_uart_read);
2022-02-23 16:19:09 -08:00
// I2C
i2c_started = false;
lbm_add_extension("i2c-start", ext_i2c_start);
lbm_add_extension("i2c-tx-rx", ext_i2c_tx_rx);
lbm_add_extension("i2c-restore", ext_i2c_restore);
2022-03-13 14:53:25 -07:00
// GPIO
lbm_add_extension("gpio-configure", ext_gpio_configure);
lbm_add_extension("gpio-write", ext_gpio_write);
lbm_add_extension("gpio-read", ext_gpio_read);
2022-03-07 01:26:51 -08:00
// String manipulation
lbm_add_extension("str-from-n", ext_str_from_n);
lbm_add_extension("str-merge", ext_str_merge);
lbm_add_extension("str-to-i", ext_str_to_i);
lbm_add_extension("str-to-f", ext_str_to_f);
lbm_add_extension("str-part", ext_str_part);
lbm_add_extension("str-split", ext_str_split);
lbm_add_extension("str-replace", ext_str_replace);
lbm_add_extension("str-to-lower", ext_str_to_lower);
lbm_add_extension("str-to-upper", ext_str_to_upper);
lbm_add_extension("str-cmp", ext_str_cmp);
2022-11-09 07:00:43 -08:00
lbm_add_extension("to-str", ext_to_str);
2022-11-10 04:01:13 -08:00
lbm_add_extension("to-str-delim", ext_to_str_delim);
lbm_add_extension("str-len", ext_str_len);
2022-03-07 01:26:51 -08:00
2022-03-31 10:37:36 -07:00
// Configuration
lbm_add_extension("conf-set", ext_conf_set);
lbm_add_extension("conf-get", ext_conf_get);
lbm_add_extension("conf-store", ext_conf_store);
2022-05-06 10:49:58 -07:00
lbm_add_extension("conf-detect-foc", ext_conf_detect_foc);
lbm_add_extension("conf-set-pid-offset", ext_conf_set_pid_offset);
lbm_add_extension("conf-measure-res", ext_conf_measure_res);
2022-03-31 10:37:36 -07:00
2022-02-21 17:37:23 -08:00
// Array extensions
lbm_array_extensions_init();
2022-04-14 04:45:36 -07:00
lbm_add_extension("bufclear", ext_bufclear);
2022-04-18 09:06:21 -07:00
lbm_add_extension("bufcpy", ext_bufcpy);
lbm_add_extension("bufset-bit", ext_bufset_bit);
// Macro expanders
lbm_add_extension("me-defun", ext_me_defun);
lbm_add_extension("me-loopfor", ext_me_loopfor);
lbm_add_extension("me-loopwhile", ext_me_loopwhile);
lbm_add_extension("me-looprange", ext_me_looprange);
lbm_add_extension("me-loopforeach", ext_me_loopforeach);
// Native libraries
lbm_add_extension("load-native-lib", ext_load_native_lib);
lbm_add_extension("unload-native-lib", ext_unload_native_lib);
// UAVCAN
lbm_add_extension("uavcan-last-rawcmd", ext_uavcan_last_rawcmd);
lbm_add_extension("uavcan-last-rpmcmd", ext_uavcan_last_rpmcmd);
2022-09-04 13:18:15 -07:00
// Lbm settings
lbm_add_extension("lbm-set-quota", ext_lbm_set_quota);
2022-09-26 04:32:41 -07:00
// Plot
lbm_add_extension("plot-init", ext_plot_init);
lbm_add_extension("plot-add-graph", ext_plot_add_graph);
lbm_add_extension("plot-set-graph", ext_plot_set_graph);
lbm_add_extension("plot-send-points", ext_plot_send_points);
2022-10-14 04:48:17 -07:00
// IO-boards
lbm_add_extension("ioboard-get-adc", ext_ioboard_get_adc);
lbm_add_extension("ioboard-get-digital", ext_ioboard_get_digital);
lbm_add_extension("ioboard-set-digital", ext_ioboard_set_digital);
lbm_add_extension("ioboard-set-pwm", ext_ioboard_set_pwm);
// Logging
lbm_add_extension("log-start", ext_log_start);
lbm_add_extension("log-stop", ext_log_stop);
lbm_add_extension("log-config-field", ext_log_config_field);
lbm_add_extension("log-send-f32", ext_log_send_f32);
2022-11-09 08:10:07 -08:00
// GNSS
lbm_add_extension("gnss-lat-lon", ext_gnss_lat_lon);
lbm_add_extension("gnss-height", ext_gnss_height);
lbm_add_extension("gnss-speed", ext_gnss_speed);
lbm_add_extension("gnss-hdop", ext_gnss_hdop);
lbm_add_extension("gnss-date-time", ext_gnss_date_time);
lbm_add_extension("gnss-age", ext_gnss_age);
if (ext_callback) {
ext_callback();
}
2022-01-14 11:54:24 -08:00
}
2022-02-16 15:49:11 -08:00
void lispif_process_can(uint32_t can_id, uint8_t *data8, int len, bool is_ext) {
if (!event_can_sid_en && !is_ext) {
return;
}
if (!event_can_eid_en && is_ext) {
return;
}
2022-11-04 12:04:17 -07:00
ext_event e = {0};
2022-11-04 11:46:36 -07:00
e.type = EXT_EVENT_SYM_INT_ARRAY;
e.sym = is_ext ? sym_event_can_eid : sym_event_can_sid;
e.i = can_id;
event_add(e, data8, len);
2022-02-16 15:49:11 -08:00
}
void lispif_process_custom_app_data(unsigned char *data, unsigned int len) {
if (!event_data_rx_en) {
return;
}
2022-11-04 12:04:17 -07:00
ext_event e = {0};
2022-11-04 11:46:36 -07:00
e.type = EXT_EVENT_SYM_ARRAY;
e.sym = sym_event_data_rx;
event_add(e, data, len);
2022-02-16 15:49:11 -08:00
}
void lispif_process_shutdown(void) {
if (!event_shutdown_en) {
return;
2022-02-16 15:49:11 -08:00
}
2022-11-04 12:04:17 -07:00
ext_event e = {0};
2022-11-04 11:46:36 -07:00
e.type = EXT_EVENT_SYM;
e.sym = sym_event_shutdown;
event_add(e, NULL, 0);
2022-02-16 15:49:11 -08:00
}
void lispif_set_ext_load_callback(void (*p_func)(void)) {
ext_callback = p_func;
}
2022-02-16 15:49:11 -08:00
void lispif_disable_all_events(void) {
2022-05-10 10:05:43 -07:00
lispif_stop_lib();
2022-02-16 15:49:11 -08:00
event_handler_registered = false;
event_can_sid_en = false;
event_can_eid_en = false;
2022-11-04 11:46:36 -07:00
// Give thread a chance to stop
chThdSleepMilliseconds(5);
2022-02-16 15:49:11 -08:00
}
2022-11-10 11:36:46 -08:00
// Note: This function is only safe to use from LBM extensions or while the evaluator is paused
bool lispif_symbol_to_io(lbm_uint sym, stm32_gpio_t **port, uint32_t *pin) {
if (compare_symbol(sym, &syms_vesc.pin_rx)) {
#ifdef HW_UART_RX_PORT
*port = HW_UART_RX_PORT; *pin = HW_UART_RX_PIN;
return true;
#endif
} else if (compare_symbol(sym, &syms_vesc.pin_tx)) {
#ifdef HW_UART_TX_PORT
*port = HW_UART_TX_PORT; *pin = HW_UART_TX_PIN;
return true;
#endif
} else if (compare_symbol(sym, &syms_vesc.pin_swdio)) {
*port = GPIOA; *pin = 13;
return true;
} else if (compare_symbol(sym, &syms_vesc.pin_swclk)) {
*port = GPIOA; *pin = 14;
return true;
} else if (compare_symbol(sym, &syms_vesc.pin_hall1)) {
*port = HW_HALL_ENC_GPIO1; *pin = HW_HALL_ENC_PIN1;
return true;
} else if (compare_symbol(sym, &syms_vesc.pin_hall2)) {
*port = HW_HALL_ENC_GPIO2; *pin = HW_HALL_ENC_PIN2;
return true;
} else if (compare_symbol(sym, &syms_vesc.pin_hall3)) {
*port = HW_HALL_ENC_GPIO3; *pin = HW_HALL_ENC_PIN3;
return true;
} else if (compare_symbol(sym, &syms_vesc.pin_adc1)) {
#ifdef HW_ADC_EXT_GPIO
*port = HW_ADC_EXT_GPIO; *pin = HW_ADC_EXT_PIN;
return true;
#endif
} else if (compare_symbol(sym, &syms_vesc.pin_adc2)) {
#ifdef HW_ADC_EXT2_GPIO
*port = HW_ADC_EXT2_GPIO; *pin = HW_ADC_EXT2_PIN;
return true;
#endif
} else if (compare_symbol(sym, &syms_vesc.pin_ppm)) {
#ifdef HW_ICU_GPIO
*port = HW_ICU_GPIO; *pin = HW_ICU_PIN;
return true;
#endif
}
return false;
}