miniature-town/firmware/components/led_matrix/led_matrix.c

#include "led_matrix.h"

#include "esp_log.h"
#include "freertos/FreeRTOS.h"
#include "led_strip.h"
#include "sdkconfig.h"

typedef struct
{
    led_strip_handle_t led_strip;
    uint32_t size;
} led_matrix_t;

led_matrix_t led_matrix;

static void led_strip_init(uint8_t gpio_pin, uint32_t size)
{
    led_matrix.size = size;

    led_strip_config_t strip_config = {.strip_gpio_num = gpio_pin,
                                       .max_leds = size,
                                       .led_model = LED_MODEL_WS2812,
                                       .color_component_format = LED_STRIP_COLOR_COMPONENT_FMT_RGB,
                                       .flags = {
                                           .invert_out = false,
                                       }};

    led_strip_rmt_config_t rmt_config = {.clk_src = RMT_CLK_SRC_DEFAULT,
                                         .resolution_hz = 0,
                                         .mem_block_symbols = 0,
                                         .flags = {
                                             .with_dma = true,
                                         }};

    ESP_ERROR_CHECK(led_strip_new_rmt_device(&strip_config, &rmt_config, &led_matrix.led_strip));
}

uint32_t led_matrix_get_size()
{
    return led_matrix.size;
}

void led_matrix_set_pixel(uint32_t index, uint8_t red, uint8_t green, uint8_t blue)
{
    led_strip_set_pixel(led_matrix.led_strip, index, red, green, blue);
}

void led_matrix_init(void *args)
{
    ESP_LOGI(pcTaskGetName(NULL), "Calling led_matrix_init()");

    led_strip_init(CONFIG_WLED_DIN_PIN, CONFIG_WLED_LED_COUNT);
    int value = 0;
    for (int i = 0; i < led_matrix.size; i++)
    {
        led_matrix_set_pixel(i, 0, value, 0);
    }

    while (true)
    {
        led_strip_refresh(led_matrix.led_strip);
        vTaskDelay(pdMS_TO_TICKS(100));
    }

    ESP_LOGI(pcTaskGetName(NULL), "Exiting led_matrix_init()");
    vTaskDelete(NULL);
}