rusEFI
The most advanced open source ECU
Loading...
Searching...
No Matches
Functions | Variables
WS2812.cpp File Reference

Detailed Description

WS2812 RGB LED driver.

Date
25.03.2023
Author
Benas Brazdziunas

Definition in file WS2812.cpp.

Functions

void initWS2812 ()
 
void clearWS2812One (uint32_t num)
 
void clearWS2812All ()
 
void setWS2812One (uint32_t num, WS2812_RGB_t rgb_col)
 
void setWS2812All (WS2812_RGB_t rgb_col)
 
void setWS2812Brightness (uint8_t num)
 
void calcBuf ()
 

Variables

static WS2812_RGB_t WS2812_LED_BUF [WS2812_LED_N]
 
static uint32_t WS2812_TIM_BUF [WS2812_BUFLEN]
 
static uint8_t WS2812_BRIGHTNESS = 20
 

Function Documentation

◆ calcBuf()

void calcBuf ( )

Calculate Timer DMA buffer

Definition at line 130 of file WS2812.cpp.

131{
132 uint32_t pos = 0;
133 // set timings for all LEDs
134 for (uint32_t num = 0; num < WS2812_LED_N; num++)
135 {
136 WS2812_RGB_t led = WS2812_LED_BUF[num];
137 float brightness = WS2812_BRIGHTNESS / 100.0;
138
139 led.red = (uint8_t)(led.red * brightness);
140 led.green = (uint8_t)(led.green * brightness);
141 led.blue = (uint8_t)(led.blue * brightness);
142
143 // Col:Green , Bit:7..0
144 WS2812_TIM_BUF[pos++] = ((led.green & 0x80) != 0) ? WS2812_DUTYCYCLE_1 : WS2812_DUTYCYCLE_0;
145 WS2812_TIM_BUF[pos++] = ((led.green & 0x40) != 0) ? WS2812_DUTYCYCLE_1 : WS2812_DUTYCYCLE_0;
146 WS2812_TIM_BUF[pos++] = ((led.green & 0x20) != 0) ? WS2812_DUTYCYCLE_1 : WS2812_DUTYCYCLE_0;
147 WS2812_TIM_BUF[pos++] = ((led.green & 0x10) != 0) ? WS2812_DUTYCYCLE_1 : WS2812_DUTYCYCLE_0;
148 WS2812_TIM_BUF[pos++] = ((led.green & 0x08) != 0) ? WS2812_DUTYCYCLE_1 : WS2812_DUTYCYCLE_0;
149 WS2812_TIM_BUF[pos++] = ((led.green & 0x04) != 0) ? WS2812_DUTYCYCLE_1 : WS2812_DUTYCYCLE_0;
150 WS2812_TIM_BUF[pos++] = ((led.green & 0x02) != 0) ? WS2812_DUTYCYCLE_1 : WS2812_DUTYCYCLE_0;
151 WS2812_TIM_BUF[pos++] = ((led.green & 0x01) != 0) ? WS2812_DUTYCYCLE_1 : WS2812_DUTYCYCLE_0;
152
153 // Col:Red , Bit:7..0
154 WS2812_TIM_BUF[pos++] = ((led.red & 0x80) != 0) ? WS2812_DUTYCYCLE_1 : WS2812_DUTYCYCLE_0;
155 WS2812_TIM_BUF[pos++] = ((led.red & 0x40) != 0) ? WS2812_DUTYCYCLE_1 : WS2812_DUTYCYCLE_0;
156 WS2812_TIM_BUF[pos++] = ((led.red & 0x20) != 0) ? WS2812_DUTYCYCLE_1 : WS2812_DUTYCYCLE_0;
157 WS2812_TIM_BUF[pos++] = ((led.red & 0x10) != 0) ? WS2812_DUTYCYCLE_1 : WS2812_DUTYCYCLE_0;
158 WS2812_TIM_BUF[pos++] = ((led.red & 0x08) != 0) ? WS2812_DUTYCYCLE_1 : WS2812_DUTYCYCLE_0;
159 WS2812_TIM_BUF[pos++] = ((led.red & 0x04) != 0) ? WS2812_DUTYCYCLE_1 : WS2812_DUTYCYCLE_0;
160 WS2812_TIM_BUF[pos++] = ((led.red & 0x02) != 0) ? WS2812_DUTYCYCLE_1 : WS2812_DUTYCYCLE_0;
161 WS2812_TIM_BUF[pos++] = ((led.red & 0x01) != 0) ? WS2812_DUTYCYCLE_1 : WS2812_DUTYCYCLE_0;
162
163 // Col:Blue , Bit:7..0
164 WS2812_TIM_BUF[pos++] = ((led.blue & 0x80) != 0) ? WS2812_DUTYCYCLE_1 : WS2812_DUTYCYCLE_0;
165 WS2812_TIM_BUF[pos++] = ((led.blue & 0x40) != 0) ? WS2812_DUTYCYCLE_1 : WS2812_DUTYCYCLE_0;
166 WS2812_TIM_BUF[pos++] = ((led.blue & 0x20) != 0) ? WS2812_DUTYCYCLE_1 : WS2812_DUTYCYCLE_0;
167 WS2812_TIM_BUF[pos++] = ((led.blue & 0x10) != 0) ? WS2812_DUTYCYCLE_1 : WS2812_DUTYCYCLE_0;
168 WS2812_TIM_BUF[pos++] = ((led.blue & 0x08) != 0) ? WS2812_DUTYCYCLE_1 : WS2812_DUTYCYCLE_0;
169 WS2812_TIM_BUF[pos++] = ((led.blue & 0x04) != 0) ? WS2812_DUTYCYCLE_1 : WS2812_DUTYCYCLE_0;
170 WS2812_TIM_BUF[pos++] = ((led.blue & 0x02) != 0) ? WS2812_DUTYCYCLE_1 : WS2812_DUTYCYCLE_0;
171 WS2812_TIM_BUF[pos++] = ((led.blue & 0x01) != 0) ? WS2812_DUTYCYCLE_1 : WS2812_DUTYCYCLE_0;
172 }
173}
WS2812_TIM_BUF
static uint32_t WS2812_TIM_BUF[WS2812_BUFLEN]
Definition WS2812.cpp:18
WS2812_BRIGHTNESS
static uint8_t WS2812_BRIGHTNESS
Definition WS2812.cpp:19
WS2812_LED_BUF
static WS2812_RGB_t WS2812_LED_BUF[WS2812_LED_N]
Definition WS2812.cpp:17
WS2812_RGB_t
Definition WS2812.h:10
WS2812_RGB_t::green
uint8_t green
Definition WS2812.h:12
WS2812_RGB_t::blue
uint8_t blue
Definition WS2812.h:13
WS2812_RGB_t::red
uint8_t red
Definition WS2812.h:11

