hyperion.ng/libsrc/leddevice/LedDeviceWS281x.cpp

#include <iostream>

#include "LedDeviceWS281x.h"

// Constructor
LedDeviceWS281x::LedDeviceWS281x(const int gpio, const int leds, const uint32_t freq, const int dmanum, const int pwmchannel, const int invert, const int rgbw, const std::string& whiteAlgorithm)
{
	_whiteAlgorithm = whiteAlgorithm;
	Debug( _log, "whiteAlgorithm : %s", whiteAlgorithm.c_str());

	initialized = false;
	led_string.freq = freq;
	led_string.dmanum = dmanum;
	if (pwmchannel != 0 && pwmchannel != 1) {
		Error( _log, "WS281x: invalid PWM channel; must be 0 or 1.");
		throw -1;
	}
	chan = pwmchannel;
	led_string.channel[chan].gpionum = gpio;
	led_string.channel[chan].invert = invert;
	led_string.channel[chan].count = leds;
	led_string.channel[chan].brightness = 255;
	if (rgbw == 1) {
		led_string.channel[chan].strip_type = SK6812_STRIP_GRBW;
	} else {
		led_string.channel[chan].strip_type = WS2811_STRIP_RGB;
	}

	led_string.channel[!chan].gpionum = 0;
	led_string.channel[!chan].invert = invert;
	led_string.channel[!chan].count = 0;
	led_string.channel[!chan].brightness = 0;
	led_string.channel[!chan].strip_type = WS2811_STRIP_RGB;
	if (ws2811_init(&led_string) < 0) {
		Error( _log,  "Unable to initialize ws281x library.");
		throw -1;
	}
	initialized = true;
}

// Send new values down the LED chain
int LedDeviceWS281x::write(const std::vector<ColorRgb> &ledValues)
{
	if (!initialized)
		return -1;

	int idx = 0;
	for (const ColorRgb& color : ledValues)
	{
		if (idx >= led_string.channel[chan].count)
			break;

		_temp_rgbw.red = color.red;
		_temp_rgbw.green = color.green;
		_temp_rgbw.blue = color.blue;
		_temp_rgbw.white = 0;

		if (led_string.channel[chan].strip_type == SK6812_STRIP_GRBW) {
			Rgb_to_Rgbw(color, &_temp_rgbw, _whiteAlgorithm);
		}

		led_string.channel[chan].leds[idx++] = 
			((uint32_t)_temp_rgbw.white << 24) + ((uint32_t)_temp_rgbw.red << 16) + ((uint32_t)_temp_rgbw.green << 8) + _temp_rgbw.blue;

	}
	while (idx < led_string.channel[chan].count)
		led_string.channel[chan].leds[idx++] = 0;

	if (ws2811_render(&led_string))
		return -1;

	return 0;
}

// Turn off the LEDs by sending 000000's
// TODO Allow optional power switch out another gpio, if this code handles it can
// make it more likely we don't accidentally drive data into an off strip
int LedDeviceWS281x::switchOff()
{
	if (!initialized)
		return -1;

	int idx = 0;
	while (idx < led_string.channel[chan].count)
		led_string.channel[chan].leds[idx++] = 0;

	if (ws2811_render(&led_string))
		return -1;

	return 0;
}

// Destructor
LedDeviceWS281x::~LedDeviceWS281x()
{
	if (initialized)
	{
		ws2811_fini(&led_string);
	}
	initialized = false;
}