New branch-generator system

play.py

@@ -1,15 +1,47 @@
-from enum import Enum, auto
+from typing import Any, Generator
+import scgenerator as sc
+import itertools
+
+import numpy as np
 
 
-class Test:
-    class State(Enum):
-        complete = auto()
-        partial = auto()
-        absent = auto()
+class DataPather:
+    def __init__(self, dl: list[dict[str, Any]]):
+        self.dict_list = dl
+        self.n = len(self.dict_list)
+        self.final_list = list(self.dico_iterator(self.n))
 
-    def state(self):
-        return self.State.complete
+    def dico_iterator(self, index: int) -> Generator[list[list[tuple[str, Any]]], None, None]:
+        d_tem_list = [el for d in self.dict_list[: index + 1] for el in d.items()]
+        dict_pos = np.cumsum([0] + [len(d) for d in self.dict_list[: index + 1]])
+        ranges = [range(len(l)) for _, l in d_tem_list]
+
+        for r in itertools.product(*ranges):
+            flat = [(d_tem_list[i][0], d_tem_list[i][1][j]) for i, j in enumerate(r)]
+            out = [flat[left:right] for left, right in zip(dict_pos[:-1], dict_pos[1:])]
+            yield out
+
+    def all_vary_list(self, index):
+        for l in self.dico_iterator(index):
+            yield sc.utils.parameter.format_variable_list(
+                sc.utils.parameter.reduce_all_variable(l[:index])
+            ), sc.utils.parameter.format_variable_list(
+                sc.utils.parameter.reduce_all_variable(l)
+            ), l[
+                index
+            ]
 
 
-a = Test()
-print(a.state() == Test.State.complete)
+configs, name = sc.utils.load_config_sequence(
+    "/Users/benoitsierro/Nextcloud/PhD/Supercontinuum/PCF Simulations/Test/NewStyle.toml"
+)
+
+dp = DataPather([config["variable"] for config in configs])
+# pprint(list(dp.dico_iterator(1)))
+for i in range(3):
+    for prev_path, this_path, this_vary in dp.all_vary_list(i):
+        print(prev_path)
+        print(this_path)
+        print(this_vary)
+        print()
+    print()