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/mouse_trim_instantly.py | |
parent | da5a1175e30c347fbce05e49e2f5f895be30bd5b (diff) |
Add the VSE addon Power Sequencer
Diffstat (limited to 'power_sequencer/operators/mouse_trim_instantly.py')
-rw-r--r-- | power_sequencer/operators/mouse_trim_instantly.py | 113 |
1 files changed, 113 insertions, 0 deletions
diff --git a/power_sequencer/operators/mouse_trim_instantly.py b/power_sequencer/operators/mouse_trim_instantly.py new file mode 100644 index 00000000..dd2bb1fd --- /dev/null +++ b/power_sequencer/operators/mouse_trim_instantly.py @@ -0,0 +1,113 @@ +# +# 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 math import floor + +from .utils.functions import find_strips_mouse +from .utils.functions import trim_strips +from .utils.functions import get_frame_range +from .utils.doc import doc_name, doc_idname, doc_brief, doc_description +from .utils.functions import sequencer_workaround_2_80_audio_bug + + +class POWER_SEQUENCER_OT_mouse_trim_instantly(bpy.types.Operator): + """ + *brief* Trim strip from a start to an end frame instantly + + + Trims a frame range or a selection from a start to an end frame. + If there's no precise time range, auto trims based on the closest cut + + Args: + - frame_start and frame_end (int) define the frame range to trim + """ + + doc = { + "name": doc_name(__qualname__), + "demo": "", + "description": doc_description(__doc__), + "shortcuts": [ + ( + {"type": "RIGHTMOUSE", "value": "PRESS", "ctrl": True, "alt": True}, + {"select_mode": "CONTEXT"}, + "Trim strip, keep gap", + ), + ( + {"type": "RIGHTMOUSE", "value": "PRESS", "ctrl": True, "alt": True, "shift": True}, + {"select_mode": "CURSOR"}, + "Trim strip, remove gap", + ), + ], + "keymap": "Sequencer", + } + bl_idname = doc_idname(__qualname__) + bl_label = doc["name"] + bl_description = doc_brief(doc["description"]) + bl_options = {"REGISTER", "UNDO"} + + select_mode: bpy.props.EnumProperty( + items=[ + ("CONTEXT", "Smart", "Uses the selection if possible, else uses the other modes"), + ("CURSOR", "Time cursor", "Select all of the strips the time cursor overlaps"), + ], + name="Selection mode", + description="Auto-select the strip you click on or that the time cursor overlaps", + default="CONTEXT", + ) + select_linked: bpy.props.BoolProperty( + name="Use linked time", + description="If auto-select, cut linked strips if checked", + default=False, + ) + gap_remove: bpy.props.BoolProperty( + name="Remove gaps", + description="When trimming the sequences, remove gaps automatically", + default=True, + ) + + to_select = [] + + @classmethod + def poll(cls, context): + return context.sequences + + def invoke(self, context, event): + to_select = [] + frame, channel = -1, -1 + x, y = context.region.view2d.region_to_view( + x=event.mouse_region_x, y=event.mouse_region_y + ) + frame, channel = round(x), floor(y) + + mouse_clicked_strip = find_strips_mouse(context, frame, channel, self.select_linked) + to_select.extend(mouse_clicked_strip) + if self.select_mode == "CURSOR": + to_select.extend([s for s in context.sequences if s.frame_final_start <= frame <= s.frame_final_end and not s.lock]) + + frame_cut_closest = min(get_frame_range(context, to_select), key=lambda f: abs(frame - f)) + frame_start = min(frame, frame_cut_closest) + frame_end = max(frame, frame_cut_closest) + + trim_strips(context, frame_start, frame_end, self.select_mode, to_select) + + context.scene.frame_current = frame_start + if self.gap_remove and self.select_mode == "CURSOR": + bpy.ops.power_sequencer.gap_remove() + + # FIXME: Workaround Blender 2.80's audio bug, remove when fixed in Blender + sequencer_workaround_2_80_audio_bug(context) + return {"FINISHED"} |