From 83929b43337c3adad48309704c2f248427464288 Mon Sep 17 00:00:00 2001
From: Danny Robson <danny@nerdcruft.net>
Date: Sun, 1 Mar 2020 12:05:55 +1100
Subject: [PATCH] layout: add initial layout test script

---
 layout.py | 75 +++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 75 insertions(+)
 create mode 100755 layout.py

diff --git a/layout.py b/layout.py
new file mode 100755
index 0000000..a0679da
--- /dev/null
+++ b/layout.py
@@ -0,0 +1,75 @@
+#!/usr/bin/env python3
+
+"""
+This script takes a list of target components and tries to visualise a
+(hopefully efficient) factory layout to produce the components.
+"""
+
+import satisfactory
+
+import os
+
+from dataclasses import dataclass
+from typing import Iterable, Dict
+
+
+@dataclass
+class entry:
+    rank: int = -1
+
+    def fold(self, rhs):
+        self.rank = max(self.rank, rhs.rank)
+
+    def advance(self):
+        self.rank += 1
+
+
+class layout(object):
+    cache: Dict[str, entry] = dict()
+    cookbook: satisfactory.Cookbook
+
+    def __init__(self, cookbook):
+        self.cookbook = cookbook
+
+    def score(self, target: str):
+        if target in self.cache:
+            return self.cache[target]
+
+        res = entry()
+        for required in self.cookbook.recipes(target)[0].input:
+            res.fold(self.score(required))
+
+        res.advance()
+        self.cache[target] = res
+        return res
+
+
+if __name__ == '__main__':
+    def main():
+        import argparse
+
+        root = os.path.dirname(__file__)
+        recipe_root = os.path.join(root, 'data', 'recipes')
+
+        parser = argparse.ArgumentParser()
+        parser.add_argument('--data', type=str, default=recipe_root)
+        parser.add_argument('--target', type=str, nargs="*")
+        parser.add_argument('--prefer', type=str, nargs="*")
+
+        args = parser.parse_args()
+
+        recipes = satisfactory.Cookbook(args.data)
+
+        if args.prefer:
+            for alternative in args.prefer:
+                recipes.prefer(alternative)
+
+        l = layout(recipes)
+        targets = args.target if args.target else recipes.components()
+        for t in targets:
+            l.score(t)
+
+        for name, data in l.cache.items():
+            print(name, data)
+
+    main()