From ca0e6592db03dfeb8d0186a443683f7bbb530f92 Mon Sep 17 00:00:00 2001
From: Jakub Jezek <jakubjezek001@gmail.com>
Date: Wed, 27 Apr 2022 17:19:59 +0200
Subject: [PATCH] hiero: adding frame family extractor

---
 .../hiero/plugins/publish/extract_frames.py   | 82 +++++++++++++++++++
 1 file changed, 82 insertions(+)
 create mode 100644 openpype/hosts/hiero/plugins/publish/extract_frames.py

diff --git a/openpype/hosts/hiero/plugins/publish/extract_frames.py b/openpype/hosts/hiero/plugins/publish/extract_frames.py
new file mode 100644
index 0000000000..5396298aac
--- /dev/null
+++ b/openpype/hosts/hiero/plugins/publish/extract_frames.py
@@ -0,0 +1,82 @@
+import os
+import pyblish.api
+import openpype
+
+
+class ExtractFrames(openpype.api.Extractor):
+    """Extracts frames"""
+
+    order = pyblish.api.ExtractorOrder
+    label = "Extract Frames"
+    hosts = ["hiero"]
+    families = ["frame"]
+    movie_extensions = ["mov", "mp4"]
+
+    def process(self, instance):
+        oiio_tool_path = openpype.lib.get_oiio_tools_path()
+        staging_dir = self.staging_dir(instance)
+        output_template = os.path.join(staging_dir, instance.data["name"])
+        sequence = instance.context.data["activeTimeline"]
+
+        files = []
+        for frame in instance.data["frames"]:
+            track_item = sequence.trackItemAt(frame)
+            media_source = track_item.source().mediaSource()
+            input_path = media_source.fileinfos()[0].filename()
+            input_frame = (
+                track_item.mapTimelineToSource(frame) +
+                track_item.source().mediaSource().startTime()
+            )
+            output_ext = instance.data["format"]
+            output_path = output_template
+            output_path += ".{:04d}.{}".format(int(frame), output_ext)
+
+            args = [oiio_tool_path]
+
+            ext = os.path.splitext(input_path)[1][1:]
+            if ext in self.movie_extensions:
+                args.extend(["--subimage", str(int(input_frame))])
+            else:
+                args.extend(["--frames", str(int(input_frame))])
+
+            if ext == "exr":
+                args.extend(["--powc", "0.45,0.45,0.45,1.0"])
+
+            args.extend([input_path, "-o", output_path])
+            output = openpype.api.run_subprocess(args)
+
+            failed_output = "oiiotool produced no output."
+            if failed_output in output:
+                raise ValueError(
+                    "oiiotool processing failed. Args: {}".format(args)
+                )
+
+            files.append(output_path)
+
+            # Feedback to user because "oiiotool" can make the publishing
+            # appear unresponsive.
+            self.log.info(
+                "Processed {} of {} frames".format(
+                    instance.data["frames"].index(frame) + 1,
+                    len(instance.data["frames"])
+                )
+            )
+
+        if len(files) == 1:
+            instance.data["representations"] = [
+                {
+                    "name": output_ext,
+                    "ext": output_ext,
+                    "files": os.path.basename(files[0]),
+                    "stagingDir": staging_dir
+                }
+            ]
+        else:
+            instance.data["representations"] = [
+                {
+                    "name": output_ext,
+                    "ext": output_ext,
+                    "files": [os.path.basename(x) for x in files],
+                    "stagingDir": staging_dir
+                }
+            ]
\ No newline at end of file