ref: 550ca496d606c8c85745e31e7246da411f7989a8
parent: 8cfa4bee12b21f7af94bd368165b3b599d18997c
author: Bryan Bishop <[email protected]>
date: Tue Mar 13 12:01:52 EDT 2012
function to find most text addresses
--- a/extras/crystal.py
+++ b/extras/crystal.py
@@ -467,6 +467,110 @@
output += "\n"
return output
+def find_all_text_pointers_in_script_engine_script(script, bank):
+ """returns a list of text pointers
+ based on each script-engine script command"""
+ #TODO: recursively follow any jumps in the script
+ if script == None: return []
+ addresses = set()
+ for (k, command) in script.items():
+ if command["type"] == 0x4B:
+ addresses.add(command["pointer"])
+ elif command["type"] == 0x4C:
+ addresses.add(command["pointer"])
+ elif command["type"] == 0x51:
+ addresses.add(command["text_pointer"])
+ elif command["type"] == 0x53:
+ addresses.add(command["text_pointer"])
+ elif command["type"] == 0x64:
+ addresses.add(command["won_pointer"])
+ addresses.add(command["lost_pointer"])
+ return addresses
+
+def find_text_addresses():
+ """returns a list of text pointers
+ useful for testing parse_text_engine_script_at
+
+ Note that this list is not exhaustive. There are some texts that
+ are only pointed to from some script that a current script just
+ points to. So find_all_text_pointers_in_script_engine_script will
+ have to recursively follow through each script to find those.
+ """
+ addresses = set()
+ #for each map group
+ for map_group in map_names:
+ #for each map id
+ for map_id in map_names[map_group]:
+ #skip the offset key
+ if map_id == "offset": continue
+ #dump this into smap
+ smap = map_names[map_group][map_id]
+ #signposts
+ signposts = smap["signposts"]
+ #for each signpost
+ for signpost in signposts:
+ #dump this into script
+ script = signpost["script"]
+ #skip signposts with no bytes
+ if len(script) == 0: continue
+ #find all text pointers in script
+ texts = find_all_text_pointers_in_script_engine_script(script, smap["event_bank"])
+ #dump these addresses in
+ addresses.update(texts)
+ #xy triggers
+ xy_triggers = smap["xy_triggers"]
+ #for each xy trigger
+ for xy_trigger in xy_triggers:
+ #dump this into script
+ script = xy_trigger["script"]
+ #find all text pointers in script
+ texts = find_all_text_pointers_in_script_engine_script(script, smap["event_bank"])
+ #dump these addresses in
+ addresses.update(texts)
+ #trigger scripts
+ triggers = smap["trigger_scripts"]
+ #for each trigger
+ for (i, trigger) in triggers.items():
+ #dump this into script
+ script = trigger["script"]
+ #find all text pointers in script
+ texts = find_all_text_pointers_in_script_engine_script(script, calculate_bank(trigger["address"]))
+ #dump these addresses in
+ addresses.update(texts)
+ #callback scripts
+ callbacks = smap["callback_scripts"]
+ #for each callback
+ for (k, callback) in callbacks.items():
+ #dump this into script
+ script = callback["script"]
+ #find all text pointers in script
+ texts = find_all_text_pointers_in_script_engine_script(script, calculate_bank(callback["address"]))
+ #dump these addresses in
+ addresses.update(texts)
+ #people-events
+ events = smap["people_events"]
+ #for each event
+ for event in events:
+ if event["event_type"] == "script":
+ #dump this into script
+ script = event["script"]
+ #find all text pointers in script
+ texts = find_all_text_pointers_in_script_engine_script(script, smap["event_bank"])
+ #dump these addresses in
+ addresses.update(texts)
+ if event["event_type"] == "trainer":
+ trainer_data = event["trainer_data"]
+ addresses.update([trainer_data["text_when_seen_ptr"]])
+ addresses.update([trainer_data["text_when_trainer_beaten_ptr"]])
+ trainer_bank = calculate_bank(event["trainer_data_address"])
+ script1 = trainer_data["script_talk_again"]
+ texts1 = find_all_text_pointers_in_script_engine_script(script1, trainer_bank)
+ addresses.update(texts1)
+ script2 = trainer_data["script_when_lost"]
+ texts2 = find_all_text_pointers_in_script_engine_script(script2, trainer_bank)
+ addresses.update(texts2)
+ return addresses
+
def parse_text_engine_script_at(address, map_group=None, map_id=None, debug=True):
return {}@@ -2802,6 +2906,7 @@
text_when_seen = parse_text_engine_script_at(text_when_seen_ptr, map_group=map_group, map_id=map_id)
text_when_trainer_beaten_ptr = calculate_pointer_from_bytes_at(address+5, bank=bank)
text_when_trainer_beaten = parse_text_engine_script_at(text_when_trainer_beaten_ptr, map_group=map_group, map_id=map_id)
+
if [ord(rom[address+7]), ord(rom[address+8])] == [0, 0]:
script_when_lost_ptr = 0
script_when_lost = None
@@ -2808,10 +2913,15 @@
else:
print "parsing script-when-lost"
script_when_lost_ptr = calculate_pointer_from_bytes_at(address+7, bank=bank)
- script_when_lost = None #parse_script_engine_script_at(script_when_lost_ptr, map_group=map_group, map_id=map_id)
+ script_when_lost = None
+ #if script_when_lost_ptr > 0x4000:
+ # script_when_lost = parse_script_engine_script_at(script_when_lost_ptr, map_group=map_group, map_id=map_id)
+
print "parsing script-talk-again" #or is this a text?
script_talk_again_ptr = calculate_pointer_from_bytes_at(address+9, bank=bank)
- script_talk_again = None #parse_script_engine_script_at(script_talk_again_ptr, map_group=map_group, map_id=map_id)
+ script_talk_again = None
+ #if script_talk_again_ptr > 0x4000:
+ # script_talk_again = parse_script_engine_script_at(script_talk_again_ptr, map_group=map_group, map_id=map_id)
return {"bit_number": bit_number,
@@ -2904,8 +3014,7 @@
"trainer_data_address": ptr_address,
"trainer_data": parsed_trainer,
}
-
-
+
#XXX not sure what's going on here
#bit no. of bit table 1 (hidden if set)
#note: FFFF for none
@@ -2933,6 +3042,10 @@
#"text_bank": text_bank, #script pointer byte 2
"when_byte": when_byte, #bit no. of bit table 1 (hidden if set)
"hide": hide, #note: FFFF for none
+
+ "is_trainer": is_trainer,
+ "is_regular_script": is_regular_script,
+ "is_give_item": is_give_item,
}
people_event.update(extra_portion)
people_events.append(people_event)