supercasiobros/levelconverter/levelconv.py

109 lines
2.7 KiB
Python

import os
import tempfile
import subprocess
from PIL import Image
# Palette de couleurs
pierre = (0, 0, 0)
brique = (255, 0, 0)
empty = (255, 255, 255)
piece = (203, 255, 0)
boite_piece = (255, 153, 0)
boite_champi = (204, 0, 255)
brique_piece = (101, 127, 0)
beton = (84, 84, 84)
tuyau_milieu = (0, 255, 102)
tuyau_bout = (50, 255, 0)
drapeau = (0, 255, 216)
goomba = (127, 76, 0)
koopa_vert = (25, 127, 0)
koopa_rouge = (127, 0, 0)
def color_compare(color1, color2):
if color1[0] == color2[0] and color1[1] == color2[1] and color1[2] == color2[2]:
return 1
else:
return 0
def write_char(val):
a = val%256
return a.to_bytes(1,'little')
#if a>128:
# a -= 256
# return chr(a)
# return '%c' % int(val%256)
# Load image
filename = input("File name ?\n> ")
img = Image.open(filename)
print("Converting " + filename + " to assets-fx/" + filename + " //size =", img.size)
#rough binary
# code = str().encode('latin1')
code = write_char(int(img.size[0]))
code += write_char(int(img.size[1]))
nombre_ennemis=0
ennemis = "\nennemi_t ennemies0[]={"
# Balayage des pixels : passe pour dessiner les murs
pixels = img.load()
for x in range(0,img.size[0]):
for i in range(0, img.size[1]):
y=img.size[1]-i-1
if color_compare(pixels[x,y],empty):
code += write_char(0)
elif color_compare(pixels[x,y], pierre):
code += write_char(1)
elif color_compare(pixels[x,y], piece):
code += write_char(2)
elif color_compare(pixels[x,y], brique):
code += write_char(3)
elif color_compare(pixels[x,y],brique_piece):
code += write_char(4)
elif color_compare(pixels[x,y],boite_piece):
code += write_char(5)
elif color_compare(pixels[x,y],boite_champi):
code += write_char(6)
elif color_compare(pixels[x,y],beton):
code += write_char(7)
elif color_compare(pixels[x,y],tuyau_bout):
code += write_char(8)
elif color_compare(pixels[x,y], tuyau_milieu):
code += write_char(9)
elif color_compare(pixels[x,y], drapeau):
code += write_char(10) # temporary code+= "{END_LEVEL,0x1102},"
else:
code += write_char(0)
#code += str(pixels[x,y])
if color_compare(pixels[x,y], goomba):
nombre_ennemis += 1
ennemis += "\n GOOMBA(" + str(8*x) + ", " + str(8*i) + ", -1),"
elif color_compare(pixels[x,y], koopa_vert):
ennemis += "\n KOOPA_V(" + str(8*x) + ", " + str(8*i) + ", -1),"
elif color_compare(pixels[x,y], koopa_rouge):
ennemis += "\n KOOPA_R(" + str(8*x) + ", " + str(8*i) + ", -1),"
ennemis += "\n};\nennemis_global_size=" + str(nombre_ennemis) + ";\ninit_ennemies(ennemies0);\n"
f = open("../assets-fx/bin/lvl_"+filename, 'wb')
f.write(code)
f.close()
print("Converted succesfully ! (", os.path.getsize("../assets-fx/bin/lvl_"+filename), "bytes )")