added functions

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):