Transport Tycoon Forums

Code: Select all

import Image
import ImageDraw
spritesheet = Image.open('test_input.png')
spritesheetpx = spritesheet.load()
draw = ImageDraw.Draw(spritesheet)
colours = {}
for x in range(spritesheet.size[0]):
  for y in range(spritesheet.size[1]):
    colour = spritesheetpx[x,y]
    if colour != 255 and colour != 0 and colour != 15:
      colours[colour] = ''
    if spritesheetpx[x,y] == 240:
      if x%2 == 0:
        body_colour = 10
        cc_colour = 200
      else:
        body_colour = 11
        cc_colour = 201
      draw.point([(x,y)],fill=body_colour)
      draw.point([(x,y-1)],fill=cc_colour)
      draw.point([(x,y-2)],fill=body_colour)
      draw.point([(x,y-3)],fill=body_colour)
      draw.point([(x,y-4)],fill=14)
    if spritesheetpx[x,y] == 238:
      if x%2 == 0:
        body_colour = 11
        cc_colour = 201
      else:
        body_colour = 12
        cc_colour = 202
      draw.point([(x,y)],fill=body_colour)
      draw.point([(x,y-1)],fill=cc_colour)
      draw.point([(x,y-2)],fill=body_colour)
      draw.point([(x,y-3)],fill=body_colour)
      draw.point([(x,y-4)],fill=14)
    if spritesheetpx[x,y] == 209:
      draw.point([(x,y)],fill=16)
      draw.point([(x,y-1)],fill=17)
      draw.point([(x,y-2)],fill=18)
      draw.point([(x,y-3)],fill=19)
      draw.point([(x,y-4)],fill=14)
spritesheet.save('a_test_trailer.png')
print colours

Code: Select all

import Image
import ImageDraw
spritesheet = Image.open('test_input.png')
spritesheetpx = spritesheet.load()
draw = ImageDraw.Draw(spritesheet)

# global constants
body_colour = 10
cc_colour = 202

#each sequence is a tuple containing lists in format: [(x-offset,y-offset), colour]
pixel_sequences = dict (
    body_outer = ([(0, 0), body_colour], [(0, 1), cc_colour], [(0, 2), body_colour], [(0, 3), body_colour], [(0, 4), 14]),
    body_end   = ([(0, 0), body_colour], [(0, 1), cc_colour], [(0, 2), body_colour], [(0, 3), body_colour], [(0, 4), 14]),
    body_inner = ([(0, 0), 16], [(0, 1), 17], [(0, 2), 18], [(0, 3), 19], [(0, 4), 14]),
)

key_colour_mapping = {
    209 : dict(seq = 'body_inner', colour_shift= 0),
    238 : dict(seq = 'body_outer', colour_shift = 0),
    240 : dict(seq = 'body_end', colour_shift = -1),
}

def get_pixel_sequence(x, y, key_colour):
    key_map = key_colour_mapping[key_colour]
    raw_sequence = pixel_sequences[key_map['seq']]
    pixel_sequence = []
    for i in raw_sequence:
        pixel_sequence.append({
            'x' : x + i[0][0],
            'y' : y - i[0][1],
            'colour' : i[1] + key_map['colour_shift'],            
        })
    return pixel_sequence

colours = {} #used for debug
for x in range(spritesheet.size[0]):
  for y in range(spritesheet.size[1]):
    colour = spritesheetpx[x,y]
    if colour != 255 and colour != 0 and colour != 15:
      colours[colour] = ''
    if spritesheetpx[x,y] in key_colour_mapping.keys():
        seq = get_pixel_sequence(x, y, spritesheetpx[x,y])
        for i in seq:
            draw.point([(i['x'],i['y'])], fill=i['colour'])
            
spritesheet.save('a_test_trailer.png')

print colours # debug: what colours did we find in this spritesheet? 

Code: Select all

import Image
import ImageDraw
import ImagePalette
spritesheet = Image.open('test_input.png')

block_size = 30
palette_key = Image.new('P',(16*block_size,16*block_size))
draw = ImageDraw.Draw(palette_key)

print spritesheet.mode

palette_key.putpalette(spritesheet.palette)
x = 0
y = 0
for i in range (256):
  draw.rectangle([(x,y),(x+block_size,y+block_size)],fill=i)
  bg_size = draw.textsize(str(i))
  text_pos = (x+(block_size/4),y+(block_size/3))
  draw.rectangle([(text_pos[0]-1,text_pos[1]+1),(text_pos[0]+bg_size[0],text_pos[1]+bg_size[1]-2)],fill=255)
  draw.text((x+(block_size/4),y+(block_size/3)),str(i),fill=1)
  x = x+block_size
  if x == 16*block_size:
    x = 0
    y = y + block_size

palette_key.save('palette_key.png')