diff options
author | Nathan Lovato <nathan@gdquest.com> | 2019-09-05 18:22:37 +0300 |
---|---|---|
committer | Nathan Lovato <nathan@gdquest.com> | 2019-09-05 18:22:52 +0300 |
commit | 61d48c0a4be0ab8f71e6e1d35f0aa99c77fcfd33 (patch) | |
tree | f64d0ea88789184f45112b489598990d549788f2 /power_sequencer/operators/gap_remove.py | |
parent | da5a1175e30c347fbce05e49e2f5f895be30bd5b (diff) |
Add the VSE addon Power Sequencer
Diffstat (limited to 'power_sequencer/operators/gap_remove.py')
-rw-r--r-- | power_sequencer/operators/gap_remove.py | 135 |
1 files changed, 135 insertions, 0 deletions
diff --git a/power_sequencer/operators/gap_remove.py b/power_sequencer/operators/gap_remove.py new file mode 100644 index 00000000..8ef00b6b --- /dev/null +++ b/power_sequencer/operators/gap_remove.py @@ -0,0 +1,135 @@ +# +# Copyright (C) 2016-2019 by Nathan Lovato, Daniel Oakey, Razvan Radulescu, and contributors +# +# This file is part of Power Sequencer. +# +# Power Sequencer is free software: you can redistribute it and/or modify it under the terms of the +# GNU General Public License as published by the Free Software Foundation, either version 3 of the +# License, or (at your option) any later version. +# +# Power Sequencer is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; +# without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License along with Power Sequencer. If +# not, see <https://www.gnu.org/licenses/>. +# +import bpy +from operator import attrgetter + +from .utils.functions import slice_selection, sequencer_workaround_2_80_audio_bug +from .utils.doc import doc_name, doc_idname, doc_brief, doc_description + + +class POWER_SEQUENCER_OT_gap_remove(bpy.types.Operator): + """ + Remove gaps, starting from the first frame, with the ability to ignore locked strips + """ + + doc = { + "name": doc_name(__qualname__), + "demo": "", + "description": doc_description(__doc__), + "shortcuts": [], + "keymap": "Sequencer", + } + bl_idname = doc_idname(__qualname__) + bl_label = doc["name"] + bl_description = doc_brief(doc["description"]) + bl_options = {"REGISTER", "UNDO"} + + ignore_locked: bpy.props.BoolProperty( + name="Ignore Locked Strips", + description="Remove gaps without moving locked strips", + default=True, + ) + all: bpy.props.BoolProperty( + name="Remove All", + description="Remove all gaps starting from the time cursor", + default=False, + ) + frame: bpy.props.IntProperty( + name="Frame", + description="Frame to remove gaps from, defaults at the time cursor", + default=-1, + ) + + @classmethod + def poll(cls, context): + return context.sequences + + def execute(self, context): + frame = self.frame if self.frame >= 0 else context.scene.frame_current + sequences = ( + [s for s in context.sequences if not s.lock] + if self.ignore_locked + else context.sequences + ) + sequences = [ + s for s in sequences if s.frame_final_start >= frame or s.frame_final_end > frame + ] + sequence_blocks = slice_selection(context, sequences) + if not sequence_blocks: + return {"FINISHED"} + + gap_frame = self.find_gap_frame(context, frame, sequence_blocks[0]) + if gap_frame == -1: + return {"FINISHED"} + + first_block_start = min( + sequence_blocks[0], key=attrgetter("frame_final_start") + ).frame_final_start + blocks_after_gap = ( + sequence_blocks[1:] if first_block_start <= gap_frame else sequence_blocks + ) + + self.gaps_remove(context, blocks_after_gap, gap_frame) + sequencer_workaround_2_80_audio_bug(context) + return {"FINISHED"} + + def find_gap_frame(self, context, frame, sorted_sequences): + """ + Takes a list sequences sorted by frame_final_start + """ + strips_start = min(sorted_sequences, key=attrgetter("frame_final_start")).frame_final_start + strips_end = max(sorted_sequences, key=attrgetter("frame_final_end")).frame_final_end + + gap_frame = -1 + if strips_start > frame: + strips_before_frame_start = [s for s in context.sequences if s.frame_final_end <= frame] + frame_target = 0 + if strips_before_frame_start: + frame_target = max( + strips_before_frame_start, key=attrgetter("frame_final_end") + ).frame_final_end + gap_frame = frame_target if frame_target < strips_start else frame + else: + gap_frame = strips_end + return gap_frame + + def gaps_remove(self, context, sequence_blocks, gap_frame_start): + """ + Recursively removes gaps between blocks of sequences + """ + + gap_frame = gap_frame_start + for block in sequence_blocks: + gap_size = block[0].frame_final_start - gap_frame + if gap_size < 1: + continue + + for s in block: + try: + s.frame_start -= gap_size + except AttributeError: + continue + + self.move_markers(context, gap_frame, gap_size) + if not self.all: + break + gap_frame = block[-1].frame_final_end + + def move_markers(self, context, gap_frame, gap_size): + markers = (m for m in context.scene.timeline_markers if m.frame > gap_frame) + for m in markers: + m.frame -= min({gap_size, m.frame - gap_frame}) |