Initial commit

sequencer_utils.py

@@ -0,0 +1,132 @@
+# SPDX-License-Identifier: GPL-2.0-or-later
+
+import bpy
+import re
+
+from pathlib import Path
+from vse_toolbox.bl_utils import get_settings
+
+def get_strips(channel=0, selected_only=False):
+    scn = bpy.context.scene
+    
+    if isinstance(channel, str):
+        channel = scn.sequence_editor.channels.keys().index(channel)
+
+    strips = [s for s in scn.sequence_editor.sequences_all if s.channel==channel]
+    
+    if selected_only:
+        strips = [s for s in strips if s.select]
+    
+    return sorted(strips, key=lambda x : x.frame_final_start)
+
+def get_shot_sequence(shot):
+    sequences = get_strips(channel='Sequences')
+    return next((s.name for s in sequences if s.frame_final_start<=shot.frame_final_start<s.frame_final_end), '')
+
+def rename_strips(
+        strips, template, increment=10, start_number=0, by_sequence=False):
+    scn = bpy.context.scene
+    settings = get_settings()
+
+    project = settings.active_project
+    episode = settings.active_episode
+
+    previous_sequence = None
+    strip_number = 0
+    
+    for strip in strips: 
+        sequence = get_shot_sequence(strip)
+
+        if (by_sequence and previous_sequence and
+                sequence and sequence != previous_sequence):
+            strip_number = 0
+
+        name = template.format(
+            episode=episode.name,
+            index=strip_number*increment+start_number
+        )
+
+        existing_strip = scn.sequence_editor.sequences_all.get(name)
+        if existing_strip: 
+            existing_strip.name = f"{name}_tmp"
+        
+        print(f'Renaming {strip.name} -> {name}')
+        strip.name = name
+        
+        previous_sequence = sequence    
+        strip_number += 1
+
+def set_channels():
+    scn = bpy.context.scene
+    settings = get_settings()
+    items = settings.rna_type.bl_rna.properties['channels'].enum_items 
+    
+    for i, c in enumerate(items.keys(), start=1):
+        scn.sequence_editor.channels[i].name = c.title()
+
+def render_strips(strips):
+    for strip in strips:
+        print(strip.name)
+
+def import_edl(filepath, adapter="cmx_3600"):
+    import opentimelineio as otio
+    scn = bpy.context.scene
+    
+    sequences = scn.sequence_editor.sequences
+
+    edl = Path(filepath)
+
+    try:
+        timeline = otio.adapters.read_from_file(
+            str(edl), adapter, rate=scn.render.fps, ignore_timecode_mismatch=True)
+    except:
+        data = edl.read_text(encoding='latin-1')
+        timeline = otio.adapters.read_from_string(
+            data, adapter, rate=scn.render.fps, ignore_timecode_mismatch=True)
+
+    scn.frame_start = (
+        0 if timeline.global_start_time is None else timeline.global_start_time
+    )
+    scn.frame_end = otio.opentime.to_frames(timeline.duration()) - 1
+    
+    for i, track in enumerate(timeline.tracks, 1):
+        for child in track.each_child(shallow_search=False):
+            frame_start = otio.opentime.to_frames(
+                child.range_in_parent().start_time)
+
+            frame_end = otio.opentime.to_frames(
+                child.range_in_parent().duration) - 1
+            try:
+                strip = sequences.new_effect(
+                    name='',
+                    type='COLOR',
+                    channel=i,
+                    frame_start=frame_start,
+                    frame_end=frame_end,
+                )
+            except Exception as e:
+                print('e: ', e)
+                continue
+
+    """
+    t = otio.adapters.read_from_file(edl, rate=C.scene.render.fps)
+    t.video_tracks() # Video Tracks du fichiers. C'est une liste.
+    Parcourir toutes les video_tracks ?
+    Trouver comment fait Felix David 
+    """
+    print('timeline: ', timeline)
+
+    return timeline
+
+"""
+def create_strip_effect(name, strip_type, channel, frame_start, frame_end):
+    strip = scn.sequence_editor.sequences.new_effect(
+        name=name,
+        type=strip_type,
+        channel=channel,
+        frame_start=frame_start,
+        frame_end=frame_end,
+    )
+
+    return strip
+"""