You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
100 lines
1.9 KiB
100 lines
1.9 KiB
10 years ago
|
/*
|
||
|
|
* light weight WS2812 lib V2.1 - Arduino support
|
||
|
|
*
|
||
|
|
* Controls WS2811/WS2812/WS2812B RGB-LEDs
|
||
|
|
* Author: Matthias Riegler
|
||
|
|
*
|
||
|
|
* Mar 07 2014: Added Arduino and C++ Library
|
||
|
|
*
|
||
|
|
* September 6, 2014: Added option to switch between most popular color orders
|
||
|
|
* (RGB, GRB, and BRG) -- Windell H. Oskay
|
||
|
|
*
|
||
|
|
* License: GNU GPL v2 (see License.txt)
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "WS2812.h"
|
||
|
|
#include <stdlib.h>
|
||
|
|
|
||
|
|
WS2812::WS2812(uint16_t num_leds) {
|
||
|
|
count_led = num_leds;
|
||
|
|
|
||
|
|
pixels = (uint8_t*)malloc(count_led*3);
|
||
|
|
offsetGreen = 0;
|
||
|
|
offsetRed = 1;
|
||
|
|
offsetBlue = 2;
|
||
|
|
}
|
||
|
|
|
||
|
|
cRGB WS2812::get_crgb_at(uint16_t index) {
|
||
|
|
|
||
|
|
cRGB px_value;
|
||
|
|
|
||
|
|
if(index < count_led) {
|
||
|
|
|
||
|
|
uint16_t tmp;
|
||
|
|
tmp = index * 3;
|
||
|
|
|
||
|
|
px_value.r = pixels[tmp+offsetRed];
|
||
|
|
px_value.g = pixels[tmp+offsetGreen];
|
||
|
|
px_value.b = pixels[tmp+offsetBlue];
|
||
|
|
}
|
||
|
|
|
||
|
|
return px_value;
|
||
|
|
}
|
||
|
|
|
||
|
|
uint8_t WS2812::set_crgb_at(uint16_t index, cRGB px_value) {
|
||
|
|
|
||
|
|
if(index < count_led) {
|
||
|
|
|
||
|
|
uint16_t tmp;
|
||
|
|
tmp = index * 3;
|
||
|
|
|
||
|
|
pixels[tmp+offsetGreen] = px_value.g;
|
||
|
|
pixels[tmp+offsetRed] = px_value.r;
|
||
|
|
pixels[tmp+offsetBlue] = px_value.b;
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
void WS2812::sync() {
|
||
|
|
*ws2812_port_reg |= pinMask; // Enable DDR
|
||
|
|
ws2812_sendarray_mask(pixels,3*count_led,pinMask,(uint8_t*) ws2812_port,(uint8_t*) ws2812_port_reg );
|
||
|
|
}
|
||
|
|
|
||
|
|
void WS2812::setColorOrderGRB() { // Default color order
|
||
|
|
offsetGreen = 0;
|
||
|
|
offsetRed = 1;
|
||
|
|
offsetBlue = 2;
|
||
|
|
}
|
||
|
|
|
||
|
|
void WS2812::setColorOrderRGB() {
|
||
|
|
offsetRed = 0;
|
||
|
|
offsetGreen = 1;
|
||
|
|
offsetBlue = 2;
|
||
|
|
}
|
||
|
|
|
||
|
|
void WS2812::setColorOrderBRG() {
|
||
|
|
offsetBlue = 0;
|
||
|
|
offsetRed = 1;
|
||
|
|
offsetGreen = 2;
|
||
|
|
}
|
||
|
|
|
||
|
|
WS2812::~WS2812() {
|
||
|
|
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
#ifndef ARDUINO
|
||
|
|
void WS2812::setOutput(const volatile uint8_t* port, volatile uint8_t* reg, uint8_t pin) {
|
||
|
|
pinMask = (1<<pin);
|
||
|
|
ws2812_port = port;
|
||
|
|
ws2812_port_reg = reg;
|
||
|
|
}
|
||
|
|
#else
|
||
|
|
void WS2812::setOutput(uint8_t pin) {
|
||
|
|
pinMask = digitalPinToBitMask(pin);
|
||
|
|
ws2812_port = portOutputRegister(digitalPinToPort(pin));
|
||
|
|
ws2812_port_reg = portModeRegister(digitalPinToPort(pin));
|
||
|
|
}
|
||
|
|
#endif
|