diff options
Diffstat (limited to 'power_sequencer/operators/trim_left_or_right_handles.py')
-rw-r--r-- | power_sequencer/operators/trim_left_or_right_handles.py | 115 |
1 files changed, 115 insertions, 0 deletions
diff --git a/power_sequencer/operators/trim_left_or_right_handles.py b/power_sequencer/operators/trim_left_or_right_handles.py new file mode 100644 index 00000000..75a461da --- /dev/null +++ b/power_sequencer/operators/trim_left_or_right_handles.py @@ -0,0 +1,115 @@ +# +# 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/>. +# +from operator import attrgetter + +import bpy + +from .utils.doc import doc_brief, doc_description, doc_idname, doc_name + + +class POWER_SEQUENCER_OT_trim_left_or_right_handles(bpy.types.Operator): + """ + Trims or extends the handle closest to the time cursor for all selected strips. + + If you keep the Shift key down, the edit will ripple through the timeline. + Auto selects sequences under the time cursor when you don't have a selection. + """ + + doc = { + "name": doc_name(__qualname__), + "demo": "", + "description": doc_description(__doc__), + "shortcuts": [ + ( + {"type": "K", "value": "PRESS", "alt": True}, + {"side": "RIGHT", "ripple": False}, + "Smart Snap Right", + ), + ( + {"type": "K", "value": "PRESS", "alt": True, "shift": True}, + {"side": "RIGHT", "ripple": True}, + "Smart Snap Right With Ripple", + ), + ( + {"type": "K", "value": "PRESS", "ctrl": True}, + {"side": "LEFT", "ripple": False}, + "Smart Snap Left", + ), + ( + {"type": "K", "value": "PRESS", "ctrl": True, "shift": True}, + {"side": "LEFT", "ripple": True}, + "Smart Snap Left With Ripple", + ), + ], + "keymap": "Sequencer", + } + bl_idname = doc_idname(__qualname__) + bl_label = doc["name"] + bl_description = doc_brief(doc["description"]) + bl_options = {"REGISTER", "UNDO"} + + side: bpy.props.EnumProperty( + items=[("LEFT", "Left", "Left side"), ("RIGHT", "Right", "Right side")], + name="Snap side", + description="Handle side to use for the snap", + default="LEFT", + ) + ripple: bpy.props.BoolProperty(name="Ripple", default=False) + + @classmethod + def poll(cls, context): + return context.sequences + + def execute(self, context): + frame_current = context.scene.frame_current + + # Only select sequences under the time cursor + sequences = context.selected_sequences if context.selected_sequences else context.sequences + for s in sequences: + s.select = s.frame_final_start <= frame_current and s.frame_final_end >= frame_current + sequences = [s for s in sequences if s.select] + if not sequences: + return {"FINISHED"} + + for s in sequences: + if self.side == "LEFT": + s.select_left_handle = True + if self.side == "RIGHT": + s.select_right_handle = True + + # If trimming from the left, we need to save the start frame before trimming + ripple_start_frame = 0 + if self.ripple and self.side == "LEFT": + ripple_start_frame = min( + sequences, key=attrgetter("frame_final_start") + ).frame_final_start + + bpy.ops.sequencer.snap(frame=frame_current) + for s in sequences: + s.select_right_handle = False + s.select_left_handle = False + + if self.ripple and sequences: + if self.side == "RIGHT": + ripple_start_frame = max( + sequences, key=attrgetter("frame_final_end") + ).frame_final_end + bpy.ops.power_sequencer.gap_remove(frame=ripple_start_frame) + else: + bpy.ops.power_sequencer.gap_remove(frame=ripple_start_frame) + + return {"FINISHED"} |