// Ovladani signalizacni chytre LED diody
#include "wsLED.h"
#if defined(ARDUINO_ARCH_ESP32)
//# include "esp_task.h"
// https://github.com/JSchaenzle/ESP32-NeoPixel-WS2812-RMT/blob/master/ws2812_control.c
#  include <set>
#  ifndef LED_RMT_TX_CHANNEL
#    define LED_RMT_TX_CHANNEL RMT_CHANNEL_0
#  endif
// These values are determined by measuring pulse timing with logic analyzer and adjusting to match datasheet.
#  define T0H 14  // 0 bit high time
#  define T1H 52  // 1 bit high time
#  define TL  52  // low time for either bit
#endif
#if defined(ARDUINO_ARCH_ESP8266)
// due to linker overriding the ICACHE_RAM_ATTR for cpp files, these methods are
// moved into a C file so the attribute will be applied correctly
extern "C" void ICACHE_RAM_ATTR send_pixels_800(uint8_t* pixel, size_t count, uint8_t pin);
#endif
// TODO: Proverit ESP32 s https://github.com/adafruit/Adafruit_NeoPixel/blob/master/esp.c
//       ... a s https://github.com/Freenove/Freenove_WS2812_Lib_for_ESP32

#define countof(a) (sizeof(a) / sizeof(a[0]))

static const uint16_t LB[] PROGMEM = {
    0x201 , 0x202 , 0x103 , 0x104 , 0x105 , 0x106 , 0x107 , 0x109 , 0x10a , 0x10c , 0x10e , 0x110 , 0x112 , 0x114 , 0x116 , 0x119 , 0x11c , 0x11e , 0x122 , 0x125 , 0x128 , 0x12c , 0x130 , 0x134,
    0x139 , 0x13d , 0x142 , 0x147 , 0x14d , 0x152 , 0x158 , 0x15e , 0x165 , 0x16b , 0x172 , 0x179 , 0x180 , 0x187 , 0x18e , 0x196 , 0x19d , 0x1a5 , 0x1ac , 0x1b3 , 0x1bb , 0x1c2 , 0x1c9 , 0x1cf,
    0x1d6 , 0x1dc , 0x1e2 , 0x1e7 , 0x1ec , 0x1f0 , 0x1f4 , 0x1f7 , 0x1fa , 0x1fc , 0x1fd , 0x3fe , 0x1fd , 0x1fb , 0x1f8 , 0x1f5 , 0x1f2 , 0x1ed , 0x1e9 , 0x1e4 , 0x1de , 0x1d8 , 0x1d2 , 0x1cb,
    0x1c4 , 0x1bd , 0x1b6 , 0x1af , 0x1a7 , 0x1a0 , 0x198 , 0x191 , 0x18a , 0x182 , 0x17b , 0x174 , 0x16d , 0x167 , 0x161 , 0x15a , 0x154 , 0x14f , 0x149 , 0x144 , 0x13f , 0x13a , 0x136 , 0x131,
    0x12d , 0x12a , 0x126 , 0x123 , 0x11f , 0x11c , 0x11a , 0x117 , 0x115 , 0x112 , 0x110 , 0x10e , 0x10c , 0x10b , 0x109 , 0x108 , 0x107 , 0x105 , 0x104 , 0x203 , 0x102 , 0x201 , 0x2a00
};

static void ICACHE_RAM_ATTR ledTrampoline(wsLED *ptr)
{

    ptr->rtLed();
}

void wsLED::show(void)
{
    uint8_t color[3];

    for (uint8_t i = 0; i < 3; i++) {
// https://github.com/FastLED/FastLED/wiki/FastLED-Color-Correction
        uint16_t c = (uint16_t)_color[i] * (1 + (uint16_t)_scale);
        color[i] = (uint8_t)(c >> 8);
    }
#if defined(ARDUINO_ARCH_ESP32)
    uint32_t bits_to_send = ((uint32_t)color[0] << 16) | ((uint32_t)color[1] << 8) | (uint32_t)color[2];
    uint32_t mask = 1 << (BITS_PER_LED_CMD - 1);
    for (uint32_t bit = 0; bit < BITS_PER_LED_CMD; bit++) {
        uint32_t bit_is_set = bits_to_send & mask;

        led_data_buffer[bit] = bit_is_set ? (rmt_item32_t) {{{T1H, 1, TL, 0}}} : (rmt_item32_t) {{{T0H, 1, TL, 0}}};
        mask >>= 1;
    }
    ESP_ERROR_CHECK(rmt_write_items(LED_RMT_TX_CHANNEL, led_data_buffer, LED_BUFFER_ITEMS, false));
//    ESP_ERROR_CHECK(rmt_wait_tx_done(LED_RMT_TX_CHANNEL, portMAX_DELAY));
#else
    noInterrupts();
    send_pixels_800(color, 1, _pin);
    interrupts();
#endif
}


