mirror of
https://github.com/hyperion-project/hyperion.ng.git
synced 2023-10-10 13:36:59 +02:00
6143075365
2. Adds support for level shifters that invert the data Config option - "invert" (integer) if omitted, do not invert if == 0, do not invert if !=0, invert Former-commit-id: fd5b2863a85b11ac3a8444d1e02822f852feb556
84 lines
2.1 KiB
C++
84 lines
2.1 KiB
C++
#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)
|
|
{
|
|
initialized = false;
|
|
led_string.freq = freq;
|
|
led_string.dmanum = dmanum;
|
|
if (pwmchannel != 0 && pwmchannel != 1) {
|
|
std::cout << "WS281x: invalid PWM channel; must be 0 or 1." << std::endl;
|
|
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;
|
|
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) {
|
|
std::cout << "Unable to initialize ws281x library." << std::endl;
|
|
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;
|
|
led_string.channel[chan].leds[idx++] = ((uint32_t)color.red << 16) + ((uint32_t)color.green << 8) + color.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)
|
|
{
|
|
std::cout << "Shutdown WS281x PWM and DMA channel" << std::endl;
|
|
ws2811_fini(&led_string);
|
|
}
|
|
initialized = false;
|
|
}
|