satisfactory/plan.py

84 lines
2.4 KiB
Python
Executable File

#!/usr/bin/env python3
import math
import fractions
import collections
import satisfactory
if __name__ == '__main__':
recipes = satisfactory.Cookbook('data/recipes')
required_items = collections.defaultdict(fractions.Fraction)
remain = []
target_name = 'supercomputer'
target = recipes[target_name]
target_recipe = target['recipes'][0]
remain.append({
target_name: fractions.Fraction(
target_recipe['output'][target_name],
target_recipe['crafting_time']
)
})
while remain:
for dst_name, dst_rate in remain.pop().items():
required_items[dst_name] += dst_rate
dst = recipes[dst_name]
if recipes.is_resource(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})
for name, rate in required_items.items():
print(name, rate, float(rate * 60))
if rate * 60 > 450:
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")