基于辉芒微FT61E0AX 芯片驱动 WS2812B

基于辉芒微FT61E0AX 芯片驱动 WS2812B

一、硬件

1. MCU 型号 FT61E0A5-TRB
2. 驱动 WS2812B 端口 PA0 (MOSI)
3. 驱动 方式 硬件 SPI

二、软件

1. 时钟
   主时钟为16MHz

2. 端口配置
   端口 PA0 配置为输出 0

3. SPI 配置

void spi_init(void)
{
     AFP1   &= ~0B00000110;
     AFP1   |= 0B00000100;
     PCKEN  |=0B00010000;				//使能SPI模块时钟
     
     SPICTRL= 0B00000000;			   //NSS引脚禁用
     SPICFG = 0B01000000;			   //设置为主机模式，第一个时钟转换的沿是数据采样点，SPI空闲时，SCK的时钟是处于低电平状态
     SPISCR = 0B00000000;			   //波特率设置为8M，Fmaster/(2*(SPISCR+1))，Fmaster=16M
     SPIRXCRC = 0B00000000;			   //接收数据的CRC计算结果
     SPITXCRC = 0B00000000;			   //发送数据的CRC计算结果
     SPIIER = 0B00000000;				   //禁止所有中断 
     SPICTRL2 = 0B11000000;			   //全双工允许发送和接收，禁用CRC校验模块，高比特位优先发送
     SPISTAT = 0B00000000;				
     SPIEN=1;						//启用SPI
}

3. led_2812.h

#ifndef _LED_2812_H_
#define _LED_2812_H_

#define LED_SUM       30    //灯珠数量

void led_2812_init(void);
u8* get_led_buf_addr(void);
u8 get_led_index(void);
void set_led_index(u8 index);
void led_2812_send_dat(void);
void set_one_light_dat(u8*addr,u8 r, u8 b, u8 g);

#endif

4. led_2812.c

#include "../drv.h"

#define LED_BIT_0     0xC0   
#define LED_BIT_1     0xF0

u8 led_buf[24*LED_SUM];    // 24bit * LED_SUM  数据缓冲区 一个字节控制一个 WS 2812 的数据位
u8 led_index;              // 数据缓冲区索引 用于做特效

//初始化 
void led_2812_init(void)
{
    u16 i;
    u8* p;
    
    led_index = 0;
    p = led_buf; 
    for(i = 0; i < 24*LED_SUM; i++)
    {
        *p++ = LED_BIT_0;
    }
}

//对外接口获取缓冲区地址
u8* get_led_buf_addr(void)  
{
    return led_buf;
}

//获取缓冲区下标
u8 get_led_index(void)
{
    return led_index;
}
//修改缓冲区下标
void set_led_index(u8 index)
{
    led_index = index;
}

/**************************************************************************************************************
 * @brief 将缓存区数据一次发送出去
 * 输出波形不是完全符合2812的要求，似乎 WS2812B 只对 高电平的时间比较敏感，低电平的时间要求不高
**************************************************************************************************************/
void led_2812_send_dat(void)
{
    u8 i;
    u8 count;
    u8 *p;
    if(0 != GetTimer(T_LED))  //软件定时器，20ms 刷新一次
    {
        return;
    }
    SetTimer(T_LED, MS_20);

    p = led_buf;
    for(count = 0; count < LED_SUM; count++)
    {
        for(i = 0; i < 24; i++)
        {
            SPIDATA = *p++;
            //不需要等待 SPI 完成，因为 CPU 运算时就已经耗费很多时间了
        }
    }
}
/**************************************************************************************************************
 * @brief 设置一个灯的数据
 * 
 * @param addr 灯的数据起始 缓存地址 
 * @param r   红色值
 * @param b   蓝色值
 * @param g   绿色值
**************************************************************************************************************/
void set_one_light_dat(u8*addr,u8 r, u8 b, u8 g)
{
    u8 i;
    u8 color;
    u8 temp;
    for(color = 0; color < 3; color++)
    {
        switch(color)
        {
            case 0: temp = b; break;
            case 1: temp = r; break;
            case 2: temp = g; break;
        }
        for (i = 0; i < 8; i++)
        {
            if (temp & 0x80)
            {
                *addr++ = LED_BIT_1;
            }
            else
            {
                *addr++ = LED_BIT_0;
            }
            temp <<= 1;
        }
    }
}