From 68e9102f7263e70c50c7f7a959ae48ba5c16280e Mon Sep 17 00:00:00 2001
From: Petr Kalis <petr.kalis@gmail.com>
Date: Thu, 17 Jun 2021 17:04:45 +0200
Subject: [PATCH] #636 - PS created loader to load reference frame from
 'render' sequence

Loader cannot do containerization for now as updates would be too difficult. (Customer doesn't mind.)
---
 .../plugins/load/load_image_from_sequence.py  | 90 +++++++++++++++++++
 1 file changed, 90 insertions(+)
 create mode 100644 openpype/hosts/photoshop/plugins/load/load_image_from_sequence.py

diff --git a/openpype/hosts/photoshop/plugins/load/load_image_from_sequence.py b/openpype/hosts/photoshop/plugins/load/load_image_from_sequence.py
new file mode 100644
index 00000000000..1409065d437
--- /dev/null
+++ b/openpype/hosts/photoshop/plugins/load/load_image_from_sequence.py
@@ -0,0 +1,90 @@
+import os
+from avalon import photoshop
+from avalon.vendor import qargparse
+
+from openpype.hosts.photoshop.plugins.load.load_image import ImageLoader
+
+stub = photoshop.stub()
+
+
+class ImageFromSequenceLoader(ImageLoader):
+    """ Load specifing image from sequence
+
+        Used only as quick load of reference file from a sequence.
+
+        Plain ImageLoader picks first frame from sequence.
+
+        Loads only existing files - currently not possible to limit loaders
+        to single select - multiselect. If user selects multiple repres, list
+        for all of them is provided, but selection is only single file.
+        This loader will be triggered multiple times, but selected name will
+        match only to proper path.
+
+        Loader doesnt do containerization as there is currently no data model
+        of 'frame of rendered files' (only rendered sequence), update would be
+        difficult.
+    """
+
+    families = ["render"]
+    representations = ["*"]
+    options = []
+
+    def load(self, context, name=None, namespace=None, data=None):
+        if data.get("frame"):
+            self.fname = os.path.join(os.path.dirname(self.fname),
+                                      data["frame"])
+            if not os.path.exists(self.fname):
+                return
+
+        layer_name = self._get_unique_layer_name(context["asset"]["name"],
+                                                 name)
+        with photoshop.maintained_selection():
+            layer = stub.import_smart_object(self.fname, layer_name)
+
+        self[:] = [layer]
+        namespace = namespace or layer_name
+
+        return namespace
+
+    @classmethod
+    def get_options(cls, repre_contexts):
+        """
+            Returns list of files for selected 'repre_contexts'.
+
+            It returns only files with same extension as in context as it is
+            expected that context points to sequence of frames.
+
+            Returns:
+                (list) of qargparse.Choice
+        """
+        files = []
+        for context in repre_contexts:
+            fname = ImageLoader.filepath_from_context(context)
+            _, file_extension = os.path.splitext(fname)
+
+            for file_name in os.listdir(os.path.dirname(fname)):
+                if not file_name.endswith(file_extension):
+                    continue
+                files.append(file_name)
+
+        # return selection only if there is something
+        if not files or len(files) <= 1:
+            return []
+
+        return [
+            qargparse.Choice(
+                "frame",
+                label="Select specific file",
+                items=files,
+                default=0,
+                help="Which frame should be loaded?"
+            )
+        ]
+
+    def update(self, container, representation):
+        """No update possible, not containerized."""
+        pass
+
+    def remove(self, container):
+        """No update possible, not containerized."""
+        pass