#!/usr/bin/env python3
# Uses the pillow Python Imaging Library:
# https://github.com/python-pillow/Pillow
#
# ----------------------------------------------------------------------------
# "THE BEER-WARE LICENSE" (Revision 42):
# <xythobuz@xythobuz.de> wrote this file. As long as you retain this notice
# you can do whatever you want with this stuff. If we meet some day, and you
# think this stuff is worth it, you can buy me a beer in return. Thomas Buck
# ----------------------------------------------------------------------------
from PIL import Image
import time
import os
# GIF images contain Palletes for each frame.
# These can be different between frames.
# So the library is converting them to RGB.
# This means we can not replace the background easily
# for all but the first frame in animations.
# This setting changes that to keep the same palette, as long as possible.
# This works with some GIFs, but not all of them.
from PIL import GifImagePlugin
GifImagePlugin.LOADING_STRATEGY = GifImagePlugin.LoadingStrategy.RGB_AFTER_DIFFERENT_PALETTE_ONLY
class ImageScreen:
def __init__(self, g, p, t = 0.2, i = 1, to = 5.0, bg = None, target_size = None, nearest = True):
self.gui = g
self.time = t
self.iterations = i
self.timeout = to
self.background = bg
scriptDir = os.path.dirname(os.path.realpath(__file__))
self.path = os.path.join(scriptDir, "images", p)
self.image = Image.open(self.path)
# for some reason non-animated images don't even have this attribute
if not hasattr(self.image, "is_animated"):
self.image.is_animated = False
self.image.n_frames = 1
if target_size == None:
target_size = (self.gui.width, self.gui.height)
# automatically crop and scale large images
if not self.image.is_animated and ((self.image.width > target_size[0]) or (self.image.height > target_size[1])):
# crop to visible area
self.image = self.image.crop(self.image.getbbox())
# keep the aspect ratio and fit within visible area
ratio = self.image.width / self.image.height
max_width = int(ratio * target_size[1])
max_height = int(target_size[0] / ratio)
if (max_height >= target_size[1]) or (((target_size[0] - max_width) < (target_size[1] - max_height)) and ((target_size[0] - max_width) >= 0)):
width = max_width
height = target_size[1]
else:
width = target_size[0]
height = max_height
# resize
if nearest:
self.image = self.image.resize((width, height),
Image.Resampling.NEAREST)
else:
self.image = self.image.resize((width, height))
# new image object is also missing these
self.image.is_animated = False
self.image.n_frames = 1
# enlarge small images
if not self.image.is_animated and ((self.image.width * 2) <= target_size[0]) and ((self.image.height * 2) <= target_size[1]):
self.image = self.image.crop(self.image.getbbox())
self.image = self.image.resize((self.image.width * 2, self.image.height * 2),
Image.Resampling.NEAREST)
self.image.is_animated = False
self.image.n_frames = 1
# TODO cropping and scaling not working for GIF animations
print(p, self.image.width, self.image.height, self.image.is_animated, self.image.n_frames, nearest)
self.xOff = int((self.gui.width - self.image.width) / 2)
self.yOff = int((self.gui.height - self.image.height) / 2)
self.restart()
def restart(self):
self.start = time.time()
self.frame = time.time()
self.count = 0
self.done = 0
self.image.seek(0)
def finished(self):
if self.done >= self.iterations:
return True
# only apply timeout for static images
if self.image.is_animated:
return False
else:
return (time.time() - self.start) >= self.timeout
def draw(self):
if self.image.is_animated:
now = time.time()
if (now - self.frame) >= self.time:
self.frame = now
self.count = (self.count + 1) % self.image.n_frames
if self.count == 0:
self.done += 1
self.image.seek(self.count)
p = self.image.getpalette()
for x in range(0, self.image.width):
for y in range(0, self.image.height):
v = self.image.getpixel((x, y))
if isinstance(v, int):
c = None
if self.background != None:
if "transparency" in self.image.info:
if v == self.image.info["transparency"]:
c = self.background
else:
if v == self.image.info["background"]:
c = self.background
if c == None:
c = (p[v * 3 + 0], p[v * 3 + 1], p[v * 3 + 2])
self.gui.set_pixel(x + self.xOff, y + self.yOff, c)
else:
self.gui.set_pixel(x + self.xOff, y + self.yOff, v)
if __name__ == "__main__":
import sys
import util
i = util.getInput()
t = util.getTarget(i)
from manager import Manager
m = Manager(t, i, 1)
scriptDir = os.path.dirname(os.path.realpath(__file__))
imageDir = os.path.join(scriptDir, "images")
for f in os.listdir(os.fsencode(imageDir)):
filename = os.fsdecode(f)
if len(sys.argv) > 1:
if not sys.argv[1] in filename:
continue
for i in range(0, 2):
nearest = (i == 1)
d = ImageScreen(t, filename, 0.2, 1, 2.5, None, None, nearest)
m.add(d)
if filename != "camp23.png":
continue
if (t.width != 32) or (t.height != 32):
continue
try:
# dump generated image to console, for embedding in Pico script
print()
print("Dumping image to img_tmp.py")
with open("img_tmp.py", "w") as f:
f.write("# Image \"" + filename + "\"" + os.linesep)
f.write("# size:" + str(d.image.width) + "x" + str(d.image.height) + os.linesep)
f.write("img_data = [" + os.linesep)
for y in range(0, d.image.height):
f.write(" [" + os.linesep)
for x in range(0, d.image.width):
s = str(d.image.getpixel((x, y)))
f.write(" " + s + "," + os.linesep)
f.write(" ]," + os.linesep)
f.write("]" + os.linesep)
print()
except Exception as e:
print()
if hasattr(sys, "print_exception"):
sys.print_exception(e)
else:
print(e)
print()
m.restart()
util.loop(t, m.draw)