106 lines
3.6 KiB
Python
106 lines
3.6 KiB
Python
#!/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)
|