159 lines
6 KiB
Python
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/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.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()
|