Merge pull request #2606 from BigRoy/hou_arnold_load_ass

Houdini: Load Arnold .ass procedurals into Houdini

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

@@ -0,0 +1,103 @@
+import os
+
+from avalon import api
+from avalon.houdini import pipeline
+import clique
+
+
+class AssLoader(api.Loader):
+    """Load .ass with Arnold Procedural"""
+
+    families = ["ass"]
+    label = "Load Arnold Procedural"
+    representations = ["ass"]
+    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
+        procedural = obj.createNode("arnold::procedural", node_name=node_name)
+        procedural.setParms({"ar_filename": self.get_path(self.fname)})
+
+        nodes = [procedural]
+        self[:] = nodes
+
+        return pipeline.containerise(
+            node_name,
+            namespace,
+            nodes,
+            context,
+            self.__class__.__name__,
+            suffix="",
+        )
+
+    def get_path(self, path):
+
+        # Find all frames in the folder
+        ext = ".ass.gz" if path.endswith(".ass.gz") else ".ass"
+        folder = os.path.dirname(path)
+        frames = [f for f in os.listdir(folder) if f.endswith(ext)]
+
+        # Get the collection of frames to detect frame padding
+        patterns = [clique.PATTERNS["frames"]]
+        collections, remainder = clique.assemble(frames,
+                                                 minimum_items=1,
+                                                 patterns=patterns)
+        self.log.debug("Detected collections: {}".format(collections))
+        self.log.debug("Detected remainder: {}".format(remainder))
+
+        if not collections and remainder:
+            if len(remainder) != 1:
+                raise ValueError("Frames not correctly detected "
+                                 "in: {}".format(remainder))
+
+            # A single frame without frame range detected
+            filepath = remainder[0]
+            return os.path.normpath(filepath).replace("\\", "/")
+
+        # Frames detected with a valid "frame" number pattern
+        # Then we don't want to have any remainder files found
+        assert len(collections) == 1 and not remainder
+        collection = collections[0]
+
+        num_frames = len(collection.indexes)
+        if num_frames == 1:
+            # Return the input path without dynamic $F variable
+            result = path
+        else:
+            # More than a single frame detected - use $F{padding}
+            fname = "{}$F{}{}".format(collection.head,
+                                      collection.padding,
+                                      collection.tail)
+            result = os.path.join(folder, fname)
+
+        # Format file name, Houdini only wants forward slashes
+        return os.path.normpath(result).replace("\\", "/")
+
+    def update(self, container, representation):
+
+        # Update the file path
+        file_path = api.get_representation_path(representation)
+        file_path = file_path.replace("\\", "/")
+
+        procedural = container["node"]
+        procedural.setParms({"ar_filename": self.get_path(file_path)})
+
+        # Update attribute
+        procedural.setParms({"representation": str(representation["_id"])})
+
+    def remove(self, container):
+
+        node = container["node"]
+        node.destroy()