Merge branch 'feature/graph' into develop

Author: KelSolaar

colour_hdri/network/graphs.py

@@ -94,6 +94,7 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
         self.add_input_port("orientation")
         self.add_input_port("bypass_input_transform", False)
         self.add_input_port("bypass_correct_lens_aberration", False)
+        self.add_input_port("bit_depth", "float32")
         self.add_input_port("bypass_watermark", False)
         self.add_input_port("bypass_orient", False)
 
@@ -334,6 +335,11 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
             self.nodes["WriteImage"],
             "path",
         )
+        self.connect(
+            "bit_depth",
+            self.nodes["WriteImage"],
+            "bit_depth",
+        )
 
     @notify_process_state
     def process(self, **kwargs: Any) -> None:
@@ -386,6 +392,7 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
         self.add_input_port("correct_distortion", True)
         self.add_input_port("downsample", 1)
         self.add_input_port("orientation")
+        self.add_input_port("bit_depth", "float32")
         self.add_input_port("bypass_input_transform", False)
         self.add_input_port("bypass_correct_lens_aberration", False)
         self.add_input_port("bypass_watermark", False)
@@ -637,6 +644,11 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
             self.nodes["WriteImage"],
             "path",
         )
+        self.connect(
+            "bit_depth",
+            self.nodes["WriteImage"],
+            "bit_depth",
+        )
 
     @notify_process_state
     def process(self, **kwargs: Any) -> None:
@@ -1146,6 +1158,7 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
         self.add_input_port("correct_chromatic_aberration", True)
         self.add_input_port("correct_distortion", True)
         self.add_input_port("orientation", None)
+        self.add_input_port("bit_depth", "float32")
         self.add_input_port("bypass_input_transform", False)
         self.add_input_port("bypass_correct_lens_aberration", False)
         self.add_input_port("bypass_watermark", False)
@@ -1251,6 +1264,11 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
             self.nodes["GraphRawProcessingCameraSensitivities"],
             "orientation",
         )
+        self.connect(
+            "bit_depth",
+            self.nodes["GraphRawProcessingCameraSensitivities"],
+            "bit_depth",
+        )
         self.connect(
             "bypass_input_transform",
             self.nodes["GraphRawProcessingCameraSensitivities"],

colour_hdri/network/nodes.py

@@ -335,6 +335,7 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
         self.add_input_port("path")
         self.add_input_port("exif_tags")
         self.add_input_port("output_colourspace")
+        self.add_input_port("bit_depth", "float32")
         self.add_input_port("bypass", False)
 
     @required("OpenImageIO")
@@ -374,7 +375,9 @@ def process(self, **kwargs: Any) -> None:  # noqa: ARG002
         if not _is_linear_file_format(path):
             image = output_colourspace.cctf_encoding(image)
 
-        write_image_OpenImageIO(image, path, attributes=attributes)
+        bit_depth = self.get_input("bit_depth")
+
+        write_image_OpenImageIO(image, path, bit_depth=bit_depth, attributes=attributes)
 
         self.dirty = False