from __future__ import with_statement
from __future__ import absolute_import
import os
import struct
from io import open

if __name__ == u'__main__':
    if len(os.sys.argv) != 2:
        print u"Use: python splita_rom_virtual_console_3ds.py /path/to/data.bin"
        os.sys.exit(1)

    fabs = os.sys.argv[1]

    try:
        with open(fabs, u"rb") as fi:
            b = bytearray(fi.read())
            rom_size = struct.unpack(u'<I', b[0x31:0x34] + '\0')[0]
            rom_data = b[0x60:0x60 + rom_size]
            print "ROM size: {dec} ({hex})".format(dec=rom_size, hex=hex(rom_size))

            pcm_data = b[0x60 + rom_size:]
            print "PCM size: {dec} ({hex})".format(dec=len(pcm_data), hex=hex(len(pcm_data)))

            with open(u'game.rom', u'wb') as fo:
                fo.write(rom_data)

            with open(u'game.pcm', u'wb') as fo:
                fo.write(pcm_data)

        print u'SUCCESS'
    except Exception, e:
        print u'ERROR' + unicode(e)