art.py

""" Copyright 2015-2017  sta256+mpskit at gmail.com
    
    This file is part of mpskit.

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU Affero General Public License as
    published by the Free Software Foundation, either version 3 of the
    License, or (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY.

    See LICENSE file for more details.
"""
from common import *
from madspack import read_madspack, write_madspack, save_madspack, load_madspack
from collections import namedtuple
from PIL import Image, ImageDraw
from PIL.ImagePalette import ImagePalette
from fab import read_fab
from palette import read_palette_col, read_palette_rex, attach_palette, export_palette

ext = 'png'




def save_header(name, h):
	oname = name + '.json'
	with open(oname, 'w') as f:
		json.dump(h.as_dict(), f, indent=2)	
	print(oname)



	

def read_art(art_name):
	check_ext(art_name, '.ART')
			
	verbose = 0	
	
	parts = read_madspack(art_name)
	
	
	save_madspack(art_name, parts)
		
	# parts[0] -- dimensions + palette
	# parts[1] -- image data
		
		
	# header	
	h = Header()
	h.width = read_uint16(parts[0])
	h.height = read_uint16(parts[0])
	
	save_header(art_name, h)
	
	# palette
	pal = read_palette_rex(parts[0])
	
	export_palette(pal, art_name)
	
	
	
	# image -- read indexed image
	img = Image.new('P', (h.width, h.height))	
	attach_palette(img, pal)	
	pix = img.load()	 
	for j in range(h.height):
		for i in range(h.width):
			ind = read_uint8(parts[1])
			pix[i,j] = ind

	save_image(art_name, img)
	

def load_header(name):
	with open('{}.json'.format(name), 'r') as f:
		return Header.from_dict(json.load(f))
	

def write_art(art_name):
	check_ext(art_name, '.ART')
			
	verbose = 0	
	
	# parts[0] -- dimensions + palette
	# parts[1] -- image data
		
		
	# load header	
	h = load_header(art_name)
	
	
	
	# load image
	part1 = BytesIO()
			
	img = Image.open('{}.png'.format(art_name))		
	
	inds = [x for x in img.getdata()]
	assert len(inds) == h.width * h.height	
	write_raw(part1, len(inds), inds)

	img.close()
	
	parts = load_madspack(art_name)
	part1.seek(0)
	parts[1] = part1
	write_madspack(art_name, parts)