115 lines
3.3 KiB
Python
Executable File
115 lines
3.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import math
|
|
import collections
|
|
|
|
from fractions import Fraction
|
|
from typing import Dict, Iterable
|
|
|
|
import satisfactory
|
|
|
|
|
|
def calculate_rates(recipes: Iterable[str], remain: Dict[str,Fraction]):
|
|
required_items = collections.defaultdict(Fraction)
|
|
|
|
while remain:
|
|
for dst_name, dst_rate in remain.pop().items():
|
|
required_items[dst_name] += dst_rate
|
|
|
|
dst = recipes[dst_name]
|
|
if not recipes.is_component(dst_name):
|
|
continue
|
|
|
|
dst_recipe = recipes[dst_name]['recipes'][0]
|
|
src_recipe = recipes[dst_name]['recipes'][0]
|
|
|
|
normal_rate = Fraction(
|
|
dst_recipe['output'][dst_name],
|
|
dst_recipe['crafting_time']
|
|
)
|
|
|
|
scale = dst_rate / normal_rate
|
|
|
|
for src_name, src_count in src_recipe['input'].items():
|
|
src_rate = Fraction(
|
|
src_count,
|
|
dst_recipe['crafting_time']
|
|
) * scale
|
|
remain.append({src_name: src_rate})
|
|
|
|
return required_items
|
|
|
|
|
|
def basic_rate(recipe: Dict) -> Fraction:
|
|
"""
|
|
Calculate the rate at which the item is crafted with the default recipe.
|
|
:param recipe:
|
|
:return:
|
|
"""
|
|
for output, count in recipe['output'].items():
|
|
return Fraction(
|
|
count, recipe['crafting_time']
|
|
)
|
|
|
|
|
|
# The number of items per minute that each tier of conveyor can carry
|
|
conveyor_rates = [0, 60, 120, 270, 480, 780, 900]
|
|
|
|
|
|
def main():
|
|
recipes = satisfactory.Cookbook('data/recipes')
|
|
|
|
required_items = collections.defaultdict(Fraction)
|
|
|
|
# Create a list of items we want to make
|
|
#targets = [ 'supercomputer' ]
|
|
targets = recipes.components()
|
|
|
|
# Create an initial name:rate request for all of the target items, then
|
|
# create a plan for their creation.
|
|
remain = [{n: basic_rate(recipes[n]['recipes'][0]) for n in targets}]
|
|
required_items = calculate_rates(recipes, remain)
|
|
|
|
# Note if any particular item is (in aggregate) going to exceed the
|
|
# highest conveyor belt capacity.
|
|
for name, rate in required_items.items():
|
|
print(name, rate, float(rate * 60))
|
|
if rate * 60 > 780:
|
|
print("Rate exceeds mk5 conveyor")
|
|
|
|
# Calculate the number of machines required to build each item at the
|
|
# calculated rates.
|
|
def numberdict():
|
|
return collections.defaultdict(int)
|
|
required_machines = collections.defaultdict(numberdict)
|
|
|
|
for name, requested_rate in required_items.items():
|
|
if recipes.is_resource(name):
|
|
continue
|
|
|
|
descriptor = recipes[name]
|
|
|
|
normal_rate = Fraction(
|
|
descriptor['recipes'][0]['output'][name],
|
|
descriptor['recipes'][0]['crafting_time']
|
|
)
|
|
|
|
machine = descriptor['machine']
|
|
required_machines[machine][name] += requested_rate / normal_rate
|
|
|
|
# Calculate the power requirements for all the machines
|
|
required_power = 0
|
|
|
|
for machine, buckets in required_machines.items():
|
|
for result, rate in buckets.items():
|
|
count = int(math.ceil(rate))
|
|
required_power += count * recipes[machine]['power_usage']
|
|
print(machine, result, math.ceil(rate))
|
|
|
|
print(required_power)
|
|
print(math.ceil(required_power / 150), "fuel generators")
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|