Flame: Loading clip with native colorspace resolved from mapping

openpype/hosts/flame/api/plugin.py

@@ -4,13 +4,13 @@
 from copy import deepcopy
 from xml.etree import ElementTree as ET
 
+import qargparse
 from Qt import QtCore, QtWidgets
 
-import qargparse
 from openpype import style
-from openpype.settings import get_current_project_settings
 from openpype.lib import Logger
 from openpype.pipeline import LegacyCreator, LoaderPlugin
+from openpype.settings import get_current_project_settings
 
 from . import constants
 from . import lib as flib
@@ -690,6 +690,54 @@ class ClipLoader(LoaderPlugin):
  )
  ]
 
+ _mapping = None
+
+ def get_colorspace(self, context):
+ """Get colorspace name
+
+ Look either to version data or representation data.
+
+ Args:
+ context (dict): version context data
+
+ Returns:
+ str: colorspace name or None
+ """
+ version = context['version']
+ version_data = version.get("data", {})
+ colorspace = version_data.get(
+ "colorspace", None
+ )
+
+ if (
+ not colorspace
+ or colorspace == "Unknown"
+ ):
+ colorspace = context["representation"]["data"].get(
+ "colorspace", None)
+
+ return colorspace
+
+ @classmethod
+ def get_native_colorspace(cls, input_colorspace):
+ """Return native colorspace name.
+
+ Args:
+ input_colorspace (str | None): colorspace name
+
+ Returns:
+ str: native colorspace name defined in mapping or None
+ """
+ if not cls._mapping:
+ settings = get_current_project_settings()["flame"]
+ mapping = settings["imageio"]["profilesMapping"]["inputs"]
+ cls._mapping = {
+ input["ocioName"]: input["flameName"]
+ for input in mapping
+ }
+
+ return cls._mapping.get(input_colorspace)
+
 
 class OpenClipSolver(flib.MediaInfoFile):
  create_new_clip = False

openpype/hosts/flame/plugins/load/load_clip.py

@@ -36,14 +36,15 @@ def load(self, context, name, namespace, options):
  version = context['version']
  version_data = version.get("data", {})
  version_name = version.get("name", None)
- colorspace = version_data.get("colorspace", None)
+ colorspace = self.get_colorspace(context)
+
  clip_name = StringTemplate(self.clip_name_template).format(
  context["representation"]["context"])
 
- # TODO: settings in imageio
  # convert colorspace with ocio to flame mapping
  # in imageio flame section
- colorspace = colorspace
+ colorspace = self.get_native_colorspace(colorspace)
+ self.log.info("Loading with colorspace: `{}`".format(colorspace))
 
  # create workfile path
  workfile_dir = os.environ["AVALON_WORKDIR"]

openpype/hosts/flame/plugins/load/load_clip_batch.py

@@ -35,7 +35,7 @@ def load(self, context, name, namespace, options):
  version = context['version']
  version_data = version.get("data", {})
  version_name = version.get("name", None)
- colorspace = version_data.get("colorspace", None)
+ colorspace = self.get_colorspace(context)
 
  # in case output is not in context replace key to representation
  if not context["representation"]["context"].get("output"):
@@ -47,10 +47,10 @@ def load(self, context, name, namespace, options):
  clip_name = StringTemplate(self.clip_name_template).format(
  formating_data)
 
- # TODO: settings in imageio
  # convert colorspace with ocio to flame mapping
  # in imageio flame section
- colorspace = colorspace
+ colorspace = self.get_native_colorspace(colorspace)
+ self.log.info("Loading with colorspace: `{}`".format(colorspace))
 
  # create workfile path
  workfile_dir = options.get("workdir") or os.environ["AVALON_WORKDIR"]