ToolboxToolbox/CategorizeRoms.py

85 lines
1.8 KiB
Python
Raw Normal View History

2020-05-13 11:14:40 +00:00
#!/usr/bin/env python3
import sys
import struct
from os import path
biglist = []
def could_be_rom(lb):
if not lb: return False
if lb == 0x300000: return True # special supermario case!
while not (lb & 1):
lb >>= 1
if lb != 1: return False
return True
for rompath in sys.argv[1:]:
try:
if not could_be_rom(path.getsize(rompath)): continue
rom = open(rompath, 'rb').read()
except IsADirectoryError:
continue
info = []
a, = struct.unpack_from('>H', rom, 8)
b, = struct.unpack_from('>H', rom, 18)
order = a << 64
if a == 0x69:
info.append('Original')
elif a == 0x75:
info.append('Plus')
elif a == 0x178:
info.append('II')
elif a == 0x276:
info.append('SE')
elif a == 0x37a:
info.append('Portable')
elif a == 0x67c:
info.append('IIci')
if rom[18] == 0x15:
order += 1 << 63
info.append('TERROR')
elif rom[18] == 0x17:
order += 2 << 63
info.append('Zydeco')
elif rom[20] & 1:
order += 3 << 63
info.append('Horror')
elif a == 0x77d:
info.append('SuperMario')
else:
order = 0
info.append('?-%X' % a)
if a >= 0x300:
info.append('Rel-%04x' % b)
order += b << 32
if a == 0x77d: # SuperMario
info.append('SubVer-%x' % struct.unpack_from('>H', rom, 0x4c))
order += struct.unpack_from('>H', rom, 0x4c)[0]
if a == 0x67c:
if b >= 0x12f1: info.append("EricksonSndMgr")
if b == 0x12f1: info.append("EricksonMistake")
biglist.append((order, ' '.join(info), rompath))
biglist.sort()
wid = max(len(s) for _, s, *_ in biglist)
for o, s, p in biglist:
print(s.ljust(wid + 4), p)