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

Merge pull request #12 from FoamyGuy/fix_for_board_without_display

Browse source 
fix error from board.DISPLAY not existing
This commit is contained in:
Melissa LeBlanc-Williams 2020-08-10 20:05:50 -07:00 committed by GitHub
commit 043f61447b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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")