mirror of
https://github.com/weiju/amiga-stuff
synced 2025-12-08 14:58:33 +00:00
79 lines
2.8 KiB
Python
Executable File
79 lines
2.8 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
"""
|
|
For license, see gpl-3.0.txt
|
|
"""
|
|
from PIL import Image
|
|
import argparse
|
|
import math
|
|
import sys
|
|
|
|
def chunks(l, n):
|
|
for i in range(0, len(list(l)), n):
|
|
yield l[i:i+n]
|
|
|
|
def color_to_plane_bits(color, depth):
|
|
"""returns the bits for a given pixel in a list, lowest to highest plane"""
|
|
result = [0] * depth
|
|
for bit in range(depth):
|
|
if color & (1 << bit) != 0:
|
|
result[bit] = 1
|
|
return result
|
|
|
|
def write_amiga_image(image, outfile):
|
|
imdata = im.getdata()
|
|
width, height = im.size
|
|
colors = [i for i in chunks([b for b in im.palette.tobytes()], 3)]
|
|
depth = int(math.log(len(colors), 2))
|
|
|
|
map_words_per_row = int(width / 16)
|
|
if width % 16 > 0:
|
|
map_words_per_row += 1
|
|
|
|
# create the converted planar data
|
|
planes = [[0] * (map_words_per_row * height) for _ in range(depth)]
|
|
for y in range(height):
|
|
x = 0
|
|
while x < width:
|
|
# build a word for each plane
|
|
for i in range(min(16, width - x)):
|
|
# get the palette index for pixel (x + i, y)
|
|
color = imdata[y * width + x + i] # color index
|
|
planebits = color_to_plane_bits(color, depth)
|
|
# now we need to "or" the bits into the words in their respective planes
|
|
wordidx = (x + i) / 16 # word number in current row
|
|
pos = int(y * map_words_per_row + wordidx) # list index in the plane
|
|
for planeidx in range(depth):
|
|
if planebits[planeidx]:
|
|
planes[planeidx][pos] |= (1 << (15 - (x + i) % 16)) # 1 << ((x + i) % 16)
|
|
x += 16
|
|
|
|
outfile.write('#include <intuition/intuition.h>\n\n')
|
|
|
|
outfile.write("/* Ensure that this data is within chip memory or you'll see nothing !!! */\n");
|
|
outfile.write('UWORD __chip imdata[] = {\n')
|
|
indent = 4
|
|
for i, plane in enumerate(planes):
|
|
outfile.write(' ' * indent + '// plane %d\n' % i)
|
|
lines = [(' ' * indent) + ','.join(['0x%04x' % i for i in chunk])
|
|
for chunk in chunks(plane, map_words_per_row)]
|
|
outfile.write(',\n'.join(lines))
|
|
outfile.write('\n')
|
|
outfile.write('};\n\n')
|
|
|
|
|
|
planepick = 2 ** depth - 1 # currently we always use all of the planes
|
|
outfile.write('struct Image image = {\n')
|
|
outfile.write(' ' * indent + '0, 0, %d, %d, %d, imdata,\n' % (width, height, depth));
|
|
outfile.write(' ' * indent + '%d, 0, NULL\n' % planepick) # PlanePick, PlaneOnOff
|
|
outfile.write('};\n')
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(description='Amiga Image converter')
|
|
parser.add_argument('pngfile')
|
|
parser.add_argument('--plain', default=False)
|
|
|
|
args = parser.parse_args()
|
|
im = Image.open(args.pngfile)
|
|
write_amiga_image(im, sys.stdout)
|