fix naming of map size constants

hg-commit-id: 4477820444cd

1 files changed, 15 insertions, 6 deletions

diff --git a/extras/make_map_size_constants.py b/extras/make_map_size_constants.py
index 70be3b02..c40a4514 100644
--- a/extras/make_map_size_constants.py
+++ b/extras/make_map_size_constants.py
@@ -3,27 +3,36 @@
 #date: 2012-01-15
 #dump map height/width constants
 import extract_maps
-from pretty_map_headers import map_name_cleaner
+from pretty_map_headers import map_name_cleaner, map_constants
 
-def get_map_size_constants():
+def get_map_size_constants(do_sed=False):
     output = ""
+    sed_lines = ""
     for map_id in extract_maps.map_headers.keys():
         if map_id in extract_maps.bad_maps: continue #skip
 
         map2 = extract_maps.map_headers[map_id]
         base_name = map_name_cleaner(map2["name"], None)[:-2]
+        constant_name = map_constants[map_id]
 
         height = int(map2["y"], 16)
         width  = int(map2["x"], 16)
 
         output += "; " + base_name + "_h map_id=" + str(map_id) + "\n"
-        output += base_name + "Height EQU $%.2x\n" % (height)
-        output += base_name + "Width  EQU $%.2x\n" % (width)
+        output += constant_name + "_HEIGHT EQU $%.2x\n" % (height)
+        output += constant_name + "_WIDTH  EQU $%.2x\n" % (width)
         output += "\n"
-    return output
+
+        sed_lines += "sed -i 's/" + base_name + "Height/" + constant_name + "_HEIGHT" + "/g' common.asm" + "\n"
+        sed_lines += "sed -i 's/" + base_name + "Width/" + constant_name + "_WIDTH" + "/g' common.asm" + "\n"
+
+    if do_sed:
+        return sed_lines
+    else:
+        return output
 
 if __name__ == "__main__":
     extract_maps.load_rom()
     extract_maps.load_map_pointers()
     extract_maps.read_all_map_headers()
-    print get_map_size_constants()
+    print get_map_size_constants(do_sed=True)