AVR定时器PWM驱动WS2812B：汇编级精准时序控制实战

1. 项目概述：用定时器PWM驱动WS2812B

如果你玩过单片机，尤其是AVR系列的，想驱动WS2812B这类智能LED（也叫NeoPixel），大概率会先想到用现成的库，比如FastLED或者Adafruit_NeoPixel。这些库确实方便，但有时候，尤其是在资源极其有限的8位MCU上，或者当你需要极致的时序控制时，直接操作硬件、自己写驱动就成了唯一的选择。这次要聊的，就是一个用AVR的定时器和PWM模块，纯手工打造WS2812B驱动的实战案例。它不依赖任何高级语言库，核心部分直接用汇编写成，为的就是把那800kHz、纳秒级精度的时序信号拿捏得死死的。

这个项目的核心目标很明确：利用ATMEGA系列单片机（比如48/88/168）的Timer 0，配置成PWM模式，来精准生成WS2812B所需的单线归零码通信协议。听起来有点硬核，但拆解开来，其实就是理解协议、配置硬件、编写中断服务程序这三步。最终实现的效果，是驱动一个12颗灯珠的环形灯板，让一组RGB灯珠像跑马灯一样旋转起来。整个过程，你会深刻体会到，在微秒级别的世界里，C语言都可能显得“笨重”，汇编才是掌控全局的利器。无论你是想深入理解底层硬件，还是面临资源瓶颈需要优化，这个思路都极具参考价值。

2. WS2812B通信协议深度解析

在动手写代码之前，必须把WS2812B的“语言”搞明白。它不像I2C或SPI那样有明确的时钟线和数据线，它只用一根数据线（Din），靠的是特定时间宽度的方波来传递0和1。这种协议我们通常叫它“单线归零码”。

2.1 时序要求：纳秒级的精度

WS2812B的通信速率是固定的800 kHz。这意味着一个比特位（Bit）的周期是 1 / 800,000 = 1.25 微秒（μs）。在这个周期内，高电平（Thigh）的持续时间决定了这个比特是0还是1。

根据数据手册，标准时序如下（通常允许±150ns的误差）：

• 逻辑0：高电平时间（T0H）约 350 ns，随后是低电平，总周期1.25μs。
• 逻辑1：高电平时间（T1H）约 700 ns，随后是低电平，总周期同样为1.25μs。

注意：不同批次或厂商的WS2812B模块对时序的宽容度可能不同。过于严苛的时序（比如完全卡死350ns和700ns）可能导致某些灯珠无法识别。通常，将T0H设置在300ns-500ns，T1H设置在650ns-850ns之间，系统都能稳定工作。我们后面采用的400ns和650ns就是一个兼顾了稳定性和实现便利的折中值。

为什么是PWM？仔细观察这个波形：一个固定频率（800kHz）、但占空比可变（28%对应0，56%对应1）的方波。这不正是脉宽调制（PWM）的典型特征吗？所以，用MCU的PWM外设来产生这个信号，是再自然不过的想法。我们不需要用CPU去死循环延时翻转IO口，而是配置好硬件，让它自动输出波形，CPU只需要在合适的时机去更新占空比（即高电平时间）即可，极大地解放了CPU资源。

2.2 数据格式与复位信号

除了0和1的时序，整个数据流的结构也要清楚。每个WS2812B灯珠需要接收24比特的数据，来分别控制其内部的G（绿）、R（红）、B（蓝）三个LED的亮度，每个颜色8比特（256级灰度）。这24比特的顺序通常是G7-G0, R7-R0, B7-B0（GRB顺序）。多个灯珠串联时，第一个灯珠会吞掉它自己的24比特数据，然后将后续的数据流原样从它的Dout引脚输出给下一个灯珠。

当需要更新所有灯珠的状态时，必须在发送完所有灯珠的数据后，保持数据线低电平超过50微秒（通常建议>50μs）。这个长时间的“低电平”就是一个复位（Reset）信号，告诉所有灯珠：“数据发完了，你们可以更新显示了”。如果没有这个复位信号，灯珠会一直等待，不会刷新显示。

3. 硬件方案与定时器配置

理解了协议，我们就要在硬件上实现它。项目基于一颗运行在20MHz时钟下的ATMEGA48/88/168单片机。选择PIN11（PD5）作为数据输出引脚，是因为这个引脚对应着Timer 0的通道B输出（OC0B），可以直接由硬件PWM模块控制。

