#include "led_matrix.h"

#include "freertos/FreeRTOS.h"
#include "led_strip.h"
#include "esp_log.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);
}