A fully working version of the scheduling language with added examples

transformation/schedule/schedule_lib/funcs.py

@@ -0,0 +1,56 @@
+from typing import Callable, List
+
+from jinja2 import Template
+
+from .singleton import Singleton
+
+
+class IdGenerator(metaclass=Singleton):
+    exec_id = -1
+    node_id = -1
+
+    @classmethod
+    def generate_node_id(cls) -> int:
+        cls.node_id +=1
+        return cls.node_id
+
+    @classmethod
+    def generate_exec_id(cls) -> int:
+        cls.exec_id += 1
+        return cls.exec_id
+
+def generate_dot_wrap(func) -> Callable:
+    def wrapper(self, *args, **kwargs) -> str:
+        nodes = []
+        edges = []
+        self.reset_visited()
+        func(self, nodes, edges, *args, **kwargs)
+        return f"digraph G {{\n\t{"\n\t".join(nodes)}\n\t{"\n\t".join(edges)}\n}}"
+
+    return wrapper
+
+
+def not_visited(func) -> Callable:
+    def wrapper(
+        self, nodes: List[str], edges: List[str], visited: set[int], *args, **kwargs
+    ) -> None:
+        if self in visited:
+            return
+        visited.add(self)
+        func(self, nodes, edges, visited, *args, **kwargs)
+
+    return wrapper
+
+
+def generate_dot_node(self, nodes: List[str], template: Template, **kwargs) -> None:
+    nodes.append(template.module.__getattribute__("Node")(**{**kwargs, "id": self.id}))
+
+
+def generate_dot_edge(
+    self, target, edges: List[str], template: Template, kwargs
+) -> None:
+    edges.append(
+        template.module.__getattribute__("Edge")(
+            **{**kwargs, "from_id": self.id, "to_id": target.id}
+        )
+    )