150 lines
		
	
	
		
			8.2 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			150 lines
		
	
	
		
			8.2 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| from .data import ename, iname, rname
 | |
| from .stages import get_stage_info
 | |
| from .options import CV64Options
 | |
| 
 | |
| from typing import Dict, List, Tuple, Union
 | |
| 
 | |
| # # #    KEY    # # #
 | |
| # "connection" = The name of the Region the Entrance connects into. If it's a Tuple[str, str], we take the stage in
 | |
| #                active_stage_exits given in the second string and then the stage given in that stage's slot given in
 | |
| #                the first string, and take the start or end Region of that stage.
 | |
| # "rule" = What rule should be applied to the Entrance during set_rules, as defined in self.rules in the CV64Rules class
 | |
| #          definition in rules.py.
 | |
| # "add conds" = A list of player options conditions that must be satisfied for the Entrance to be added. Can be of
 | |
| #               varying length depending on how many conditions need to be satisfied. In the add_conds dict's tuples,
 | |
| #               the first element is the name of the option, the second is the option value to check for, and the third
 | |
| #               is a boolean for whether we are evaluating for the option value or not.
 | |
| entrance_info = {
 | |
|     # Forest of Silence
 | |
|     ename.forest_dbridge_gate: {"connection": rname.forest_mid},
 | |
|     ename.forest_werewolf_gate: {"connection": rname.forest_end},
 | |
|     ename.forest_end: {"connection": ("next", rname.forest_of_silence)},
 | |
|     # Castle Wall
 | |
|     ename.cw_portcullis_c: {"connection": rname.cw_exit},
 | |
|     ename.cw_lt_skip: {"connection": ("next", rname.castle_wall), "add conds": ["hard"]},
 | |
|     ename.cw_lt_door: {"connection": rname.cw_ltower, "rule": iname.left_tower_key},
 | |
|     ename.cw_end: {"connection": ("next", rname.castle_wall)},
 | |
|     # Villa
 | |
|     ename.villa_dog_gates: {"connection": rname.villa_main},
 | |
|     ename.villa_snipe_dogs: {"connection": rname.villa_start, "add conds": ["carrie", "hard"]},
 | |
|     ename.villa_to_storeroom: {"connection": rname.villa_storeroom, "rule": iname.storeroom_key},
 | |
|     ename.villa_to_archives: {"connection": rname.villa_archives, "rule": iname.archives_key},
 | |
|     ename.villa_renon: {"connection": rname.renon, "add conds": ["shopsanity"]},
 | |
|     ename.villa_to_maze: {"connection": rname.villa_maze, "rule": iname.garden_key},
 | |
|     ename.villa_from_storeroom: {"connection": rname.villa_main, "rule": iname.storeroom_key},
 | |
|     ename.villa_from_maze: {"connection": rname.villa_servants, "rule": iname.garden_key},
 | |
|     ename.villa_servant_door: {"connection": rname.villa_main},
 | |
|     ename.villa_copper_door: {"connection": rname.villa_crypt, "rule": iname.copper_key,
 | |
|                               "add conds": ["not hard"]},
 | |
|     ename.villa_copper_skip: {"connection": rname.villa_crypt, "add conds": ["hard"]},
 | |
|     ename.villa_bridge_door: {"connection": rname.villa_maze},
 | |
|     ename.villa_end_r: {"connection": ("next", rname.villa)},
 | |
|     ename.villa_end_c: {"connection": ("alt", rname.villa)},
 | |
|     # Tunnel
 | |
|     ename.tunnel_start_renon: {"connection": rname.renon, "add conds": ["shopsanity"]},
 | |
|     ename.tunnel_gondolas: {"connection": rname.tunnel_end},
 | |
|     ename.tunnel_end_renon: {"connection": rname.renon, "add conds": ["shopsanity"]},
 | |
|     ename.tunnel_end: {"connection": ("next", rname.tunnel)},
 | |
|     # Underground Waterway
 | |
|     ename.uw_renon: {"connection": rname.renon, "add conds": ["shopsanity"]},
 | |
|     ename.uw_final_waterfall: {"connection": rname.uw_end},
 | |
|     ename.uw_waterfall_skip: {"connection": rname.uw_main, "add conds": ["hard"]},
 | |
|     ename.uw_end: {"connection": ("next", rname.underground_waterway)},
 | |
|     # Castle Center
 | |
|     ename.cc_tc_door: {"connection": rname.cc_torture_chamber, "rule": iname.chamber_key},
 | |
|     ename.cc_renon: {"connection": rname.renon, "add conds": ["shopsanity"]},
 | |
|     ename.cc_lower_wall: {"connection": rname.cc_crystal, "rule": "Bomb 2"},
 | |
|     ename.cc_upper_wall: {"connection": rname.cc_library, "rule": "Bomb 1"},
 | |
|     ename.cc_elevator: {"connection": rname.cc_elev_top},
 | |
|     ename.cc_exit_r: {"connection": ("next", rname.castle_center)},
 | |
|     ename.cc_exit_c: {"connection": ("alt", rname.castle_center)},
 | |
|     # Duel Tower
 | |
|     ename.dt_start: {"connection": ("prev", rname.duel_tower)},
 | |
|     ename.dt_end: {"connection": ("next", rname.duel_tower)},
 | |
|     # Tower of Execution
 | |
|     ename.toe_start: {"connection": ("prev", rname.tower_of_execution)},
 | |
|     ename.toe_gate: {"connection": rname.toe_ledge, "rule": iname.execution_key,
 | |
|                      "add conds": ["not hard"]},
 | |
|     ename.toe_gate_skip: {"connection": rname.toe_ledge, "add conds": ["hard"]},
 | |
|     ename.toe_end: {"connection": ("next", rname.tower_of_execution)},
 | |
|     # Tower of Science
 | |
|     ename.tosci_start: {"connection": ("prev", rname.tower_of_science)},
 | |
|     ename.tosci_key1_door: {"connection": rname.tosci_three_doors, "rule": iname.science_key1},
 | |
|     ename.tosci_to_key2_door: {"connection": rname.tosci_conveyors, "rule": iname.science_key2},
 | |
|     ename.tosci_from_key2_door: {"connection": rname.tosci_start, "rule": iname.science_key2},
 | |
|     ename.tosci_key3_door: {"connection": rname.tosci_key3, "rule": iname.science_key3},
 | |
|     ename.tosci_end: {"connection": ("next", rname.tower_of_science)},
 | |
|     # Tower of Sorcery
 | |
|     ename.tosor_start: {"connection": ("prev", rname.tower_of_sorcery)},
 | |
|     ename.tosor_end: {"connection": ("next", rname.tower_of_sorcery)},
 | |
|     # Room of Clocks
 | |
|     ename.roc_gate: {"connection": ("next", rname.room_of_clocks)},
 | |
|     # Clock Tower
 | |
|     ename.ct_to_door1: {"connection": rname.ct_middle, "rule": iname.clocktower_key1},
 | |
|     ename.ct_from_door1: {"connection": rname.ct_start, "rule": iname.clocktower_key1},
 | |
|     ename.ct_to_door2: {"connection": rname.ct_end, "rule": iname.clocktower_key2},
 | |
|     ename.ct_from_door2: {"connection": rname.ct_middle, "rule": iname.clocktower_key2},
 | |
|     ename.ct_renon: {"connection": rname.renon, "add conds": ["shopsanity"]},
 | |
|     ename.ct_door_3: {"connection": ("next", rname.clock_tower), "rule": iname.clocktower_key3},
 | |
|     # Castle Keep
 | |
|     ename.ck_slope_jump: {"connection": rname.roc_main, "add conds": ["hard"]},
 | |
|     ename.ck_drac_door: {"connection": rname.ck_drac_chamber, "rule": "Dracula"}
 | |
| }
 | |
