import os,sys,struct,operator,zlib

table = [
            0x0,0x01,0x02,0x48,0x04,0x05,0x35,0x07,0x08,0x0B,0x0A,0x09,0x10,0x13,0x0E,0x0F,0x0C,
            0x19,0x12,0x0D,0x14,0x1B,0x16,0x17,0x18,0x11,0x1A,0x15,0x1E,0x1D,0x1C,0x1F,0x23,
            0x21,0x22,0x20,0x24,0x25,0x29,0x27,0x28,0x26,0x2A,0x2B,0x2F,0x2D,0x32,0x2C,0x30,
            0x31,0x2E,0x33,0x34,0x06,0x36,0x37,0x38,0x39,0x3A,0x3B,0x3C,0x3D,0x3E,0x3F,0x40,
            0x41,0x42,0x43,0x44,0x45,0x46,0x47,0x03,0x49,0x4A,0x4B,0x4C,0x4D,0x4E,0x4F,0x50,
            0x51,0x52,0x53,0x54,0x55,0x56,0x57,0x58,0x59,0x5A,0x5B,0x5C,0x5D,0x5E,0x5F,0x60,
            0x61,0x62,0x63,0x64,0x65,0x66,0x67,0x68,0x69,0x6A,0x6B,0x6C,0x6D,0x6E,0x6F,0x70,
            0x71,0x72,0x73,0x74,0x75,0x76,0x77,0x78,0x79,0x7A,0x7B,0x7C,0x7D,0x7E,0x7F,0x80,
            0x81,0x82,0x83,0x84,0x85,0x86,0x87,0x88,0x89,0x8A,0x8B,0x8C,0x8D,0x8E,0x8F,0x90,
            0x91,0x92,0x93,0x94,0x95,0x96,0x97,0x98,0x99,0x9A,0x9B,0x9C,0x9D,0x9E,0x9F,0xA0,
            0xA1,0xA2,0xA3,0xA4,0xA5,0xA6,0xA7,0xA8,0xA9,0xAA,0xAB,0xAC,0xAD,0xAE,0xAF,0xB0,
            0xB1,0xB2,0xB3,0xB4,0xB5,0xB6,0xB7,0xB8,0xB9,0xBA,0xBB,0xBC,0xBD,0xBE,0xBF,0xC0,
            0xC1,0xC2,0xC3,0xC4,0xC5,0xC6,0xC7,0xC8,0xC9,0xCA,0xCB,0xCC,0xCD,0xCE,0xCF,0xD0,
            0xD1,0xD2,0xD3,0xD4,0xD5,0xD6,0xD7,0xD8,0xD9,0xDA,0xDB,0xDC,0xDD,0xDE,0xDF,0xE0,
            0xE1,0xE2,0xE3,0xE4,0xE5,0xE6,0xE7,0xE8,0xE9,0xEA,0xEB,0xEC,0xED,0xEE,0xEF,0xF0,
            0xF1,0xF2,0xF3,0xF4,0xF5,0xF6,0xF7,0xF8,0xF9,0xFA,0xFB,0xFC,0xFD,0xFE,0xFF,
        ]

def get_data(filename):
    totalbytes = os.path.getsize(filename)
    infile = open(filename, 'rb')
    totalfiledata = infile.read(totalbytes)
    infile.close()
    return totalfiledata

def extractarchive(filename):
    filedata = open(filename,'rb')
    filedata.seek(8)
    filecount = struct.unpack('<I',filedata.read(4))[0]
    filedata.seek(0x20)

    for i in xrange(0,filecount):
        filedata.seek(4,1)
        filenamelen = struct.unpack('<B',filedata.read(1))[0]
        filenamelen = (operator.neg(filenamelen)) & 0xFF
        filenamelen = table[filenamelen-1]

        filename = ''
        for a in xrange(0,filenamelen):
            byte = struct.unpack('<B',filedata.read(1))[0]
            byte = (operator.neg(byte)) & 0xFF
            byte -= 1
            filename += struct.pack('<B',byte)

        filedata.seek(2,1)
        origlen = struct.unpack('<I',filedata.read(4))[0]
        filelen = struct.unpack('<I',filedata.read(4))[0]
        offset = struct.unpack('<I',filedata.read(4))[0]
        filedata.seek(8,1) #2 unknows here?
        currpos = filedata.tell()

        print filename
        folders = filename.split('\\')
        filename = folders[len(folders)-1]
        folders.pop(len(folders)-1)
        fullfolderpath = ''
        for a in xrange(0,len(folders)):
            if not os.path.exists(fullfolderpath + folders[a]):
                os.mkdir(fullfolderpath + folders[a])
            fullfolderpath += folders[a] + '\\'

        filedata.seek(offset,0)
        filed = filedata.read(filelen)
        if origlen != filelen:
            filed = zlib.decompress(filed)
        if filed[:4] == 'YCG\x00':
            filed = extractYCG(filed)
            filename = filename.rsplit('.',1)[0] + '.bmp'

        filedata.seek(currpos,0)
        outfile = open(fullfolderpath + filename,'wb')
        outfile.write(filed)
        outfile.close()

    filedata.close()

def decryptfilesection(filedata):
    newfiledata = ''
    xorcnt = 0
    for i in xrange(0,len(filedata)):
        byte = struct.unpack('<B',filedata[i:i+1])[0]
        if xorcnt == 0:
            newfiledata += struct.pack('<B',byte ^ 0x76)
        elif xorcnt == 1:
            newfiledata += struct.pack('<B',byte ^ 0x3)
        elif xorcnt == 2:
            newfiledata += struct.pack('<B',byte ^ 0x3b)
        elif xorcnt == 3:
            newfiledata += struct.pack('<B',byte ^ 0x26)
            xorcnt = -1
        xorcnt += 1
    return newfiledata

def extractYCG(filedata):
    newfiledata = ''

    seccount = struct.unpack('<I',filedata[0x14:0x18])[0]
    curlen = 0
    for i in range(0,seccount):
        seclen = struct.unpack('<I',filedata[0x18+(i*0x10)+0xc:0x18+(i*0x10)+0x10])[0]
        newfiledata += zlib.decompress(filedata[0x38+curlen:0x38+curlen+seclen])
        curlen += seclen

    bmpheader = 'BM'
    bmpheader += struct.pack('<I',0x36+len(newfiledata))
    bmpheader += '\x00\x00\x00\x00'
    bmpheader += struct.pack('<I',0x36) + struct.pack('<I',0x28)
    bmpheader += struct.pack('<I',struct.unpack('<I',filedata[0x4:0x8])[0]) #width
    bmpheader += struct.pack('<I',-struct.unpack('<I',filedata[0x8:0xc])[0] & 0xFFFFFFFF) #height
    bmpheader += struct.pack('<H',1) + struct.pack('<H',32)
    bmpheader += struct.pack('<I',0)
    bmpheader += struct.pack('<I',len(newfiledata))
    bmpheader += struct.pack('<I',0x2e20) + struct.pack('<I',0x2e20)
    bmpheader += struct.pack('<I',0) + struct.pack('<I',0)

    return bmpheader + newfiledata

if __name__ == '__main__':
    extractarchive(sys.argv[1])