hyperion.ng/test/TestSpi.cpp

// 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;
}