3.1 定时器工作模式选择

ATMEGA的Timer 0是一个8位定时器。要产生800kHz的方波，我们需要让定时器以这个频率周期性溢出。计算一下：系统时钟20MHz / 目标频率800kHz = 25。这意味着，定时器每计数25个系统时钟周期，就应该完成一个循环（溢出一次）。对于8位定时器，最大值是255，25这个值完全在范围内。

我们选择快速PWM模式7。这个模式特殊之处在于，它的计数上限不是固定的255，而是可以由我们通过OCR0A寄存器来设定的。这被称为“可调分辨率”的快速PWM模式。我们将OCR0A设置为24（为什么是24而不是25？后面会解释）。这样，定时器就从0计数到OCR0A（24），然后清零并产生溢出中断，周期就是 (24+1)=25 个时钟周期，完美匹配800kHz的要求。

3.2 占空比（OCR0B）的计算

PWM的输出由OCR0B寄存器控制。在“比较匹配时清零OC0B”的模式下（COM0B=0b10），当定时器计数值TCNT0等于OCR0B时，OC0B引脚输出低电平。因此，OCR0B的值直接决定了高电平的持续时间。

• 对于逻辑“1”（目标Thigh=650ns）：持续时间 = OCR0B * 时钟周期。时钟周期=1/20MHz=50ns。所以 OCR0B = 650ns / 50ns = 13。但原文中采用了12，这对应600ns，仍在650±150ns的容差范围内，是一个安全且易于实现的值。
• 对于逻辑“0”（目标Thigh=400ns）：OCR0B = 400ns / 50ns = 8。原文中采用了7，对应350ns，同样在标准范围内。

所以，在我们的配置中：

• THigh(代表‘1’) = 12
• TLow(代表‘0’) = 7
• OCR0A(周期) = 24

这里有一个关键点：当TCNT0计数到OCR0A（24）时，在下一个时钟周期TCNT0会被清零，并发生溢出中断。而我们的PWM波形，是在TCNT0从0开始计数，到等于OCR0B时拉低，直到本次周期结束（TCNT0==OCR0A后清零）。因此，高电平时间实际上是(OCR0B + 1) * 50ns。计算一下：(12+1)*50ns=650ns，(7+1)*50ns=400ns，这就完全精确了。所以OCR0A设为24，周期(24+1)*50ns=1250ns=1.25μs，频率800kHz；OCR0B设为12和7，分别得到650ns和400ns的高电平。

3.3 引脚与初始化流程

硬件连接非常简单：WS2812B灯带的数据输入引脚（Din）直接接到MCU的OC0B引脚（PD5）。如果灯带需要5V供电而MCU是3.3V，可能需要一个电平转换电路，或者选择5V耐受的MCU型号（如ATMEGA系列多数IO口可耐受5V）。

软件初始化步骤如下：

1. 配置IO口：将PD5（OC0B）设置为输出模式。
2. 配置Timer 0：
   • 设置TCCR0A和TCCR0B寄存器，选择快速PWM模式7（WGM02:0 = 0b111）。
   • 设置COM0B1:0 = 0b10，使得在比较匹配B时清零OC0B，在TCNT0为0时置位OC0B（即输出高电平）。
   • 将计算好的值写入OCR0A（24）和OCR0B（初始值，比如7）。
   • 先不开启时钟源（TCCR0B中的CS02:0保持为0b000），让定时器处于停止状态。
3. 使能中断：使能Timer 0的溢出中断（TOIE0）。
4. 全局中断使能。

4. 驱动程序设计：汇编与C的混合编程

这是整个项目的精髓所在。因为时序极其苛刻，我们必须保证在每次定时器溢出中断（每1.25μs发生一次）时，中断服务程序（ISR）能在极短的时间内判断出下一个要发送的比特是0还是1，并迅速更新OCR0B寄存器。

4.1 为什么必须用汇编？

计算一下时间预算：中断周期是1.25μs，即20MHz下的25个时钟周期。中断服务程序必须在下一个周期开始前完成工作并退出。这包括了保护现场（压栈）、执行逻辑、恢复现场（出栈）的所有时间。原文给出的极限是1.25μs（25个周期），实际上留给核心逻辑的时间可能只有十几甚至几个周期。

