mirror of
https://github.com/ArchipelagoMW/Archipelago.git
synced 2026-03-30 23:23:22 -07:00
123 lines
5.6 KiB
Python
123 lines
5.6 KiB
Python
from random import Random
|
|
from typing import Optional
|
|
from collections.abc import Iterable
|
|
from .GameLogic import GameLogic, Recipe
|
|
from .Options import SatisfactoryOptions
|
|
from .Options import SatisfactoryOptions
|
|
|
|
class CriticalPathCalculator:
|
|
logic: GameLogic
|
|
random: Random
|
|
options: SatisfactoryOptions
|
|
|
|
required_parts: set[str]
|
|
required_buildings: set[str]
|
|
required_item_names: set[str]
|
|
required_power_level: int
|
|
|
|
__potential_required_belt_speed: int
|
|
__potential_required_pipes: bool
|
|
__potential_required_radioactive: bool
|
|
|
|
def __init__(self, logic: GameLogic, random: Random, options: SatisfactoryOptions):
|
|
self.logic = logic
|
|
self.random = random
|
|
self.options = options
|
|
|
|
self.required_parts = set()
|
|
self.required_buildings = set()
|
|
self.required_power_level: int = 1
|
|
|
|
self.__potential_required_belt_speed = 1
|
|
self.__potential_required_pipes = False
|
|
self.__potential_required_radioactive = False
|
|
|
|
selected_power_infrastructure: dict[int, Recipe] = {}
|
|
|
|
self.select_minimal_required_parts_for(self.logic.space_elevator_tiers[options.final_elevator_package-1].keys())
|
|
|
|
for tree in self.logic.man_trees.values():
|
|
self.select_minimal_required_parts_for(tree.access_items)
|
|
|
|
for node in tree.nodes:
|
|
if node.minimal_tier > options.final_elevator_package:
|
|
continue
|
|
|
|
self.select_minimal_required_parts_for(node.unlock_cost.keys())
|
|
|
|
self.select_minimal_required_parts_for_building("MAM")
|
|
self.select_minimal_required_parts_for_building("AWESOME Sink")
|
|
self.select_minimal_required_parts_for_building("AWESOME Shop")
|
|
self.select_minimal_required_parts_for_building("Space Elevator")
|
|
self.select_minimal_required_parts_for_building("Conveyor Splitter")
|
|
self.select_minimal_required_parts_for_building("Conveyor Merger")
|
|
self.select_minimal_required_parts_for_building("Equipment Workshop")
|
|
self.select_minimal_required_parts_for_building("Foundation")
|
|
self.select_minimal_required_parts_for_building("Walls Orange")
|
|
self.select_minimal_required_parts_for_building("Power Storage")
|
|
self.select_minimal_required_parts_for_building("Miner Mk.2")
|
|
|
|
#equipment
|
|
self.select_minimal_required_parts_for(self.logic.recipes["Hazmat Suit"][0].inputs)
|
|
self.select_minimal_required_parts_for(self.logic.recipes["Iodine Infused Filter"][0].inputs)
|
|
|
|
for i in range(1, self.__potential_required_belt_speed + 1):
|
|
self.select_minimal_required_parts_for_building(f"Conveyor Mk.{i}")
|
|
if self.__potential_required_pipes:
|
|
self.select_minimal_required_parts_for_building("Pipes Mk.1")
|
|
self.select_minimal_required_parts_for_building("Pipes Mk.2")
|
|
self.select_minimal_required_parts_for_building("Pipeline Pump Mk.1")
|
|
self.select_minimal_required_parts_for_building("Pipeline Pump Mk.2")
|
|
if self.__potential_required_radioactive:
|
|
self.select_minimal_required_parts_for(self.logic.recipes["Hazmat Suit"][0].inputs)
|
|
self.select_minimal_required_parts_for(self.logic.recipes["Iodine Infused Filter"][0].inputs)
|
|
for i in range(1, self.required_power_level + 1):
|
|
power_recipe = random.choice(self.logic.requirement_per_powerlevel[i])
|
|
selected_power_infrastructure[i] = power_recipe
|
|
self.select_minimal_required_parts_for(power_recipe.inputs)
|
|
self.select_minimal_required_parts_for_building(power_recipe.building)
|
|
|
|
self.required_item_names = set(
|
|
recipe.name
|
|
for part in self.required_parts
|
|
for recipe in self.logic.recipes[part]
|
|
if recipe.minimal_tier <= self.options.final_elevator_package
|
|
)
|
|
self.required_item_names.update("Building: "+ building for building in self.required_buildings)
|
|
|
|
def select_minimal_required_parts_for_building(self, building: str) -> None:
|
|
self.select_minimal_required_parts_for(self.logic.buildings[building].inputs)
|
|
self.required_buildings.add(building)
|
|
|
|
def select_minimal_required_parts_for(self, parts: Optional[Iterable[str]]) -> None:
|
|
if parts is None:
|
|
return
|
|
|
|
for part in parts:
|
|
if part in self.required_parts:
|
|
continue
|
|
|
|
self.required_parts.add(part)
|
|
|
|
for recipe in self.logic.recipes[part]:
|
|
if recipe.minimal_tier > self.options.final_elevator_package:
|
|
continue
|
|
|
|
self.__potential_required_belt_speed = \
|
|
max(self.__potential_required_belt_speed, recipe.minimal_belt_speed)
|
|
|
|
self.select_minimal_required_parts_for(recipe.inputs)
|
|
|
|
if recipe.needs_pipes:
|
|
self.__potential_required_pipes = True
|
|
if recipe.is_radio_active:
|
|
self.__potential_required_radioactive = True
|
|
|
|
if recipe.building:
|
|
self.select_minimal_required_parts_for(self.logic.buildings[recipe.building].inputs)
|
|
self.required_buildings.add(recipe.building)
|
|
|
|
if self.logic.buildings[recipe.building].power_requirement:
|
|
self.required_power_level = \
|
|
max(self.required_power_level,
|
|
self.logic.buildings[recipe.building].power_requirement) |