126 lines
6.6 KiB
Python
126 lines
6.6 KiB
Python
from typing import Dict, List, NamedTuple, Optional
|
|
|
|
from BaseClasses import MultiWorld, Region, Entrance
|
|
from .Locations import location_table, RiskOfRainLocation
|
|
|
|
|
|
class RoRRegionData(NamedTuple):
|
|
locations: Optional[List[str]]
|
|
region_exits: Optional[List[str]]
|
|
|
|
|
|
def create_regions(multiworld: MultiWorld, player: int):
|
|
# Default Locations
|
|
non_dlc_regions: Dict[str, RoRRegionData] = {
|
|
"Menu": RoRRegionData(None, ["Distant Roost", "Distant Roost (2)", "Titanic Plains", "Titanic Plains (2)"]),
|
|
"Distant Roost": RoRRegionData([], ["OrderedStage_1"]),
|
|
"Distant Roost (2)": RoRRegionData([], ["OrderedStage_1"]),
|
|
"Titanic Plains": RoRRegionData([], ["OrderedStage_1"]),
|
|
"Titanic Plains (2)": RoRRegionData([], ["OrderedStage_1"]),
|
|
"Abandoned Aqueduct": RoRRegionData([], ["OrderedStage_2"]),
|
|
"Wetland Aspect": RoRRegionData([], ["OrderedStage_2"]),
|
|
"Rallypoint Delta": RoRRegionData([], ["OrderedStage_3"]),
|
|
"Scorched Acres": RoRRegionData([], ["OrderedStage_3"]),
|
|
"Abyssal Depths": RoRRegionData([], ["OrderedStage_4"]),
|
|
"Siren's Call": RoRRegionData([], ["OrderedStage_4"]),
|
|
"Sundered Grove": RoRRegionData([], ["OrderedStage_4"]),
|
|
"Sky Meadow": RoRRegionData([], ["Hidden Realm: Bulwark's Ambry", "OrderedStage_5"]),
|
|
}
|
|
# SOTV Regions
|
|
dlc_regions: Dict[str, RoRRegionData] = {
|
|
"Siphoned Forest": RoRRegionData([], ["OrderedStage_1"]),
|
|
"Aphelian Sanctuary": RoRRegionData([], ["OrderedStage_2"]),
|
|
"Sulfur Pools": RoRRegionData([], ["OrderedStage_3"])
|
|
}
|
|
other_regions: Dict[str, RoRRegionData] = {
|
|
"Commencement": RoRRegionData(None, ["Victory", "Petrichor V"]),
|
|
"OrderedStage_5": RoRRegionData(None, ["Hidden Realm: A Moment, Fractured", "Commencement"]),
|
|
"OrderedStage_1": RoRRegionData(None, ["Hidden Realm: Bazaar Between Time",
|
|
"Hidden Realm: Gilded Coast", "Abandoned Aqueduct", "Wetland Aspect"]),
|
|
"OrderedStage_2": RoRRegionData(None, ["Rallypoint Delta", "Scorched Acres"]),
|
|
"OrderedStage_3": RoRRegionData(None, ["Abyssal Depths", "Siren's Call", "Sundered Grove"]),
|
|
"OrderedStage_4": RoRRegionData(None, ["Sky Meadow"]),
|
|
"Hidden Realm: A Moment, Fractured": RoRRegionData(None, ["Hidden Realm: A Moment, Whole"]),
|
|
"Hidden Realm: A Moment, Whole": RoRRegionData(None, ["Victory"]),
|
|
"Void Fields": RoRRegionData(None, []),
|
|
"Victory": RoRRegionData(None, None),
|
|
"Petrichor V": RoRRegionData(None, ["Victory"]),
|
|
"Hidden Realm: Bulwark's Ambry": RoRRegionData(None, None),
|
|
"Hidden Realm: Bazaar Between Time": RoRRegionData(None, ["Void Fields"]),
|
|
"Hidden Realm: Gilded Coast": RoRRegionData(None, None)
|
|
}
|
|
dlc_other_regions: Dict[str, RoRRegionData] = {
|
|
"The Planetarium": RoRRegionData(None, ["Victory"]),
|
|
"Void Locus": RoRRegionData(None, ["The Planetarium"])
|
|
}
|
|
# Totals of each item
|
|
chests = int(multiworld.chests_per_stage[player])
|
|
shrines = int(multiworld.shrines_per_stage[player])
|
|
scavengers = int(multiworld.scavengers_per_stage[player])
|
|
scanners = int(multiworld.scanner_per_stage[player])
|
|
newt = int(multiworld.altars_per_stage[player])
|
|
all_location_regions = {**non_dlc_regions}
|
|
if multiworld.dlc_sotv[player]:
|
|
all_location_regions = {**non_dlc_regions, **dlc_regions}
|
|
|
|
# Locations
|
|
for key in all_location_regions:
|
|
if key == "Menu":
|
|
continue
|
|
# Chests
|
|
for i in range(0, chests):
|
|
all_location_regions[key].locations.append(f"{key}: Chest {i + 1}")
|
|
# Shrines
|
|
for i in range(0, shrines):
|
|
all_location_regions[key].locations.append(f"{key}: Shrine {i + 1}")
|
|
# Scavengers
|
|
if scavengers > 0:
|
|
for i in range(0, scavengers):
|
|
all_location_regions[key].locations.append(f"{key}: Scavenger {i + 1}")
|
|
# Radio Scanners
|
|
if scanners > 0:
|
|
for i in range(0, scanners):
|
|
all_location_regions[key].locations.append(f"{key}: Radio Scanner {i + 1}")
|
|
# Newt Altars
|
|
if newt > 0:
|
|
for i in range(0, newt):
|
|
all_location_regions[key].locations.append(f"{key}: Newt Altar {i + 1}")
|
|
regions_pool: Dict = {**all_location_regions, **other_regions}
|
|
|
|
# DLC Locations
|
|
if multiworld.dlc_sotv[player]:
|
|
non_dlc_regions["Menu"].region_exits.append("Siphoned Forest")
|
|
other_regions["OrderedStage_1"].region_exits.append("Aphelian Sanctuary")
|
|
other_regions["OrderedStage_2"].region_exits.append("Sulfur Pools")
|
|
other_regions["Void Fields"].region_exits.append("Void Locus")
|
|
regions_pool: Dict = {**all_location_regions, **other_regions, **dlc_other_regions}
|
|
|
|
# Create all the regions
|
|
for name, data in regions_pool.items():
|
|
multiworld.regions.append(create_region(multiworld, player, name, data))
|
|
|
|
# Connect all the regions to their exits
|
|
for name, data in regions_pool.items():
|
|
create_connections_in_regions(multiworld, player, name, data)
|
|
|
|
|
|
def create_region(multiworld: MultiWorld, player: int, name: str, data: RoRRegionData):
|
|
region = Region(name, player, multiworld)
|
|
if data.locations:
|
|
for location_name in data.locations:
|
|
location_data = location_table.get(location_name)
|
|
location = RiskOfRainLocation(player, location_name, location_data, region)
|
|
region.locations.append(location)
|
|
|
|
return region
|
|
|
|
|
|
def create_connections_in_regions(multiworld: MultiWorld, player: int, name: str, data: RoRRegionData):
|
|
region = multiworld.get_region(name, player)
|
|
if data.region_exits:
|
|
for region_exit in data.region_exits:
|
|
r_exit_stage = Entrance(player, region_exit, region)
|
|
exit_region = multiworld.get_region(region_exit, player)
|
|
r_exit_stage.connect(exit_region)
|
|
region.exits.append(r_exit_stage)
|