Merge branch 'blender2.7'
[blender.git] / release / scripts / startup / bl_operators / constraint.py
1 # ##### BEGIN GPL LICENSE BLOCK #####
2 #
3 #  This program is free software; you can redistribute it and/or
4 #  modify it under the terms of the GNU General Public License
5 #  as published by the Free Software Foundation; either version 2
6 #  of the License, or (at your option) any later version.
7 #
8 #  This program is distributed in the hope that it will be useful,
9 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
10 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11 #  GNU General Public License for more details.
12 #
13 #  You should have received a copy of the GNU General Public License
14 #  along with this program; if not, write to the Free Software Foundation,
15 #  Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
16 #
17 # ##### END GPL LICENSE BLOCK #####
18
19 # <pep8-80 compliant>
20
21 import bpy
22 from bpy.types import (
23     Operator,
24 )
25 from bpy.props import (
26     IntProperty,
27 )
28
29
30 class CONSTRAINT_OT_add_target(Operator):
31     """Add a target to the constraint"""
32     bl_idname = "constraint.add_target"
33     bl_label = "Add Target"
34     bl_options = {'UNDO', 'INTERNAL'}
35
36     def execute(self, context):
37         context.constraint.targets.new()
38         return {'FINISHED'}
39
40
41 class CONSTRAINT_OT_remove_target(Operator):
42     """Remove the target from the constraint"""
43     bl_idname = "constraint.remove_target"
44     bl_label = "Remove Target"
45     bl_options = {'UNDO', 'INTERNAL'}
46
47     index: IntProperty()
48
49     def execute(self, context):
50         tgts = context.constraint.targets
51         tgts.remove(tgts[self.index])
52         return {'FINISHED'}
53
54
55 class CONSTRAINT_OT_normalize_target_weights(Operator):
56     """Normalize weights of all target bones"""
57     bl_idname = "constraint.normalize_target_weights"
58     bl_label = "Normalize Weights"
59     bl_options = {'UNDO', 'INTERNAL'}
60
61     def execute(self, context):
62         tgts = context.constraint.targets
63         total = sum(t.weight for t in tgts)
64
65         if total > 0:
66             for t in tgts:
67                 t.weight = t.weight / total
68
69         return {'FINISHED'}
70
71
72 classes = (
73     CONSTRAINT_OT_add_target,
74     CONSTRAINT_OT_remove_target,
75     CONSTRAINT_OT_normalize_target_weights,
76 )