lodextract/lodextract.py

#!/usr/bin/env python

import zlib
import struct
import os
from PIL import Image, ImageDraw

from common import crc24_func, get_complement, font

def is_pcx(data):
    size,width,height = struct.unpack("<III",data[:12])
    return size == width*height or size == width*height*3

def read_pcx(data):
    size,width,height = struct.unpack("<III",data[:12])
    if size == width*height:
        im = Image.fromstring('P', (width,height),data[12:12+width*height])
        palette = []
        for i in range(256):
            offset=12+width*height+i*3
            r,g,b = struct.unpack("<BBB",data[offset:offset+3])
            palette.extend((r,g,b))
        im.putpalette(palette)
        return im
    elif size == width*height*3:
        return Image.fromstring('RGB', (width,height),data[12:])
    else:
        return None

def unpack_lod(infile,outdir,shred=True):
    f = open(infile)

    header = f.read(4)
    if header != 'LOD\0':
        print "not LOD file: %s"%header
        return False

    f.seek(8)
    total, = struct.unpack("<I", f.read(4))
    f.seek(92)

    files=[]
    for i in range(total):
        filename, = struct.unpack("16s", f.read(16))
        filename = filename[:filename.index('\0')].lower()
        offset,size,_,csize = struct.unpack("<IIII", f.read(16))
        files.append((filename,offset,size,csize))

    for filename,offset,size,csize in files:
        filename=os.path.join(outdir,filename)
        print filename
        f.seek(offset)
        if csize != 0:
            data = zlib.decompress(f.read(csize))
        else:
            data = f.read(size)
        if is_pcx(data):
            im = read_pcx(data)
            if im:
                if shred:
                    crc = crc24_func(filename)
                    r = crc>>16
                    g = (crc&0xff00)>>8
                    b = crc&0xff
                    w,h = im.size
                    pixels = im.load()
                    for i in range(w):
                        for j in range(h):
                            if pixels[i,j] > 7:
                                if im.mode == 'P':
                                    pixels[i,j] = 8+crc%248
                                else:
                                    pixels[i,j] = (r,g,b)
                    im.resize((w*3,h*3))
                    draw = ImageDraw.Draw(im)
                    tw,th = draw.textsize(os.path.basename(filename),font=font)
                    tpos = ((w*3-tw)/2,(h*3-th)/2)
                    if im.mode == 'P':
                        # we can't really have a complement in palette mode, so just get some color
                        draw.text(tpos,os.path.basename(filename),255,font=font)
                    else:
                        draw.text(tpos,os.path.basename(filename),get_complement(r,g,b),font=font)
                    im = im.resize((w,h),Image.ANTIALIAS)
                im.save(filename, "PNG")
            else:
                return False
        else:
            o = open(filename,"w+")
            o.write(data)
            o.close()

    return True

if __name__ == '__main__':
    import sys
    if len(sys.argv) != 3:
        print "usage: %s infile.lod ./outdir"%sys.argv[0]
        print ""
        print "usually after installing the normal way:"
        print "    %s .vcmi/Data/H3bitmap.lod .vcmi/Mods/vcmi/Data/"%sys.argv[0]
        print "    rm .vcmi/Data/H3bitmap.lod"
        print "    %s .vcmi/Data/H3sprite.lod .vcmi/Mods/vcmi/Data/"%sys.argv[0]
        print "    rm .vcmi/Data/H3sprite.lod"
        exit(1)
    ret = unpack_lod(sys.argv[1], sys.argv[2])
    exit(0 if ret else 1)