STM32F1使用PWM+DMA驱动WS2812

泰骏兴电子

使用定时器3的PWM驱动
使用WS2812_send函数可以直接驱动LED

1. .h文件
   
2. 
   
3. 
   
4. #define TIMING_ONE  50                        //搞电平时间
   
5. #define TIMING_ZERO 25                        //低电平时间
   
6. #define W2812_SIZE        29                        //LED个数
   
7. 
   
8. extern uint16_t LED_BUF[10+24*W2812_SIZE];//LED数据缓存
   
9. 
   
10. 
    
11. 
    
12. 
    
13. void TIME_3_init(void);
    
14. void WS2812_send(uint8_t (*color)[3], uint16_t len);
    
15. void WS2812_Set_color_bit(u8 r,u8 g,u8 b,u8 lan);//设置单个LED颜色
    
16. 
    
17. 
    
18. void WS2812_update(void);//更新显示
    
19. void WS2812_OFF(void);
    
20. void WS2812_ALL(u8 r,u8 g,u8 b);

复制代码

1. .c文件
   
2. 
   
3. uint16_t LED_BUF[10+24*W2812_SIZE];//LED数据缓存
   
4. 
   
5. 
   
6. #define DMA_CH_Channe        DMA1_Channel3
   
7. #define DMA_CH_Flag                   DMA1_FLAG_TC3
   
8. #define TIME_CH                           TIM3
   
9. #define TIME_CCR                CCR3
   
10. 
    
11. 
    
12. void TIME_3_init(void)
    
13. {
    
14.                         TIM_TimeBaseInitTypeDef  TIM_TimeBaseStructure;
    
15.                         TIM_OCInitTypeDef TIM_OCInitStructure;
    
16.                         GPIO_InitTypeDef GPIO_InitStructure;
    
17.                         DMA_InitTypeDef DMA_InitStructure;
    
18. 
    
19.                         RCC_APB2PeriphClockCmd(RCC_APB2Periph_GPIOB, ENABLE);
    
20.                         GPIO_InitStructure.GPIO_Pin = GPIO_Pin_0;
    
21.                         GPIO_InitStructure.GPIO_Mode = GPIO_Mode_AF_PP;
    
22.                         GPIO_InitStructure.GPIO_Speed = GPIO_Speed_50MHz;
    
23.                         GPIO_Init(GPIOB, &GPIO_InitStructure);
    
24. 
    
25.                         RCC_APB1PeriphClockCmd(RCC_APB1Periph_TIM3, ENABLE);
    
26.                         TIM_TimeBaseStructure.TIM_Period = 90-1;
    
27.                         TIM_TimeBaseStructure.TIM_Prescaler = 0;
    
28.                         TIM_TimeBaseStructure.TIM_ClockDivision = 0;
    
29.                         TIM_TimeBaseStructure.TIM_CounterMode = TIM_CounterMode_Up;
    
30.                         TIM_TimeBaseInit(TIME_CH, &TIM_TimeBaseStructure);
    
31. 
    
32.                         TIM_OCInitStructure.TIM_OCMode = TIM_OCMode_PWM1;
    
33.                         TIM_OCInitStructure.TIM_OutputState = TIM_OutputState_Enable;
    
34.                         TIM_OCInitStructure.TIM_Pulse = 45;
    
35.                         TIM_OCInitStructure.TIM_OCPolarity = TIM_OCPolarity_High;
    
36.                         TIM_OC3Init(TIME_CH, &TIM_OCInitStructure);
    
37. 
    
38.                         RCC_AHBPeriphClockCmd(RCC_AHBPeriph_DMA1, ENABLE);
    
39. 
    
40. 
    
41. 
    
42.                         DMA_DeInit(DMA_CH_Channe);
    
43. 
    
44.                         DMA_InitStructure.DMA_PeripheralBaseAddr = (uint32_t)(&TIME_CH->TIME_CCR);
    
45.                         DMA_InitStructure.DMA_MemoryBaseAddr = (uint32_t)LED_BUF;
    
46.                         DMA_InitStructure.DMA_DIR = DMA_DIR_PeripheralDST;
    
47.                         DMA_InitStructure.DMA_BufferSize = 42;
    
48.                         DMA_InitStructure.DMA_PeripheralInc = DMA_PeripheralInc_Disable;
    
49.                         DMA_InitStructure.DMA_MemoryInc = DMA_MemoryInc_Enable;
    
50.                         DMA_InitStructure.DMA_PeripheralDataSize = DMA_PeripheralDataSize_HalfWord;
    
51.                         DMA_InitStructure.DMA_MemoryDataSize = DMA_MemoryDataSize_HalfWord;
    
52.                         DMA_InitStructure.DMA_Mode = DMA_Mode_Normal;
    
53.                         DMA_InitStructure.DMA_Priority = DMA_Priority_High;
    
54.                         DMA_InitStructure.DMA_M2M = DMA_M2M_Disable;
    
55. 
    
56.                         DMA_Init(DMA_CH_Channe, &DMA_InitStructure);
    
57. 
    
58.                         TIM_DMACmd(TIME_CH, TIM_DMA_Update, ENABLE);
    
59. }
    
