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
|
# ##### BEGIN GPL LICENSE BLOCK #####
#
# This program 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 2
# of the License, or (at your option) any later version.
#
# This program 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 this program; if not, write to the Free Software Foundation,
# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
#
# ##### END GPL LICENSE BLOCK #####
# <pep8 compliant>
from bpy.types import Panel
from bpy.props import (StringProperty,
BoolProperty,
IntProperty,
FloatProperty,
EnumProperty,
PointerProperty,
CollectionProperty,
FloatVectorProperty,
IntVectorProperty,
)
def get_active_strip(context):
return next([strip for strip in context.selected_nla_strips if strip.active],None)
class OBJECT_OT_nla_preblend_add_bone(bpy.types.Operator):
bl_idname = "pose.null_op"
bl_label = "NULL OP"
bl_options = {'REGISTER', 'UNDO'}
preblend_index : IntProperty()
@classmethod
def poll(cls,context):
return context.selected_nla_strips
def execute(self, context):
active_strip = get_active_strip()
preblend = active_strip.preblend_transforms[self.preblend_index]
preblend.bones.add()
return {'FINISHED'}
class OBJECT_OT_nla_preblend_remove_bone(bpy.types.Operator):
bl_idname = "pose.null_op"
bl_label = "NULL OP"
bl_options = {'REGISTER', 'UNDO'}
preblend_index : IntProperty()
bone_index : IntProperty()
@classmethod
def poll(cls,context):
return context.selected_nla_strips
def execute(self, context):
active_strip = get_active_strip()
preblend = active_strip.preblend_transforms[self.preblend_index]
preblend.bones.remove(self.bone_index)
return {'FINISHED'}
class OBJECT_PT_nla_alignment(Panel):
bl_space_type = 'NLA_EDITOR'
bl_region_type = 'UI'
bl_label = "Alignment"
def draw(self, context):
#Q: not in a poll() so it doesnt disappear on nla strip deselection (annoying)
if not context.selected_nla_strips:
return
layout = self.layout
active_strip = get_active_strip()
box = layout.box()
for i,preblend in enumerate(active_strip.preblend_transforms):
box.prop(preblend,"location")
box.prop(preblend,"euler")
box.prop(preblend,"scale")
col = box.column(align=True)
row = col.row(align=True)
#todo: support for objects?
row.label(text="Bones")
row.operator(OBJECT_OT_nla_preblend_add_bone.bl_idname,text='',icon='ADD').preblend_index = i
for j,bone in enumerate(preblend.bones):
row = box.row(align=True)
row.prop_search(bone,"name",context.active_object.data,"bones",text='')
op = row.operator(OBJECT_OT_nla_preblend_remove_bone.bl_idname,text='',icon='REMOVE')
op.preblend_index = i
op.bone_index = j
classes = (
# Object Panels
OBJECT_PT_nla_alignment,
OBJECT_OT_nla_preblend_remove_bone,
OBJECT_OT_nla_preblend_add_bone,
)
if __name__ == "__main__": # only for live edit.
from bpy.utils import register_class
for cls in classes:
register_class(cls)
|
