boards: simulator: Add an screenshot facility

The screenshot is automatically named after the application currently
running and copied into the res/ directory. This allows the application
screenshots to be quickly updates if/when the screenshots get out of date
as the applications are improved.

Signed-off-by: Daniel Thompson <daniel@redfelineninja.org.uk>

1 files changed, 28 insertions, 1 deletions

diff --git a/wasp/boards/simulator/display.py b/wasp/boards/simulator/display.py
index 260173c..144004f 100644
--- a/wasp/boards/simulator/display.py
+++ b/wasp/boards/simulator/display.py
@@ -6,6 +6,9 @@
 import sys
 import sdl2
 import sdl2.ext
+import numpy as np
+from PIL import Image
+import wasp
 
 CASET = 0x2a
 RASET = 0x2b
@@ -189,6 +192,26 @@ window.refresh()
 spi_st7789_sim = ST7789Sim()
 i2c_cst816s_sim = CST816SSim()
 
+def save_image(surface, fname):
+    """Save a surface as an image."""
+    raw = sdl2.ext.pixels2d(surface)
+
+    # Crop and swap the axes to ensure the final rotation is correct
+    cropped = raw[SKIN['top_pad']:-SKIN['bottom_pad']]
+    cropped = np.swapaxes(cropped, 0, 1)
+    cropped = cropped[SKIN['left_pad']:-SKIN['right_pad']]
+
+    # Split into r, g and b
+    r = cropped >> 16
+    g = (cropped >> 8) & 0xff
+    b = cropped & 0xff
+
+    # Combine into the final pixel data
+    rgb = np.uint8(np.dstack((r, g, b)))
+
+    # Save the image
+    Image.fromarray(rgb).save(fname)
+
 def tick(pins):
     events = sdl2.ext.get_events()
     for event in events:
@@ -200,7 +223,11 @@ def tick(pins):
         elif event.type == sdl2.SDL_MOUSEBUTTONUP:
             i2c_cst816s_sim.handle_mousebuttonup(event.button, pins)
         elif event.type == sdl2.SDL_KEYDOWN:
-            if event.key.keysym.sym == sdl2.SDLK_TAB:
+            if event.key.keysym.sym == sdl2.SDLK_s:
+                fname = f'res/{wasp.system.app.NAME}App.png'.replace(' ', '')
+                save_image(windowsurface, fname)
+                print(f'Saved: {fname}')
+            elif event.key.keysym.sym == sdl2.SDLK_TAB:
                 pins['BUTTON'].value(0)
             else:
                 i2c_cst816s_sim.handle_key(event.key, pins)