satisfactory/plan.py

#!/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()