Merge pull request #15 from ynput/enhancement/tray_publish_set_explicit_colorspace

TrayPublisher: colorspace overrides from defined ocio config

openpype/hosts/traypublisher/plugins/publish/collect_explicit_colorspace.py

@@ -1,7 +1,8 @@
 import pyblish.api
-
+from openpype.pipeline import registered_host
 from openpype.pipeline import publish
-from openpype.lib import TextDef
+from openpype.lib import EnumDef
+from openpype.pipeline import colorspace
 
 
 class CollectColorspace(pyblish.api.InstancePlugin,
@@ -13,12 +14,19 @@ class CollectColorspace(pyblish.api.InstancePlugin,
     order = pyblish.api.CollectorOrder + 0.49
     hosts = ["traypublisher"]
 
+    colorspace_items = [
+        (None, "Don't override")
+    ]
+    colorspace_attr_show = False
+
     def process(self, instance):
         values = self.get_attr_values_from_data(instance.data)
         colorspace = values.get("colorspace", None)
-        if not colorspace:
+        if colorspace is None:
             return
 
+        self.log.debug("Explicit colorspace set to: {}".format(colorspace))
+
         context = instance.context
         for repre in instance.data.get("representations", {}):
             self.set_representation_colorspace(
@@ -27,10 +35,32 @@ class CollectColorspace(pyblish.api.InstancePlugin,
                 colorspace=colorspace
             )
 
+    @classmethod
+    def apply_settings(cls, project_settings):
+        host = registered_host()
+        host_name = host.name
+        project_name = host.get_current_project_name()
+        config_data = colorspace.get_imageio_config(
+            project_name, host_name,
+            project_settings=project_settings
+        )
+
+        if config_data:
+            filepath = config_data["path"]
+            config_items = colorspace.get_ocio_config_colorspaces(filepath)
+            cls.colorspace_items.extend((
+                (name, name) for name in config_items.keys()
+            ))
+            cls.colorspace_attr_show = True
+
     @classmethod
     def get_attribute_defs(cls):
         return [
-            TextDef("colorspace",
-                    label="Override Colorspace",
-                    placeholder="")
+            EnumDef(
+                "colorspace",
+                cls.colorspace_items,
+                default="Don't override",
+                label="Override Colorspace",
+                hidden=not cls.colorspace_attr_show
+            )
         ]