switch to event based expander interface (WIP)

main/drivers/CMakeLists.txt

@@ -5,19 +5,23 @@ set(SOURCES
     "SparkFunBQ27441/SparkFunBQ27441.cpp"
     "esp_lcd_ili9488/esp_lcd_ili9488.c"
     # "bottom_half.cpp"
-    "event_based_bottom_half.cpp"
-    "inputs.cpp"
     "char_lcd.cpp"
+    "game_info.cpp"
     "game_timer.cpp"
     "i2c_lcd_pcf8574.c"
+    "i2c.cpp"
+    "inputs.cpp"
     "leds.cpp"
+    "perh.cpp"
     "power.cpp"
     "sd.cpp"
     "speaker.cpp"
     "sseg.cpp"
+    "starcode.cpp"
     "state_tracking.cpp"
     "tft.cpp"
     "wires.cpp"
+    "wlvgl.cpp"
 )
 
 target_sources(${COMPONENT_LIB} PRIVATE ${SOURCES})

main/drivers/SparkFunBQ27441/SparkFunBQ27441.cpp

@@ -32,6 +32,7 @@ Arduino Uno (any 'duino should do)
 ******************************************************************************/
 
 #include "SparkFunBQ27441.h"
+#include "../i2c.h"
 
 /*****************************************************************************
  ************************** Initialization Functions *************************
@@ -710,7 +711,9 @@ bool BQ27441::writeExtendedData(uint8_t classID, uint8_t offset, uint8_t * data,
 int16_t BQ27441::i2cReadBytes(uint8_t subAddress, uint8_t * dest, uint8_t count)
 {
 	int16_t timeout = BQ72441_I2C_TIMEOUT;
+	xSemaphoreTake(main_i2c_mutex, portMAX_DELAY);
 	i2c_master_write_read_device(BQ72441_I2C_NUM, _deviceAddress, &subAddress, 1, dest, count, timeout);
+	xSemaphoreGive(main_i2c_mutex);
 	return timeout;
 }
 

main/drivers/all.cpp

@@ -1,12 +1,10 @@
 #include "all.h"
 
-static const char *TAG = "driver_all";
-static void init_i2c();
-
 void init_drivers() {
     init_i2c();
     // init char_lcd so we can use it to report other initialization errors.
     init_lcd();
+    init_star_code_system();
     // init the bottom half so that we can get user input.
     init_bottom_half();
     init_sd();
@@ -16,36 +14,6 @@ void init_drivers() {
     init_tft();
     init_leds();
     init_power_board();
-}
 
-/// @brief Initializes I2C_NUM_0. 
+    set_lcd_header_enabled(true);
-///
-/// This is hooked up the to:
-///  - The bottom half
-///  - The char lcd
-///  - The power board
-///  - The MPU6050
-///  - The PERH port
-///  - The Capacitive Touch Panel
-static void init_i2c() {
-    ESP_LOGI(TAG, "Initializing i2c...");
-
-    i2c_config_t conf = {
-        .mode = I2C_MODE_MASTER,
-        .sda_io_num = PIN_I2C_SDA,
-        .scl_io_num = PIN_I2C_SCL,
-        .sda_pullup_en = GPIO_PULLUP_DISABLE,
-        .scl_pullup_en = GPIO_PULLUP_DISABLE,
-        // .sda_pullup_en = GPIO_PULLUP_ENABLE,
-        // .scl_pullup_en = GPIO_PULLUP_ENABLE,
-        .master = {
-            .clk_speed = 100*1000,
-        },
-        .clk_flags = I2C_SCLK_SRC_FLAG_FOR_NOMAL
-    };
-
-    ESP_ERROR_CHECK(i2c_param_config(I2C_NUM_0, &conf));
-    ESP_ERROR_CHECK(i2c_driver_install(I2C_NUM_0, conf.mode, 0, 0, 0));
-
-    ESP_LOGI(TAG, "i2c initialized!");
 }

main/drivers/all.h

@@ -1,17 +1,19 @@
 #ifndef ALL_H
 #define ALL_H
 
-#define PIN_I2C_SDA GPIO_NUM_7
-#define PIN_I2C_SCL GPIO_NUM_15
-
-#include "char_lcd.h"
 #include "bottom_half.h"
+#include "char_lcd.h"
+#include "game_timer.h"
+#include "i2c.h"
+#include "leds.h"
+#include "power.h"
 #include "sd.h"
 #include "speaker.h"
-#include "game_timer.h"
+#include "sseg.h"
-#include "drivers/tft.h"
+#include "starcode.h"
-#include "drivers/leds.h"
+#include "state_tracking.h"
-#include "drivers/power.h"
+#include "tft.h"
+#include "wires.h"
 
 void init_drivers();
 

main/drivers/bottom_half.cpp

@@ -1,6 +1,7 @@
 #include "bottom_half.h"
 #include <esp_log.h>
 #include "state_tracking.h"
+#include "starcode.h"
 
 static const char *TAG = "bottom_half";
 
@@ -45,8 +46,14 @@ static bool replay_handler(const char* event, char* arg) {
 void init_bottom_half() {
     ESP_LOGI(TAG, "Initializing bottom half...");
 
-    ESP_ERROR_CHECK(gpio_set_direction(BOTTOM_PIN_INTERUPT, GPIO_MODE_INPUT));
+    gpio_config_t int_conf = {
-    ESP_ERROR_CHECK(gpio_set_pull_mode(BOTTOM_PIN_INTERUPT, GPIO_PULLUP_ONLY));
+        .pin_bit_mask = 1ULL << BOTTOM_PIN_INTERUPT,
+        .mode = GPIO_MODE_INPUT,
+        .pull_up_en = GPIO_PULLUP_ENABLE,
+        .pull_down_en = GPIO_PULLDOWN_DISABLE,
+        .intr_type = GPIO_INTR_DISABLE,
+    };
+    ESP_ERROR_CHECK(gpio_config(&int_conf));
 
     // TODO: do interupt stuff.
     // ESP_ERROR_CHECK(gpio_intr_enable(BOTTOM_PIN_INTERUPT));
@@ -69,20 +76,25 @@ void init_bottom_half() {
 
 static uint8_t receive_delta(void) {
     uint8_t reg = 1;
-    buf[0] = 0;
+    esp_err_t result = i2c_master_write_read_device(BOTTOM_I2C_NUM, BOTTOM_I2C_ADDR, &reg, 1, buf, 1, (100 / portTICK_PERIOD_MS));
-    ESP_ERROR_CHECK_WITHOUT_ABORT(i2c_master_write_read_device(BOTTOM_I2C_NUM, BOTTOM_I2C_ADDR, &reg, 1, buf, 1, (100 / portTICK_PERIOD_MS)));
+    ESP_ERROR_CHECK_WITHOUT_ABORT(result);
+    if (result != ESP_OK) {
+        return 0;
+    }
     return buf[0];
 }
 
 static void receive_keypad(void) {
+    // TODO: use mutex
+    // TODO: change the bottom half polling scheme from a state-based protocol to an event based protocol
     uint8_t reg = 2;
-    buf[0] = 0;
+    esp_err_t result = i2c_master_write_read_device(BOTTOM_I2C_NUM, BOTTOM_I2C_ADDR, &reg, 1, buf, 2, (100 / portTICK_PERIOD_MS));
-    buf[1] = 0;
+    ESP_ERROR_CHECK_WITHOUT_ABORT(result);
-    ESP_ERROR_CHECK_WITHOUT_ABORT(i2c_master_write_read_device(BOTTOM_I2C_NUM, BOTTOM_I2C_ADDR, &reg, 1, buf, 2, (100 / portTICK_PERIOD_MS)));
+    if (result != ESP_OK) {
+        return;
+    }
     uint16_t new_keypad_state = buf[0] | (buf[1] << 8);
-
     uint16_t just_pressed = new_keypad_state & ~keypad_state;
-    keypad_pressed |= just_pressed;
     if (is_state_tracking() && just_pressed) {
         char buf[6];
         sprintf(buf, "%d", just_pressed);
@@ -90,19 +102,27 @@ static void receive_keypad(void) {
     }
 
     uint16_t just_released = ~new_keypad_state & keypad_state;
-    keypad_released |= just_released;
     if (is_state_tracking() && just_released) {
         char buf[6];
         sprintf(buf, "%d", just_released);
         event_occured("KP_RELEASE", buf);
     }
 
+    star_code_handle_keypad(&just_pressed, &just_released);
+
+    keypad_pressed |= just_pressed;
+    keypad_released |= just_released;
     keypad_state = new_keypad_state;
 }
 
 static void receive_button_switch(void) {
     uint8_t reg = 3;
-    ESP_ERROR_CHECK_WITHOUT_ABORT(i2c_master_write_read_device(BOTTOM_I2C_NUM, BOTTOM_I2C_ADDR, &reg, 1, buf, 2, (100 / portTICK_PERIOD_MS)));
+    esp_err_t result = i2c_master_write_read_device(BOTTOM_I2C_NUM, BOTTOM_I2C_ADDR, &reg, 1, buf, 2, (100 / portTICK_PERIOD_MS));
+    ESP_ERROR_CHECK_WITHOUT_ABORT(result);
+    if (result != ESP_OK) {
+        return;
+    }
+
     uint8_t new_button_state = buf[1] & 0xF;
     uint8_t new_switch_state = (~buf[0]) & 0xF;
     uint8_t new_switch_touch_state = (buf[1] >> 4) & 0xF;
@@ -190,12 +210,10 @@ static void poll_bottom_task(void *arg) {
     // TODO: if using an interupt, switch this to use a queue
     while (1) {
         bool new_data = gpio_get_level(BOTTOM_PIN_INTERUPT) == 0;
-        // bool new_data = 1;
         if (new_data) {
             uint8_t delta = receive_delta();
             // ESP_LOGI(_TAG, "delta: %d", delta);
-            // if (delta == 0) ESP_LOGW(TAG, "delta pin was low, but delta register returned 0");
+            if (delta == 0) ESP_LOGW(TAG, "delta pin was low, but delta register returned 0");
-            if (delta != 0) ESP_LOGI(TAG, "delta!");
 
             if (delta & (1 << DELTA_BIT_KP)) receive_keypad();
             if (delta & (1 << DELTA_BIT_BUTTON_SWITCH)) receive_button_switch();
@@ -219,7 +237,8 @@ void clear_all_pressed_released(void) {
     touch_released = 0;
 }
 
-static bool _take_key(KeypadKey* kp, uint16_t* keypad_bitfield) {
+// TODO: this is public, but it won't need to be after the event-based protocol refactor
+bool take_key(KeypadKey* kp, uint16_t* keypad_bitfield) {
     for (int i = 0; i < 16; i++) {
         int bit_selector = (1 << i);
         if ((*keypad_bitfield) & bit_selector) {
@@ -235,10 +254,10 @@ static bool _take_key(KeypadKey* kp, uint16_t* keypad_bitfield) {
 }
 
 bool get_keypad_pressed(KeypadKey* kp) {
-    return _take_key(kp, &keypad_pressed);
+    return take_key(kp, &keypad_pressed);
 }
 bool get_keypad_released(KeypadKey* kp) {
-    return _take_key(kp, &keypad_released);
+    return take_key(kp, &keypad_released);
 }
 
 char char_of_keypad_key(KeypadKey kp) {

main/drivers/bottom_half.h

@@ -14,22 +14,22 @@
 
 /// @brief An enum for the possible keypad buttons.
 typedef enum {
-    k1 = 0,
+    kd = 0,
-    k4 = 1,
+    pound = 1,
-    k7 = 2,
+    k0 = 2,
     star = 3,
-    k2 = 4,
+    kc = 4,
-    k5 = 5,
+    k9 = 5,
     k8 = 6,
-    k0 = 7,
+    k7 = 7,
-    k3 = 8,
+    kb = 8,
     k6 = 9,
-    k9 = 10,
+    k5 = 10,
-    pound = 11,
+    k4 = 11,
     ka = 12,
-    kb = 13,
+    k3 = 13,
-    kc = 14,
+    k2 = 14,
-    kd = 15,
+    k1 = 15,
 } KeypadKey;
 
 /// @brief An enum for the possible buttons.
@@ -126,6 +126,14 @@ bool get_touch_pressed();
 /// @return true if the touch sensor was just released
 bool get_touch_released();
 
+/// @brief A helper function for internal use.
+/// 
+/// Takes one key from the bitfield and sets the `kp` variable accordingly if the bitfield is not 0.
+/// @param kp Out. The keypad key to set.
+/// @param keypad_bitfield A pointer to the keypad bitfield to take a key from
+/// @return true if a key was taken from the bitfield
+bool take_key(KeypadKey* kp, uint16_t* keypad_bitfield);
+
 // TODO: add touch sensor for switch
 
 #endif /* BOTTOM_HALF_HPP */

main/drivers/char_lcd.cpp

@@ -4,13 +4,28 @@
 #include <esp_log.h>
 #include "state_tracking.h"
 #include <cstring>
+#include "power.h"
+#include "starcode.h"
+#include "game_info.h"
 
 i2c_lcd_pcf8574_handle_t lcd;
 
+static volatile bool header_enabled = false;
+
 static const char *TAG = "char_lcd";
+static const char* EMPTY_ROW = "                    ";
 
 static char buf[65];
 
+static void monitor_battery_task(void* _arg) {
+    (void) _arg;
+
+    while (true) {
+        vTaskDelay(pdMS_TO_TICKS(1'000));
+        lcd_print_header_bat();
+    }
+}
+
 static bool replay_handler(const char* event, char* arg) {
     if (strcmp(event, "LCD_CLEAR") == 0) {
         lcd_clear();
@@ -57,8 +72,7 @@ static bool replay_handler(const char* event, char* arg) {
         return true;
     }
     if (strcmp(event, "LCD_BACKLIGHT") == 0) {
-        uint32_t brightness = atoi(arg);
+        lcd_set_backlight(strcmp(arg, "true") == 0);
-        lcd_set_backlight(brightness);
         return true;
     }
     if (strcmp(event, "LCD_CREATE_CHAR") == 0) {
@@ -74,14 +88,9 @@ static bool replay_handler(const char* event, char* arg) {
         lcd_create_char(location, charmap);
         return true;
     }
-    if (strcmp(event, "LCD_WRITE") == 0) {
-        uint8_t value = atoi(arg);
-        lcd_write(value);
-        return true;
-    }
     if (strcmp(event, "LCD_PRINT") == 0) {
         // TODO: handle \r and \n
-        lcd_print(arg);
+        lcd_print(&lcd, arg);
         return true;
     }
 
@@ -98,25 +107,31 @@ void init_lcd() {
 
     register_replay_fn(replay_handler);
 
+    xTaskCreate(monitor_battery_task, "bat_monitor", 1024*2, nullptr, 0, nullptr);
+
     ESP_LOGI(TAG, "LCD initialized!");
 }
 
 void lcd_clear() {
+    if (!header_enabled) {
         lcd_clear(&lcd);
 
         if (is_state_tracking()) {
             event_occured("LCD_CLEAR", NULL);
         }
-}
+    } else {
-
+        lcd_print(0, 1, EMPTY_ROW);
-void lcd_cursor_home() {
+        lcd_print(0, 2, EMPTY_ROW);
-    lcd_home(&lcd);
+        lcd_print(0, 3, EMPTY_ROW);
-
-    if (is_state_tracking()) {
-        event_occured("LCD_CURSOR", "0,0");
     }
 }
 
+// TODO: rm
+void lcd_cursor_home() {
+    lcd_set_cursor_pos(0, 0);
+}
+
+// TODO: with print requiring you to set a pos every time, this function is not helpful
 void lcd_set_cursor_pos(uint8_t col, uint8_t row) {
     lcd_set_cursor(&lcd, col, row);
 
@@ -204,16 +219,16 @@ void lcd_set_autoscroll(bool autoscroll) {
     }
 }
 
-void lcd_set_backlight(uint8_t brightness) {
+void lcd_set_backlight(bool backlight) {
-    lcd_set_backlight(&lcd, brightness);
+    lcd_set_backlight(&lcd, backlight);
 
     if (is_state_tracking()) {
-        sprintf(buf, "%d", brightness);
+        sprintf(buf, "%d", backlight);
-        event_occured("LCD_BACKLIGHT", buf);
+        event_occured("LCD_BACKLIGHT", backlight ? "true" : "false");
     }
 }
 
-void lcd_create_char(uint8_t location, uint8_t* charmap) {
+void lcd_create_char(uint8_t location, const uint8_t charmap[]) {
     lcd_create_char(&lcd, location, charmap);
 
     if (is_state_tracking()) {
@@ -225,16 +240,9 @@ void lcd_create_char(uint8_t location, uint8_t* charmap) {
     }
 }
 
-void lcd_write(uint8_t value) {
+// TODO: switch to row, col
-    lcd_write(&lcd, value);
+void lcd_print(uint8_t col, uint8_t row, const char* str) {
-
+    lcd_set_cursor_pos(col, row);
-    if (is_state_tracking()) {
-        sprintf(buf, "%d", value);
-        event_occured("LCD_WRITE", buf);
-    }
-}
-
-void lcd_print(const char* str) {
     lcd_print(&lcd, str);
 
     if (is_state_tracking()) {
@@ -243,7 +251,46 @@ void lcd_print(const char* str) {
     }
 }
 
-void lcd_print(uint8_t col, uint8_t row, const char* str) {
+void set_lcd_header_enabled(bool enable) {
-    lcd_set_cursor_pos(col, row);
+    bool old_header_enabled = header_enabled;
-    lcd_print(str);
+    header_enabled = enable;
+
+    // update header in response to enabling/disabling the header
+    if (enable && !old_header_enabled) {
+        lcd_print_header();
+    } else if (!enable && old_header_enabled) {
+        lcd_print(0, 0, EMPTY_ROW);
+    }
+}
+
+bool lcd_header_enabled() {
+    return header_enabled;
+}
+
+void lcd_print_header() {
+    lcd_print_header_star_code();
+    lcd_print_header_step();
+    lcd_print_header_bat();
+}
+
+void lcd_do_splash() {
+    const uint8_t custom_char[6][8] = {
+        { 0x01, 0x01, 0x02, 0x02, 0x07, 0x07, 0x0F, 0x0D },
+        { 0x10, 0x10, 0x18, 0x18, 0x1C, 0x0C, 0x0E, 0x06 },
+        { 0x00, 0x00, 0x01, 0x01, 0x03, 0x03, 0x07, 0x07 },
+        { 0x19, 0x1B, 0x13, 0x17, 0x07, 0x0F, 0x0F, 0x1F },
+        { 0x13, 0x1B, 0x1F, 0x1F, 0x00, 0x1F, 0x1F, 0x1F },
+        { 0x00, 0x00, 0x10, 0x10, 0x00, 0x18, 0x1C, 0x1C },
+    };
+
+    // TODO: make the lcd_lib somehow support the custom character 0 which would otherwise be a null terminator
+    lcd_create_char(1, custom_char[0]);
+    lcd_create_char(2, custom_char[1]);
+    lcd_create_char(3, custom_char[2]);
+    lcd_create_char(4, custom_char[3]);
+    lcd_create_char(5, custom_char[4]);
+    lcd_create_char(6, custom_char[5]);
+
+    lcd_print(6, 1, "\x01\x02Marino");
+    lcd_print(5, 2, "\x03\x04\x05\x06""DEV");
 }

main/drivers/char_lcd.h

@@ -44,18 +44,29 @@ void lcd_right_to_left();
 void lcd_set_autoscroll(bool autoscroll);
 
 // Set backlight brightness
-void lcd_set_backlight(uint8_t brightness);
+void lcd_set_backlight(bool backlight);
 
 // Create a custom character
-void lcd_create_char(uint8_t location, uint8_t charmap[]);
+void lcd_create_char(uint8_t location, const uint8_t charmap[]);
 
-// Write a character to the LCD
+/// @brief Print a string to the LCD at a given pos. 
-void lcd_write(uint8_t value);
+/// @param col the column to print the string at.
-
+/// @param row the row the print the string at.
-// Print a string to the LCD
+/// @param str the string to print.
-void lcd_print(const char* str);
-
-// Print a string to the LCD at a given pos
 void lcd_print(uint8_t col, uint8_t row, const char* str);
 
+/// @brief Enables or disables the header on the LCD.
+/// @param enable `true` to enable the header, `false` to disable.
+void set_lcd_header_enabled(bool enable);
+
+/// @brief Returns weather or not the lcd_header is enabled.
+/// @return `true` if the header is enabled, `false` otherwise.
+bool lcd_header_enabled();
+
+/// @brief Prints the header in the LCD.
+void lcd_print_header();
+
+/// @brief Prints the splash screen for the BLK_BOX.
+void lcd_do_splash();
+
 #endif /* CHAR_LCD_H */

main/drivers/esp_lcd_ili9488/esp_lcd_ili9488.c

@@ -128,7 +128,7 @@ static esp_err_t panel_ili9488_init(esp_lcd_panel_t *panel)
     // ORIGINAL
     lcd_init_cmd_t ili9488_init[] =
     {
-#if CONFIG_USE_NEW_DISPLAY || 1
+#if CONFIG_USE_NEW_DISPLAY
         { ILI9488_POSITIVE_GAMMA_CTL, { 0x00, 0x08, 0x0c, 0x02, 0x0e, 0x04, 0x30, 0x45, 0x47, 0x04, 0x0C, 0x0a, 0x2e, 0x34, 0x0F }, 15 },
         { ILI9488_NEGATIVE_GAMMA_CTL, { 0x00, 0x11, 0x0d, 0x01, 0x0f, 0x05, 0x39, 0x36, 0x51, 0x06, 0x0f, 0x0d, 0x33, 0x37, 0x0F }, 15 },
 #else

main/drivers/event_based_bottom_half.cpp

@@ -1,326 +0,0 @@
-#include "bottom_half.h"
-#include "inputs.hpp"
-
-#include <array>
-
-static uint8_t reverse_4_bits(uint8_t value) {
-    return static_cast<uint8_t>(((value & 0x1) << 3) |
-                                ((value & 0x2) << 1) |
-                                ((value & 0x4) >> 1) |
-                                ((value & 0x8) >> 3));
-}
-
-static KeypadKey map_input_keypad_key(InputKeypadKey key) {
-    switch (key) {
-        case InputKeypadKey::K0: return KeypadKey::k0;
-        case InputKeypadKey::K1: return KeypadKey::k1;
-        case InputKeypadKey::K2: return KeypadKey::k2;
-        case InputKeypadKey::K3: return KeypadKey::k3;
-        case InputKeypadKey::K4: return KeypadKey::k4;
-        case InputKeypadKey::K5: return KeypadKey::k5;
-        case InputKeypadKey::K6: return KeypadKey::k6;
-        case InputKeypadKey::K7: return KeypadKey::k7;
-        case InputKeypadKey::K8: return KeypadKey::k8;
-        case InputKeypadKey::K9: return KeypadKey::k9;
-        case InputKeypadKey::A:  return KeypadKey::ka;
-        case InputKeypadKey::B:  return KeypadKey::kb;
-        case InputKeypadKey::C:  return KeypadKey::kc;
-        case InputKeypadKey::D:  return KeypadKey::kd;
-        case InputKeypadKey::STAR:  return KeypadKey::star;
-        case InputKeypadKey::POUND: return KeypadKey::pound;
-        default: return KeypadKey::k0;
-    }
-}
-
-static uint8_t get_fingerprint_touch_state() {
-    InputsState current = InputsController::get_input_state();
-    return static_cast<uint8_t>((current.touch_state >> 4) & 0x1);
-}
-
-static bool touch_state_initialized = false;
-static bool touch_state_last = false;
-
-static bool update_fingerprint_transition(bool want_pressed) {
-    bool current = get_fingerprint_touch_state();
-    if (!touch_state_initialized) {
-        touch_state_last = current;
-        touch_state_initialized = true;
-        return false;
-    }
-
-    bool transition = want_pressed ? (current && !touch_state_last)
-                                   : (!current && touch_state_last);
-    touch_state_last = current;
-    return transition;
-}
-
-static std::array<SwitchFlip, 8> pending_switch_flips;
-static size_t pending_switch_flip_count = 0;
-
-static void push_pending_switch_flip(const SwitchFlip& event) {
-    if (pending_switch_flip_count < pending_switch_flips.size()) {
-        pending_switch_flips[pending_switch_flip_count++] = event;
-        return;
-    }
-
-    // Drop the oldest event if the buffer is full.
-    for (size_t i = 1; i < pending_switch_flips.size(); ++i) {
-        pending_switch_flips[i - 1] = pending_switch_flips[i];
-    }
-    pending_switch_flips.back() = event;
-}
-
-static bool pop_pending_switch_flip(bool want_up, SwitchFlip& out) {
-    for (size_t i = 0; i < pending_switch_flip_count; ++i) {
-        if (pending_switch_flips[i].is_up() == want_up) {
-            out = pending_switch_flips[i];
-            for (size_t j = i + 1; j < pending_switch_flip_count; ++j) {
-                pending_switch_flips[j - 1] = pending_switch_flips[j];
-            }
-            --pending_switch_flip_count;
-            return true;
-        }
-    }
-    return false;
-}
-
-static void clear_pending_switch_flips() {
-    pending_switch_flip_count = 0;
-}
-
-static std::array<SwitchTouch, 8> pending_switch_touches;
-static size_t pending_switch_touch_count = 0;
-
-static void push_pending_switch_touch(const SwitchTouch& event) {
-    if (pending_switch_touch_count < pending_switch_touches.size()) {
-        pending_switch_touches[pending_switch_touch_count++] = event;
-        return;
-    }
-
-    for (size_t i = 1; i < pending_switch_touches.size(); ++i) {
-        pending_switch_touches[i - 1] = pending_switch_touches[i];
-    }
-    pending_switch_touches.back() = event;
-}
-
-static bool pop_pending_switch_touch(bool want_touched, SwitchTouch& out) {
-    for (size_t i = 0; i < pending_switch_touch_count; ++i) {
-        if (pending_switch_touches[i].is_touched() == want_touched) {
-            out = pending_switch_touches[i];
-            for (size_t j = i + 1; j < pending_switch_touch_count; ++j) {
-                pending_switch_touches[j - 1] = pending_switch_touches[j];
-            }
-            --pending_switch_touch_count;
-            return true;
-        }
-    }
-    return false;
-}
-
-static void clear_pending_switch_touches() {
-    pending_switch_touch_count = 0;
-}
-
-void init_bottom_half() {
-    init_expander();
-    clear_all_pressed_released();
-}
-
-void clear_all_pressed_released() {
-    InputsController::clear_all_events();
-    clear_pending_switch_flips();
-    clear_pending_switch_touches();
-    touch_state_initialized = false;
-}
-
-bool get_keypad_pressed(KeypadKey* kp) {
-    auto opt = InputsController::get_keypad_press();
-    if (!opt.has_value()) {
-        return false;
-    }
-
-    if (kp != nullptr) {
-        *kp = map_input_keypad_key(opt.value());
-    }
-    return true;
-}
-
-bool get_keypad_released(KeypadKey* kp) {
-    auto opt = InputsController::get_keypad_release();
-    if (!opt.has_value()) {
-        return false;
-    }
-
-    if (kp != nullptr) {
-        *kp = map_input_keypad_key(opt.value());
-    }
-    return true;
-}
-
-char char_of_keypad_key(KeypadKey kp) {
-    switch (kp) {
-        case KeypadKey::k1: return '1';
-        case KeypadKey::k2: return '2';
-        case KeypadKey::k3: return '3';
-        case KeypadKey::k4: return '4';
-        case KeypadKey::k5: return '5';
-        case KeypadKey::k6: return '6';
-        case KeypadKey::k7: return '7';
-        case KeypadKey::k8: return '8';
-        case KeypadKey::k9: return '9';
-        case KeypadKey::k0: return '0';
-        case KeypadKey::ka: return 'A';
-        case KeypadKey::kb: return 'B';
-        case KeypadKey::kc: return 'C';
-        case KeypadKey::kd: return 'D';
-        case KeypadKey::star: return '*';
-        case KeypadKey::pound: return '#';
-        default: return ' ';
-    }
-}
-
-static bool take_button(ButtonKey* button, std::optional<Button> opt) {
-    if (!opt.has_value()) {
-        return false;
-    }
-    if (button != nullptr) {
-        *button = static_cast<ButtonKey>(static_cast<uint8_t>(opt.value()));
-    }
-    return true;
-}
-
-bool get_button_pressed(ButtonKey* button) {
-    return take_button(button, InputsController::get_button_press());
-}
-
-bool get_button_released(ButtonKey* button) {
-    return take_button(button, InputsController::get_button_release());
-}
-
-uint8_t get_button_state() {
-    return reverse_4_bits(InputsController::button_state() & 0xF);
-}
-
-static bool take_switch_key(SwitchKey* switch_, const SwitchFlip& event) {
-    if (switch_ != nullptr) {
-        *switch_ = static_cast<SwitchKey>(static_cast<uint8_t>(event.get_switch()));
-    }
-    return true;
-}
-
-bool get_switch_flipped_up(SwitchKey* switch_) {
-    SwitchFlip event;
-    if (pop_pending_switch_flip(true, event)) {
-        return take_switch_key(switch_, event);
-    }
-
-    while (true) {
-        auto opt = InputsController::get_switch_flip();
-        if (!opt.has_value()) {
-            return false;
-        }
-        if (opt->is_up()) {
-            return take_switch_key(switch_, *opt);
-        }
-        push_pending_switch_flip(*opt);
-    }
-}
-
-bool get_switch_flipped_down(SwitchKey* switch_) {
-    SwitchFlip event;
-    if (pop_pending_switch_flip(false, event)) {
-        return take_switch_key(switch_, event);
-    }
-
-    while (true) {
-        auto opt = InputsController::get_switch_flip();
-        if (!opt.has_value()) {
-            return false;
-        }
-        if (!opt->is_up()) {
-            return take_switch_key(switch_, *opt);
-        }
-        push_pending_switch_flip(*opt);
-    }
-}
-
-bool get_switch_flipped(SwitchKey* switch_) {
-    if (pending_switch_flip_count > 0) {
-        SwitchFlip event = pending_switch_flips[0];
-        for (size_t i = 1; i < pending_switch_flip_count; ++i) {
-            pending_switch_flips[i - 1] = pending_switch_flips[i];
-        }
-        --pending_switch_flip_count;
-        return take_switch_key(switch_, event);
-    }
-
-    auto opt = InputsController::get_switch_flip();
-    if (!opt.has_value()) {
-        return false;
-    }
-
-    return take_switch_key(switch_, *opt);
-}
-
-uint8_t get_switch_state() {
-    return reverse_4_bits(InputsController::switch_state() & 0xF);
-}
-
-static bool take_switch_touch(SwitchKey* switch_, const SwitchTouch& event) {
-    if (switch_ != nullptr) {
-        *switch_ = static_cast<SwitchKey>(static_cast<uint8_t>(event.get_switch()));
-    }
-    return true;
-}
-
-bool get_switch_touch_pressed(SwitchKey* switch_) {
-    SwitchTouch event;
-    if (pop_pending_switch_touch(true, event)) {
-        return take_switch_touch(switch_, event);
-    }
-
-    while (true) {
-        auto opt = InputsController::get_switch_touch();
-        if (!opt.has_value()) {
-            return false;
-        }
-        if (opt->is_touched()) {
-            return take_switch_touch(switch_, *opt);
-        }
-        push_pending_switch_touch(*opt);
-    }
-}
-
-bool get_switch_touch_released(SwitchKey* switch_) {
-    SwitchTouch event;
-    if (pop_pending_switch_touch(false, event)) {
-        return take_switch_touch(switch_, event);
-    }
-
-    while (true) {
-        auto opt = InputsController::get_switch_touch();
-        if (!opt.has_value()) {
-            return false;
-        }
-        if (!opt->is_touched()) {
-            return take_switch_touch(switch_, *opt);
-        }
-        push_pending_switch_touch(*opt);
-    }
-}
-
-uint8_t get_switch_touch_state() {
-    return reverse_4_bits(InputsController::switch_touch_state() & 0xF);
-}
-
-bool get_touch_state() {
-    return get_fingerprint_touch_state() != 0;
-}
-
-bool get_touch_pressed() {
-    return update_fingerprint_transition(true);
-}
-
-bool get_touch_released() {
-    return update_fingerprint_transition(false);
-}

main/drivers/game_info.cpp

@@ -0,0 +1,21 @@
+#include "game_info.h"
+#include "starcode.h"
+#include <stdio.h>
+#include "char_lcd.h"
+
+static char game_state[GAME_STATE_MAX_LEN+2] = " MENU ";
+
+void set_game_state(const char* new_state) {
+    snprintf(game_state, sizeof(game_state), " %-5s", new_state);
+}
+
+void reset_game_state() {
+    set_game_state("");
+}
+
+void lcd_print_header_step() {
+    if (!lcd_header_enabled()) return;
+    if (lcd_starcode_displaying_result()) return;
+
+    lcd_print(10, 0, game_state);
+}

main/drivers/game_info.h

@@ -0,0 +1,17 @@
+#ifndef GAME_INFO_H
+#define GAME_INFO_H
+
+#define GAME_STATE_MAX_LEN 5
+
+/// @brief Sets the game state, used for the header.
+///
+/// Must be <= 5 characters
+void set_game_state(const char* new_state);
+
+/// @brief Resets the game state to be blank.
+void reset_game_state();
+
+/// @brief Prints the game state section of the header to the char_lcd. (row 0, columns 11-15)
+void lcd_print_header_step();
+
+#endif /* GAME_INFO_H */

main/drivers/i2c.cpp

@@ -0,0 +1,35 @@
+#include "i2c.h"
+#include "esp_log.h"
+#include "esp_err.h"
+#include "driver/i2c.h"
+
+static const char *TAG = "i2c";
+
+SemaphoreHandle_t main_i2c_mutex;
+
+void init_i2c() {
+    ESP_LOGI(TAG, "Initializing i2c...");
+
+    i2c_config_t conf = {
+        .mode = I2C_MODE_MASTER,
+        .sda_io_num = PIN_I2C_SDA,
+        .scl_io_num = PIN_I2C_SCL,
+        .sda_pullup_en = GPIO_PULLUP_DISABLE,
+        .scl_pullup_en = GPIO_PULLUP_DISABLE,
+        // .sda_pullup_en = GPIO_PULLUP_ENABLE,
+        // .scl_pullup_en = GPIO_PULLUP_ENABLE,
+        .master = {
+            // TODO: 400k?
+            .clk_speed = 100*1000,
+        },
+        .clk_flags = I2C_SCLK_SRC_FLAG_FOR_NOMAL
+    };
+
+    ESP_ERROR_CHECK(i2c_param_config(MAIN_I2C_BUS_NUM, &conf));
+    ESP_ERROR_CHECK(i2c_driver_install(MAIN_I2C_BUS_NUM, conf.mode, 0, 0, 0));
+
+    main_i2c_mutex = xSemaphoreCreateMutex();
+    assert(main_i2c_mutex != NULL);
+
+    ESP_LOGI(TAG, "i2c initialized!");
+}

main/drivers/i2c.h

@@ -0,0 +1,26 @@
+#ifndef I2C_H
+#define I2C_H
+
+#include "freertos/FreeRTOS.h"
+#include "freertos/semphr.h"
+
+#define MAIN_I2C_BUS_NUM I2C_NUM_0
+
+#define PIN_I2C_SDA GPIO_NUM_7
+#define PIN_I2C_SCL GPIO_NUM_15
+
+/// The mutex for accessing `I2C_NUM_0`.
+extern SemaphoreHandle_t main_i2c_mutex;
+
+/// @brief Initializes `I2C_NUM_0`.
+///
+/// This is hooked up the to:
+///  - The bottom half
+///  - The char lcd
+///  - The power board
+///  - The MPU6050
+///  - The PERH port
+///  - The Capacitive Touch Panel
+void init_i2c();
+
+#endif /* I2C_H */

main/drivers/i2c_lcd_pcf8574.c

@@ -12,6 +12,7 @@
 #include "esp_check.h"
 #include "freertos/FreeRTOS.h"
 #include "freertos/task.h"
+#include "i2c.h"
 
 #define TAG "I2C_LCD_PCF8574"
 
@@ -58,6 +59,7 @@ void lcd_begin(i2c_lcd_pcf8574_handle_t* lcd, uint8_t cols, uint8_t rows) {
     lcd->entrymode = 0x02;
 
     // The following are the reset sequence: Please see "Initialization instruction in the PCF8574 datasheet."
+    xSemaphoreTake(main_i2c_mutex, portMAX_DELAY);
     i2c_cmd_handle_t cmd = i2c_cmd_link_create();
     i2c_master_start(cmd);
     // We left-shift the device addres and add the read/write command
@@ -95,6 +97,7 @@ void lcd_begin(i2c_lcd_pcf8574_handle_t* lcd, uint8_t cols, uint8_t rows) {
     i2c_master_stop(cmd);
     i2c_master_cmd_begin(lcd->i2c_port, cmd, I2C_MASTER_TIMEOUT_MS / portTICK_PERIOD_MS);
     i2c_cmd_link_delete(cmd);
+    xSemaphoreGive(main_i2c_mutex);
 
     // Instruction: function set = 0x20
     lcd_send(lcd, 0x20 | (rows > 1 ? 0x08 : 0x00), false);
@@ -110,7 +113,7 @@ void lcd_clear(i2c_lcd_pcf8574_handle_t* lcd) {
     // Instruction: Clear display = 0x01
     lcd_send(lcd, 0x01, false);
     // Clearing the display takes a while: takes approx. 1.5ms
-    esp_rom_delay_us(1600);
+    esp_rom_delay_us(2000);
 }  // lcd_clear()
 
 // Set the display to home
@@ -118,7 +121,7 @@ void lcd_home(i2c_lcd_pcf8574_handle_t* lcd) {
     // Instruction: Return home = 0x02
     lcd_send(lcd, 0x02, false);
     // Same as clearing the display: takes approx. 1.5ms
-    esp_rom_delay_us(1600);
+    esp_rom_delay_us(2000);
 }  // lcd_home()
 
 // Set the cursor to a new position.
@@ -238,7 +241,7 @@ void lcd_set_backlight(i2c_lcd_pcf8574_handle_t* lcd, uint8_t brightness) {
 }  // lcd_set_backlight()
 
 // Custom character creation: allows us to create up to 8 custom characters in the CGRAM locations
-void lcd_create_char(i2c_lcd_pcf8574_handle_t* lcd, uint8_t location, uint8_t charmap[]) {
+void lcd_create_char(i2c_lcd_pcf8574_handle_t* lcd, uint8_t location, const uint8_t charmap[]) {
     location &= 0x7;  // Only 8 locations are available
     // Set the CGRAM address
     lcd_send(lcd, 0x40 | (location << 3), false);
@@ -295,6 +298,7 @@ void lcd_print_number(i2c_lcd_pcf8574_handle_t* lcd, uint8_t col, uint8_t row, u
 
 
 static void lcd_send(i2c_lcd_pcf8574_handle_t* lcd, uint8_t value, bool is_data) {
+    xSemaphoreTake(main_i2c_mutex, portMAX_DELAY);
     i2c_cmd_handle_t cmd = i2c_cmd_link_create();
     i2c_master_start(cmd);
     i2c_master_write_byte(cmd, (lcd->i2c_addr << 1) | I2C_MASTER_WRITE, true);
@@ -303,6 +307,7 @@ static void lcd_send(i2c_lcd_pcf8574_handle_t* lcd, uint8_t value, bool is_data)
     i2c_master_stop(cmd);
     esp_err_t ret = i2c_master_cmd_begin(lcd->i2c_port, cmd, I2C_MASTER_TIMEOUT_MS / portTICK_PERIOD_MS);
     i2c_cmd_link_delete(cmd);
+    xSemaphoreGive(main_i2c_mutex);
     
     if (ret != ESP_OK) {
         ESP_LOGE(TAG, "Failed to send data to LCD: %s", esp_err_to_name(ret));
@@ -341,6 +346,7 @@ static void lcd_write_i2c(i2c_lcd_pcf8574_handle_t* lcd, uint8_t data, bool is_d
         data |= lcd->backlight_mask;
     }
 
+    xSemaphoreTake(main_i2c_mutex, portMAX_DELAY);
     i2c_cmd_handle_t cmd = i2c_cmd_link_create();
     i2c_master_start(cmd);
     i2c_master_write_byte(cmd, (lcd->i2c_addr << 1) | I2C_MASTER_WRITE, true);
@@ -348,6 +354,7 @@ static void lcd_write_i2c(i2c_lcd_pcf8574_handle_t* lcd, uint8_t data, bool is_d
     i2c_master_stop(cmd);
     esp_err_t ret = i2c_master_cmd_begin(lcd->i2c_port, cmd, I2C_MASTER_TIMEOUT_MS / portTICK_PERIOD_MS);
     i2c_cmd_link_delete(cmd);
+    xSemaphoreGive(main_i2c_mutex);
 
     if (ret != ESP_OK) {
         ESP_LOGE(TAG, "Failed to write to LCD: %s", esp_err_to_name(ret));

main/drivers/i2c_lcd_pcf8574.h

@@ -94,7 +94,7 @@ void lcd_no_autoscroll(i2c_lcd_pcf8574_handle_t* lcd);
 void lcd_set_backlight(i2c_lcd_pcf8574_handle_t* lcd, uint8_t brightness);
 
 // Create a custom character
-void lcd_create_char(i2c_lcd_pcf8574_handle_t* lcd, uint8_t location, uint8_t charmap[]);
+void lcd_create_char(i2c_lcd_pcf8574_handle_t* lcd, uint8_t location, const uint8_t charmap[]);
 
 // Write a character to the LCD
 void lcd_write(i2c_lcd_pcf8574_handle_t* lcd, uint8_t value);

main/drivers/inputs.cpp

@@ -1,4 +1,6 @@
-#include "inputs.hpp"
+#include "blk_box_drivers/inputs.hpp"
+
+#include "bottom_half.h"
 #include "pins.h"
 #include "driver/i2c.h"
 #include "driver/gpio.h"
@@ -11,6 +13,8 @@ static const char *TAG = "INPUTS";
 
 static TaskHandle_t expander_task_handle = NULL;
 
+static i2c_master_dev_handle_t expander_i2c_dev_handle;
+
 const static uint8_t REG_WHOAMI = 0x01;
 const static uint8_t REG_SW_VERSION = 0x02;
 const static uint8_t REG_EVENT_QUEUE_POP = 0x10;
@@ -73,12 +77,19 @@ static void IRAM_ATTR expander_isr_handler(void *arg) {
 void init_expander() {
     ESP_LOGI(TAG, "Initializing expander...");
 
-    // legacy I2C driver: use the shared I2C_NUM_0 bus already configured elsewhere.
+    i2c_device_config_t dev_config = {
-    // TODO: replace all these ESP_ERROR_CHECK with proper error handling that doesn't just crash the program
+        .dev_addr_length = I2C_ADDR_BIT_LEN_7,
+        .device_address = EXPANDER_I2C_ADDR,
+        .scl_speed_hz = EXPANDER_I2C_SPEED,
+        .scl_wait_us = 0, // default
+        .flags = {
+            .disable_ack_check = 0,
+        }
+    };
 
-    // setup interrupt on PIN_EXPANDER_INT
+    // setup interrupt on BOTTOM_PIN_INTERUPT
     gpio_config_t io_conf = {
-        .pin_bit_mask = (1ULL << PIN_EXPANDER_INT),
+        .pin_bit_mask = (1ULL << BOTTOM_PIN_INTERUPT),
         .mode = GPIO_MODE_INPUT,
         .pull_up_en = GPIO_PULLUP_ENABLE,
         .pull_down_en = GPIO_PULLDOWN_DISABLE,
@@ -91,12 +102,12 @@ void init_expander() {
     ESP_ERROR_CHECK(gpio_install_isr_service(0));
 
     // Attach the ISR to the expander pin
-    ESP_ERROR_CHECK(gpio_isr_handler_add(PIN_EXPANDER_INT, expander_isr_handler, NULL));
+    ESP_ERROR_CHECK(gpio_isr_handler_add(BOTTOM_PIN_INTERUPT, expander_isr_handler, NULL));
 
     // verify the expander connection status by reading the WHOAMI register
     uint8_t read_buf[2] = {0};
 
-    ESP_ERROR_CHECK(i2c_master_write_read_device(I2C_NUM_0, EXPANDER_I2C_ADDR, &REG_WHOAMI, 1, read_buf, 1, pdMS_TO_TICKS(EXPANDER_TIMEOUT_MS)));
+    i2c_master_write_read_device(I2C_NUM_0, BOTTOM_I2C_ADDR, &REG_WHOAMI, 1, read_buf, 1, 1000);
 
     if (read_buf[0] != EXPANDER_WHOAMI_VALUE) {
         ESP_LOGE(TAG, "WHOAMI mismatch, expected 0x%02X, got 0x%02X", EXPANDER_WHOAMI_VALUE, read_buf[0]);
@@ -105,7 +116,7 @@ void init_expander() {
 
     ESP_LOGD(TAG, "Expander WHOAMI check passed");
 
-    ESP_ERROR_CHECK(i2c_master_write_read_device(I2C_NUM_0, EXPANDER_I2C_ADDR, &REG_SW_VERSION, 1, read_buf, 2, pdMS_TO_TICKS(EXPANDER_TIMEOUT_MS)));
+    ESP_ERROR_CHECK(i2c_master_transmit_receive(expander_i2c_dev_handle, &REG_SW_VERSION, 1, read_buf, 2, EXPANDER_TIMEOUT_MS));
 
     // init the peripheral struct
     expander_peripheral_singleton.state_mutex = xSemaphoreCreateMutex();
@@ -114,8 +125,8 @@ void init_expander() {
     expander_peripheral_singleton.switch_flip_events= xQueueCreate(EXPANDER_EVENT_QUEUE_SIZE, sizeof(SwitchFlip));
     expander_peripheral_singleton.switch_touch_events= xQueueCreate(EXPANDER_EVENT_QUEUE_SIZE, sizeof(SwitchTouch));
     expander_peripheral_singleton.touch_events= xQueueCreate(EXPANDER_EVENT_QUEUE_SIZE, sizeof(TouchedReleased));
-    expander_peripheral_singleton.keypad_press_events= xQueueCreate(EXPANDER_KEYPAD_QUEUE_SIZE, sizeof(InputKeypadKey));
+    expander_peripheral_singleton.keypad_press_events= xQueueCreate(EXPANDER_KEYPAD_QUEUE_SIZE, sizeof(KeypadKey));
-    expander_peripheral_singleton.keypad_release_events= xQueueCreate(EXPANDER_KEYPAD_QUEUE_SIZE, sizeof(InputKeypadKey));
+    expander_peripheral_singleton.keypad_release_events= xQueueCreate(EXPANDER_KEYPAD_QUEUE_SIZE, sizeof(KeypadKey));
 
     ESP_LOGI(TAG, "Expander initialized! SW version: v%d.%d", read_buf[0], read_buf[1]);
 
@@ -148,7 +159,7 @@ static void expander_task(void *arg) {
 static void get_events() {
     uint8_t recv;
     while (gpio_get_level(PIN_EXPANDER_INT) == 0) {
-        ESP_ERROR_CHECK(i2c_master_write_read_device(I2C_NUM_0, EXPANDER_I2C_ADDR, &REG_EVENT_QUEUE_POP, 1, &recv, 1, pdMS_TO_TICKS(EXPANDER_TIMEOUT_MS)));
+        ESP_ERROR_CHECK(i2c_master_transmit_receive(expander_i2c_dev_handle, &REG_EVENT_QUEUE_POP, 1, &recv, 1, EXPANDER_TIMEOUT_MS));
         handle_event(recv);
     }
 }
@@ -238,7 +249,7 @@ static void handle_keypad_event(uint8_t event) {
 
     bool pressed = (event & PRESSED_NOT_RELEASED_BIT) != 0;
     uint8_t number = event & KEY_MASK;
-    InputKeypadKey key = static_cast<InputKeypadKey>(number);
+    KeypadKey key = static_cast<KeypadKey>(number);
 
     // starcode system gets first dibs
     // TODO: do starcode inbetweener
@@ -429,8 +440,8 @@ bool InputsController::has_keypad_press() {
 }
 
 /// Gets the next keypad press event (if any).
-std::optional<InputKeypadKey> InputsController::get_keypad_press() {
+std::optional<KeypadKey> InputsController::get_keypad_press() {
-    InputKeypadKey k;
+    KeypadKey k;
     if (xQueueReceive(expander_peripheral_singleton.keypad_press_events, &k, 0) == pdTRUE) {
         return k;
     }
@@ -438,8 +449,8 @@ std::optional<InputKeypadKey> InputsController::get_keypad_press() {
 }
 
 /// Gets the next keypad press event, waiting if neccesary.
-InputKeypadKey InputsController::wait_keypad_press() {
+KeypadKey InputsController::wait_keypad_press() {
-    InputKeypadKey k;
+    KeypadKey k;
     xQueueReceive(expander_peripheral_singleton.keypad_press_events, &k, portMAX_DELAY);
     return k;
 }
@@ -450,8 +461,8 @@ bool InputsController::has_keypad_release() {
 }
 
 /// Gets the next keypad release event (if any).
-std::optional<InputKeypadKey> InputsController::get_keypad_release() {
+std::optional<KeypadKey> InputsController::get_keypad_release() {
-    InputKeypadKey k;
+    KeypadKey k;
     if (xQueueReceive(expander_peripheral_singleton.keypad_release_events, &k, 0) == pdTRUE) {
         return k;
     }
@@ -459,8 +470,8 @@ std::optional<InputKeypadKey> InputsController::get_keypad_release() {
 }
 
 /// Gets the next keypad release event, waiting if neccesary.
-InputKeypadKey InputsController::wait_keypad_release() {
+KeypadKey InputsController::wait_keypad_release() {
-    InputKeypadKey k;
+    KeypadKey k;
     xQueueReceive(expander_peripheral_singleton.keypad_release_events, &k, portMAX_DELAY);
     return k;
 }

main/drivers/inputs.hpp

@@ -1,275 +0,0 @@
-#ifndef INPUTS_H
-#define INPUTS_H
-
-#include <freertos/FreeRTOS.h>
-#include <freertos/queue.h>
-#include <freertos/semphr.h>
-#include <optional>
-
-#define EXPANDER_I2C_ADDR (0x7E)
-#define EXPANDER_I2C_SPEED (400000)
-// the actual transaction takes ~0.3ms, but for some reason a timout of ~10 or lower causes issues.
-#define EXPANDER_TIMEOUT_MS (100)
-
-#define EXPANDER_WHOAMI_VALUE (0x85)
-
-// queue sizes
-#define EXPANDER_EVENT_QUEUE_SIZE 4
-#define EXPANDER_KEYPAD_QUEUE_SIZE 16
-
-void init_expander();
-
-/// The four buttons on the bottom half.
-enum class Button: uint8_t {
-    B1 = 0,
-    B2 = 1,
-    B3 = 2,
-    B4 = 3,
-    GREEN = 0,
-    RED = 1,
-    YELLOW = 2,
-    BLUE = 3,
-};
-
-constexpr uint8_t raw_value(Button v) { return static_cast<uint8_t>(v); }
-constexpr Button button_from_raw(uint8_t raw) { return static_cast<Button>(raw & 0b11); }
-
-/// The four switches on the bottom half.
-enum class Switch: uint8_t {
-    S1 = 0,
-    S2 = 1,
-    S3 = 2,
-    S4 = 3,
-};
-
-constexpr uint8_t raw_value(Switch v) { return static_cast<uint8_t>(v); }
-constexpr Switch switch_from_raw(uint8_t raw) { return static_cast<Switch>(raw & 0b11); }
-
-enum class TouchedReleased: uint8_t {
-    Released = 0,
-    Touched = 1,
-};
-
-constexpr uint8_t raw_value(TouchedReleased v) { return static_cast<uint8_t>(v); }
-constexpr TouchedReleased touched_released_from_raw(uint8_t raw) { return static_cast<TouchedReleased>(raw & 0b1); }
-
-/// One of the keys on the keypad.
-enum class InputKeypadKey: uint8_t {
-    K0 = 0,
-    K1 = 1,
-    K2 = 2,
-    K3 = 3,
-    K4 = 4,
-    K5 = 5,
-    K6 = 6,
-    K7 = 7,
-    K8 = 8,
-    K9 = 9,
-    A = 10,
-    B = 11,
-    C = 12,
-    D = 13,
-    STAR = 14,
-    POUND = 15,
-};
-
-constexpr uint8_t raw_value(InputKeypadKey v) { return static_cast<uint8_t>(v); }
-constexpr InputKeypadKey keypad_key_from_raw(uint8_t raw) { return static_cast<InputKeypadKey>(raw & 0b1111); }
-
-constexpr char keypad_key_to_char(InputKeypadKey key) {
-    static constexpr char lookup[16] = {
-        '0', '1', '2', '3', '4', '5', '6', '7',
-        '8', '9', 'A', 'B', 'C', 'D', '*', '#'
-    };
-    return lookup[static_cast<uint8_t>(key) & 0b1111];
-}
-
-
-struct SwitchFlip {
-private:
-    // [bit2: up] [bit1-0: switch] 
-    uint8_t data;
-
-public:
-    // Constructor
-    SwitchFlip(Switch sw, bool up)
-        : data((static_cast<uint8_t>(sw) & 0b11) |
-               ((up ? 1 : 0) << 2)) {}
-
-    // Default constructor
-    SwitchFlip() : data(0) {}
-
-    // Raw value constructor
-    explicit SwitchFlip(uint8_t raw_data) : data(raw_data) {}
-
-    // Raw value getter
-    uint8_t raw() const { return data; }
-
-    // Getters
-    Switch get_switch() const {
-        return static_cast<Switch>(data & 0b11);
-    }
-
-    bool is_up() const {
-        return (data >> 2) & 1;
-    }
-
-    // Setters
-    void set_switch(Switch sw) {
-        data = (data & ~0b11) | (static_cast<uint8_t>(sw) & 0b11);
-    }
-
-    void set_up(bool up) {
-        data = (data & ~(1 << 2)) | ((up ? 1 : 0) << 2);
-    }
-};
-static_assert(sizeof(SwitchFlip) == 1);
-
-struct SwitchTouch {
-private:
-    // [bit2: touched] [bit1-0: switch] 
-    uint8_t data;
-
-public:
-    // Constructor
-    SwitchTouch(Switch sw, bool touched)
-        : data((static_cast<uint8_t>(sw) & 0b11) |
-               ((touched ? 1 : 0) << 2)) {}
-
-    // Default constructor
-    SwitchTouch() : data(0) {}
-
-    // Raw value constructor
-    explicit SwitchTouch(uint8_t raw_data) : data(raw_data) {}
-
-    // Raw value getter
-    uint8_t raw() const { return data; }
-
-    // Getters
-    Switch get_switch() const {
-        return static_cast<Switch>(data & 0b11);
-    }
-
-    bool is_touched() const {
-        return (data >> 2) & 1;
-    }
-
-    // Setters
-    void set_switch(Switch sw) {
-        data = (data & ~0b11) | (static_cast<uint8_t>(sw) & 0b11);
-    }
-
-    void set_touched(bool touched) {
-        data = (data & ~(1 << 2)) | ((touched ? 1 : 0) << 2);
-    }
-};
-static_assert(sizeof(SwitchTouch) == 1);
-
-struct ButtonOrSwitch {
-private:
-    // [bit2: is_switch] [bit1-0: number]
-    uint8_t data;
-
-public:
-    // Constructor
-    ButtonOrSwitch(uint8_t number, bool is_switch)
-        : data((number & 0b11) |
-               ((is_switch ? 1 : 0) << 2)) {}
-
-    ButtonOrSwitch() : data(0) {}
-
-    // Raw value constructor
-    explicit ButtonOrSwitch(uint8_t raw_data) : data(raw_data) {}
-
-    // Raw value getter
-    uint8_t raw() const { return data; }
-
-    // Getters
-    uint8_t number() const {
-        return data & 0b11;
-    }
-
-    bool is_switch() const {
-        return (data >> 2) & 1;
-    }
-
-    // Setters
-    void set_number(uint8_t number) {
-        data = (data & ~0b11) | (number & 0b11);
-    }
-
-    void set_is_switch(bool is_switch) {
-        data = (data & ~(1 << 2)) | ((is_switch ? 1 : 0) << 2);
-    }
-};
-static_assert(sizeof(ButtonOrSwitch) == 1);
-
-
-/// @brief The state of the bottom half of the box.
-struct InputsState {
-    /// The touch state of the switches in the lower 4 bits.
-    /// The touch pad state in bit 4.
-    uint8_t touch_state;
-    /// The current state of the buttons in the lower 4 bits.
-    uint8_t button_state;
-    /// The current state of the switches. Up switches are stored
-    /// in the lower 4 bits, switches that are down are stored in
-    /// the upper 4 bits. If switches are in the middle, the
-    /// corresponding bit will be `0` in the upper and lower 4.
-    uint8_t switch_state;
-    /// The state of the keypad.
-    uint16_t keypad_state;
-    /// The sensitivity of the `hal` value to auto update.
-    uint16_t hal_sense;
-    /// The sensitivity of the `close_hal` value to auto update.
-    uint16_t close_hal_sense;
-    /// A non-exact hal value reading.
-    /// This only gets updated when it changes by `hal_sense`
-    uint16_t hal;
-    /// A non-exact hal value reading.
-    /// This only gets updated when it changes by `close_hal_sense`
-    uint16_t close_hal;
-    /// The RFID card that was presented last.
-    uint32_t rfid_state;
-
-    InputsState() : touch_state(0), button_state(0), switch_state(0), keypad_state(0), hal_sense(0), close_hal_sense(0), hal(0), close_hal(0), rfid_state(0) {}
-};
-
-class InputsController {
-public:
-    static void clear_all_events();
-
-    static InputsState get_input_state();
-    
-    static bool has_button_press();
-    static std::optional<Button> get_button_press();
-    static Button wait_button_press();
-    static uint8_t button_state();
-
-    static bool has_button_release();
-    static std::optional<Button> get_button_release();
-    static Button wait_button_release();
-
-    static bool has_switch_flip();
-    static std::optional<SwitchFlip> get_switch_flip();
-    static SwitchFlip wait_switch_flip();
-    static uint8_t switch_state();
-
-    static bool has_switch_touch();
-    static std::optional<SwitchTouch> get_switch_touch();
-    static SwitchTouch wait_switch_touch();
-    static uint8_t switch_touch_state();
-
-    static bool has_keypad_press();
-    static std::optional<InputKeypadKey> get_keypad_press();
-    static InputKeypadKey wait_keypad_press();
-
-    static bool has_keypad_release();
-    static std::optional<InputKeypadKey> get_keypad_release();
-    static InputKeypadKey wait_keypad_release();
-    static uint16_t keypad_state();
-
-    // TODO: impl and add the hal and RFID stuff
-};
-
-#endif // INPUTS_H

main/drivers/leds.cpp

@@ -8,11 +8,11 @@ static const char* TAG = "leds";
 
 static led_strip_handle_t leds;
 
+// TODO: rename these to playback_handler
 static bool replay_handler(const char* event, char* arg) {
     if (strcmp(event, "LED_SET") == 0) {
         uint32_t led = atoi(strtok(arg, ","));
         uint32_t color = atoi(strtok(NULL, ","));
-        ESP_LOGI("leds", "color: %ld", color);
         led_set(led, color);
         return true;
     }

main/drivers/leds.h

@@ -1,23 +1,10 @@
 #ifndef LEDS_H
 #define LEDS_H
 
-#include "main.h"
 #include <stdint.h>
 
 #define LED_COUNT 21
-
-#ifdef CONTROL_REV_2_0
 #define NEOPIXEL_PIN GPIO_NUM_0
-#endif
-#ifdef CONTROL_REV_2_1
-#define NEOPIXEL_PIN GPIO_NUM_21
-#endif
-
-#ifndef CONTROL_REV_2_0
-#ifndef CONTROL_REV_2_1
-#error "define rev2.0 or rev2.1"
-#endif
-#endif
 // 10MHz resolution, 1 tick = 0.1us (led strip needs a high resolution)
 #define LED_STRIP_RMT_RES_HZ  (10 * 1000 * 1000)
 
@@ -39,6 +26,7 @@ enum LEDColor: uint32_t {
     LED_COLOR_WHITE_STRONG = 0xFF'FF'FF,
 };
 
+// TODO: sepperate the indicator leds from the shape display.
 enum IndicatorLED {
     LED_SHAPE1 = 0u,
     LED_SHAPE2 = 1u,

main/drivers/perh.cpp

@@ -0,0 +1,2 @@
+#include "perh.h"
+

main/drivers/perh.h

@@ -0,0 +1,12 @@
+#ifndef PERH_H
+#define PERH_H
+
+#include "driver/gpio.h"
+
+#define PIN_PERH0 GPIO_NUM_6
+#define PIN_PERH1 GPIO_NUM_5
+#define PIN_PERH2 GPIO_NUM_4
+#define PIN_PERH3 GPIO_NUM_2
+#define PIN_PERH4 GPIO_NUM_1
+
+#endif /* PERH_H */

main/drivers/pins.h

@@ -1,48 +0,0 @@
-#ifndef PINS_H
-#define PINS_H
-
-#include "driver/gpio.h"
-
-// ONLY INPUTS.HPP uses this file
-
-#define PIN_SDA          (GPIO_NUM_7)
-#define PIN_SCL          (GPIO_NUM_15)
-
-#define PIN_LCD_MISO     (GPIO_NUM_16)
-#define PIN_LCD_MOSI     (GPIO_NUM_17)
-#define PIN_LCD_CLK      (GPIO_NUM_18)
-#define PIN_LCD_RS       (GPIO_NUM_8)
-#define PIN_LCD_RST      (GPIO_NUM_9)
-
-#define PIN_USB_DM       (GPIO_NUM_19)
-#define PIN_USB_DP       (GPIO_NUM_20)
-
-#define PIN_I2S_DAT      (GPIO_NUM_3)
-#define PIN_I2S_BCLK     (GPIO_NUM_11)
-#define PIN_I2S_LRCLK    (GPIO_NUM_12)
-
-#define PIN_SSEG_DAT     (GPIO_NUM_46)
-#define PIN_SSEG_CLK     (GPIO_NUM_48)
-
-#define PIN_MPU_INT      (GPIO_NUM_10)
-#define PIN_EXPANDER_INT (GPIO_NUM_13)
-
-#define PIN_IR_RCV       (GPIO_NUM_14)
-
-// #define PIN_NEOPIXEL     (GPIO_NUM_21) // Rev 2.1
-#define PIN_NEOPIXEL     (GPIO_NUM_0) // Rev 2.0
-
-#define PIN_SD_DAT0      (GPIO_NUM_38)
-#define PIN_SD_DAT1      (GPIO_NUM_47)
-#define PIN_SD_DAT2      (GPIO_NUM_42)
-#define PIN_SD_DAT3      (GPIO_NUM_41)
-#define PIN_SD_CMD       (GPIO_NUM_40)
-#define PIN_SD_CLK       (GPIO_NUM_39)
-
-#define PIN_PERH0        (GPIO_NUM_6)
-#define PIN_PERH1        (GPIO_NUM_5)
-#define PIN_PERH2        (GPIO_NUM_4)
-#define PIN_PERH3        (GPIO_NUM_2)
-#define PIN_PERH4        (GPIO_NUM_1)
-
-#endif // PINS_H

main/drivers/power.cpp

@@ -1,5 +1,6 @@
 #include "power.h"
 #include "char_lcd.h"
+#include "starcode.h"
 #include <esp_log.h>
 
 static const char* TAG = "power";
@@ -50,4 +51,28 @@ void init_power_board() {
     ESP_LOGI(TAG, "Power board initialized!");
 }
 
+uint16_t get_bat_voltage() {
+    return lipo.voltage();
+}
 
+void lcd_print_header_bat() {
+    if (!lcd_header_enabled()) return;
+    if (lcd_starcode_displaying_result()) return;
+    
+    uint8_t soc = lipo.soc();
+    uint8_t current = lipo.current();
+    char buf[6];
+    if (soc < 5 && current <= 0) {
+        snprintf(buf, sizeof(buf), " LOW");
+    } else if (soc == 100) {
+        snprintf(buf, sizeof(buf), " 100");
+    } else {
+        if (current > 0) {
+            snprintf(buf, sizeof(buf), " %2d+", soc);
+        } else {
+            snprintf(buf, sizeof(buf), " %2d%%", soc);
+        }
+    }
+
+    lcd_print(16, 0, buf);
+}

main/drivers/power.h

@@ -3,13 +3,18 @@
 
 #include "SparkFunBQ27441/SparkFunBQ27441.h"
 
+extern volatile uint32_t battery_charge;
+
 void bat_monitor_task(void* arg);
 
 /// Initializes the battery gas guage for getting battery stats.
 void init_power_board();
 
-/// @brief Gets the battery voltage
+/// @brief Gets the battery voltage.
 /// @return battery voltage in mV.
 uint16_t get_bat_voltage();
 
+/// @brief Prints the battery section of the header to the char_lcd. (row 0, columns 17-19)
+void lcd_print_header_bat();
+
 #endif /* POWER_H */

main/drivers/sd.h

@@ -1,7 +1,6 @@
 #ifndef SD_H
 #define SD_H
 
-#include "main.h"
 #include <string.h>
 #include <sys/unistd.h>
 #include <sys/stat.h>
@@ -16,16 +15,10 @@ extern sdmmc_card_t *card;
 #define SD_PIN_CLK GPIO_NUM_39
 #define SD_PIN_CMD GPIO_NUM_40
 #define SD_PIN_D0  GPIO_NUM_38
+#define SD_PIN_D1  GPIO_NUM_45
 #define SD_PIN_D2  GPIO_NUM_42
 #define SD_PIN_D3  GPIO_NUM_41
 
-#ifdef CONTROL_REV_2_0
-#define SD_PIN_D1  GPIO_NUM_45
-#endif
-#ifdef CONTROL_REV_2_1
-#define SD_PIN_D1  GPIO_NUM_47
-#endif
-
 /// @brief Initializes the SD card
 ///
 /// This requires the char_lcd to have been initialized.

main/drivers/starcode.cpp

@@ -0,0 +1,256 @@
+#include "starcode.h"
+#include <vector>
+#include <algorithm>
+#include <string.h>
+#include <stdint.h>
+#include <esp_log.h>
+#include "drivers/bottom_half.h"
+#include "char_lcd.h"
+#include "esp_timer.h"
+
+static const char* TAG = "star_code";
+
+volatile bool handling_new_starcodes = false;
+static volatile bool system_initialized = false;
+
+// TODO: use the semaphore, convert to RWLock?
+static volatile SemaphoreHandle_t star_codes_sem;
+static std::vector<StarCodeEntry> star_codes;
+
+static const char EMPTY_STAR_CODE_HEADER[] = "          ";
+
+esp_timer_handle_t starcode_delay_timer;
+/// @brief `true` if we are delaying for a starcode
+static volatile bool delaying_for_starcode;
+static volatile StarCodeEntry* current_starcode = nullptr;
+/// @brief `true` when we are handling user input for a starcode
+static volatile bool doing_starcode = false;
+static uint16_t starcode_waiting_on_release;
+static char current[STARCODE_MAX_LEN + 1];
+static size_t current_idx;
+
+static void starcode_trigger_cb(void* arg) {
+    (void) arg;
+
+    delaying_for_starcode = false;
+
+    if (current_starcode != nullptr) {
+        if (current_starcode->triggered_sem != nullptr)
+            xSemaphoreGive(current_starcode->triggered_sem);
+        if (current_starcode->callback != nullptr) 
+            (current_starcode->callback)();
+        
+        current_starcode = nullptr;
+    }
+
+    // TODO: rename star code everywhere to starcode
+    lcd_print_header();
+}
+
+
+void star_code_handle_keypad(uint16_t* just_pressed, uint16_t* just_released) {
+    if ((!delaying_for_starcode) && handling_new_starcodes &&  (*just_pressed & (1 << KeypadKey::star))) {
+        current_idx = 0;
+        current[current_idx] = '\0';
+        doing_starcode = true;
+    }
+    if (doing_starcode) {
+        // If we get a press while handling a starcode, we also want to capture the release of that key.
+        starcode_waiting_on_release |= *just_pressed;
+
+        KeypadKey key;
+        while (take_key(&key, just_pressed)) {
+            if (key == KeypadKey::star) {
+                current_idx = 0;
+                current[current_idx] = '\0';
+            } else if (key == KeypadKey::pound) {
+                doing_starcode = false;
+                if (current_idx != 0) {
+                    trigger_star_code(current);
+                }
+            } else {
+                // shift the digits left if neccesary
+                if (current_idx >= STARCODE_MAX_LEN) {
+                    for (int i = 1; i < current_idx; i++) {
+                        current[i-1] = current[i];
+                    }
+                    current_idx--;
+                }
+                // append the character
+                current[current_idx++] = char_of_keypad_key(key);
+                current[current_idx] = '\0';
+            }
+            lcd_print_header_star_code();
+        }
+    }
+
+    // capture any releases from starcodes
+    uint16_t new_just_released = (*just_released) & (~starcode_waiting_on_release);
+    starcode_waiting_on_release = starcode_waiting_on_release & (~*just_released);
+    *just_released = new_just_released;
+}
+
+void init_star_code_system() {
+    star_codes_sem = xSemaphoreCreateBinary();
+    xSemaphoreGive(star_codes_sem);
+
+    const esp_timer_create_args_t timer_args = {
+        .callback = &starcode_trigger_cb,
+        .arg = nullptr,
+        .dispatch_method = ESP_TIMER_TASK,
+        .name = "starcode_trigger",
+        .skip_unhandled_events = false,
+    };
+    
+    ESP_ERROR_CHECK(esp_timer_create(&timer_args, &starcode_delay_timer));
+
+    handling_new_starcodes = true;
+    system_initialized = true;
+}
+
+/// Checks if a triggered code matches an expected code.
+/// @return true iff the codes match, where '*'s in the expected code can match any character in the triggered code
+static bool check_code_match(const char* triggered, const char* expected) {
+    size_t triggered_len = strlen(triggered);
+    size_t match_len = strlen(triggered);
+
+    if (triggered_len != match_len) 
+        return false;
+
+    for (int i = 0; i < triggered_len; i++) {
+        if (!(expected[i] == '*' || expected[i] == triggered[i])) {
+            return false;
+        }
+    }
+
+    return true;
+}
+
+bool add_star_code(StarCodeEntry code) {
+    ESP_LOGI(TAG, "Adding starcode: %s", code.code);
+    if (code.code == nullptr || strlen(code.code) > STARCODE_MAX_LEN) {
+        ESP_LOGW(TAG, "invalid code");
+        return false;
+    }
+    if (code.display_text != nullptr && strlen(code.display_text) > STARCODE_DISPLAY_TEXT_MAX_LEN) {
+        ESP_LOGW(TAG, "invalid display_text");
+        return false;
+    }
+
+    // check for a existing entry
+    auto it = std::find_if(star_codes.begin(), star_codes.end(), [&](const StarCodeEntry& other) {
+        return check_code_match(code.code, other.code);
+    });
+
+    if (it != star_codes.end()) {
+        // existing star code found!
+        ESP_LOGW(TAG, "Duplicate starcode %s", code.code);
+        return false;
+    }
+
+    star_codes.push_back(code);
+    return true;
+}
+
+bool add_star_codes(const StarCodeEntry* codes, size_t len) {
+    bool success = true;
+    for (int i = 0; i < len; i++) {
+        if (!add_star_code(codes[i])) {
+            success = false;
+        }
+    }
+    return success;
+}
+
+bool rm_star_code(const char* code) {
+    ESP_LOGI(TAG, "Removing starcode: %s", code);
+
+    auto it = std::find_if(star_codes.begin(), star_codes.end(), [&](const StarCodeEntry& star_code) {
+        return strcmp(code, star_code.code) == 0;
+    });
+    
+    if (it == star_codes.end()) {
+        ESP_LOGW(TAG, "Failed to remove star code %s", code);
+        return false;
+    }
+
+    star_codes.erase(it);
+    return true;
+}
+
+bool rm_star_codes(const StarCodeEntry* codes, size_t len) {
+    bool success = true;
+    for (int i = 0; i < len; i++) {
+        if (!rm_star_code(codes[i].code)) {
+            success = false;
+        }
+    }
+    return success;
+}
+
+bool rm_star_codes_str(const char** codes, size_t len) {
+    bool success = true;
+    for (int i = 0; i < len; i++) {
+        if (!rm_star_code(codes[i])) {
+            success = false;
+        }
+    }
+    return success;
+}
+
+void clear_star_codes() {
+    star_codes.clear();
+}
+
+bool trigger_star_code(const char* code) {
+    auto it = std::find_if(star_codes.begin(), star_codes.end(), [&](const StarCodeEntry& other) {
+        return check_code_match(code, other.code);
+    });
+
+    uint64_t delay_us = 2'000'000;
+    delaying_for_starcode = true;
+    if (it != star_codes.end()) {
+        current_starcode = &*it;
+        delay_us = current_starcode->delay_us;
+    }
+
+    ESP_ERROR_CHECK(esp_timer_start_once(starcode_delay_timer, delay_us));
+
+    return current_starcode != nullptr;
+}
+
+void pause_star_code_system() {
+    doing_starcode = false;
+    handling_new_starcodes = false;
+}
+
+void resume_star_code_system() {
+    handling_new_starcodes = system_initialized;
+}
+
+void lcd_print_header_star_code() {
+    if (!lcd_header_enabled()) return;
+
+    // TODO: consider upping the display text size to be able to overwrite the game_state area.
+    if (delaying_for_starcode) {
+        if (current_starcode == nullptr) {
+            lcd_print(0, 0, "Invalid starcode    ");
+        } else if (current_starcode->display_text != nullptr) {
+            char buf[21];
+            snprintf(buf, sizeof(buf), "%-20s", current_starcode->display_text);
+            lcd_print(0, 0, buf);
+        } else {
+            lcd_print(0, 0, EMPTY_STAR_CODE_HEADER);
+        }
+    } else if (doing_starcode) {
+        char buf[STARCODE_MAX_LEN + 2];
+        snprintf(buf, sizeof(buf), "*%-9s", current);
+        lcd_print(0, 0, buf);
+    } else {
+        lcd_print(0, 0, EMPTY_STAR_CODE_HEADER);
+    }
+}
+
+bool lcd_starcode_displaying_result() {
+    return delaying_for_starcode;
+}

main/drivers/starcode.h

@@ -0,0 +1,90 @@
+#ifndef STAR_CODE_H
+#define STAR_CODE_H
+
+#include <stddef.h>
+#include <freertos/FreeRTOS.h>
+#include <freertos/semphr.h>
+
+/// The max length of a starcode (not counting the star)
+#define STARCODE_MAX_LEN 9
+#define STARCODE_DISPLAY_TEXT_MAX_LEN 20
+
+/// @brief A handler for a specific star code
+struct StarCodeEntry {
+    /// @brief The star code without the star
+    ///
+    /// This must be <= 9 characters.
+    ///
+    /// You may include a * in the code to match on any character
+    const char* code;
+    /// @brief The text to display when the star code is entered (or null).
+    /// 
+    /// This must be <= 20 characters.
+    const char* display_text;
+    /// @brief The number of microseconds to delay when the star code is entered before calling the handler.
+    uint64_t delay_us;
+    /// @brief The function to call when the star code is entered.
+    /// Can be null.
+    void (*callback)(void);
+    /// @brief The binary semaphore that will be given when this star code is triggered.
+    /// Can be null.
+    SemaphoreHandle_t triggered_sem;
+};
+
+/// @brief Initializes the star code system.
+void init_star_code_system();
+
+/// @brief Handles any keypad presses and releases before they bubble up to the rest of the BLK_BOX.
+void star_code_handle_keypad(uint16_t* just_pressed, uint16_t* just_released);
+
+/// @brief Adds a star code to be handled.
+/// @param code the star code to add
+/// @return true iff the star code was added
+bool add_star_code(StarCodeEntry code);
+
+/// @brief Adds a list of star codes to be handled.
+/// @param codes the list of star codes to add
+/// @param len the length of the list
+/// @return true iff all the star codes were added
+bool add_star_codes(const StarCodeEntry* codes, size_t len);
+
+/// @brief removes a star code to stop handling it.
+/// @param code the star code to remove
+/// @return true iff the star code was removed
+bool rm_star_code(const char* code);
+
+/// @brief removes all given star codes to stop handling them.
+/// @param codes the list of star codes to remove
+/// @param len the length of the list
+/// @return true iff all star codes were removed
+bool rm_star_codes(const StarCodeEntry* codes, size_t len);
+
+/// @brief removes all given star codes to stop handling them.
+/// @param codes the list of star codes to remove
+/// @param len the length of the list
+/// @return true iff all star codes were removed
+bool rm_star_codes_str(const char** codes, size_t len);
+
+/// @brief clears all star codes.
+void clear_star_codes();
+
+/// @brief Triggers the given star code.
+/// @param code the star code to trigger (without the *)
+/// @return true iff a star code was triggered
+bool trigger_star_code(const char* code);
+
+
+/// @brief Starts/stops the star code system from handling new star codes.
+/// If one is being handled currently, it is canceled.
+void set_star_code_sys_enabled(bool enable);
+
+/// @return `true` iff the star code system is handling star codes.
+bool star_code_sys_enabled();
+
+/// @brief Prints the star code section of the header to the char_lcd. (row 0, columns 0-9)
+void lcd_print_header_star_code();
+
+/// @return `true` iff the starcode system is using the full header.
+bool lcd_starcode_displaying_result();
+
+#endif /* STAR_CODE_H */

main/drivers/state_tracking.cpp

@@ -1,6 +1,7 @@
 #include "state_tracking.h"
 #include <unistd.h>
 #include <vector>
+#include "wlvgl.h"
 
 static const char* PLAYBACK_TAG = "playback";
 
@@ -24,7 +25,6 @@ TaskHandle_t playback_task_handle;
 
 static volatile state_t state = STATE_IDLE;
 
-
 /// @brief Periodically flushes and syncs (if neccesary) the output stream.
 /// @param arg unused.
 static void flush_file_task(void* arg) {
@@ -171,6 +171,7 @@ bool start_recording() {
     state = STATE_RECORDING;
     recording_start_time = xTaskGetTickCount();
     xTaskCreate(flush_file_task, "flush_recording", 2048, NULL, 2, &flush_file_task_handle);
+    reset_wlv_tables(); // TODO: generify this
     return true;
 }
 
@@ -198,6 +199,7 @@ bool start_playback() {
     state = STATE_PLAYBACK;
     playback_start_time = xTaskGetTickCount();
     xTaskCreate(playback_task, "playback", 4096, NULL, 2, &playback_task_handle);
+    reset_wlv_tables(); // TODO: generify this
     return true;
 }
 

main/drivers/tft.cpp

@@ -26,7 +26,7 @@ static bool notify_lvgl_flush_ready(
     esp_lcd_panel_io_event_data_t *edata,
     void *user_ctx
 ) {
-    lv_disp_drv_t *disp_driver = (lv_disp_drv_t *)user_ctx;
+    lv_disp_drv_t* disp_driver = (lv_disp_drv_t *)user_ctx;
     lv_disp_flush_ready(disp_driver);
     return false;
 }

main/drivers/wires.cpp

@@ -32,7 +32,7 @@ void init_wires(void) {
         .scl_pullup_en = GPIO_PULLUP_ENABLE,
         .master = {
             .clk_speed = 100000,
-        }
+        },
     };
 
     gpio_reset_pin(PIN_WIRES_SDA);

main/drivers/wires.h

@@ -8,13 +8,7 @@
 #include "drivers/char_lcd.h"
 #include "drivers/game_timer.h"
 #include "main.h"
-
+#include "perh.h"
-
-#define PIN_PERH0 GPIO_NUM_6
-#define PIN_PERH1 GPIO_NUM_5
-#define PIN_PERH2 GPIO_NUM_4
-#define PIN_PERH3 GPIO_NUM_2
-#define PIN_PERH4 GPIO_NUM_1
 
 #define WIRES_PIN_DELTA PIN_PERH3
 #define PIN_WIRES_SDA PIN_PERH1

main/drivers/wlvgl.cpp

@@ -0,0 +1,152 @@
+#include "wlvgl.h"
+#include <vector>
+
+/// A table that maps an incrementing integer to a style reference
+static std::vector<lv_style_t*> style_table;
+
+/// A table that maps an incrementing integer to a style reference
+static std::vector<lv_obj_t*> obj_table;
+
+static int index_of_style(lv_style_t* style) {
+    for (size_t i = 0; i < style_table.size(); i++) {
+        if (style_table[i] == style) {
+            return i;
+        }
+    }
+    return -1;
+}
+
+static int index_of_obj(lv_obj_t* obj) {
+    for (size_t i = 0; i < obj_table.size(); i++) {
+        if (obj_table[i] == obj) {
+            return i;
+        }
+    }
+    return -1;
+}
+
+
+void reset_wlv_tables() {
+    style_table.clear();
+    obj_table.clear();
+}
+
+lv_obj_t * wlv_obj_create(lv_obj_t* parent) {
+    // initialize the obj
+    lv_obj_t* value = lv_obj_create(parent);
+    
+    if (is_state_tracking()) {
+        // add the style to the table
+        obj_table.push_back(value);
+        size_t obj_index = obj_table.size();
+
+		char buf[8];
+		sprintf(buf, "%d", obj_index);
+		event_occured("lv_obj_create", buf);
+	}
+    return value;
+}
+
+void wlv_style_init(lv_style_t* style) {
+    // initialize the style
+    lv_style_init(style);
+
+    if (is_state_tracking()) {
+        // add the style to the table
+        style_table.push_back(style);
+        size_t style_index = style_table.size();
+
+		char buf[8];
+		sprintf(buf, "%d", style_index);
+		event_occured("lv_style_init", buf);
+	}
+}
+
+void wlv_obj_set_style_bg_color(lv_obj_t* obj, lv_color_t value, lv_style_selector_t selector) {
+    lv_obj_set_style_bg_color(obj, value, selector);
+
+    if (is_state_tracking()) {
+        int obj_index = index_of_obj(obj);
+		char buf[64];
+		sprintf(buf, "%d,%d,%ld", obj_index, value.full, selector);
+		event_occured("lv_obj_set_style_bg_color", buf);
+	}
+}
+
+
+void wlv_style_set_text_color() {
+    
+}
+
+void wlv_style_set_text_align() {
+
+}
+
+void wlv_obj_align() {
+
+}
+
+void wlv_obj_set_size() {
+
+}
+
+void wlv_obj_add_style() {
+
+}
+
+void wlv_obj_set_style_text_align() {
+
+}
+
+void wlv_label_set_text() {
+
+}
+
+void wlv_scr_load() {
+
+}
+
+void wlv_scr_act() {
+
+}
+
+void wlv_style_set_bg_color() {
+
+}
+
+void wlv_style_set_bg_opa() {
+
+}
+
+void wlv_obj_set_width() {
+
+}
+
+void wlv_style_set_text_font() {
+    
+}
+
+void wlv_label_create() {
+
+}
+
+void wlv_obj_del() {
+
+}
+
+void wlv_obj_center() {
+
+}
+
+void wlv_obj_remove_style() {
+
+}
+
+void wlv_obj_add_flag() {
+
+}
+
+void wlv_obj_clear_flag() {
+
+}
+

main/drivers/wlvgl.h

@@ -0,0 +1,32 @@
+#ifndef WLVGL_H
+#define WLVGL_H
+
+#include "lvgl.h"
+#include "state_tracking.h"
+
+void reset_wlv_tables();
+
+lv_obj_t* wlv_obj_create(lv_obj_t* parent);
+void wlv_style_init(lv_style_t* style);
+void wlv_obj_set_style_bg_color(struct _lv_obj_t * obj, lv_color_t value, lv_style_selector_t selector);
+void wlv_style_set_text_color();
+void wlv_style_set_text_align();
+void wlv_obj_align();
+void wlv_obj_set_size();
+void wlv_obj_add_style();
+void wlv_obj_set_style_text_align();
+void wlv_label_set_text();
+void wlv_scr_load();
+void wlv_scr_act();
+void wlv_style_set_bg_color();
+void wlv_style_set_bg_opa();
+void wlv_obj_set_width();
+void wlv_style_set_text_font();
+void wlv_label_create();
+void wlv_obj_del();
+void wlv_obj_center();
+void wlv_obj_remove_style();
+void wlv_obj_add_flag();
+void wlv_obj_clear_flag();
+
+#endif /* WLVGL_H */

main/helper.cpp

@@ -17,76 +17,10 @@ void clean_bomb(void) {
     set_module_sseg_raw(clear);
 
     // clear char lcd
-    lcd_clear();
     lcd_set_cursor_vis(false);
-    lcd_cursor_home();
-}
-
-static const int STRING_MAX_LEN = 8;
-void do_star_codes(StarCodeHandler* star_codes, int star_codes_len) {
-    KeypadKey key;
-
-    int current_idx = 0;
-    char current[STRING_MAX_LEN+1] = {0};
-
-    while (1) {
-        while (get_keypad_pressed(&key)) {
-            if (key == KeypadKey::star) {
-                current[0] = '*';
-                for (int i = 1; i < STRING_MAX_LEN; i++) {
-                    current[i] = 0;
-                }
-                current_idx = 1;
-            } else if (key == KeypadKey::pound) {
-                // submit
-                if (current[0] == '\0') {
-                    continue;
-                }
-
-                bool hit = false;
-                for (int i = 0; i < star_codes_len; i++) {
-                    StarCodeHandler sch = star_codes[i];
-                    if (strcmp(current, sch.code) == 0) {
-                        hit = true;
-                        lcd_print(1, 2, sch.display_text);
-                        vTaskDelay(pdMS_TO_TICKS(2000));
-                        if (sch.callback != nullptr) {
-                            (sch.callback)();
-                        }
-                        if (sch.should_exit) {
-                            return;
-                        }
-                        break;
-                    }
-                }
-                
-                if (!hit) {
-                    lcd_print(1, 2, "Invalid Star Code");
-                    vTaskDelay(pdMS_TO_TICKS(2000));
-                }
-                
-                // clear
-                for (int i = 0; i < STRING_MAX_LEN; i++) {
-                    current[i] = 0;
-                }
-                current_idx = 0;
-            } else {
-                // out of room. skip
-                if (current_idx >= STRING_MAX_LEN) break;
-                // no code started.
-                if (current[0] != '*') continue;
-
-                char c = char_of_keypad_key(key);
-                current[current_idx++] = c;
-            }
-            // ESP_LOGI(STEP0_TAG, "Pressed: %c", c);
-
     lcd_clear();
-            lcd_print(1, 1, current);
-        }
 
-        vTaskDelay(pdMS_TO_TICKS(10));
+    // TODO: add stuff for starcode system
-    }
 }
 
 static lv_obj_t* old_scr;
@@ -120,7 +54,6 @@ void display_game_results(void) {
     
     lv_style_init(&game_results_style);
     lv_style_set_text_color(&game_results_style, lv_color_white());
-    // lv_style_set_bg_color(&game_results_style, lv_color_black());
     lv_style_set_text_align(&game_results_style, LV_TEXT_ALIGN_LEFT);
 
     overall_results_label = lv_label_create(scr);

main/helper.h

@@ -10,20 +10,11 @@
 #include "drivers/speaker.h"
 #include "drivers/tft.h"
 
-struct StarCodeHandler {
-    const char* code;
-    const char* display_text;
-    bool should_exit;
-    void (*callback)(void);
-};
-
 // TODO: add something for RNG.
-// TODO: add something for colors, to make everything consistant.
 
 /// Clears most persistant bomb state
 void clean_bomb(void);
 void poster_child_task(void* arg);
-void do_star_codes(StarCodeHandler* star_codes, int star_codes_len);
 void display_game_results();
 
 #endif /* HELPER_H */

main/main.cpp

@@ -13,102 +13,19 @@
 #include "helper.h"
 
 #include "steps/step0.h"
-#include "steps/p001_step1.h"
+#include "steps/step1.h"
-#include "steps/p001_step2.h"
+#include "steps/step2.h"
-#include "steps/p001_step3.h"
+#include "steps/step3.h"
-#include "steps/p001_step4.h"
+#include "steps/step4.h"
-#include "steps/p001_step5.h"
+#include "steps/step5.h"
-#include "steps/p001_step6.h"
+#include "steps/step6.h"
-#include "steps/p002_step1.h"
-#include "steps/p002_step2.h"
-#include "steps/p002_step3.h"
-#include "steps/p002_step4.h"
-#include "steps/p002_step5.h"
-#include "steps/p002_step6.h"
 
 static const char *TAG = "main";
 uint32_t initial_game_time = 90*60*1000 + 1000;
 uint32_t skip_to_step = 0;
 uint32_t current_step = 0;
-uint32_t puzzle = 0;
 extern uint32_t total_strikes;
 
-static void do_p001() {
-    set_game_time(initial_game_time);
-    start_game_timer();
-    total_strikes = 0;
-    clean_bomb();
-    current_step = 1;
-    if (skip_to_step <= 1) p001_step1();
-    step_finish_times[current_step-1] = get_game_time();
-    clean_bomb();
-    current_step = 2;
-    if (skip_to_step <= 2) p001_step2();
-    step_finish_times[current_step-1] = get_game_time();
-    clean_bomb();
-    current_step = 3;
-    if (skip_to_step <= 3) p001_step3();
-    step_finish_times[current_step-1] = get_game_time();
-    clean_bomb();
-    current_step = 4;
-    if (skip_to_step <= 4) p001_step4();
-    step_finish_times[current_step-1] = get_game_time();
-    clean_bomb();
-    current_step = 5;
-    if (skip_to_step <= 5) p001_step5();
-    step_finish_times[current_step-1] = get_game_time();
-    clean_bomb();
-    current_step = 6;
-    if (skip_to_step <= 6) p001_step6();
-    start_game_timer();
-    step_finish_times[current_step-1] = get_game_time();
-    clean_bomb();
-
-    stop_game_timer();
-    ESP_LOGI(TAG, "Bomb has been diffused. Counter-Terrorists win.");
-    play_clip_wav(MOUNT_POINT "/diffuse.wav", true, false, 3, 0);
-
-    display_game_results();
-}
-
-static void do_p002() {
-    set_game_time(initial_game_time);
-    start_game_timer();
-    total_strikes = 0;
-    clean_bomb();
-    current_step = 1;
-    if (skip_to_step <= 1) p002_step1();
-    step_finish_times[current_step-1] = get_game_time();
-    clean_bomb();
-    current_step = 2;
-    if (skip_to_step <= 2) p002_step2();
-    step_finish_times[current_step-1] = get_game_time();
-    clean_bomb();
-    current_step = 3;
-    if (skip_to_step <= 3) p002_step3();
-    step_finish_times[current_step-1] = get_game_time();
-    clean_bomb();
-    current_step = 4;
-    if (skip_to_step <= 4) p002_step4();
-    step_finish_times[current_step-1] = get_game_time();
-    clean_bomb();
-    current_step = 5;
-    if (skip_to_step <= 5) p002_step5();
-    step_finish_times[current_step-1] = get_game_time();
-    clean_bomb();
-    current_step = 6;
-    if (skip_to_step <= 6) p002_step6();
-    start_game_timer();
-    step_finish_times[current_step-1] = get_game_time();
-    clean_bomb();
-
-    stop_game_timer();
-    ESP_LOGI(TAG, "Bomb has been diffused. Counter-Terrorists win.");
-    play_clip_wav(MOUNT_POINT "/diffuse.wav", true, false, 3, 0);
-
-    display_game_results();
-}
-
 extern "C" void app_main(void) {
     printf("app_main\n");
 
@@ -120,11 +37,52 @@ extern "C" void app_main(void) {
     vTaskDelay(pdMS_TO_TICKS(1000));
 
     clean_bomb();
+    lcd_do_splash();
     step0();
 
-    if (puzzle == 1) {
+    // set_recording_source(stdout, false);
-        do_p001();
+    FILE* record_file = fopen(MOUNT_POINT "/record.txt", "w");
-    } else {
+    if (record_file == nullptr) {
-        do_p002();
+        ESP_LOGE("main", "failed to open record.txt");
     }
+    set_recording_source(record_file, true);
+    start_recording();
+
+    set_game_time(initial_game_time);
+    start_game_timer();
+    total_strikes = 0;
+    clean_bomb();
+    current_step = 1;
+    if (skip_to_step <= 1) step1();
+    step_finish_times[current_step-1] = get_game_time();
+    clean_bomb();
+    current_step = 2;
+    if (skip_to_step <= 2) step2();
+    step_finish_times[current_step-1] = get_game_time();
+    clean_bomb();
+    current_step = 3;
+    if (skip_to_step <= 3) step3();
+    step_finish_times[current_step-1] = get_game_time();
+    clean_bomb();
+    current_step = 4;
+    if (skip_to_step <= 4) step4();
+    step_finish_times[current_step-1] = get_game_time();
+    clean_bomb();
+    current_step = 5;
+    if (skip_to_step <= 5) step5();
+    step_finish_times[current_step-1] = get_game_time();
+    clean_bomb();
+    current_step = 6;
+    if (skip_to_step <= 6) step6();
+    start_game_timer();
+    step_finish_times[current_step-1] = get_game_time();
+    clean_bomb();
+
+    stop_game_timer();
+    ESP_LOGI(TAG, "Bomb has been diffused. Counter-Terrorists win.");
+    play_clip_wav(MOUNT_POINT "/diffuse.wav", true, false, 3, 0);
+
+    display_game_results();
+    stop_recording();
+
 }

main/main.h

@@ -1,9 +1,6 @@
 #ifndef MAIN_H
 #define MAIN_H
 
-#define CONTROL_REV_2_0
-// #define CONTROL_REV_2_1
-
 #include <cstddef>
 
 constexpr size_t N_STEPS = 6;

main/steps/CMakeLists.txt

@@ -1,18 +1,12 @@
 set(SOURCES 
     "setup_wires.cpp"
     "step0.cpp"
-    "p001_step1.cpp"
+    "step1.cpp"
-    "p001_step2.cpp"
+    "step2.cpp"
-    "p001_step3.cpp"
+    "step3.cpp"
-    "p001_step4.cpp"
+    "step4.cpp"
-    "p001_step5.cpp"
+    "step5.cpp"
-    "p001_step6.cpp"
+    "step6.cpp"
-    "p002_step1.cpp"
-    "p002_step2.cpp"
-    "p002_step3.cpp"
-    "p002_step4.cpp"
-    "p002_step5.cpp"
-    "p002_step6.cpp"
     "wires_puzzle.cpp"
 )
 

main/steps/p001_step1.h

@@ -1,10 +0,0 @@
-#ifndef P001_STEP_1_H
-#define P001_STEP_1_H
-
-#include <random>
-#include "../drivers/all.h"
-#include "../helper.h"
-
-void p001_step1(void);
-
-#endif /* P001_STEP_1_H */

main/steps/p001_step2.h

@@ -1,16 +0,0 @@
-#ifndef P001_STEP_2_H
-#define P001_STEP_2_H
-
-#include "../drivers/bottom_half.h"
-#include "../drivers/wires.h"
-#include "../drivers/game_timer.h"
-#include "../drivers/leds.h"
-#include "../drivers/speaker.h"
-#include "../helper.h"
-#include <iostream>
-#include <random>
-#include <map>
-
-void p001_step2(void);
-
-#endif /* P001_STEP_2_H */

main/steps/p001_step3.h

@@ -1,10 +0,0 @@
-#ifndef P001_STEP_3_H
-#define P001_STEP_3_H
-
-#include <random>
-#include "../drivers/all.h"
-#include "../helper.h"
-
-void p001_step3(void);
-
-#endif /* P001_STEP_3_H */

main/steps/p001_step4.h

@@ -1,18 +0,0 @@
-#ifndef P001_STEP_4_H
-#define P001_STEP_4_H
-
-#include <random>
-#include "../drivers/tft.h"
-#include "../drivers/speaker.h"
-#include "../drivers/bottom_half.h"
-#include "../drivers/game_timer.h"
-#include "../drivers/wires.h"
-#include "../drivers/char_lcd.h"
-#include "../helper.h"
-
-#define TETRIS_USE_FLASH_IMG
-#define TETRIS_USE_FLASH_BG_IMG
-
-void p001_step4(void);
-
-#endif /* P001_STEP_4_H */

main/steps/p001_step5.h

@@ -1,20 +0,0 @@
-#ifndef P001_STEP_5_H
-#define P001_STEP_5_H
-
-#include "../drivers/bottom_half.h"
-#include "../drivers/game_timer.h"
-#include "../drivers/char_lcd.h"
-#include "../drivers/leds.h"
-#include "../drivers/wires.h"
-#include "../helper.h"
-#include <random>
-#include <iostream>
-#include <set>
-#include <map>
-#include <vector>
-#include <cmath>
-#include <array>
-
-void p001_step5(void);
-
-#endif /* P001_STEP_5_H */

main/steps/p001_step6.h

@@ -1,10 +0,0 @@
-#ifndef P001_STEP_6_H
-#define P001_STEP_6_H
-
-#include "wires_puzzle.h"
-#include "../drivers/all.h"
-#include "../helper.h"
-
-void p001_step6(void);
-
-#endif /* P001_STEP_6_H */

main/steps/p002_step1.cpp

@@ -1,8 +0,0 @@
-#include "p002_step1.h"
-
-__attribute__((unused))
-static const char *TAG = "step1";
-
-void p002_step1(void) {
-    // TODO: implement step 1
-}

main/steps/p002_step1.h

@@ -1,8 +0,0 @@
-#ifndef P002_STEP_1_H
-#define P002_STEP_1_H
-
-#include "../drivers/all.h"
-
-void p002_step1(void);
-
-#endif /* P002_STEP_1_H */

main/steps/p002_step2.cpp

@@ -1,272 +0,0 @@
-#include "p002_step2.h"
-#include "helper.h"
-#include <cstdio>
-#include <cstring>
-#include <random>
-
-__attribute__((unused))
-static const char *TAG = "step2";
-
-static const uint8_t BOARD_COUNT = 9;
-static const uint8_t BOARD_SIZE = 6;
-static const uint8_t PART_SHOTS[3] = {20, 15, 12};
-static const char *PART_LABELS[3] = {"Alpha", "Bravo", "Charlie"};
-static const char ROW_LABELS[BOARD_SIZE] = {'A', 'B', 'C', 'D', 'E', 'F'};
-
-static std::random_device random_device;
-static std::mt19937 random_engine(random_device());
-static std::uniform_int_distribution<int> random_board(0, BOARD_COUNT - 1);
-
-static const uint8_t ALL_BOARDS[BOARD_COUNT][BOARD_SIZE][BOARD_SIZE] = {
-    {
-        {0,0,0,0,0,0},
-        {0,0,1,1,1,1},
-        {0,1,0,0,0,0},
-        {0,1,0,1,1,0},
-        {0,1,0,0,0,0},
-        {0,0,0,1,0,0},
-    },
-    {
-        {0,1,0,0,0,0},
-        {0,1,0,1,1,0},
-        {0,1,0,0,0,0},
-        {0,0,0,1,0,0},
-        {0,0,0,0,0,0},
-        {1,1,1,1,0,0},
-    },
-    {
-        {0,0,1,1,1,1},
-        {0,1,0,0,0,0},
-        {0,1,0,0,0,0},
-        {0,1,0,0,0,0},
-        {0,0,1,0,0,0},
-        {1,0,1,0,0,0},
-    },
-    {
-        {1,0,0,0,0,0},
-        {1,0,0,0,0,0},
-        {1,0,0,0,0,1},
-        {1,0,0,0,0,1},
-        {0,0,0,0,1,0},
-        {0,1,1,1,0,0},
-    },
-    {
-        {1,1,0,1,0,0},
-        {0,0,0,1,0,0},
-        {0,1,0,1,0,0},
-        {0,0,0,0,0,0},
-        {0,1,1,1,1,0},
-        {0,0,0,0,0,0},
-    },
-    {
-        {0,0,1,0,1,0},
-        {1,0,1,0,1,0},
-        {0,0,1,0,1,0},
-        {0,0,1,0,0,0},
-        {0,0,0,0,0,0},
-        {1,1,0,0,0,0},
-    },
-    {
-        {0,0,0,0,0,1},
-        {1,1,0,0,0,0},
-        {0,0,0,0,1,0},
-        {1,1,1,0,1,0},
-        {0,0,0,0,1,0},
-        {0,0,0,0,1,0},
-    },
-    {
-        {0,1,1,0,0,0},
-        {0,0,0,0,1,0},
-        {0,0,0,0,1,0},
-        {0,0,1,0,1,0},
-        {0,0,0,0,0,0},
-        {1,1,1,1,0,0},
-    },
-    {
-        {0,0,0,0,0,0},
-        {1,0,0,0,0,0},
-        {0,1,0,0,1,1},
-        {0,1,0,1,0,0},
-        {0,1,0,1,0,0},
-        {0,1,0,1,0,0},
-    },
-};
-
-static bool row_from_button_state(uint8_t state, uint8_t *row) {
-    switch (state) {
-        case 0b1100: *row = 0; return true; // B1 + B2 -> A
-        case 0b1010: *row = 1; return true; // B1 + B3 -> B
-        case 0b1001: *row = 2; return true; // B1 + B4 -> C
-        case 0b0110: *row = 3; return true; // B2 + B3 -> D
-        case 0b0101: *row = 4; return true; // B2 + B4 -> E
-        case 0b0011: *row = 5; return true; // B3 + B4 -> F
-        default: return false;
-    }
-}
-
-static bool column_from_switch_state(uint8_t state, uint8_t *col) {
-    uint8_t value = state & 0xF;
-    if (value >= 1 && value <= 6) {
-        *col = value - 1;
-        return true;
-    }
-    return false;
-}
-
-static int count_ship_segments(const uint8_t board[BOARD_SIZE][BOARD_SIZE]) {
-    int count = 0;
-    for (uint8_t row = 0; row < BOARD_SIZE; ++row) {
-        for (uint8_t col = 0; col < BOARD_SIZE; ++col) {
-            if (board[row][col] == 1) {
-                ++count;
-            }
-        }
-    }
-    return count;
-}
-
-static void copy_board(uint8_t dest[BOARD_SIZE][BOARD_SIZE], const uint8_t src[BOARD_SIZE][BOARD_SIZE]) {
-    memcpy(dest, src, BOARD_SIZE * BOARD_SIZE);
-}
-
-static void render_status(const char *part_name,
-                          int shots_left,
-                          int hits,
-                          int misses,
-                          int row,
-                          int col,
-                          const char *info) {
-    char line0[21] = {0};
-    char line1[21] = {0};
-    char line2[21] = {0};
-    char line3[21] = {0};
-
-    snprintf(line0, sizeof(line0), "%s Shots:%2d", part_name, shots_left);
-    if (row >= 0 && col >= 0) {
-        snprintf(line1, sizeof(line1), "Row %c Col %d", ROW_LABELS[row], col + 1);
-    } else {
-        snprintf(line1, sizeof(line1), "Row - Col -");
-    }
-    snprintf(line2, sizeof(line2), "Hits:%2d Misses:%2d", hits, misses);
-    snprintf(line3, sizeof(line3), "%s", info);
-
-    lcd_clear();
-    lcd_print(0, 0, line0);
-    lcd_print(0, 1, line1);
-    lcd_print(0, 2, line2);
-    lcd_print(0, 3, line3);
-}
-
-void p002_step2(void) {
-    clean_bomb();
-
-    for (int part = 0; part < 3; ++part) {
-        const char *part_name = PART_LABELS[part];
-        const int max_shots = PART_SHOTS[part];
-        bool completed = false;
-
-        while (!completed) {
-            uint8_t board[BOARD_SIZE][BOARD_SIZE];
-            copy_board(board, ALL_BOARDS[random_board(random_engine)]);
-
-            int shots_taken = 0;
-            int hits = 0;
-            int misses = 0;
-            int current_row = -1;
-            int current_col = -1;
-            const char *status = "Hold 2 buttons + set switches";
-
-            render_status(part_name, max_shots - shots_taken, hits, misses, -1, -1, status);
-            clean_bomb();
-
-            while (true) {
-                uint8_t button_state = get_button_state();
-                uint8_t switch_state = get_switch_state();
-                uint8_t row = 0;
-                uint8_t col = 0;
-                bool row_valid = row_from_button_state(button_state, &row);
-                bool col_valid = column_from_switch_state(switch_state, &col);
-
-                if (row_valid) {
-                    current_row = row;
-                } else {
-                    current_row = -1;
-                }
-                if (col_valid) {
-                    current_col = col;
-                } else {
-                    current_col = -1;
-                }
-
-                KeypadKey key;
-                if (get_keypad_pressed(&key)) {
-                    if (key == KeypadKey::k5) {
-                        if (!row_valid) {
-                            status = "Invalid row selection";
-                        } else if (!col_valid) {
-                            status = "Invalid column value";
-                        } else {
-                            uint8_t cell = board[row][col];
-                            if (cell == 2 || cell == 3) {
-                                status = "Already fired there";
-                            } else {
-                                shots_taken += 1;
-                                if (cell == 1) {
-                                    board[row][col] = 3;
-                                    hits += 1;
-                                    status = "Hit!";
-                                    led_set(IndicatorLED::LED_LCD, LEDColor::LED_COLOR_RED);
-                                } else {
-                                    board[row][col] = 2;
-                                    misses += 1;
-                                    status = "Miss!";
-                                    led_set(IndicatorLED::LED_LCD, LEDColor::LED_COLOR_WHITE);
-                                }
-                                leds_flush();
-                                vTaskDelay(pdMS_TO_TICKS(250));
-                                led_set(IndicatorLED::LED_LCD, LEDColor::LED_COLOR_OFF);
-                                leds_flush();
-                            }
-                        }
-                    }
-                }
-
-                render_status(part_name,
-                              max_shots - shots_taken,
-                              hits,
-                              misses,
-                              current_row,
-                              current_col,
-                              status);
-
-                if (count_ship_segments(board) == 0) {
-                    completed = true;
-                    break;
-                }
-
-                if (shots_taken >= max_shots) {
-                    strike("Out of shots!");
-                    status = "Out of shots! Retry";
-                    render_status(part_name,
-                                  0,
-                                  hits,
-                                  misses,
-                                  current_row,
-                                  current_col,
-                                  status);
-                    vTaskDelay(pdMS_TO_TICKS(2000));
-                    break;
-                }
-
-                vTaskDelay(pdMS_TO_TICKS(50));
-            }
-        }
-
-        if (part < 2)
-            play_clip_wav(MOUNT_POINT "/partdone.wav", true, false, 0, 0);
-        clean_bomb();
-    }
-
-    play_clip_wav(MOUNT_POINT "/stepdone.wav", true, false, 1, 0);
-    lcd_clear();
-}

main/steps/p002_step2.h

@@ -1,8 +0,0 @@
-#ifndef P002_STEP_2_H
-#define P002_STEP_2_H
-
-#include "../drivers/all.h"
-
-void p002_step2(void);
-
-#endif /* P002_STEP_2_H */

main/steps/p002_step3.cpp

@@ -1,8 +0,0 @@
-#include "p002_step3.h"
-
-__attribute__((unused))
-static const char *TAG = "step3";
-
-void p002_step3(void) {
-    // TODO: implement step 3
-}

main/steps/p002_step3.h

@@ -1,8 +0,0 @@
-#ifndef P002_STEP_3_H
-#define P002_STEP_3_H
-
-#include "../drivers/all.h"
-
-void p002_step3(void);
-
-#endif /* P002_STEP_3_H */

main/steps/p002_step4.cpp

@@ -1,8 +0,0 @@
-#include "p002_step4.h"
-
-__attribute__((unused))
-static const char *TAG = "step4";
-
-void p002_step4(void) {
-    // TODO: implement step 4
-}

main/steps/p002_step4.h

@@ -1,8 +0,0 @@
-#ifndef P002_STEP_4_H
-#define P002_STEP_4_H
-
-#include "../drivers/all.h"
-
-void p002_step4(void);
-
-#endif /* P002_STEP_4_H */

main/steps/p002_step5.cpp

@@ -1,8 +0,0 @@
-#include "p002_step5.h"
-
-__attribute__((unused))
-static const char *TAG = "step5";
-
-void p002_step5(void) {
-    // TODO: implement step 5
-}

main/steps/p002_step5.h

@@ -1,8 +0,0 @@
-#ifndef P002_STEP_5_H
-#define P002_STEP_5_H
-
-#include "../drivers/all.h"
-
-void p002_step5(void);
-
-#endif /* P002_STEP_5_H */

main/steps/p002_step6.cpp

@@ -1,8 +0,0 @@
-#include "p002_step6.h"
-
-__attribute__((unused))
-static const char *TAG = "step6";
-
-void p002_step6(void) {
-    // TODO: implement step 6
-}

main/steps/p002_step6.h

@@ -1,8 +0,0 @@
-#ifndef P002_STEP_6_H
-#define P002_STEP_6_H
-
-#include "../drivers/all.h"
-
-void p002_step6(void);
-
-#endif /* P002_STEP_6_H */

main/steps/step0.cpp

@@ -5,7 +5,6 @@ static const char* TAG = "step0";
 
 extern uint32_t initial_game_time;
 extern uint32_t skip_to_step;
-extern uint32_t puzzle;
 
 static void set_game_time();
 static void skip_to_step1() { skip_to_step = 1; }
@@ -14,12 +13,12 @@ static void skip_to_step3() { skip_to_step = 3; }
 static void skip_to_step4() { skip_to_step = 4; }
 static void skip_to_step5() { skip_to_step = 5; }
 static void skip_to_step6() { skip_to_step = 6; }
-static void try_step1() { clean_bomb(); p001_step1(); }
+static void try_step1() { clean_bomb(); step1(); }
-static void try_step2() { clean_bomb(); p001_step2(); }
+static void try_step2() { clean_bomb(); step2(); }
-static void try_step3() { clean_bomb(); p001_step3(); }
+static void try_step3() { clean_bomb(); step3(); }
-static void try_step4() { clean_bomb(); p001_step4(); }
+static void try_step4() { clean_bomb(); step4(); }
-static void try_step5() { clean_bomb(); p001_step5(); }
+static void try_step5() { clean_bomb(); step5(); }
-static void try_step6() { clean_bomb(); p001_step6(); }
+static void try_step6() { clean_bomb(); step6(); }
 static void issue_strike() { strike("Strike Issued"); }
 static void flashbang();
 static void debug_switches();
@@ -46,151 +45,166 @@ static void replay_last() {
     start_playback();
 }
 
-static void do_p001() {
-    puzzle = 1;
-}
 
-static void do_p002() {
-    puzzle = 2;
-}
-
-/// Wait for "*9819"
 void step0() {
     led_set(IndicatorLED::LED_SPEAKER, LEDColor::LED_COLOR_BLUE);
     leds_flush();
 
-    StarCodeHandler star_codes[] = {
+    SemaphoreHandle_t continue_sem = xSemaphoreCreateBinary();
+    if (continue_sem == nullptr) {
+        ESP_LOGE(TAG, "could not create semaphore");
+        return;
+    }
+
+    StarCodeEntry star_codes[] = {
         {
-            .code = "*9819",
+            .code = "9819",
-            .display_text = "Start P001",
+            .display_text = "Diffusal Initiated",
-            .should_exit = true,
+            .delay_us = 2'000'000,
-            .callback = do_p001,
+            .callback = nullptr,
+            .triggered_sem = continue_sem,
         },
         {
-            .code = "*3141",
+            .code = "59861",
-            .display_text = "Start P002",
-            .should_exit = true,
-            .callback = do_p002,
-        },
-        {
-            .code = "*59861",
             .display_text = "Setup Wires",
-            .should_exit = false,
+            .delay_us = 10'000'000,
             .callback = setup_wires,
+            .triggered_sem = nullptr,
         },
         {
-            .code = "*59862",
+            .code = "59862",
             .display_text = "Set Game Time",
-            .should_exit = false,
+            .delay_us = 2'000'000,
             .callback = set_game_time,
+            .triggered_sem = nullptr,
         },
         {
-            .code = "*59863",
+            .code = "59863",
-            .display_text = "Debug Switches",
+            .display_text = "Debug switches",
-            .should_exit = false,
+            .delay_us = 2'000'000,
             .callback = debug_switches,
+            .triggered_sem = nullptr,
         },
         {
-            .code = "*59864",
+            .code = "59864",
             .display_text = "Battery Stats",
-            .should_exit = false,
+            .delay_us = 2'000'000,
             .callback = battery_stats,
+            .triggered_sem = nullptr,
         },
         {
-            .code = "*59871",
+            .code = "59871",
             .display_text = "Try Step 1",
-            .should_exit = false,
+            .delay_us = 2'000'000,
             .callback = try_step1,
+            .triggered_sem = nullptr,
         },
         {
-            .code = "*59872",
+            .code = "59872",
             .display_text = "Try Step 2",
-            .should_exit = false,
+            .delay_us = 2'000'000,
             .callback = try_step2,
+            .triggered_sem = nullptr,
         },
         {
-            .code = "*59873",
+            .code = "59873",
             .display_text = "Try Step 3",
-            .should_exit = false,
+            .delay_us = 2'000'000,
             .callback = try_step3,
+            .triggered_sem = nullptr,
         },
         {
-            .code = "*59874",
+            .code = "59874",
             .display_text = "Try Step 4",
-            .should_exit = false,
+            .delay_us = 2'000'000,
             .callback = try_step4,
+            .triggered_sem = nullptr,
         },
         {
-            .code = "*59875",
+            .code = "59875",
             .display_text = "Try Step 5",
-            .should_exit = false,
+            .delay_us = 2'000'000,
             .callback = try_step5,
+            .triggered_sem = nullptr,
         },
         {
-            .code = "*59876",
+            .code = "59876",
             .display_text = "Try Step 6",
-            .should_exit = false,
+            .delay_us = 2'000'000,
             .callback = try_step6,
+            .triggered_sem = nullptr,
         },
         {
-            .code = "*59881",
+            .code = "59881",
             .display_text = "Skip To Step 1",
-            .should_exit = true,
+            .delay_us = 2'000'000,
             .callback = skip_to_step1,
+            .triggered_sem = continue_sem,
         },
         {
-            .code = "*59882",
+            .code = "59882",
             .display_text = "Skip To Step 2",
-            .should_exit = true,
+            .delay_us = 2'000'000,
             .callback = skip_to_step2,
+            .triggered_sem = continue_sem,
         },
         {
-            .code = "*59883",
+            .code = "59883",
             .display_text = "Skip To Step 3",
-            .should_exit = true,
+            .delay_us = 2'000'000,
             .callback = skip_to_step3,
+            .triggered_sem = continue_sem,
         },
         {
-            .code = "*59884",
+            .code = "59884",
             .display_text = "Skip To Step 4",
-            .should_exit = true,
+            .delay_us = 2'000'000,
             .callback = skip_to_step4,
+            .triggered_sem = continue_sem,
         },
         {
-            .code = "*59885",
+            .code = "59885",
             .display_text = "Skip To Step 5",
-            .should_exit = true,
+            .delay_us = 2'000'000,
             .callback = skip_to_step5,
+            .triggered_sem = continue_sem,
         },
         {
-            .code = "*59886",
+            .code = "59886",
             .display_text = "Skip To Step 6",
-            .should_exit = true,
+            .delay_us = 2'000'000,
             .callback = skip_to_step6,
+            .triggered_sem = continue_sem,
         },
         {
-            .code = "*1111",
+            .code = "1111",
             .display_text = "Issue Strike",
-            .should_exit = false,
+            .delay_us = 2'000'000,
             .callback = issue_strike,
+            .triggered_sem = nullptr,
         },
         {
-            .code = "*1112",
+            .code = "1112",
             .display_text = "????",
-            .should_exit = false,
+            .delay_us = 2'000'000,
             .callback = flashbang,
+            .triggered_sem = nullptr,
         },
         {
-            .code = "*1113",
+            .code = "1113",
-            .display_text = "replay_last",
+            .display_text = "replay",
-            .should_exit = false,
+            .delay_us = 2'000'000,
             .callback = replay_last,
+            .triggered_sem = continue_sem,
         },
     };
-    int len = sizeof(star_codes)/sizeof(StarCodeHandler);
+    size_t len = sizeof(star_codes)/sizeof(star_codes[0]);
-    do_star_codes(star_codes, len);
-}
 
+    add_star_codes(star_codes, len);
+    xSemaphoreTake(continue_sem, portMAX_DELAY);
+    rm_star_codes(star_codes, len);
+    vSemaphoreDelete(continue_sem);
+}
 
 static const int CURSOR_POS_MAP[5] = {1, 3, 4, 6, 7};
 static char str_buf[18] = {0};

main/steps/step0.h

@@ -1,19 +1,15 @@
 #ifndef STEP_0_H
 #define STEP_0_H
 
-#include "../drivers/bottom_half.h"
+#include "../drivers/all.h"
-#include "../drivers/char_lcd.h"
-#include "../drivers/wires.h"
-#include "../drivers/power.h"
-#include "setup_wires.h"
-#include "../helper.h"
 
-#include "p001_step1.h"
+#include "setup_wires.h"
-#include "p001_step2.h"
+#include "step1.h"
-#include "p001_step3.h"
+#include "step2.h"
-#include "p001_step4.h"
+#include "step3.h"
-#include "p001_step5.h"
+#include "step4.h"
-#include "p001_step6.h"
+#include "step5.h"
+#include "step6.h"
 
 /// Wait for "*9819"
 void step0(void);

main/steps/step1.cpp

@@ -1,4 +1,4 @@
-#include "p001_step1.h"
+#include "step1.h"
 
 __attribute__((unused))
 static const char *TAG = "step1";
@@ -206,18 +206,17 @@ static bool play_part(uint32_t time) {
     set_module_time(time);
 
     lcd_clear();
-    lcd_set_cursor_pos(1, 1);
     switch (part) {
         case 0:
-        lcd_print("COLOR");
+        lcd_print(1, 1, "COLOR");
         led_set(IndicatorLED::LED_LCD, LEDColor::LED_COLOR_PINK);
         break;
         case 1:
-        lcd_print("NUMBER");
+        lcd_print(1, 1, "NUMBER");
         led_set(IndicatorLED::LED_LCD, LEDColor::LED_COLOR_BLUE);
         break;
         case 2:
-        lcd_print("SWITCH");
+        lcd_print(1, 1, "SWITCH");
         led_set(IndicatorLED::LED_LCD, LEDColor::LED_COLOR_YELLOW);
         break;
     }
@@ -276,7 +275,7 @@ static bool play_part(uint32_t time) {
     return true;
 }
 
-void p001_step1(void) {
+void step1(void) {
     while (get_switch_flipped(nullptr));
 
     init_step();

main/steps/step1.h

@@ -0,0 +1,10 @@
+#ifndef STEP_1_H
+#define STEP_1_H
+
+#include <random>
+#include "../drivers/all.h"
+#include "../helper.h"
+
+void step1(void);
+
+#endif /* STEP_1_H */

main/steps/step2.cpp

@@ -1,4 +1,4 @@
-#include "p001_step2.h"
+#include "step2.h"
 
 __attribute__((unused))
 static const char *TAG = "step2";
@@ -91,7 +91,7 @@ static void new_puzzle(void) {
     set_module_sseg_raw(display_map);
 }
 
-void p001_step2(void) {
+void step2(void) {
     KeypadKey key;
     int solved_times = 0;
 

main/steps/step2.h

@@ -0,0 +1,12 @@
+#ifndef STEP_2_H
+#define STEP_2_H
+
+#include "../drivers/all.h"
+#include "../helper.h"
+#include <iostream>
+#include <random>
+#include <map>
+
+void step2(void);
+
+#endif /* STEP_2_H */

main/steps/step3.cpp

@@ -1,4 +1,4 @@
-#include "p001_step3.h"
+#include "step3.h"
 
 #define ONE_SECOND_TIME 90'000
 #define THREE_SECOND_TIME 90'000
@@ -21,6 +21,8 @@ static const char* TONE_FILES[] = {
     MOUNT_POINT "/high-6.wav",
 };
 
+static const size_t LCD_STRING_SOMETHING = 0;
+static const size_t LCD_STRING_NOTHING = 1;
 static const char* LCD_STRINGS[] = {
     "something",
     "nothing",
@@ -49,13 +51,20 @@ static std::uniform_int_distribution<> lcd_rand_char_dist(0, sizeof(lcd_random_c
 static std::uniform_int_distribution<> has_coconut_dist(0, 2);
 static std::uniform_int_distribution<> coconut_position_dist(0, 13);
 
+const static uint32_t NEOPIXEL_COLOR_IDX_RED = 0;
+const static uint32_t NEOPIXEL_COLOR_IDX_YELLOW = 1;
+const static uint32_t NEOPIXEL_COLOR_IDX_GREEN = 2;
+const static uint32_t NEOPIXEL_COLOR_IDX_BLUE = 3;
+const static uint32_t NEOPIXEL_COLOR_IDX_PINK = 4;
+const static uint32_t NEOPIXEL_COLOR_IDX_WHITE = 5;
+const static uint32_t NEOPIXEL_COLOR_IDX_OFF = 6;
 static uint32_t NEOPIXEL_COLORS[7] = {
     LEDColor::LED_COLOR_RED,
-    LEDColor::LED_COLOR_ORANGE,
     LEDColor::LED_COLOR_YELLOW,
     LEDColor::LED_COLOR_GREEN,
     LEDColor::LED_COLOR_BLUE,
     LEDColor::LED_COLOR_PINK,
+    LEDColor::LED_COLOR_WHITE,
     LEDColor::LED_COLOR_OFF,
 };
 
@@ -63,17 +72,25 @@ static bool one_second();
 static bool three_second();
 static bool six_second();
 
-void p001_step3(void) {
+void step3(void) {
-    StarCodeHandler star_codes[] = {
+    SemaphoreHandle_t continue_sem = xSemaphoreCreateBinary();
-        {
+    if (continue_sem == nullptr) {
-            .code = "*1642",
+        ESP_LOGE(TAG, "could not create semaphore");
+        return;
+    }
+
+    StarCodeEntry start_code = {
+        .code = "1642",
         .display_text = "Starting...",
-            .should_exit = true,
+        .delay_us = 2'000'000,
         .callback = nullptr,
-        },
+        .triggered_sem = continue_sem,
     };
-    int len = sizeof(star_codes)/sizeof(StarCodeHandler);
+
-    do_star_codes(star_codes, len);
+    add_star_code(start_code);
+    xSemaphoreTake(continue_sem, portMAX_DELAY);
+    rm_star_code(start_code.code);
+    vSemaphoreDelete(continue_sem);
 
     while (times < TIMES_TO_COMPLETE) {
         tone = tone_dist(gen);
@@ -82,7 +99,7 @@ void p001_step3(void) {
 
         play_clip_wav(MOUNT_POINT "/ready.wav", true, false, 3, 0);
         // The high pitched tones need to be scaled down by 3 more
-        play_clip_wav(TONE_FILES[tone], false, false, 2 + (tone/3) * 3, 0);
+        play_clip_wav(TONE_FILES[tone], false, false, 1 + (tone/3) * 4, 0);
         
         bool correct = false;
         switch (tone % 3) {
@@ -221,16 +238,16 @@ static bool one_second() {
     int red_led_count = 0;
     int blue_led_count = 0;
     for (int i = 0; i < LED_COUNT; i++) {
-        if (indicator_led_idxs[i] == 0) {
+        if (indicator_led_idxs[i] == NEOPIXEL_COLOR_IDX_RED) {
             red_led_count++;
-        } else if (indicator_led_idxs[i] == 4) {
+        } else if (indicator_led_idxs[i] == NEOPIXEL_COLOR_IDX_BLUE) {
             blue_led_count++;
         }
     }
 
     uint8_t correct_switches = four_bit_flag(
-        speaker_color == 0 || speaker_color == 1 || speaker_color == 2,
+        speaker_color == NEOPIXEL_COLOR_IDX_RED || speaker_color == NEOPIXEL_COLOR_IDX_YELLOW || speaker_color == NEOPIXEL_COLOR_IDX_PINK,
-        lcd_string_idx == 0 || lcd_string_idx == 1,
+        lcd_string_idx == LCD_STRING_SOMETHING || lcd_string_idx == LCD_STRING_NOTHING,
         was_high,
         !was_high
     );
@@ -280,9 +297,9 @@ static bool three_second() {
     int red_led_count = 0;
     int blue_led_count = 0;
     for (int i = 0; i < LED_COUNT; i++) {
-        if (indicator_led_idxs[i] == 0) {
+        if (indicator_led_idxs[i] == NEOPIXEL_COLOR_IDX_RED) {
             red_led_count++;
-        } else if (indicator_led_idxs[i] == 4) {
+        } else if (indicator_led_idxs[i] == NEOPIXEL_COLOR_IDX_BLUE) {
             blue_led_count++;
         }
     }
@@ -345,7 +362,7 @@ static bool six_second() {
 
     bool was_high = (tone / 3) == 1;
 
-    bool second_switch_correct_state = (indicator_led_idxs[IndicatorLED::LED_S2] == 0) || (indicator_led_idxs[IndicatorLED::LED_S2] == 6);
+    bool second_switch_correct_state = (indicator_led_idxs[IndicatorLED::LED_S2] == NEOPIXEL_COLOR_IDX_RED) || (indicator_led_idxs[IndicatorLED::LED_S2] == NEOPIXEL_COLOR_IDX_OFF);
     second_switch_correct_state = second_switch_correct_state || was_high;
 
     rng_leds();
@@ -354,16 +371,16 @@ static bool six_second() {
     int green_led_count = 0;
     int blue_led_count = 0;
     for (int i = 0; i < LED_COUNT; i++) {
-        if (indicator_led_idxs[i] == 4) {
+        if (indicator_led_idxs[i] == NEOPIXEL_COLOR_IDX_BLUE) {
             blue_led_count++;
-        } else if (indicator_led_idxs[i] == 3) {
+        } else if (indicator_led_idxs[i] == NEOPIXEL_COLOR_IDX_GREEN) {
             green_led_count++;
         }
     }
 
     int pink_led_on_bottom_count = 0;
     for (int i = IndicatorLED::LED_RFID; i < LED_COUNT; i++) {
-        if (indicator_led_idxs[i] == 5) {
+        if (indicator_led_idxs[i] == NEOPIXEL_COLOR_IDX_PINK) {
             pink_led_on_bottom_count++;
         }
     }

main/steps/step3.h

@@ -0,0 +1,10 @@
+#ifndef STEP_3_H
+#define STEP_3_H
+
+#include <random>
+#include "../drivers/all.h"
+#include "../helper.h"
+
+void step3(void);
+
+#endif /* STEP_3_H */

main/steps/step4.cpp

@@ -1,4 +1,4 @@
-#include "p001_step4.h"
+#include "step4.h"
 
 __attribute__((unused))
 static const char *TAG = "step4";
@@ -233,8 +233,8 @@ bool play_game(int time, int required_score) {
     target_score = required_score;
     score = 0;
     update_score();
-    // set_module_time(time);
+    set_module_time(time);
-    // start_module_timer();
+    start_module_timer();
 
     clear_board();
 
@@ -272,7 +272,7 @@ bool play_game(int time, int required_score) {
             show_board();
 
             if (score >= required_score) {
-                // stop_module_timer();
+                stop_module_timer();
                 return true;
             } 
         }
@@ -307,19 +307,18 @@ bool play_game(int time, int required_score) {
                     show_board();
 
                     if (score >= required_score) {
-                        // stop_module_timer();
+                        stop_module_timer();
                         return true;
                     } 
                 }
             }
         }
 
-        // if (get_module_time() <= 0) {
+        if (get_module_time() <= 0) {
-        //     stop_module_timer();
+            stop_module_timer();
-        //     strike("Out of time");
+            strike("Out of time");
-        //     return false;
+            return false;
-        // }
+        }
-
         // if (get_switch_flipped(&switch_)) {
         //     printf("%d\n", piece);
         //     for (int i = 0; i < sizeof(piece_nodes)/sizeof(piece_nodes[0]); i++) {
@@ -334,7 +333,7 @@ bool play_game(int time, int required_score) {
 
     // game over
     ESP_LOGI(TAG, "Game Over. Score: %d", score);
-    // stop_module_timer();
+    stop_module_timer();
     strike("Out of room");
     return false;
 }
@@ -380,23 +379,36 @@ static void fail() {
     }
 }
 
-void p001_step4() {
+void step4() {
-    StarCodeHandler star_code = {
+    // TODO: extract to helper function
-        .code = "*3850",
+    SemaphoreHandle_t continue_sem = xSemaphoreCreateBinary();
+    if (continue_sem == nullptr) {
+        ESP_LOGE(TAG, "could not create semaphore");
+    }
+
+    StarCodeEntry start_code = {
+        .code = "3850",
         .display_text = "Starting...",
-        .should_exit = true,
+        .delay_us = 2'000'000,
         .callback = nullptr,
+        .triggered_sem = continue_sem,
     };
-    do_star_codes(&star_code, 1);
+
+    add_star_code(start_code);
+    xSemaphoreTake(continue_sem, portMAX_DELAY);
+    rm_star_code(start_code.code);
+    vSemaphoreDelete(continue_sem);
+
     init_screen();
 
     while (!play_game(4*60*1000, 2)) fail();
+    // TODO: create constants for common assets, and put them in a folder.
     play_clip_wav(MOUNT_POINT "/partdone.wav", true, false, 0, 0);
     complete();
-    while (!play_game(4*60*1000, 5)) fail();
+    while (!play_game(4*60*1000, 4)) fail();
     play_clip_wav(MOUNT_POINT "/partdone.wav", true, false, 0, 0);
     complete();
-    while (!play_game(7*60*1000, 12)) fail();
+    while (!play_game(7*60*1000, 8)) fail();
     play_clip_wav(MOUNT_POINT "/stepdone.wav", true, false, 1, 0);
     complete();
 

main/steps/step4.h

@@ -0,0 +1,14 @@
+#ifndef STEP_4_H
+#define STEP_4_H
+
+#include <random>
+#include "../drivers/all.h"
+#include "../helper.h"
+#include "esp_log.h"
+
+#define TETRIS_USE_FLASH_IMG
+#define TETRIS_USE_FLASH_BG_IMG
+
+void step4(void);
+
+#endif /* STEP_4_H */

main/steps/step5.cpp

@@ -1,4 +1,4 @@
-#include "p001_step5.h"
+#include "step5.h"
 
 #define TIME_CLEAR   30'000
 #define TIME_PLANK   40'000
@@ -7,6 +7,7 @@
 #define TIME_BLINK   35'000
 #define TIME_UMMM    35'000
 #define TIME_BLANK   40'000
+#define TIME_WHAT    60'000
 #define TIME_PLINK   40'000
 
 __attribute__((unused))
@@ -177,17 +178,24 @@ bool submit_6(bool* buttons_cycling, bool button_turned_on, int led_off) {
     return true;
 }
 
-void p001_step5(void) {
+void step5(void) {
-    StarCodeHandler star_codes[] = {
+    SemaphoreHandle_t continue_sem = xSemaphoreCreateBinary();
-        {
+    if (continue_sem == nullptr) {
-            .code = "*2648",
+        ESP_LOGE(TAG, "could not create semaphore");
+    }
+
+    StarCodeEntry start_code = {
+        .code = "2648",
         .display_text = "Starting...",
-            .should_exit = true,
+        .delay_us = 2'000'000,
         .callback = nullptr,
-        },
+        .triggered_sem = continue_sem,
     };
-    int len = sizeof(star_codes)/sizeof(StarCodeHandler);
+
-    do_star_codes(star_codes, len);
+    add_star_code(start_code);
+    xSemaphoreTake(continue_sem, portMAX_DELAY);
+    rm_star_code(start_code.code);
+    vSemaphoreDelete(continue_sem);
 
     std::vector<int> all_leds;
 
@@ -221,15 +229,13 @@ void p001_step5(void) {
     clean_bomb();
     int solved_puzzles = 0;
     while (solved_puzzles < TIMES_TO_SOLVE) {
-        lcd_set_cursor_pos(1, 1);
         bool solved_correctly = false;
 
         int puzzle = puzzle_dist(gen);
-        // int puzzle = 2;
 
         switch (puzzle) {
             case 0: {
-                lcd_print("Clear");
+                lcd_print(1, 1, "Clear");
                 set_module_time(TIME_CLEAR);
                 start_module_timer();
 
@@ -264,7 +270,7 @@ void p001_step5(void) {
                 break;
             }
             case 1: {
-                lcd_print("Blank");
+                lcd_print(1, 1, "Blank");
                 set_module_time(TIME_BLANK);
                 start_module_timer();
                 
@@ -378,7 +384,7 @@ void p001_step5(void) {
                 break;
             }
             case 3: {
-                lcd_print("Nothing");
+                lcd_print(1, 1, "Nothing");
                 set_module_time(TIME_NOTHING);
                 start_module_timer();
 
@@ -437,7 +443,7 @@ void p001_step5(void) {
                 break;
             }
             case 4: {
-                lcd_print("Blink");
+                lcd_print(1, 1, "Blink");
                 set_module_time(TIME_BLINK);
                 start_module_timer();
 
@@ -501,7 +507,7 @@ void p001_step5(void) {
                 break;
             }
             case 5: {
-                lcd_print("Ummm");
+                lcd_print(1, 1, "Ummm");
                 set_module_time(TIME_UMMM);
                 start_module_timer();
 
@@ -557,7 +563,7 @@ void p001_step5(void) {
                 break;
             }
             case 6: {
-                lcd_print("Plank");
+                lcd_print(1, 1, "Plank");
                 set_module_time(TIME_PLANK);
                 start_module_timer();
 
@@ -650,7 +656,155 @@ void p001_step5(void) {
                 break;
             }
             case 7: {
-                lcd_print("Plink");
+                lcd_print(1, 1, "What");
+                set_module_time(TIME_WHAT);
+                start_module_timer();
+
+                std::uniform_int_distribution<> math_number_dist(1, 9);
+
+                std::vector<float> math_numbers;
+                std::vector<int> math_operations;
+
+                std::map<int, char> operation_map = {
+                    {0, '+'},
+                    {1, '-'},
+                    {2, '*'},
+                    {3, '/'},
+                };
+
+                int expression_answer = -1;
+                std::string display_expression;
+
+                while (expression_answer < 0) {
+                    math_numbers = {static_cast<float>(math_number_dist(gen)), static_cast<float>(math_number_dist(gen)), static_cast<float>(math_number_dist(gen)), static_cast<float>(math_number_dist(gen))};
+                    std::vector<int> possible_math_operations = {0, 1, 2, 3}; 
+                    math_operations = unique_values(possible_math_operations, 3);
+
+                    display_expression = std::to_string(static_cast<int>(math_numbers[0]));
+                    for (int i = 0; i < 3; i++) {
+                        display_expression += operation_map[math_operations[i]];
+                        display_expression += std::to_string(static_cast<int>(math_numbers[i + 1]));
+                    }
+
+                    // Solve
+                    for (int j = 0; j < 3; j++) {
+                        bool found = false;
+                        for (int i = 0; i < math_operations.size(); i++) {
+                            if (math_operations[i] == 2) {
+                                // ESP_LOGI(TAG, "i = %i, condensing %f * %f to %f", i, math_numbers[i], math_numbers[i + 1], (math_numbers[i] * math_numbers[i+1]));
+                                math_numbers[i] = math_numbers[i] * math_numbers[i + 1];
+                                math_numbers.erase(math_numbers.begin() + i + 1);
+                                math_operations.erase(math_operations.begin() + i);
+                                found = true;
+                                break;
+                            } else if (math_operations[i] == 3) {
+                                // ESP_LOGI(TAG, "i = %i, condensing %f / %f to %f", i, math_numbers[i], math_numbers[i + 1], (math_numbers[i] / math_numbers[i+1]));
+                                math_numbers[i] = math_numbers[i] / math_numbers[i + 1];
+                                math_numbers.erase(math_numbers.begin() + i + 1);
+                                math_operations.erase(math_operations.begin() + i);
+                                found = true;
+                                break;
+                            }
+                        }
+                        if (found) continue;
+                        for (int i = 0; i < math_operations.size(); i++) {
+                            if (math_operations[i] == 0) {
+                                // ESP_LOGI(TAG, "i = %i, condensing %f + %f to %f", i, math_numbers[i], math_numbers[i + 1], (math_numbers[i] + math_numbers[i+1]));
+                                math_numbers[i] = math_numbers[i] + math_numbers[i + 1];
+                                math_numbers.erase(math_numbers.begin() + i + 1);
+                                math_operations.erase(math_operations.begin() + i);
+                                found = true;
+                                break;
+                            } else if (math_operations[i] == 1) {
+                                // ESP_LOGI(TAG, "i = %i, condensing %f - %f to %f", i, math_numbers[i], math_numbers[i + 1], (math_numbers[i] - math_numbers[i+1]));
+                                math_numbers[i] = math_numbers[i] - math_numbers[i + 1];
+                                math_numbers.erase(math_numbers.begin() + i + 1);
+                                math_operations.erase(math_operations.begin() + i);
+                                found = true;
+                                break;
+                            }
+                        }
+                    }
+                    expression_answer = static_cast<int>(std::floor(math_numbers[0]));
+                }
+
+                // display expression
+                lcd_print(1, 2, display_expression.c_str());
+
+                // set LEDs
+                const uint32_t COLORS[] = {
+                    LEDColor::LED_COLOR_BLUE,
+                    LEDColor::LED_COLOR_RED,
+                    LEDColor::LED_COLOR_GREEN,
+                    LEDColor::LED_COLOR_YELLOW,
+                };
+
+                std::uniform_int_distribution<> add_sub_indicator_dist(1, 6);
+                std::uniform_int_distribution<> mult_div_indicator_dist(1, 3);
+
+                int modifier_indicators[4] = {add_sub_indicator_dist(gen), add_sub_indicator_dist(gen), mult_div_indicator_dist(gen), mult_div_indicator_dist(gen)};
+
+
+                while ((((expression_answer + (modifier_indicators[0] * 3) - modifier_indicators[1]) * std::pow(3, modifier_indicators[2])) / std::pow(2, modifier_indicators[3])) < 0) {
+                    modifier_indicators[0] = add_sub_indicator_dist(gen);
+                    modifier_indicators[1] = add_sub_indicator_dist(gen);
+                    modifier_indicators[2] = mult_div_indicator_dist(gen);
+                    modifier_indicators[3] = mult_div_indicator_dist(gen);
+                }
+
+                expression_answer += modifier_indicators[0] * 3;
+                expression_answer -= modifier_indicators[1];
+                expression_answer *= std::pow(3, modifier_indicators[2]);
+                expression_answer /= std::pow(2, modifier_indicators[3]);
+
+                std::vector<int> led_options = all_leds;
+                for (int i = 0; i < 4; i++) {
+                    set_unique_leds(led_options, modifier_indicators[i], COLORS[i]);
+                }
+
+                leds_flush();
+
+                std::string answer_string = std::to_string(expression_answer);
+                std::string entered_string = "";
+
+                ESP_LOGI(TAG, "Solved full answer: %s", answer_string.c_str());
+
+                KeypadKey key;
+
+                while (1) {
+                    if (get_keypad_pressed(&key)) {
+                        if (key == KeypadKey::star) {
+                            // clear
+                            entered_string = "";
+                        } else if (key == KeypadKey::pound) {
+                            // submit
+                            if (entered_string != answer_string) {
+                                strike("Incorrect answer!");
+                            } else {
+                                solved_correctly = true;
+                            }
+                            break;
+                        } else {
+                            entered_string += char_of_keypad_key(key);
+                        }
+
+                        lcd_clear();
+                        lcd_print(1, 1, "What");
+                        lcd_print(1, 2, display_expression.c_str());
+                        lcd_print(1, 3, entered_string.c_str());
+                    }
+                    if (get_module_time() <= 0) {
+                        strike("Ran out of time!");
+                        break;
+                    }
+
+                    vTaskDelay(pdMS_TO_TICKS(10));
+                }
+
+                break;
+            }
+            case 8: {
+                lcd_print(1, 1, "Plink");
                 set_module_time(TIME_PLINK);
                 start_module_timer();
 
@@ -744,22 +898,20 @@ void p001_step5(void) {
                 if (!failed) {
                     solved_correctly = true;
                 }
-                break;
-            }
-            default: {
-                ESP_LOGE(TAG, "Invalid puzzle number %d", puzzle);
             }
+
+
         }
         stop_module_timer();
         if (solved_correctly) {
             solved_puzzles++;
-            play_clip_wav(MOUNT_POINT "/correct.wav", true, false, 3, 0);
+            play_clip_wav(MOUNT_POINT "/partdone.wav", true, false, 0, 0);
             vTaskDelay(pdMS_TO_TICKS(500));
             solved_correctly = false;
         } else {
             vTaskDelay(pdMS_TO_TICKS(3000));
         }
-        clear_all_pressed_released();
+        play_clip_wav(MOUNT_POINT "/stepdone.wav", true, false, 1, 0);
         clean_bomb();
     }
 }

main/steps/step5.h

@@ -0,0 +1,16 @@
+#ifndef STEP_5_H
+#define STEP_5_H
+
+#include "../drivers/all.h"
+#include "../helper.h"
+#include <random>
+#include <iostream>
+#include <set>
+#include <map>
+#include <vector>
+#include <cmath>
+#include <array>
+
+void step5(void);
+
+#endif /* STEP_5_H */

main/steps/step6.cpp

@@ -1,11 +1,11 @@
-#include "p001_step6.h"
+#include "step6.h"
 
 __attribute__((unused))
 static const char *TAG = "step6";
 
 static uint8_t cut_wires = 0;
 
-void p001_step6(void) {
+void step6(void) {
     get_cut_wires();
     clear_all_pressed_released();
 

main/steps/step6.h

@@ -0,0 +1,10 @@
+#ifndef STEP_6_H
+#define STEP_6_H
+
+#include "wires_puzzle.h"
+#include "../drivers/all.h"
+#include "../helper.h"
+
+void step6(void);
+
+#endif /* STEP_6_H */

partitions.csv

@@ -0,0 +1,8 @@
+# ESP-IDF Partition Table
+# Name, Type, SubType, Offset, Size, Flags
+nvs,data,nvs,0x9000,0x6000,,
+phy_init,data,phy,0xf000,0x1000,,
+ota,data,ota,0x10000,0x2000,,
+factory,app,factory,0x20000,2M,,
+ota0,app,ota_0,0x220000,2M,,
+ota1,app,ota_1,0x420000,2M,,