123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639 |
- #include "module.h"
- #include "lauxlib.h"
- #include "lmem.h"
- #include "platform.h"
- #include <stdlib.h>
- #include <math.h>
- #include <string.h>
- #include "user_interface.h"
- #include "driver/uart.h"
- #include "osapi.h"
- #include "ws2812.h"
- #define CANARY_VALUE 0x32383132
- #define MODE_SINGLE 0
- #define MODE_DUAL 1
- // Init UART1 to be able to stream WS2812 data to GPIO2 pin
- // If DUAL mode is selected, init UART0 to stream to TXD0 as well
- // You HAVE to redirect LUA's output somewhere else
- static int ws2812_init(lua_State* L) {
- const int mode = luaL_optinteger(L, 1, MODE_SINGLE);
- luaL_argcheck(L, mode == MODE_SINGLE || mode == MODE_DUAL, 1, "ws2812.SINGLE or ws2812.DUAL expected");
- // Configure UART1
- // Set baudrate of UART1 to 3200000
- WRITE_PERI_REG(UART_CLKDIV(1), UART_CLK_FREQ / 3200000);
- // Set UART Configuration No parity / 6 DataBits / 1 StopBits / Invert TX
- WRITE_PERI_REG(UART_CONF0(1), UART_TXD_INV | (1 << UART_STOP_BIT_NUM_S) | (1 << UART_BIT_NUM_S));
- if (mode == MODE_DUAL) {
- // Configure UART0
- // Set baudrate of UART0 to 3200000
- WRITE_PERI_REG(UART_CLKDIV(0), UART_CLK_FREQ / 3200000);
- // Set UART Configuration No parity / 6 DataBits / 1 StopBits / Invert TX
- WRITE_PERI_REG(UART_CONF0(0), UART_TXD_INV | (1 << UART_STOP_BIT_NUM_S) | (1 << UART_BIT_NUM_S));
- }
- // Pull GPIO2 down
- platform_gpio_mode(4, PLATFORM_GPIO_OUTPUT, PLATFORM_GPIO_FLOAT);
- platform_gpio_write(4, 0);
- // Waits 10us to simulate a reset
- os_delay_us(10);
- // Redirect UART1 to GPIO2
- // Disable GPIO2
- GPIO_REG_WRITE(GPIO_ENABLE_W1TC_ADDRESS, BIT2);
- // Enable Function 2 for GPIO2 (U1TXD)
- PIN_FUNC_SELECT(PERIPHS_IO_MUX_GPIO2_U, FUNC_U1TXD_BK);
- return 0;
- }
- // Stream data using UART1 routed to GPIO2
- // ws2812.init() should be called first
- //
- // NODE_DEBUG should not be activated because it also uses UART1
- void ICACHE_RAM_ATTR ws2812_write_data(const uint8_t *pixels, uint32_t length, const uint8_t *pixels2, uint32_t length2) {
- // Data are sent LSB first, with a start bit at 0, an end bit at 1 and all inverted
- // 0b00110111 => 110111 => [0]111011[1] => 10001000 => 00
- // 0b00000111 => 000111 => [0]111000[1] => 10001110 => 01
- // 0b00110100 => 110100 => [0]001011[1] => 11101000 => 10
- // 0b00000100 => 000100 => [0]001000[1] => 11101110 => 11
- // Array declared as static const to avoid runtime generation
- // But declared in ".data" section to avoid read penalty from FLASH
- static const __attribute__((section(".data._uartData"))) uint8_t _uartData[4] = { 0b00110111, 0b00000111, 0b00110100, 0b00000100 };
- const uint8_t *end = pixels + length;
- const uint8_t *end2 = pixels2 + length2;
- do {
- // If something to send for first buffer and enough room
- // in FIFO buffer (we wants to write 4 bytes, so less than
- // 124 in the buffer)
- if (pixels < end && (((READ_PERI_REG(UART_STATUS(1)) >> UART_TXFIFO_CNT_S) & UART_TXFIFO_CNT) <= 124)) {
- uint8_t value = *pixels++;
- // Fill the buffer
- WRITE_PERI_REG(UART_FIFO(1), _uartData[(value >> 6) & 3]);
- WRITE_PERI_REG(UART_FIFO(1), _uartData[(value >> 4) & 3]);
- WRITE_PERI_REG(UART_FIFO(1), _uartData[(value >> 2) & 3]);
- WRITE_PERI_REG(UART_FIFO(1), _uartData[(value >> 0) & 3]);
- }
- // Same for the second buffer
- if (pixels2 < end2 && (((READ_PERI_REG(UART_STATUS(0)) >> UART_TXFIFO_CNT_S) & UART_TXFIFO_CNT) <= 124)) {
- uint8_t value = *pixels2++;
- // Fill the buffer
- WRITE_PERI_REG(UART_FIFO(0), _uartData[(value >> 6) & 3]);
- WRITE_PERI_REG(UART_FIFO(0), _uartData[(value >> 4) & 3]);
- WRITE_PERI_REG(UART_FIFO(0), _uartData[(value >> 2) & 3]);
- WRITE_PERI_REG(UART_FIFO(0), _uartData[(value >> 0) & 3]);
- }
- } while(pixels < end || pixels2 < end2); // Until there is still something to send
- }
- // Lua: ws2812.write("string")
- // Byte triples in the string are interpreted as G R B values.
- //
- // ws2812.init() should be called first
- //
- // ws2812.write(string.char(0, 255, 0)) sets the first LED red.
- // ws2812.write(string.char(0, 0, 255):rep(10)) sets ten LEDs blue.
- // ws2812.write(string.char(255, 0, 0, 255, 255, 255)) first LED green, second LED white.
- //
- // In DUAL mode 'ws2812.init(ws2812.DUAL)', you may pass a second string as parameter
- // It will be sent through TXD0 in parallel
- static int ws2812_write(lua_State* L) {
- size_t length1, length2;
- const char *buffer1, *buffer2;
- // First mandatory parameter
- int type = lua_type(L, 1);
- if (type == LUA_TNIL)
- {
- buffer1 = 0;
- length1 = 0;
- }
- else if(type == LUA_TSTRING)
- {
- buffer1 = lua_tolstring(L, 1, &length1);
- }
- else if (type == LUA_TUSERDATA)
- {
- ws2812_buffer * buffer = (ws2812_buffer*)luaL_checkudata(L, 1, "ws2812.buffer");
- buffer1 = buffer->values;
- length1 = buffer->colorsPerLed*buffer->size;
- }
- else
- {
- luaL_argerror(L, 1, "ws2812.buffer or string expected");
- }
- // Second optionnal parameter
- type = lua_type(L, 2);
- if (type == LUA_TNONE || type == LUA_TNIL)
- {
- buffer2 = 0;
- length2 = 0;
- }
- else if (type == LUA_TSTRING)
- {
- buffer2 = lua_tolstring(L, 2, &length2);
- }
- else if (type == LUA_TUSERDATA)
- {
- ws2812_buffer * buffer = (ws2812_buffer*)luaL_checkudata(L, 2, "ws2812.buffer");
- buffer2 = buffer->values;
- length2 = buffer->colorsPerLed*buffer->size;
- }
- else
- {
- luaL_argerror(L, 2, "ws2812.buffer or string expected");
- }
- // Send the buffers
- ws2812_write_data(buffer1, length1, buffer2, length2);
- return 0;
- }
- static ptrdiff_t posrelat(ptrdiff_t pos, size_t len) {
- /* relative string position: negative means back from end */
- if (pos < 0) pos += (ptrdiff_t)len + 1;
- return MIN(MAX(pos, 1), len);
- }
- static ws2812_buffer *allocate_buffer(lua_State *L, int leds, int colorsPerLed) {
- // Allocate memory
- size_t size = sizeof(ws2812_buffer) + colorsPerLed*leds;
- ws2812_buffer * buffer = (ws2812_buffer*)lua_newuserdata(L, size);
- // Associate its metatable
- luaL_getmetatable(L, "ws2812.buffer");
- lua_setmetatable(L, -2);
- // Save led strip size
- buffer->size = leds;
- buffer->colorsPerLed = colorsPerLed;
- return buffer;
- }
- // Handle a buffer where we can store led values
- static int ws2812_new_buffer(lua_State *L) {
- const int leds = luaL_checkint(L, 1);
- const int colorsPerLed = luaL_checkint(L, 2);
- luaL_argcheck(L, leds > 0, 1, "should be a positive integer");
- luaL_argcheck(L, colorsPerLed > 0, 2, "should be a positive integer");
- ws2812_buffer * buffer = allocate_buffer(L, leds, colorsPerLed);
- memset(buffer->values, 0, colorsPerLed * leds);
- return 1;
- }
- int ws2812_buffer_fill(ws2812_buffer * buffer, int * colors) {
- // Grab colors
- int i, j;
- // Fill buffer
- uint8_t * p = &buffer->values[0];
- for(i = 0; i < buffer->size; i++)
- {
- for (j = 0; j < buffer->colorsPerLed; j++)
- {
- *p++ = colors[j];
- }
- }
- return 0;
- }
- static int ws2812_buffer_fill_lua(lua_State* L) {
- ws2812_buffer * buffer = (ws2812_buffer*)luaL_checkudata(L, 1, "ws2812.buffer");
- // Grab colors
- int i;
- int * colors = luaM_malloc(L, buffer->colorsPerLed * sizeof(int));
- for (i = 0; i < buffer->colorsPerLed; i++)
- {
- colors[i] = luaL_checkinteger(L, 2+i);
- }
- ws2812_buffer_fill(buffer, colors);
- // Free memory
- luaM_free(L, colors);
- return 0;
- }
- void ws2812_buffer_fade(ws2812_buffer * buffer, int fade, unsigned direction) {
- uint8_t * p = &buffer->values[0];
- int val = 0;
- int i;
- for (i = 0; i < buffer->size * buffer->colorsPerLed; i++)
- {
- if (direction == FADE_OUT)
- {
- *p++ /= fade;
- }
- else
- {
- // as fade in can result in value overflow, an int is used to perform the check afterwards
- val = *p * fade;
- if (val > 255) val = 255;
- *p++ = val;
- }
- }
- }
- static int ws2812_buffer_fade_lua(lua_State* L) {
- ws2812_buffer * buffer = (ws2812_buffer*)luaL_checkudata(L, 1, "ws2812.buffer");
- const int fade = luaL_checkinteger(L, 2);
- unsigned direction = luaL_optinteger( L, 3, FADE_OUT );
- luaL_argcheck(L, fade > 0, 2, "fade value should be a strict positive number");
- ws2812_buffer_fade(buffer, fade, direction);
- return 0;
- }
- int ws2812_buffer_shift(lua_State* L, ws2812_buffer * buffer, int shiftValue, unsigned shift_type, int pos_start, int pos_end){
-
- ws2812_buffer_shift_prepare* prepare = ws2812_buffer_get_shift_prepare(L, buffer, shiftValue, shift_type, pos_start, pos_end);
- ws2812_buffer_shift_prepared(prepare);
- // Free memory
- luaM_freemem(L, prepare, sizeof(ws2812_buffer_shift_prepare) + prepare->shift_len);
- return 0;
- }
- ws2812_buffer_shift_prepare* ws2812_buffer_get_shift_prepare(lua_State* L, ws2812_buffer * buffer, int shiftValue, unsigned shift_type, int pos_start, int pos_end){
- ptrdiff_t start = posrelat(pos_start, buffer->size);
- ptrdiff_t end = posrelat(pos_end, buffer->size);
- start--;
- int size = end - start;
- size_t offset = start * buffer->colorsPerLed;
- luaL_argcheck(L, shiftValue >= 0-size && shiftValue <= size, 2, "shifting more elements than buffer size");
- int shift = shiftValue >= 0 ? shiftValue : -shiftValue;
- size_t shift_len, remaining_len;
- // calculate length of shift section and remaining section
- shift_len = shift*buffer->colorsPerLed;
- remaining_len = (size-shift)*buffer->colorsPerLed;
- ws2812_buffer_shift_prepare* prepare = luaM_malloc(L, sizeof(ws2812_buffer_shift_prepare) + shift_len);
- prepare->offset = offset;
- prepare->tmp_pixels = (uint8_t*)(prepare+1);
- prepare->shiftValue = shiftValue;
- prepare->shift_len = shift_len;
- prepare->remaining_len = remaining_len;
- prepare->shift_type = shift_type;
- prepare->buffer = buffer;
- return prepare;
- }
- void ws2812_buffer_shift_prepared(ws2812_buffer_shift_prepare* prepare) {
- // check if we want to shift at all
- if (prepare->shift_len == 0 || (prepare->shift_len + prepare->remaining_len) <= 0)
- {
- return;
- }
- if (prepare->shiftValue > 0)
- {
- // Store the values which are moved out of the array (last n pixels)
- memcpy(prepare->tmp_pixels, &prepare->buffer->values[prepare->offset + prepare->remaining_len], prepare->shift_len);
- // Move pixels to end
- os_memmove(&prepare->buffer->values[prepare->offset + prepare->shift_len], &prepare->buffer->values[prepare->offset], prepare->remaining_len);
- // Fill beginning with temp data
- if (prepare->shift_type == SHIFT_LOGICAL)
- {
- memset(&prepare->buffer->values[prepare->offset], 0, prepare->shift_len);
- }
- else
- {
- memcpy(&prepare->buffer->values[prepare->offset], prepare->tmp_pixels, prepare->shift_len);
- }
- }
- else
- {
- // Store the values which are moved out of the array (last n pixels)
- memcpy(prepare->tmp_pixels, &prepare->buffer->values[prepare->offset], prepare->shift_len);
- // Move pixels to end
- os_memmove(&prepare->buffer->values[prepare->offset], &prepare->buffer->values[prepare->offset + prepare->shift_len], prepare->remaining_len);
- // Fill beginning with temp data
- if (prepare->shift_type == SHIFT_LOGICAL)
- {
- memset(&prepare->buffer->values[prepare->offset + prepare->remaining_len], 0, prepare->shift_len);
- }
- else
- {
- memcpy(&prepare->buffer->values[prepare->offset + prepare->remaining_len], prepare->tmp_pixels, prepare->shift_len);
- }
- }
- }
- static int ws2812_buffer_shift_lua(lua_State* L) {
- ws2812_buffer * buffer = (ws2812_buffer*)luaL_checkudata(L, 1, "ws2812.buffer");
- const int shiftValue = luaL_checkinteger(L, 2);
- const unsigned shift_type = luaL_optinteger( L, 3, SHIFT_LOGICAL );
- const int pos_start = luaL_optinteger(L, 4, 1);
- const int pos_end = luaL_optinteger(L, 5, -1);
- ws2812_buffer_shift(L, buffer, shiftValue, shift_type, pos_start, pos_end);
- return 0;
- }
- static int ws2812_buffer_dump(lua_State* L) {
- ws2812_buffer * buffer = (ws2812_buffer*)luaL_checkudata(L, 1, "ws2812.buffer");
- lua_pushlstring(L, buffer->values, buffer->size * buffer->colorsPerLed);
- return 1;
- }
- static int ws2812_buffer_replace(lua_State* L) {
- ws2812_buffer * buffer = (ws2812_buffer*)luaL_checkudata(L, 1, "ws2812.buffer");
- ptrdiff_t start = posrelat(luaL_optinteger(L, 3, 1), buffer->size);
- uint8_t *src;
- size_t srcLen;
- if (lua_type(L, 2) == LUA_TSTRING) {
- size_t length;
- src = (uint8_t *) lua_tolstring(L, 2, &length);
- srcLen = length / buffer->colorsPerLed;
- } else {
- ws2812_buffer * rhs = (ws2812_buffer*)luaL_checkudata(L, 2, "ws2812.buffer");
- src = rhs->values;
- srcLen = rhs->size;
- luaL_argcheck(L, rhs->colorsPerLed == buffer->colorsPerLed, 2, "Buffers have different colors");
- }
- luaL_argcheck(L, srcLen + start - 1 <= buffer->size, 2, "Does not fit into destination");
- memcpy(buffer->values + (start - 1) * buffer->colorsPerLed, src, srcLen * buffer->colorsPerLed);
- return 0;
- }
- // buffer:mix(factor1, buffer1, ..)
- // factor is 256 for 100%
- // uses saturating arithmetic (one buffer at a time)
- static int ws2812_buffer_mix(lua_State* L) {
- ws2812_buffer * buffer = (ws2812_buffer*)luaL_checkudata(L, 1, "ws2812.buffer");
- int pos = 2;
- size_t cells = buffer->size * buffer->colorsPerLed;
- int n_sources = (lua_gettop(L) - 1) / 2;
- struct {
- int factor;
- const uint8_t *values;
- } source[n_sources];
- int src;
- for (src = 0; src < n_sources; src++, pos += 2) {
- int factor = luaL_checkinteger(L, pos);
- ws2812_buffer *src_buffer = (ws2812_buffer*) luaL_checkudata(L, pos + 1, "ws2812.buffer");
- luaL_argcheck(L, src_buffer->size == buffer->size && src_buffer->colorsPerLed == buffer->colorsPerLed, pos + 1, "Buffer not same shape");
- source[src].factor = factor;
- source[src].values = src_buffer->values;
- }
- size_t i;
- for (i = 0; i < cells; i++) {
- int32_t val = 0;
- for (src = 0; src < n_sources; src++) {
- val += (int32_t)(source[src].values[i] * source[src].factor);
- }
- val += 128; // rounding istead of floor
- val /= 256; // do not use implemetation dependant right shift
- if (val < 0) {
- val = 0;
- } else if (val > 255) {
- val = 255;
- }
- buffer->values[i] = (uint8_t)val;
- }
- return 0;
- }
- // Returns the total of all channels
- static int ws2812_buffer_power(lua_State* L) {
- ws2812_buffer * buffer = (ws2812_buffer*)luaL_checkudata(L, 1, "ws2812.buffer");
- size_t cells = buffer->size * buffer->colorsPerLed;
- size_t i;
- int total = 0;
- for (i = 0; i < cells; i++) {
- total += buffer->values[i];
- }
- lua_pushinteger(L, total);
- return 1;
- }
- static int ws2812_buffer_get(lua_State* L) {
- ws2812_buffer * buffer = (ws2812_buffer*)luaL_checkudata(L, 1, "ws2812.buffer");
- const int led = luaL_checkinteger(L, 2) - 1;
- luaL_argcheck(L, led >= 0 && led < buffer->size, 2, "index out of range");
- int i;
- for (i = 0; i < buffer->colorsPerLed; i++)
- {
- lua_pushinteger(L, buffer->values[buffer->colorsPerLed*led+i]);
- }
- return buffer->colorsPerLed;
- }
- static int ws2812_buffer_set(lua_State* L) {
- ws2812_buffer * buffer = (ws2812_buffer*)luaL_checkudata(L, 1, "ws2812.buffer");
- const int led = luaL_checkinteger(L, 2) - 1;
- luaL_argcheck(L, led >= 0 && led < buffer->size, 2, "index out of range");
- int type = lua_type(L, 3);
- if(type == LUA_TTABLE)
- {
- int i;
- for (i = 0; i < buffer->colorsPerLed; i++)
- {
- // Get value and push it on stack
- lua_rawgeti(L, 3, i+1);
- // Convert it as int and store them in buffer
- buffer->values[buffer->colorsPerLed*led+i] = lua_tointeger(L, -1);
- }
- // Clean up the stack
- lua_pop(L, buffer->colorsPerLed);
- }
- else if(type == LUA_TSTRING)
- {
- size_t len;
- const char * buf = lua_tolstring(L, 3, &len);
- // Overflow check
- if( buffer->colorsPerLed*led + len > buffer->colorsPerLed*buffer->size )
- {
- return luaL_error(L, "string size will exceed strip length");
- }
- memcpy(&buffer->values[buffer->colorsPerLed*led], buf, len);
- }
- else
- {
- int i;
- for (i = 0; i < buffer->colorsPerLed; i++)
- {
- buffer->values[buffer->colorsPerLed*led+i] = luaL_checkinteger(L, 3+i);
- }
- }
- return 0;
- }
- static int ws2812_buffer_size(lua_State* L) {
- ws2812_buffer * buffer = (ws2812_buffer*)luaL_checkudata(L, 1, "ws2812.buffer");
- lua_pushinteger(L, buffer->size);
- return 1;
- }
- static int ws2812_buffer_sub(lua_State* L) {
- ws2812_buffer * lhs = (ws2812_buffer*)luaL_checkudata(L, 1, "ws2812.buffer");
- size_t l = lhs->size;
- ptrdiff_t start = posrelat(luaL_checkinteger(L, 2), l);
- ptrdiff_t end = posrelat(luaL_optinteger(L, 3, -1), l);
- if (start <= end) {
- ws2812_buffer *result = allocate_buffer(L, end - start + 1, lhs->colorsPerLed);
- memcpy(result->values, lhs->values + lhs->colorsPerLed * (start - 1), lhs->colorsPerLed * (end - start + 1));
- } else {
- ws2812_buffer *result = allocate_buffer(L, 0, lhs->colorsPerLed);
- }
- return 1;
- }
- static int ws2812_buffer_concat(lua_State* L) {
- ws2812_buffer * lhs = (ws2812_buffer*)luaL_checkudata(L, 1, "ws2812.buffer");
- ws2812_buffer * rhs = (ws2812_buffer*)luaL_checkudata(L, 2, "ws2812.buffer");
- luaL_argcheck(L, lhs->colorsPerLed == rhs->colorsPerLed, 1, "Can only concatenate buffers with same colors");
- int colorsPerLed = lhs->colorsPerLed;
- int leds = lhs->size + rhs->size;
- ws2812_buffer * buffer = allocate_buffer(L, leds, colorsPerLed);
- memcpy(buffer->values, lhs->values, lhs->colorsPerLed * lhs->size);
- memcpy(buffer->values + lhs->colorsPerLed * lhs->size, rhs->values, rhs->colorsPerLed * rhs->size);
- return 1;
- }
- static int ws2812_buffer_tostring(lua_State* L) {
- ws2812_buffer * buffer = (ws2812_buffer*)luaL_checkudata(L, 1, "ws2812.buffer");
- luaL_Buffer result;
- luaL_buffinit(L, &result);
- luaL_addchar(&result, '[');
- int i;
- int p = 0;
- for (i = 0; i < buffer->size; i++) {
- int j;
- if (i > 0) {
- luaL_addchar(&result, ',');
- }
- luaL_addchar(&result, '(');
- for (j = 0; j < buffer->colorsPerLed; j++, p++) {
- if (j > 0) {
- luaL_addchar(&result, ',');
- }
- char numbuf[5];
- sprintf(numbuf, "%d", buffer->values[p]);
- luaL_addstring(&result, numbuf);
- }
- luaL_addchar(&result, ')');
- }
- luaL_addchar(&result, ']');
- luaL_pushresult(&result);
- return 1;
- }
- LROT_BEGIN(ws2812_buffer_map, NULL, LROT_MASK_INDEX)
- LROT_FUNCENTRY( __concat, ws2812_buffer_concat )
- LROT_TABENTRY( __index, ws2812_buffer_map )
- LROT_FUNCENTRY( __tostring, ws2812_buffer_tostring )
- LROT_FUNCENTRY( dump, ws2812_buffer_dump )
- LROT_FUNCENTRY( fade, ws2812_buffer_fade_lua)
- LROT_FUNCENTRY( fill, ws2812_buffer_fill_lua )
- LROT_FUNCENTRY( get, ws2812_buffer_get )
- LROT_FUNCENTRY( replace, ws2812_buffer_replace )
- LROT_FUNCENTRY( mix, ws2812_buffer_mix )
- LROT_FUNCENTRY( power, ws2812_buffer_power )
- LROT_FUNCENTRY( set, ws2812_buffer_set )
- LROT_FUNCENTRY( shift, ws2812_buffer_shift_lua )
- LROT_FUNCENTRY( size, ws2812_buffer_size )
- LROT_FUNCENTRY( sub, ws2812_buffer_sub )
- LROT_END(ws2812_buffer_map, NULL, LROT_MASK_INDEX)
- LROT_BEGIN(ws2812, NULL, 0)
- LROT_FUNCENTRY( init, ws2812_init )
- LROT_FUNCENTRY( newBuffer, ws2812_new_buffer )
- LROT_FUNCENTRY( write, ws2812_write )
- LROT_NUMENTRY( FADE_IN, FADE_IN )
- LROT_NUMENTRY( FADE_OUT, FADE_OUT )
- LROT_NUMENTRY( MODE_SINGLE, MODE_SINGLE )
- LROT_NUMENTRY( MODE_DUAL, MODE_DUAL )
- LROT_NUMENTRY( SHIFT_LOGICAL, SHIFT_LOGICAL )
- LROT_NUMENTRY( SHIFT_CIRCULAR, SHIFT_CIRCULAR )
- LROT_END(ws2812, NULL, 0)
- int luaopen_ws2812(lua_State *L) {
- // TODO: Make sure that the GPIO system is initialized
- luaL_rometatable(L, "ws2812.buffer", LROT_TABLEREF(ws2812_buffer_map));
- return 0;
- }
- NODEMCU_MODULE(WS2812, "ws2812", ws2812, luaopen_ws2812);
|