mirror of
https://github.com/hyperion-project/hyperion.ng.git
synced 2023-10-10 13:36:59 +02:00
218 lines
5.3 KiB
C++
218 lines
5.3 KiB
C++
// 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);
|
|
}
|
|
}
|