64 lines
1.6 KiB
Python
64 lines
1.6 KiB
Python
import hyperion
|
|
import time
|
|
import colorsys
|
|
import socket
|
|
import errno
|
|
import struct
|
|
|
|
# Get the parameters
|
|
ListenPort = int(hyperion.args.get('ListenPort', 2801))
|
|
ListenIP = hyperion.args.get('ListenIP', "")
|
|
octets = ListenIP.split('.');
|
|
|
|
UDPSock = socket.socket(socket.AF_INET,socket.SOCK_DGRAM, socket.IPPROTO_UDP)
|
|
UDPSock.setblocking(False)
|
|
|
|
listen_addr = (ListenIP,ListenPort)
|
|
UDPSock.bind(listen_addr)
|
|
|
|
if ListenIP == "":
|
|
print "udp.py: Listening on *.*.*.*:"+str(ListenPort)
|
|
else:
|
|
print "udp.py: Listening on "+ListenIP+":"+str(ListenPort)
|
|
|
|
if len(octets) == 4 and int(octets[0]) >= 224 and int(octets[0]) < 240:
|
|
print "ListenIP is a multicast address\n"
|
|
# Multicast handling
|
|
try:
|
|
mreq = struct.pack("4sl", socket.inet_aton(ListenIP), socket.INADDR_ANY)
|
|
UDPSock.setsockopt(socket.IPPROTO_IP, socket.IP_ADD_MEMBERSHIP, mreq)
|
|
except socket.error:
|
|
print "ERROR enabling multicast\n"
|
|
|
|
hyperion.setColor(hyperion.ledCount * bytearray((int(0), int(0), int(0))) )
|
|
|
|
# Start the write data loop
|
|
while not hyperion.abort():
|
|
try:
|
|
data,addr = UDPSock.recvfrom(4500)
|
|
# print data.strip(),len(data),addr
|
|
if (len(data)%3 == 0):
|
|
# print "numleds ",len(data)/3
|
|
ledData = bytearray()
|
|
for i in range(hyperion.ledCount):
|
|
if (i<(len(data)/3)):
|
|
ledData += data[i*3+0]
|
|
ledData += data[i*3+1]
|
|
ledData += data[i*3+2]
|
|
else:
|
|
ledData += bytearray((int(0), int(0), int(0)))
|
|
|
|
hyperion.setColor(ledData)
|
|
|
|
else:
|
|
print "not div 3"
|
|
except IOError as e:
|
|
if e.errno == errno.EWOULDBLOCK:
|
|
pass
|
|
else:
|
|
print "errno:", e.errno
|
|
|
|
print "udp.py: closing socket"
|
|
UDPSock.close()
|
|
|