shithub: pokered

Download patch

ref: ce6c6f1f7262db043045f7babf7405e9443cb1ea
parent: 863b08346c078bf8b5db0072e48f925677d76ffe
author: Bryan Bishop <[email protected]>
date: Thu Jan 26 23:51:20 EST 2012 

small script to make better trainer header labels

hg-commit-id: 6ef15ef40471



--- /dev/null
+++ b/extras/fix_labels.py
@@ -1,0 +1,83 @@
+#!/usr/bin/python
+#author: Bryan Bishop <[email protected]>
+#date: 2012-01-27
+#fix trainer header labels to not suck so much
+import analyze_incbins
+
+def replace_trainer_header_labels(debug=False):
+    """trainer header labels could be better"""
+    asm = analyze_incbins.asm
+    if debug: print str(type(asm))
+    single_asm = "\n".join(asm)
+    current_map_name = "asdjkl;"
+    line_id = 0
+    trainer_header_counter = 1
+
+    for line in asm:
+        trainer_header_base = current_map_name + "TrainerHeader"
+        trainer_header_name = trainer_header_base + str(trainer_header_counter)
+
+        #we've found a TrainerHeaders thing
+        if "TrainerHeaders:" in line:
+            current_map_name = line.split("TrainerHeaders")[0]
+            if line[0:len(current_map_name)] == current_map_name:
+                trainer_header_counter = 1
+
+        #replace a trainer header label
+        elif "TrainerHeader_" in line and line[0:14] == "TrainerHeader_":
+            temp = line.split("TrainerHeader_")[1]
+            temp = temp.split(": ;")[0]
+            
+            old_label = "TrainerHeader_" + temp
+            new_label = current_map_name + "TH" + str(trainer_header_counter) #trainer_header_name
+            single_asm = single_asm.replace(old_label + ":", new_label + ":")
+            single_asm = single_asm.replace(old_label + "\n", new_label + "\n")
+            if debug: print "old_label = " + old_label
+            if debug: print "new_label = " + new_label
+
+            trainer_header_counter += 1
+        
+        elif trainer_header_base in line and line[0:len(trainer_header_base)] == trainer_header_base and (line[len(trainer_header_base)+1:len(trainer_header_base)+2] == ":" or line[len(trainer_header_base)+2:len(trainer_header_base)+3] == ":"):
+            if line[len(trainer_header_base)+1:len(trainer_header_base)+2] == ":":
+                trainer_header_counter = int(line[len(trainer_header_base):len(trainer_header_base)+1])
+            elif line[len(trainer_header_base)+2:len(trainer_header_base)+3] == ":":
+                trainer_header_counter = int(line[len(trainer_header_base):len(trainer_header_base)+2])
+            trainer_header_name = trainer_header_base + str(trainer_header_counter)
+
+        #replace a text label
+        elif " TextBeforeBattle" in line and not current_map_name in line:
+            old_label = line.split("dw ")[1].split(" ;")[0]
+            new_label = current_map_name + "B4BattleTxt" + str(trainer_header_counter) #trainer_header_name + "BeforeBattleText"
+            single_asm = single_asm.replace(old_label + ":", new_label + ":")
+            single_asm = single_asm.replace(old_label + "\n", new_label + "\n")
+            single_asm = single_asm.replace(old_label + " ;", new_label + " ;")
+            if debug: print "old_label = " + old_label
+            if debug: print "new_label = " + new_label
+        #replace a text label
+        elif " TextAfterBattle" in line and not current_map_name in line:
+            old_label = line.split("dw ")[1].split(" ;")[0]
+            new_label = current_map_name + "AfterBattleTxt" + str(trainer_header_counter) #trainer_header_name + "AfterBattleText"
+            single_asm = single_asm.replace(old_label + ":", new_label + ":")
+            single_asm = single_asm.replace(old_label + "\n", new_label + "\n")
+            single_asm = single_asm.replace(old_label + " ;", new_label + " ;")
+            if debug: print "old_label = " + old_label
+            if debug: print "new_label = " + new_label
+        #replace a text label
+        elif " TextEndBattle" in line and not current_map_name in line:
+            old_label = line.split("dw ")[1].split(" ;")[0]
+            new_label = current_map_name + "EndBattleTxt" + str(trainer_header_counter) #trainer_header_name + "EndBattleText"
+            single_asm = single_asm.replace(old_label + ":", new_label + ":")
+            single_asm = single_asm.replace(old_label + "\n", new_label + "\n")
+            single_asm = single_asm.replace(old_label + " ;", new_label + " ;")
+            if debug: print "old_label = " + old_label
+            if debug: print "new_label = " + new_label
+
+        line_id += 1
+
+    print single_asm
+
+if __name__ == "__main__":
+    analyze_incbins.load_asm()
+
+    replace_trainer_header_labels()
+