hyperion.ng/libsrc/udplistener/UDPListener.cpp

// system includes
#include <stdexcept>

// project includes
#include <udplistener/UDPListener.h>

// hyperion util includes
#include "hyperion/ImageProcessorFactory.h"
#include "hyperion/ImageProcessor.h"
#include "utils/ColorRgb.h"
#include "HyperionConfig.h"

UDPListener::UDPListener(const int priority, const int timeout, const std::string& address, quint16 listenPort, bool shared) :
	QObject(),
	_hyperion(Hyperion::getInstance()),
	_server(),
	_openConnections(),
	_priority(priority),
	_timeout(timeout),
	_ledColors(Hyperion::getInstance()->getLedCount(), ColorRgb::BLACK),
	_log(Logger::getInstance("UDPLISTENER"))
{
	_server = new QUdpSocket(this);

	QAbstractSocket::BindFlag bondage = QAbstractSocket::DefaultForPlatform;
	if (shared) {
		bondage = QAbstractSocket::ShareAddress;
	}
	QHostAddress listenAddress = QHostAddress(QHostAddress::Any);
	QHostAddress mcastGroup;

	if (!address.empty()) {
		listenAddress = QHostAddress( QString::fromStdString(address) );
	}
	if (listenAddress.isInSubnet(QHostAddress::parseSubnet("224.0.0.0/4"))) {
		mcastGroup = listenAddress;
		listenAddress = QHostAddress(QHostAddress::AnyIPv4);
	}

	if (!_server->bind(listenAddress, listenPort, bondage))
	{
		Warning(_log, "Could not bind to %s:%d parsed from %s", listenAddress.toString().toStdString().c_str(), listenPort, address.c_str());
	} else {
		Info(_log, "Started, listening on %s:%d", listenAddress.toString().toStdString().c_str(), listenPort);
		if (!mcastGroup.isNull()) {
			if (_server->joinMulticastGroup(mcastGroup)) {
				Info(_log, "Multicast listening on %s",mcastGroup.toString().toStdString().c_str());
			} else {
				Warning(_log, "Multicast failed for %s", mcastGroup.toString().toStdString().c_str());
			}
		}
	}

	// Set trigger for incoming connections
	connect(_server, SIGNAL(readyRead()), this, SLOT(readPendingDatagrams()));
}

UDPListener::~UDPListener()
{
	// clear the current channel
	_hyperion->clear(_priority);
}

uint16_t UDPListener::getPort() const
{
	return _server->localPort();
}


void UDPListener::readPendingDatagrams()
{
	while (_server->hasPendingDatagrams()) {
		QByteArray datagram;
		datagram.resize(_server->pendingDatagramSize());
		QHostAddress sender;
		quint16 senderPort;

		_server->readDatagram(datagram.data(), datagram.size(),
					&sender, &senderPort);

		processTheDatagram(&datagram);

	}
}


void UDPListener::processTheDatagram(const QByteArray * datagram)
{
	int packlen = datagram->size()/3;
	int ledlen = _ledColors.size();
	int maxled = std::min(packlen , ledlen);

	for (int ledIndex=0; ledIndex < maxled; ledIndex++) {
		ColorRgb & rgb =  _ledColors[ledIndex];
		rgb.red = datagram->at(ledIndex*3+0);
		rgb.green = datagram->at(ledIndex*3+1);
		rgb.blue = datagram->at(ledIndex*3+2);
	}

	_hyperion->setColors(_priority, _ledColors, _timeout, -1);

}