60. 
    
61. void WS2812_send(uint8_t (*color)[3], uint16_t len)
    
62. {
    
63.                         uint8_t i;
    
64.                         uint16_t memaddr;
    
65.                         uint16_t buffersize;
    
66.                         buffersize = (len*24)+43;
    
67.                         memaddr = 0;
    
68. 
    
69.                         while (len)
    
70.         {   
    
71.                         for(i=0; i<8; i++)
    
72.                 {
    
73.                         LED_BUF[memaddr] = ((color[0][1]<<i) & 0x0080) ? TIMING_ONE : TIMING_ZERO;
    
74.                         memaddr++;
    
75.                 }
    
76.                         for(i=0; i<8; i++)
    
77.                 {
    
78.                         LED_BUF[memaddr] = ((color[0][0]<<i) & 0x0080) ? TIMING_ONE : TIMING_ZERO;
    
79.                         memaddr++;
    
80.                 }
    
81.                         for(i=0; i<8; i++)
    
82.                 {
    
83.                         LED_BUF[memaddr] = ((color[0][2]<<i) & 0x0080) ? TIMING_ONE : TIMING_ZERO;
    
84.                         memaddr++;
    
85.                 }
    
86.                         len--;
    
87.         }
    
88. 
    
89. 
    
90. 
    
91.                         LED_BUF[memaddr] = ((color[0][2]<<8) & 0x0080) ? TIMING_ONE : TIMING_ZERO;
    
92.                         memaddr++;  
    
93. 
    
94.                         LED_BUF[memaddr] = ((color[0][2]<<8) & 0x0080) ? TIMING_ONE : TIMING_ZERO;
    
95.                         memaddr++;
    
96. 
    
97.                         while(memaddr < buffersize)
    
98.         {
    
99.                         LED_BUF[memaddr] = 0;
    
100.                         memaddr++;
     
101.         }
     
102. 
     
103. 
     
104. 
     
105. 
     
106.                         DMA_SetCurrDataCounter(DMA_CH_Channe, buffersize);
     
107.                         DMA_Cmd(DMA_CH_Channe, ENABLE);
     
108.                         TIM_Cmd(TIME_CH, ENABLE);
     
109.                         while(!DMA_GetFlagStatus(DMA_CH_Flag)) ;  
     
110.                         TIM_Cmd(TIME_CH, DISABLE);  
     
111.                         DMA_Cmd(DMA_CH_Channe, DISABLE);
     
112.                         DMA_ClearFlag(DMA_CH_Flag);
     
113. }
     
114. void WS2812_Set_color_bit(u8 r,u8 g,u8 b,u8 lan)//设置单个LED颜色
     
115. {
     
116.                         uint8_t i;
     
117.                         
     
118.                         for(i=0; i<8; i++)
     
119.                 {
     
120.                         LED_BUF[lan*24+i] = ((g<<i) & 0x0080) ? TIMING_ONE : TIMING_ZERO;
     
121.                         LED_BUF[lan*24+i+8] = ((r<<i) & 0x0080) ? TIMING_ONE : TIMING_ZERO;
     
122.                         LED_BUF[lan*24+i+16] = ((b<<i) & 0x0080) ? TIMING_ONE : TIMING_ZERO;
     
123.                 }
     
124.                         LED_BUF[W2812_SIZE*24+1]=0;
     
125.                         LED_BUF[W2812_SIZE*24+2]=0;
     
126. }
     
127. 
     
128. void WS2812_update()//更新显示
     
129. {
     
130.                         DMA_SetCurrDataCounter(DMA_CH_Channe, W2812_SIZE*24);
     
131.                         DMA_Cmd(DMA_CH_Channe, ENABLE);
     
132.                         TIM_Cmd(TIME_CH, ENABLE);
     
133.                         while(!DMA_GetFlagStatus(DMA_CH_Flag)) ;  
     
134.                         TIM_Cmd(TIME_CH, DISABLE);  
     
135.                         DMA_Cmd(DMA_CH_Channe, DISABLE);
     
136.                         DMA_ClearFlag(DMA_CH_Flag);
     
137. }
     
138. 
     
139. 
     
140. 
     
141. 
     
142. void WS2812_OFF()
     
143. {
     
144.                         u16 q;
     
145.                         for(q=0;q<W2812_SIZE;q++)
     
146.         {
     
147.                         WS2812_Set_color_bit(0,0,0,q);
     
148.         }
     
149.                         WS2812_update();
     
150. }
     
151. void WS2812_ALL(u8 r,u8 g,u8 b)
     
152. {
     
153.                         u16 q;
     
154.                         for(q=0;q<W2812_SIZE;q++)
     
155.         {
     
156.                         WS2812_Set_color_bit(r,g,b,q);
     
157.         }
     
158.                         WS2812_update();
     
159. }
     
160. 
     
161. 
     
162. 
     

复制代码