diff options
Diffstat (limited to 'extras/analyze_incbins.py')
| -rw-r--r-- | extras/analyze_incbins.py | 18 |
1 files changed, 16 insertions, 2 deletions
diff --git a/extras/analyze_incbins.py b/extras/analyze_incbins.py index b52f0e15..544cd55f 100644 --- a/extras/analyze_incbins.py +++ b/extras/analyze_incbins.py @@ -27,9 +27,23 @@ def offset_to_pointer(offset): return int(offset) % 0x4000 + 0x4000 def load_asm(filename="../main.asm"): - "loads the asm source code into memory" + """loads the asm source code into memory + this also detects if the revision of the repository + is using main.asm, common.asm or pokered.asm, which is + useful when generating images in romvisualizer.py""" global asm - asm = open(filename, "r").read().split("\n") + defaults = ["../main.asm", "../common.asm", "../pokered.asm"] + if filename in defaults: + if os.path.exists("../main.asm"): + asm = open("../main.asm", "r").read().split("\n") + elif os.path.exists("../common.asm"): + asm = open("../common.asm", "r").read().split("\n") + elif os.path.exists("../pokered.asm"): + asm = open("../pokered.asm", "r").read().split("\n") + else: + raise "this shouldn't happen" + else: + asm = open(filename, "r").read().split("\n") return asm def isolate_incbins(): |