1
0
Fork 0
wasp-os/apps/gallery.py
Adam Blair e494eab608 Wasp A La carte
Configure wasp using a central toml file

Signed-off-by: Adam Blair <adampblair@protonmail.com>
2023-03-19 15:17:18 +00:00

159 lines
6 KiB
Python

# SPDX-License-Identifier: LGPL-3.0-or-later
# Copyright (C) 2021 Francesco Gazzetta
"""Image gallery
~~~~~~~~~~~~~~~~~~~~~~~~~
An application that shows images stored in the filesystem.
.. figure:: res/screenshots/GalleryApp.png
:width: 179
The images have to be uploaded in the "gallery" directory.
The images have to be encoded as BMP RGB565 data in big endian byte order.
To encode them, you can use GIMP (File → Export, select the BMP format,
set "R5 G6 B5" in "Advanced Options"), or ImageMagick:
.. code-block:: sh
convert -define bmp:subtype=RGB565 my_image.png my_image.bmp
And to upload:
.. code-block:: sh
./tools/wasptool --binary --upload my_image.bmp --as gallery/my_image
"""
import wasp
import icons
from apps.system.pager import PagerApp
class GalleryApp():
NAME = 'Gallery'
# 2-bit RLE, 96x64, generated from res/gallery_icon.png, 370 bytes
ICON = (
b'\x02'
b'`@'
b'\x1e@\x81d<d<d;f?X\xec2\xf0/'
b'\xf2-\xf4,\xc3.\xc3,\xc3.\xc3,\xc3.\xc3,'
b'\xc3.\xc3,\xc3\x04\x80\x1d\xa6\x04\xc3,\xc3\x04\xa6\x04'
b'\xc3,\xc3\x04\x83\xc0\xd2\xc2\xa1\x04@\xd7C,C\x04'
b'\x82\xc4\xa0\x04C,C\x04\x81\xc6\x9f\x04C,C\x04'
b'\x82\xc4\xa0\x04C,C\x04\x83\xc2\x9c\x80\xc1\x81\xc0\x1d'
b'\xc4\x04C,C\x04\xe0\x83\xc3\x04C,C\x04\xdf\x85'
b'\xc2\x04C,C\x04\xde\x87\xc1\x04C,C\x04\xdd\x89'
b'\x04C,C\x04\xdc\x8a\x04C,C\x04\xce\x82\xcb\x8b'
b'\x04C,C\x04\xcc\x85\xc9\x8c\x04C+D\x04\xca\x89'
b'\xc6\x8d\x04C*E\x04\xc8\x8d\xc3\x8e\x04C*E\x04'
b'\xc6\x91\xc2\x8d\x04C*E\x04\xc4\x95\xc2\x8b\x04C*'
b'E\x04\xc2\x99\xc2\x89\x04C*E\x04\xa6\x04C*E'
b'\x04\xa6\x04C*E\x04\xa6\x04C+D\x04\xa6\x04C'
b',C\x04\xa6\x04C,C\x04\x86@CB\x9bC\x04'
b'\x80\xd7\x83,\x83\x04\xc0\xc1\xc2I\xd5F\x04\x83,\x83'
b'\x04\xc1L\xcfJ\x04\x83,\x83\x04P\xc7O\x04\x83,'
b'\x83\x04f\x04\x83,\x83\x04f\x04\x83,\x83\x04f\x04'
b'\x83,\x83\x04f\x04\x83,\x83\x04f\x04\x83,\x83\x04'
b'f\x04\x83,\x83\x04f\x04\x83,\x83\x04f\x04\x83,'
b'\x83\x04f\x04\x83,\x83\x04f\x04\x83,\x83.\x83,'
b'\x83.\x83,\x83.\x83,\x83.\x83,\xb4-\xb2/'
b'\xb02\xac?X@\x81f;d<d<d\x1e'
)
def foreground(self):
try:
self.files = wasp.watch.os.listdir("gallery")
except FileNotFoundError:
self.files = []
# In case some images were deleted, we reset the index every time
self.index = 0
self._draw()
wasp.system.request_event(wasp.EventMask.SWIPE_LEFTRIGHT)
def background(self):
# We will read the contents of gallery again on foreground(),
# so let's free some memory
self.files = []
def swipe(self, event):
if event[0] == wasp.EventType.LEFT:
increment = 1
elif event[0] == wasp.EventType.RIGHT:
increment = -1
else:
increment = 0
self.index = (self.index + increment) % len(self.files)
self._draw()
def _invalid_file(self, filename):
draw = wasp.watch.drawable
draw.string('Invalid BMP file', 0, 10, width=240)
draw.blit(self.ICON, 72, 72)
draw.line(72,52, 168,148, 3, 0xf800)
def _draw(self):
draw = wasp.watch.drawable
draw.fill()
if not self.files:
draw.string('No files', 0, 60, width=240)
draw.string('in gallery/', 0, 98, width=240)
else:
filename = self.files[self.index]
# The file name will only show until the image overwrites it,
# so let's put it at the bottom so the user has a chance to see it
draw.string(filename[:(draw.wrap(filename, 240)[1])], 0, 200)
file = open("gallery/{}".format(filename), "rb")
display = wasp.watch.display
# check that we are reading a RGB565 BMP
magic = file.read(2)
if magic != b'BM': # check BMP magic number
self._invalid_file(filename)
return
file.seek(0x0A)
data_offset = int.from_bytes(file.read(4), 'little')
file.seek(0x0E)
dib_len = int.from_bytes(file.read(4), 'little')
if dib_len != 124: # check header V5
self._invalid_file(filename)
return
width = int.from_bytes(file.read(4), 'little')
height = int.from_bytes(file.read(4), 'little')
# width and height are signed, but only height can actually be negative
if height >= 2147483648:
height = 4294967296 - height
bottom_up = False
else: bottom_up = True
if width > 240 or height > 240: # check size <= 240x240
self._invalid_file(filename)
return
file.seek(0x1C)
bit_count = int.from_bytes(file.read(2), 'little')
if bit_count != 16: # check 16 bpp
self._invalid_file(filename)
return
compression = int.from_bytes(file.read(4), 'little')
if compression != 3: # check bitmask mode
self._invalid_file(filename)
return
file.seek(0x36)
bitmask = file.read(4), file.read(4), file.read(4)
if bitmask != (b'\x00\xF8\x00\x00', b'\xE0\x07\x00\x00', b'\x1F\x00\x00\x00'): # check bitmask RGB565
self._invalid_file(filename)
return
display.set_window((240 - width) // 2, 0, width, height)
file.seek(data_offset)
# We don't have enough memory to load the entire image at once, so
# we stream it from flash memory to the display
buf = display.linebuffer[:2*width]
for y in reversed(range(0, height)):
if bottom_up: file.seek(data_offset + y * width * 2)
file.readinto(buf)
for x in range(0, width):
buf[x*2], buf[x*2+1] = buf[x*2+1], buf[x*2]
display.write_data(buf)
file.close()