BoblightServer added

Former-commit-id: deb3479ee673d763ad2e5451fcd35a0febedb4f3

libsrc/boblightserver/BoblightClientConnection.cpp

@@ -0,0 +1,217 @@
+// system includes
+#include <stdexcept>
+#include <cassert>
+#include <iomanip>
+#include <cstdio>
+
+// stl includes
+#include <iostream>
+#include <sstream>
+#include <iterator>
+
+// Qt includes
+#include <QResource>
+#include <QDateTime>
+
+// hyperion util includes
+#include "hyperion/ImageProcessorFactory.h"
+#include "hyperion/ImageProcessor.h"
+#include "utils/RgbColor.h"
+
+// project includes
+#include "BoblightClientConnection.h"
+
+BoblightClientConnection::BoblightClientConnection(QTcpSocket *socket, Hyperion * hyperion) :
+	QObject(),
+	_locale(QLocale::C),
+	_socket(socket),
+	_imageProcessor(ImageProcessorFactory::getInstance().newImageProcessor()),
+	_hyperion(hyperion),
+	_receiveBuffer(),
+	_priority(255),
+	_ledColors(hyperion->getLedCount(), RgbColor::BLACK)
+{
+	// initalize the locale. Start with the default C-locale
+	_locale.setNumberOptions(QLocale::OmitGroupSeparator | QLocale::RejectGroupSeparator);
+
+	// connect internal signals and slots
+	connect(_socket, SIGNAL(disconnected()), this, SLOT(socketClosed()));
+	connect(_socket, SIGNAL(readyRead()), this, SLOT(readData()));
+}
+
+BoblightClientConnection::~BoblightClientConnection()
+{
+	if (_priority < 255)
+	{
+		// clear the current channel
+		_hyperion->clear(_priority);
+		_priority = 255;
+	}
+
+	delete _socket;
+}
+
+void BoblightClientConnection::readData()
+{
+	_receiveBuffer += _socket->readAll();
+
+	int bytes = _receiveBuffer.indexOf('\n') + 1;
+	while(bytes > 0)
+	{
+		// create message string (strip the newline)
+		QString message = QString::fromAscii(_receiveBuffer.data(), bytes-1);
+
+		// remove message data from buffer
+		_receiveBuffer = _receiveBuffer.mid(bytes);
+
+		// handle message
+		handleMessage(message);
+
+		// try too look up '\n' again
+		bytes = _receiveBuffer.indexOf('\n') + 1;
+	}
+}
+
+void BoblightClientConnection::socketClosed()
+{
+	if (_priority < 255)
+	{
+		// clear the current channel
+		_hyperion->clear(_priority);
+		_priority = 255;
+	}
+
+	emit connectionClosed(this);
+}
+
+void BoblightClientConnection::handleMessage(const QString & message)
+{
+	//std::cout << "boblight message: " << message.toStdString() << std::endl;
+
+	QStringList messageParts = message.split(" ", QString::SkipEmptyParts);
+
+	if (messageParts.size() > 0)
+	{
+		if (messageParts[0] == "hello")
+		{
+			sendMessage("hello\n");
+			return;
+		}
+		else if (messageParts[0] == "ping")
+		{
+			sendMessage("ping 1\n");
+			return;
+		}
+		else if (messageParts[0] == "get" && messageParts.size() > 1)
+		{
+			if (messageParts[1] == "version")
+			{
+				sendMessage("version 5\n");
+				return;
+			}
+			else if (messageParts[1] == "lights")
+			{
+				sendLightMessage();
+				return;
+			}
+		}
+		else if (messageParts[0] == "set" && messageParts.size() > 2)
+		{
+			if (messageParts.size() > 3 && messageParts[1] == "light")
+			{
+				bool rc;
+				unsigned ledIndex = messageParts[2].toUInt(&rc);
+				if (rc && ledIndex < _ledColors.size())
+				{
+					if (messageParts[3] == "rgb" && messageParts.size() == 7)
+					{
+						bool rc1, rc2, rc3;
+						uint8_t red = 255 * messageParts[4].toFloat(&rc1);
+						if (!rc1)
+						{
+							// maybe a locale issue. switch to a locale with a comma instead of a dot as decimal seperator (or vice versa)
+							_locale = QLocale((_locale.decimalPoint() == QChar('.')) ? QLocale::Dutch : QLocale::C);
+							_locale.setNumberOptions(QLocale::OmitGroupSeparator | QLocale::RejectGroupSeparator);
+
+							// try again
+							red = 255 * messageParts[4].toFloat(&rc1);
+						}
+
+						uint8_t green = 255 * messageParts[5].toFloat(&rc2);
+						uint8_t blue  = 255 * messageParts[6].toFloat(&rc3);
+
+						if (rc1 && rc2 && rc3)
+						{
+							RgbColor & rgb =  _ledColors[ledIndex];
+							rgb.red = red;
+							rgb.green = green;
+							rgb.blue = blue;
+							return;
+						}
+					}
+					else if(messageParts[3] == "speed" ||
+							messageParts[3] == "interpolation" ||
+							messageParts[3] == "use" ||
+							messageParts[3] == "singlechange")
+					{
+						// these message are ignored by Hyperion
+						return;
+					}
+				}
+			}
+			else if (messageParts.size() == 3 && messageParts[1] == "priority")
+			{
+				bool rc;
+				int prio = messageParts[2].toInt(&rc);
+				if (rc && prio != _priority)
+				{
+					if (_priority < 255)
+					{
+						// clear the current channel
+						_hyperion->clear(_priority);
+					}
+
+					_priority = prio;
+					return;
+				}
+			}
+		}
+		else if (messageParts[0] == "sync")
+		{
+			// send current color values to hyperion
+			if (_priority < 255)
+			{
+				_hyperion->setColors(_priority, _ledColors, -1);
+			}
+		}
+	}
+
+	std::cout << "unknown boblight message: " << message.toStdString() << std::endl;
+}
+
+void BoblightClientConnection::sendMessage(const std::string & message)
+{
+	//std::cout << "send boblight message: " << message;
+	_socket->write(message.c_str(), message.size());
+}
+
+void BoblightClientConnection::sendMessage(const char * message, int size)
+{
+	//std::cout << "send boblight message: " << std::string(message, size);
+	_socket->write(message, size);
+}
+
+void BoblightClientConnection::sendLightMessage()
+{
+	char buffer[256];
+	int n = snprintf(buffer, sizeof(buffer), "lights %d\n", _hyperion->getLedCount());
+	sendMessage(buffer, n);
+
+	for (unsigned i = 0; i < _hyperion->getLedCount(); ++i)
+	{
+		double h0, h1, v0, v1;
+		_imageProcessor->getScanParameters(i, h0, h1, v0, v1);
+		n = snprintf(buffer, sizeof(buffer), "light %03d scan %f %f %f %f\n", i, 100*v0, 100*v1, 100*h0, 100*h1);
+		sendMessage(buffer, n);
+	}
+}