shithub: pokecrystal

Download patch

ref: f22bbdd722bad8e2700b5801b45d935b631de6c8
parent: 0ee57e94db13be9654c8bd1bbcb2b7bb2aa4e929
author: Bryan Bishop <[email protected]>
date: Sun Jan 27 12:00:51 EST 2013

remove a duplication of load_rom and load_asm

--- a/extras/comparator.py
+++ b/extras/comparator.py
@@ -7,6 +7,8 @@
     get_label_from_line,
     get_address_from_line_comment,
     AsmSection,
+    direct_load_rom,
+    direct_load_asm,
 )
 
 from romstr import (
@@ -17,22 +19,12 @@
 def load_rom(path):
     """ Loads a ROM file into an abbreviated RomStr object.
     """
+    return direct_load_rom(filename=path)
 
-    fh = open(path, "r")
-    x  = RomStr(fh.read())
-    fh.close()
-
-    return x
-
 def load_asm(path):
     """ Loads source ASM into an abbreviated AsmList object.
     """
-
-    fh = open(path, "r")
-    x = AsmList(fh.read().split("\n"))
-    fh.close()
-
-    return x
+    return direct_load_asm(filename=path)
 
 def findall_iter(sub, string):
     # url: http://stackoverflow.com/a/3874760/687783
--- a/extras/crystal.py
+++ b/extras/crystal.py
@@ -145,12 +145,16 @@
     elif os.lstat(filename).st_size != len(rom):
         return direct_load_rom(filename)
 
+def direct_load_asm(filename="../main.asm"):
+    """returns asm source code (AsmList) from a file"""
+    asm = open(filename, "r").read().split("\n")
+    asm = AsmList(asm)
+    return asm
 
 def load_asm(filename="../main.asm"):
-    """loads the asm source code into memory"""
+    """returns asm source code (AsmList) from a file (uses a global)"""
     global asm
-    asm = open(filename, "r").read().split("\n")
-    asm = AsmList(asm)
+    asm = direct_load_asm(filename=filename)
     return asm
 
 def grouper(some_list, count=2):