stm32按键控制ws2812灯带，使其产生白色光和黄色光

首先，你需要连接好STM32和WS2812灯带，并确保可以正常控制灯带。然后，你需要编写STM32的代码来控制WS2812灯带，以下是一个简单的示例代码：

#include "stm32f10x.h"

#define LED_NUM 10 // 灯珠数量
#define LED_BYTE_NUM (LED_NUM * 3) // 灯带字节数

// 初始化WS2812灯带
void init_ws2812(void)
{
    // 确保GPIO时钟已开启
    RCC_APB2PeriphClockCmd(RCC_APB2Periph_GPIOA, ENABLE);
    
    // 配置GPIO为推挽输出模式
    GPIO_InitTypeDef GPIO_InitStructure;
    GPIO_InitStructure.GPIO_Pin = GPIO_Pin_5;
    GPIO_InitStructure.GPIO_Mode = GPIO_Mode_Out_PP;
    GPIO_InitStructure.GPIO_Speed = GPIO_Speed_50MHz;
    GPIO_Init(GPIOA, &GPIO_InitStructure);
}

// 发送一个24位的颜色数据到WS2812灯带
void send_color(uint8_t r, uint8_t g, uint8_t b)
{
    for (int i = 0; i < 8; i++)
    {
        if (r & 0x80)
            GPIO_SetBits(GPIOA, GPIO_Pin_5);
        else
            GPIO_ResetBits(GPIOA, GPIO_Pin_5);
        r <<= 1;
        
        if (g & 0x80)
            GPIO_SetBits(GPIOA, GPIO_Pin_5);
        else
            GPIO_ResetBits(GPIOA, GPIO_Pin_5);
        g <<= 1;
        
        if (b & 0x80)
            GPIO_SetBits(GPIOA, GPIO_Pin_5);
        else
            GPIO_ResetBits(GPIOA, GPIO_Pin_5);
        b <<= 1;
        
        // 发送一个 bit0，保持高电平 0.4us ~ 0.6us
        GPIO_SetBits(GPIOA, GPIO_Pin_5);
        __NOP(); __NOP(); __NOP(); __NOP(); __NOP(); __NOP(); __NOP(); __NOP(); __NOP(); __NOP();
        __NOP(); __NOP(); __NOP(); __NOP(); __NOP(); __NOP(); __NOP(); __NOP(); __NOP(); __NOP();
        GPIO_ResetBits(GPIOA, GPIO_Pin_5);
    }
}

// 发送所有灯珠的颜色数据到WS2812灯带
void send_led_data(uint8_t* data)
{
    for (int i = 0; i < LED_BYTE_NUM; i += 3)
        send_color(data[i], data[i + 1], data[i + 2]);
}

int main(void)
{
    uint8_t led_data[LED_BYTE_NUM] = {0}; // 所有灯珠的颜色数据
    
    init_ws2812(); // 初始化WS2812灯带
    
    while (1)
    {
        // 判断按键是否按下
        if (GPIO_ReadInputDataBit(GPIOA, GPIO_Pin_0) == Bit_RESET)
        {
            // 产生白色光
            for (int i = 0; i < LED_BYTE_NUM; i += 3)
            {
                led_data[i] = 0xff; // 红色通道
                led_data[i + 1] = 0xff; // 绿色通道
                led_data[i + 2] = 0xff; // 蓝色通道
            }
            send_led_data(led_data); // 发送所有灯珠的颜色数据到WS2812灯带
        }
        else if (GPIO_ReadInputDataBit(GPIOA, GPIO_Pin_1) == Bit_RESET)
        {
            // 产生黄色光
            for (int i = 0; i < LED_BYTE_NUM; i += 3)
            {
                led_data[i] = 0xff; // 红色通道
                led_data[i + 1] = 0xff; // 绿色通道
                led_data[i + 2] = 0x00; // 蓝色通道
            }
            send_led_data(led_data); // 发送所有灯珠的颜色数据到WS2812灯带
        }
    }
}

上述代码中，我们使用了PA5作为控制WS2812灯带的数据线，PA0和PA1作为按键输入。当按下PA0时，会产生白色光；当按下PA1时，会产生黄色光。

在发送颜色数据到WS2812灯带时，需要按照一定的时间序列发送24个bit，其中每个bit都需要保持高电平0.4us~0.6us，然后再根据bit值发送相应的低电平时间。这里我们使用了一个简单的位移操作来获取每个颜色通道的值，并使用了内嵌汇编指令NOP来实现一定的延时。当然，这种方法只是一个简单的示例，实际应用中可能需要更加精细的控制时间序列。