diff --git a/pycardium/meson.build b/pycardium/meson.build index 4d976e0a4ba020cc9a938be041fc467d107118d3..8d3b447a1750f369ff595d17271e43fc8e3d5707 100644 --- a/pycardium/meson.build +++ b/pycardium/meson.build @@ -4,6 +4,7 @@ modsrc = files( 'modules/buttons.c', 'modules/fat_file.c', 'modules/fat_reader_import.c', + 'modules/gpio.c', 'modules/interrupt.c', 'modules/sys_leds.c', 'modules/light_sensor.c', diff --git a/pycardium/modules/gpio.c b/pycardium/modules/gpio.c new file mode 100644 index 0000000000000000000000000000000000000000..8816c12406bd1b5cf1834e6778c18c0c6972df2a --- /dev/null +++ b/pycardium/modules/gpio.c @@ -0,0 +1,80 @@ +#include "py/obj.h" + +#include "epicardium.h" + +static mp_obj_t mp_gpio_set_mode(mp_obj_t pin_obj, mp_obj_t mode_obj) +{ + int pin = mp_obj_get_int(pin_obj); + int mode = mp_obj_get_int(mode_obj); + epic_gpio_set_pin_mode(pin, mode); + return mp_const_none; +}; +static MP_DEFINE_CONST_FUN_OBJ_2(gpio_set_mode, mp_gpio_set_mode); + +static mp_obj_t mp_gpio_get_mode(mp_obj_t pin_obj) +{ + int pin = mp_obj_get_int(pin_obj); + uint8_t mode = epic_gpio_get_pin_mode(pin); + return MP_OBJ_NEW_SMALL_INT(mode); +}; +static MP_DEFINE_CONST_FUN_OBJ_1(gpio_get_mode, mp_gpio_get_mode); + +static mp_obj_t mp_gpio_read(mp_obj_t pin_obj) +{ + int pin = mp_obj_get_int(pin_obj); + uint32_t value = epic_gpio_read_pin(pin); + return MP_OBJ_NEW_SMALL_INT(value); +}; +static MP_DEFINE_CONST_FUN_OBJ_1(gpio_read, mp_gpio_read); + +static mp_obj_t mp_gpio_write(mp_obj_t pin_obj, mp_obj_t on_obj) +{ + int pin = mp_obj_get_int(pin_obj); + bool on = mp_obj_is_true(on_obj); + epic_gpio_write_pin(pin, on); + return mp_const_none; +}; +static MP_DEFINE_CONST_FUN_OBJ_2(gpio_write, mp_gpio_write); + +static const mp_rom_map_elem_t gpio_module_modes_table[] = { + { MP_ROM_QSTR(MP_QSTR___name__), MP_ROM_QSTR(MP_QSTR_mode) }, + { MP_ROM_QSTR(MP_QSTR_INPUT), MP_OBJ_NEW_SMALL_INT(GPIO_MODE_IN) }, + { MP_ROM_QSTR(MP_QSTR_OUTPUT), MP_OBJ_NEW_SMALL_INT(GPIO_MODE_OUT) }, + { MP_ROM_QSTR(MP_QSTR_PULL_UP), MP_OBJ_NEW_SMALL_INT(GPIO_PULL_UP) }, + { MP_ROM_QSTR(MP_QSTR_PULL_DOWN), + MP_OBJ_NEW_SMALL_INT(GPIO_PULL_DOWN) }, +}; +static MP_DEFINE_CONST_DICT(gpio_module_modes_dict, gpio_module_modes_table); + +const mp_obj_module_t gpio_module_modes = { + .base = { &mp_type_module }, + .globals = (mp_obj_dict_t *)&gpio_module_modes_dict, +}; + +/* The globals table for this module */ +static const mp_rom_map_elem_t gpio_module_globals_table[] = { + { MP_ROM_QSTR(MP_QSTR___name__), MP_ROM_QSTR(MP_QSTR_gpio) }, + { MP_ROM_QSTR(MP_QSTR_set_mode), MP_ROM_PTR(&gpio_set_mode) }, + { MP_ROM_QSTR(MP_QSTR_get_mode), MP_ROM_PTR(&gpio_get_mode) }, + { MP_ROM_QSTR(MP_QSTR_read), MP_ROM_PTR(&gpio_read) }, + { MP_ROM_QSTR(MP_QSTR_write), MP_ROM_PTR(&gpio_write) }, + { MP_ROM_QSTR(MP_QSTR_WRISTBAND_1), + MP_OBJ_NEW_SMALL_INT(GPIO_WRISTBAND_1) }, + { MP_ROM_QSTR(MP_QSTR_WRISTBAND_2), + MP_OBJ_NEW_SMALL_INT(GPIO_WRISTBAND_2) }, + { MP_ROM_QSTR(MP_QSTR_WRISTBAND_3), + MP_OBJ_NEW_SMALL_INT(GPIO_WRISTBAND_3) }, + { MP_ROM_QSTR(MP_QSTR_WRISTBAND_4), + MP_OBJ_NEW_SMALL_INT(GPIO_WRISTBAND_4) }, + { MP_ROM_QSTR(MP_QSTR_mode), MP_ROM_PTR(&gpio_module_modes) }, +}; +static MP_DEFINE_CONST_DICT(gpio_module_globals, gpio_module_globals_table); + +const mp_obj_module_t gpio_module = { + .base = { &mp_type_module }, + .globals = (mp_obj_dict_t *)&gpio_module_globals, +}; + +/* This is a special macro that will make MicroPython aware of this module */ +/* clang-format off */ +MP_REGISTER_MODULE(MP_QSTR_gpio, gpio_module, MODULE_GPIO_ENABLED); diff --git a/pycardium/modules/qstrdefs.h b/pycardium/modules/qstrdefs.h index e1bb0a02f0e69da49c6ab6b968b42437c2056797..49ad16f40b8dd7ae88e4a756be3270739b7625d8 100644 --- a/pycardium/modules/qstrdefs.h +++ b/pycardium/modules/qstrdefs.h @@ -100,3 +100,18 @@ Q(exit) Q(exec) Q(listdir) Q(unlink) + +/* gpio */ +Q(gpio) +Q(set_mode) +Q(get_mode) +Q(read) +Q(write) +Q(WRISTBAND_1) +Q(WRISTBAND_2) +Q(WRISTBAND_3) +Q(WRISTBAND_4) +Q(INPUT) +Q(OUTPUT) +Q(PULL_UP) +Q(PULL_DOWN) diff --git a/pycardium/mpconfigport.h b/pycardium/mpconfigport.h index 00430b2e457276b1eea295cb91d383e8c1007962..0708ebe52ad745d888ac5fa0cbace2b7fc655558 100644 --- a/pycardium/mpconfigport.h +++ b/pycardium/mpconfigport.h @@ -41,6 +41,7 @@ /* Modules */ #define MODULE_BUTTONS_ENABLED (1) #define MODULE_DISPLAY_ENABLED (1) +#define MODULE_GPIO_ENABLED (1) #define MODULE_INTERRUPT_ENABLED (1) #define MODULE_LEDS_ENABLED (1) #define MODULE_LIGHT_SENSOR_ENABLED (1)