badge.py: add png support

pimoroni · Jun 21, 2024 · 21e27e8 · 21e27e8
1 parent 9e67d95
commit 21e27e8
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/badger_os/examples/badge.py b/badger_os/examples/badge.py
@@ -1,5 +1,6 @@
 import badger2040
 import jpegdec
+import pngdec
 
 
 # Global Constants
@@ -56,9 +57,14 @@ def draw_badge():
     display.set_pen(0)
     display.clear()
 
-    # Draw badge image
-    jpeg.open_file(badge_image)
-    jpeg.decode(WIDTH - IMAGE_WIDTH, 0)
+    if badge_image.endswith('.png' or '.PNG' or '.Png'):
+        # Draw badge image
+        png.open_file(badge_image)
+        png.decode(WIDTH - IMAGE_WIDTH, 0)
+    else:
+        # Draw badge image
+        jpeg.open_file(badge_image)
+        jpeg.decode(WIDTH - IMAGE_WIDTH, 0)
 
     # Draw a border around the image
     display.set_pen(0)
@@ -123,6 +129,7 @@ def draw_badge():
 display.set_thickness(2)
 
 jpeg = jpegdec.JPEG(display.display)
+png = pngdec.PNG(display.display)
 
 # Open the badge file
 try: