mirror of
https://github.com/MarioSpore/Grinch-AP.git
synced 2025-10-21 20:21:32 -06:00
568a71cdbe
(There's still a lot of work ahead, such as: registering locations and items to the World, as well as methods to create_item_from_name() many more method names for various stages embedding Options into the world type and many more...)
194 lines
7.4 KiB
Python
194 lines
7.4 KiB
Python
from typing import Dict, List, Set
|
|
|
|
from BaseClasses import MultiWorld
|
|
from Options import TechTreeLayout
|
|
|
|
funnel_layers = {TechTreeLayout.option_small_funnels: 3,
|
|
TechTreeLayout.option_medium_funnels: 4,
|
|
TechTreeLayout.option_large_funnels: 5}
|
|
|
|
funnel_slice_sizes = {TechTreeLayout.option_small_funnels: 6,
|
|
TechTreeLayout.option_medium_funnels: 10,
|
|
TechTreeLayout.option_large_funnels: 15}
|
|
|
|
def get_shapes(world: MultiWorld, player: int) -> Dict[str, List[str]]:
|
|
prerequisites: Dict[str, Set[str]] = {}
|
|
layout = world.tech_tree_layout[player].value
|
|
custom_technologies = world.custom_data[player]["custom_technologies"]
|
|
tech_names: List[str] = list(set(custom_technologies) - world.worlds[player].static_nodes)
|
|
tech_names.sort()
|
|
world.random.shuffle(tech_names)
|
|
|
|
if layout == TechTreeLayout.option_small_diamonds:
|
|
slice_size = 4
|
|
while len(tech_names) > slice_size:
|
|
slice = tech_names[:slice_size]
|
|
tech_names = tech_names[slice_size:]
|
|
slice.sort(key=lambda tech_name: len(custom_technologies[tech_name].get_prior_technologies()))
|
|
diamond_0, diamond_1, diamond_2, diamond_3 = slice
|
|
|
|
# 0 |
|
|
# 1 2 |
|
|
# 3 V
|
|
prerequisites[diamond_3] = {diamond_1, diamond_2}
|
|
prerequisites[diamond_2] = prerequisites[diamond_1] = {diamond_0}
|
|
|
|
elif layout == TechTreeLayout.option_medium_diamonds:
|
|
slice_size = 9
|
|
while len(tech_names) > slice_size:
|
|
slice = tech_names[:slice_size]
|
|
tech_names = tech_names[slice_size:]
|
|
slice.sort(key=lambda tech_name: len(custom_technologies[tech_name].get_prior_technologies()))
|
|
|
|
# 0 |
|
|
# 1 2 |
|
|
# 3 4 5 |
|
|
# 6 7 |
|
|
# 8 V
|
|
|
|
prerequisites[slice[1]] = {slice[0]}
|
|
prerequisites[slice[2]] = {slice[0]}
|
|
|
|
prerequisites[slice[3]] = {slice[1]}
|
|
prerequisites[slice[4]] = {slice[1], slice[2]}
|
|
prerequisites[slice[5]] = {slice[2]}
|
|
|
|
prerequisites[slice[6]] = {slice[3], slice[4]}
|
|
prerequisites[slice[7]] = {slice[4], slice[5]}
|
|
|
|
prerequisites[slice[8]] = {slice[6], slice[7]}
|
|
|
|
elif layout == TechTreeLayout.option_large_diamonds:
|
|
slice_size = 16
|
|
while len(tech_names) > slice_size:
|
|
slice = tech_names[:slice_size]
|
|
tech_names = tech_names[slice_size:]
|
|
slice.sort(key=lambda tech_name: len(custom_technologies[tech_name].get_prior_technologies()))
|
|
|
|
# 0 |
|
|
# 1 2 |
|
|
# 3 4 5 |
|
|
# 6 7 8 9 |
|
|
# 10 11 12 |
|
|
# 13 14 |
|
|
# 15 |
|
|
|
|
prerequisites[slice[1]] = {slice[0]}
|
|
prerequisites[slice[2]] = {slice[0]}
|
|
|
|
prerequisites[slice[3]] = {slice[1]}
|
|
prerequisites[slice[4]] = {slice[1], slice[2]}
|
|
prerequisites[slice[5]] = {slice[2]}
|
|
|
|
prerequisites[slice[6]] = {slice[3]}
|
|
prerequisites[slice[7]] = {slice[3], slice[4]}
|
|
prerequisites[slice[8]] = {slice[4], slice[5]}
|
|
prerequisites[slice[9]] = {slice[5]}
|
|
|
|
prerequisites[slice[10]] = {slice[6], slice[7]}
|
|
prerequisites[slice[11]] = {slice[7], slice[8]}
|
|
prerequisites[slice[12]] = {slice[8], slice[9]}
|
|
|
|
prerequisites[slice[13]] = {slice[10], slice[11]}
|
|
prerequisites[slice[14]] = {slice[11], slice[12]}
|
|
|
|
prerequisites[slice[15]] = {slice[13], slice[14]}
|
|
|
|
elif layout == TechTreeLayout.option_small_pyramids:
|
|
slice_size = 6
|
|
while len(tech_names) > slice_size:
|
|
slice = tech_names[:slice_size]
|
|
tech_names = tech_names[slice_size:]
|
|
slice.sort(key=lambda tech_name: len(custom_technologies[tech_name].get_prior_technologies()))
|
|
|
|
# 0 |
|
|
# 1 2 |
|
|
# 3 4 5 |
|
|
|
|
prerequisites[slice[1]] = {slice[0]}
|
|
prerequisites[slice[2]] = {slice[0]}
|
|
|
|
prerequisites[slice[3]] = {slice[1]}
|
|
prerequisites[slice[4]] = {slice[1], slice[2]}
|
|
prerequisites[slice[5]] = {slice[2]}
|
|
|
|
elif layout == TechTreeLayout.option_medium_pyramids:
|
|
slice_size = 10
|
|
while len(tech_names) > slice_size:
|
|
slice = tech_names[:slice_size]
|
|
tech_names = tech_names[slice_size:]
|
|
slice.sort(key=lambda tech_name: len(custom_technologies[tech_name].get_prior_technologies()))
|
|
|
|
# 0 |
|
|
# 1 2 |
|
|
# 3 4 5 |
|
|
# 6 7 8 9 |
|
|
|
|
|
|
prerequisites[slice[1]] = {slice[0]}
|
|
prerequisites[slice[2]] = {slice[0]}
|
|
|
|
prerequisites[slice[3]] = {slice[1]}
|
|
prerequisites[slice[4]] = {slice[1], slice[2]}
|
|
prerequisites[slice[5]] = {slice[2]}
|
|
|
|
prerequisites[slice[6]] = {slice[3]}
|
|
prerequisites[slice[7]] = {slice[3], slice[4]}
|
|
prerequisites[slice[8]] = {slice[4], slice[5]}
|
|
prerequisites[slice[9]] = {slice[5]}
|
|
|
|
elif layout == TechTreeLayout.option_large_pyramids:
|
|
slice_size = 15
|
|
while len(tech_names) > slice_size:
|
|
slice = tech_names[:slice_size]
|
|
tech_names = tech_names[slice_size:]
|
|
slice.sort(key=lambda tech_name: len(custom_technologies[tech_name].get_prior_technologies()))
|
|
|
|
# 0 |
|
|
# 1 2 |
|
|
# 3 4 5 |
|
|
# 6 7 8 9 |
|
|
# 10 11 12 13 14 |
|
|
|
|
|
|
prerequisites[slice[1]] = {slice[0]}
|
|
prerequisites[slice[2]] = {slice[0]}
|
|
|
|
prerequisites[slice[3]] = {slice[1]}
|
|
prerequisites[slice[4]] = {slice[1], slice[2]}
|
|
prerequisites[slice[5]] = {slice[2]}
|
|
|
|
prerequisites[slice[6]] = {slice[3]}
|
|
prerequisites[slice[7]] = {slice[3], slice[4]}
|
|
prerequisites[slice[8]] = {slice[4], slice[5]}
|
|
prerequisites[slice[9]] = {slice[5]}
|
|
|
|
prerequisites[slice[10]] = {slice[6]}
|
|
prerequisites[slice[11]] = {slice[6], slice[7]}
|
|
prerequisites[slice[12]] = {slice[7], slice[8]}
|
|
prerequisites[slice[13]] = {slice[8], slice[9]}
|
|
prerequisites[slice[14]] = {slice[9]}
|
|
|
|
elif layout in funnel_layers:
|
|
slice_size = funnel_slice_sizes[layout]
|
|
|
|
world.random.shuffle(tech_names)
|
|
tech_names.sort(key=lambda tech_name: len(custom_technologies[tech_name].get_prior_technologies()))
|
|
|
|
while len(tech_names) > slice_size:
|
|
tech_names = tech_names[slice_size:]
|
|
current_tech_names = tech_names[:slice_size]
|
|
layer_size = funnel_layers[layout]
|
|
previous_slice = []
|
|
for layer in range(funnel_layers[layout]):
|
|
slice = current_tech_names[:layer_size]
|
|
current_tech_names = current_tech_names[layer_size:]
|
|
if previous_slice:
|
|
for i, tech_name in enumerate(slice):
|
|
prerequisites.setdefault(tech_name, set()).update(previous_slice[i:i+2])
|
|
previous_slice = slice
|
|
layer_size -= 1
|
|
|
|
world.tech_tree_layout_prerequisites[player] = prerequisites
|
|
return prerequisites
|