ii-pix/convert.py
2021-01-21 23:33:12 +00:00

102 lines
3.4 KiB
Python

import argparse
import os.path
import time
from PIL import Image
import numpy as np
import dither as dither_pyx
import dither_pattern
import image as image_py
import palette as palette_py
import screen as screen_py
# TODO:
# - support alternate palettes properly
# - compare to bmp2dhr and a2bestpix
# - support LR/DLR
# - support HGR
# - README
def main():
parser = argparse.ArgumentParser()
parser.add_argument("input", type=str, help="Input image file to process.")
parser.add_argument("output", type=str, help="Output file for converted "
"Apple II image.")
parser.add_argument(
"--lookahead", type=int, default=6,
help=("How many pixels to look ahead to compensate for NTSC colour "
"artifacts."))
parser.add_argument(
'--dither', type=str, choices=list(dither_pattern.PATTERNS.keys()),
default=dither_pattern.DEFAULT_PATTERN,
help="Error distribution pattern to apply when dithering.")
parser.add_argument(
'--show_input', action=argparse.BooleanOptionalAction, default=False,
help="Whether to show the input image before conversion.")
parser.add_argument(
'--show_output', action=argparse.BooleanOptionalAction, default=True,
help="Whether to show the output image after conversion.")
parser.add_argument(
'--resolution', type=int, choices=(560, 140), default=560,
help=("Double hi-res resolution to target. 140 treats pixels in "
"groups of 4, with 16 colours that can be chosen independently, "
"and ignores NTSC fringing. 560 treats each pixel individually, "
"with choice of 2 colours (depending on NTSC colour phase), "
"and looking ahead over next --lookahead pixels to optimize the "
"colour sequence.")
)
args = parser.parse_args()
palette = palette_py.Palette()
if args.resolution == 560:
screen = screen_py.DHGR560Screen(palette)
lookahead = args.lookahead
else:
screen = screen_py.DHGR140Screen(palette)
lookahead = 0
image = image_py.open(args.input)
# TODO: better performance with malloc'ed array?
resized = np.array(
image_py.resize(image, screen.X_RES, screen.Y_RES)).astype(np.float32)
if args.show_input:
Image.fromarray(resized.astype(np.uint8)).show()
dither = dither_pattern.PATTERNS[args.dither]()
start = time.time()
output_4bit, output_rgb = dither_pyx.dither_image(
screen, resized, dither, lookahead)
print(time.time() - start)
if args.resolution == 140:
# Show un-fringed 140px output image
out_image = Image.fromarray(image_py.linear_to_srgb(output_rgb).astype(
np.uint8))
if args.show_output:
image_py.resize(out_image, 560, 384, srgb_output=True).show()
bitmap = screen.pack(output_4bit)
output_rgb = screen.bitmap_to_image_rgb(bitmap)
# Show output image
out_image = Image.fromarray(image_py.linear_to_srgb(output_rgb).astype(
np.uint8))
if args.show_output:
# out_image.show()
image_py.resize(out_image, 560, 384, srgb_output=True).show()
outfile = os.path.join(os.path.splitext(args.output)[0] + "_preview.png")
out_image.save(outfile, "PNG")
with open(args.output, "wb") as f:
f.write(bytes(screen.main))
f.write(bytes(screen.aux))
if __name__ == "__main__":
main()