基于单片机的WS2812和SK6812通用驱动程序

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

---

前言

在现代的照明系统中，WS2812和SK6812 LED灯带因其丰富的色彩和灵活的控制而广泛应用。本文将介绍如何使用STM32微控制器来控制这些LED灯带, 我们将使用STM32微控制器的GPIO引脚来控制WS2812或SK6812 LED灯带，通过精确的时序控制来设置每个LED的颜色和亮度。

一、代码概述

1.LED缓冲区和初始化

Flow_led_buffer用于存储LED的控制数据。Flow_led_init函数初始化用于控制LED的GPIO引脚。GPIO引脚被配置为输出模式，并启用了时钟。

uint8_t Flow_led_buffer[Flow_led_num * led_data_lenght] = {0};

void Flow_led_init(void)
{
    RCC_AHBPeriphClockCmd(RCC_AHBPeriph_GPIOA, ENABLE);

    GPIO_InitTypeDef GPIO_InitStructure;
    GPIO_InitStructure.GPIO_Pin = GPIO_Pin_1;
    GPIO_InitStructure.GPIO_Mode = GPIO_Mode_OUT;
    GPIO_InitStructure.GPIO_Speed = GPIO_Speed_10MHz;
    GPIO_Init(GPIOA, &GPIO_InitStructure);
}

2.发送LED数据

Flow_led_send函数将缓冲区中的数据发送到LED灯带。通过GPIO引脚精确控制LED的颜色和亮度。

void Flow_led_send(void) {
    for (int i = 0; i < sizeof(Flow_led_buffer); i++) {
        if (Flow_led_buffer[i]) {
            GPIOA->BSRR = BSRR_VAL;
            __NOP();
            __NOP();
            __NOP();
            __NOP();
            __NOP();
            __NOP();
            GPIOA->BRR = BSRR_VAL;
            __NOP();
        } else {
            GPIOA->BSRR = BSRR_VAL;
            __NOP();
            __NOP();
            GPIOA->BRR = BSRR_VAL;
            __NOP();
            __NOP();
            __NOP();
            __NOP();
            __NOP();
        }
    }
}

3.设置WS2812 LED亮度

ws2812_set_index_brightness函数用于设置WS2812 LED的亮度。通过RGB颜色值调整指定通道的LED亮度。该函数将颜色值转换为适合LED控制的数据格式，并存储在Flow_led_buffer中。

void ws2812_set_index_brightness(uint32_t RGB, uint16_t channelIndex) {
    static uint8_t green = 0, red = 0, blue = 0; 
    red  	= RGB >> 16;
    green = RGB >> 8;
    blue 	=	RGB;
    uint8_t* ptr = &Flow_led_buffer[channelIndex * led_data_lenght];
	
    for (uint8_t j = 0; j < 3; j++) {
        for (uint8_t i = 0; i < 8; i++) {
            if (j == 0) {
                ptr[i] = (green & (1 << (7 - i))) ? 1 : 0;
            }
            if (j == 1) {
                ptr[i + 8]  = (red & (1 << (7 - i))) ? 1 : 0;
            }
            if (j == 2) {
                ptr[i + 16] = (blue & (1 << (7 - i))) ? 1 : 0;
            }
        }		
    }
}

4.设置SK6812 LED亮度

sk6812_set_index_brightness函数用于设置SK6812 LED的亮度，支持RGBW模式。通过RGBW颜色值调整指定通道的LED亮度。

void sk6812_set_index_brightness(uint32_t RGBW, uint16_t channelIndex) {
    static uint8_t green = 0, red = 0, blue = 0, white = 0; 
    uint8_t* ptr = &Flow_led_buffer[channelIndex * led_data_lenght];
    red  	= RGBW >> 24;
    green = RGBW >> 16;
    blue 	=	RGBW >> 8;
    white = RGBW;
    
    for (uint8_t j = 0; j < 4; j++) {
        for (uint8_t i = 0; i < 8; i++) {
            if (j == 0) {
                ptr[i] = (green & (1 << (7 - i))) ? 1 : 0;
            }
            if (j == 1) {
                ptr[i + 8] = (red & (1 << (7 - i))) ? 1 : 0;
            }
            if (j == 2) {
                ptr[i + 16] = (blue & (1 << (7 - i))) ? 1 : 0;
            }
            if (j == 3) {
                ptr[i + 24] = (white & (1 << (7 - i))) ? 1 : 0;
            }
        }		
    }
}

5.初始化LED状态

/**
 * @brief 初始化所有WS2812 LED的亮度
 *
 * 此函数初始化所有WS2812 LED通道，将每个LED的亮度设置为零（关闭LED），并将初始状态发送到LED灯带。
 *
 * @note 此函数遍历所有LED通道，将每个通道的RGB颜色值设置为0，然后通过 `Flow_led_send` 函数将状态更新到LED灯带上。
 */
void ws2812_Init_Led_Bit(void)
{
    for (int i = 0; i < Flow_led_num; i++) {
        uint8_t* ptr = &Flow_led_buffer[i * led_data_lenght];

        for (int i = 0; i < 8; i++) {
            ptr[i]      = (0 & (1 << (7 - i))) ? 1 : 0;
            ptr[i + 8]  = (0 & (1 << (7 - i))) ? 1 : 0;
            ptr[i + 16] = (0 & (1 << (7 - i))) ? 1 : 0;
        }
    }
    Flow_led_send(); 
}

/**
 * @brief 初始化所有SK6812 LED的亮度
 *
 * 此函数初始化所有SK6812 LED通道，将每个LED的亮度设置为零（关闭LED），并将初始状态发送到LED灯带。
 *
 * @note 此函数遍历所有LED通道，将每个通道的RGBW颜色值设置为0（关闭所有颜色通道），
 *       然后通过 `Flow_led_send` 函数将状态更新到LED灯带上。
 */
void sk6812_Init_Led_Bit(void)
{
    for (int i = 0; i < Flow_led_num; i++) {
        uint8_t* ptr = &Flow_led_buffer[i * led_data_lenght];
        for (int i = 0; i < 8; i++) {
            ptr[i]      = (0 & (1 << (7 - i))) ? 1 : 0;
            ptr[i + 8]  = (0 & (1 << (7 - i))) ? 1 : 0;
            ptr[i + 16] = (0 & (1 << (7 - i))) ? 1 : 0;
            ptr[i + 24] = (0 & (1 << (7 - i))) ? 1 : 0;
        }
    }
    Flow_led_send(); 
}

总结

本文介绍了如何使用STM32微控制器来控制WS2812和SK6812 LED灯带。通过精确的GPIO时序控制，我们可以实现丰富的灯光效果。代码展示了如何初始化GPIO、发送LED数据以及设置LED亮度。

希望这个实现能够为你的项目提供帮助，如果有任何问题或改进建议，欢迎在评论区留言！

驱动下载地址：WS2812和SK6812通用驱动程序