From d13b74cb29b41d8fb94a7b6e3fbdd0146619ade9 Mon Sep 17 00:00:00 2001
From: Toke Stuart Jepsen <tokejepsen@gmail.com>
Date: Thu, 16 Feb 2023 10:47:51 +0000
Subject: [PATCH] Support more parameters.

---
 .../tools/mayalookassigner/arnold_standin.py  | 65 +++++++++++++++++++
 1 file changed, 65 insertions(+)

diff --git a/openpype/tools/mayalookassigner/arnold_standin.py b/openpype/tools/mayalookassigner/arnold_standin.py
index 0ba1526cda..b7568ec45f 100644
--- a/openpype/tools/mayalookassigner/arnold_standin.py
+++ b/openpype/tools/mayalookassigner/arnold_standin.py
@@ -25,11 +25,49 @@ log = logging.getLogger(__name__)
 
 
 ATTRIBUTE_MAPPING = {
+    "primaryVisibility": "visibility",  # Camera
+    "castsShadows": "visibility",  # Shadow
+    "receiveShadows": "receive_shadows",
+    "aiSelfShadows": "self_shadows",
+    "aiOpaque": "opaque",
+    "aiMatte": "matte",
+    "aiVisibleInDiffuseTransmission": "visibility",
+    "aiVisibleInSpecularTransmission": "visibility",
+    "aiVisibleInVolume": "visibility",
+    "aiVisibleInDiffuseReflection": "visibility",
+    "aiVisibleInSpecularReflection": "visibility",
+    "aiSubdivUvSmoothing": "subdiv_uv_smoothing",
+    "aiDispHeight": "disp_height",
+    "aiDispPadding": "disp_padding",
+    "aiDispZeroValue": "disp_zero_value",
+    "aiStepSize": "step_size",
+    "aiVolumePadding": "volume_padding",
     "aiSubdivType": "subdiv_type",
     "aiSubdivIterations": "subdiv_iterations"
 }
 
 
+def calculate_visibility_mask(attributes):
+    # https://arnoldsupport.com/2018/11/21/backdoor-setting-visibility/
+    mapping = {
+        "primaryVisibility": 1,  # Camera
+        "castsShadows": 2,  # Shadow
+        "aiVisibleInDiffuseTransmission": 4,
+        "aiVisibleInSpecularTransmission": 8,
+        "aiVisibleInVolume": 16,
+        "aiVisibleInDiffuseReflection": 32,
+        "aiVisibleInSpecularReflection": 64
+    }
+    mask = 255
+    for attr, value in mapping.items():
+        if attributes.get(attr, True):
+            continue
+
+        mask -= value
+
+    return mask
+
+
 def get_cbid_by_node(path):
     """Get cbid from Arnold Scene Source.
 
@@ -216,6 +254,7 @@ def assign_look(standin, subset):
                 )
 
             if edit["action"] == "setattr":
+                visibility = False
                 for attr, value in edit["attributes"].items():
                     if attr not in ATTRIBUTE_MAPPING:
                         log.warning(
@@ -227,11 +266,37 @@ def assign_look(standin, subset):
                     if isinstance(value, str):
                         value = "'{}'".format(value)
 
+                    if ATTRIBUTE_MAPPING[attr] == "visibility":
+                        visibility = True
+                        continue
+
                     assignment = "{}={}".format(ATTRIBUTE_MAPPING[attr], value)
 
                     for node in edit["nodes"]:
                         node_assignments[node].append(assignment)
 
+                if visibility:
+                    # https://arnoldsupport.com/2018/11/21/backdoor-setting-visibility/
+                    mapping = {
+                        "primaryVisibility": 1,  # Camera
+                        "castsShadows": 2,  # Shadow
+                        "aiVisibleInDiffuseTransmission": 4,
+                        "aiVisibleInSpecularTransmission": 8,
+                        "aiVisibleInVolume": 16,
+                        "aiVisibleInDiffuseReflection": 32,
+                        "aiVisibleInSpecularReflection": 64
+                    }
+                    mask = 255
+                    for attr, value in mapping.items():
+                        if edit["attributes"].get(attr, True):
+                            continue
+                        mask -= value
+
+                    assignment = "visibility={}".format(mask)
+
+                    for node in edit["nodes"]:
+                        node_assignments[node].append(assignment)
+
         # Assign shader
         # Clear all current shader assignments
         plug = standin + ".operators"