267 lines
7.9 KiB
C++
267 lines
7.9 KiB
C++
/*
|
|
* UDP notifier
|
|
*/
|
|
|
|
#define WLEDPACKETSIZE 24
|
|
#define UDP_IN_MAXSIZE 1472
|
|
|
|
void notify(byte callMode, bool followUp=false)
|
|
{
|
|
if (!udpConnected) return;
|
|
switch (callMode)
|
|
{
|
|
case 0: return;
|
|
case 1: if (!notifyDirect) return; break;
|
|
case 2: if (!notifyButton) return; break;
|
|
case 4: if (!notifyDirect) return; break;
|
|
case 6: if (!notifyDirect) return; break; //fx change
|
|
case 7: if (!notifyHue) return; break;
|
|
case 8: if (!notifyDirect) return; break;
|
|
case 9: if (!notifyDirect) return; break;
|
|
default: return;
|
|
}
|
|
byte udpOut[WLEDPACKETSIZE];
|
|
udpOut[0] = 0; //0: wled notifier protocol 1: WARLS protocol
|
|
udpOut[1] = callMode;
|
|
udpOut[2] = bri;
|
|
udpOut[3] = col[0];
|
|
udpOut[4] = col[1];
|
|
udpOut[5] = col[2];
|
|
udpOut[6] = nightlightActive;
|
|
udpOut[7] = nightlightDelayMins;
|
|
udpOut[8] = effectCurrent;
|
|
udpOut[9] = effectSpeed;
|
|
udpOut[10] = white;
|
|
udpOut[11] = 5; //compatibilityVersionByte: 0: old 1: supports white 2: supports secondary color 3: supports FX intensity, 24 byte packet 4: supports transitionDelay 5: sup palette
|
|
udpOut[12] = colSec[0];
|
|
udpOut[13] = colSec[1];
|
|
udpOut[14] = colSec[2];
|
|
udpOut[15] = whiteSec;
|
|
udpOut[16] = effectIntensity;
|
|
udpOut[17] = (transitionDelay >> 0) & 0xFF;
|
|
udpOut[18] = (transitionDelay >> 8) & 0xFF;
|
|
udpOut[19] = effectPalette;
|
|
|
|
IPAddress broadcastIp;
|
|
broadcastIp = ~WiFi.subnetMask() | WiFi.gatewayIP();
|
|
|
|
notifierUdp.beginPacket(broadcastIp, udpPort);
|
|
notifierUdp.write(udpOut, WLEDPACKETSIZE);
|
|
notifierUdp.endPacket();
|
|
notificationSentCallMode = callMode;
|
|
notificationSentTime = millis();
|
|
notificationTwoRequired = (followUp)? false:notifyTwice;
|
|
}
|
|
|
|
void arlsLock(uint32_t timeoutMs)
|
|
{
|
|
if (!realtimeActive){
|
|
for (uint16_t i = 0; i < ledCount; i++)
|
|
{
|
|
strip.setPixelColor(i,0,0,0,0);
|
|
}
|
|
strip.setMode(0);
|
|
}
|
|
realtimeActive = true;
|
|
realtimeTimeout = millis() + timeoutMs;
|
|
if (arlsForceMaxBri) strip.setBrightness(255);
|
|
}
|
|
|
|
void initE131(){
|
|
if (WiFi.status() == WL_CONNECTED && e131Enabled)
|
|
{
|
|
e131 = new E131();
|
|
e131->begin((e131Multicast) ? E131_MULTICAST : E131_UNICAST , e131Universe);
|
|
} else {
|
|
e131Enabled = false;
|
|
}
|
|
}
|
|
|
|
void handleE131(){
|
|
//E1.31 protocol support
|
|
if(e131Enabled) {
|
|
uint16_t len = e131->parsePacket();
|
|
if (!len || e131->universe < e131Universe || e131->universe > e131Universe +4) return;
|
|
len /= 3; //one LED is 3 DMX channels
|
|
|
|
uint16_t multipacketOffset = (e131->universe - e131Universe)*170; //if more than 170 LEDs (510 channels), client will send in next higher universe
|
|
if (ledCount <= multipacketOffset) return;
|
|
|
|
arlsLock(realtimeTimeoutMs);
|
|
if (len + multipacketOffset > ledCount) len = ledCount - multipacketOffset;
|
|
|
|
for (uint16_t i = 0; i < len; i++) {
|
|
int j = i * 3;
|
|
setRealtimePixel(i + multipacketOffset, e131->data[j], e131->data[j+1], e131->data[j+2], 0);
|
|
}
|
|
strip.show();
|
|
}
|
|
}
|
|
|
|
void handleNotifications()
|
|
{
|
|
//send second notification if enabled
|
|
if(udpConnected && notificationTwoRequired && millis()-notificationSentTime > 250){
|
|
notify(notificationSentCallMode,true);
|
|
}
|
|
|
|
handleE131();
|
|
|
|
//unlock strip when realtime UDP times out
|
|
if (realtimeActive && millis() > realtimeTimeout)
|
|
{
|
|
strip.unlockAll();
|
|
strip.setBrightness(bri);
|
|
realtimeActive = false;
|
|
strip.setMode(effectCurrent);
|
|
realtimeIP[0] = 0;
|
|
}
|
|
|
|
//receive UDP notifications
|
|
if(udpConnected && (receiveNotifications || receiveDirect)){
|
|
uint16_t packetSize = notifierUdp.parsePacket();
|
|
|
|
//hyperion / raw RGB
|
|
if (!packetSize && udpRgbConnected) {
|
|
packetSize = rgbUdp.parsePacket();
|
|
if (!receiveDirect) return;
|
|
if (packetSize > UDP_IN_MAXSIZE || packetSize < 3) return;
|
|
realtimeIP = rgbUdp.remoteIP();
|
|
DEBUG_PRINTLN(rgbUdp.remoteIP());
|
|
olen = 0;
|
|
rgbUdp.read(obuf, packetSize);
|
|
arlsLock(realtimeTimeoutMs);
|
|
uint16_t id = 0;
|
|
for (uint16_t i = 0; i < packetSize -2; i += 3)
|
|
{
|
|
setRealtimePixel(id, obuf[i], obuf[i+1], obuf[i+2], 0);
|
|
|
|
id++; if (id >= ledCount) break;
|
|
}
|
|
strip.show();
|
|
return;
|
|
}
|
|
|
|
if (packetSize > UDP_IN_MAXSIZE) return;
|
|
if(packetSize && notifierUdp.remoteIP() != WiFi.localIP()) //don't process broadcasts we send ourselves
|
|
{
|
|
olen = 0;
|
|
notifierUdp.read(obuf, packetSize);
|
|
char* udpIn = obuf;
|
|
|
|
if (udpIn[0] == 0 && !realtimeActive && receiveNotifications) //wled notifier, block if realtime packets active
|
|
{
|
|
if (receiveNotificationColor)
|
|
{
|
|
col[0] = udpIn[3];
|
|
col[1] = udpIn[4];
|
|
col[2] = udpIn[5];
|
|
}
|
|
if (udpIn[11] > 0 && receiveNotificationColor) //check if sending modules white val is inteded
|
|
{
|
|
white = udpIn[10];
|
|
if (udpIn[11] > 1 )
|
|
{
|
|
colSec[0] = udpIn[12];
|
|
colSec[1] = udpIn[13];
|
|
colSec[2] = udpIn[14];
|
|
whiteSec = udpIn[15];
|
|
}
|
|
}
|
|
if (udpIn[8] != effectCurrent && receiveNotificationEffects)
|
|
{
|
|
effectCurrent = udpIn[8];
|
|
strip.setMode(effectCurrent);
|
|
}
|
|
if (udpIn[9] != effectSpeed && receiveNotificationEffects)
|
|
{
|
|
effectSpeed = udpIn[9];
|
|
strip.setSpeed(effectSpeed);
|
|
}
|
|
if (udpIn[11] > 2 && udpIn[16] != effectIntensity && receiveNotificationEffects)
|
|
{
|
|
effectIntensity = udpIn[16];
|
|
strip.setIntensity(effectIntensity);
|
|
}
|
|
if (udpIn[11] > 3)
|
|
{
|
|
transitionDelayTemp = ((udpIn[17] << 0) & 0xFF) + ((udpIn[18] << 8) & 0xFF00);
|
|
}
|
|
if (udpIn[11] > 4 && udpIn[19] != effectPalette && receiveNotificationEffects)
|
|
{
|
|
effectPalette = udpIn[19];
|
|
strip.setPalette(effectPalette);
|
|
}
|
|
nightlightActive = udpIn[6];
|
|
if (!nightlightActive)
|
|
{
|
|
if (receiveNotificationBrightness) bri = udpIn[2];
|
|
colorUpdated(3);
|
|
}
|
|
} else if (udpIn[0] > 0 && udpIn[0] < 4 && receiveDirect) //1 warls //2 drgb //3 drgbw
|
|
{
|
|
realtimeIP = notifierUdp.remoteIP();
|
|
DEBUG_PRINTLN(notifierUdp.remoteIP());
|
|
if (packetSize > 1) {
|
|
if (udpIn[1] == 0)
|
|
{
|
|
realtimeActive = false;
|
|
return;
|
|
} else {
|
|
arlsLock(udpIn[1]*1000);
|
|
}
|
|
if (udpIn[0] == 1) //warls
|
|
{
|
|
for (uint16_t i = 2; i < packetSize -3; i += 4)
|
|
{
|
|
setRealtimePixel(udpIn[i], udpIn[i+1], udpIn[i+2], udpIn[i+3], 0);
|
|
}
|
|
} else if (udpIn[0] == 2) //drgb
|
|
{
|
|
uint16_t id = 0;
|
|
for (uint16_t i = 2; i < packetSize -2; i += 3)
|
|
{
|
|
setRealtimePixel(id, udpIn[i], udpIn[i+1], udpIn[i+2], 0);
|
|
|
|
id++; if (id >= ledCount) break;
|
|
}
|
|
} else if (udpIn[0] == 3) //drgbw
|
|
{
|
|
uint16_t id = 0;
|
|
for (uint16_t i = 2; i < packetSize -3; i += 4)
|
|
{
|
|
setRealtimePixel(id, udpIn[i], udpIn[i+1], udpIn[i+2], udpIn[i+3]);
|
|
|
|
id++; if (id >= ledCount) break;
|
|
}
|
|
} else if (udpIn[0] == 4) //dnrgb
|
|
{
|
|
uint16_t id = ((udpIn[3] << 0) & 0xFF) + ((udpIn[2] << 8) & 0xFF00);
|
|
for (uint16_t i = 4; i < packetSize -2; i += 3)
|
|
{
|
|
if (id >= ledCount) break;
|
|
setRealtimePixel(id, udpIn[i], udpIn[i+1], udpIn[i+2], udpIn[i+3]);
|
|
id++;
|
|
}
|
|
}
|
|
strip.show();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void setRealtimePixel(uint16_t i, byte r, byte g, byte b, byte w)
|
|
{
|
|
uint16_t pix = i + arlsOffset;
|
|
if (pix < ledCount)
|
|
{
|
|
if (!arlsDisableGammaCorrection && useGammaCorrectionRGB)
|
|
{
|
|
strip.setPixelColor(pix, gamma8[r], gamma8[g], gamma8[b], gamma8[w]);
|
|
} else {
|
|
strip.setPixelColor(pix, r, g, b, w);
|
|
}
|
|
}
|
|
}
|