// Local-Hyperion includes #include "LedDeviceAtmoOrb.h" #include // qt includes #include #include #include #include #include // Constants namespace { const bool verbose = false; const bool verbose3 = false; 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["host"].toString(MULTICAST_GROUP_DEFAULT_ADDRESS); _multiCastGroupPort = static_cast(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 : %d", 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)); } } int numberOrbs = _orbIds.size(); int 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 [%d] than configured LEDs [%d].", 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 &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(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(0xC0); bytes[1] = static_cast(0xFF); bytes[2] = static_cast(0xEE); // Command type bytes[3] = static_cast(commandType); // Orb ID bytes[4] = static_cast(orbId); // RED / GREEN / BLUE bytes[5] = static_cast(color.red); bytes[6] = static_cast(color.green); bytes[7] = static_cast(color.blue); sendCommand(bytes); } void LedDeviceAtmoOrb::sendCommand(const QByteArray &bytes) { DebugIf(verbose3, _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(const QJsonObject& params) { DebugIf(verbose, _log, "params: [%s]", QString(QJsonDocument(params).toJson(QJsonDocument::Compact)).toUtf8().constData()); QJsonObject devicesDiscovered; devicesDiscovered.insert("ledDeviceType", _activeDeviceType ); QJsonArray deviceList; _multicastGroup = params["multiCastGroup"].toString(MULTICAST_GROUP_DEFAULT_ADDRESS); _multiCastGroupPort = static_cast(params["multiCastPort"].toInt(MULTICAST_GROUP_DEFAULT_PORT)); if ( open() == 0 ) { Debug ( _log, "Send discovery requests to all AtmoOrbs listening to %s:%d", QSTRING_CSTR(_multicastGroup),_multiCastGroupPort ); 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::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); DebugIf(verbose, _log, "devicesDiscovered: [%s]", QString(QJsonDocument(devicesDiscovered).toJson(QJsonDocument::Compact)).toUtf8().constData() ); return devicesDiscovered; } void LedDeviceAtmoOrb::identify(const QJsonObject& params) { DebugIf(verbose, _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); } }