Implement initial support for *Graph Editor*.

colour-science · Jan 3, 2025 · 6b536bd · 6b536bd
1 parent dbda4a1
commit 6b536bd
Show file tree

Hide file tree

Showing 10 changed files with 39,345 additions and 1,114 deletions.
diff --git a/colour_hdri/examples/examples_advanced_processing_with_the_nodegraph.ipynb b/colour_hdri/examples/examples_advanced_processing_with_the_nodegraph.ipynb
diff --git a/colour_hdri/network/graph_editor.py b/colour_hdri/network/graph_editor.py
diff --git a/colour_hdri/network/graphs.py b/colour_hdri/network/graphs.py
@@ -19,6 +19,7 @@
     ExecutionNode,
     ParallelForMultiprocess,
     PortGraph,
+    notify_process_state,
 )
 
 from colour_hdri.generation import double_sigmoid_anchored_function
@@ -244,6 +245,11 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
                 output_port,
             )
 
+        self.connect(
+            "execution_input",
+            self.nodes["ConvertRawFileToDNGFile"],
+            "execution_input",
+        )
         self.connect(
             "raw_file_path",
             self.nodes["ConvertRawFileToDNGFile"],
@@ -330,6 +336,7 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
             "path",
         )
 
+    @notify_process_state
     def process(self, **kwargs: Any) -> None:
         """
         Process the node-graph.
@@ -536,6 +543,11 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
                 output_port,
             )
 
+        self.connect(
+            "execution_input",
+            self.nodes["ConvertRawFileToDNGFile"],
+            "execution_input",
+        )
         self.connect(
             "raw_file_path",
             self.nodes["ConvertRawFileToDNGFile"],
@@ -627,6 +639,7 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
             "path",
         )
 
+    @notify_process_state
     def process(self, **kwargs: Any) -> None:
         """
         Process the node-graph.
@@ -736,6 +749,11 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
                 output_port,
             )
 
+        self.connect(
+            "execution_input",
+            self.nodes["CreateImageStack"],
+            "execution_input",
+        )
         self.connect(
             "exr_file_paths",
             self.nodes["CreateImageStack"],
@@ -782,6 +800,7 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
             "path",
         )
 
+    @notify_process_state
     def process(self, **kwargs: Any) -> None:
         """
         Process the node-graph.
@@ -866,6 +885,10 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
             self.add_node(node)
 
         for connection in [
+            (
+                ("NormaliseExposure", "execution_output"),
+                ("ParallelForMultiprocess", "execution_input"),
+            ),
             (
                 ("ParallelForMultiprocess", "loop_output"),
                 ("WritePreviewImage", "execution_input"),
@@ -882,6 +905,11 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
                 output_port,
             )
 
+        self.connect(
+            "execution_input",
+            self.nodes["NormaliseExposure"],
+            "execution_input",
+        )
         self.connect(
             "array",
             self.nodes["NormaliseExposure"],
@@ -916,6 +944,7 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
             "task", _task_multiprocess_post_merge_hdr
         )
 
+    @notify_process_state
     def process(self, **kwargs: Any) -> None:
         """
         Process the node-graph.
@@ -991,6 +1020,11 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
                 output_port,
             )
 
+        self.connect(
+            "execution_input",
+            self.nodes["CreateBatches"],
+            "execution_input",
+        )
         self.connect(
             "array",
             self.nodes["CreateBatches"],
@@ -1043,6 +1077,7 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
             "results", self.nodes["GraphPostMergeHDRI"], "array"
         )
 
+    @notify_process_state
     def process(self, **kwargs: Any) -> None:
         """
         Process the node-graph.
@@ -1162,6 +1197,11 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
                 output_port,
             )
 
+        self.connect(
+            "execution_input",
+            self.nodes["ParallelForMultiprocess"],
+            "execution_input",
+        )
         self.connect(
             "array",
             self.nodes["ParallelForMultiprocess"],
@@ -1274,6 +1314,7 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
             "Watermark"
         ].set_input("include_exposure_information", False)
 
+    @notify_process_state
     def process(self, **kwargs: Any) -> None:
         """
         Process the node-graph.