用C语言编写ISR，编译器会产生额外的指令（如寄存器保存、参数传递、函数调用开销），很难保证在这个极限时间内完成。而汇编语言允许我们对每一个时钟周期进行精确控制。在这个驱动中，ISR的核心逻辑被设计成只使用单周期指令，并且将关键变量（如当前数据字节、位掩码、字节计数器）保存在通用寄存器中，避免访问速度较慢的SRAM，从而将ISR的执行时间压缩到20个周期以内，稳稳地满足要求。

4.2 程序结构与变量定义

项目采用了混合编程。主循环和初始化等非实时性任务用高级语言（如Bascom-AVR或C）编写，而时序关键的发送函数和中断服务程序则用汇编编写。

关键的数据结构是一个字节数组LED_data[]，它存储了所有灯珠的GRB颜色数据。例如，驱动12个灯珠，就需要 12 * 3 = 36 个字节。原文中数组有39个字节，多出的3个字节用作数据移动时的缓冲区，这在实现灯珠颜色旋转效果时很方便。

几个核心的全局变量（在汇编和C中都需要访问）：

• LED_data[39]: 颜色数据数组。
• Num_Bytes: 需要发送的有效字节数（例如36）。
• Mask: 位测试掩码，初始值为0x80（二进制10000000），用于从字节的最高位（MSB）开始提取每一个比特。
• Datapointer: 指向LED_data数组当前字节的指针。

4.3 核心发送流程详解

发送过程由高级语言调用一个汇编函数WS2812_send启动。

第一步：发送函数初始化 (WS2812_send)

1. 保存所有即将用到的寄存器到堆栈（上下文保护）。
2. 加载常量：R16=TLow(7)， R17=THigh(12)， R18=1（用作辅助）， R19/R20用作控制寄存器。
3. 从内存中加载Mask到寄存器R1和R22，加载Num_Bytes到R21。
4. 将数据指针X指向LED_data数组的首地址，并取出第一个字节到R2。
5. 检查R2的最高位（利用R1中的掩码）：如果是1，则将R17（THigh）写入OCR0B；如果是0，则将R16（TLow）写入OCR0B。这设定了第一个比特的PWM占空比。
6. 将位掩码R1右移一位，准备测试下一个比特。
7. 启动Timer 0（设置TCCR0B的时钟源，如不分频CS=0b001），PWM波形开始输出。
8. 进入一个循环，等待发送完成。循环的退出条件由中断服务程序设置。

第二步：中断服务程序 (ISR_Transmit)这是每秒被执行800,000次的核心。每次进入ISR，意味着一个比特（1.25μs周期）已经发送完毕，需要准备下一个比特。

1. 判断当前字节是否发送完：检查位掩码寄存器（R1或R22）。如果掩码已经右移到了0（即(mask & 0xFF) == 0），说明当前字节的8个比特全部发完。
2. 如果当前字节发完：
   • 字节计数器R21减1。如果R21为0，说明所有字节都已发送完毕。
   • 发送完成处理：停止Timer 0（清除时钟源），将一个“结束标志”（如0x80）写入控制寄存器R20，然后退出ISR。主循环中的等待循环检测到这个标志就会跳出。
   • 如果还有字节：数据指针X加1，指向下一个颜色字节，加载到R2。重置位掩码为0x80。然后根据新字节的最高位，加载相应的THigh/TLow到OCR0B，为发送下一个字节的第一个比特做好准备。
3. 如果当前字节未发完：
   • 根据当前位掩码测试R2中的字节，判断下一个待发送比特是1还是0。
   • 将对应的值（R17或R16）加载到OCR0B寄存器。
   • 将位掩码右移一位，指向下一个比特。
4. 退出ISR。

整个流程就像一条精密的流水线：主函数设定好初始状态并启动引擎；随后，每次定时器溢出中断这个“节拍器”响起，ISR就迅速决定下一个“音符”（比特电平）的长短，并更新PWM。所有比特发送完毕后，ISR关闭定时器，通知主程序。主程序在发送完成后，需要额外等待至少50μs（即至少执行一段空循环延时），以产生复位信号，灯珠才会更新显示。

5. 实际应用：LED旋转效果实现

