123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226 |
- /*
- * console.c
- */
-
- #include <inttypes.h>
- #include <string.h>
- #include "pico/stdlib.h"
-
- #include "config.h"
- #include "log.h"
- #include "pmw3360.h"
- #include "util.h"
- #include "usb_msc.h"
- #include "debug.h"
- #include "console.h"
-
- #define CNSL_BUFF_SIZE 1024
- #define CNSL_REPEAT_MS 500
-
- //#define CNSL_REPEAT_PMW_STATUS_BY_DEFAULT
-
- static char cnsl_line_buff[CNSL_BUFF_SIZE + 1];
- static uint32_t cnsl_buff_pos = 0;
-
- static char cnsl_last_command[CNSL_BUFF_SIZE + 1];
-
- static char cnsl_repeated_command[CNSL_BUFF_SIZE + 1];
- static bool repeat_command = false;
- static uint32_t last_repeat_time = 0;
-
- static void cnsl_interpret(const char *line) {
- if (strlen(line) == 0) {
- if ((strlen(cnsl_last_command) > 0) && (strcmp(cnsl_last_command, "repeat") != 0)) {
- // repeat last command once
- println("repeating command \"%s\"", cnsl_last_command);
- cnsl_interpret(cnsl_last_command);
- println();
- }
- return;
- } else if (strcmp(line, "repeat") == 0) {
- if (!repeat_command) {
- // mark last command to be repeated multiple times
- strncpy(cnsl_repeated_command, cnsl_last_command, CNSL_BUFF_SIZE + 1);
- last_repeat_time = to_ms_since_boot(get_absolute_time()) - 1001;
- repeat_command = true;
- } else {
- // stop repeating
- repeat_command = false;
- }
- } else if ((strcmp(line, "help") == 0)
- || (strcmp(line, "h") == 0)
- || (strcmp(line, "?") == 0)) {
- println("Trackball Firmware Usage:");
- println(" cpi - print current sensitivity");
- println(" cpi N - set sensitivity");
- println(" pmws - print PMW3360 status");
- println(" pmwf - print PMW3360 frame capture");
- println(" pmwd - print PMW3360 data dump");
- println(" pmwr - reset PMW3360");
- println(" reset - reset back into this firmware");
- println(" \\x18 - reset to bootloader");
- println(" repeat - repeat last command every %d milliseconds", CNSL_REPEAT_MS);
- println(" help - print this message");
- println(" stats - put statistics on mass storage medium");
- println(" data - put PMW3360 data on mass storage medium");
- println(" mount - make mass storage medium (un)available");
- println("Press Enter with no input to repeat last command.");
- println("Use repeat to continuously execute last command.");
- println("Stop this by calling repeat again.");
- } else if (strcmp(line, "pmws") == 0) {
- char status_buff[1024];
- pmw_print_status(status_buff, sizeof(status_buff));
- print("%s", status_buff);
- } else if (strcmp(line, "pmwd") == 0) {
- pmw_dump_data(true);
- } else if (strcmp(line, "pmwf") == 0) {
- uint8_t frame[PMW_FRAME_CAPTURE_LEN];
- ssize_t r = pmw_frame_capture(frame, PMW_FRAME_CAPTURE_LEN);
- if (r == PMW_FRAME_CAPTURE_LEN) {
- println("PMW3360 frame capture:");
- hexdump(frame, PMW_FRAME_CAPTURE_LEN);
-
- println("Re-Initializing PMW3360");
- pmw_init();
- } else {
- println("error capturing frame (%d)", r);
- }
- } else if (strcmp(line, "pmwr") == 0) {
- println("user requests re-initializing of PMW3360");
- int r = pmw_init();
- if (r < 0) {
- println("error initializing PMW3360");
- } else {
- println("PMW3360 re-initialized successfully");
- }
- } else if (strcmp(line, "cpi") == 0) {
- uint8_t sense = pmw_get_sensitivity();
- uint16_t cpi = PMW_SENSE_TO_CPI(sense);
- println("current cpi: %u (0x%02X)", cpi, sense);
- } else if (str_startswith(line, "cpi ")) {
- const char *num_str = line + 4;
- uintmax_t num = strtoumax(num_str, NULL, 10);
- if ((num < 100) || (num > 12000)) {
- println("invalid cpi %llu, needs to be %u <= cpi <= %u", num, 100, 12000);
- } else {
- num = PMW_CPI_TO_SENSE(num);
- println("setting cpi to 0x%02llX", num);
- pmw_set_sensitivity(num);
- }
- } else if (strcmp(line, "reset") == 0) {
- reset_to_main();
- } else if ((strcmp(line, "stats") == 0) || (strcmp(line, "data") == 0)) {
- if (msc_is_medium_available()) {
- println("Currently mounted. Unplugging now.");
- msc_set_medium_available(false);
- }
-
- if (debug_msc_mount() != 0) {
- println("Error mounting file system.");
- println();
- return;
- }
-
- println("Writing data to file system");
-
- if (strcmp(line, "data") == 0) {
- debug_msc_pmw3360();
- }
-
- debug_msc_stats();
-
- if (debug_msc_unmount() != 0) {
- println("Error unmounting file system.");
- }
-
- println("Done. Plugging in now.");
- msc_set_medium_available(true);
- } else if (strcmp(line, "mount") == 0) {
- bool state = msc_is_medium_available();
- println("Currently %s. %s now.",
- state ? "mounted" : "unmounted",
- state ? "Unplugging" : "Plugging in");
- msc_set_medium_available(!state);
- } else {
- println("unknown command \"%s\"", line);
- }
-
- println();
- }
-
- void cnsl_init(void) {
- cnsl_buff_pos = 0;
- for (int i = 0; i < CNSL_BUFF_SIZE + 1; i++) {
- cnsl_line_buff[i] = '\0';
- cnsl_last_command[i] = '\0';
- cnsl_repeated_command[i] = '\0';
- }
-
- #ifdef CNSL_REPEAT_PMW_STATUS_BY_DEFAULT
- strcpy(cnsl_repeated_command, "pmws");
- repeat_command = true;
- #endif // CNSL_REPEAT_PMW_STATUS_BY_DEFAULT
- }
-
- static int32_t cnsl_find_line_end(void) {
- for (int32_t i = 0; i < cnsl_buff_pos; i++) {
- if ((cnsl_line_buff[i] == '\r') || (cnsl_line_buff[i] == '\n')) {
- return i;
- }
- }
- return -1;
- }
-
- void cnsl_run(void) {
- if (repeat_command && (strlen(cnsl_repeated_command) > 0)
- && (strcmp(cnsl_repeated_command, "repeat") != 0)) {
- uint32_t now = to_ms_since_boot(get_absolute_time());
- if (now >= (last_repeat_time + CNSL_REPEAT_MS)) {
- println("repeating command \"%s\"", cnsl_repeated_command);
- cnsl_interpret(cnsl_repeated_command);
- println();
-
- last_repeat_time = now;
- }
- } else {
- if (repeat_command) {
- println("nothing to repeat");
- }
- repeat_command = false;
- }
- }
-
- void cnsl_handle_input(const char *buf, uint32_t len) {
- if ((cnsl_buff_pos + len) > CNSL_BUFF_SIZE) {
- debug("error: console input buffer overflow! %lu > %u", cnsl_buff_pos + len, CNSL_BUFF_SIZE);
- cnsl_init();
- }
-
- memcpy(cnsl_line_buff + cnsl_buff_pos, buf, len);
- cnsl_buff_pos += len;
-
- int32_t line_len = cnsl_find_line_end();
- if (line_len < 0) {
- // user has not pressed enter yet
- return;
- }
-
- // convert line to C-style string
- cnsl_line_buff[line_len] = '\0';
-
- // TODO handle backspace and other terminal commands?
-
- cnsl_interpret(cnsl_line_buff);
-
- // store command for eventual repeats
- strncpy(cnsl_last_command, cnsl_line_buff, CNSL_BUFF_SIZE + 1);
-
- // clear string and move following data over
- uint32_t cnt = line_len + 1;
- if (cnsl_line_buff[line_len + 1] == '\n') {
- cnt++;
- }
- memset(cnsl_line_buff, '\0', cnt);
- memmove(cnsl_line_buff, cnsl_line_buff + cnt, sizeof(cnsl_line_buff) - cnt);
- cnsl_buff_pos -= cnt;
- }
|