From 75fa36c8e9f38df65bb749dbfb2ac3f3cc2d5739 Mon Sep 17 00:00:00 2001
From: Jakub Jezek <jakub@orbi.tools>
Date: Wed, 23 Dec 2020 15:14:09 +0100
Subject: [PATCH] resolve: wip subset resources collection

---
 .../publish/collect_otio_subset_resources.py  | 115 ++++++++----------
 1 file changed, 52 insertions(+), 63 deletions(-)

diff --git a/pype/plugins/global/publish/collect_otio_subset_resources.py b/pype/plugins/global/publish/collect_otio_subset_resources.py
index f6bfe11941..019b9f902f 100644
--- a/pype/plugins/global/publish/collect_otio_subset_resources.py
+++ b/pype/plugins/global/publish/collect_otio_subset_resources.py
@@ -1,9 +1,10 @@
+# TODO: this head doc string
 """
 Requires:
+    instance -> otio_clip
+
+Optional:
     instance -> review
-    instance -> masterLayer
-    instance -> otioClip
-    context -> otioTimeline
 
 Provides:
     instance -> otioReviewClips
@@ -30,71 +31,59 @@ class CollectOcioSubsetResources(pyblish.api.InstancePlugin):
         # generate range in parent
         otio_src_range = otio_clip.source_range
         otio_avalable_range = otio_clip.available_range()
-        otio_visible_range = otio_clip.visible_range()
         otio_src_range_handles = pype.lib.otio_range_with_handles(
             otio_src_range, instance)
         trimmed_media_range = pype.lib.trim_media_range(
             otio_avalable_range, otio_src_range_handles)
 
-        self.log.debug(
-            "_ otio_avalable_range: {}".format(otio_avalable_range))
-        self.log.debug(
-            "_ otio_visible_range: {}".format(otio_visible_range))
-        self.log.debug(
-            "_ otio_src_range: {}".format(otio_src_range))
-        self.log.debug(
-            "_ otio_src_range_handles: {}".format(otio_src_range_handles))
         self.log.debug(
             "_ trimmed_media_range: {}".format(trimmed_media_range))
 
-        #
-        # media_ref = otio_clip.media_reference
-        # metadata = media_ref.metadata
-        #
-        # if isinstance(media_ref, otio.schema.ImageSequenceReference):
-        #     dirname = media_ref.target_url_base
-        #     head = media_ref.name_prefix
-        #     tail = media_ref.name_suffix
-        #     first, last = pype.lib.otio_range_to_frame_range(
-        #         available_range)
-        #     collection = clique.Collection(
-        #         head=head,
-        #         tail=tail,
-        #         padding=media_ref.frame_zero_padding
-        #     )
-        #     collection.indexes.update(
-        #         [i for i in range(first, (last + 1))])
-        #     # render segment
-        #     self._render_seqment(
-        #         sequence=[dirname, collection])
-        #     # generate used frames
-        #     self._generate_used_frames(
-        #         len(collection.indexes))
-        # elif metadata.get("padding"):
-        #     # in case it is file sequence but not new OTIO schema
-        #     # `ImageSequenceReference`
-        #     path = media_ref.target_url
-        #     dir_path, collection = self._make_sequence_collection(
-        #         path, available_range, metadata)
-        #
-        #     # render segment
-        #     self._render_seqment(
-        #         sequence=[dir_path, collection])
-        #     # generate used frames
-        #     self._generate_used_frames(
-        #         len(collection.indexes))
-        # else:
-        #     path = media_ref.target_url
-        #     # render video file to sequence
-        #     self._render_seqment(
-        #         video=[path, available_range])
-        #     # generate used frames
-        #     self._generate_used_frames(
-        #         available_range.duration.value)
-        #
-        # instance.data["otioReviewClips"] = otio_review_clips
-        # self.log.debug(
-        #     "_ otio_review_clips: {}".format(otio_review_clips))
-        #
-        # self.log.debug(
-        #     "_ instance.data: {}".format(pformat(instance.data)))
+        media_ref = otio_clip.media_reference
+        metadata = media_ref.metadata
+
+        # check in two way if it is sequence
+        if hasattr(otio.schema, "ImageSequenceReference"):
+            # for OpenTimelineIO 0.13 and newer
+            if isinstance(media_ref,
+                          otio.schema.ImageSequenceReference):
+                is_sequence = True
+        else:
+            # for OpenTimelineIO 0.12 and older
+            if metadata.get("padding"):
+                is_sequence = True
+
+        first, last = pype.lib.otio_range_to_frame_range(
+            trimmed_media_range)
+
+        self.log.info(
+            "first-last: {}-{}".format(first, last))
+
+        if is_sequence:
+            # file sequence way
+            if hasattr(media_ref, "target_url_base"):
+                dirname = media_ref.target_url_base
+                head = media_ref.name_prefix
+                tail = media_ref.name_suffix
+                collection = clique.Collection(
+                    head=head,
+                    tail=tail,
+                    padding=media_ref.frame_zero_padding
+                )
+                collection.indexes.update(
+                    [i for i in range(first, (last + 1))])
+                # TODO: add representation
+                self.log.debug((dirname, collection))
+            else:
+                # in case it is file sequence but not new OTIO schema
+                # `ImageSequenceReference`
+                path = media_ref.target_url
+                dir_path, collection = pype.lib.make_sequence_collection(
+                    path, trimmed_media_range, metadata)
+
+                # TODO: add representation
+                self.log.debug((dir_path, collection))
+        else:
+            path = media_ref.target_url
+            # TODO: add representation
+            self.log.debug(path)