标题: stm32 anh led RGB WS2812b部分程序 [打印本页]

---

作者: ngoluc    时间: 2021-9-14 11:03
标题: stm32 anh led RGB WS2812b部分程序
stm32 anh led WS2812b

1. #include "ws2812.h"
   
2. 
   
3. /* Variables -----------------------------------------------*/
   
4. static uint8_t LEDbuffer[LED_BUFFER_SIZE];
   
5. 
   
6. TIM_HandleTypeDef TimHandle;
   
7. TIM_OC_InitTypeDef sConfig;
   
8. GPIO_InitTypeDef GPIO_InitStruct;
   
9. DMA_HandleTypeDef hdma_tim;
   
10. 
    
11. /* Functions -----------------------------------------------*/
    
12. 
    
13. /**
    
14. * @brief TIM MSP Initialization
    
15. *        This function configures the hardware resources used in this example:
    
16. *           - Peripheral's clock enable
    
17. *           - Peripheral's GPIO Configuration
    
18. *           - DMA configuration for transmission request by peripheral
    
19. * @param htim: TIM handle pointer
    
20. * @retval None
    
21. */
    
22. void HAL_TIM_PWM_MspInit(TIM_HandleTypeDef *htim) {
    
23.         /*##-1- Enable peripherals and GPIO Clocks #################################*/
    
24.         /* TIMx clock enable */
    
25.         TIMx_CLK_ENABLE();
    
26. 
    
27.         /* Enable GPIO Channel Clock */
    
28.         TIMx_CHANNEL1_GPIO_CLK_ENABLE();
    
29. 
    
30.         /* Enable DMA clock */
    
31.         DMAx_CLK_ENABLE();
    
32. 
    
33.         /* Configure TIM1_Channel1 in output, push-pull & alternate function mode */
    
34.         GPIO_InitStruct.Pin = GPIO_PIN_CHANNEL1;
    
35.         GPIO_InitStruct.Mode = GPIO_MODE_AF_PP;
    
36.         GPIO_InitStruct.Pull = GPIO_PULLUP;
    
37.         GPIO_InitStruct.Speed = GPIO_SPEED_FREQ_HIGH;
    
38.         GPIO_InitStruct.Alternate = GPIO_AF_TIMx;
    
39.         HAL_GPIO_Init(TIMx_GPIO_CHANNEL1_PORT, &GPIO_InitStruct);
    
40. 
    
41.         /* Set the parameters to be configured */
    
42.         hdma_tim.Init.Request = TIMx_CC1_DMA_REQUEST;
    
43.         hdma_tim.Init.Direction = DMA_MEMORY_TO_PERIPH;
    
44.         hdma_tim.Init.PeriphInc = DMA_PINC_DISABLE;
    
45.         hdma_tim.Init.MemInc = DMA_MINC_ENABLE;
    
46.         hdma_tim.Init.PeriphDataAlignment = DMA_PDATAALIGN_WORD;
    
47.         hdma_tim.Init.MemDataAlignment = DMA_MDATAALIGN_BYTE;
    
48.         hdma_tim.Init.Mode = DMA_CIRCULAR;
    
49.         hdma_tim.Init.Priority = DMA_PRIORITY_HIGH;
    
50. 
    
51.         /* Set hdma_tim instance */
    
52.         hdma_tim.Instance = TIMx_CC1_DMA_INST;
    
53. 
    
54.         /* Link hdma_tim to hdma[TIM_DMA_ID_CC1] (channel1) */
    
55.         __HAL_LINKDMA(htim, hdma[TIM_DMA_ID_CC1], hdma_tim);
    
56. 
    
57.         /* Initialize TIMx DMA handle */
    
58.         HAL_DMA_Init(htim->hdma[TIM_DMA_ID_CC1]);
    
59. 
    
60.         /*##-2- Configure the NVIC for DMA #########################################*/
    
61.         /* NVIC configuration for DMA transfer complete interrupt */
    
62.         HAL_NVIC_SetPriority(TIMx_DMA_IRQn, 0, 0);
    
63.         HAL_NVIC_EnableIRQ(TIMx_DMA_IRQn);
    
64. }
    
65. 
    
66. void ws2812_init(void) {
    
67.         fillBufferBlack();
    
68. 
    
69.         TimHandle.Instance = TIMx;
    
70. 
    
71.         TimHandle.Init.Period = TIMER_PERIOD - 1;
    
72.         TimHandle.Init.RepetitionCounter = LED_BUFFER_SIZE + 1; //LED_BUFFER_SIZE + 1;//0xFFFF;
    
73.         TimHandle.Init.Prescaler = (uint32_t)(
    
74.                         (SystemCoreClock / TIMER_CLOCK_FREQ) - 1);
    
75.         TimHandle.Init.ClockDivision = TIM_CLOCKDIVISION_DIV1;
    
76.         TimHandle.Init.CounterMode = TIM_COUNTERMODE_UP;
    
77.         HAL_TIM_PWM_Init(&TimHandle);
    
78. 
    
79.         /*##-2- Configure the PWM channel 3 ########################################*/
    
80.         sConfig.OCMode = TIM_OCMODE_PWM1;
    
81.         sConfig.OCPolarity = TIM_OCPOLARITY_HIGH;
    
82.         sConfig.OCFastMode = TIM_OCFAST_DISABLE;
    
83.         sConfig.OCIdleState = TIM_OCIDLESTATE_RESET;
    
84.         //sConfig.Pulse        = 0;
    
85.         HAL_TIM_PWM_ConfigChannel(&TimHandle, &sConfig, TIM_CHANNEL_1);
    
86. 
    
87.         /*##-3- Start PWM signal generation in DMA mode ############################*/
    
88.         HAL_TIM_PWM_Start_DMA(&TimHandle, TIM_CHANNEL_1, (uint32_t *) LEDbuffer,
    
89.                         LED_BUFFER_SIZE);
    
90. }
    