| 
 | |
| add_conds = {"carrie": ("carrie_logic", True, True),
 | |
|              "hard": ("hard_logic", True, True),
 | |
|              "not hard": ("hard_logic", False, True),
 | |
|              "shopsanity": ("shopsanity", True, True)}
 | |
| 
 | |
| stage_connection_types = {"prev": "end region",
 | |
|                           "next": "start region",
 | |
|                           "alt": "start region"}
 | |
| 
 | |
| 
 | |
| def get_entrance_info(entrance: str, info: str) -> Union[str, Tuple[str, str], List[str], None]:
 | |
|     return entrance_info[entrance].get(info, None)
 | |
| 
 | |
| 
 | |
| def get_warp_entrances(active_warp_list: List[str]) -> Dict[str, str]:
 | |
|     # Create the starting stage Entrance.
 | |
|     warp_entrances = {get_stage_info(active_warp_list[0], "start region"): "Start stage"}
 | |
| 
 | |
|     # Create the warp Entrances.
 | |
|     for i in range(1, len(active_warp_list)):
 | |
|         mid_stage_region = get_stage_info(active_warp_list[i], "mid region")
 | |
|         warp_entrances.update({mid_stage_region: f"Warp {i}"})
 | |
| 
 | |
|     return warp_entrances
 | |
| 
 | |
| 
 | |
| def verify_entrances(options: CV64Options, entrances: List[str],
 | |
|                      active_stage_exits: Dict[str, Dict[str, Union[str, int, None]]]) -> Dict[str, str]:
 | |
|     verified_entrances = {}
 | |
| 
 | |
|     for ent_name in entrances:
 | |
|         ent_add_conds = get_entrance_info(ent_name, "add conds")
 | |
| 
 | |
|         # Check any options that might be associated with the Entrance before adding it.
 | |
|         add_it = True
 | |
|         if ent_add_conds is not None:
 | |
|             for cond in ent_add_conds:
 | |
|                 if not ((getattr(options, add_conds[cond][0]).value == add_conds[cond][1]) == add_conds[cond][2]):
 | |
|                     add_it = False
 | |
| 
 | |
|         if not add_it:
 | |
|             continue
 | |
| 
 | |
|         # Add the Entrance to the verified Entrances if the above check passes.
 | |
|         connection = get_entrance_info(ent_name, "connection")
 | |
| 
 | |
|         # If the Entrance is a connection to a different stage, get the corresponding other stage Region.
 | |
|         if isinstance(connection, tuple):
 | |
|             connecting_stage = active_stage_exits[connection[1]][connection[0]]
 | |
|             # Stages that lead backwards at the beginning of the line will appear leading to "Menu".
 | |
|             if connecting_stage in ["Menu", None]:
 | |
|                 continue
 | |
|             connection = get_stage_info(connecting_stage, stage_connection_types[connection[0]])
 | |
|         verified_entrances.update({connection: ent_name})
 | |
| 
 | |
|     return verified_entrances
 | 
