From 55ed1d1d71a4713975d12446f5b429801bf2066a Mon Sep 17 00:00:00 2001
From: Florin Tobler <florin.tobler@gpv-group.com>
Date: Thu, 28 Jul 2022 16:46:48 +0200
Subject: [PATCH] added functions

---
 framebuffer.py | 49 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 49 insertions(+)

diff --git a/framebuffer.py b/framebuffer.py
index d27ddfb..50af118 100644
--- a/framebuffer.py
+++ b/framebuffer.py
@@ -1,4 +1,53 @@
 from PIL import Image
+import numpy
+from utils import *
+
+
+def _read_and_convert_to_ints(filename):
+    with open(filename, "r") as fp:
+        content = fp.read()
+        tokens = content.strip().split(",")
+        return [int(t) for t in tokens if t]
+
+
+def _converter_argb(image: Image):
+    return bytes([x for r, g, b in image.getdata() for x in (255, r, g, b)])
+def _converter_rgb565(image: Image):
+    return bytes([x for r, g, b in image.getdata()
+                  for x in ((g & 0x1c) << 3 | (b >> 3), r & 0xf8 | (g >> 3))])
+def _converter_1_argb(image: Image):
+    return bytes([x for p in image.getdata()
+                  for x in (255, p, p, p)])
+def _converter_1_rgb(image: Image):
+    return bytes([x for p in image.getdata()
+                  for x in (p, p, p)])
+def _converter_1_rgb565(image: Image):
+    return bytes([(255 if x else 0) for p in image.getdata()
+                  for x in (p, p)])
+def _converter_rgba_rgb565_numpy(image: Image):
+    flat = numpy.frombuffer(image.tobytes(), dtype=numpy.uint32)
+    # note,  this is assumes little endian byteorder and results in
+    # the following packing of an integer:
+    # bits 0-7: red, 8-15: green, 16-23: blue, 24-31: alpha
+    flat = ((flat & 0xf8) << 8) | ((flat & 0xfc00) >> 5) | ((flat & 0xf80000) >> 19)
+    return flat.astype(numpy.uint16).tobytes()
+def _converter_no_change(image: Image):
+    return image.tobytes()
+
+# anything that does not use numpy is hopelessly slow
+_CONVERTER = {
+    ("RGBA", 16): _converter_rgba_rgb565_numpy,
+    ("RGB", 16): _converter_rgb565,
+    ("RGB", 24): _converter_no_change,
+    ("RGB", 32): _converter_argb,
+    ("RGBA", 32): _converter_no_change,
+    # note numpy does not work well with mode="1" images as
+    # image.tobytes() loses pixel color info
+    ("1", 16): _converter_1_rgb565,
+    ("1", 24): _converter_1_rgb,
+    ("1", 32): _converter_1_argb,
+}
+
 
 class Framebuffer(object):
     def __init__(self, device_no: int):