blackmagic-esp32-s2/main/cli/cli-commands-gpio.c

281 lines
7.7 KiB
C

#include <driver/gpio.h>
#include "cli.h"
#include "cli-args.h"
#include "cli-commands.h"
#include "helpers.h"
typedef struct {
const char* name;
const gpio_num_t gpio_num;
const bool get;
const bool set;
} GPIOItem;
#define MAKE_IO(pin_index) \
{ .name = "IO" #pin_index, .gpio_num = GPIO_NUM_##pin_index, .get = true, .set = true }
#define MAKE_IO_NAMED(pin_name, pin_index) \
{ .name = pin_name, .gpio_num = GPIO_NUM_##pin_index, .get = true, .set = true }
#define MAKE_IO_READONLY(pin_index) \
{ .name = "IO" #pin_index, .gpio_num = GPIO_NUM_##pin_index, .get = true, .set = false }
#define MAKE_IO_NAMED_READONLY(pin_name, pin_index) \
{ .name = pin_name, .gpio_num = GPIO_NUM_##pin_index, .get = true, .set = false }
static const GPIOItem gpio_items[] = {
//BOOT
MAKE_IO_READONLY(0),
MAKE_IO_NAMED_READONLY("BOOT", 0),
// SWD
MAKE_IO(1),
MAKE_IO_NAMED("SWCLK", 1),
MAKE_IO(2),
MAKE_IO_NAMED("SWDIO", 2),
// I2C_PULL_UP
MAKE_IO(3),
MAKE_IO_NAMED("I2C_PULL_UP", 3),
// I2C_PULL_UP
MAKE_IO(4),
MAKE_IO_NAMED("LED_BLUE", 4),
MAKE_IO(5),
MAKE_IO_NAMED("LED_GREEN", 5),
MAKE_IO(6),
MAKE_IO_NAMED("LED_RED", 6),
// GPIO
MAKE_IO(7),
MAKE_IO(8),
MAKE_IO(9),
// SPI
MAKE_IO(10),
MAKE_IO_NAMED("SWO", 10),
MAKE_IO_NAMED("SPI_CS", 10),
MAKE_IO(11),
MAKE_IO_NAMED("SPI_MOSI", 11),
MAKE_IO(12),
MAKE_IO_NAMED("SPI_CLK", 12),
MAKE_IO(13),
MAKE_IO_NAMED("SPI_MISO", 13),
// GPIO
MAKE_IO(14),
MAKE_IO(15),
MAKE_IO(16),
// CLI UART
MAKE_IO(17),
MAKE_IO_NAMED("CLI_TXD", 17),
MAKE_IO(18),
MAKE_IO_NAMED("CLI_RXD", 18),
// GPIO
MAKE_IO(21),
MAKE_IO(33),
MAKE_IO(34),
MAKE_IO(35),
MAKE_IO(36),
MAKE_IO(37),
// JTAG
MAKE_IO(38),
MAKE_IO_NAMED("JTAG_SRST", 38),
MAKE_IO(39),
MAKE_IO_NAMED("JTAG_TCK", 39),
MAKE_IO(40),
MAKE_IO_NAMED("JTAG_TDI", 40),
MAKE_IO(41),
MAKE_IO_NAMED("JTAG_TDO", 41),
MAKE_IO(42),
MAKE_IO_NAMED("JTAG_TMS", 41),
};
static void cli_gpio_print_list_set(Cli* cli) {
cli_write_str(cli, "Pin names: ");
for(size_t i = 0; i < COUNT_OF(gpio_items); i++) {
if(gpio_items[i].set) {
cli_write_str(cli, gpio_items[i].name);
if((i + 1) < COUNT_OF(gpio_items)) {
cli_write_str(cli, " ");
}
}
}
}
static gpio_num_t gpio_search_set(mstring_t* gpio_name) {
gpio_num_t gpio_num = GPIO_NUM_NC;
for(size_t i = 0; i < COUNT_OF(gpio_items); i++) {
if(mstring_cmp_cstr(gpio_name, gpio_items[i].name) == 0 && gpio_items[i].set) {
gpio_num = gpio_items[i].gpio_num;
break;
}
}
return gpio_num;
}
static void cli_gpio_print_list_get(Cli* cli) {
cli_write_str(cli, "Pin names: ");
for(size_t i = 0; i < COUNT_OF(gpio_items); i++) {
if(gpio_items[i].get) {
cli_write_str(cli, gpio_items[i].name);
if((i + 1) < COUNT_OF(gpio_items)) {
cli_write_str(cli, " ");
}
}
}
}
static gpio_num_t gpio_search_get(mstring_t* gpio_name) {
gpio_num_t gpio_num = GPIO_NUM_NC;
for(size_t i = 0; i < COUNT_OF(gpio_items); i++) {
if(mstring_cmp_cstr(gpio_name, gpio_items[i].name) == 0 && gpio_items[i].get) {
gpio_num = gpio_items[i].gpio_num;
break;
}
}
return gpio_num;
}
void cli_gpio_set_print_usage(Cli* cli) {
cli_write_str(cli, "Usage: gpio_set <pin_names> <value>");
cli_write_eol(cli);
cli_gpio_print_list_set(cli);
cli_write_eol(cli);
cli_write_str(cli, "Examples:");
cli_write_eol(cli);
cli_write_str(cli, " gpio_set IO8 1");
cli_write_eol(cli);
cli_write_str(cli, " gpio_set IO8 IO10 SPI_MOSI 1");
}
void cli_gpio_get_print_usage(Cli* cli) {
cli_write_str(cli, "Usage: gpio_get <pin_names>");
cli_write_eol(cli);
cli_gpio_print_list_get(cli);
cli_write_eol(cli);
cli_write_str(cli, "Examples:");
cli_write_eol(cli);
cli_write_str(cli, " gpio_get IO8");
cli_write_eol(cli);
cli_write_str(cli, " gpio_get IO8 IO10 SPI_MOSI");
}
void cli_gpio_set(Cli* cli, mstring_t* args) {
mstring_t* gpio_name = mstring_alloc();
int gpio_value = 0;
size_t argc = mstring_count_char(args, ' ') + 1;
do {
if(argc < 2) {
cli_gpio_set_print_usage(cli);
break;
}
gpio_config_t io_conf;
io_conf.intr_type = GPIO_PIN_INTR_DISABLE;
io_conf.mode = GPIO_MODE_OUTPUT;
io_conf.pin_bit_mask = 0;
io_conf.pull_down_en = 0;
io_conf.pull_up_en = 0;
bool error = false;
for(size_t i = 0; i < (argc - 1); i++) {
if(cli_args_read_string_and_trim(args, gpio_name)) {
gpio_num_t gpio_num = gpio_search_set(gpio_name);
if(gpio_num == GPIO_NUM_NC) {
cli_printf(cli, "Invalid <pin_name> %s", mstring_get_cstr(gpio_name));
cli_write_eol(cli);
error = true;
break;
} else {
io_conf.pin_bit_mask |= (1LL << (uint64_t)gpio_num);
}
} else {
error = true;
break;
}
}
if(!cli_args_read_int_and_trim(args, &gpio_value) || error) {
cli_gpio_set_print_usage(cli);
break;
}
gpio_config(&io_conf);
for(size_t i = 0; i < 64; i++) {
if((io_conf.pin_bit_mask & (1LL << (uint64_t)i)) > 0) {
gpio_set_level(i, (gpio_value > 0));
}
}
cli_write_str(cli, "OK");
} while(false);
mstring_free(gpio_name);
}
void cli_gpio_get(Cli* cli, mstring_t* args) {
mstring_t* gpio_name = mstring_alloc();
size_t argc = mstring_count_char(args, ' ') + 1;
do {
if(argc < 1) {
cli_gpio_get_print_usage(cli);
break;
}
gpio_config_t io_conf;
io_conf.intr_type = GPIO_PIN_INTR_DISABLE;
io_conf.mode = GPIO_MODE_INPUT;
io_conf.pin_bit_mask = 0;
io_conf.pull_down_en = 0;
io_conf.pull_up_en = 0;
bool error = false;
gpio_num_t last_gpio_num = GPIO_NUM_NC;
for(size_t i = 0; i < argc; i++) {
if(cli_args_read_string_and_trim(args, gpio_name)) {
gpio_num_t gpio_num = gpio_search_get(gpio_name);
if(gpio_num == GPIO_NUM_NC) {
cli_printf(cli, "Invalid <pin_name> %s", mstring_get_cstr(gpio_name));
cli_write_eol(cli);
error = true;
break;
} else {
if(gpio_num <= last_gpio_num) {
cli_printf(
cli,
"<pin_name> %s is less than or equal to the previous",
mstring_get_cstr(gpio_name));
cli_write_eol(cli);
error = true;
break;
} else {
last_gpio_num = gpio_num;
io_conf.pin_bit_mask |= (1LL << (uint64_t)gpio_num);
}
}
} else {
error = true;
break;
}
}
if(error) {
cli_gpio_get_print_usage(cli);
break;
}
gpio_config(&io_conf);
for(size_t i = 0; i < 64; i++) {
if((io_conf.pin_bit_mask & (1LL << (uint64_t)i)) > 0) {
cli_printf(cli, "%d ", gpio_get_level(i));
}
}
} while(false);
mstring_free(gpio_name);
}