91. 
    
92. void ws2812_update(void) {
    
93.         HAL_TIM_PWM_ConfigChannel(&TimHandle, &sConfig, TIM_CHANNEL_1);
    
94.         HAL_TIM_PWM_Start_DMA(&TimHandle, TIM_CHANNEL_1, (uint32_t *) LEDbuffer,
    
95.                         LED_BUFFER_SIZE);
    
96. }
    
97. 
    
98. void setLEDcolor(uint32_t LEDnumber, uint8_t RED, uint8_t GREEN, uint8_t BLUE) {
    
99.         uint8_t tempBuffer[24];
    
100.         uint32_t i;
     
101.         uint32_t LEDindex;
     
102.         LEDindex = LEDnumber % LED_NUMBER;
     
103. 
     
104.         for (i = 0; i < 8; i++) // GREEN data
     
105.                 tempBuffer[i] = ((GREEN << i) & 0x80) ? WS2812_1 : WS2812_0;
     
106.         for (i = 0; i < 8; i++) // RED
     
107.                 tempBuffer[8 + i] = ((RED << i) & 0x80) ? WS2812_1 : WS2812_0;
     
108.         for (i = 0; i < 8; i++) // BLUE
     
109.                 tempBuffer[16 + i] = ((BLUE << i) & 0x80) ? WS2812_1 : WS2812_0;
     
110. 
     
111.         for (i = 0; i < 24; i++)
     
112.                 LEDbuffer[RESET_SLOTS_BEGIN + LEDindex * 24 + i] = tempBuffer[i];
     
113. }
     
114. 
     
115. void setWHOLEcolor(uint8_t RED, uint8_t GREEN, uint8_t BLUE) {
     
116.         uint32_t index;
     
117. 
     
118.         for (index = 0; index < LED_NUMBER; index++)
     
119.                 setLEDcolor(index, RED, GREEN, BLUE);
     
120. }
     
121. 
     
122. void fillBufferBlack(void) {
     
123.         /*Fill LED buffer - ALL OFF*/
     
124.         uint32_t index, buffIndex;
     
125.         buffIndex = 0;
     
126. 
     
127.         for (index = 0; index < RESET_SLOTS_BEGIN; index++) {
     
128.                 LEDbuffer[buffIndex] = WS2812_RESET;
     
129.                 buffIndex++;
     
130.         }
     
131.         for (index = 0; index < LED_DATA_SIZE; index++) {
     
132.                 LEDbuffer[buffIndex] = WS2812_0;
     
133.                 buffIndex++;
     
134.         }
     
135.         LEDbuffer[buffIndex] = WS2812_0;
     
136.         buffIndex++;
     
137.         for (index = 0; index < RESET_SLOTS_END; index++) {
     
138.                 LEDbuffer[buffIndex] = 0;
     
139.                 buffIndex++;
     
140.         }
     
141. }
     
142. 
     
143. void fillBufferWhite(void) {
     
144.         /*Fill LED buffer - ALL OFF*/
     
145.         uint32_t index, buffIndex;
     
146.         buffIndex = 0;
     
147. 
     
148.         for (index = 0; index < RESET_SLOTS_BEGIN; index++) {
     
149.                 LEDbuffer[buffIndex] = WS2812_RESET;
     
150.                 buffIndex++;
     
151.         }
     
152.         for (index = 0; index < LED_DATA_SIZE; index++) {
     
153.                 LEDbuffer[buffIndex] = WS2812_1;
     
154.                 buffIndex++;
     
155.         }
     
156.         LEDbuffer[buffIndex] = WS2812_0;
     
157.         buffIndex++;
     
158.         for (index = 0; index < RESET_SLOTS_END; index++) {
     
159.                 LEDbuffer[buffIndex] = 0;
     
160.                 buffIndex++;
     
161.         }
     
162. }
     
163. 
     
164. void TIMx_DMA_IRQHandler(void) {
     
165.         HAL_DMA_IRQHandler(TimHandle.hdma[TIM_DMA_ID_CC1]);
     
166. }
     

复制代码

WS2812.zip (25.38 KB, 下载次数: 23)

2021-9-14 11:03 上传

点击文件名下载附件
下载积分: 黑币 -5

---

欢迎光临 (http://www.51hei.com/bbs/)

Powered by Discuz! X3.1