Simple image display

tools/bitplanify.py

@@ -0,0 +1,79 @@
+#!/usr/bin/env python
+
+import sys
+import math
+from PIL import Image
+from bitmap import (
+    image_to_bitplanes,
+    write_bitplane,
+    write_interleaved,
+    to_amiga_colors,
+)
+
+
+def main(args):
+    source_file = args.source
+    bit_depth = args.depth
+    verbose = args.verbose
+
+    img = Image.open(source_file)
+    if img.mode != 'P':
+        print 'Image is not palette-based'
+        sys.exit(1)
+
+    palette = img.getpalette()  # always 256*3 values
+    if not bit_depth:
+        max_index = img.getextrema()[1]
+        bit_depth = int(math.log(max_index) / math.log(2)) + 1
+        if verbose:
+            print 'Maximum palette index used: %d: assuming bit depth %d' % (max_index, bit_depth)
+
+    amiga_palette = to_amiga_colors(
+        palette[:(1 << bit_depth) * 3]
+    )
+
+    width, height = img.size
+    if (width & 7) != 0:
+        print 'Width is not divisable by 8:', width
+        sys.exit(1)
+
+    bitmap_image = image_to_bitplanes(img, bit_depth)
+    if verbose:
+        print 'Image width: %d pixels = %d bytes. Height: %d pixels' % (
+            width, bitmap_image.byte_width, height)
+
+    if args.separate:
+        for plane_index in range(bit_depth):
+            with open(args.out % (plane_index + 1), 'wb') as out:
+                write_bitplane(out, bitmap_image, plane_index)
+    else:
+        with open(args.out, 'wb') as out:
+            write_interleaved(out, bitmap_image)
+
+    if args.copper:
+        with open(args.copper, 'w') as out:
+            for index, rgb in enumerate(amiga_palette):
+                out.write('\tdc.w\t$%x,$%03x\n' % (0x180 + index * 2, rgb))
+
+if __name__ == '__main__':
+    import argparse
+    parser = argparse.ArgumentParser(description='Convert an image to Amiga bitplanes')
+    parser.add_argument(
+        '--depth', metavar='N', type=int, default=None,
+        help='Set number of bitplanes in output')
+    parser.add_argument(
+        '--copper',
+        help='Write palette as copper source code (default is to auto-detect)')
+    parser.add_argument(
+        '--separate', action='store_true',
+        help='Write one bitplane per file. Use %%s in filename as replacement for bitplane number.')
+    parser.add_argument(
+        '--verbose', '-v', action='store_true',
+        help='Write more information to stdout')
+    parser.add_argument('source', metavar='IMAGE_FILE',
+        help='Image file to convert to bitplanes')
+    parser.add_argument('out', metavar='BITPLANES_FILE',
+        help='File to write bitmaps to')
+
+    args = parser.parse_args()
+    main(args)