notwa/mm
1
0
Fork 0
mirror of https://github.com/notwa/mm synced 2025-02-05 05:23:22 -08:00
Code Issues Releases Wiki Activity

preliminary support for extracting iQue files

Browse source
This commit is contained in:
Connor Olding 2018-04-28 06:36:30 +02:00
parent 7243d1f181
commit bd6cf49f6b
1 changed files with 25 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
z64dump.py
View file

@ -29,6 +29,7 @@ lament = lambda *args, **kwargs: print(*args, file=sys.stderr, **kwargs)
# shoutouts to spinout182 # shoutouts to spinout182
# assume first entry is makerom (0x1060), and second entry begins from makerom # assume first entry is makerom (0x1060), and second entry begins from makerom
dma_sig = b"\x00\x00\x00\x00\x00\x00\x10\x60\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x10\x60" dma_sig = b"\x00\x00\x00\x00\x00\x00\x10\x60\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x10\x60"
dma_sig_ique = b"\x00\x00\x00\x00\x00\x00\x10\x50\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x10\x50"
# hacky # hacky
heresay = os.path.split(sys.argv[0])[0] heresay = os.path.split(sys.argv[0])[0]
@ -44,6 +45,17 @@ def dump_wrap(data, fn, size):
fn += '.' + kind fn += '.' + kind
dump_as(data, fn, size) dump_as(data, fn, size)
def try_deflate(fn, compressed, size):
# love you zoinkity
import zlib
decomp = zlib.decompressobj(-zlib.MAX_WBITS)
data = bytearray()
data.extend(decomp.decompress(compressed))
while decomp.unconsumed_tail:
data.extend(decomp.decompress(decomp.unconsumed_tail))
data.extend(decomp.flush())
return data
def z_dump_file(f, i=0, name=None, uncompress=True): def z_dump_file(f, i=0, name=None, uncompress=True):
vs = R4(f.read(4)) # virtual start vs = R4(f.read(4)) # virtual start
ve = R4(f.read(4)) # virtual end ve = R4(f.read(4)) # virtual end
@ -85,8 +97,12 @@ def z_dump_file(f, i=0, name=None, uncompress=True):
dump(compressed, fn+'.Yaz0', len(compressed)) dump(compressed, fn+'.Yaz0', len(compressed))
else: else:
if uncompress: if uncompress:
data = try_deflate(fn, compressed, size)
if data is None or len(data) == 0:
lament('unknown compression; skipping:', fn) lament('unknown compression; skipping:', fn)
lament(compressed[:4]) lament(compressed[:4])
else:
dump(data, fn, size)
else: else:
lament('unknown compression:', fn) lament('unknown compression:', fn)
dump(compressed, fn, len(compressed)) dump(compressed, fn, len(compressed))
@ -100,8 +116,9 @@ def z_find_dma(f):
data = f.read(16) data = f.read(16)
if len(data) == 0: # EOF if len(data) == 0: # EOF
break break
if data == dma_sig[:16]: for sig in (dma_sig, dma_sig_ique):
rest = dma_sig[16:] if data == sig[:16]:
rest = sig[16:]
if f.read(len(rest)) == rest: if f.read(len(rest)) == rest:
return f.tell() - len(rest) - 16 return f.tell() - len(rest) - 16
else: else: