diff --git a/extras/crystal.py b/extras/crystal.py index d778d339d..3eb0b96bc 100644 --- a/extras/crystal.py +++ b/extras/crystal.py @@ -467,6 +467,110 @@ def parse_text_at2(address, count=10): 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,16 +2906,22 @@ def parse_trainer_header_at(address, map_group=None, map_id=None): 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 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 @@ def parse_people_event_bytes(some_bytes, address=None, map_group=None, map_id=No "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 @@ def parse_people_event_bytes(some_bytes, address=None, map_group=None, map_id=No #"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)