找回密码
 立即注册

QQ登录

只需一步,快速开始

搜索
查看: 5928|回复: 3
收起左侧

ws2812b芯片驱动程序

[复制链接]
ID:576951 发表于 2019-7-3 10:36 | 显示全部楼层 |阅读模式
ws2812b驱动程序

单片机源程序如下:
  1. #include "bsp_ws2812b.h"
  2. #include "bsp.h"


  3. #define TIM4CCR1ADDR        0x40000834
  4. #define TIM4CCR2ADDR        0x40000838

  5. volatile  uint8_t LEDbuffer[LED_BUFFER_SIZE];



  6. static GPIO_InitTypeDef tim4ch1pwm =
  7. {
  8.     .GPIO_Pin = GPIO_Pin_6,
  9.     .GPIO_Speed = GPIO_Speed_50MHz,
  10.     .GPIO_Mode = GPIO_Mode_AF_PP
  11. };
  12. static GPIO_InitTypeDef tim4ch2pwm =
  13. {
  14.     .GPIO_Pin = GPIO_Pin_7,
  15.     .GPIO_Speed = GPIO_Speed_50MHz,
  16.     .GPIO_Mode = GPIO_Mode_AF_PP
  17. };


  18. static DMA_InitTypeDef dmachan1 =
  19. {
  20.     .DMA_PeripheralBaseAddr = ( uint32_t )&TIM4->CCR1,
  21.     .DMA_MemoryBaseAddr = ( uint32_t ) ( &LEDbuffer[0] ),
  22.     .DMA_DIR = DMA_DIR_PeripheralDST,
  23.     .DMA_BufferSize = LED_BUFFER_SIZE + 1,
  24.     .DMA_PeripheralInc = DMA_PeripheralInc_Disable,
  25.     .DMA_MemoryInc = DMA_MemoryInc_Enable,
  26.     .DMA_PeripheralDataSize = DMA_PeripheralDataSize_HalfWord,
  27.     .DMA_MemoryDataSize = DMA_MemoryDataSize_Byte,
  28.     .DMA_Mode = DMA_Mode_Circular,
  29.     .DMA_Priority = DMA_Priority_High,
  30.     .DMA_M2M = DMA_M2M_Disable
  31. };

  32. static DMA_InitTypeDef dmachan4 =
  33. {
  34.     .DMA_PeripheralBaseAddr = ( uint32_t )&TIM4->CCR2,
  35.     .DMA_MemoryBaseAddr = ( uint32_t ) ( &LEDbuffer[0] ),
  36.     .DMA_DIR = DMA_DIR_PeripheralDST,
  37.     .DMA_BufferSize = LED_BUFFER_SIZE + 1,
  38.     .DMA_PeripheralInc = DMA_PeripheralInc_Disable,
  39.     .DMA_MemoryInc = DMA_MemoryInc_Enable,
  40.     .DMA_PeripheralDataSize = DMA_PeripheralDataSize_HalfWord,
  41.     .DMA_MemoryDataSize = DMA_MemoryDataSize_Byte,
  42.     .DMA_Mode = DMA_Mode_Circular,
  43.     .DMA_Priority = DMA_Priority_High,
  44.     .DMA_M2M = DMA_M2M_Disable
  45. };

  46. static TIM_TimeBaseInitTypeDef tim4 =
  47. {
  48.     .TIM_Period = TIMER_PERIOD - 1,
  49.     .TIM_Prescaler = 0,
  50.     .TIM_ClockDivision = TIM_CKD_DIV1,
  51.     .TIM_CounterMode = TIM_CounterMode_Up
  52. };
  53. static TIM_OCInitTypeDef timingmode =
  54. {
  55.     .TIM_OCMode = TIM_OCMode_PWM1,
  56.     .TIM_OutputState = TIM_OutputState_Enable,
  57.     .TIM_Pulse = 0,
  58.     .TIM_OCPolarity = TIM_OCPolarity_High
  59. };



  60. /*************************************************************
  61. * 函 数 名 : WS2812_Init
  62. * 函数描述 : 初始化
  63. * 返 回 值 : none
  64. * 输入参数 : none
  65. * 输出参数 : none

  66. *************************************************************/

  67. void WS2812_Init ()
  68. {

  69.     RCC_APB2PeriphClockCmd ( RCC_APB2Periph_GPIOB | RCC_APB2Periph_AFIO, ENABLE );
  70. //    RCC_APB2PeriphClockCmd ( RCC_APB2Periph_GPIOB, ENABLE );

  71.     RCC_APB1PeriphClockCmd ( RCC_APB1Periph_TIM4, ENABLE );
  72.     GPIO_Init ( GPIOB, &tim4ch1pwm );
  73.     GPIO_Init ( GPIOB, &tim4ch2pwm );

  74.     /* 初始化时基 */
  75.     TIM_TimeBaseInit ( TIM4, &tim4 );

  76.     /* 初始化channel1 */
  77.     TIM_OC1PreloadConfig ( TIM4, TIM_OCPreload_Enable );
  78.     TIM_OC1Init ( TIM4, &timingmode );

  79.     /* 初始化channel2 */
  80.     TIM_OC2PreloadConfig ( TIM4, TIM_OCPreload_Enable );
  81.     TIM_OC2Init ( TIM4, &timingmode );

  82. //    /* 初始化DMA1 */
  83.     RCC_AHBPeriphClockCmd ( RCC_AHBPeriph_DMA1, ENABLE );

  84.     /* 初始化DMA1通道 */
  85.     DMA_Init ( DMA1_Channel1, &dmachan1 );
  86.     DMA_Init ( DMA1_Channel4, &dmachan4 );

  87.     TIM_DMACmd ( TIM4, TIM_DMA_CC1, ENABLE );
  88.     TIM_DMACmd ( TIM4, TIM_DMA_CC2, ENABLE );

  89.     /* 使能DMA中断 */
  90.     DMA_ITConfig ( DMA1_Channel1, DMA_IT_TC | DMA_IT_TE, ENABLE );
  91.     DMA_ITConfig ( DMA1_Channel4, DMA_IT_TC | DMA_IT_TE, ENABLE );


  92. //    TIM_Cmd ( TIM4, ENABLE );

  93. }

  94. /*************************************************************
  95. * 函 数 名 : WS2812_update
  96. * 函数描述 : DMA发送数据
  97. * 输入参数 : none
  98. * 输出参数 : none
  99. * 返 回 值 : none
  100. *************************************************************/

  101. void WS2812_update ( uint8_t group )
  102. {

  103.     TIM_Cmd ( TIM4, ENABLE );

  104.     if ( group == GROUP_A )                                             // GROUP_A只有1盏LED灯
  105.     {


  106.         DMA_SetCurrDataCounter ( DMA1_Channel1, LED_BUFFER_SIZE );      

  107.         DMA_Cmd ( DMA1_Channel1, ENABLE );                              

  108.         TIM_DMACmd ( TIM4, TIM_DMA_Update, ENABLE );                    // 使能TIM4 DMA请求
  109.     }
  110.     else
  111.     {


  112.         DMA_SetCurrDataCounter ( DMA1_Channel4, LED_BUFFER_SIZE );      

  113.         DMA_Cmd ( DMA1_Channel4, ENABLE );                              

  114.         TIM_DMACmd ( TIM4, TIM_DMA_Update, ENABLE );                    // 使能TIM4 DMA请求

  115.     }
  116. }

  117. /*************************************************************
  118. * 函 数 名 : SetLedColor
  119. * 函数描述 : 设置某一盏LED颜色
  120. * 输入参数 : index  需要点亮的LED序号(第一盏、第二盏.....)
  121. * 输出参数 : none
  122. * 返 回 值 : none
  123. *************************************************************/

  124. void SetLedColor ( uint8_t index, uint8_t RED, uint8_t GREEN, uint8_t BLUE )
  125. {
  126.     uint8_t tempBuffer[24] = {0};
  127.     uint8_t i;

  128.     for ( i = 0; i < 8; i++ ) // GREEN data
  129.         tempBuffer[i] = ( ( GREEN << i ) & 0x80 ) ? WS2812_1 : WS2812_0;

  130.     for ( i = 0; i < 8; i++ ) // RED
  131.         tempBuffer[8 + i] = ( ( RED << i ) & 0x80 ) ? WS2812_1 : WS2812_0;

  132.     for ( i = 0; i < 8; i++ ) // BLUE
  133.         tempBuffer[16 + i] = ( ( BLUE << i ) & 0x80 ) ? WS2812_1 : WS2812_0;

  134.     for ( i = 0; i < 24; i++ )
  135.         LEDbuffer[RESET_SLOTS_BEGIN + index * 24 + i] = tempBuffer[i];



  136. }

  137. void SetWholeColor ( uint8_t group, uint8_t red, uint8_t green, uint8_t blue )
  138. {
  139.     uint32_t index;

  140.     if ( group == GROUP_A ) // GROUP_A只有一盏LED灯
  141.     {
  142. //        SetLedColor ( 0, RED, GREEN, BLUE );

  143.         for ( index = 0; index < LED_NUMBER; index++ )
  144.             SetLedColor ( index, red, green, blue );

  145.     }
  146.     else
  147.     {
  148.         for ( index = 0; index < LED_NUMBER; index++ )
  149.             SetLedColor ( index, red, green, blue );

  150.     }
  151. }




  152. /****************************************************************
  153. * 函 数 名:DMA1_Channel1_IRQHandler
  154. * 函数描述:channel1中断处理
  155. * 输入参数:none
  156. * 输出参数:none
  157. * 返回类型:void
  158. *****************************************************************/

  159. void DMA1_Channel1_IRQHandler()
  160. {
  161.     OSIntEnter();

  162.     TIM_Cmd ( TIM4, DISABLE );

  163.     TIM_DMACmd ( TIM4, TIM_DMA_Update, DISABLE );           // 禁止请求
  164.     DMA_Cmd ( DMA1_Channel1, DISABLE );                     
  165.     DMA_ClearFlag ( DMA1_FLAG_TC1 );
  166.     memset ( ( char * ) LEDbuffer, 0, LED_BUFFER_SIZE );
  167. ……………………

  168. …………限于本文篇幅 余下代码请从51黑下载附件…………
复制代码

所有资料51hei提供下载:
ws2812b.rar (3.5 KB, 下载次数: 99)
回复

使用道具 举报

ID:59830 发表于 2020-5-6 17:22 | 显示全部楼层
手上一块 8个WS2812彩灯,没有资料不知道怎么驱动,想找一下资料。参考一下.谢谢你的分享
回复

使用道具 举报

ID:877820 发表于 2021-1-15 14:55 | 显示全部楼层
参考一下
回复

使用道具 举报

ID:861657 发表于 2021-1-27 10:03 | 显示全部楼层
16个WS2812也可以驱动吗
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

手机版|小黑屋|51黑电子论坛 |51黑电子论坛6群 QQ 管理员QQ:125739409;技术交流QQ群281945664

Powered by 单片机教程网

快速回复 返回顶部 返回列表