1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
|
#
# 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.doc import doc_name, doc_idname, doc_brief, doc_description
class POWER_SEQUENCER_OT_swap_strips(bpy.types.Operator):
"""
*brief* Swaps the 2 strips between them
Places the first strip in the channel and starting frame (frame_final_start) of the second
strip, and places the second strip in the channel and starting frame (frame_final_end) of
the first strip. If there is no space for the swap, it does nothing.
"""
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"}
direction: bpy.props.EnumProperty(
name="Direction",
description="The direction to find the closest strip",
items=[
("up", "Up", "The direction up from the selected strip"),
("down", "Down", "The direction down from the selected strip"),
],
default="up",
)
@classmethod
def poll(cls, context):
return context.selected_sequences
def execute(self, context):
strip_1 = context.selected_sequences[0]
if len(context.selected_sequences) == 1:
strip_2 = self.find_closest_strip_vertical(context, strip_1, self.direction)
else:
strip_2 = context.selected_sequences[1]
if not strip_2 or strip_1.lock or strip_2.lock:
return {"CANCELLED"}
# Swap a strip and one of its effects
if hasattr(strip_1, "input_1") or hasattr(strip_2, "input_1"):
if not self.are_linked(strip_1, strip_2):
return {"CANCELLED"}
self.swap_with_effect(strip_1, strip_2)
return {"FINISHED"}
s1_start, s1_channel = strip_1.frame_final_start, strip_1.channel
s2_start, s2_channel = strip_2.frame_final_start, strip_2.channel
self.move_to_end(strip_1, context)
self.move_to_end(strip_2, context)
s1_start_2 = strip_1.frame_final_start
s2_start_2 = strip_2.frame_final_start
group_1 = {
s: s.channel
for s in context.sequences
if s.frame_final_start == s1_start_2 and s != strip_1
}
group_2 = {
s: s.channel
for s in context.sequences
if s.frame_final_start == s2_start_2 and s != strip_2
}
strip_2.select = False
bpy.ops.transform.seq_slide(
value=(s2_start - strip_1.frame_final_start, s2_channel - strip_1.channel)
)
bpy.ops.sequencer.select_all(action="DESELECT")
strip_2.select = True
bpy.ops.transform.seq_slide(
value=(s1_start - strip_2.frame_final_start, s1_channel - strip_2.channel)
)
if not self.fits(
strip_1, group_1, s2_start, s1_channel, s2_channel, context
) or not self.fits(strip_2, group_2, s1_start, s2_channel, s1_channel, context):
self.reconstruct(strip_1, s1_channel, group_1, context)
self.reconstruct(strip_2, s2_channel, group_2, context)
bpy.ops.sequencer.select_all(action="DESELECT")
strip_1.select = True
bpy.ops.transform.seq_slide(
value=(s1_start - strip_1.frame_final_start, s1_channel - strip_1.channel)
)
bpy.ops.sequencer.select_all(action="DESELECT")
strip_2.select = True
bpy.ops.transform.seq_slide(
value=(s2_start - strip_2.frame_final_start, s2_channel - strip_2.channel)
)
bpy.ops.sequencer.select_all(action="DESELECT")
strip_1.select = True
strip_2.select = True
return {"CANCELLED"}
bpy.ops.sequencer.select_all(action="DESELECT")
strip_1.select = True
strip_2.select = True
return {"FINISHED"}
def move_to_frame(self, strip, frame, context):
"""
Moves a strip based on its frame_final_start without changing its
duration.
Args:
- strip: The strip to be moved.
- frame: The frame, the frame_final_start of the strip will be placed
at.
"""
selected_strips = context.selected_sequences
bpy.ops.sequencer.select_all(action="DESELECT")
strip.select = True
bpy.ops.transform.seq_slide(value=(frame - strip.frame_final_start, 0))
bpy.ops.sequencer.select_all(action="DESELECT")
for s in selected_strips:
s.select = True
def move_to_end(self, strip, context):
"""
Moves a strip to an empty slot at the end of the sequencer, different
than its initial slot.
Args:
- strip: The strip to move.
"""
end_frame = max(context.sequences, key=attrgetter("frame_final_end")).frame_final_end
self.move_to_frame(strip, end_frame, context)
def fits(self, strip, group, frame, init_channel, target_channel, context):
"""
Checks if a swap has been successful or not.
Args:
- strip: The core strip of the swap.
- group: The effect strips of the core strip.
- frame: The starting frame of the target location.
- init_channel: The initial channel of the strip, before the swap took
place.
- target_channel: The channel of the target location.
Returns: True if the swap was successful, otherwise False.
"""
if strip.frame_final_start != frame or strip.channel != target_channel:
return False
offset = strip.channel - init_channel
for s in group.keys():
if s.channel != group[s] + offset:
return False
return context.selected_sequences
def reconstruct(self, strip, init_channel, group, context):
"""
Reconstructs a failed swap, based on a core strip. After its done, the
core strip is placed at the end of the sequencer, in an empty slot.
Args:
- strip: The core strip of the swap.
- init_channel: The initial channel of the core strip.
- group: A dictionary with the effect strips of the core strip, and
their target channels.
"""
self.move_to_end(strip, context)
bpy.ops.sequencer.select_all(action="DESELECT")
strip.select = True
bpy.ops.transform.seq_slide(value=(0, init_channel - strip.channel))
for s in group.keys():
channel = group[s]
for u in group.keys():
if u.channel == channel and u != s:
u.channel += 1
s.channel = channel
def find_closest_strip_vertical(self, context, strip, direction):
"""
Finds the closest strip to a given strip in a specific direction.
Args:
- strip: The base strip.
Returns: The closest strip to the given strip, in the proper direction.
If no strip is found, returns None.
"""
strips_in_range = (
s
for s in context.sequences
if strip.frame_final_start <= s.frame_final_start
and s.frame_final_end <= strip.frame_final_end
)
if direction == "up":
strips_above = [s for s in strips_in_range if s.channel > strip.channel]
if not strips_above:
return
return min(strips_above, key=attrgetter("channel"))
elif direction == "down":
strips_below = [s for s in strips_in_range if s.channel < strip.channel]
if not strips_below:
return
return max(strips_below, key=attrgetter("channel"))
def are_linked(self, strip_1, strip_2):
return (
strip_1.frame_final_start == strip_2.frame_final_start
and strip_1.frame_final_end == strip_2.frame_final_end
)
def swap_with_effect(self, strip_1, strip_2):
effect_strip = strip_1 if hasattr(strip_1, "input_1") else strip_2
other_strip = strip_1 if effect_strip != strip_1 else strip_2
effect_strip_channel = effect_strip.channel
other_strip_channel = other_strip.channel
effect_strip.channel -= 1
other_strip.channel = effect_strip_channel
effect_strip.channel = other_strip_channel
|