/*
* 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 "lw_ws2812.h"
#include <stdlib.h>

WS2812::WS2812(uint16_t num_leds, uint16_t offset) {
    // used if we know the length of the strip and
    // the offset from the start of the master array
    // up front
    init(num_leds, offset);
}

WS2812::WS2812(uint16_t num_leds) {
    // used when we know the lengths up front.
    init(num_leds);
}

WS2812::WS2812() {
    // used as an empty constructor for setup.
    init(0);
}

void WS2812::init(uint16_t num_leds, uint16_t offset) {

    set_length(num_leds);
    set_offset(offset);
}

void WS2812::init(uint16_t num_leds) {
    // assume offset is zeron in this case
    init(num_leds, 0);
}

void WS2812::set_offset(uint16_t master_offset) {
    offset = master_offset;
}

void WS2812::set_length(uint16_t num_leds) {
    count_led = num_leds;
}

uint16_t WS2812::get_length() {
    return count_led;
}

// TODO: Remove magic threes
void WS2812::sync(uint8_t *px_array, uint8_t pixel_depth) {
	*ws2812_port_reg |= pinMask; // Enable DDR
	ws2812_sendarray_mask(px_array+(offset*pixel_depth),
            count_led * pixel_depth,
            pinMask,(uint8_t*) ws2812_port,(uint8_t*) ws2812_port_reg
    );
}

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