jepler/Adafruit_CircuitPython_BitmapSaver
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

fix error from board.DISPLAY not existing

Browse source
This commit is contained in:
FoamyGuy 2020-07-03 09:28:55 -05:00
parent fbe707d184
commit 3fb0d561a2
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
adafruit_bitmapsaver.py
View file

@ -121,7 +121,7 @@ def _write_pixels(output_file, pixel_source, palette):
# pylint:enable=too-many-locals
def save_pixels(file_or_filename, pixel_source=board.DISPLAY, palette=None):
def save_pixels(file_or_filename, pixel_source=None, palette=None):
"""Save pixels to a 24 bit per pixel BMP file.
If pixel_source if a displayio.Bitmap, save it's pixels through palette.
If it's a displayio.Display, a palette isn't required.
@ -130,6 +130,11 @@ def save_pixels(file_or_filename, pixel_source=board.DISPLAY, palette=None):
:param pixel_source: the Bitmap or Display to save
:param palette: the Palette to use for looking up colors in the bitmap
"""
if not pixel_source:
if "DISPLAY" in dir(board):
pixel_source = board.DISPLAY
else:
raise ValueError("Second argument must be a Bitmap or Display")
if isinstance(pixel_source, Bitmap):
if not isinstance(palette, Palette):
raise ValueError("Third argument must be a Palette for a Bitmap save")