Pokitto-Raycasting/convert_png.py
2018-09-27 13:23:27 +02:00

49 lines
963 B
Python

# Convert png image to 256 palette image for Pokitto.
#
# author: Miloslav Ciz
# license: CC0
import sys
from PIL import Image
def findClosest(pixel, palettePixels):
result = 0
closestDist = 255 * 255 * 255
for i in range(256):
p = palettePixels[i % 16,i / 16]
d = abs(p[0] - pixel[0]) + abs(p[1] - pixel[1]) + abs(p[2] - pixel[2])
if d < closestDist:
result = i
closestDist = d
if d == 0:
break
return result
image = Image.open(sys.argv[1])
pixels = image.load()
pal = Image.open(sys.argv[2])
palPixels = pal.load()
sys.stdout.write("const unsigned char image[] =\n{ ")
sys.stdout.write(str(image.size[0]) + ", " + str(image.size[1]) + " // width, height")
count = 0
for x in range(image.size[0]):
for y in range(image.size[1]):
if count % 14 == 0:
sys.stdout.write("\n ");
p = pixels[x,y]
sys.stdout.write(",0x{:02x}".format(findClosest(p,palPixels)))
count += 1
print("\n};")