From 007c51567ad0ed4ccb4488753aec3cc7f7dbb395 Mon Sep 17 00:00:00 2001
From: Hel Gibbons <hel@pimoroni.com>
Date: Fri, 21 Jun 2024 15:29:42 +0100
Subject: [PATCH] badge.py: add png support

badge.py: add png support
---
 badger_os/examples/badge.py | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/badger_os/examples/badge.py b/badger_os/examples/badge.py
index 8b8ec77..ed986e8 100644
--- 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)
+    try:
+        # Draw badge image
+        png.open_file(badge_image)
+        png.decode(WIDTH - IMAGE_WIDTH, 0)
+    except (OSError, RuntimeError):
+        # 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: