simplify load_rom in gbz80disasm

This commit is contained in:
Bryan Bishop 2013-02-03 15:18:06 -06:00
parent 82b8b5121a
commit 57200b6cf7
2 changed files with 7 additions and 8 deletions

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"

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)