2020-12-29 18:24:29 +00:00
|
|
|
import argparse
|
2021-01-08 22:44:28 +00:00
|
|
|
import os.path
|
2021-01-12 10:00:56 +00:00
|
|
|
import time
|
2020-12-29 18:24:29 +00:00
|
|
|
|
2021-01-03 22:32:04 +00:00
|
|
|
from PIL import Image
|
2020-12-29 18:24:29 +00:00
|
|
|
import numpy as np
|
2021-01-15 22:18:25 +00:00
|
|
|
|
2021-01-15 22:20:28 +00:00
|
|
|
import dither as dither_pyx
|
2021-01-15 22:18:25 +00:00
|
|
|
import dither_pattern
|
|
|
|
import image as image_py
|
|
|
|
import palette as palette_py
|
|
|
|
import screen as screen_py
|
2021-01-03 23:23:15 +00:00
|
|
|
|
2021-01-11 20:21:00 +00:00
|
|
|
|
2020-12-29 21:03:17 +00:00
|
|
|
# TODO:
|
2021-01-15 22:58:01 +00:00
|
|
|
# - include fringing in 140px output
|
2020-12-30 10:27:33 +00:00
|
|
|
# - compare to bmp2dhr and a2bestpix
|
2021-01-15 22:34:03 +00:00
|
|
|
# - support LR/DLR
|
|
|
|
# - support HGR
|
2021-01-03 22:32:04 +00:00
|
|
|
|
2020-12-30 10:27:33 +00:00
|
|
|
|
2020-12-29 18:24:29 +00:00
|
|
|
def main():
|
|
|
|
parser = argparse.ArgumentParser()
|
2021-01-15 22:28:44 +00:00
|
|
|
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.")
|
2021-01-08 22:44:28 +00:00
|
|
|
parser.add_argument(
|
2021-01-15 22:25:32 +00:00
|
|
|
"--lookahead", type=int, default=6,
|
2021-01-08 22:44:28 +00:00
|
|
|
help=("How many pixels to look ahead to compensate for NTSC colour "
|
|
|
|
"artifacts."))
|
2021-01-15 22:28:44 +00:00
|
|
|
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.")
|
2021-01-15 22:34:03 +00:00
|
|
|
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.")
|
2021-01-15 22:58:01 +00:00
|
|
|
parser.add_argument(
|
|
|
|
'--resolution', type=int, choices=(560, 140), default=560,
|
|
|
|
help=("Double hi-res resolution to target. 140 treats pixels as "
|
|
|
|
"groups of 4 and ignores NTSC fringing.")
|
|
|
|
)
|
2021-01-08 22:44:28 +00:00
|
|
|
args = parser.parse_args()
|
|
|
|
|
2021-01-15 22:18:25 +00:00
|
|
|
palette = palette_py.Palette()
|
2021-01-15 22:58:01 +00:00
|
|
|
if args.resolution == 560:
|
|
|
|
screen = screen_py.DHGR560Screen(palette)
|
|
|
|
lookahead = args.lookahead
|
|
|
|
else:
|
|
|
|
screen = screen_py.DHGR140Screen(palette)
|
|
|
|
lookahead = 0
|
2020-12-30 10:27:33 +00:00
|
|
|
|
2021-01-15 22:18:25 +00:00
|
|
|
image = image_py.open(screen.X_RES, screen.Y_RES, args.input)
|
2021-01-15 22:34:03 +00:00
|
|
|
if args.show_input:
|
|
|
|
Image.fromarray(image.astype(np.uint8)).show()
|
2021-01-03 22:32:04 +00:00
|
|
|
|
2021-01-15 22:25:06 +00:00
|
|
|
dither = dither_pattern.PATTERNS[args.dither]()
|
2020-12-30 10:27:33 +00:00
|
|
|
|
2021-01-12 10:00:56 +00:00
|
|
|
start = time.time()
|
2021-01-15 22:20:28 +00:00
|
|
|
output_4bit, output_rgb = dither_pyx.dither_image(
|
2021-01-15 22:58:01 +00:00
|
|
|
screen, image, dither, lookahead)
|
2021-01-12 10:00:56 +00:00
|
|
|
print(time.time() - start)
|
2021-01-10 16:06:14 +00:00
|
|
|
screen.pack(output_4bit)
|
2020-12-30 10:27:33 +00:00
|
|
|
|
2021-01-15 22:18:25 +00:00
|
|
|
out_image = Image.fromarray(image_py.linear_to_srgb(output_rgb).astype(
|
|
|
|
np.uint8))
|
2021-01-08 22:44:28 +00:00
|
|
|
outfile = os.path.join(os.path.splitext(args.output)[0] + ".png")
|
|
|
|
out_image.save(outfile, "PNG")
|
2021-01-15 22:34:03 +00:00
|
|
|
if args.show_output:
|
|
|
|
out_image.show()
|
2021-01-04 21:08:29 +00:00
|
|
|
|
2021-01-03 22:32:04 +00:00
|
|
|
with open(args.output, "wb") as f:
|
2021-01-03 23:23:15 +00:00
|
|
|
f.write(bytes(screen.main))
|
|
|
|
f.write(bytes(screen.aux))
|
2020-12-29 18:24:29 +00:00
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
2021-01-09 18:05:36 +00:00
|
|
|
main()
|