Pokitto-Raycasting/convert_png.py

# 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};")