STM32

stm32 驱动ws2812

2020-01-07 19:22发布

站内问答 / STM32/STM8
7453 37 37
最近正在用stm32做一个台灯，使用到了ws2812 LED,到网上找ws2812的stm32驱动，找半天找到一个，结果还是用不了。所以决定还是自己动手来吧。
下面的驱动代码贴出来，时候大家要使用也不错的
ws2812.h

  2. #ifndef __WS2812_H
  3. #define __WS2812_H

  5. #ifdef __cplusplus
  6. extern "C"
  7. {
  8. #endif

  10. #include "stm32f10x_conf.h"

  12. // The following macro must be ported to  your platform
  13. // 以下的宏定义请移植到你的IO Pin
  14. #define   ws2812DIN_HIGH()     (GPIOA->BSRR = 0x00001000)
  15. #define   ws2812DIN_LOW()      (GPIOA->BSRR = 0x10000000)
  16. #define   ws2812DIN_INIT()     {
  17.                                                                 RCC->APB2ENR |= 0x01<<2;
  18.                                                                 GPIOA->CRH &= ~(0x000f0000);
  19.                                                                 GPIOA->CRH |= (0x00030000);
  20.                                                                 GPIOA->BSRR |= (0x10000000);
  21.                                                                 }

  23. // RGB structure
  24. typedef struct
  25. {
  26.         // R,G,B color value
  27.     uint8_t ucRedVal;
  28.     uint8_t ucGreenVal;
  29.     uint8_t ucBlueVal;
  30. }RGB_t;
  31. // when you want to use ws2812 , you must call this function
  32. void ws2812_Init(void);
  33. void ws2812_SendRes(void);
  34. void ws2812_SendRGBData(RGB_t xRGB);


  37. /*** typical usage for above API， 典型的使用方法
  38. 　ws2812_Init();
  39.     GRB_t xRGB;
  40.     xRGB.ucRedVal = 0xff;
  41.     xRGB.ucGreenVal = 0x00;
  42.     xRGB.ucBlueVal = 0x00;
  43.     while(1)
  44.     {
  45.             // The first ws2812
  46.             ws2812_SendRGBData(xRGB);
  47.             // The second ws2812
  48.             ws2812_SendRGBData(xRGB);
  49.             // The third ws2812
  50.             ws2812_SendRGBData(xRGB);
  51.             //  The fourth ws2812
  52.             ws2812_SendRGBData(xRGB);
  53.             // send frame seperator
  54.             ws2812_SendRes();


  57.     }

  59. */

  61. #ifdef __cplusplus
  62. }
  63. #endif
  64. #endif
复制代码

