Leahnaya/pokemon-reverse-engineering-tools
1
0
Fork 0
mirror of https://github.com/pret/pokemon-reverse-engineering-tools.git synced 2026-07-13 14:51:04 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

simplify load_rom in gbz80disasm

Browse Source 
original-commit-id: 57200b6cf75040b9696ae54bdac69d5e452a2c48
This commit is contained in:
Bryan Bishop 2013-02-03 15:18:06 -06:00
parent c789a2962a
commit d3a1ef222a
2 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
gbz80disasm.py
View File

@ -14,11 +14,8 @@ if not hasattr(json, "read"):
from romstr import RomStr
def load_rom(filename="../baserom.gbc"):
"""loads bytes into memory"""
global rom
file_handler = open(filename, "rb")
rom = RomStr(file_handler.read())
file_handler.close()
rom = RomStr.load(filename=filename)
return rom
spacing = "\t"

10
romstr.py
View File

@ -46,15 +46,17 @@ class RomStr(str):
return "RomStr(too long)"
@classmethod
def load(cls, crystal=True, red=False):
def load(cls, filename=None, crystal=True, red=False):
""" Loads a ROM into a RomStr.
"""
if crystal and not red:
if crystal and not red and not filename:
file_handler = open("../baserom.gbc", "r")
elif red and not crystal:
elif red and not crystal and not filename:
file_handler = open("../pokered-baserom.gbc", "r")
elif filename not in ["", None]:
file_handler = open(filename, "rb")
else:
raise Exception, "not sure which rom to load?"
raise Exception("not sure which rom to load?")
bytes = file_handler.read()
file_handler.close()
return RomStr(bytes)