Merge pull request #324 from tokejepsen/2.x/feature/support_palettes

Support palettes.

pype/hosts/harmony/__init__.py

@@ -121,7 +121,12 @@ def check_inventory():
     }
     func
     """
-    outdated_nodes = [x["node"] for x in outdated_containers]
+    outdated_nodes = []
+    for container in outdated_containers:
+        if container["loader"] == "ImageSequenceLoader":
+            outdated_nodes.append(
+                harmony.find_node_by_name(container["name"], "READ")
+            )
     harmony.send({"function": func, "args": [outdated_nodes]})
 
     # Warn about outdated containers.

pype/plugins/global/publish/integrate_new.py

@@ -83,6 +83,7 @@ class IntegrateAssetNew(pyblish.api.InstancePlugin):
                 "textures",
                 "action",
                 "harmony.template",
+                "harmony.palette",
                 "editorial"
                 ]
     exclude_families = ["clip"]
@@ -605,7 +606,7 @@ class IntegrateAssetNew(pyblish.api.InstancePlugin):
                 "type": "subset",
                 "name": subset_name,
                 "data": {
-                    "families": instance.data.get('families')
+                    "families": instance.data.get("families", [])
                 },
                 "parent": asset["_id"]
             }).inserted_id

pype/plugins/harmony/load/load_imagesequence.py

@@ -1,8 +1,10 @@
 import os
+import uuid
 
 import clique
 
 from avalon import api, harmony
+import pype.lib
 
 copy_files = """function copyFile(srcFilename, dstFilename)
 {
@@ -251,15 +253,15 @@ class ImageSequenceLoader(api.Loader):
                 ).replace("\\", "/")
             )
 
+        name = context["subset"]["name"]
+        name += "_{}".format(uuid.uuid4())
         read_node = harmony.send(
             {
                 "function": copy_files + import_files,
-                "args": ["Top", files, context["subset"]["name"], 1]
+                "args": ["Top", files, name, 1]
             }
         )["result"]
 
-        self[:] = [read_node]
-
         return harmony.containerise(
             name,
             namespace,
@@ -269,7 +271,7 @@ class ImageSequenceLoader(api.Loader):
         )
 
     def update(self, container, representation):
-        node = container.pop("node")
+        node = harmony.find_node_by_name(container["name"], "READ")
 
         path = api.get_representation_path(representation)
         collections, remainder = clique.assemble(
@@ -297,12 +299,34 @@ class ImageSequenceLoader(api.Loader):
             }
         )
 
+        # Colour node.
+        func = """function func(args){
+            for( var i =0; i <= args[0].length - 1; ++i)
+            {
+                var red_color = new ColorRGBA(255, 0, 0, 255);
+                var green_color = new ColorRGBA(0, 255, 0, 255);
+                if (args[1] == "red"){
+                    node.setColor(args[0], red_color);
+                }
+                if (args[1] == "green"){
+                    node.setColor(args[0], green_color);
+                }
+            }
+        }
+        func
+        """
+        if pype.lib.is_latest(representation):
+            harmony.send({"function": func, "args": [node, "green"]})
+        else:
+            harmony.send({"function": func, "args": [node, "red"]})
+
         harmony.imprint(
             node, {"representation": str(representation["_id"])}
         )
 
     def remove(self, container):
-        node = container.pop("node")
+        node = harmony.find_node_by_name(container["name"], "READ")
+
         func = """function deleteNode(_node)
         {
             node.deleteNode(_node, true, true);

pype/plugins/harmony/load/load_palette.py

@@ -0,0 +1,66 @@
+import os
+import shutil
+
+from avalon import api, harmony
+from avalon.vendor import Qt
+
+
+class ImportPaletteLoader(api.Loader):
+    """Import palettes."""
+
+    families = ["harmony.palette"]
+    representations = ["plt"]
+    label = "Import Palette"
+
+    def load(self, context, name=None, namespace=None, data=None):
+        name = self.load_palette(context["representation"])
+
+        return harmony.containerise(
+            name,
+            namespace,
+            name,
+            context,
+            self.__class__.__name__
+        )
+
+    def load_palette(self, representation):
+        subset_name = representation["context"]["subset"]
+        name = subset_name.replace("palette", "")
+
+        # Overwrite palette on disk.
+        scene_path = harmony.send(
+            {"function": "scene.currentProjectPath"}
+        )["result"]
+        src = api.get_representation_path(representation)
+        dst = os.path.join(
+            scene_path,
+            "palette-library",
+            "{}.plt".format(name)
+        )
+        shutil.copy(src, dst)
+
+        harmony.save_scene()
+
+        # Dont allow instances with the same name.
+        message_box = Qt.QtWidgets.QMessageBox()
+        message_box.setIcon(Qt.QtWidgets.QMessageBox.Warning)
+        msg = "Updated {}.".format(subset_name)
+        msg += " You need to reload the scene to see the changes."
+        message_box.setText(msg)
+        message_box.exec_()
+
+        return name
+
+    def remove(self, container):
+        harmony.remove(container["name"])
+
+    def switch(self, container, representation):
+        self.update(container, representation)
+
+    def update(self, container, representation):
+        self.remove(container)
+        name = self.load_palette(representation)
+
+        container["representation"] = str(representation["_id"])
+        container["name"] = name
+        harmony.imprint(name, container)

pype/plugins/harmony/publish/collect_palettes.py

@@ -0,0 +1,45 @@
+import os
+import json
+
+import pyblish.api
+from avalon import harmony
+
+
+class CollectPalettes(pyblish.api.ContextPlugin):
+    """Gather palettes from scene when publishing templates."""
+
+    label = "Palettes"
+    order = pyblish.api.CollectorOrder
+    hosts = ["harmony"]
+
+    def process(self, context):
+        func = """function func()
+        {
+            var palette_list = PaletteObjectManager.getScenePaletteList();
+
+            var palettes = {};
+            for(var i=0; i < palette_list.numPalettes; ++i)
+            {
+                var palette = palette_list.getPaletteByIndex(i);
+                palettes[palette.getName()] = palette.id;
+            }
+
+            return palettes;
+        }
+        func
+        """
+        palettes = harmony.send({"function": func})["result"]
+
+        for name, id in palettes.items():
+            instance = context.create_instance(name)
+            instance.data.update({
+                "id": id,
+                "family": "harmony.palette",
+                "asset": os.environ["AVALON_ASSET"],
+                "subset": "palette" + name
+            })
+            self.log.info(
+                "Created instance:\n" + json.dumps(
+                    instance.data, sort_keys=True, indent=4
+                )
+            )

pype/plugins/harmony/publish/extract_palette.py

@@ -0,0 +1,34 @@
+import os
+
+from avalon import harmony
+import pype.api
+import pype.hosts.harmony
+
+
+class ExtractPalette(pype.api.Extractor):
+    """Extract palette."""
+
+    label = "Extract Palette"
+    hosts = ["harmony"]
+    families = ["harmony.palette"]
+
+    def process(self, instance):
+        func = """function func(args)
+        {
+            var palette_list = PaletteObjectManager.getScenePaletteList();
+            var palette = palette_list.getPaletteById(args[0]);
+            return (palette.getPath() + "/" + palette.getName() + ".plt");
+        }
+        func
+        """
+        palette_file = harmony.send(
+            {"function": func, "args": [instance.data["id"]]}
+        )["result"]
+
+        representation = {
+            "name": "plt",
+            "ext": "plt",
+            "files": os.path.basename(palette_file),
+            "stagingDir": os.path.dirname(palette_file)
+        }
+        instance.data["representations"] = [representation]