wsLED::wsLED(int pin, int order) : _pin(pin), _order(order), _ledState(LED_STOP)
{
}

wsLED::wsLED(void) : _pin(NOT_A_PIN), _order(RGB), _ledState(LED_STOP)
{
}

void wsLED::begin(int pin)
{

    _pin = pin;
    begin();
}

void wsLED::begin(void)
{

    if (NOT_A_PIN != _pin) {
#if defined(ARDUINO_ARCH_ESP8266)
        pinMode(_pin, OUTPUT);
#endif
#if defined(ARDUINO_ARCH_ESP32)
// TODO: mozna nejak zkusit najit volny RMT kanal?
        rmt_config_t config;

        config.rmt_mode = RMT_MODE_TX;
        config.channel = LED_RMT_TX_CHANNEL;
        config.gpio_num = (gpio_num_t)_pin;
        config.mem_block_num = 3;
        config.tx_config.loop_en = false;
        config.tx_config.carrier_en = false;
        config.tx_config.idle_output_en = true;
        config.tx_config.idle_level = RMT_IDLE_LEVEL_LOW;
        config.clk_div = 2;

        ESP_ERROR_CHECK(rmt_config(&config));
        ESP_ERROR_CHECK(rmt_driver_install(config.channel, 0, 0));
#endif
        _color[0] = _color[1] = _color[2] = 0;
        show();
    }
}

void wsLED::setColor(LEDRGB color)
{

    _color[(_order >> 6) & 7] = color.r;
    _color[(_order >> 3) & 7] = color.g;
    _color[_order & 7] = color.b;
}

void ICACHE_RAM_ATTR wsLED::rtLed(void)
{
    uint32_t timing;

    _handler.detach();
    switch (_ledState) {
    case LED_BLINK:
        if (_subState) {
            setColor(_color2);
        } else {
            setColor(_color1);
        }
        _subState = !_subState;
        timing = _blinkSpeed;
        break;


    case LED_BREATH: {
        ++_subState;
        _subState %= countof(LB);
        uint16_t val = pgm_read_word(&LB[_subState]);
        _scale = (uint8_t)val;
        val >>= 8;
        timing = 30ul * val;
        if (timing < 30) {
            timing = 30;
        }
        timing = 30;
        break;
    }

    case LED_PULSE: {
        _ledState = _savedState;
        _subState = 0;
        setColor(_color1);
        timing = 5;
        break;
    }

    case LED_PULSE_START:
        setColor(_pulseColor);
        _ledState = LED_PULSE;
        timing = _pulseDuration;
        break;

    default:
        timing = 5000;
        break;
    }
    show();
    _handler.attach_ms(timing, ledTrampoline, this);
}

void wsLED::setColors(LEDRGB color1, LEDRGB color2)
{

    _color1 = color1;
    _color2 = color2;
    if (LED_BREATH == _ledState) {
        setColor(_color1);
    }
}

void wsLED::blink(uint32_t speed)
{

    if (NOT_A_PIN != _pin) {
        if (LED_BLINK != _ledState) {
            _handler.detach();
            _ledState = LED_BLINK;
            _scale = 0xff;
            _subState = 1;
            _blinkSpeed = speed;
            _handler.attach_ms(5, ledTrampoline, this);
        }
    }
}

void wsLED::breath(void)
{

    if (NOT_A_PIN != _pin) {
        if (LED_BREATH != _ledState) {
            _handler.detach();
            _ledState = LED_BREATH;
            _subState = 0;
            setColor(_color1);
            _handler.attach_ms(5, ledTrampoline, this);
        }
    }
}

void wsLED::setOrder(int neworder)
{

    _order = neworder;
}

void wsLED::pulse(LEDRGB color, uint32_t duration)
{

    if (NOT_A_PIN != _pin) {
        _handler.detach();
        _scale = 0xff;
        _pulseColor = color;
        if (LED_BREATH == _ledState || LED_BLINK == _ledState) {
            _savedState = _ledState;
        }
        _pulseDuration = duration;
        _ledState = LED_PULSE_START;
        _handler.attach_ms(5, ledTrampoline, this);
    }
}

ledstate_t wsLED::getState(void)
{

    return _ledState;
}

void wsLED::stop(void)
{

    if (NOT_A_PIN != _pin) {
        _handler.detach();
        _ledState = LED_STOP;
        _scale = 0xff;
        setColor(LEDRGB(0, 0, 0));
        show();
    }
}