ESP32 RMT 驱动 WS2812B：精确计算与消除时钟抖动导致的色彩偏差

背景与问题

WS2812B 采用单线归零码通信，每个 bit 由高电平和低电平的持续时间决定（T0H、T0L、T1H、T1L）。ESP32 的 RMT（Remote Control）外设专为红外遥控设计，但因其灵活的脉冲生成能力，常被用于驱动 WS2812B。然而，RMT 的时钟源（APB 时钟）默认 80 MHz，且存在分频和舍入误差，导致实际脉冲宽度与理想值偏差，进而引起色彩失真（尤其是白色和低亮度时）。

RMT 时钟与抖动根源

RMT 的时序由 clk_div（分频系数）和 duration（计数周期数）决定。实际脉冲宽度 = (duration / (APB_CLK / clk_div))。问题在于：

  • APB_CLK 可能不是精确的 80 MHz（受 PLL 影响）。
  • clk_div 和 duration 均为整数，无法精确表示任意时间。
  • 每个 bit 的误差累积，导致数据帧整体偏移。

例如，理想 T0H=350ns，T0L=800ns，T1H=700ns，T1L=600ns。若 clk_div=1，则每个计数周期为 12.5ns，350ns 对应 28 个周期，但 28*12.5=350ns 精确；但若 APB_CLK 实际为 79.9MHz，则周期为 12.516ns，28 周期为 350.45ns，误差 0.45ns，看似微小，但连续 24 bit 后误差达 10ns，可能超出 WS2812B 的容忍范围（±150ns）。

精确计算策略

1. 测量实际 APB 时钟

使用 esp_clk_apb_freq() 获取实际频率，而非假设 80MHz。

2. 选择最优分频系数

clk_div 应使计数周期接近 1ns 的整数倍，以减少舍入误差。例如，若 APB=80MHz，clk_div=2 则周期 25ns，但 WS2812B 时序要求 ns 级精度，25ns 粒度太粗。因此推荐 clk_div=1，周期 12.5ns，误差最小。

3. 计算 duration 并取整

对每个时序参数，计算 duration = round(time_ns * APB_CLK / clk_div / 1000)。注意使用四舍五入而非截断。

4. 补偿累积误差

由于每个 bit 的 duration 取整，误差累积。可在帧间隙（RESET 信号）中调整，或在每字节后插入补偿脉冲。但更简单的方法是：确保总帧时间（24 bit + RESET）在 WS2812B 允许范围内（>280us）。

配置步骤

  1. 初始化 RMT 驱动：使用 rmt_config_t 结构体，设置 clk_div 为 1，mem_block_num 为 1。
  2. 定义时序结构：创建 rmt_item32_t 数组，每个 bit 对应一个 item。
  3. 计算 duration：根据实际 APB 频率计算。
  4. 发送数据：使用 rmt_write_items() 发送，并等待完成。

完整代码示例

以下代码演示如何精确驱动 WS2812B，并消除抖动。

#include "driver/rmt.h"
#include "esp_clk.h"

#define RMT_TX_CHANNEL RMT_CHANNEL_0
#define GPIO_NUM 18
#define LED_NUM 10

// 理想时序（ns）
#define T0H_NS 350
#define T0L_NS 800
#define T1H_NS 700
#define T1L_NS 600

static uint32_t apb_clk_hz;
static uint32_t tick_ns; // 每个计数周期对应的 ns
static rmt_item32_t bit_0, bit_1;

void rmt_init_ws2812b() {
    apb_clk_hz = esp_clk_apb_freq();
    tick_ns = 1000000000 / (apb_clk_hz / 1); // clk_div=1
    
    // 计算 duration（四舍五入）
    bit_0.duration0 = (T0H_NS + tick_ns/2) / tick_ns;
    bit_0.level0 = 1;
    bit_0.duration1 = (T0L_NS + tick_ns/2) / tick_ns;
    bit_0.level1 = 0;
    
    bit_1.duration0 = (T1H_NS + tick_ns/2) / tick_ns;
    bit_1.level0 = 1;
    bit_1.duration1 = (T1L_NS + tick_ns/2) / tick_ns;
    bit_1.level1 = 0;
    
    rmt_config_t config = {
        .rmt_mode = RMT_MODE_TX,
        .channel = RMT_TX_CHANNEL,
        .gpio_num = GPIO_NUM,
        .clk_div = 1,
        .mem_block_num = 1,
        .tx_config.loop_en = false,
        .tx_config.carrier_en = false,
        .tx_config.idle_output_en = true,
        .tx_config.idle_level = 0,
    };
    rmt_config(&config);
    rmt_driver_install(RMT_TX_CHANNEL, 0, 0);
}

void ws2812b_send_color(uint8_t r, uint8_t g, uint8_t b) {
    rmt_item32_t items[24];
    uint32_t bits = (g << 16) | (r << 8) | b; // GRB 顺序
    for (int i = 0; i < 24; i++) {
        items[i] = (bits & (1 << (23 - i))) ? bit_1 : bit_0;
    }
    rmt_write_items(RMT_TX_CHANNEL, items, 24, true);
    rmt_wait_tx_done(RMT_TX_CHANNEL, portMAX_DELAY);
}

void app_main() {
    rmt_init_ws2812b();
    while (1) {
        ws2812b_send_color(255, 0, 0); // 红色
        vTaskDelay(pdMS_TO_TICKS(1000));
        ws2812b_send_color(0, 255, 0); // 绿色
        vTaskDelay(pdMS_TO_TICKS(1000));
    }
}

注意事项

  • 时钟源选择：RMT 默认使用 APB 时钟，但可通过 rmt_set_source_clk() 选择 REF_TICK（1MHz），但精度更低，不推荐。
  • 中断与优先级：rmt_write_items 的 wait_tx_done 参数设为 true 会阻塞，若需非阻塞，可设为 false 并使用回调。
  • 多灯珠：连续发送时，每帧之间需保持 RESET 信号（>280us 低电平），可发送一个全零 item 或延迟。
  • 温度漂移：APB 时钟受温度影响，但 ESP32 内部 PLL 稳定，误差可忽略。
  • 调试技巧：使用逻辑分析仪验证实际波形，调整 duration 微调。

总结

通过测量实际 APB 时钟、选择合适分频、四舍五入计算 duration，并注意帧间复位，可有效消除 RMT 驱动 WS2812B 时的色彩偏差。本文提供的方法适用于所有基于 RMT 的时序敏感应用，确保嵌入式系统的高可靠性。