ws2812.c 文件


  3. #include "ws2812.h"

  5. /*
  6. * @Desc: delay a approximate us
  7. * @Args: ulUs, us about to delay
  8. * @Returns: None
  9. */
  10. static void
  11. DelayUs(uint32_t ulUs)
  12. {
  13.         uint32_t j;
  14.         while(ulUs--)
  15.         {
  16.                 j = 12;
  17.                 while(j)
  18.                 {
  19.                         j--;
  20.                 }
  21.         }
  22. }


  25. /*
  26. * @Desc: delay a number of nop for your platform(one nop equal to 13.9 ns)
  27. * @Args: ulNopNum, nop number
  28. * @Returns: None
  29. *
  30. */
  31. static void
  32. DelayNop(uint32_t ulNopNum)
  33. {
  34.         while(ulNopNum)
  35.         {
  36.                 __NOP();
  37.                 ulNopNum--;
  38.         }
  39. }


  42. /*
  43. * @Desc: init ws2812, just init io Pin,
  44. *   you must call this function firstly whenever
  45. *   you want to use ws28128
  46. * @Args: None
  47. * @Returns: None
  48. *
  49. */
  50. void
  51. ws2812_Init(void)
  52. {
  53.         ws2812DIN_INIT();
  54. }


  57. /*
  58. * @Desc: send rgb value to a ws2812,
  59. * @Args: xRGB, rgb value container variable
  60. * @Returns: NOne
  61. *
  62. */
  63. void
  64. ws2812_SendRGBData(RGB_t xRGB)
  65. {
  66.         uint32_t i = 0;
  67.         uint32_t ulColor = 0;
  68.         // put blue color  bit to ulColor
  69.         for(i = 0; i < 8; i++)
  70.         {
  71.                 ulColor = ((xRGB.ucBlueVal & 0x01) | (ulColor << 1));
  72.                 xRGB.ucBlueVal >>= 1;
  73.         }
  74.         // put red color  bit to ulColor
  75.         for(i = 8; i < 16; i++)
  76.         {
  77.                 ulColor = ((xRGB.ucRedVal & 0x01) | (ulColor << 1));
  78.                 xRGB.ucRedVal >>= 1;
  79.         }
  80.         // put green color bit to ulColor
  81.         for(i = 16; i < 24; i++)
  82.         {
  83.                 ulColor = ((xRGB.ucGreenVal & 0x01) | (ulColor << 1));
  84.                 xRGB.ucGreenVal >>= 1;
  85.         }
  86.         // send 24 bits color data
  87.         for(i = 0; i < 24; i++)
  88.         {
  89.                 if((ulColor & (uint32_t)0x01))
  90.                 {
  91.                         ws2812DIN_HIGH();
  92.                         DelayNop(20);
  93.                         ws2812DIN_LOW();
  94.                 }
  95.                 else
  96.                 {
  97.                         ws2812DIN_HIGH();
  98.                         ws2812DIN_LOW();
  99.                         DelayNop(20);
  100.                 }
  101.                 ulColor >>= 1;
  102.         }
  103. }

  105. /*
  106. * @Desc: send　a frame seperator
  107. * @Args: None
  108. * @Returns: None
  109. */
  110. void
  111. ws2812_SendRes(void)
  112. {
  113.     ws2812DIN_LOW();
  114.     DelayUs(100);
  115. }

复制代码
友情提示: 此问题已得到解决，问题已经关闭,关闭后问题禁止继续编辑，回答。
37条回答
helloforworld
2020-01-07 23:11
仔细想了一下把，又把驱动重新的写了一下，(主要把数据结构改了，把ws812_SendRGBData() 换成了ws2812_SendColorData() ),具体地如下:
ws2812.h文件

  2. /**
  3. * Copyright ©2017 Thogo Tech. All Rights Reserved.
  4. *
  5. *@description: This file contains data structure and API
  6.      definition for RGB LED ws2812.it's sensible for time
  7. *@author: infinite.ft
  8. *@create_at: 2017/03/09
  9. *@update_at: 2017/03/12
  10. *@version: 0.0.1
  11. *
  12. */
  13. #ifndef __WS2812_H
  14. #define __WS2812_H

  16. #ifdef __cplusplus
  17. extern "C"
  18. {
  19. #endif

  21. #include "stm32f10x_conf.h"

  23. // The following macro must be ported to  your platform
  24. #define   ws2812DIN_HIGH()     (GPIOA->BSRR = 0x00001000)
  25. #define   ws2812DIN_LOW()      (GPIOA->BSRR = 0x10000000)
  26. #define   ws2812DIN_INIT()     {
  27.                                                                 RCC->APB2ENR |= 0x01<<2;
  28.                                                                 GPIOA->CRH &= ~(0x000f0000);
  29.                                                                 GPIOA->CRH |= (0x00030000);
  30.                                                                 GPIOA->BSRR |= (0x10000000);
  31.                                                                 }
  32. // RGB color type structure
  33. typedef union
  34. {
  35.         struct {
  36.                 uint8_t ucPadding;
  37.                 uint8_t ucBlue;
  38.                 uint8_t ucRed;
  39.                 uint8_t ucGreen;
  40.         }xRGB;
  41.         uint32_t ulBits;
  42. }Color_t;

  44. // when you want to use ws2812 , you must call this function
  45. void ws2812_Init(void);
  46. void ws2812_SendRes(void);
  47. void ws2812_SendColorData(Color_t xColor);

  49. /*** typical usage for above API
  50.     Color_t xColor;
  51.     xColor.xRGB.ucRed = 0xff;
  52.     xColor.xRGB.ucGreen = 0x00;
  53.     xColor.xRGB.ucBlue = 0x00;
  54.     while(1)
  55.     {
  56.             // first ws2812
  57.             ws2812_SendColorData(xColor);
  58.             // second ws2812
  59.             ws2812_SendColorData(xColor);
  60.             // third ws2812
  61.             ws2812_SendColorData(xColor);
  62.             // fourth ws2812
  63.             ws2812_SendColorData(xColor);
  64.             // send frame seperator
  65.             ws2812_SendRes();


  68.     }

  70. */

  72. #ifdef __cplusplus
  73. }
  74. #endif
  75. #endif

