aboutsummaryrefslogtreecommitdiff
path: root/experiments/conv/png2pc
blob: 21b3b29fdb3bcf658e2ba376d70920cb34625480 (plain)
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
#!/bin/python3
import itertools
import io
from PIL import Image, ImageDraw

from consts import *
from shared import main

palette_img = Image.new('P', (1, 1))
palette_img.putpalette(list(itertools.chain.from_iterable([
	*PICTOCHAT_PALETTE,
	*[PICTOCHAT_PALETTE[0] for x in range(256 - len(PICTOCHAT_PALETTE))],
])))

def convert_chunk(img, offset):
	palette_indices = []
	for y in range(DS_TILE_SIZE):
		for x in range(DS_TILE_SIZE):
			palette_indices.append(img.getpixel((x + offset[0], y + offset[1])))

	bytes = bytearray()
	for i in range(0, len(palette_indices), 2):
		bytes.append((palette_indices[i+0] << 0) | (palette_indices[i+1] << 4))
	
	return bytes

def png2pc(data):
	img = Image.open(io.BytesIO(data))
	if img.width != TILES_HORIZONTAL * DS_TILE_SIZE:
		print(f"error: input image is not {TILES_HORIZONTAL * DS_TILE_SIZE} pixels wide")
		exit(1)
	if img.height % DS_TILE_SIZE != 0:
		print(f"error: input image height not a muliple of {DS_TILE_SIZE}")
		exit(1)
	tile_count = (img.width * img.height) // (DS_TILE_SIZE ** 2)
	
	img = img.convert('RGB')
	img = img.quantize(palette=palette_img)

	draw = ImageDraw.Draw(img)
	draw.rectangle(((0, 0), (24, img.height)), fill=0)
	draw.rectangle(((252, 0), (256, img.height)), fill=0)
	draw.rectangle(((24, 0), (79, 16)), fill=0)

	output = b""
	for tile_idx in range(tile_count):
		tile = (
			tile_idx % TILES_HORIZONTAL,
			tile_idx // TILES_HORIZONTAL,
		)
		offset = (
			tile[0] * DS_TILE_SIZE,
			tile[1] * DS_TILE_SIZE,
		)
		output += convert_chunk(img, offset)

	return output

if __name__ == "__main__":
	main("png", "bin", png2pc)