STM32G431RBT6 CubeMX 驱动WS2812教程

MCU:STM32G431RBT6

功能介绍

界面

界面1：控制灯条总开关、灯条亮度
界面2：控制灯条模式、灯条流动时间

按键

B1:切换界面键。
B2:
界面1：灯条总开关键。
界面2：灯条模式键。
B3:
界面1：灯条流动时间加键。
界面2：灯条亮度加键。
B3:
界面1：灯条流动时间减键。
界面2：灯条亮度减键。

注：所有按键均支持长按

模式

0-红绿蓝交替变化 1-灯条随机变化
2-每颗LED随机变化 3-亮红 4-亮绿 5-亮蓝

CubeMX配置

时钟配置

PWM配置

随机函数RNG配置

其他IO配置

WS2812驱动程序

ws2812.c

#include "main.h"
 
uint8_t WS2812_buf[WS2812_NUMS+1][WS2812_BIT]={0};

/*
函数功能：计算出24位RGB数据
形参：
	R:R的值
	G:R的值
	B:B的值
返回值：24位RGB数据
*/
u32 WS2812_RGB(u8 R,u8 G,u8 B)
{
	return (u32)(G<<16|R<<8|B);	//根据实际RGB顺序修改，这里的顺序为GRB
}

/*
函数功能：打开/关闭灯带
形参：
	mode: 0-关闭 1-RGB最大值
返回值：无
*/
void WS2812_offoron(u8 mode)
{
	u8 a,b;
	u8 data;
	
	if(mode) data=WS2812_ONE;
	else     data=WS2812_ZERO;

	for(a=0;a<WS2812_NUMS;a++)
	{
		for(b=0;b<WS2812_BIT;b++)
		{
			WS2812_buf[a][b]=data;
		}
	}
}

/*
函数功能：修改单个LED数据
形参：
	led: 要修改的LED位置
	R:R的值
	G:R的值
	B:B的值
返回值：无
*/
void WS2812_writr_bit(u8 led,u8 R,u8 G,u8 B)
{
	u32 data;
	u8 b;
	
	data=WS2812_RGB(R,G,B);
	
	for(b=0;b<WS2812_BIT;b++)
	{
		WS2812_buf[led][b]=(data<<b)&0x800000?WS2812_ONE:WS2812_ZERO;
	}
}	

/*
函数功能：修改整条灯带LED数据
形参：
	R:R的值
	G:R的值
	B:B的值
返回值：无
*/
void WS2812_writr_all(u8 R,u8 G,u8 B)
{
	u32 data;
	u8 a,b;
	
	data=WS2812_RGB(R,G,B);
	
	for(a=0;a<WS2812_NUMS;a++)
	{
		for(b=0;b<WS2812_BIT;b++)
		{
			WS2812_buf[a][b]=(data<<b)&0x800000?WS2812_ONE:WS2812_ZERO;
		}
	}
}

/*
函数功能：刷新灯带显示
形参：无
返回值：无
*/
void WS2812_display(void)
{
	HAL_TIM_PWM_Stop_DMA(&htim2,TIM_CHANNEL_2);
	HAL_TIM_PWM_Start_DMA(&htim2,TIM_CHANNEL_2,(uint32_t *)WS2812_buf[0],WS2812_DATA_LEN);
}

/*
函数功能：灯带初始化
形参：无
返回值：无
*/
void WS2812_init(void)
{	
	WS2812_offoron(0);
	WS2812_display();
	HAL_Delay(10);
}

/*
函数功能：灯带功能函数
形参：无
返回值：无
*/
void WS2812_function(void)
{
	u32 data[1];
	static u8 R,G,B;
	static u8 state=0,led=0;
	
	if(!TaskTime.WS2812)
	{
		TaskTime.WS2812=WS2812_TIME;
		
		if(!WS2812_ALL_OFF_ON)
		{
			WS2812_offoron(0);
			WS2812_display();
			return;
		}
		
		if(WS2812_MODE==0)
		{
			if(state==0) 	  WS2812_writr_bit(led,WS2812_brightness,0,0);
			else if(state==1) WS2812_writr_bit(led,0,WS2812_brightness,0);
			else if(state==2) WS2812_writr_bit(led,0,0,WS2812_brightness);
			if(++led==WS2812_NUMS)
			{
				led=0;
				if(++state==3) state=0;
			}	  
		}
		else if(WS2812_MODE==1)
		{
									
			WS2812_writr_bit(led,R,G,B);

			if(++led==WS2812_NUMS)
			{
				HAL_RNG_GenerateRandomNumber(&hrng,data);
				R=data[0]%WS2812_brightness;
				HAL_RNG_GenerateRandomNumber(&hrng,data);
				G=data[0]%WS2812_brightness;
				HAL_RNG_GenerateRandomNumber(&hrng,data);
				B=data[0]%WS2812_brightness;				
				led=0;
				if(++state==3) state=0;
			}		
		}	
		else if(WS2812_MODE==2)
		{
			HAL_RNG_GenerateRandomNumber(&hrng,data);
			R=data[0]%WS2812_brightness;
			HAL_RNG_GenerateRandomNumber(&hrng,data);
			G=data[0]%WS2812_brightness;
			HAL_RNG_GenerateRandomNumber(&hrng,data);
			B=data[0]%WS2812_brightness;			
						
			WS2812_writr_bit(led,R,G,B);

			if(++led==WS2812_NUMS)
			{
				led=0;
			}			
		}	
		else if(WS2812_MODE==3) WS2812_writr_all(WS2812_brightness,0,0);
		else if(WS2812_MODE==4) WS2812_writr_all(0,WS2812_brightness,0);
		else if(WS2812_MODE==5) WS2812_writr_all(0,0,WS2812_brightness);
		WS2812_display();	
	}
}
	

ws2812.h

#ifndef __WS2812_
#define __WS2812_
 
#define WS2812_ONE     66	//1 码 		(0.0125*66=0.825us)
#define WS2812_ZERO    33	//0 码 		(0.0125*33=0.4125us)
#define WS2812_NUMS    30	//led数量	(根据实际数量修改)
#define WS2812_BIT     24	//24bit		(每颗灯有24位数据，R:8 G:8 B:8)
#define WS2812_DATA_LEN  ((WS2812_NUMS+1)*WS2812_BIT) 	//数组长度

void WS2812_offoron(unsigned char mode);
void WS2812_writr_all(unsigned char R,unsigned char G,unsigned char B);
void WS2812_writr_bit(unsigned char led,unsigned char R,unsigned char G,unsigned char B);
void WS2812_init(void);
void WS2812_display(void);
void WS2812_function(void);

#endif

效果展示

基于STM32G431RBT6 HAL库驱动WS2812

注：若LED数量过多，电脑USB供电可能不足，需充电宝或其他电源供电。

程序下载

链接：https://pan.baidu.com/s/11YaIc71mewoaz65Ggsf2lg
提取码：bnn2