复制代码

ws2812.c 文件

  2. /**
  3. *
  4. * Copyright ©2017 Thogo Tech. All Rights Reserved.
  5. *
  6. * @description: this file contains functions that responsible
  7.      for performing operation for RGB LED ws2812
  8. * @author: infinite.ft
  9. * @create_at: 2017/03/09
  10. * @update_at: 2017/03/12
  11. * @version: 0.0.1
  12. *
  13. */

  15. #include "ws2812.h"

  17. /*
  18. * @Desc: delay a approximate us
  19. * @Args: ulUs, us about to delay
  20. * @Returns: None
  21. */
  22. static void
  23. DelayUs(uint32_t ulUs)
  24. {
  25.         uint32_t j;
  26.         while(ulUs--)
  27.         {
  28.                 j = 12;
  29.                 while(j)
  30.                 {
  31.                         j--;
  32.                 }
  33.         }
  34. }


  37. /*
  38. * @Desc: delay a number of nop for your platform(one nop equal to 13.9 ns)
  39. * @Args: ulNopNum, nop number
  40. * @Returns: None
  41. *
  42. */
  43. static void
  44. DelayNop(uint32_t ulNopNum)
  45. {
  46.         while(ulNopNum)
  47.         {
  48.                 __NOP();
  49.                 ulNopNum--;
  50.         }
  51. }


  54. /*
  55. * @Desc: init ws2812, just init io Pin,
  56. *   you must call this function firstly whenever
  57. *   you want to use ws28128
  58. * @Args: None
  59. * @Returns: None
  60. *
  61. */
  62. void
  63. ws2812_Init(void)
  64. {
  65.         ws2812DIN_INIT();
  66. }


  69. /*
  70. * @Desc: send RGB color value to a ws2812,
  71. * @Args: xColor, rgb color value  variable
  72. * @Returns: NOne
  73. *
  74. */
  75. void ws2812_SendColorData(Color_t xColor)
  76. {
  77.         uint32_t i = 0;
  78.         // send 24 bits color data
  79.         for(i = 0; i < 24; i++)
  80.         {
  81.                 if((xColor.ulBits & (uint32_t)(0x01 << (31-i))))
  82.                 {
  83.                         ws2812DIN_HIGH();
  84.                         DelayNop(20);
  85.                         ws2812DIN_LOW();
  86.                 }
  87.                 else
  88.                 {
  89.                         ws2812DIN_HIGH();
  90.                         ws2812DIN_LOW();
  91.                         DelayNop(20);
  92.                 }
  93.         }
  94. }
  95. /*
  96. * @Desc: send　a frame seperator
  97. * @Args: None
  98. * @Returns: None
  99. */
  100. void
  101. ws2812_SendRes(void)
  102. {
  103.     ws2812DIN_LOW();
  104.     DelayUs(100);
  105. }


复制代码

一周热门 更多>