"""
    Extract sprites and background graphics from "Plan 9 From Outer Space" (PC)
    Copyright (C) 2010 Gil Megidish

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU 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; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>
"""
from PIL import Image
from struct import unpack

ndx = open("plan9.ndx", "rb").read()

offset = 2
index = 1

palette = None

while offset < len(ndx):
	c = (ord(ndx[offset+0])) | (ord(ndx[offset+1]) << 8) | (ord(ndx[offset+2]) << 16)
	code = ord(ndx[offset+3])
	print "%x code %d" % (c, code)
	offset += 4

	if code == 2:
		f = open("plan9.gfx", "rb")
		f.seek(c, 0)
		size = unpack("<i", f.read(4))[0]
		height = unpack("<h", f.read(2))[0]
		width = unpack("<h", f.read(2))[0]
		raw = f.read(size)
		im = Image.fromstring("P", (width, height), raw)
		if palette is not None:
			im.putpalette(palette)
	
		im.save("%04d.png" % index)
		f.close()
		index = index + 1

	if code == 3:
		f = open("plan9.gfx", "rb")
		f.seek(c, 0)
		ncolors = unpack("<h", f.read(2))[0]
		colors = f.read(ncolors * 3)
		size = unpack("<i", f.read(4))[0]
		height = unpack("<h", f.read(2))[0]
		width = unpack("<h", f.read(2))[0]
		print ncolors, width, height, size
		raw = f.read(size)
       	
		palette = [0] * 768
		for i in range(ncolors):
			palette[i*3+0] = ord(colors[i*3+0]) << 2
			palette[i*3+1] = ord(colors[i*3+1]) << 2
			palette[i*3+2] = ord(colors[i*3+2]) << 2
       	
		im = Image.fromstring("P", (width, height), raw)
		im.putpalette(palette)
		im.save("%04d.png" % index)
		index = index + 1