mirror of
https://github.com/hyperion-project/hyperion.ng.git
synced 2023-10-10 13:36:59 +02:00
7a685185f4
* AtmoOrb UdpSocket-Bind Fix * Cleanup and update defaults (to work via PowerLan) * Cleanup and update defaults (to work via PowerLan) * AtmoOrb identification support, small updates * AtmoOrb discovery & identification support, fixes and stability updates * Small clean-ups * Type fix * Add missing include * Adalight - Update default config and levels * Update Atmoorb sketch * Yeelight - Update default value
384 lines
9.7 KiB
C++
384 lines
9.7 KiB
C++
// Local-Hyperion includes
|
|
#include "LedDeviceAtmoOrb.h"
|
|
#include <utils/QStringUtils.h>
|
|
|
|
// qt includes
|
|
#include <QUdpSocket>
|
|
#include <QNetworkInterface>
|
|
#include <QUrl>
|
|
#include <QHostInfo>
|
|
|
|
#include <chrono>
|
|
|
|
// Constants
|
|
namespace {
|
|
|
|
const QString MULTICAST_GROUP_DEFAULT_ADDRESS = "239.255.255.250";
|
|
const quint16 MULTICAST_GROUP_DEFAULT_PORT = 49692;
|
|
|
|
constexpr std::chrono::milliseconds DEFAULT_DISCOVERY_TIMEOUT{2000};
|
|
|
|
} //End of constants
|
|
|
|
LedDeviceAtmoOrb::LedDeviceAtmoOrb(const QJsonObject &deviceConfig)
|
|
: LedDevice(deviceConfig)
|
|
, _udpSocket (nullptr)
|
|
, _multicastGroup(MULTICAST_GROUP_DEFAULT_ADDRESS)
|
|
, _multiCastGroupPort (MULTICAST_GROUP_DEFAULT_PORT)
|
|
, _joinedMulticastgroup (false)
|
|
, _useOrbSmoothing (false)
|
|
, _skipSmoothingDiff (0)
|
|
{
|
|
}
|
|
|
|
LedDevice* LedDeviceAtmoOrb::construct(const QJsonObject &deviceConfig)
|
|
{
|
|
return new LedDeviceAtmoOrb(deviceConfig);
|
|
}
|
|
|
|
LedDeviceAtmoOrb::~LedDeviceAtmoOrb()
|
|
{
|
|
delete _udpSocket;
|
|
}
|
|
|
|
bool LedDeviceAtmoOrb::init(const QJsonObject &deviceConfig)
|
|
{
|
|
bool isInitOK = false;
|
|
|
|
if ( LedDevice::init(deviceConfig) )
|
|
{
|
|
|
|
_multicastGroup = deviceConfig["output"].toString(MULTICAST_GROUP_DEFAULT_ADDRESS);
|
|
_multiCastGroupPort = static_cast<quint16>(deviceConfig["port"].toInt(MULTICAST_GROUP_DEFAULT_PORT));
|
|
_useOrbSmoothing = deviceConfig["useOrbSmoothing"].toBool(false);
|
|
_skipSmoothingDiff = deviceConfig["skipSmoothingDiff"].toInt(0);
|
|
QStringList orbIds = QStringUtils::split(deviceConfig["orbIds"].toString().simplified().remove(" "),",", QStringUtils::SplitBehavior::SkipEmptyParts);
|
|
|
|
Debug(_log, "DeviceType : %s", QSTRING_CSTR( this->getActiveDeviceType() ));
|
|
Debug(_log, "LedCount : %u", this->getLedCount());
|
|
Debug(_log, "ColorOrder : %s", QSTRING_CSTR( this->getColorOrder() ));
|
|
Debug(_log, "RefreshTime : %d", _refreshTimerInterval_ms);
|
|
Debug(_log, "LatchTime : %d", this->getLatchTime());
|
|
|
|
Debug(_log, "MulticastGroup : %s", QSTRING_CSTR(_multicastGroup));
|
|
Debug(_log, "MulticastGroupPort: %d", _multiCastGroupPort);
|
|
Debug(_log, "Orb ID list : %s", QSTRING_CSTR(deviceConfig["orbIds"].toString()));
|
|
Debug(_log, "Use Orb Smoothing : %d", _useOrbSmoothing);
|
|
Debug(_log, "Skip SmoothingDiff: %d", _skipSmoothingDiff);
|
|
|
|
_orbIds.clear();
|
|
|
|
for (auto & id_str : orbIds)
|
|
{
|
|
bool ok;
|
|
int id = id_str.toInt(&ok);
|
|
if (ok)
|
|
{
|
|
if ( id < 1 || id > 255 )
|
|
{
|
|
Warning(_log, "Skip orb id '%d'. IDs must be in range 1-255", id);
|
|
}
|
|
else
|
|
{
|
|
_orbIds.append(id);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
Error(_log, "orb id '%s' is not a number", QSTRING_CSTR(id_str));
|
|
}
|
|
}
|
|
|
|
uint numberOrbs = _orbIds.size();
|
|
uint configuredLedCount = this->getLedCount();
|
|
|
|
if ( _orbIds.empty() )
|
|
{
|
|
this->setInError("No valid OrbIds found!");
|
|
isInitOK = false;
|
|
}
|
|
else
|
|
{
|
|
if ( numberOrbs < configuredLedCount )
|
|
{
|
|
QString errorReason = QString("Not enough Orbs [%1] for configured LEDs [%2] found!")
|
|
.arg(numberOrbs)
|
|
.arg(configuredLedCount);
|
|
this->setInError(errorReason);
|
|
isInitOK = false;
|
|
}
|
|
else
|
|
{
|
|
if ( numberOrbs > configuredLedCount )
|
|
{
|
|
Info(_log, "%s: More Orbs [%u] than configured LEDs [%u].", QSTRING_CSTR(this->getActiveDeviceType()), numberOrbs, configuredLedCount );
|
|
}
|
|
|
|
isInitOK = true;
|
|
}
|
|
}
|
|
}
|
|
return isInitOK;
|
|
}
|
|
|
|
int LedDeviceAtmoOrb::open()
|
|
{
|
|
int retval = -1;
|
|
_isDeviceReady = false;
|
|
|
|
if ( _udpSocket == nullptr )
|
|
{
|
|
_udpSocket = new QUdpSocket();
|
|
}
|
|
|
|
// Try to bind the UDP-Socket
|
|
if ( _udpSocket != nullptr )
|
|
{
|
|
if ( _udpSocket->state() != QAbstractSocket::BoundState )
|
|
{
|
|
if ( !_udpSocket->bind(QHostAddress(QHostAddress::AnyIPv4), 0 ) )
|
|
{
|
|
QString errortext = QString ("Socket bind failed: (%1) %2, MulticastGroup: (%3)").arg(_udpSocket->error()).arg(_udpSocket->errorString(), _multicastGroup);
|
|
this->setInError( errortext );
|
|
}
|
|
else
|
|
{
|
|
_groupAddress = QHostAddress(_multicastGroup);
|
|
_joinedMulticastgroup = _udpSocket->joinMulticastGroup(_groupAddress);
|
|
if ( !_joinedMulticastgroup )
|
|
{
|
|
QString errortext = QString ("Joining Multicastgroup failed: (%1) %2, MulticastGroup: (%3)").arg(_udpSocket->error()).arg(_udpSocket->errorString(), _multicastGroup);
|
|
this->setInError( errortext );
|
|
}
|
|
}
|
|
}
|
|
|
|
if ( ! _isDeviceInError )
|
|
{
|
|
// Everything is OK, device is ready
|
|
_isDeviceReady = true;
|
|
retval = 0;
|
|
}
|
|
}
|
|
return retval;
|
|
}
|
|
|
|
int LedDeviceAtmoOrb::close()
|
|
{
|
|
int retval = 0;
|
|
_isDeviceReady = false;
|
|
|
|
if ( _udpSocket != nullptr )
|
|
{
|
|
if ( _udpSocket->state() == QAbstractSocket::BoundState )
|
|
{
|
|
_udpSocket->leaveMulticastGroup(_groupAddress);
|
|
}
|
|
|
|
// Test, if device requires closing
|
|
if ( _udpSocket->isOpen() )
|
|
{
|
|
Debug(_log,"Close UDP-device: %s", QSTRING_CSTR( this->getActiveDeviceType() ) );
|
|
_udpSocket->close();
|
|
// Everything is OK -> device is closed
|
|
}
|
|
}
|
|
return retval;
|
|
}
|
|
|
|
int LedDeviceAtmoOrb::write(const std::vector <ColorRgb> &ledValues)
|
|
{
|
|
// If not in multicast group return
|
|
if (!_joinedMulticastgroup)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
// Command options:
|
|
//
|
|
// 1 = force off
|
|
// 2 = use lamp smoothing and validate by Orb ID
|
|
// 4 = validate by Orb ID
|
|
|
|
// When setting _useOrbSmoothing = true it's recommended to disable Hyperion's own smoothing as it will conflict (double smoothing)
|
|
int commandType = 4;
|
|
if(_useOrbSmoothing)
|
|
{
|
|
commandType = 2;
|
|
}
|
|
|
|
ColorRgb color;
|
|
for (int idx = 0; idx < _orbIds.size(); idx++ )
|
|
{
|
|
if ( idx < static_cast<int>(ledValues.size()) )
|
|
{
|
|
color = ledValues[idx];
|
|
}
|
|
else
|
|
{
|
|
color = ColorRgb::BLACK;
|
|
}
|
|
// Retrieve last send colors
|
|
int lastRed = lastColorRedMap[idx];
|
|
int lastGreen = lastColorGreenMap[idx];
|
|
int lastBlue = lastColorBlueMap[idx];
|
|
|
|
// If color difference is higher than _skipSmoothingDiff than we skip Orb smoothing (if enabled) and send it right away
|
|
if ((_skipSmoothingDiff != 0 && _useOrbSmoothing) && (abs(color.red - lastRed) >= _skipSmoothingDiff || abs(color.blue - lastBlue) >= _skipSmoothingDiff ||
|
|
abs(color.green - lastGreen) >= _skipSmoothingDiff))
|
|
{
|
|
// Skip Orb smoothing when using (command type 4)
|
|
commandType = 4;
|
|
}
|
|
|
|
// Send color
|
|
setColor(_orbIds[idx], color, commandType);
|
|
|
|
// Store last colors send for light id
|
|
lastColorRedMap[idx] = color.red;
|
|
lastColorGreenMap[idx] = color.green;
|
|
lastColorBlueMap[idx] = color.blue;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
void LedDeviceAtmoOrb::setColor(int orbId, const ColorRgb &color, int commandType)
|
|
{
|
|
QByteArray bytes;
|
|
|
|
// 5 bytes command-header + 3 bytes color information
|
|
bytes.resize(5 + 3);
|
|
bytes.fill('\0');
|
|
|
|
// Command identifier: C0FFEE
|
|
bytes[0] = static_cast<char>(0xC0);
|
|
bytes[1] = static_cast<char>(0xFF);
|
|
bytes[2] = static_cast<char>(0xEE);
|
|
|
|
// Command type
|
|
bytes[3] = static_cast<char>(commandType);
|
|
|
|
// Orb ID
|
|
bytes[4] = static_cast<char>(orbId);
|
|
|
|
// RED / GREEN / BLUE
|
|
bytes[5] = static_cast<char>(color.red);
|
|
bytes[6] = static_cast<char>(color.green);
|
|
bytes[7] = static_cast<char>(color.blue);
|
|
|
|
sendCommand(bytes);
|
|
}
|
|
|
|
void LedDeviceAtmoOrb::sendCommand(const QByteArray &bytes)
|
|
{
|
|
//Debug ( _log, "command: [%s] -> %s:%u", QSTRING_CSTR( QString(bytes.toHex())), QSTRING_CSTR(_groupAddress.toString()), _multiCastGroupPort );
|
|
_udpSocket->writeDatagram(bytes.data(), bytes.size(), _groupAddress, _multiCastGroupPort);
|
|
}
|
|
|
|
QJsonObject LedDeviceAtmoOrb::discover()
|
|
{
|
|
QJsonObject devicesDiscovered;
|
|
devicesDiscovered.insert("ledDeviceType", _activeDeviceType );
|
|
|
|
QJsonArray deviceList;
|
|
|
|
if ( open() == 0 )
|
|
{
|
|
Debug ( _log, "Send discovery requests to all AtmoOrbs" );
|
|
setColor(0, ColorRgb::BLACK, 8);
|
|
|
|
if ( _udpSocket->waitForReadyRead(DEFAULT_DISCOVERY_TIMEOUT.count()) )
|
|
{
|
|
while (_udpSocket->waitForReadyRead(500))
|
|
{
|
|
QByteArray datagram;
|
|
|
|
while (_udpSocket->hasPendingDatagrams())
|
|
{
|
|
datagram.resize(_udpSocket->pendingDatagramSize());
|
|
QHostAddress senderIP;
|
|
quint16 senderPort;
|
|
|
|
_udpSocket->readDatagram(datagram.data(), datagram.size(), &senderIP, &senderPort);
|
|
|
|
if ( datagram.size() == 1 )
|
|
{
|
|
unsigned char orbId = datagram[0];
|
|
if ( orbId > 0 )
|
|
{
|
|
Debug(_log, "Orb ID (%d) discovered at [%s]", orbId, QSTRING_CSTR(senderIP.toString()));
|
|
_services.insert(orbId, senderIP);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
close();
|
|
}
|
|
|
|
QMap<int, QHostAddress>::iterator i;
|
|
for (i = _services.begin(); i != _services.end(); ++i)
|
|
{
|
|
QJsonObject obj;
|
|
|
|
obj.insert("id", i.key());
|
|
obj.insert("ip", i.value().toString());
|
|
|
|
QHostInfo hostInfo = QHostInfo::fromName(i.value().toString());
|
|
if (hostInfo.error() == QHostInfo::NoError )
|
|
{
|
|
QString hostname = hostInfo.hostName();
|
|
//Seems that for Windows no local domain name is resolved
|
|
if (!hostInfo.localDomainName().isEmpty() )
|
|
{
|
|
obj.insert("hostname", hostname.remove("."+hostInfo.localDomainName()));
|
|
obj.insert("domain", hostInfo.localDomainName());
|
|
}
|
|
else
|
|
{
|
|
int domainPos = hostname.indexOf('.');
|
|
obj.insert("hostname", hostname.left(domainPos));
|
|
obj.insert("domain", hostname.mid(domainPos+1));
|
|
}
|
|
}
|
|
|
|
deviceList << obj;
|
|
}
|
|
|
|
devicesDiscovered.insert("devices", deviceList);
|
|
Debug(_log, "devicesDiscovered: [%s]", QString(QJsonDocument(devicesDiscovered).toJson(QJsonDocument::Compact)).toUtf8().constData() );
|
|
|
|
return devicesDiscovered;
|
|
}
|
|
|
|
void LedDeviceAtmoOrb::identify(const QJsonObject& params)
|
|
{
|
|
//Debug(_log, "params: [%s]", QString(QJsonDocument(params).toJson(QJsonDocument::Compact)).toUtf8().constData());
|
|
|
|
int orbId = 0;
|
|
if ( params["id"].isString() )
|
|
{
|
|
orbId = params["id"].toString().toInt();
|
|
}
|
|
else
|
|
{
|
|
orbId = params["id"].toInt();
|
|
}
|
|
|
|
if ( orbId >0 && orbId < 256 )
|
|
{
|
|
Debug (_log, "Orb ID [%d]", orbId);
|
|
if ( open() == 0 )
|
|
{
|
|
setColor(orbId, ColorRgb::BLACK, 9);
|
|
close();
|
|
}
|
|
}
|
|
else
|
|
{
|
|
Warning(_log, "Identification of Orb with ID='%d' skipped. ID must be in range 1-255", orbId);
|
|
}
|
|
}
|