68 lines
1.8 KiB
Python
68 lines
1.8 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)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
askQuestion()
|
|
# mainMatrix = Matrix()
|
|
mainAnimation = Animation()
|
|
file = open('matrix.json', 'w')
|
|
|
|
matrixJSONData = mainAnimation.toJSON()
|
|
file.write(matrixJSONData)
|
|
file.close()
|
|
with zipfile.ZipFile('matrix.zip', 'w', zipfile.ZIP_DEFLATED) as myzip:
|
|
myzip.write('matrix.json')
|
|
|
|
|
|
# arr = bytearray(mainAnimation)
|
|
# print(arr)
|
|
# file.write(arr)
|
|
|