unleashed-firmware/applications/system/js_app/modules/js_gpio.c
2024-07-15 20:02:45 +03:00

396 lines
12 KiB
C

#include "../js_modules.h"
#include <furi_hal_gpio.h>
#include <furi_hal_resources.h>
#include <expansion/expansion.h>
typedef struct {
FuriHalAdcHandle* handle;
} JsGpioInst;
typedef struct {
const GpioPin* pin;
const char* name;
const FuriHalAdcChannel channel;
} GpioPinCtx;
static const GpioPinCtx js_gpio_pins[] = {
{.pin = &gpio_ext_pa7, .name = "PA7", .channel = FuriHalAdcChannel12}, // 2
{.pin = &gpio_ext_pa6, .name = "PA6", .channel = FuriHalAdcChannel11}, // 3
{.pin = &gpio_ext_pa4, .name = "PA4", .channel = FuriHalAdcChannel9}, // 4
{.pin = &gpio_ext_pb3, .name = "PB3", .channel = FuriHalAdcChannelNone}, // 5
{.pin = &gpio_ext_pb2, .name = "PB2", .channel = FuriHalAdcChannelNone}, // 6
{.pin = &gpio_ext_pc3, .name = "PC3", .channel = FuriHalAdcChannel4}, // 7
{.pin = &gpio_swclk, .name = "PA14", .channel = FuriHalAdcChannelNone}, // 10
{.pin = &gpio_swdio, .name = "PA13", .channel = FuriHalAdcChannelNone}, // 12
{.pin = &gpio_usart_tx, .name = "PB6", .channel = FuriHalAdcChannelNone}, // 13
{.pin = &gpio_usart_rx, .name = "PB7", .channel = FuriHalAdcChannelNone}, // 14
{.pin = &gpio_ext_pc1, .name = "PC1", .channel = FuriHalAdcChannel2}, // 15
{.pin = &gpio_ext_pc0, .name = "PC0", .channel = FuriHalAdcChannel1}, // 16
{.pin = &gpio_ibutton, .name = "PB14", .channel = FuriHalAdcChannelNone}, // 17
};
bool js_gpio_get_gpio_pull(const char* pull, GpioPull* value) {
if(strcmp(pull, "no") == 0) {
*value = GpioPullNo;
return true;
} else if(strcmp(pull, "up") == 0) {
*value = GpioPullUp;
return true;
} else if(strcmp(pull, "down") == 0) {
*value = GpioPullDown;
return true;
} else {
*value = GpioPullNo;
return true;
}
return false;
}
bool js_gpio_get_gpio_mode(const char* mode, GpioMode* value) {
if(strcmp(mode, "input") == 0) {
*value = GpioModeInput;
return true;
} else if(strcmp(mode, "outputPushPull") == 0) {
*value = GpioModeOutputPushPull;
return true;
} else if(strcmp(mode, "outputOpenDrain") == 0) {
*value = GpioModeOutputOpenDrain;
return true;
} else if(strcmp(mode, "altFunctionPushPull") == 0) {
*value = GpioModeAltFunctionPushPull;
return true;
} else if(strcmp(mode, "altFunctionOpenDrain") == 0) {
*value = GpioModeAltFunctionOpenDrain;
return true;
} else if(strcmp(mode, "analog") == 0) {
*value = GpioModeAnalog;
return true;
} else if(strcmp(mode, "interruptRise") == 0) {
*value = GpioModeInterruptRise;
return true;
} else if(strcmp(mode, "interruptFall") == 0) {
*value = GpioModeInterruptFall;
return true;
} else if(strcmp(mode, "interruptRiseFall") == 0) {
*value = GpioModeInterruptRiseFall;
return true;
} else if(strcmp(mode, "eventRise") == 0) {
*value = GpioModeEventRise;
return true;
} else if(strcmp(mode, "eventFall") == 0) {
*value = GpioModeEventFall;
return true;
} else if(strcmp(mode, "eventRiseFall") == 0) {
*value = GpioModeEventRiseFall;
return true;
} else {
return false;
}
}
const GpioPin* js_gpio_get_gpio_pin(const char* name) {
for(size_t i = 0; i < COUNT_OF(js_gpio_pins); i++) {
if(strcmp(js_gpio_pins[i].name, name) == 0) {
return js_gpio_pins[i].pin;
}
}
return NULL;
}
FuriHalAdcChannel js_gpio_get_gpio_channel(const char* name) {
for(size_t i = 0; i < COUNT_OF(js_gpio_pins); i++) {
if(strcmp(js_gpio_pins[i].name, name) == 0) {
return js_gpio_pins[i].channel;
}
}
return FuriHalAdcChannelNone;
}
static void js_gpio_init(struct mjs* mjs) {
mjs_val_t pin_arg = mjs_arg(mjs, 0);
mjs_val_t mode_arg = mjs_arg(mjs, 1);
mjs_val_t pull_arg = mjs_arg(mjs, 2);
if(!mjs_is_string(pin_arg)) {
mjs_prepend_errorf(mjs, MJS_BAD_ARGS_ERROR, "Argument must be a string");
mjs_return(mjs, MJS_UNDEFINED);
return;
}
const char* pin_name = mjs_get_string(mjs, &pin_arg, NULL);
if(!pin_name) {
mjs_prepend_errorf(mjs, MJS_BAD_ARGS_ERROR, "Failed to get pin name");
mjs_return(mjs, MJS_UNDEFINED);
return;
}
if(!mjs_is_string(mode_arg)) {
mjs_prepend_errorf(mjs, MJS_BAD_ARGS_ERROR, "Argument must be a string");
mjs_return(mjs, MJS_UNDEFINED);
return;
}
const char* mode_name = mjs_get_string(mjs, &mode_arg, NULL);
if(!mode_name) {
mjs_prepend_errorf(mjs, MJS_BAD_ARGS_ERROR, "Failed to get mode name");
mjs_return(mjs, MJS_UNDEFINED);
return;
}
if(!mjs_is_string(pull_arg)) {
mjs_prepend_errorf(mjs, MJS_BAD_ARGS_ERROR, "Argument must be a string");
mjs_return(mjs, MJS_UNDEFINED);
return;
}
const char* pull_name = mjs_get_string(mjs, &pull_arg, NULL);
if(!pull_name) {
mjs_prepend_errorf(mjs, MJS_BAD_ARGS_ERROR, "Failed to get pull name");
mjs_return(mjs, MJS_UNDEFINED);
return;
}
const GpioPin* gpio_pin = js_gpio_get_gpio_pin(pin_name);
if(gpio_pin == NULL) {
mjs_prepend_errorf(mjs, MJS_BAD_ARGS_ERROR, "Invalid pin name");
mjs_return(mjs, MJS_UNDEFINED);
return;
}
GpioMode gpio_mode;
if(!js_gpio_get_gpio_mode(mode_name, &gpio_mode)) {
mjs_prepend_errorf(mjs, MJS_BAD_ARGS_ERROR, "Invalid mode name");
mjs_return(mjs, MJS_UNDEFINED);
return;
}
GpioPull gpio_pull;
if(!js_gpio_get_gpio_pull(pull_name, &gpio_pull)) {
mjs_prepend_errorf(mjs, MJS_BAD_ARGS_ERROR, "Invalid pull name");
mjs_return(mjs, MJS_UNDEFINED);
return;
}
expansion_disable(furi_record_open(RECORD_EXPANSION));
furi_record_close(RECORD_EXPANSION);
furi_hal_gpio_init(gpio_pin, gpio_mode, gpio_pull, GpioSpeedVeryHigh);
mjs_return(mjs, MJS_UNDEFINED);
}
static void js_gpio_write(struct mjs* mjs) {
mjs_val_t pin_arg = mjs_arg(mjs, 0);
mjs_val_t value_arg = mjs_arg(mjs, 1);
if(!mjs_is_string(pin_arg)) {
mjs_prepend_errorf(mjs, MJS_BAD_ARGS_ERROR, "Argument must be a string");
mjs_return(mjs, MJS_UNDEFINED);
return;
}
const char* pin_name = mjs_get_string(mjs, &pin_arg, NULL);
if(!pin_name) {
mjs_prepend_errorf(mjs, MJS_BAD_ARGS_ERROR, "Failed to get pin name");
mjs_return(mjs, MJS_UNDEFINED);
return;
}
if(!mjs_is_boolean(value_arg)) {
mjs_prepend_errorf(mjs, MJS_BAD_ARGS_ERROR, "Argument must be a boolean");
mjs_return(mjs, MJS_UNDEFINED);
return;
}
bool value = mjs_get_bool(mjs, value_arg);
const GpioPin* gpio_pin = js_gpio_get_gpio_pin(pin_name);
if(gpio_pin == NULL) {
mjs_prepend_errorf(mjs, MJS_BAD_ARGS_ERROR, "Invalid pin name");
mjs_return(mjs, MJS_UNDEFINED);
return;
}
furi_hal_gpio_write(gpio_pin, value);
mjs_return(mjs, MJS_UNDEFINED);
}
static void js_gpio_read(struct mjs* mjs) {
mjs_val_t pin_arg = mjs_arg(mjs, 0);
if(!mjs_is_string(pin_arg)) {
mjs_prepend_errorf(mjs, MJS_BAD_ARGS_ERROR, "Argument must be a string");
mjs_return(mjs, MJS_UNDEFINED);
return;
}
const char* pin_name = mjs_get_string(mjs, &pin_arg, NULL);
if(!pin_name) {
mjs_prepend_errorf(mjs, MJS_BAD_ARGS_ERROR, "Failed to get pin name");
mjs_return(mjs, MJS_UNDEFINED);
return;
}
const GpioPin* gpio_pin = js_gpio_get_gpio_pin(pin_name);
if(gpio_pin == NULL) {
mjs_prepend_errorf(mjs, MJS_BAD_ARGS_ERROR, "Invalid pin name");
mjs_return(mjs, MJS_UNDEFINED);
return;
}
bool value = furi_hal_gpio_read(gpio_pin);
mjs_return(mjs, mjs_mk_boolean(mjs, value));
}
static void js_gpio_read_analog(struct mjs* mjs) {
mjs_val_t obj_inst = mjs_get(mjs, mjs_get_this(mjs), INST_PROP_NAME, ~0);
JsGpioInst* gpio = mjs_get_ptr(mjs, obj_inst);
furi_assert(gpio);
if(gpio->handle == NULL) {
mjs_prepend_errorf(mjs, MJS_INTERNAL_ERROR, "Analog mode not started");
mjs_return(mjs, MJS_UNDEFINED);
return;
}
mjs_val_t pin_arg = mjs_arg(mjs, 0);
if(!mjs_is_string(pin_arg)) {
mjs_prepend_errorf(mjs, MJS_BAD_ARGS_ERROR, "Argument must be a string");
mjs_return(mjs, MJS_UNDEFINED);
return;
}
const char* pin_name = mjs_get_string(mjs, &pin_arg, NULL);
if(!pin_name) {
mjs_prepend_errorf(mjs, MJS_BAD_ARGS_ERROR, "Failed to get pin name");
mjs_return(mjs, MJS_UNDEFINED);
return;
}
FuriHalAdcChannel channel = js_gpio_get_gpio_channel(pin_name);
if(channel == FuriHalAdcChannelNone) {
mjs_prepend_errorf(mjs, MJS_BAD_ARGS_ERROR, "Invalid pin name");
mjs_return(mjs, MJS_UNDEFINED);
return;
}
uint16_t adc_value = furi_hal_adc_read(gpio->handle, channel);
float adc_mv = furi_hal_adc_convert_to_voltage(gpio->handle, adc_value);
mjs_return(mjs, mjs_mk_number(mjs, adc_mv));
}
static void js_gpio_start_analog(struct mjs* mjs) {
mjs_val_t obj_inst = mjs_get(mjs, mjs_get_this(mjs), INST_PROP_NAME, ~0);
JsGpioInst* gpio = mjs_get_ptr(mjs, obj_inst);
furi_assert(gpio);
FuriHalAdcScale scale = FuriHalAdcScale2048;
if(mjs_nargs(mjs) > 0) {
mjs_val_t scale_arg = mjs_arg(mjs, 0);
if(!mjs_is_number(scale_arg)) {
mjs_prepend_errorf(mjs, MJS_BAD_ARGS_ERROR, "Argument must be a number");
mjs_return(mjs, MJS_UNDEFINED);
return;
}
int32_t scale_num = mjs_get_int32(mjs, scale_arg);
if(scale_num == 2048 || scale_num == 2000) { // 2 volt reference
scale = FuriHalAdcScale2048;
} else if(scale_num == 2500) { // 2.5 volt reference
scale = FuriHalAdcScale2500;
} else {
mjs_prepend_errorf(mjs, MJS_BAD_ARGS_ERROR, "Invalid scale");
mjs_return(mjs, MJS_UNDEFINED);
return;
}
}
if(gpio->handle != NULL) {
mjs_prepend_errorf(mjs, MJS_INTERNAL_ERROR, "Analog mode already started");
mjs_return(mjs, MJS_UNDEFINED);
return;
}
gpio->handle = furi_hal_adc_acquire();
furi_hal_adc_configure_ex(
gpio->handle,
scale,
FuriHalAdcClockSync64,
FuriHalAdcOversample64,
FuriHalAdcSamplingtime247_5);
}
static void js_gpio_stop_analog(struct mjs* mjs) {
mjs_val_t obj_inst = mjs_get(mjs, mjs_get_this(mjs), INST_PROP_NAME, ~0);
JsGpioInst* gpio = mjs_get_ptr(mjs, obj_inst);
furi_assert(gpio);
if(gpio->handle == NULL) {
mjs_prepend_errorf(mjs, MJS_INTERNAL_ERROR, "Analog mode not started");
mjs_return(mjs, MJS_UNDEFINED);
return;
}
furi_hal_adc_release(gpio->handle);
gpio->handle = NULL;
}
static void* js_gpio_create(struct mjs* mjs, mjs_val_t* object) {
JsGpioInst* gpio = malloc(sizeof(JsGpioInst));
gpio->handle = NULL;
mjs_val_t gpio_obj = mjs_mk_object(mjs);
mjs_set(mjs, gpio_obj, INST_PROP_NAME, ~0, mjs_mk_foreign(mjs, gpio));
mjs_set(mjs, gpio_obj, "init", ~0, MJS_MK_FN(js_gpio_init));
mjs_set(mjs, gpio_obj, "write", ~0, MJS_MK_FN(js_gpio_write));
mjs_set(mjs, gpio_obj, "read", ~0, MJS_MK_FN(js_gpio_read));
mjs_set(mjs, gpio_obj, "readAnalog", ~0, MJS_MK_FN(js_gpio_read_analog));
mjs_set(mjs, gpio_obj, "startAnalog", ~0, MJS_MK_FN(js_gpio_start_analog));
mjs_set(mjs, gpio_obj, "stopAnalog", ~0, MJS_MK_FN(js_gpio_stop_analog));
*object = gpio_obj;
return (void*)gpio;
}
static void js_gpio_destroy(void* inst) {
if(inst != NULL) {
JsGpioInst* gpio = (JsGpioInst*)inst;
if(gpio->handle != NULL) {
furi_hal_adc_release(gpio->handle);
gpio->handle = NULL;
}
free(gpio);
}
// loop through all pins and reset them to analog mode
for(size_t i = 0; i < COUNT_OF(js_gpio_pins); i++) {
furi_hal_gpio_write(js_gpio_pins[i].pin, false);
furi_hal_gpio_init(js_gpio_pins[i].pin, GpioModeAnalog, GpioPullNo, GpioSpeedVeryHigh);
}
expansion_enable(furi_record_open(RECORD_EXPANSION));
furi_record_close(RECORD_EXPANSION);
}
static const JsModuleDescriptor js_gpio_desc = {
"gpio",
js_gpio_create,
js_gpio_destroy,
};
static const FlipperAppPluginDescriptor plugin_descriptor = {
.appid = PLUGIN_APP_ID,
.ep_api_version = PLUGIN_API_VERSION,
.entry_point = &js_gpio_desc,
};
const FlipperAppPluginDescriptor* js_gpio_ep(void) {
return &plugin_descriptor;
}