设为首页收藏本站
开启辅助访问 切换到宽版

 找回密码
 立即注册

QQ登录

只需一步，快速开始

热门： 51单片机 | 24小时必答区 | 单片机教程 | 单片机DIY制作 | STM32 | Cortex M3 | 模数电子 | 电子DIY制作 | 音响/功放 | 拆机乐园 | Arduino | 嵌入式OS | 程序设计

搜索
»论坛 › 嵌入式/单片机论坛 › STM32/8 › stm32 anh led RGB WS2812b部分程序
返回列表 发新帖
查看: 1351|回复: 0
收起左侧

stm32 anh led RGB WS2812b部分程序

[复制链接]
ID:963502 发表于 2021-9-14 11:03 | 显示全部楼层 |阅读模式
stm32 anh led WS2812b
  1. #include "ws2812.h"

  3. /* Variables -----------------------------------------------*/
  4. static uint8_t LEDbuffer[LED_BUFFER_SIZE];

  6. TIM_HandleTypeDef TimHandle;
  7. TIM_OC_InitTypeDef sConfig;
  8. GPIO_InitTypeDef GPIO_InitStruct;
  9. DMA_HandleTypeDef hdma_tim;

  11. /* Functions -----------------------------------------------*/

  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();

  27.         /* Enable GPIO Channel Clock */
  28.         TIMx_CHANNEL1_GPIO_CLK_ENABLE();

  30.         /* Enable DMA clock */
  31.         DMAx_CLK_ENABLE();

  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);

  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;

  51.         /* Set hdma_tim instance */
  52.         hdma_tim.Instance = TIMx_CC1_DMA_INST;

  54.         /* Link hdma_tim to hdma[TIM_DMA_ID_CC1] (channel1) */
  55.         __HAL_LINKDMA(htim, hdma[TIM_DMA_ID_CC1], hdma_tim);

  57.         /* Initialize TIMx DMA handle */
  58.         HAL_DMA_Init(htim->hdma[TIM_DMA_ID_CC1]);

  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. }

  66. void ws2812_init(void) {
  67.         fillBufferBlack();

  69.         TimHandle.Instance = TIMx;

  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);

  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);

  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. }

  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. }

  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;

  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;

  111.         for (i = 0; i < 24; i++)
  112.                 LEDbuffer[RESET_SLOTS_BEGIN + LEDindex * 24 + i] = tempBuffer[i];
  113. }

  115. void setWHOLEcolor(uint8_t RED, uint8_t GREEN, uint8_t BLUE) {
  116.         uint32_t index;

  118.         for (index = 0; index < LED_NUMBER; index++)
  119.                 setLEDcolor(index, RED, GREEN, BLUE);
  120. }

  122. void fillBufferBlack(void) {
  123.         /*Fill LED buffer - ALL OFF*/
  124.         uint32_t index, buffIndex;
  125.         buffIndex = 0;

  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. }

  143. void fillBufferWhite(void) {
  144.         /*Fill LED buffer - ALL OFF*/
  145.         uint32_t index, buffIndex;
  146.         buffIndex = 0;

  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. }

  164. void TIMx_DMA_IRQHandler(void) {
  165.         HAL_DMA_IRQHandler(TimHandle.hdma[TIM_DMA_ID_CC1]);
  166. }
复制代码
51hei.png
WS2812.zip (25.38 KB, 下载次数: 23)

评分

参与人数 1黑币 +50 收起 理由
admin + 50 共享资料的黑币奖励！

查看全部评分

回复

使用道具 举报

返回列表 发新帖
高级模式
B Color Image Link Quote Code Smilies
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

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

Powered by 单片机教程网

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