arduino通过自定义协议控制ws2812led全彩彩灯

#include <Adafruit_NeoPixel.h>
#ifdef __AVR__
#include <avr/power.h>
#endif
#define PIN_LED1_DI    1   // led1 di口
#define PIN_LED2_DI    2   // led2 di口#define LED_COMMAND_SET_FLASH                'A'      // 设置灯闪烁
#define LED_COMMAND_STOP_FLASH               'B'      // 停止灯闪烁
#define LED_COMMAND_SET_COLOR                'C'      // 设置灯颜色char temp;
char recvData[1024];
int  nRecvDataLen;
int FlashFlag;
char FlashValue;
Adafruit_NeoPixel strip = Adafruit_NeoPixel(8, PIN_LED1_DI, NEO_GRB + NEO_KHZ800);//8代表有8个小灯
//如果一个板子控制多个灯圈，在这新建，组成数组，然后做处理
void setup() {// put your setup code here, to run once:Serial.begin(115200); //设置串口波特率115200while (Serial.read() >= 0) {} //清除缓存pinMode(PIN_LED1_DI, OUTPUT);memset(recvData, 0, sizeof(recvData));nRecvDataLen = 0;FlashFlag=0;
}// Fill the dots one after the other with a color
void colorWipe(uint32_t c, uint8_t wait) {for(uint16_t i=0; i<strip.numPixels(); i++) {strip.setPixelColor(i, c);strip.show();delay(wait);}
}void rainbow(uint8_t wait) {uint16_t i, j;for(j=0; j<256; j++) {for(i=0; i<strip.numPixels(); i++) {strip.setPixelColor(i, Wheel((i+j) & 255));}strip.show();delay(wait);}
}// Slightly different, this makes the rainbow equally distributed throughout
void rainbowCycle(uint8_t wait) {uint16_t i, j;for(j=0; j<256*5; j++) { // 5 cycles of all colors on wheelfor(i=0; i< strip.numPixels(); i++) {strip.setPixelColor(i, Wheel(((i * 256 / strip.numPixels()) + j) & 255));}strip.show();delay(wait);}
}//Theatre-style crawling lights.
void theaterChase(uint32_t c, uint8_t wait) {for (int j=0; j<10; j++) {  //do 10 cycles of chasingfor (int q=0; q < 3; q++) {for (int i=0; i < strip.numPixels(); i=i+3) {strip.setPixelColor(i+q, c);    //turn every third pixel on}strip.show();delay(wait);for (int i=0; i < strip.numPixels(); i=i+3) {strip.setPixelColor(i+q, 0);        //turn every third pixel off}}}
}//Theatre-style crawling lights with rainbow effect
void theaterChaseRainbow(uint8_t wait) {for (int j=0; j < 256; j++) {     // cycle all 256 colors in the wheelfor (int q=0; q < 3; q++) {for (int i=0; i < strip.numPixels(); i=i+3) {strip.setPixelColor(i+q, Wheel( (i+j) % 255));    //turn every third pixel on}strip.show();delay(wait);for (int i=0; i < strip.numPixels(); i=i+3) {strip.setPixelColor(i+q, 0);        //turn every third pixel off}}}
}// Input a value 0 to 255 to get a color value.
// The colours are a transition r - g - b - back to r.
uint32_t Wheel(byte WheelPos) {WheelPos = 255 - WheelPos;if(WheelPos < 85) {return strip.Color(255 - WheelPos * 3, 0, WheelPos * 3);}if(WheelPos < 170) {WheelPos -= 85;return strip.Color(0, WheelPos * 3, 255 - WheelPos * 3);}WheelPos -= 170;return strip.Color(WheelPos * 3, 255 - WheelPos * 3, 0);
}void FlashControl(char value) {if (value == 'R') {theaterChase(strip.Color(127, 0, 0), 10); // Red} else if (value == 'G') {theaterChase(strip.Color(127, 127, 127), 10);// White} else if (value == 'B') {theaterChase(strip.Color(0, 0, 127), 10); // Blue}else{theaterChaseRainbow(0);//七色轮闪光}
}
void SetColor(char value) {if (value == 'R') {colorWipe(strip.Color(255, 0, 0), 0);} else if (value == 'G') {colorWipe(strip.Color(0, 255, 0), 0);} else if (value == 'B') {colorWipe(strip.Color(0, 0, 255), 0);}else if (value == '0') {colorWipe(strip.Color(0, 0, 0), 0);}
}int processOneCommand()
{int nLedNo = recvData[1] - 0x30;char nCmdNo = recvData[2];char cValue = recvData[3];// 看是哪种命令，并调用相应的接口if (nCmdNo == LED_COMMAND_SET_FLASH){FlashFlag=1;FlashValue=cValue;}else if (nCmdNo == LED_COMMAND_SET_COLOR){FlashFlag=0;SetColor(cValue);}else if(nCmdNo == LED_COMMAND_STOP_FLASH){FlashFlag=0;SetColor('0');}}
void loop() {// put your main code here, to run repeatedly:while (Serial.available() > 0){temp = Serial.read();//获取串口接收到的数据一个一个收if (temp == '$'){nRecvDataLen = 0;}if (nRecvDataLen < 6){recvData[nRecvDataLen++] = temp;}if (recvData[0] == '$' && recvData[5] == '@'){processOneCommand();memset(recvData, 0, sizeof(recvData));nRecvDataLen = 0; // 已有的数据缓冲区清空}}if(FlashFlag){FlashControl(FlashValue);}}