162 lines
2.8 KiB
C++
162 lines
2.8 KiB
C++
|
|
// STL includes
|
|
#include <vector>
|
|
#include <ctime>
|
|
#include <cstring>
|
|
#include <string>
|
|
#include <unistd.h>
|
|
#include <iostream>
|
|
|
|
// Local includes
|
|
#include <utils/ColorRgb.h>
|
|
|
|
#include "../libsrc/leddevice/LedDeviceWs2801.h"
|
|
|
|
void setColor(char* colorStr)
|
|
{
|
|
ColorRgb color = ColorRgb::BLACK;
|
|
std::cout << "Switching all leds to: ";
|
|
if (strncmp("red", colorStr, 3) == 0)
|
|
{
|
|
std::cout << "red";
|
|
color = ColorRgb::RED;
|
|
}
|
|
else if (strncmp("green", colorStr, 5) == 0)
|
|
{
|
|
std::cout << "green";
|
|
color = ColorRgb::GREEN;
|
|
}
|
|
else if (strncmp("blue", colorStr, 5) == 0)
|
|
{
|
|
std::cout << "blue";
|
|
color = ColorRgb::BLUE;
|
|
}
|
|
else if (strncmp("cyan", colorStr, 5) == 0)
|
|
{
|
|
std::cout << "cyan";
|
|
}
|
|
else if (strncmp("gray", colorStr, 4) == 0)
|
|
{
|
|
std::cout << "gray";
|
|
}
|
|
else if (strncmp("white", colorStr, 5) == 0)
|
|
{
|
|
std::cout << "white";
|
|
color = ColorRgb::WHITE;
|
|
}
|
|
else if (strncmp("black", colorStr, 5) == 0)
|
|
{
|
|
std::cout << "black";
|
|
color = ColorRgb::BLACK;
|
|
}
|
|
std::cout << std::endl;
|
|
|
|
unsigned ledCnt = 50;
|
|
std::vector<ColorRgb> buff(ledCnt, color);
|
|
|
|
LedDeviceWs2801 ledDevice("/dev/spidev0.0", 40000);
|
|
ledDevice.open();
|
|
ledDevice.write(buff);
|
|
}
|
|
|
|
bool _running = true;
|
|
void doCircle()
|
|
{
|
|
ColorRgb color_1 = ColorRgb::RED;
|
|
ColorRgb color_2 = ColorRgb::YELLOW;
|
|
|
|
unsigned ledCnt = 50;
|
|
std::vector<ColorRgb> data(ledCnt, ColorRgb::BLACK);
|
|
|
|
LedDeviceWs2801 ledDevice("/dev/spidev0.0", 40000);
|
|
ledDevice.open();
|
|
|
|
timespec loopTime;
|
|
loopTime.tv_sec = 0;
|
|
loopTime.tv_nsec = 100000000; // 100 ms
|
|
|
|
int curLed_1 = 0;
|
|
int nextLed_1 = 1;
|
|
|
|
int curLed_2 = 49;
|
|
int nextLed_2 = 48;
|
|
|
|
|
|
while (_running)
|
|
{
|
|
data[curLed_1] = ColorRgb::BLACK;
|
|
data[curLed_2] = ColorRgb::BLACK;
|
|
|
|
// Move the current and the next pointer
|
|
curLed_1 = nextLed_1;
|
|
curLed_2 = nextLed_2;
|
|
++nextLed_1;
|
|
--nextLed_2;
|
|
if (nextLed_1 == int(ledCnt))
|
|
{
|
|
nextLed_1 = 0;
|
|
}
|
|
if (nextLed_2 < 0)
|
|
{
|
|
nextLed_2 = 49;
|
|
}
|
|
|
|
data[curLed_1] = color_1;
|
|
|
|
data[curLed_2] = color_2;
|
|
|
|
ledDevice.write(data);
|
|
|
|
nanosleep(&loopTime, NULL);
|
|
}
|
|
|
|
// Switch the current leds off
|
|
data[curLed_1] = ColorRgb::BLACK;
|
|
data[curLed_2] = ColorRgb::BLACK;
|
|
|
|
ledDevice.write(data);
|
|
}
|
|
|
|
#include <csignal>
|
|
|
|
void signal_handler(int signum)
|
|
{
|
|
_running = false;
|
|
}
|
|
|
|
int main(int argc, char** argv)
|
|
{
|
|
if (sizeof(ColorRgb) != 3)
|
|
{
|
|
std::cout << "sizeof(ColorRgb) = " << sizeof(ColorRgb) << std::endl;
|
|
return -1;
|
|
}
|
|
|
|
// Install signal handlers to stop loops
|
|
signal(SIGTERM, &signal_handler);
|
|
signal(SIGINT, &signal_handler);
|
|
|
|
if (argc < 2)
|
|
{
|
|
std::cerr << "Missing argument" << std::endl;
|
|
return -1;
|
|
}
|
|
|
|
if (strncmp("fixed", argv[1], 5) == 0)
|
|
{
|
|
setColor(argv[2]);
|
|
return 0;
|
|
}
|
|
else if (strncmp("circle", argv[1], 6) == 0)
|
|
{
|
|
doCircle();
|
|
}
|
|
else
|
|
{
|
|
std::cerr << "Unknown option: " << argv[1] << std::endl;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|