← Back to branch summary

~ubuntu-branches/ubuntu/intrepid/blender/intrepid-updates

« back to all changes in this revision

Viewing changes to release/scripts/armature_symetry.py

  • Committer: Bazaar Package Importer
  • Author(s): Cyril Brulebois
  • Date: 2008-08-08 02:45:40 UTC
  • mfrom: (12.1.14 intrepid)
  • Revision ID: james.westby@ubuntu.com-20080808024540-kkjp7ekfivzhuw3l
Tags: 2.46+dfsg-4
http://bugs.debian.org/492280
* Fix python syntax warning in import_dxf.py, which led to nasty output
  in installation/upgrade logs during byte-compilation, using a patch
  provided by the script author (Closes: #492280):
   - debian/patches/45_fix_python_syntax_warning

Show diffs side-by-side

added added

removed removed

Lines of Context:
release/scripts/armature_symetry.py
1
 
#!BPY
2
 
 
3
 
"""
4
 
Name: 'Armature Symmetry'
5
 
Blender: 234
6
 
Group: 'Animation'
7
 
Tooltip: 'Make an armature symetrical'
8
 
"""
9
 
 
10
 
__author__ = "Jonas Petersen"
11
 
__url__ = ("blender", "elysiun", "Script's homepage, http://www.mindfloaters.de/blender/", "thread at blender.org, http://www.blender.org/modules.php?op=modload&name=phpBB2&file=viewtopic&t=4858")
12
 
__version__ = "0.9 2004-11-10"
13
 
 
14
 
__doc__ = """\
15
 
This script creates perfectly symmetrical armatures.
16
 
 
17
 
With default configuration it will:<br>
18
 
  - Look for bones that have the reference suffix (".L") and
19
 
adjust/create the according opposite bone (suffix ".R");<br>
20
 
  - Center align all bones that _don't_ have the suffix ".X".
21
 
 
22
 
Please check the script's homepage and the thread at blender.org (last link button above) for more info.
23
 
 
24
 
For this version users need to edit the script code to change default options.
25
 
"""
26
 
 
27
 
# --------------------------------------------------------------------------
28
 
# "Armature Symmetry" by Jonas Petersen
29
 
# Version 0.9 - 10th November 2004 - first public release
30
 
# --------------------------------------------------------------------------
31
 
#
32
 
# A script for creating perfectly symmetrical armatures.
33
 
#
34
 
# It is available in Object Mode via the menu item:
35
 
#
36
 
#   Object -> Scripts -> Armature Symmetry
37
 
#
38
 
# With default configuration it will:
39
 
#
40
 
#   - Look for bones that have the reference suffix (".L") and
41
 
#     adjust/create the according opposite bone (suffix ".R").
42
 
#
43
 
#   - Center align all bones that _don't_ have the suffix ".X"
44
 
#
45
 
# Find the latest version at: http://www.mindfloaters.de/blender/
46
 
#
47
 
# --------------------------------------------------------------------------
48
 
# $Id: armature_symetry.py,v 1.1 2005/05/17 07:17:52 ianwill Exp $
49
 
# --------------------------------------------------------------------------
50
 
# ***** BEGIN GPL LICENSE BLOCK *****
51
 
#
52
 
# Copyright (C) 2004: Jonas Petersen, jonas at mindfloaters dot de
53
 
#
54
 
# This program is free software; you can redistribute it and/or
55
 
# modify it under the terms of the GNU General Public License
56
 
# as published by the Free Software Foundation; either version 2
57
 
# of the License, or (at your option) any later version.
58
 
#
59
 
# This program is distributed in the hope that it will be useful,
60
 
# but WITHOUT ANY WARRANTY; without even the implied warranty of
61
 
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
62
 
# GNU General Public License for more details.
63
 
#
64
 
# You should have received a copy of the GNU General Public License
65
 
# along with this program; if not, write to the Free Software Foundation,
66
 
# Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
67
 
#
68
 
# ***** END GPL LICENCE BLOCK *****
69
 
 
70
 
# --------------------------------------------------------------------------
71
 
# CONFIGURATION
72
 
# --------------------------------------------------------------------------
73
 
 
74
 
# Note: Theses values will later be editable via a gui interface
75
 
# within Blender.
76
 
 
77
 
# CENTER_SUFFIX is the suffix for bones that should (or shouldn't) get
78
 
# center aligned. The default is '.X'.
79
 
CENTER_SUFFIX = '.X'
80
 
 
81
 
# CENTER_SUFFIX_MODE:
82
 
#
83
 
#   'include'  only bones with the CENTER_SUFFIX appended
84
 
#              get center aligned.
85
 
#
86
 
#   'exclude'  (default)
87
 
#              all bones except those with the CENTER_SUFFIX
88
 
#              appended get center aligned.
89
 
#
90
 
#
91
 
#   'off'      bones will not get center aligned at all.
92
 
#
93
 
CENTER_SUFFIX_MODE = 'exclude'
94
 
 
95
 
# The suffix for the reference and opposite side of the
96
 
# armature. Bone positions of the opposite side will be overwritten by
97
 
# the mirrored values of the reference side.
98
 
# The default is REF_SUFFIX = '.L' and OPP_SUFFIX = '.R'.
99
 
REF_SUFFIX = '.L'
100
 
OPP_SUFFIX = '.R'
101
 
 
102
 
# MIRROR_AXIS defines the axis in which bones are mirrored/aligned.
103
 
# Values:
104
 
#   0 for X (default)
105
 
#   1 for Y
106
 
#   2 for Z
107
 
MIRROR_AXIS = 0
108
 
 
109
 
# --------------------------------------------------------------------------
110
 
# END OF CONFIGURATION
111
 
# --------------------------------------------------------------------------
112
 
 
113
 
import Blender
114
 
 
115
 
def splitName(bone):
116
 
        name = bone.getName()
117
 
        base = name[0:len(name)-ref_suffix_len]
118
 
        rsuff = name[-ref_suffix_len:len(name)]
119
 
        csuff = name[-center_suffix_len:len(name)]
120
 
        return name, base, rsuff, csuff
121
 
 
122
 
ref_suffix_len = len(REF_SUFFIX);
123
 
center_suffix_len = len(CENTER_SUFFIX);
124
 
armature_selected = False
125
 
 
126
 
obj_list = Blender.Object.GetSelected()
127
 
for obj in obj_list:
128
 
    if obj.getType() == "Armature":
129
 
                armature_selected = True
130
 
                arm = obj.getData()
131
 
                bones = arm.getBones()
132
 
                bonehash = {}
133
 
 
134
 
                for bone in bones:
135
 
                        bonehash[bone.getName()] = bone
136
 
 
137
 
                for bone in bones:
138
 
                        name, base, rsuff, csuff = splitName(bone)
139
 
 
140
 
                        # reference bone?
141
 
                        if (rsuff == REF_SUFFIX):
142
 
                                oppname = base + OPP_SUFFIX
143
 
 
144
 
                                # create opposite bone if necessary
145
 
                                if not bonehash.has_key(oppname):
146
 
                                        bonehash[oppname]=Blender.Armature.Bone.New(oppname)
147
 
                                        parent = bone.getParent()
148
 
                                        if parent:
149
 
                                                pname, pbase, prsuff, pcsuff = splitName(parent)
150
 
                                                if prsuff == REF_SUFFIX:
151
 
                                                        poppname = pbase + OPP_SUFFIX
152
 
                                                        if bonehash.has_key(poppname):
153
 
                                                                bonehash[oppname].setParent(bonehash[poppname])
154
 
                                                else:
155
 
                                                        bonehash[oppname].setParent(parent)
156
 
                                        arm.addBone(bonehash[oppname])
157
 
 
158
 
                                # mirror bone coords
159
 
 
160
 
                                tail = bone.getTail()
161
 
                                tail[MIRROR_AXIS] *= -1;
162
 
                                bonehash[oppname].setTail(tail)
163
 
 
164
 
                                head = bone.getHead()
165
 
                                head[MIRROR_AXIS] *= -1;
166
 
                                bonehash[oppname].setHead(head)
167
 
 
168
 
                                roll = -bone.getRoll()
169
 
                                bonehash[oppname].setRoll(roll)
170
 
 
171
 
                                # Write access to ik flag not (yet?) supported in Blender (2.34)
172
 
                                #if bone.hasParent():
173
 
                                #       bonehash[oppname].setIK(not bone.getIK())
174
 
 
175
 
                        # center bone?
176
 
                        elif (rsuff != OPP_SUFFIX) and \
177
 
                             (CENTER_SUFFIX_MODE != 'off') and \
178
 
                             ((CENTER_SUFFIX_MODE == 'exclude' and csuff != CENTER_SUFFIX) or \
179
 
                             (CENTER_SUFFIX_MODE == 'include' and csuff == CENTER_SUFFIX)):
180
 
 
181
 
                                # center bone coords
182
 
 
183
 
                                tail = bone.getTail()
184
 
                                tail[MIRROR_AXIS] = 0.0;
185
 
                                bone.setTail(tail)
186
 
 
187
 
                                head = bone.getHead()
188
 
                                head[MIRROR_AXIS] = 0.0;
189
 
                                bone.setHead(head)
190
 
 
191
 
                                # Setting set roll in python rotates all child bones.
192
 
                                # Not so if set via the Transform Properties in Blender.
193
 
                                # Bug?
194
 
                                bone.setRoll(0.0)
195
 
 
196
 
if not armature_selected:
197
 
        Blender.Draw.PupMenu("Armature Symmetry%t|Please select an Armature object!")