okhsunrog/esp_ws28xx
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

code formatting

Browse source
This commit is contained in:
Данила Горнушко 2023-09-21 05:04:24 +03:00
parent 5502b04fcb
commit 2176892c3b
3 changed files with 147 additions and 127 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
.editorconfig Normal file
View file

@ -0,0 +1,26 @@
# EditorConfig is awesome: https://EditorConfig.org
# top-most EditorConfig file
root = true
# Unix-style newlines with a newline ending every file
[*]
end_of_line = lf
insert_final_newline = true
# Set default charset
charset = utf-8
# 4 space indentation
[*.{c, h, cpp}]
indent_style = space
indent_size = 4
# Tab indentation (no size specified)
[Makefile]
indent_style = tab
# Matches the exact files either package.json or .travis.yml
[{package.json,.travis.yml}]
indent_style = space
indent_size = 2

190
esp_ws28xx.c
View file

@ -1,115 +1,107 @@
#include "esp_ws28xx.h" #include "esp_ws28xx.h"
uint16_t *dma_buffer; uint16_t* dma_buffer;
CRGB *ws28xx_pixels; CRGB* ws28xx_pixels;
static int n_of_leds, reset_delay, dma_buf_size; static int n_of_leds, reset_delay, dma_buf_size;
led_strip_model_t led_model; led_strip_model_t led_model;
static spi_settings_t spi_settings = { static spi_settings_t spi_settings = {.host = SPI2_HOST,
.host = SPI2_HOST, .dma_chan = SPI_DMA_CH_AUTO,
.dma_chan = SPI_DMA_CH_AUTO, .buscfg =
.buscfg = {
{ .miso_io_num = -1,
.miso_io_num = -1, .sclk_io_num = -1,
.sclk_io_num = -1, .quadwp_io_num = -1,
.quadwp_io_num = -1, .quadhd_io_num = -1,
.quadhd_io_num = -1, },
}, .devcfg = {
.devcfg = { .clock_speed_hz = 3.2 * 1000 * 1000, // Clock out at 3.2 MHz
.clock_speed_hz = 3.2 * 1000 * 1000, // Clock out at 3.2 MHz .mode = 0, // SPI mode 0
.mode = 0, // SPI mode 0 .spics_io_num = -1, // CS pin
.spics_io_num = -1, // CS pin .queue_size = 1,
.queue_size = 1, .command_bits = 0,
.command_bits = 0, .address_bits = 0,
.address_bits = 0, .flags = SPI_DEVICE_TXBIT_LSBFIRST,
.flags = SPI_DEVICE_TXBIT_LSBFIRST, }};
}};
esp_err_t ws28xx_init(int pin, led_strip_model_t model, int num_of_leds, esp_err_t ws28xx_init(int pin, led_strip_model_t model, int num_of_leds, CRGB** led_buffer_ptr) {
CRGB **led_buffer_ptr) { esp_err_t err = ESP_OK;
esp_err_t err = ESP_OK; n_of_leds = num_of_leds;
n_of_leds = num_of_leds; led_model = model;
led_model = model; dma_buf_size =
dma_buf_size = n_of_leds * 12 + (reset_delay + 1) * 2; // 12 bytes for each led + bytes for initial zero and reset state
n_of_leds * 12 + ws28xx_pixels = malloc(sizeof(CRGB) * n_of_leds);
(reset_delay + 1) * if (ws28xx_pixels == NULL) {
2; // 12 bytes for each led + bytes for initial zero and reset state return ESP_ERR_NO_MEM;
ws28xx_pixels = malloc(sizeof(CRGB) * n_of_leds); }
if (ws28xx_pixels == NULL) { *led_buffer_ptr = ws28xx_pixels;
return ESP_ERR_NO_MEM; spi_settings.buscfg.mosi_io_num = pin;
} spi_settings.buscfg.max_transfer_sz = dma_buf_size;
*led_buffer_ptr = ws28xx_pixels; err = spi_bus_initialize(spi_settings.host, &spi_settings.buscfg, spi_settings.dma_chan);
spi_settings.buscfg.mosi_io_num = pin; if (err != ESP_OK) {
spi_settings.buscfg.max_transfer_sz = dma_buf_size; free(ws28xx_pixels);
err = spi_bus_initialize(spi_settings.host, &spi_settings.buscfg, return err;
spi_settings.dma_chan); }
if (err != ESP_OK) { err = spi_bus_add_device(spi_settings.host, &spi_settings.devcfg, &spi_settings.spi);
free(ws28xx_pixels); if (err != ESP_OK) {
return err; free(ws28xx_pixels);
} return err;
err = spi_bus_add_device(spi_settings.host, &spi_settings.devcfg, }
&spi_settings.spi); reset_delay = (model == WS2812B) ? 3 : 30; // insrease if something breaks. values are less that
if (err != ESP_OK) { // recommended in datasheets but seem stable
free(ws28xx_pixels); dma_buffer = heap_caps_malloc(dma_buf_size,
return err; MALLOC_CAP_DMA); // Critical to be DMA memory.
} if (dma_buffer == NULL) {
reset_delay = (model == WS2812B) free(ws28xx_pixels);
? 3 return ESP_ERR_NO_MEM;
: 30; // insrease if something breaks. values are less that }
// recommended in datasheets but seem stable return ESP_OK;
dma_buffer = heap_caps_malloc(dma_buf_size,
MALLOC_CAP_DMA); // Critical to be DMA memory.
if (dma_buffer == NULL) {
free(ws28xx_pixels);
return ESP_ERR_NO_MEM;
}
return ESP_OK;
} }
void ws28xx_fill_all(CRGB color) { void ws28xx_fill_all(CRGB color) {
for (int i = 0; i < n_of_leds; i++) { for (int i = 0; i < n_of_leds; i++) {
ws28xx_pixels[i] = color; ws28xx_pixels[i] = color;
} }
} }
esp_err_t ws28xx_update() { esp_err_t ws28xx_update() {
uint16_t timing_bits[16] = {0x1111, 0x7111, 0x1711, 0x7711, 0x1171, 0x7171, uint16_t timing_bits[16] = {0x1111, 0x7111, 0x1711, 0x7711, 0x1171, 0x7171, 0x1771, 0x7771,
0x1771, 0x7771, 0x1117, 0x7117, 0x1717, 0x7717, 0x1117, 0x7117, 0x1717, 0x7717, 0x1177, 0x7177, 0x1777, 0x7777};
0x1177, 0x7177, 0x1777, 0x7777}; esp_err_t err;
esp_err_t err; int n = 0;
int n = 0; memset(dma_buffer, 0, dma_buf_size);
memset(dma_buffer, 0, dma_buf_size);
dma_buffer[n++] = 0;
for (int i = 0; i < n_of_leds; i++) {
uint32_t temp = ws28xx_pixels[i].num; // Data you want to write to each LEDs
if (led_model == WS2815) {
// Red
dma_buffer[n++] = timing_bits[0x0f & (temp >> 4)];
dma_buffer[n++] = timing_bits[0x0f & (temp)];
// Green
dma_buffer[n++] = timing_bits[0x0f & (temp >> 12)];
dma_buffer[n++] = timing_bits[0x0f & (temp) >> 8];
} else {
// Green
dma_buffer[n++] = timing_bits[0x0f & (temp >> 12)];
dma_buffer[n++] = timing_bits[0x0f & (temp) >> 8];
// Red
dma_buffer[n++] = timing_bits[0x0f & (temp >> 4)];
dma_buffer[n++] = timing_bits[0x0f & (temp)];
}
// Blue
dma_buffer[n++] = timing_bits[0x0f & (temp >> 20)];
dma_buffer[n++] = timing_bits[0x0f & (temp) >> 16];
}
for (int i = 0; i < reset_delay; i++)
dma_buffer[n++] = 0; dma_buffer[n++] = 0;
for (int i = 0; i < n_of_leds; i++) {
uint32_t temp = ws28xx_pixels[i].num; // Data you want to write to each LEDs
if (led_model == WS2815) {
// Red
dma_buffer[n++] = timing_bits[0x0f & (temp >> 4)];
dma_buffer[n++] = timing_bits[0x0f & (temp)];
err = spi_device_transmit(spi_settings.spi, &(spi_transaction_t){ // Green
.length = dma_buf_size * 8, dma_buffer[n++] = timing_bits[0x0f & (temp >> 12)];
.tx_buffer = dma_buffer, dma_buffer[n++] = timing_bits[0x0f & (temp) >> 8];
}); } else {
return err; // Green
dma_buffer[n++] = timing_bits[0x0f & (temp >> 12)];
dma_buffer[n++] = timing_bits[0x0f & (temp) >> 8];
// Red
dma_buffer[n++] = timing_bits[0x0f & (temp >> 4)];
dma_buffer[n++] = timing_bits[0x0f & (temp)];
}
// Blue
dma_buffer[n++] = timing_bits[0x0f & (temp >> 20)];
dma_buffer[n++] = timing_bits[0x0f & (temp) >> 16];
}
for (int i = 0; i < reset_delay; i++) {
dma_buffer[n++] = 0;
}
err = spi_device_transmit(spi_settings.spi, &(spi_transaction_t){
.length = dma_buf_size * 8,
.tx_buffer = dma_buffer,
});
return err;
} }

58
esp_ws28xx.h
View file

@ -1,46 +1,48 @@
#ifndef MAIN_ESP_WS28XX_H_ #ifndef MAIN_ESP_WS28XX_H_
#define MAIN_ESP_WS28XX_H_ #define MAIN_ESP_WS28XX_H_
#include "driver/gpio.h"
#include "driver/spi_master.h"
#include <stdio.h> #include <stdio.h>
#include <string.h> #include <string.h>
#include "driver/gpio.h"
#include "driver/spi_master.h"
typedef struct { typedef struct {
union { union {
struct { struct {
union { union {
uint8_t r; uint8_t r;
uint8_t red; uint8_t red;
}; };
union {
uint8_t g; union {
uint8_t green; uint8_t g;
}; uint8_t green;
union { };
uint8_t b;
uint8_t blue; union {
}; uint8_t b;
uint8_t blue;
};
};
uint8_t raw[3];
uint32_t num;
}; };
uint8_t raw[3];
uint32_t num;
};
} CRGB; } CRGB;
typedef struct { typedef struct {
spi_host_device_t host; spi_host_device_t host;
spi_device_handle_t spi; spi_device_handle_t spi;
int dma_chan; int dma_chan;
spi_device_interface_config_t devcfg; spi_device_interface_config_t devcfg;
spi_bus_config_t buscfg; spi_bus_config_t buscfg;
} spi_settings_t; } spi_settings_t;
typedef enum { typedef enum {
WS2812B = 0, WS2812B = 0,
WS2815, WS2815,
} led_strip_model_t; } led_strip_model_t;
esp_err_t ws28xx_init(int pin, led_strip_model_t model, int num_of_leds, esp_err_t ws28xx_init(int pin, led_strip_model_t model, int num_of_leds, CRGB** led_buffer_ptr);
CRGB **led_buffer_ptr);
void ws28xx_fill_all(CRGB color); void ws28xx_fill_all(CRGB color);
esp_err_t ws28xx_update(); esp_err_t ws28xx_update();