fix romstr for compiling (thanks pasky)

This commit is contained in:
Bryan Bishop 2012-07-01 08:41:14 -05:00
parent 3359121ba7
commit fcf43b13da

View File

@ -72,7 +72,7 @@ class RomStr(str):
# scan the asm source code for labels # scan the asm source code for labels
if generate_labels: if generate_labels:
asm = open("../main.asm", "r").read().split("\n") asm = open(os.path.dirname(__file__) + "/../main.asm", "r").read().split("\n")
for line in asm: for line in asm:
label = get_label_from_line(line) label = get_label_from_line(line)
@ -446,7 +446,7 @@ class DisAsm:
addresses = set() addresses = set()
for (id, command) in self.asm_commands.items(): for (id, command) in self.asm_commands.items():
if command.has_key("target_address"): if command.has_key("target_address") and command["id"] in call_commands:
addresses.add(command["target_address"]) addresses.add(command["target_address"])
return addresses return addresses