107 lines
2.9 KiB
Python
Executable File
107 lines
2.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import math
|
|
import fractions
|
|
import collections
|
|
|
|
from typing import Dict
|
|
|
|
import satisfactory
|
|
|
|
|
|
def calculate_rates(recipes, remain):
|
|
required_items = collections.defaultdict(fractions.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 = fractions.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 = fractions.Fraction(
|
|
src_count,
|
|
dst_recipe['crafting_time']
|
|
) * scale
|
|
remain.append({src_name: src_rate})
|
|
|
|
return required_items
|
|
|
|
|
|
def basic_rate(recipe: Dict) -> fractions.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 fractions.Fraction(
|
|
count, recipe['crafting_time']
|
|
)
|
|
|
|
|
|
def main():
|
|
recipes = satisfactory.Cookbook('data/recipes')
|
|
|
|
required_items = collections.defaultdict(fractions.Fraction)
|
|
|
|
#targets = [ 'supercomputer' ]
|
|
targets = recipes.components()
|
|
|
|
target_name = 'supercomputer'
|
|
target = recipes[target_name]
|
|
target_recipe = target['recipes'][0]
|
|
|
|
remain = [{n: basic_rate(recipes[n]['recipes'][0]) for n in targets}]
|
|
required_items = calculate_rates(recipes, remain)
|
|
|
|
for name, rate in required_items.items():
|
|
print(name, rate, float(rate * 60))
|
|
if rate * 60 > 780:
|
|
print("Rate exceeds mk4 conveyer")
|
|
|
|
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 = fractions.Fraction(
|
|
descriptor['recipes'][0]['output'][name],
|
|
descriptor['recipes'][0]['crafting_time']
|
|
)
|
|
|
|
machine = descriptor['machine']
|
|
required_machines[machine][name] += requested_rate / normal_rate
|
|
|
|
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()
|