From 66cb5336d354b501f7c467854d74e37ec9faffcc Mon Sep 17 00:00:00 2001
From: Simone Barbieri <me@barbierisimone.com>
Date: Tue, 13 Feb 2024 18:03:44 +0000
Subject: [PATCH] Added an option to disable composite output

---
 .../ayon_core/hosts/blender/api/render_lib.py | 29 ++++++++++++++-----
 .../server/settings/render_settings.py        |  4 +++
 2 files changed, 25 insertions(+), 8 deletions(-)

diff --git a/client/ayon_core/hosts/blender/api/render_lib.py b/client/ayon_core/hosts/blender/api/render_lib.py
index 73d327b1a8..91913f7913 100644
--- a/client/ayon_core/hosts/blender/api/render_lib.py
+++ b/client/ayon_core/hosts/blender/api/render_lib.py
@@ -55,6 +55,14 @@ def get_renderer(settings):
                     ["renderer"])
 
 
+def get_compositing(settings):
+    """Get compositing from blender settings."""
+
+    return (settings["blender"]
+                    ["RenderSettings"]
+                    ["compositing"])
+
+
 def get_render_product(output_path, name, aov_sep):
     """
     Generate the path to the render product. Blender interprets the `#`
@@ -184,7 +192,9 @@ def _create_aov_slot(name, aov_sep, slots, rpass_name, multi_exr, output_path):
     return slot, filepath
 
 
-def set_node_tree(output_path, render_product, name, aov_sep, ext, multilayer):
+def set_node_tree(
+    output_path, render_product, name, aov_sep, ext, multilayer, compositing
+):
     # Set the scene to use the compositor node tree to render
     bpy.context.scene.use_nodes = True
 
@@ -250,11 +260,12 @@ def set_node_tree(output_path, render_product, name, aov_sep, ext, multilayer):
         name, aov_sep, slots, pass_name, multi_exr, output_path)
     tree.links.new(render_layer_node.outputs["Image"], slot)
 
-    # Create a new socket for the composite output
-    pass_name = "composite"
-    comp_socket, filepath = _create_aov_slot(
-        name, aov_sep, slots, pass_name, multi_exr, output_path)
-    aov_file_products.append(("Composite", filepath))
+    if compositing:
+        # Create a new socket for the composite output
+        pass_name = "composite"
+        comp_socket, filepath = _create_aov_slot(
+            name, aov_sep, slots, pass_name, multi_exr, output_path)
+        aov_file_products.append(("Composite", filepath))
 
     # For each active render pass, we add a new socket to the output node
     # and link it
@@ -278,7 +289,7 @@ def set_node_tree(output_path, render_product, name, aov_sep, ext, multilayer):
         tree.links.remove(link)
 
     # If there's a composite node, we connect its input with the new output
-    if composite_node:
+    if compositing and composite_node:
         for link in tree.links:
             if link.to_node == composite_node:
                 tree.links.new(link.from_socket, comp_socket)
@@ -321,6 +332,7 @@ def prepare_rendering(asset_group):
     ext = get_image_format(settings)
     multilayer = get_multilayer(settings)
     renderer = get_renderer(settings)
+    compositing = get_compositing(settings)
 
     set_render_format(ext, multilayer)
     bpy.context.scene.render.engine = renderer
@@ -330,7 +342,8 @@ def prepare_rendering(asset_group):
 
     render_product = get_render_product(output_path, name, aov_sep)
     aov_file_product = set_node_tree(
-        output_path, render_product, name, aov_sep, ext, multilayer)
+        output_path, render_product, name, aov_sep,
+        ext, multilayer, compositing)
 
     # Clear the render filepath, so that the output is handled only by the
     # output node in the compositor.
diff --git a/server_addon/blender/server/settings/render_settings.py b/server_addon/blender/server/settings/render_settings.py
index a1f6d3114a..f992ea6fcc 100644
--- a/server_addon/blender/server/settings/render_settings.py
+++ b/server_addon/blender/server/settings/render_settings.py
@@ -127,6 +127,9 @@ class RenderSettingsModel(BaseSettingsModel):
         title="Renderer",
         enum_resolver=renderers_enum
     )
+    compositing: bool = SettingsField(
+        title="Enable Compositing"
+    )
     aov_list: list[str] = SettingsField(
         default_factory=list,
         enum_resolver=aov_list_enum,
@@ -149,6 +152,7 @@ DEFAULT_RENDER_SETTINGS = {
     "image_format": "exr",
     "multilayer_exr": True,
     "renderer": "CYCLES",
+    "compositing": True,
     "aov_list": ["combined"],
     "custom_passes": []
 }