ws2812b_spi.c 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100
  1. #include "ws2812b_spi.h"
  2. #include "user_ws2812b.h"
  3. /*******************************************************************************
  4. * @函数名称 user_spi_dma_init
  5. * @函数说明 SPI DMA初始化
  6. * @输入参数 txData:数据指针
  7. txSize:数据长度
  8. * @输出参数 无
  9. * @返回参数 无
  10. *******************************************************************************/
  11. void Ws2812b_spi_dma_init(uint8_t *txData, uint32_t txSize)
  12. {
  13. DMA_InitTypeDef DMA_InitStructure;
  14. NVIC_InitTypeDef NVIC_InitStructure;
  15. DMA_DeInit(USER_WS2812B_DMA_CHANNEL); // Channel 3 for SPI1_TX
  16. DMA_InitStructure.DMA_PeripheralBaseAddr = (uint32_t)&USER_WS2812B_SPI->DATAR;
  17. DMA_InitStructure.DMA_MemoryBaseAddr = (uint32_t)txData;
  18. DMA_InitStructure.DMA_DIR = DMA_DIR_PeripheralDST;
  19. DMA_InitStructure.DMA_BufferSize = txSize;
  20. DMA_InitStructure.DMA_PeripheralInc = DMA_PeripheralInc_Disable;
  21. DMA_InitStructure.DMA_MemoryInc = DMA_MemoryInc_Enable;
  22. DMA_InitStructure.DMA_PeripheralDataSize = DMA_PeripheralDataSize_Byte;
  23. DMA_InitStructure.DMA_MemoryDataSize = DMA_MemoryDataSize_Byte;
  24. DMA_InitStructure.DMA_Mode = DMA_Mode_Normal;
  25. DMA_InitStructure.DMA_Priority = DMA_Priority_High;
  26. DMA_InitStructure.DMA_M2M = DMA_M2M_Disable;
  27. NVIC_InitStructure.NVIC_IRQChannelPreemptionPriority = 1;
  28. NVIC_InitStructure.NVIC_IRQChannelSubPriority = 1;
  29. NVIC_InitStructure.NVIC_IRQChannelCmd = ENABLE;
  30. DMA_Init(USER_WS2812B_DMA_CHANNEL, &DMA_InitStructure);
  31. // 启用 DMA 传输完成中断
  32. DMA_ITConfig(USER_WS2812B_DMA_CHANNEL, DMA_IT_TC, ENABLE);
  33. NVIC_InitStructure.NVIC_IRQChannel = USER_WS2812B_DMA_IRQn;
  34. NVIC_Init(&NVIC_InitStructure);
  35. }
  36. /*******************************************************************************
  37. * @函数名称 SPI_FullDuplex_Init
  38. * @函数说明 SPI初始化
  39. * @输入参数 无
  40. * @输出参数 无
  41. * @返回参数 无
  42. *******************************************************************************/
  43. void ws2812b_spi_Init(void)
  44. {
  45. GPIO_InitTypeDef GPIO_InitStructure = {0};
  46. SPI_InitTypeDef SPI_InitStructure = {0};
  47. USER_WS2812B_GPIO_RCC_ENABLE;
  48. USER_WS2812B_SPI_RCC_ENABLE;
  49. USER_WS2812B_DMA_RCC_ENABLE;
  50. GPIO_InitStructure.GPIO_Pin = USER_WS2812B_SPI_MOSI_PIN;
  51. GPIO_InitStructure.GPIO_Mode = GPIO_Mode_AF_PP;
  52. GPIO_InitStructure.GPIO_Speed = GPIO_Speed_50MHz;
  53. GPIO_Init(USER_WS2812B_SPI_MOSI_GPIO, &GPIO_InitStructure);
  54. SPI_InitStructure.SPI_Direction = SPI_Direction_1Line_Tx;
  55. SPI_InitStructure.SPI_Mode = SPI_Mode_Master;
  56. SPI_InitStructure.SPI_DataSize = SPI_DataSize_8b;
  57. SPI_InitStructure.SPI_CPOL = SPI_CPOL_Low;
  58. SPI_InitStructure.SPI_CPHA = SPI_CPHA_2Edge;
  59. SPI_InitStructure.SPI_NSS = SPI_NSS_Soft;
  60. SPI_InitStructure.SPI_FirstBit = SPI_FirstBit_MSB;
  61. SPI_InitStructure.SPI_BaudRatePrescaler = SPI_BaudRatePrescaler_8;
  62. SPI_InitStructure.SPI_CRCPolynomial = 7;
  63. SPI_Init(USER_WS2812B_SPI, &SPI_InitStructure);
  64. SPI_CalculateCRC(USER_WS2812B_SPI, DISABLE);
  65. SPI_Cmd(USER_WS2812B_SPI, ENABLE);
  66. }
  67. /*******************************************************************************
  68. * Function Name : bUser_spi_WriteByte
  69. * Description : SPI写字节
  70. * Input : unsigned char TxData 写入字节
  71. * Output : None
  72. * Return : 结果
  73. Flash_ReadWriteByte
  74. *******************************************************************************/
  75. bool bWs2812b_spi_WriteByte(volatile uint8_t *TxData, uint32_t TxSize)
  76. {
  77. // 清除传输完成标志
  78. vUser_ws2812b_set_tx_stat(true);
  79. // 配置 DMA
  80. DMA_SetCurrDataCounter(USER_WS2812B_DMA_CHANNEL, TxSize);
  81. // 启用 SPI DMA 请求
  82. SPI_I2S_DMACmd(USER_WS2812B_SPI, SPI_I2S_DMAReq_Tx, ENABLE);
  83. // 启用 DMA 通道
  84. DMA_Cmd(USER_WS2812B_DMA_CHANNEL, ENABLE);
  85. return true;
  86. }