250 lines
6.7 KiB
C++
250 lines
6.7 KiB
C++
// 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 , 0x2900
|
|
};
|
|
|
|
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();
|
|
}
|
|
}
|