Adalight-FastLED_rgbwMod/Arduino/LEDstream_FastLED/LEDstream_FastLED.ino

267 lines
7.0 KiB
Arduino
Raw Normal View History

/* LEDstream_FastLED
2016-12-21 17:02:29 +00:00
*
2017-06-08 21:06:36 +00:00
* Modified version of Adalight that uses the FastLED
2016-12-21 17:02:29 +00:00
* library (http://fastled.io) for driving led strips.
*
* http://github.com/dmadison/Adalight-FastLED
*
* --------------------------------------------------------------------
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
* --------------------------------------------------------------------
2016-12-21 17:02:29 +00:00
*/
2016-12-21 16:01:46 +00:00
2017-03-19 22:48:14 +00:00
// --- General Settings
static const uint16_t
Num_Leds = 80; // strip length
static const uint8_t
2017-03-27 21:35:44 +00:00
Brightness = 255; // maximum brightness
2016-12-21 16:19:03 +00:00
2016-12-21 16:29:46 +00:00
// --- FastLED Setings
#define LED_TYPE WS2812B // led strip type for FastLED
#define COLOR_ORDER GRB // color order for bitbang
#define PIN_DATA 6 // led data output pin
//#define PIN_CLOCK 7 // led data clock pin (uncomment if you're using a 4-wire LED type)
2016-12-21 16:29:46 +00:00
2016-12-21 16:19:03 +00:00
// --- Serial Settings
static const unsigned long
SerialSpeed = 115200; // serial port speed
static const uint16_t
SerialTimeout = 60; // time before LEDs are shut off if no data (in seconds), 0 to disable
2017-03-27 21:35:44 +00:00
2017-03-19 22:48:14 +00:00
// --- Optional Settings (uncomment to add)
2017-06-08 21:10:57 +00:00
#define SERIAL_FLUSH // Serial buffer cleared on LED latch
//#define CLEAR_ON_START // LEDs are cleared on reset
2016-12-21 16:19:03 +00:00
//#define GROUND_PIN 10 // additional grounding pin (optional)
2016-12-21 19:13:59 +00:00
//#define CALIBRATE // sets all LEDs to the color of the first
// --- Debug Settings (uncomment to add)
//#define DEBUG_LED 13 // toggles the Arduino's built-in LED on header match
2017-05-05 09:40:40 +00:00
//#define DEBUG_FPS 8 // enables a pulse on LED latch
2016-12-21 16:19:03 +00:00
// --------------------------------------------------------------------
2016-12-21 16:29:46 +00:00
#include <FastLED.h>
CRGB leds[Num_Leds];
uint8_t * ledsRaw = (uint8_t *)leds;
// A 'magic word' (along with LED count & checksum) precedes each block
// of LED data; this assists the microcontroller in syncing up with the
// host-side software and properly issuing the latch (host I/O is
2017-03-27 21:35:44 +00:00
// likely buffered, making usleep() unreliable for latch). You may see
// an initial glitchy frame or two until the two come into alignment.
// The magic word can be whatever sequence you like, but each character
// should be unique, and frequent pixel values like 0 and 255 are
2017-03-27 21:35:44 +00:00
// avoided -- fewer false positives. The host software will need to
// generate a compatible header: immediately following the magic word
// are three bytes: a 16-bit count of the number of LEDs (high byte
// first) followed by a simple checksum value (high byte XOR low byte
2017-03-27 21:35:44 +00:00
// XOR 0x55). LED data follows, 3 bytes per LED, in order R, G, B,
// where 0 = off and 255 = max brightness.
static const uint8_t magic[] = {
2017-03-27 21:35:44 +00:00
'A','d','a'};
#define MAGICSIZE sizeof(magic)
// Check values are header byte # - 1, as they are indexed from 0
#define HICHECK (MAGICSIZE)
#define LOCHECK (MAGICSIZE + 1)
#define CHECKSUM (MAGICSIZE + 2)
2017-06-08 21:04:22 +00:00
enum processModes_t {Header, Data} mode = Header;
static int16_t
c;
static uint16_t
outPos;
static uint32_t
bytesRemaining;
static unsigned long
t,
lastByteTime,
lastAckTime;
2017-05-05 09:40:40 +00:00
// Debug macros initialized
2018-10-18 22:19:00 +00:00
#ifdef DEBUG_CI
#undef LED_TYPE
#define LED_TYPE LED_CHIPSET // Use command line chipset definition
#endif
#ifdef DEBUG_LED
#define ON 1
#define OFF 0
#define D_LED(x) do {digitalWrite(DEBUG_LED, x);} while(0)
#else
#define D_LED(x)
#endif
2017-05-05 09:40:40 +00:00
#ifdef DEBUG_FPS
#define D_FPS do {digitalWrite(DEBUG_FPS, HIGH); digitalWrite(DEBUG_FPS, LOW);} while (0)
#else
#define D_FPS
#endif
void setup(){
2017-03-27 21:35:44 +00:00
#ifdef GROUND_PIN
pinMode(GROUND_PIN, OUTPUT);
digitalWrite(GROUND_PIN, LOW);
#endif
2016-12-21 16:01:46 +00:00
2017-04-19 02:52:37 +00:00
#ifdef DEBUG_LED
pinMode(DEBUG_LED, OUTPUT);
digitalWrite(DEBUG_LED, LOW);
#endif
2017-05-05 09:40:40 +00:00
#ifdef DEBUG_FPS
pinMode(DEBUG_FPS, OUTPUT);
#endif
#ifdef PIN_CLOCK
FastLED.addLeds<LED_TYPE, PIN_DATA, PIN_CLOCK, COLOR_ORDER>(leds, Num_Leds);
#else
FastLED.addLeds<LED_TYPE, PIN_DATA, COLOR_ORDER>(leds, Num_Leds);
#endif
2017-03-27 21:35:44 +00:00
FastLED.setBrightness(Brightness);
2017-03-27 21:35:44 +00:00
#ifdef CLEAR_ON_START
FastLED.show();
#endif
2017-03-27 19:14:02 +00:00
2017-03-27 21:35:44 +00:00
Serial.begin(SerialSpeed);
Serial.print("Ada\n"); // Send ACK string to host
lastByteTime = lastAckTime = millis(); // Set initial counters
}
void loop(){
2017-03-27 21:35:44 +00:00
adalight();
}
void adalight(){
2017-05-05 13:16:01 +00:00
t = millis(); // Save current time
2017-05-05 13:16:01 +00:00
// If there is new serial data
if((c = Serial.read()) >= 0){
lastByteTime = lastAckTime = t; // Reset timeout counters
switch(mode) {
2017-06-08 21:04:22 +00:00
case Header:
headerMode();
break;
2017-06-08 21:04:22 +00:00
case Data:
dataMode();
break;
}
}
else {
2017-05-05 13:16:01 +00:00
// No new data
timeouts();
}
}
void headerMode(){
static uint8_t
headPos,
hi, lo, chk;
if(headPos < MAGICSIZE){
2017-05-05 13:16:01 +00:00
// Check if magic word matches
if(c == magic[headPos]) {headPos++;}
else {headPos = 0;}
}
else{
2017-05-05 13:16:01 +00:00
// Magic word matches! Now verify checksum
switch(headPos){
case HICHECK:
hi = c;
headPos++;
break;
case LOCHECK:
lo = c;
headPos++;
break;
case CHECKSUM:
chk = c;
if(chk == (hi ^ lo ^ 0x55)) {
// Checksum looks valid. Get 16-bit LED count, add 1
// (# LEDs is always > 0) and multiply by 3 for R,G,B.
D_LED(ON);
bytesRemaining = 3L * (256L * (long)hi + (long)lo + 1L);
outPos = 0;
memset(leds, 0, Num_Leds * sizeof(struct CRGB));
2017-06-08 21:04:22 +00:00
mode = Data; // Proceed to latch wait mode
}
headPos = 0; // Reset header position regardless of checksum result
break;
}
}
}
void dataMode(){
2017-05-05 13:16:01 +00:00
// If LED data is not full
if (outPos < sizeof(leds)){
dataSet();
}
bytesRemaining--;
if(bytesRemaining == 0) {
// End of data -- issue latch:
2017-06-08 21:04:22 +00:00
mode = Header; // Begin next header search
FastLED.show();
D_FPS;
D_LED(OFF);
2017-06-08 20:28:17 +00:00
#ifdef SERIAL_FLUSH
serialFlush();
#endif
}
}
void dataSet(){
#ifdef CALIBRATE
if(outPos < 3)
ledsRaw[outPos++] = c;
else{
ledsRaw[outPos] = ledsRaw[outPos%3]; // Sets RGB data to first LED color
outPos++;
}
#else
ledsRaw[outPos++] = c; // Issue next byte
#endif
}
void timeouts(){
// No data received. If this persists, send an ACK packet
// to host once every second to alert it to our presence.
2017-12-05 00:11:30 +00:00
if((t - lastAckTime) >= 1000) {
Serial.print("Ada\n"); // Send ACK string to host
lastAckTime = t; // Reset counter
// If no data received for an extended time, turn off all LEDs.
2017-12-05 00:43:20 +00:00
if(SerialTimeout != 0 && (t - lastByteTime) >= (uint32_t) SerialTimeout * 1000) {
memset(leds, 0, Num_Leds * sizeof(struct CRGB)); //filling Led array by zeroes
FastLED.show();
mode = Header;
lastByteTime = t; // Reset counter
}
}
}
2017-06-08 20:28:17 +00:00
void serialFlush(){
while(Serial.available() > 0) {
Serial.read();
2017-06-08 20:28:17 +00:00
}
}