add loader

openpype/hosts/houdini/plugins/load/load_fbx.py

@@ -0,0 +1,118 @@
+# -*- coding: utf-8 -*-
+"""Fbx Loader for houdini.
+
+It's an exact copy of
+'load_bgeo.py'
+however this one includes extra comments for demonstration.
+
+This plugin is part of publish process guide.
+"""
+import os
+import re
+
+from openpype.pipeline import (
+    load,
+    get_representation_path,
+)
+from openpype.hosts.houdini.api import pipeline
+
+
+class FbxLoader(load.LoaderPlugin):
+    """Load fbx files to Houdini."""
+
+    label = "Load FBX"
+    families = ["filmboxfbx", "fbx"]
+    representations = ["fbx"]
+    order = -10
+    icon = "code-fork"
+    color = "orange"
+
+    def load(self, context, name=None, namespace=None, data=None):
+
+        import hou
+
+        # Get the root node
+        obj = hou.node("/obj")
+
+        # Define node name
+        namespace = namespace if namespace else context["asset"]["name"]
+        node_name = "{}_{}".format(namespace, name) if namespace else name
+
+        # Create a new geo node
+        container = obj.createNode("geo", node_name=node_name)
+        is_sequence = bool(context["representation"]["context"].get("frame"))
+
+        # Remove the file node, it only loads static meshes
+        # Houdini 17 has removed the file node from the geo node
+        file_node = container.node("file1")
+        if file_node:
+            file_node.destroy()
+
+        # Explicitly create a file node
+        path = self.filepath_from_context(context)
+        file_node = container.createNode("file", node_name=node_name)
+        file_node.setParms(
+            {"file": self.format_path(path, context["representation"])})
+
+        # Set display on last node
+        file_node.setDisplayFlag(True)
+
+        nodes = [container, file_node]
+        self[:] = nodes
+
+        return pipeline.containerise(
+            node_name,
+            namespace,
+            nodes,
+            context,
+            self.__class__.__name__,
+            suffix="",
+        )
+
+    @staticmethod
+    def format_path(path, representation):
+        """Format file path correctly for single bgeo or bgeo sequence."""
+        if not os.path.exists(path):
+            raise RuntimeError("Path does not exist: %s" % path)
+
+        is_sequence = bool(representation["context"].get("frame"))
+        # The path is either a single file or sequence in a folder.
+        if not is_sequence:
+            filename = path
+        else:
+            filename = re.sub(r"(.*)\.(\d+)\.(bgeo.*)", "\\1.$F4.\\3", path)
+
+            filename = os.path.join(path, filename)
+
+        filename = os.path.normpath(filename)
+        filename = filename.replace("\\", "/")
+
+        return filename
+
+    def update(self, container, representation):
+
+        node = container["node"]
+        try:
+            file_node = next(
+                n for n in node.children() if n.type().name() == "file"
+            )
+        except StopIteration:
+            self.log.error("Could not find node of type `file`")
+            return
+
+        # Update the file path
+        file_path = get_representation_path(representation)
+        file_path = self.format_path(file_path, representation)
+
+        file_node.setParms({"file": file_path})
+
+        # Update attribute
+        node.setParms({"representation": str(representation["_id"])})
+
+    def remove(self, container):
+
+        node = container["node"]
+        node.destroy()
+
+    def switch(self, container, representation):
+        self.update(container, representation)