2025-12-01

scripts/addons/contactsheet/opsdata.py

@@ -0,0 +1,95 @@
+# SPDX-FileCopyrightText: 2021 Blender Studio Tools Authors
+#
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+from typing import Optional, List, Tuple
+
+import bpy
+
+from . import checksqe
+from .log import LoggerFactory
+
+logger = LoggerFactory.getLogger(name=__name__)
+
+
+def get_valid_cs_sequences(
+    sequence_list: List["bpy.types.Strip"],
+) -> List["bpy.types.Strip"]:
+    """
+    Returns list of valid sequences out of input sequence list
+    """
+    valid_sequences = [
+        s for s in sequence_list if s.type in ["MOVIE", "IMAGE"] and not s.mute
+    ]
+    return valid_sequences
+
+
+def get_sqe_editor(context: bpy.types.Context) -> Optional[bpy.types.Area]:
+    sqe_editor = None
+
+    for area in context.screen.areas:
+        if area.type == "SEQUENCE_EDITOR":
+            sqe_editor = area
+
+    return sqe_editor
+
+
+def fit_frame_range_to_strips(
+    context: bpy.types.Context, strips: Optional[List["bpy.types.Strip"]] = None
+) -> Tuple[int, int]:
+    def get_sort_tuple(strip: "bpy.types.Strip") -> Tuple[int, int]:
+        return (strip.frame_final_start, strip.frame_final_duration)
+
+    if not strips:
+        strips = context.scene.sequence_editor.sequences_all
+
+    if not strips:
+        return (0, 0)
+
+    strips = list(strips)
+    strips.sort(key=get_sort_tuple)
+
+    context.scene.frame_start = strips[0].frame_final_start
+    context.scene.frame_end = strips[-1].frame_final_end
+
+    return (context.scene.frame_start, context.scene.frame_end)
+
+
+def get_top_level_valid_strips_continuous(
+    context: bpy.types.Context,
+) -> List["bpy.types.Strip"]:
+
+    sequences_tmp = get_valid_cs_sequences(
+        list(context.scene.sequence_editor.sequences_all)
+    )
+
+    sequences_tmp.sort(key=lambda s: (s.channel, s.frame_final_start), reverse=True)
+    sequences: List["bpy.types.Strip"] = []
+
+    for strip in sequences_tmp:
+
+        occ_ranges = checksqe.get_occupied_ranges_for_strips(sequences)
+        s_range = range(strip.frame_final_start, strip.frame_final_end + 1)
+
+        if not checksqe.is_range_occupied(s_range, occ_ranges):
+            sequences.append(strip)
+
+    return sequences
+
+
+def poll_make_contactsheet(context: bpy.types.Context) -> bool:
+
+    if not context.scene.sequence_editor.sequences_all:
+        return False
+
+    sequences = context.selected_sequences
+
+    if not sequences:
+        valid_sequences = get_top_level_valid_strips_continuous(context)
+    else:
+        valid_sequences = get_valid_cs_sequences(sequences)
+
+    return bool(valid_sequences)
+
+
+# TODO: add function to actually get sequences, same structure in 3 places.