Badger2040: Add image example, update convert.py

examples/badger2040/image_converter/convert.py

@@ -13,12 +13,14 @@ from pathlib import Path
 parser = argparse.ArgumentParser(description='Converts images into the format used by Badger2040.')
 parser.add_argument('file', nargs="+", help='input files to convert')
 parser.add_argument('--binary', action="store_true", help='output binary file for MicroPython')
+parser.add_argument('--resize', action="store_true", help='force images to 296x128 pixels')
 
 options = parser.parse_args()
 
 
 def convert_image(img):
-    # img = img.resize((296, 128)) # resize and crop
+    if options.resize:
+        img = img.resize((296, 128)) # resize
     enhancer = ImageEnhance.Contrast(img)
     img = enhancer.enhance(2.0)
     img = img.convert("1") # convert to black and white
@@ -37,8 +39,8 @@ for input_filename in options.file:
         output_data = [~b & 0xff for b in list(img.tobytes())]
 
         if options.binary:
-            output_filename = input_filename + ".bin"
-            print(f"Saving to {output_filename}")
+            output_filename = Path(input_filename).with_suffix(".bin")
+            print(f"Saving to {output_filename}, {w}x{h}")
             with open(output_filename, "wb") as out:
                 out.write(bytearray(output_data))
         else:

micropython/examples/badger2040/image.py

@@ -0,0 +1,123 @@
+import os
+import sys
+import time
+import machine
+import badger2040
+from badger2040 import WIDTH, HEIGHT
+
+
+"""
+Images must be 296x128 pixel with 1bit colour depth.
+
+You can use examples/badger200/image_converter/convert.py to convert them:
+
+python3 convert.py --binary --resize image_file_1.png image_file_2.png image_file_3.png
+
+Create a new "images" directory via Thonny, and upload the .bin files there.
+"""
+
+OVERLAY_BORDER = 40
+OVERLAY_SPACING = 20
+OVERLAY_TEXT_SIZE = 0.5
+
+TOTAL_IMAGES = 0
+
+try:
+    IMAGES = [f for f in os.listdir("/images") if f.endswith(".bin")]
+    TOTAL_IMAGES = len(IMAGES)
+except OSError:
+    pass
+
+
+display = badger2040.Badger2040()
+
+button_up = machine.Pin(badger2040.BUTTON_UP, machine.Pin.IN, machine.Pin.PULL_DOWN)
+button_down = machine.Pin(badger2040.BUTTON_DOWN, machine.Pin.IN, machine.Pin.PULL_DOWN)
+
+image = bytearray(int(296 * 128 / 8))
+current_image = 0
+
+
+# Draw an overlay box with a given message within it
+def draw_overlay(message, width, height, line_spacing, text_size):
+
+    # Draw a light grey background
+    display.pen(12)
+    display.rectangle((WIDTH - width) // 2, (HEIGHT - height) // 2, width, height)
+
+    # Take the provided message and split it up into
+    # lines that fit within the specified width
+    words = message.split(" ")
+    lines = []
+    line = ""
+    appended_line = ""
+    for word in words:
+        if len(word) > 0:
+            appended_line += " "
+        appended_line += word
+        if display.measure_text(appended_line, text_size) >= width:
+            lines.append(line)
+            appended_line = word
+        else:
+            line = appended_line
+    if len(line) != 0:
+        lines.append(line)
+
+    display.pen(0)
+    display.thickness(2)
+
+    # Display each line of text from the message, centre-aligned
+    num_lines = len(lines)
+    for i in range(num_lines):
+        length = display.measure_text(lines[i], text_size)
+        current_line = (i * line_spacing) - ((num_lines - 1) * line_spacing) // 2
+        display.text(lines[i], (WIDTH - length) // 2, (HEIGHT // 2) + current_line, text_size)
+
+
+def show_image(n):
+    file = IMAGES[n]
+    name = file.split(".")[0]
+    open("images/{}".format(file), "r").readinto(image)
+    display.image(image)
+    name_length = display.measure_text(name, 0.5)
+    display.pen(0)
+    display.rectangle(0, HEIGHT - 21, name_length + 11, 21)
+    display.pen(15)
+    display.rectangle(0, HEIGHT - 20, name_length + 10, 20)
+    display.pen(0)
+    display.text(name, 5, HEIGHT - 10, 0.5)
+
+    for i in range(TOTAL_IMAGES):
+        x = 286
+        y = int((128 / 2) - (TOTAL_IMAGES * 10 / 2) + (i * 10))
+        display.pen(0)
+        display.rectangle(x, y, 8, 8)
+        if current_image != i:
+            display.pen(15)
+            display.rectangle(x + 1, y + 1, 6, 6)
+
+    display.update()
+
+
+if TOTAL_IMAGES == 0:
+    display.pen(15)
+    display.clear()
+    draw_overlay("To run this demo, create an /images directory on your device and upload some 1bit 296x128 pixel images.", WIDTH - OVERLAY_BORDER, HEIGHT - OVERLAY_BORDER, OVERLAY_SPACING, OVERLAY_TEXT_SIZE)
+    display.update()
+    sys.exit()
+
+
+show_image(current_image)
+
+
+while True:
+    if button_up.value():
+        if current_image > 0:
+            current_image -= 1
+        show_image(current_image)
+    if button_down.value():
+        if current_image < TOTAL_IMAGES:
+            current_image += 1
+        show_image(current_image)
+
+    time.sleep(0.01)