diff options
| -rwxr-xr-x | tools/rle_encode.py | 44 |
1 files changed, 38 insertions, 6 deletions
diff --git a/tools/rle_encode.py b/tools/rle_encode.py index aea7d40..c8c2de6 100755 --- a/tools/rle_encode.py +++ b/tools/rle_encode.py @@ -1,5 +1,6 @@ #!/usr/bin/env python3 +import argparse import sys from PIL import Image @@ -32,6 +33,20 @@ def encode(im): return (im.width, im.height, bytes(rle)) +def render_c(image, fname): + print(f'// 1-bit RLE, generated from {fname}, {len(image[2])} bytes') + print('static const uint8_t rle[] = {\n ', end='') + i = 0 + for rl in image[2]: + print(f' {hex(rl)},', end='') + + i += 1 + if i == 12: + print('\n ', end='') + i = 0 + + print('\n};') + def decode_to_ascii(image): (sx, sy, rle) = image data = bytearray(2*sx) @@ -59,9 +74,26 @@ def decode_to_ascii(image): # Check the image is the correct length assert(dp == 0) -image = encode(Image.open(sys.argv[1])) -# This is kinda cool for testing but let's leave this disabled until we add -# proper argument processing! -#decode_to_ascii(image) -print(f'# 1-bit RLE, generated from {sys.argv[1]}') -print(image) + + +parser = argparse.ArgumentParser(description='RLE encoder tool.') +parser.add_argument('files', nargs='+', + help='files to be encoded') +parser.add_argument('--ascii', action='store_true', + help='Run the resulting image(s) through an ascii art decoder') +parser.add_argument('--c', action='store_true', + help='Render the output as C instead of python') +args = parser.parse_args() + +for fname in args.files: + image = encode(Image.open(fname)) + + if args.c: + render_c(image, fname) + else: + print(f'# 1-bit RLE, generated from {fname}, {len(image[2])} bytes') + print(f'rle = {image}') + + if args.ascii: + print() + decode_to_ascii(image) |