leds

2026-04-01 22:11:48 -05:00
parent 8bc5a982d1
commit 0b79eb8399
7 changed files with 183 additions and 2 deletions
@@ -0,0 +1,64 @@
+#include "blk_box_drivers/leds.hpp"
+#include "pins.h"
+#include "esp_log.h"
+#include "esp_err.h"
+#include "led_strip.h"
+
+const LEDColor LEDColor::OFF = LEDColor::from_rgb(0x00, 0x00, 0x00);
+const LEDColor LEDColor::RED = LEDColor::from_rgb(0x17, 0x00, 0x00);
+const LEDColor LEDColor::RED_STRONG = LEDColor::from_rgb(0xFF, 0x00, 0x00);
+const LEDColor LEDColor::ORANGE = LEDColor::from_rgb(0x17, 0x02, 0x00);
+const LEDColor LEDColor::ORANGE_STRONG = LEDColor::from_rgb(0xFF, 0x20, 0x00);
+const LEDColor LEDColor::YELLOW = LEDColor::from_rgb(0x07, 0x07, 0x00);
+const LEDColor LEDColor::YELLOW_STRONG = LEDColor::from_rgb(0xFF, 0xFF, 0x00);
+const LEDColor LEDColor::GREEN = LEDColor::from_rgb(0x00, 0x07, 0x00);
+const LEDColor LEDColor::GREEN_STRONG = LEDColor::from_rgb(0x00, 0xFF, 0x00);
+const LEDColor LEDColor::BLUE = LEDColor::from_rgb(0x00, 0x00, 0x17);
+const LEDColor LEDColor::BLUE_STRONG = LEDColor::from_rgb(0x00, 0x00, 0xFF);
+const LEDColor LEDColor::PINK = LEDColor::from_rgb(0x10, 0x00, 0x04);
+const LEDColor LEDColor::PINK_STRONG = LEDColor::from_rgb(0xFF, 0x00, 0x80);
+const LEDColor LEDColor::WHITE = LEDColor::from_rgb(0x04, 0x04, 0x04);
+const LEDColor LEDColor::WHITE_STRONG = LEDColor::from_rgb(0xFF, 0xFF, 0xFF);
+
+const static char* TAG = "leds";
+
+static led_strip_handle_t led_strip = NULL;
+
+void init_leds() {
+    /// LED strip common configuration
+    led_strip_config_t strip_config = {
+        .strip_gpio_num = PIN_NEOPIXEL,
+        .max_leds = LED_COUNT,
+        // TODO: switch this over when we switch to the different LEDs
+        .led_model = LED_MODEL_WS2812,
+        .color_component_format = LED_STRIP_COLOR_COMPONENT_FMT_GRB,
+        .flags = {
+            .invert_out = false,
+        }
+    };
+
+    /// RMT backend specific configuration
+    led_strip_rmt_config_t rmt_config = {
+        .clk_src = RMT_CLK_SRC_DEFAULT,
+        .resolution_hz = LED_STRIP_RMT_RES_HZ,
+        .mem_block_symbols = RMT_LED_SYMBOLS * RMT_SYMBOLS_PER_LED,
+        .flags = {
+            .with_dma = false,
+        }
+    };
+
+    /// Create the LED strip object
+    ESP_ERROR_CHECK(led_strip_new_rmt_device(&strip_config, &rmt_config, &led_strip));
+}
+
+void LEDController::set_led(uint32_t led, uint32_t color) {
+    ESP_ERROR_CHECK(led_strip_set_pixel(led_strip, led, color >> 16 & 0xFF, color >> 8 & 0xFF, color & 0xFF));
+}
+
+void LEDController::flush() {
+    ESP_ERROR_CHECK(led_strip_refresh(led_strip));
+}
+
+void LEDController::clear() {
+    ESP_ERROR_CHECK(led_strip_clear(led_strip));
+}