1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
import struct
VERSION = 0x0001
MAGIC = 0x4adc0104
PORTOUT = 0x0108
UNI = 0
import pdb
import util.TimeOps as timeops
argDict = {'flags': 0, 'startcode': 0, 'pad':0}
def composePixelStripData(pixelStrip,currentTime=timeops.time()):
packet = bytearray()
for light in pixelStrip:
color = light.state(currentTime)
for channel in color: #skip the last value, its an
#alpha value
packet.append(struct.pack('B', channel))
return packet
# packet = [0]*len(pixelStrip.pixels)*3 #preallocate for speed
# for i in range(len(pixelStrip.pixels)):
#color = pixelStrip.pixels[i].state()
#packet[i:i+2] = color
# return bytearray(packet)
cache = {}
def memoize(f):
def helper(x):
if x not in cache:
cache[x] = f(x)
return cache[x]
return helper
@memoize
def cachePacketHeader(port):
packet = bytearray()
subDict = dict(argDict)
subDict['len'] = 38500 #I have no idea why this works.
subDict['port'] = port
packet.extend(portOutPacket(subDict))
packet.append(0x0)
return packet
def composePixelStripPacket(pixelStrip,port, currentTime):
packet = bytearray(cachePacketHeader(port))
data = composePixelStripData(pixelStrip, currentTime)
packet.extend(data)
return packet
def packheader():
header = bytearray()
header.extend(struct.pack('L', MAGIC))
header.extend(struct.pack('H', VERSION))
header.extend(struct.pack('H', PORTOUT))
header.extend(struct.pack('L', 0))
return header
def portOut():
header = packheader()
header.extend(struct.pack('L', UNI))
return header
def portOutPayload(argDict):
payload = bytearray()
payload.extend(struct.pack('B', argDict['port']))
payload.extend(struct.pack('H', argDict['flags']))
#payload.append(0x00) #somepadding? lolwtf.
payload.extend(struct.pack('H', argDict['len']))
payload.extend(struct.pack('H', argDict['startcode']))
return payload
def portOutPacket(payloadArgs):
packet = bytearray()
packet.extend(portOut())
packet.extend(portOutPayload(payloadArgs))
return packet
|
