diff --git a/MultiClient.py b/MultiClient.py index 5418c16d..12be383b 100644 --- a/MultiClient.py +++ b/MultiClient.py @@ -663,7 +663,7 @@ async def process_server_cmd(ctx : Context, cmd, args): ctx.player_names = {p: n for p, n in args[1]} msgs = [] if ctx.locations_checked: - msgs.append(['LocationChecks', [Regions.location_table[loc][0] for loc in ctx.locations_checked]]) + msgs.append(['LocationChecks', [Regions.lookup_name_to_id[loc] for loc in ctx.locations_checked]]) if ctx.locations_scouted: msgs.append(['LocationScouts', list(ctx.locations_scouted)]) if msgs: @@ -676,7 +676,7 @@ async def process_server_cmd(ctx : Context, cmd, args): elif start_index != len(ctx.items_received): sync_msg = [['Sync']] if ctx.locations_checked: - sync_msg.append(['LocationChecks', [Regions.location_table[loc][0] for loc in ctx.locations_checked]]) + sync_msg.append(['LocationChecks', [Regions.lookup_name_to_id[loc] for loc in ctx.locations_checked]]) await send_msgs(ctx.socket, sync_msg) if start_index == len(ctx.items_received): for item in items: @@ -688,7 +688,7 @@ async def process_server_cmd(ctx : Context, cmd, args): if location not in ctx.locations_info: replacements = {0xA2: 'Small Key', 0x9D: 'Big Key', 0x8D: 'Compass', 0x7D: 'Map'} item_name = replacements.get(item, get_item_name_from_id(item)) - logging.info(f"Saw {color(item_name, 'red', 'bold')} at {list(Regions.location_table.keys())[location - 1]}") + logging.info(f"Saw {color(item_name, 'red', 'bold')} at {list(Regions.lookup_id_to_name.keys())[location - 1]}") ctx.locations_info[location] = (item, player) ctx.watcher_event.set() @@ -769,7 +769,7 @@ async def console_loop(ctx : Context): get_location_name_from_address(item.location), index, len(ctx.items_received))) if command[0] == '/missing': - for location in [k for k, v in Regions.location_table.items() if type(v[0]) is int]: + for location in [k for k, v in Regions.lookup_name_to_id.items() if type(v[0]) is int]: if location not in ctx.locations_checked: logging.info('Missing: ' + location) if command[0] == '/getitem' and len(command) > 1: @@ -784,23 +784,26 @@ async def console_loop(ctx : Context): await snes_flush_writes(ctx) + def get_item_name_from_id(code): items = [k for k, i in Items.item_table.items() if type(i[3]) is int and i[3] == code] - return items[0] if items else 'Unknown item' + return items[0] if items else f'Unknown item (ID:{code})' + def get_location_name_from_address(address): if type(address) is str: return address - locs = [k for k, l in Regions.location_table.items() if type(l[0]) is int and l[0] == address] - return locs[0] if locs else 'Unknown location' + return Regions.lookup_id_to_name.get(address, f'Unknown location (ID:{address})') + async def track_locations(ctx : Context, roomid, roomdata): new_locations = [] + def new_check(location): ctx.locations_checked.add(location) logging.info("New check: %s (%d/216)" % (location, len(ctx.locations_checked))) - new_locations.append(Regions.location_table[location][0]) + new_locations.append(Regions.lookup_name_to_id[location]) for location, (loc_roomid, loc_mask) in location_table_uw.items(): if location not in ctx.locations_checked and loc_roomid == roomid and (roomdata << 4) & loc_mask != 0: @@ -915,7 +918,7 @@ async def game_watcher(ctx : Context): if scout_location > 0 and scout_location not in ctx.locations_scouted: ctx.locations_scouted.add(scout_location) - logging.info(f'Scouting item at {list(Regions.location_table.keys())[scout_location - 1]}') + logging.info(f'Scouting item at {list(Regions.lookup_id_to_name.keys())[scout_location - 1]}') await send_msgs(ctx.socket, [['LocationScouts', [scout_location]]]) await track_locations(ctx, roomid, roomdata) diff --git a/MultiServer.py b/MultiServer.py index f81ad629..cd333edf 100644 --- a/MultiServer.py +++ b/MultiServer.py @@ -154,7 +154,8 @@ def send_new_items(ctx : Context): client.send_index = len(items) def forfeit_player(ctx : Context, team, slot): - all_locations = [values[0] for values in Regions.location_table.values() if type(values[0]) is int] + all_locations = {values[0] for values in Regions.location_table.values() if type(values[0]) is int} + all_locations.update({values[1] for values in Regions.key_drop_data.values()}) notify_all(ctx, "%s (Team #%d) has forfeited" % (ctx.player_names[(team, slot)], team + 1)) register_location_checks(ctx, team, slot, all_locations) @@ -248,11 +249,11 @@ async def process_client_cmd(ctx : Context, client : Client, cmd, args): return locs = [] for location in args: - if type(location) is not int or 0 >= location > len(Regions.location_table): + if type(location) is not int or 0 >= location > len(Regions.lookup_id_to_name.keys()): await send_msgs(client.socket, [['InvalidArguments', 'LocationScouts']]) return - loc_name = list(Regions.location_table.keys())[location - 1] - target_item, target_player = ctx.locations[(Regions.location_table[loc_name][0], client.slot)] + loc_name = list(Regions.lookup_id_to_name.keys())[location - 1] + target_item, target_player = ctx.locations[(Regions.lookup_name_to_id[loc_name], client.slot)] replacements = {'SmallKey': 0xA2, 'BigKey': 0x9D, 'Compass': 0x8D, 'Map': 0x7D} item_type = [i[2] for i in Items.item_table.values() if type(i[3]) is int and i[3] == target_item] diff --git a/Regions.py b/Regions.py index 10301399..de6ca1f2 100644 --- a/Regions.py +++ b/Regions.py @@ -1195,3 +1195,8 @@ location_table = {'Mushroom': (0x180013, 0x186338, False, 'in the woods'), 'Ice Palace - Prize': ([0x120A4, 0x53F5A, 0x53F5B, 0x180059, 0x180073, 0xC705], None, True, 'Ice Palace'), 'Misery Mire - Prize': ([0x120A2, 0x53F48, 0x53F49, 0x180057, 0x180075, 0xC703], None, True, 'Misery Mire'), 'Turtle Rock - Prize': ([0x120A7, 0x53F24, 0x53F25, 0x18005C, 0x180079, 0xC708], None, True, 'Turtle Rock')} + +lookup_id_to_name = {data[0]: name for name, data in location_table.items() if type(data[0]) == int} +lookup_id_to_name = {**lookup_id_to_name, **{data[1]: name for name, data in key_drop_data.items()}} +lookup_name_to_id = {name: data[0] for name, data in location_table.items() if type(data[0]) == int} +lookup_name_to_id = {**lookup_name_to_id, **{name: data[1] for name, data in key_drop_data.items()}}