Referenced by clearWS2812All(), clearWS2812One(), setWS2812All(), setWS2812Brightness(), and setWS2812One().

Here is the caller graph for this function:

◆ clearWS2812All()

void clearWS2812All ( )

Set all LEDs to 0 (off)

Definition at line 83 of file WS2812.cpp.

84{
85 for (uint16_t num = 0; num < WS2812_LED_N; num++)
86 {
87 WS2812_LED_BUF[num] = (WS2812_RGB_t){0, 0, 0};
88 }
89 calcBuf();
90}
calcBuf
void calcBuf()
Definition WS2812.cpp:130

Referenced by initWS2812().

Here is the call graph for this function:
Here is the caller graph for this function:

◆ clearWS2812One()

void clearWS2812One ( uint32_t  num)

Set one LEDs to 0 (off)

Definition at line 71 of file WS2812.cpp.

72{
73 if (num < WS2812_LED_N)
74 {
75 WS2812_LED_BUF[num] = (WS2812_RGB_t){0, 0, 0};
76 }
77 calcBuf();
78}

Referenced by initWS2812().

Here is the call graph for this function:
Here is the caller graph for this function:

◆ initWS2812()

void initWS2812 ( )

Definition at line 21 of file WS2812.cpp.

22{
23 palSetPadMode(WS2812_PORT, WS2812_PIN, PAL_MODE_ALTERNATE(1));
24
25 static const PWMConfig ws2812_pwm_config = {
26 .frequency = WS2812_PWM_FREQUENCY,
27 .period = WS2812_PWM_PERIOD,
28 .callback = NULL,
29 .channels = {
30 [0] = {.mode = WS2812_TIM_CH == 0 ? PWM_OUTPUT_ACTIVE_HIGH : PWM_OUTPUT_DISABLED, .callback = NULL}, // Turn on the channel we care about
31 [1] = {.mode = WS2812_TIM_CH == 1 ? PWM_OUTPUT_ACTIVE_HIGH : PWM_OUTPUT_DISABLED, .callback = NULL}, // Turn on the channel we care about
32 [2] = {.mode = WS2812_TIM_CH == 2 ? PWM_OUTPUT_ACTIVE_HIGH : PWM_OUTPUT_DISABLED, .callback = NULL}, // Turn on the channel we care about
33 [3] = {.mode = WS2812_TIM_CH == 3 ? PWM_OUTPUT_ACTIVE_HIGH : PWM_OUTPUT_DISABLED, .callback = NULL}, // Turn on the channel we care about
34 },
35 .cr2 = 0,
36 .bdtr = 0,
37 .dier = TIM_DIER_UDE, // DMA on update event for next period
38 };
39
40 const stm32_dma_stream_t *dma = dmaStreamAlloc(WS2812_DMA_STREAM, 10, NULL, NULL);
41 dmaStreamSetPeripheral(dma, &(WS2812_PWM_DRIVER.tim->CCR[WS2812_TIM_CH]));
42 dmaStreamSetMemory0(dma, WS2812_TIM_BUF);
43 dmaStreamSetTransactionSize(dma, WS2812_BUFLEN);
44 // M2P: Memory 2 Periph; PL: Priority Level
45 dmaStreamSetMode(dma,
46 STM32_DMA_CR_CHSEL(WS2812_DMA_CHANNEL) |
47 STM32_DMA_CR_DIR_M2P |
48 STM32_DMA_CR_PSIZE_WORD |
49 STM32_DMA_CR_MSIZE_WORD |
50 STM32_DMA_CR_MINC |
51 STM32_DMA_CR_CIRC |
52 STM32_DMA_CR_PL(3)
53 );
54
55 dmaStreamEnable(dma);
56 pwmStart(&PWMD1, &ws2812_pwm_config);
57 pwmEnableChannel(&PWMD1, WS2812_TIM_CH, 0); //(WS2812_PWM_PERIOD/2)
58
59 //TEST FUNCTIONALITY
60 clearWS2812All();
61 setWS2812Brightness(5);
62 setWS2812All((WS2812_RGB_t){255, 0, 0});
63 setWS2812One(19,(WS2812_RGB_t){0, 0, 255});
64 setWS2812One(9,(WS2812_RGB_t){0, 255, 0});
65 clearWS2812One(18);
66}
clearWS2812All
void clearWS2812All()
Definition WS2812.cpp:83
setWS2812One
void setWS2812One(uint32_t num, WS2812_RGB_t rgb_col)
Definition WS2812.cpp:95
clearWS2812One
void clearWS2812One(uint32_t num)
Definition WS2812.cpp:71
setWS2812Brightness
void setWS2812Brightness(uint8_t num)
Definition WS2812.cpp:119
setWS2812All
void setWS2812All(WS2812_RGB_t rgb_col)
Definition WS2812.cpp:107
PWMD1
PWMDriver PWMD1
PWMD1 driver identifier.
Definition hal_pwm_lld.c:49
PWMConfig
Type of a PWM driver configuration structure.
Definition hal_pwm_lld.h:158
PWMConfig::frequency
uint32_t frequency
Timer clock in Hz.
Definition hal_pwm_lld.h:164

