Badger 2040, tests Ferreol et Capucine

badger2040/libs/image_converter/convert.py

@@ -0,0 +1,136 @@
+#!/usr/bin/env python3
+"""
+Converts images into a format suitable for display on Badger 2040.
+
+Optionally resizes images to 296x128 to fit the display.
+
+Crunches images down to dithered, 1bit colour depth.
+
+Outputs either in raw binary format or as a .py file for embedding into MicroPython.
+
+Output to py functionality is borrwed from data_to_py.py, Copyright (c) 2016 Peter Hinch
+"""
+
+import io
+import argparse
+from PIL import Image, ImageEnhance
+from pathlib import Path
+
+
+PY_HEADER = """# Code generated by convert.py.
+"""
+
+PY_FOOTER = """_mvdata = memoryview(_data)
+
+def data():
+    return _mvdata
+
+"""
+
+
+parser = argparse.ArgumentParser(description='Converts images into the format used by Badger2040.')
+parser.add_argument('file', nargs="+", help='input files to convert')
+parser.add_argument('--out_dir', type=Path, default=None, help='output directory')
+parser.add_argument('--binary', action="store_true", help='output binary file for MicroPython')
+parser.add_argument('--py', action="store_true", help='output .py file for MicroPython embedding')
+parser.add_argument('--resize', action="store_true", help='force images to 296x128 pixels')
+
+options = parser.parse_args()
+
+
+class ByteWriter(object):
+    bytes_per_line = 16
+
+    def __init__(self, stream, varname):
+        self.stream = stream
+        self.stream.write('{} =\\\n'.format(varname))
+        self.bytecount = 0  # For line breaks
+
+    def _eol(self):
+        self.stream.write("'\\\n")
+
+    def _eot(self):
+        self.stream.write("'\n")
+
+    def _bol(self):
+        self.stream.write("b'")
+
+    # Output a single byte
+    def obyte(self, data):
+        if not self.bytecount:
+            self._bol()
+        self.stream.write('\\x{:02x}'.format(data))
+        self.bytecount += 1
+        self.bytecount %= self.bytes_per_line
+        if not self.bytecount:
+            self._eol()
+
+    # Output from a sequence
+    def odata(self, bytelist):
+        for byt in bytelist:
+            self.obyte(byt)
+
+    # ensure a correct final line
+    def eot(self):  # User force EOL if one hasn't occurred
+        if self.bytecount:
+            self._eot()
+        self.stream.write('\n')
+
+
+def convert_image(img):
+    if options.resize:
+        img = img.resize((296, 128))  # resize
+    try:
+        enhancer = ImageEnhance.Contrast(img)
+        img = enhancer.enhance(2.0)
+    except ValueError:
+        pass
+    img = img.convert("1")  # convert to black and white
+    return img
+
+
+def write_stream(header, footer, ip_stream, op_stream):
+    op_stream.write(header)
+    op_stream.write('\n')
+    data = ip_stream.read()
+    bw_data = ByteWriter(op_stream, '_data')
+    bw_data.odata(data)
+    bw_data.eot()
+    op_stream.write(footer)
+
+
+# create map of images based on input filenames
+for input_filename in options.file:
+    with Image.open(input_filename) as img:
+        img = convert_image(img)
+
+        image_name = Path(input_filename).stem
+
+        w, h = img.size
+
+        output_data = [~b & 0xff for b in list(img.tobytes())]
+
+        if options.binary:
+            if options.out_dir is not None:
+                output_filename = (options.out_dir / image_name).with_suffix(".bin")
+            else:
+                output_filename = Path(input_filename).with_suffix(".bin")
+            print(f"Saving to {output_filename}, {w}x{h}")
+            with open(output_filename, "wb") as out:
+                out.write(bytearray(output_data))
+        elif options.py:
+            if options.out_dir is not None:
+                output_filename = (options.out_dir / image_name).with_suffix(".py")
+            else:
+                output_filename = Path(input_filename).with_suffix(".py")
+            print(f"Saving to {output_filename}, {w}x{h}")
+            with open(output_filename, "w") as out:
+                write_stream(PY_HEADER, PY_FOOTER, io.BytesIO(bytes(output_data)), out)
+        else:
+            image_code = '''\
+static const uint8_t {image_name}[{count}] = {{
+    {byte_data}
+}};
+    '''.format(image_name=image_name, count=len(output_data), byte_data=", ".join(str(b) for b in output_data))
+
+            print(image_code)

