Alembic Loader as Arnold Standin

openpype/hosts/maya/plugins/load/load_abc_to_standin.py

@@ -0,0 +1,132 @@
+import os
+
+from openpype.pipeline import (
+    legacy_io,
+    load,
+    get_representation_path
+)
+from openpype.settings import get_project_settings
+
+
+class AlembicStandinLoader(load.LoaderPlugin):
+    """Load Alembic as Arnold Standin"""
+
+    families = ["animation", "model", "pointcache"]
+    representations = ["abc"]
+
+    label = "Import Alembic as Arnold Standin"
+    order = -5
+    icon = "code-fork"
+    color = "orange"
+
+    def load(self, context, name, namespace, options):
+
+        import maya.cmds as cmds
+        import mtoa.ui.arnoldmenu
+        from openpype.hosts.maya.api.pipeline import containerise
+        from openpype.hosts.maya.api.lib import unique_namespace
+
+        version = context["version"]
+        version_data = version.get("data", {})
+        family = version["data"]["families"]
+        self.log.info("version_data: {}\n".format(version_data))
+        self.log.info("family: {}\n".format(family))
+        frameStart = version_data.get("frameStart", None)
+
+        asset = context["asset"]["name"]
+        namespace = namespace or unique_namespace(
+            asset + "_",
+            prefix="_" if asset[0].isdigit() else "",
+            suffix="_",
+        )
+
+        # Root group
+        label = "{}:{}".format(namespace, name)
+        root = cmds.group(name=label, empty=True)
+
+        settings = get_project_settings(os.environ['AVALON_PROJECT'])
+        colors = settings["maya"]["load"]["colors"]
+        fps = legacy_io.Session["AVALON_FPS"]
+        c = colors.get(family[0])
+        if c is not None:
+            r = (float(c[0]) / 255)
+            g = (float(c[1]) / 255)
+            b = (float(c[2]) / 255)
+            cmds.setAttr(root + ".useOutlinerColor", 1)
+            cmds.setAttr(root + ".outlinerColor",
+                         r, g, b)
+
+        transform_name = label + "_ABC"
+
+        standinShape = cmds.ls(mtoa.ui.arnoldmenu.createStandIn())[0]
+        standin = cmds.listRelatives(standinShape, parent=True,
+                                     typ="transform")
+        standin = cmds.rename(standin, transform_name)
+        standinShape = cmds.listRelatives(standin, children=True)[0]
+
+        cmds.parent(standin, root)
+
+        # Set the standin filepath
+        cmds.setAttr(standinShape + ".dso", self.fname, type="string")
+        cmds.setAttr(standinShape + ".abcFPS", float(fps))
+
+        if frameStart is None:
+            cmds.setAttr(standinShape + ".useFrameExtension", 0)
+
+        elif "model" in family:
+            cmds.setAttr(standinShape + ".useFrameExtension", 0)
+
+        else:
+            cmds.setAttr(standinShape + ".useFrameExtension", 1)
+
+        nodes = [root, standin]
+        self[:] = nodes
+
+        return containerise(
+            name=name,
+            namespace=namespace,
+            nodes=nodes,
+            context=context,
+            loader=self.__class__.__name__)
+
+    def update(self, container, representation):
+
+        import pymel.core as pm
+
+        path = get_representation_path(representation)
+        fps = legacy_io.Session["AVALON_FPS"]
+        # Update the standin
+        standins = list()
+        members = pm.sets(container['objectName'], query=True)
+        self.log.info("container:{}".format(container))
+        for member in members:
+            shape = member.getShape()
+            if (shape and shape.type() == "aiStandIn"):
+                standins.append(shape)
+
+        for standin in standins:
+            standin.dso.set(path)
+            standin.abcFPS.set(float(fps))
+            if "modelMain" in container['objectName']:
+                standin.useFrameExtension.set(0)
+            else:
+                standin.useFrameExtension.set(1)
+
+        container = pm.PyNode(container["objectName"])
+        container.representation.set(str(representation["_id"]))
+
+    def switch(self, container, representation):
+        self.update(container, representation)
+
+    def remove(self, container):
+        import maya.cmds as cmds
+        members = cmds.sets(container['objectName'], query=True)
+        cmds.lockNode(members, lock=False)
+        cmds.delete([container['objectName']] + members)
+
+        # Clean up the namespace
+        try:
+            cmds.namespace(removeNamespace=container['namespace'],
+                           deleteNamespaceContent=True)
+        except RuntimeError:
+            pass