Referenced by initHardware().

Here is the call graph for this function:
Here is the caller graph for this function:

◆ setWS2812All()

void setWS2812All ( WS2812_RGB_t  rgb_col)

Set all LEDs (R, G, B values).

Definition at line 107 of file WS2812.cpp.

108{
109 for (uint16_t num = 0; num < WS2812_LED_N; num++)
110 {
111 WS2812_LED_BUF[num] = rgb_col;
112 }
113 calcBuf();
114}

Referenced by initWS2812().

Here is the call graph for this function:
Here is the caller graph for this function:

◆ setWS2812Brightness()

void setWS2812Brightness ( uint8_t  num)

Set all LEDs Brightness.

Definition at line 119 of file WS2812.cpp.

120{
121 num = num >= 100 ? 100 : num;
122 num = num <= 0 ? 0 : num;
123 WS2812_BRIGHTNESS = num;
124 calcBuf();
125}

Referenced by initWS2812().

Here is the call graph for this function:
Here is the caller graph for this function:

◆ setWS2812One()

void setWS2812One ( uint32_t  num,
WS2812_RGB_t  rgb_col 
)

Set one LED (R, G, B values).

Definition at line 95 of file WS2812.cpp.

96{
97 if (num < WS2812_LED_N)
98 {
99 WS2812_LED_BUF[num] = rgb_col;
100 }
101 calcBuf();
102}

Referenced by initWS2812().

Here is the call graph for this function:
Here is the caller graph for this function:

Variable Documentation

◆ WS2812_BRIGHTNESS

uint8_t WS2812_BRIGHTNESS = 20
static

Definition at line 19 of file WS2812.cpp.

Referenced by calcBuf(), and setWS2812Brightness().

◆ WS2812_LED_BUF

WS2812_RGB_t WS2812_LED_BUF[WS2812_LED_N]
static

Definition at line 17 of file WS2812.cpp.

Referenced by calcBuf(), clearWS2812All(), clearWS2812One(), setWS2812All(), and setWS2812One().

◆ WS2812_TIM_BUF

uint32_t WS2812_TIM_BUF[WS2812_BUFLEN]
static

Definition at line 18 of file WS2812.cpp.

Referenced by calcBuf(), and initWS2812().

Go to the source code of this file.

Generated on Sat Sep 27 2025 00:10:09 for rusEFI by doxygen 1.9.8