110 lines
3.8 KiB
C
110 lines
3.8 KiB
C
#include "status_led.h"
|
|
#include "freertos/FreeRTOS.h"
|
|
#include "freertos/task.h"
|
|
#include "driver/gpio.h"
|
|
#include "led_strip.h"
|
|
#include "esp_log.h"
|
|
|
|
static const char *TAG = "status_led";
|
|
|
|
static led_strip_handle_t s_led_strip = NULL;
|
|
static bool s_is_rgb = false;
|
|
static int s_gpio_pin = -1;
|
|
static volatile led_state_t s_current_state = LED_STATE_NO_CONFIG;
|
|
|
|
static void set_color(uint8_t r, uint8_t g, uint8_t b) {
|
|
if (s_is_rgb && s_led_strip) {
|
|
led_strip_set_pixel(s_led_strip, 0, r, g, b);
|
|
led_strip_refresh(s_led_strip);
|
|
} else if (!s_is_rgb && s_gpio_pin >= 0) {
|
|
// Simple LED: On if any color component is > 0
|
|
gpio_set_level(s_gpio_pin, (r+g+b) > 0);
|
|
}
|
|
}
|
|
|
|
static void led_task(void *arg) {
|
|
int toggle = 0;
|
|
while (1) {
|
|
switch (s_current_state) {
|
|
case LED_STATE_NO_CONFIG: // Yellow (Solid for RGB, Blink for Single)
|
|
if (s_is_rgb) { set_color(20, 20, 0); vTaskDelay(pdMS_TO_TICKS(1000)); }
|
|
else { set_color(1,1,1); vTaskDelay(pdMS_TO_TICKS(100)); set_color(0,0,0); vTaskDelay(pdMS_TO_TICKS(100)); }
|
|
break;
|
|
case LED_STATE_WAITING: // Blue Blink (Searching)
|
|
set_color(0, 0, toggle ? 50 : 0);
|
|
toggle = !toggle;
|
|
vTaskDelay(pdMS_TO_TICKS(500));
|
|
break;
|
|
case LED_STATE_CONNECTED: // Green Solid (Idle)
|
|
set_color(0, 20, 0);
|
|
vTaskDelay(pdMS_TO_TICKS(1000));
|
|
break;
|
|
case LED_STATE_MONITORING: // Blue Solid (Sniffer Mode)
|
|
set_color(0, 0, 50);
|
|
vTaskDelay(pdMS_TO_TICKS(1000));
|
|
break;
|
|
case LED_STATE_TRANSMITTING: // Fast Purple Flash (Busy)
|
|
set_color(toggle ? 50 : 0, 0, toggle ? 50 : 0);
|
|
toggle = !toggle;
|
|
vTaskDelay(pdMS_TO_TICKS(50));
|
|
break;
|
|
case LED_STATE_TRANSMITTING_SLOW: // Slow Purple Pulse (Pacing)
|
|
set_color(toggle ? 50 : 0, 0, toggle ? 50 : 0);
|
|
toggle = !toggle;
|
|
vTaskDelay(pdMS_TO_TICKS(250));
|
|
break;
|
|
case LED_STATE_STALLED: // Purple Solid (Stalled)
|
|
set_color(50, 0, 50);
|
|
vTaskDelay(pdMS_TO_TICKS(1000));
|
|
break;
|
|
case LED_STATE_FAILED: // Red Blink (Error)
|
|
set_color(toggle ? 50 : 0, 0, 0);
|
|
toggle = !toggle;
|
|
vTaskDelay(pdMS_TO_TICKS(200));
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
void status_led_init(int gpio_pin, bool is_rgb_strip) {
|
|
s_gpio_pin = gpio_pin;
|
|
s_is_rgb = is_rgb_strip;
|
|
|
|
ESP_LOGI(TAG, "Initializing LED on GPIO %d (RGB=%d)", gpio_pin, is_rgb_strip);
|
|
|
|
if (s_is_rgb) {
|
|
led_strip_config_t s_cfg = {
|
|
.strip_gpio_num = gpio_pin,
|
|
.max_leds = 1,
|
|
.led_pixel_format = LED_PIXEL_FORMAT_GRB, // Standard for WS2812
|
|
.led_model = LED_MODEL_WS2812, // Specific model
|
|
.flags.invert_out = false,
|
|
};
|
|
led_strip_rmt_config_t r_cfg = {
|
|
.resolution_hz = 10 * 1000 * 1000, // 10MHz
|
|
.flags.with_dma = false,
|
|
};
|
|
|
|
esp_err_t ret = led_strip_new_rmt_device(&s_cfg, &r_cfg, &s_led_strip);
|
|
if (ret != ESP_OK) {
|
|
ESP_LOGE(TAG, "Failed to create RMT LED strip: %s", esp_err_to_name(ret));
|
|
return;
|
|
}
|
|
led_strip_clear(s_led_strip);
|
|
} else {
|
|
gpio_reset_pin(gpio_pin);
|
|
gpio_set_direction(gpio_pin, GPIO_MODE_OUTPUT);
|
|
gpio_set_level(gpio_pin, 0);
|
|
}
|
|
|
|
xTaskCreate(led_task, "led_task", 2048, NULL, 5, NULL);
|
|
}
|
|
|
|
void status_led_set_state(led_state_t state) {
|
|
s_current_state = state;
|
|
}
|
|
|
|
led_state_t status_led_get_state(void) {
|
|
return s_current_state;
|
|
}
|