90 lines
2.1 KiB
C
90 lines
2.1 KiB
C
|
//ws2812 by zyx
|
|||
|
|
#include "RGB.h"
|
|||
|
|
#include "main.h"
|
|||
|
|
#include "spi.h"
|
|||
|
|
#include "bsp_spi.h"
|
|||
|
|
|
|||
|
|
//灯条显存SPI数据缓存
|
|||
|
|
uint8_t gWs2812bDat_SPI[WS2812B_AMOUNT * 24] = {0};
|
|||
|
|
//灯条显存
|
|||
|
|
tWs2812bCache_TypeDef gWs2812bDat[WS2812B_AMOUNT] = {
|
|||
|
|
|
|||
|
|
//R G B
|
|||
|
|
0XFF, 0X00, 0X00, //0
|
|||
|
|
0X00, 0XFF, 0X00, //1
|
|||
|
|
0X00, 0X00, 0XFF, //2
|
|||
|
|
0X00, 0XFF, 0XFF, //3
|
|||
|
|
0XFF, 0X00, 0XFF, //4
|
|||
|
|
0XFF, 0XFF, 0X00, //5
|
|||
|
|
0XFF, 0XFF, 0XFF, //6
|
|||
|
|
0X00, 0X00, 0X00, //7
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
void WS2812b_Set(uint16_t Ws2b812b_NUM, uint8_t r,uint8_t g,uint8_t b)
|
|||
|
|
{
|
|||
|
|
uint8_t *pR = &gWs2812bDat_SPI[(Ws2b812b_NUM) * 24 + 8];
|
|||
|
|
uint8_t *pG = &gWs2812bDat_SPI[(Ws2b812b_NUM) * 24];
|
|||
|
|
uint8_t *pB = &gWs2812bDat_SPI[(Ws2b812b_NUM) * 24 + 16];
|
|||
|
|
|
|||
|
|
for(uint8_t i = 0; i < 8; i++) {
|
|||
|
|
if(g & 0x80) {
|
|||
|
|
*pG = CODE_1;
|
|||
|
|
}
|
|||
|
|
else {
|
|||
|
|
*pG = CODE_0;
|
|||
|
|
}
|
|||
|
|
if(r & 0x80) {
|
|||
|
|
*pR = CODE_1;
|
|||
|
|
}
|
|||
|
|
else {
|
|||
|
|
*pR = CODE_0;
|
|||
|
|
}
|
|||
|
|
if(b & 0x80) {
|
|||
|
|
*pB = CODE_1;
|
|||
|
|
}
|
|||
|
|
else {
|
|||
|
|
*pB = CODE_0;
|
|||
|
|
}
|
|||
|
|
r <<= 1;
|
|||
|
|
g <<= 1;
|
|||
|
|
b <<= 1;
|
|||
|
|
pR++;
|
|||
|
|
pG++;
|
|||
|
|
pB++;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
void WS2812B_Task(void)
|
|||
|
|
{
|
|||
|
|
uint8_t dat = 0;
|
|||
|
|
|
|||
|
|
//将gWs2812bDat数据解析成SPI数据
|
|||
|
|
for(uint8_t iLED = 0; iLED < WS2812B_AMOUNT; iLED++)
|
|||
|
|
{
|
|||
|
|
WS2812b_Set(iLED, gWs2812bDat[iLED].R, gWs2812bDat[iLED].G, gWs2812bDat[iLED].B);
|
|||
|
|
}
|
|||
|
|
//总线输出数据
|
|||
|
|
HAL_SPI_Transmit(&hspi1, gWs2812bDat_SPI, sizeof(gWs2812bDat_SPI),0XFFFF);
|
|||
|
|
//使总线输出低电平
|
|||
|
|
HAL_SPI_Transmit(&hspi1, &dat, 1,0XFFFF);
|
|||
|
|
//帧信号：一个大于50us的低电平
|
|||
|
|
HAL_Delay(1);
|
|||
|
|
}
|
|||
|
|
void move_Front()
|
|||
|
|
{
|
|||
|
|
uint8_t i;
|
|||
|
|
uint8_t temp[3];
|
|||
|
|
temp[0] = gWs2812bDat[0].R;
|
|||
|
|
temp[1] = gWs2812bDat[0].G;
|
|||
|
|
temp[2] = gWs2812bDat[0].B;
|
|||
|
|
for (i = 0; i < WS2812B_AMOUNT-1; i++)
|
|||
|
|
{
|
|||
|
|
gWs2812bDat[i].R = gWs2812bDat[i+1].R;
|
|||
|
|
gWs2812bDat[i].G = gWs2812bDat[i+1].G;
|
|||
|
|
gWs2812bDat[i].B = gWs2812bDat[i+1].B;
|
|||
|
|
}
|
|||
|
|
gWs2812bDat[7].R = temp[0];
|
|||
|
|
gWs2812bDat[7].G = temp[1];
|
|||
|
|
gWs2812bDat[7].B = temp[2];
|
|||
|
|
}
|
|||
|
|
|