mirror of
https://github.com/ADElectronics/RTL00_WEB_WS2812.git
synced 2026-07-08 21:55:38 +00:00
SDK update
This commit is contained in:
A_D
parent
7ac60bf6e1
commit
68362feaa1
1153 changed files with 97410 additions and 592771 deletions
329
Firmware/Project/WS2812/ws2812.c
Normal file
329
Firmware/Project/WS2812/ws2812.c
Normal file
|
|
@ -0,0 +1,329 @@
|
|||
#include "ws2812.h"
|
||||
|
||||
/* Quick and dirty, we use one big DMA buffer for the whole strip length.
|
||||
* TODO: use smaller DMA buffer and fill in bit patterns on the fly */
|
||||
uint8_t dma_buffer[WS2812_DMABUF_LEN(WS2812_LEDS_MAX)];
|
||||
|
||||
/* scale uint8 value from range 2-255 to range 0-scale */
|
||||
static inline uint8_t WS2812_Scale(uint8_t value, uint8_t scale)
|
||||
{
|
||||
uint32_t tmp;
|
||||
|
||||
tmp = value * scale;
|
||||
tmp /= 256;
|
||||
|
||||
return (uint8_t)tmp;
|
||||
}
|
||||
|
||||
// wake up waiting tasks when DMA transfer is complete
|
||||
static void master_tr_done_callback(void *pdata, SpiIrq event)
|
||||
{
|
||||
BaseType_t task_woken, result;
|
||||
ws2812_t *cfg;
|
||||
|
||||
task_woken = pdFALSE;
|
||||
cfg = (ws2812_t *) pdata;
|
||||
|
||||
switch (event)
|
||||
{
|
||||
case SpiRxIrq:
|
||||
break;
|
||||
|
||||
case SpiTxIrq:
|
||||
result = xEventGroupSetBitsFromISR(cfg->events, BIT_DONE, &task_woken);
|
||||
if(result == pdPASS)
|
||||
{
|
||||
portYIELD_FROM_ISR(task_woken);
|
||||
}
|
||||
break;
|
||||
|
||||
default:
|
||||
DBG_8195A("WS2812: Unknown SPI irq event!\n");
|
||||
}
|
||||
}
|
||||
|
||||
static void WS2812_HSV2RGB(ws2812_hsv_t *hsv, ws2812_rgb_t *rgb)
|
||||
{
|
||||
//uint8_t r, g, b;
|
||||
uint8_t hue, sat, val;
|
||||
uint8_t base, sector, offset;
|
||||
uint8_t rise, fall;
|
||||
|
||||
// scale hue to range 0- 3*64. Makes subsequent calculations easier
|
||||
hue = WS2812_Scale(hsv->hue, 192);
|
||||
sat = hsv->sat;
|
||||
val = hsv->value;
|
||||
|
||||
sector = hue / 64;
|
||||
offset = hue % 64;
|
||||
|
||||
// get common white base level and remaining colour amplitude
|
||||
base = 255 - sat;
|
||||
|
||||
rise = (offset * sat * 4) / 256;
|
||||
fall = 255 - base - rise;
|
||||
|
||||
rise = (rise * val) / 256;
|
||||
fall = (fall * val) / 256;
|
||||
base = (base * val) / 256;
|
||||
|
||||
rgb->r = base;
|
||||
rgb->g = base;
|
||||
rgb->b = base;
|
||||
|
||||
switch (sector)
|
||||
{
|
||||
case 0:
|
||||
rgb->r += fall;
|
||||
rgb->g += rise;
|
||||
break;
|
||||
case 1:
|
||||
rgb->g += fall;
|
||||
rgb->b += rise;
|
||||
break;
|
||||
case 2:
|
||||
rgb->r += rise;
|
||||
rgb->b += fall;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
// convert a RGB byte into SPI data stream with 2 bits per byte
|
||||
static uint8_t *WS2812_RGB2PWM(uint8_t *dst, const uint8_t colour)
|
||||
{
|
||||
uint8_t cnt, data = colour;
|
||||
|
||||
for (cnt = 0; cnt < 4; cnt++)
|
||||
{
|
||||
switch (data & 0xC0)
|
||||
{
|
||||
case 0x00:
|
||||
*dst = WS2812_BITS_00;
|
||||
break;
|
||||
case 0x40:
|
||||
*dst = WS2812_BITS_01;
|
||||
break;
|
||||
case 0x80:
|
||||
*dst = WS2812_BITS_10;
|
||||
break;
|
||||
case 0xC0:
|
||||
*dst = WS2812_BITS_11;
|
||||
break;
|
||||
}
|
||||
dst++;
|
||||
data <<= 2;
|
||||
}
|
||||
return dst;
|
||||
}
|
||||
|
||||
int32_t WS2812_Tx(ws2812_t *cfg, uint16_t delay)
|
||||
{
|
||||
EventBits_t rcvd_events;
|
||||
TickType_t timeout;
|
||||
BaseType_t status;
|
||||
|
||||
// wait for any SPI transfer to finish
|
||||
while(cfg->spi_master.state & SPI_STATE_TX_BUSY)
|
||||
{
|
||||
vTaskDelay(0);
|
||||
}
|
||||
|
||||
// obey requested delay
|
||||
if(delay > 0)
|
||||
{
|
||||
vTaskDelay(delay);
|
||||
}
|
||||
|
||||
// lock the DMA buffer mutex while it is transferred
|
||||
status = xSemaphoreTake(cfg->mutex, configTICK_RATE_HZ);
|
||||
if(status != pdTRUE)
|
||||
{
|
||||
printf("WS2812: [%s] Timeout waiting for config mutex.\n", __func__);
|
||||
xSemaphoreGive(cfg->mutex);
|
||||
return -1;
|
||||
}
|
||||
|
||||
if(cfg->dma_buff == NULL || cfg->buff_len == 0)
|
||||
{
|
||||
printf("WS2812: [%s] DMA buffer invalid\n", __func__);
|
||||
xSemaphoreGive(cfg->mutex);
|
||||
return -1;
|
||||
}
|
||||
|
||||
xEventGroupClearBits(cfg->events, BIT_DONE);
|
||||
spi_master_write_stream_dma(&cfg->spi_master, (char *)(cfg->dma_buff), cfg->buff_len);
|
||||
|
||||
timeout = 1000 / portTICK_PERIOD_MS;
|
||||
rcvd_events = xEventGroupWaitBits(
|
||||
cfg->events,
|
||||
BIT_DONE, // wait for DMA TX done
|
||||
pdTRUE, // clear event bit
|
||||
pdFALSE, // do not wait for all bits to be set
|
||||
timeout );
|
||||
|
||||
if(!(rcvd_events & BIT_DONE))
|
||||
{
|
||||
printf("WS2812: [%s] DMA timeout\n", __func__);
|
||||
xSemaphoreGive(cfg->mutex);
|
||||
return -1;
|
||||
}
|
||||
|
||||
xSemaphoreGive(cfg->mutex);
|
||||
return 1;
|
||||
}
|
||||
|
||||
int32_t ws2812_Update(ws2812_t *cfg, ws2812_hsv_t hsv_values[], uint32_t strip_len, uint16_t delay)
|
||||
{
|
||||
uint32_t i;
|
||||
uint8_t *bufp;
|
||||
uint16_t len;
|
||||
BaseType_t status;
|
||||
ws2812_rgb_t rgb;
|
||||
|
||||
// Æäåì, ïîêà îñâîáîäèòñÿ áóôôåð DMA
|
||||
while(cfg->spi_master.state & SPI_STATE_TX_BUSY)
|
||||
{
|
||||
vTaskDelay(0);
|
||||
}
|
||||
|
||||
status = xSemaphoreTake(cfg->mutex, configTICK_RATE_HZ);
|
||||
if(status != pdTRUE)
|
||||
{
|
||||
printf("WS2812: [%s] Timeout waiting for config mutex.\n", __func__);
|
||||
return -1;
|
||||
}
|
||||
bufp = &(cfg->dma_buff[0]);
|
||||
|
||||
// Ïðîâåðêà íà âìåùàåìîñòü è êîïèðîâàíèå
|
||||
len = min(strip_len, cfg->strip_len);
|
||||
for(i = 0; i < len; i++)
|
||||
{
|
||||
WS2812_HSV2RGB(&hsv_values[i], &rgb);
|
||||
bufp = WS2812_RGB2PWM(bufp, rgb.g);
|
||||
bufp = WS2812_RGB2PWM(bufp, rgb.r);
|
||||
bufp = WS2812_RGB2PWM(bufp, rgb.b);
|
||||
}
|
||||
|
||||
// Âñå íå èñïîëüçóåìûå ïèêñåëè - âûêëþ÷àåì
|
||||
if(cfg->strip_len > len)
|
||||
{
|
||||
memset(bufp, WS2812_BITS_00, cfg->strip_len - len);
|
||||
bufp += cfg->strip_len - len;
|
||||
}
|
||||
|
||||
memset(bufp, 0x0, WS2812_RESET_LEN);
|
||||
xSemaphoreGive(cfg->mutex);
|
||||
|
||||
return WS2812_Tx(cfg, delay);
|
||||
}
|
||||
|
||||
ws2812_t *ws2812_Init(uint16_t strip_len)
|
||||
{
|
||||
int32_t result;
|
||||
ws2812_t *cfg;
|
||||
|
||||
result = 0;
|
||||
|
||||
cfg = malloc(sizeof(ws2812_t));
|
||||
if(cfg == NULL)
|
||||
{
|
||||
printf("WS2812: [%s] malloc for cfg failed\n", __func__);
|
||||
result = -1;
|
||||
goto err_out;
|
||||
}
|
||||
|
||||
memset(cfg, 0x0, sizeof(ws2812_t));
|
||||
|
||||
cfg->mutex = xSemaphoreCreateMutex();
|
||||
if(cfg->mutex == NULL)
|
||||
{
|
||||
printf("WS2812: [%s] Mutex creation failed\n", __func__);
|
||||
result = -1;
|
||||
goto err_out;
|
||||
}
|
||||
|
||||
cfg->events = xEventGroupCreate();
|
||||
if(cfg->events == NULL)
|
||||
{
|
||||
printf("WS2812: [%s] Creating event group failed\n", __func__);
|
||||
result = -1;
|
||||
goto err_out;
|
||||
}
|
||||
|
||||
spi_init(&(cfg->spi_master), WS2812_SPI_MOSI, WS2812_SPI_MISO, WS2812_SPI_SCLK, WS2812_SPI_CS);
|
||||
spi_format(&(cfg->spi_master), 8, 3, 0);
|
||||
spi_frequency(&(cfg->spi_master), WS2812_SPI_FREQ);
|
||||
spi_irq_hook(&(cfg->spi_master), (spi_irq_handler)master_tr_done_callback, (uint32_t)cfg);
|
||||
|
||||
result = ws2812_SetLen(cfg, strip_len);
|
||||
if(!result)
|
||||
{
|
||||
printf("WS2812: [%s] ws2812_set_len() failed\n", __func__);
|
||||
}
|
||||
|
||||
err_out:
|
||||
if(!result && cfg != NULL)
|
||||
{
|
||||
if(cfg->mutex != NULL)
|
||||
{
|
||||
vQueueDelete(cfg->mutex);
|
||||
}
|
||||
|
||||
if(cfg->events != NULL)
|
||||
{
|
||||
vEventGroupDelete(cfg->events);
|
||||
}
|
||||
|
||||
if(cfg->dma_buff != NULL)
|
||||
{
|
||||
free(cfg->dma_buff);
|
||||
}
|
||||
|
||||
free(cfg);
|
||||
cfg = NULL;
|
||||
}
|
||||
|
||||
return cfg;
|
||||
}
|
||||
|
||||
int32_t ws2812_SetLen(ws2812_t *cfg, uint16_t strip_len)
|
||||
{
|
||||
BaseType_t status;
|
||||
uint32_t reset_off;
|
||||
|
||||
if(cfg == NULL)
|
||||
{
|
||||
printf("WS2812: [%s] no config given\n", __func__);
|
||||
return -1;
|
||||
}
|
||||
|
||||
// lock the config mutex
|
||||
status = xSemaphoreTake(cfg->mutex, configTICK_RATE_HZ);
|
||||
if(status != pdTRUE)
|
||||
{
|
||||
printf("WS2812: [%s] Timeout waiting for config mutex.\n", __func__);
|
||||
return -1;
|
||||
}
|
||||
|
||||
if(strip_len <= WS2812_LEDS_MAX)
|
||||
{
|
||||
/* TODO: use dynamically allocated buffer */
|
||||
cfg->dma_buff = dma_buffer;
|
||||
|
||||
/* initialise LEDs to off and add reset pulse at end of strip */
|
||||
reset_off = WS2812_DMABUF_LEN(strip_len) - WS2812_RESET_LEN;
|
||||
memset(&(cfg->dma_buff[0]), WS2812_BITS_00, reset_off);
|
||||
memset(&(cfg->dma_buff[reset_off]), 0x0, WS2812_RESET_LEN);
|
||||
cfg->strip_len = strip_len;
|
||||
cfg->buff_len = WS2812_DMABUF_LEN(strip_len);
|
||||
}
|
||||
else
|
||||
{
|
||||
printf("WS2812: [%s] Strip too long for DMA buffer\n", __func__);
|
||||
return -1;
|
||||
}
|
||||
|
||||
xSemaphoreGive(cfg->mutex);
|
||||
return 1;
|
||||
}
|
||||
|
||||
Loading…
Add table
Add a link
Reference in a new issue