badger2040/libs/image_converter/data_to_py.py

@@ -0,0 +1,154 @@
+#! /usr/bin/python3
+# -*- coding: utf-8 -*-
+
+# The MIT License (MIT)
+#
+# Copyright (c) 2016 Peter Hinch
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+
+import argparse
+import sys
+import os
+
+# UTILITIES FOR WRITING PYTHON SOURCECODE TO A FILE
+
+# ByteWriter takes as input a variable name and data values and writes
+# Python source to an output stream of the form
+# my_variable = b'\x01\x02\x03\x04\x05\x06\x07\x08'\
+
+# Lines are broken with \ for readability.
+
+
+class ByteWriter(object):
+    bytes_per_line = 16
+
+    def __init__(self, stream, varname):
+        self.stream = stream
+        self.stream.write('{} =\\\n'.format(varname))
+        self.bytecount = 0  # For line breaks
+
+    def _eol(self):
+        self.stream.write("'\\\n")
+
+    def _eot(self):
+        self.stream.write("'\n")
+
+    def _bol(self):
+        self.stream.write("b'")
+
+    # Output a single byte
+    def obyte(self, data):
+        if not self.bytecount:
+            self._bol()
+        self.stream.write('\\x{:02x}'.format(data))
+        self.bytecount += 1
+        self.bytecount %= self.bytes_per_line
+        if not self.bytecount:
+            self._eol()
+
+    # Output from a sequence
+    def odata(self, bytelist):
+        for byt in bytelist:
+            self.obyte(byt)
+
+    # ensure a correct final line
+    def eot(self):  # User force EOL if one hasn't occurred
+        if self.bytecount:
+            self._eot()
+        self.stream.write('\n')
+
+
+# PYTHON FILE WRITING
+
+STR01 = """# Code generated by data_to_py.py.
+version = '0.1'
+"""
+
+STR02 = """_mvdata = memoryview(_data)
+
+def data():
+    return _mvdata
+
+"""
+
+
+def write_func(stream, name, arg):
+    stream.write('def {}():\n    return {}\n\n'.format(name, arg))
+
+
+def write_data(op_path, ip_path):
+    try:
+        with open(ip_path, 'rb') as ip_stream:
+            try:
+                with open(op_path, 'w') as op_stream:
+                    write_stream(ip_stream, op_stream)
+            except OSError:
+                print("Can't open", op_path, 'for writing')
+                return False
+    except OSError:
+        print("Can't open", ip_path)
+        return False
+    return True
+
+
+def write_stream(ip_stream, op_stream):
+    op_stream.write(STR01)
+    op_stream.write('\n')
+    data = ip_stream.read()
+    bw_data = ByteWriter(op_stream, '_data')
+    bw_data.odata(data)
+    bw_data.eot()
+    op_stream.write(STR02)
+
+
+# PARSE COMMAND LINE ARGUMENTS
+
+def quit(msg):
+    print(msg)
+    sys.exit(1)
+
+
+DESC = """data_to_py.py
+Utility to convert an arbitrary binary file to Python source.
+Sample usage:
+data_to_py.py image.jpg image.py
+
+"""
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(__file__, description=DESC,
+                                     formatter_class=argparse.RawDescriptionHelpFormatter)
+    parser.add_argument('infile', type=str, help='Input file path')
+    parser.add_argument('outfile', type=str,
+                        help='Path and name of output file. Must have .py extension.')
+
+    args = parser.parse_args()
+
+    if not os.path.isfile(args.infile):
+        quit("Data filename does not exist")
+
+    if not os.path.splitext(args.outfile)[1].upper() == '.PY':
+        quit('Output filename must have a .py extension.')
+
+    print('Writing Python file.')
+    if not write_data(args.outfile, args.infile):
+        sys.exit(1)
+
+    print(args.outfile, 'written successfully.')