add vray render creator

2026-01-02 00:44:52 +01:00 · 2023-04-12 22:28:15 +08:00 · 2023-04-12 22:28:15 +08:00 · d925322aa4
commit d925322aa4
parent 80a5982a24
1 changed files with 124 additions and 0 deletions
--- a/openpype/hosts/houdini/plugins/create/create_vray_rop.py
+++ b/openpype/hosts/houdini/plugins/create/create_vray_rop.py
@ -0,0 +1,124 @@
+# -*- coding: utf-8 -*-
+"""Creator plugin to create VRay ROP."""
+from openpype.hosts.houdini.api import plugin
+from openpype.pipeline import CreatedInstance
+from openpype.lib import EnumDef, BoolDef
+
+
+class CreateVrayROP(plugin.HoudiniCreator):
+    """VRay ROP"""
+
+    identifier = "io.openpype.creators.houdini.vray_rop"
+    label = "VRay ROP"
+    family = "vray_rop"
+    icon = "magic"
+    defaults = ["master"]
+
+    ext = "exr"
+
+    def create(self, subset_name, instance_data, pre_create_data):
+        import hou  #
+
+        instance_data.pop("active", None)
+        instance_data.update({"node_type": "vray_renderer"})
+        # Add chunk size attribute
+        instance_data["chunkSize"] = 10
+
+        instance = super(CreateVrayROP, self).create(
+            subset_name,
+            instance_data,
+            pre_create_data)  # type: CreatedInstance
+
+        instance_node = hou.node(instance.get("instance_node"))
+
+        # Add IPR for Vray
+        basename = instance_node.name()
+        try:
+            ipr_rop = instance_node.parent().createNode(
+                "vray", node_name=basename + "_IPR"
+            )
+        except hou.OperationFailed:
+            raise plugin.OpenPypeCreatorError(
+                "Cannot create Vray render node. "
+                "Make sure Vray installed and enabled!"
+            )
+
+        ipr_rop.setPosition(instance_node.position() + hou.Vector2(0, -1))
+        ipr_rop.parm("rop").set(instance_node.path())
+
+        ext = pre_create_data.get("image_format")
+
+        filepath = "{}{}".format(
+            hou.text.expandString("$HIP/pyblish/"),
+            "{}.$F4.{}".format(subset_name, ext)
+        )
+
+        parms = {
+            "trange": 1,
+            "SettingsOutput_img_file_path": filepath,
+             "SettingsEXR_bits_per_channel": "16"   # half precision
+        }
+
+        if self.selected_nodes:
+        # set up the render camera from the selected node
+            camera = None
+            for node in self.selected_nodes:
+                if node.type().name() == "cam":
+                    camera = node.path()
+            parms.update({
+                "render_camera": camera or ""
+            })
+
+        #TODO:Add Render Element
+        has_re = pre_create_data.get("render_element_enabled")
+        if has_re:
+            re_rop = instance_node.parent().createNode(
+                "vray_render_channels",
+                node_name=basename + "_render_element"
+            )
+            # move the render element node next to the vray renderer node
+            re_rop.setPosition(instance_node.position() + hou.Vector2(0, 1))
+            re_path = re_rop.path()
+            parms.update({
+                "use_render_channels": 1,
+                "render_network_render_channels": re_path
+            })
+        else:
+            parms.update({
+                "use_render_channels": 0
+            })
+
+        custom_res = pre_create_data.get("override_resolution")
+        if custom_res:
+            parms.update({"override_camerares": 1})
+
+        instance_node.setParms(parms)
+
+        # lock parameters from AVALON
+        to_lock = ["family", "id"]
+        self.lock_parameters(instance_node, to_lock)
+
+    def get_pre_create_attr_defs(self):
+        attrs = super(CreateVrayROP, self).get_pre_create_attr_defs()
+        image_format_enum = [
+            "bmp", "cin", "exr", "jpg", "pic", "pic.gz", "png",
+            "rad", "rat", "rta", "sgi", "tga", "tif",
+        ]
+
+        return attrs + [
+            EnumDef("image_format",
+                    image_format_enum,
+                    default=self.ext,
+                    label="Image Format Options"),
+            BoolDef("override_resolution",
+                    label="Override Camera Resolution",
+                    tooltip="Override the current camera "
+                            "resolution, recommended for IPR.",
+                    default=False),
+            BoolDef("render_element_enabled",
+                    label="Render Element",
+                    tooltip="Create Render Element Node "
+                            "if enabled",
+                    default=False)
+        ]
+# ${HIP}/render/${HIPNAME}.${AOV}.$F4.exr