Commit 328b4fb8 authored by schneider's avatar schneider
Browse files

fix(pycardium): Fix code style of some py/ modules

parent 0a00d50c
...@@ -75,7 +75,9 @@ def kitt( ...@@ -75,7 +75,9 @@ def kitt(
# set the color values to the LEDs by multiplying the given color # set the color values to the LEDs by multiplying the given color
# value with the corresponding brightness value in the kitt table # value with the corresponding brightness value in the kitt table
output = [[int(x * y) for y in rgb] for x in kitt_table[j : (j + used_leds)]] output = [
[int(x * y) for y in rgb] for x in kitt_table[j : (j + used_leds)]
]
else: else:
used_leds = len(spectrum) used_leds = len(spectrum)
......
#this script reads a BMP file and displays it as POV. For optimal results use a BMP that is 11 pixels wide. # this script reads a BMP file and displays it as POV. For optimal results use a BMP that is 11 pixels wide.
import leds import leds
#default filename. Depending on how you want to run this, change it
# default filename. Depending on how you want to run this, change it
FILENAME = "Smiley.bmp" FILENAME = "Smiley.bmp"
#Determines the Frame Rate, currently not used. # Determines the Frame Rate, currently not used.
DELAY = 0 DELAY = 0
#Determine Brightness Level. You might not wanna use the full values from the BMP # Determine Brightness Level. You might not wanna use the full values from the BMP
BRIGHTNESS = 1.0 BRIGHTNESS = 1.0
class BMPError(Exception): class BMPError(Exception):
pass pass
class POV: class POV:
def __init__(self, filename=FILENAME): def __init__(self, filename=FILENAME):
self.filename = filename self.filename = filename
f = open("/" + self.filename, "rb") f = open("/" + self.filename, "rb")
if f.read(2) != b'BM': # check signature if f.read(2) != b"BM": # check signature
raise BMPError("Not BitMap file") raise BMPError("Not BitMap file")
#turn into int.from_bytes # turn into int.from_bytes
bmpFileSize = int.from_bytes(f.read(4), 'little') bmpFileSize = int.from_bytes(f.read(4), "little")
f.read(4) # Read & ignore creator bytes f.read(4) # Read & ignore creator bytes
self.bmpImageOffset = self.read_le(f.read(4)) # Start of image data self.bmpImageOffset = self.read_le(f.read(4)) # Start of image data
...@@ -29,8 +30,10 @@ class POV: ...@@ -29,8 +30,10 @@ class POV:
self.bmpHeight = self.read_le(f.read(4)) self.bmpHeight = self.read_le(f.read(4))
flip = False flip = False
print("Size: %d\nImage offset: %d\nHeader size: %d" % print(
(bmpFileSize, self.bmpImageOffset, headerSize)) "Size: %d\nImage offset: %d\nHeader size: %d"
% (bmpFileSize, self.bmpImageOffset, headerSize)
)
print("Width: %d\nHeight: %d" % (self.bmpWidth, self.bmpHeight)) print("Width: %d\nHeight: %d" % (self.bmpWidth, self.bmpHeight))
if self.read_le(f.read(2)) != 1: if self.read_le(f.read(2)) != 1:
...@@ -45,21 +48,25 @@ class POV: ...@@ -45,21 +48,25 @@ class POV:
print("Image OK!") print("Image OK!")
self.rowSize = (self.bmpWidth * 3 + 3) & ~3 # 32-bit line boundary self.rowSize = (self.bmpWidth * 3 + 3) & ~3 # 32-bit line boundary
self.fileOffset = f.tell() self.fileOffset = f.tell()
print("File Offset"+str(self.fileOffset)) print("File Offset" + str(self.fileOffset))
f.close() f.close()
self.read_image() self.read_image()
def correct_brightness(self, color): def correct_brightness(self, color):
return int(pow((color * BRIGHTNESS) / 255, 2.7) * 255 + 0.5) return int(pow((color * BRIGHTNESS) / 255, 2.7) * 255 + 0.5)
def getWidth(self): def getWidth(self):
return self.bmpWidth return self.bmpWidth
def setFileOffset(self, fileOffset): def setFileOffset(self, fileOffset):
self.fileOffset = fileOffset self.fileOffset = fileOffset
def getFileOffset(self): def getFileOffset(self):
return self.fileOffset return self.fileOffset
def read_le(self, bytes): def read_le(self, bytes):
return int.from_bytes(bytes, 'little') return int.from_bytes(bytes, "little")
def read_image(self): def read_image(self):
self.imageArray = [] self.imageArray = []
flip = True flip = True
...@@ -75,7 +82,7 @@ class POV: ...@@ -75,7 +82,7 @@ class POV:
for col in range(self.bmpWidth): for col in range(self.bmpWidth):
rgb = [f.read(1), f.read(1), f.read(1)] rgb = [f.read(1), f.read(1), f.read(1)]
rgb = [int.from_bytes(x, "little") for x in rgb] rgb = [int.from_bytes(x, "little") for x in rgb]
#rgb = [self.correct_brightness(x) for x in rgb] # rgb = [self.correct_brightness(x) for x in rgb]
rowArray += [rgb] rowArray += [rgb]
self.imageArray += [rowArray] self.imageArray += [rowArray]
...@@ -86,6 +93,7 @@ class POV: ...@@ -86,6 +93,7 @@ class POV:
for i in self.imageArray: for i in self.imageArray:
leds.set_all(i) leds.set_all(i)
def play(imagename): def play(imagename):
myPov = POV(imagename) myPov = POV(imagename)
while True: while True:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment