allow setting shader object through python dictionary (#136)

shader-slang · Dec 6, 2024 · 502626d · 502626d
1 parent 2bf7a15
commit 502626d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/sgl/device/python/cursor_utils.h b/src/sgl/device/python/cursor_utils.h
@@ -383,6 +383,14 @@ class WriteConverterTable {
             if (kind != TypeReflection::Kind::struct_)
                 type_layout = type_layout->element_type_layout();
 
+            // Handle shader object if possible.
+            if constexpr (requires { self.set_object(nullptr); }) {
+                if (nb::isinstance<MutableShaderObject>(nbval)) {
+                    self.set_object(nb::cast<ref<MutableShaderObject>>(nbval));
+                    return;
+                }
+            }
+
             // Expect a dict for a slang struct.
             if (nb::isinstance<nb::dict>(nbval)) {
                 auto dict = nb::cast<nb::dict>(nbval);