77 lines
2.4 KiB
Python
77 lines
2.4 KiB
Python
import json
|
|
import zipfile
|
|
from ledMatrix import Animation
|
|
|
|
matrixWith = 39
|
|
matrixHeight = 15
|
|
|
|
|
|
def askQuestion():
|
|
global matrixWith, matrixHeight
|
|
while True:
|
|
col = int(input("Column [0-" + str(matrixWith) + "]: "))
|
|
if col <= matrixWith and col >= 0:
|
|
break
|
|
else:
|
|
print("Column Number Must be Between 0 and " + str(matrixWith))
|
|
|
|
while True:
|
|
row = int(input("Row [0-" + str(matrixHeight) + "]: "))
|
|
if row <= matrixHeight and row >= 0:
|
|
break
|
|
else:
|
|
print("Row Number Must be Between 0 and " + str(matrixHeight))
|
|
|
|
print("LED Number: " + str(getLedNumber(col, row)))
|
|
|
|
|
|
def getLedNumber(column, row):
|
|
global matrixWith, matrixHeight
|
|
pcbNum = int(column / 2)
|
|
colNum = column % 2
|
|
if colNum == 0:
|
|
return ((matrixHeight - row) + (pcbNum * ((matrixHeight + 1) * 2)))
|
|
else:
|
|
return (((matrixHeight + 1) + row) + (pcbNum * ((matrixHeight + 1) * 2)))
|
|
|
|
|
|
def compressFileToString(inputFile):
|
|
"""
|
|
read the given open file, compress the data and return it as string.
|
|
"""
|
|
stream = StringIO()
|
|
compressor = gzip.GzipFile(fileobj=stream, mode='w')
|
|
while True: # until EOF
|
|
chunk = inputFile.read(8192)
|
|
if not chunk: # EOF?
|
|
compressor.close()
|
|
return stream.getvalue()
|
|
compressor.write(chunk)
|
|
|
|
|
|
def saveAnimationData(animation, fileName):
|
|
datFile = open(fileName, 'wb')
|
|
for l in range(0, len(animation.matrices)):
|
|
for i in range(0, len(animation.matrices[0].panels)):
|
|
for j in range(0, len(animation.matrices[0].panels[0].leds)):
|
|
datFile.write(
|
|
animation.matrices[l].panels[i].leds[j].r.to_bytes(1, byteorder='big'))
|
|
datFile.write(
|
|
animation.matrices[l].panels[i].leds[j].g.to_bytes(1, byteorder='big'))
|
|
datFile.write(
|
|
animation.matrices[l].panels[i].leds[j].b.to_bytes(1, byteorder='big'))
|
|
datFile.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# askQuestion()
|
|
# mainMatrix = Matrix()
|
|
mainAnimation = Animation()
|
|
|
|
saveAnimationData(mainAnimation, 'testAnimation.dat')
|
|
with zipfile.ZipFile('testAnimation.zip', 'w', zipfile.ZIP_DEFLATED) as myzip:
|
|
myzip.write('testAnimation.dat')
|
|
|
|
# arr = bytearray(mainAnimation)
|
|
# print(arr)
|
|
# file.write(arr)
|