理解了底层驱动，上层应用就灵活多了。原文的例子是实现12颗灯珠的旋转效果。思路很简单：

1. 数据准备：在LED_data数组中，按顺序存放12个灯珠的GRB数据。假设我们想让第1、2、3个灯珠分别显示纯绿、纯红、纯蓝，其余为熄灭（0），那么数组前9个字节可能是：[0xFF, 0x00, 0x00, 0x00, 0xFF, 0x00, 0x00, 0x00, 0xFF, ...]。
2. 旋转算法：在主循环中，每隔一段时间（比如100ms），将LED_data数组中的数据整体向左或向右移动3个字节（一个灯珠的数据量）。例如，向右旋转一次，就把最后一个灯珠的数据（最后3字节）移到数组开头，其余数据依次后移。
3. 刷新显示：每次移动数据后，调用WS2812_send()函数，将新的数组数据发送给灯带，然后延时产生复位信号。人眼看到的就是绿、红、蓝三个光点在环形灯带上追逐旋转的效果。

这个例子清晰地展示了分层设计的好处：底层汇编驱动确保时序硬实时，毫秒不差；上层应用用C语言编写，专注于业务逻辑（颜色计算、动画效果），清晰易维护。

6. 移植与调试中的关键问题

虽然这个驱动是针对20MHz的ATMEGA168编写的，但其思想可以移植到其他平台，如STM32、ESP8266等。关键在于抓住几个核心参数的计算。

6.1 关键参数重计算

移植到不同时钟频率的MCU时，必须重新计算三个核心参数：OCR0A（周期）、THigh（逻辑1）、TLow（逻辑0）。 公式如下：

• OCR0A = (MCU_Clock / Target_Frequency) - 1。例如，16MHz下驱动800kHz，OCR0A = (16,000,000 / 800,000) - 1 = 19。
• THigh = (T1H_Desired * MCU_Clock) - 1。例如，16MHz下想要650ns高电平，THigh = (650e-9 * 16e6) - 1 = 9.4，取整为9（对应562.5ns）或10（对应625ns），需在容差范围内测试。
• TLow = (T0H_Desired * MCU_Clock) - 1。例如，16MHz下想要400ns，TLow = (400e-9 * 16e6) - 1 = 5.4，取整为5（对应375ns）或6（对应437.5ns）。

实操心得：计算出的值最好在示波器下验证。由于取整误差，实际波形可能与理论有微小偏差。只要高电平时间在数据手册的容差范围内（逻辑0：200ns-500ns；逻辑1：550ns-850ns），系统通常都能稳定工作。优先保证周期（1.25μs±150ns）的准确性。

6.2 常见问题排查表

6.3 性能优化与扩展

• 减少中断开销：这是汇编驱动的核心优势。确保ISR中只做最必要的操作：判断比特、更新OCR0B、管理指针和计数器。所有计算和查表操作都应在主循环中完成。
• 使用DMA（针对高级MCU）：在像STM32这样的ARM Cortex-M芯片上，可以利用定时器触发DMA，将预先计算好的PWM占空比序列（一个比特对应一个OCR值）自动搬运到定时器寄存器中，实现“零CPU开销”驱动WS2812B。这是性能最优的方案。
• 支持更多灯珠：本驱动中，灯珠数量受限于LED_data数组的大小和SRAM容量。对于ATMEGA168，有1KB SRAM，驱动上百个灯珠（300字节）是可行的。但要注意，发送所有数据的时间会变长（N241.25μs），在需要高速刷新的场合（如视频流）会成为瓶颈。
• 亮度与颜色校正：WS2812B在不同电压、温度下，颜色和亮度可能有偏差。可以在上层应用中预先建立一个校正查找表（Gamma校正表、白平衡校正），在设置颜色值前进行转换，使显示效果更专业。

这个项目虽然小，但“麻雀虽小，五脏俱全”。它涉及了硬件定时器、PWM、中断、汇编优化、混合编程等多个嵌入式开发的核心知识点。通过亲手实现一遍，你对MCU如何与外部器件进行精确时序通信的理解，会远比单纯调用库函数深刻得多。当看到自己编写的汇编代码精准地控制着每一颗LED发出预定的色彩时，那种对硬件完全掌控的成就感，是使用高级库无法比拟的。