SpriteToCode/transformation.py at master · Mokona/SpriteToCode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
from PIL import Image


def columnize(image, sprite_size):
    sprite_width, _ = sprite_size
    image_width, image_height = image.size

    vertical_bands_count = int(image_width / sprite_width)

    new_image_width = sprite_width
    new_image_height = int(image_height * vertical_bands_count)
    new_size = new_image_width, new_image_height

    new_image = Image.new(image.mode, new_size)

    palette = image.getpalette()
    if palette:
        new_image.putpalette(palette)

    for count in range(vertical_bands_count):
        band_x_coordinates = count * sprite_width, ((count + 1) * sprite_width)
        band = image.crop((band_x_coordinates[0], 0, band_x_coordinates[1], image_height))

        new_image.paste(band, (0, count * image_height))

    return new_image


gb_palette = [
    0x000000, 0x004385, 0x960040, 0x008B50, 0xCF8E44, 0x544D43, 0xA89987,
    0xFFFFFF, 0xDB1D23, 0xFFA811, 0xF5E700, 0x85CF44, 0x7DBBFF, 0x4485CF,
    0xCF4485, 0xFFD690
]


def rgb_to_gamebuino_palette_index(rgb):
    return gb_palette.index(rgb)


def palettize(source_image):
    palette_image = source_image.convert('P', palette=Image.ADAPTIVE)
    return palette_image


def im_palette_as_rgb(im_palette):
    from itertools import islice

    reds = islice(im_palette, 0, None, 3)
    greens = islice(im_palette, 1, None, 3)
    blues = islice(im_palette, 2, None, 3)

    return enumerate(zip(reds, greens, blues))


def palette_mapping(im_palette):
    mapping = {}

    for i, (r, g, b) in im_palette_as_rgb(im_palette):
        color = (r << 16) + (g << 8) + b

        if color in gb_palette:
            mapping[i] = gb_palette.index(color)
        else:
            mapping[i] = gb_palette.index(0x000000)

    return mapping


def pack_data(image_data, mapping):
    from itertools import islice

    even = islice(image_data, 0, None, 2)
    odd = islice(image_data, 1, None, 2)
    return [(mapping.get(a, 0) << 4) + mapping.get(b, 0)
            for a, b
            in zip(even, odd)]