svn merge -r 22628:22753 https://svn.blender.org/svnroot/bf-blender/branches/blender2...
[blender.git] / source / blender / makesrna / intern / rna_pose.c
1 /**
2  * $Id$
3  *
4  * ***** BEGIN GPL LICENSE BLOCK *****
5  *
6  * This program is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU General Public License
8  * as published by the Free Software Foundation; either version 2
9  * of the License, or (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software Foundation,
18  * Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
19  *
20  * Contributor(s): Blender Foundation (2008), Roland Hess, Joshua Leung
21  *
22  * ***** END GPL LICENSE BLOCK *****
23  */
24
25 #include <stdlib.h>
26
27 #include "RNA_define.h"
28 #include "RNA_types.h"
29
30 #include "rna_internal.h"
31
32 #include "DNA_action_types.h"
33 #include "DNA_armature_types.h"
34 #include "DNA_constraint_types.h"
35 #include "DNA_object_types.h"
36 #include "DNA_scene_types.h"
37
38 #include "WM_types.h"
39
40 #ifdef RNA_RUNTIME
41
42 #include <string.h>
43
44 #include "BLI_arithb.h"
45
46 #include "DNA_userdef_types.h"
47
48 #include "BKE_context.h"
49 #include "BKE_depsgraph.h"
50 #include "BKE_idprop.h"
51
52 #include "ED_armature.h"
53
54 static void rna_Pose_update(bContext *C, PointerRNA *ptr)
55 {
56         // XXX when to use this? ob->pose->flag |= (POSE_LOCKED|POSE_DO_UNLOCK);
57
58         DAG_object_flush_update(CTX_data_scene(C), ptr->id.data, OB_RECALC_DATA);
59 }
60
61 static char *rna_PoseChannel_path(PointerRNA *ptr)
62 {
63         return BLI_sprintfN("pose.pose_channels[\"%s\"]", ((bPoseChannel*)ptr->data)->name);
64 }
65
66 static void rna_BoneGroup_color_set_set(PointerRNA *ptr, int value)
67 {
68         bActionGroup *grp= ptr->data;
69         
70         /* if valid value, set the new enum value, then copy the relevant colours? */
71         if ((value >= -1) && (value < 21))
72                 grp->customCol= value;
73         else
74                 return;
75         
76         /* only do color copying if using a custom color (i.e. not default colour)  */
77         if (grp->customCol) {
78                 if (grp->customCol > 0) {
79                         /* copy theme colors on-to group's custom color in case user tries to edit color */
80                         bTheme *btheme= U.themes.first;
81                         ThemeWireColor *col_set= &btheme->tarm[(grp->customCol - 1)];
82                         
83                         memcpy(&grp->cs, col_set, sizeof(ThemeWireColor));
84                 }
85                 else {
86                         /* init custom colors with a generic multi-color rgb set, if not initialised already (for custom color set) */
87                         if (grp->cs.solid[0] == 0) {
88                                 /* define for setting colors in theme below */
89                                 #define SETCOL(col, r, g, b, a)  col[0]=r; col[1]=g; col[2]= b; col[3]= a;
90                                 
91                                 SETCOL(grp->cs.solid, 0xff, 0x00, 0x00, 255);
92                                 SETCOL(grp->cs.select, 0x81, 0xe6, 0x14, 255);
93                                 SETCOL(grp->cs.active, 0x18, 0xb6, 0xe0, 255);
94                                 
95                                 #undef SETCOL
96                         }
97                 }
98         }
99 }
100
101 IDProperty *rna_PoseChannel_idproperties(PointerRNA *ptr, int create)
102 {
103         bPoseChannel *pchan= ptr->data;
104
105         if(create && !pchan->prop) {
106                 IDPropertyTemplate val = {0};
107                 pchan->prop= IDP_New(IDP_GROUP, val, "RNA_PoseChannel group");
108         }
109
110         return pchan->prop;
111 }
112
113 static void rna_PoseChannel_euler_rotation_get(PointerRNA *ptr, float *value)
114 {
115         bPoseChannel *pchan= ptr->data;
116
117         if(pchan->rotmode == PCHAN_ROT_QUAT)
118                 QuatToEul(pchan->quat, value);
119         else
120                 VECCOPY(value, pchan->eul);
121 }
122
123 static void rna_PoseChannel_euler_rotation_set(PointerRNA *ptr, const float *value)
124 {
125         bPoseChannel *pchan= ptr->data;
126
127         if(pchan->rotmode == PCHAN_ROT_QUAT)
128                 EulToQuat((float*)value, pchan->quat);
129         else
130                 VECCOPY(pchan->eul, value);
131 }
132
133 static void rna_PoseChannel_name_set(PointerRNA *ptr, const char *value)
134 {
135         Object *ob= (Object*)ptr->id.data;
136         bPoseChannel *pchan= (bPoseChannel*)ptr->data;
137         char oldname[32], newname[32];
138         
139         /* need to be on the stack */
140         BLI_strncpy(newname, value, 32);
141         BLI_strncpy(oldname, pchan->name, 32);
142         
143         ED_armature_bone_rename(ob->data, oldname, newname);
144 }
145
146 static int rna_PoseChannel_has_ik_get(PointerRNA *ptr)
147 {
148         Object *ob= (Object*)ptr->id.data;
149         bPoseChannel *pchan= (bPoseChannel*)ptr->data;
150
151         return ED_pose_channel_in_IK_chain(ob, pchan);
152 }
153
154 static PointerRNA rna_PoseChannel_bone_group_get(PointerRNA *ptr)
155 {
156         Object *ob= (Object*)ptr->id.data;
157         bPose *pose= (ob) ? ob->pose : NULL;
158         bPoseChannel *pchan= (bPoseChannel*)ptr->data;
159         bActionGroup *grp;
160         
161         if (pose)
162                 grp= BLI_findlink(&pose->agroups, pchan->agrp_index-1);
163         else
164                 grp= NULL;
165         
166         return rna_pointer_inherit_refine(ptr, &RNA_BoneGroup, grp);
167 }
168
169 static void rna_PoseChannel_bone_group_set(PointerRNA *ptr, PointerRNA value)
170 {
171         Object *ob= (Object*)ptr->id.data;
172         bPose *pose= (ob) ? ob->pose : NULL;
173         bPoseChannel *pchan= (bPoseChannel*)ptr->data;
174         
175         if (pose)
176                 pchan->agrp_index= BLI_findindex(&pose->agroups, value.data) + 1;
177         else
178                 pchan->agrp_index= 0;
179 }
180
181 static int rna_PoseChannel_bone_group_index_get(PointerRNA *ptr)
182 {
183         bPoseChannel *pchan= (bPoseChannel*)ptr->data;
184         return MAX2(pchan->agrp_index-1, 0);
185 }
186
187 static void rna_PoseChannel_bone_group_index_set(PointerRNA *ptr, int value)
188 {
189         bPoseChannel *pchan= (bPoseChannel*)ptr->data;
190         pchan->agrp_index= value+1;
191 }
192
193 static void rna_PoseChannel_bone_group_index_range(PointerRNA *ptr, int *min, int *max)
194 {
195         Object *ob= (Object*)ptr->id.data;
196         bPose *pose= (ob) ? ob->pose : NULL;
197         
198         *min= 0;
199         
200         if (pose) {
201                 *max= BLI_countlist(&pose->agroups)-1;
202                 *max= MAX2(0, *max);
203         }
204         else
205                 *max= 0;
206 }
207
208 static PointerRNA rna_Pose_active_bone_group_get(PointerRNA *ptr)
209 {
210         bPose *pose= (bPose*)ptr->data;
211         return rna_pointer_inherit_refine(ptr, &RNA_BoneGroup, BLI_findlink(&pose->agroups, pose->active_group-1));
212 }
213
214 static void rna_Pose_active_bone_group_set(PointerRNA *ptr, PointerRNA value)
215 {
216         bPose *pose= (bPose*)ptr->data;
217         pose->active_group= BLI_findindex(&pose->agroups, value.data) + 1;
218 }
219
220 static int rna_Pose_active_bone_group_index_get(PointerRNA *ptr)
221 {
222         bPose *pose= (bPose*)ptr->data;
223         return MAX2(pose->active_group-1, 0);
224 }
225
226 static void rna_Pose_active_bone_group_index_set(PointerRNA *ptr, int value)
227 {
228         bPose *pose= (bPose*)ptr->data;
229         pose->active_group= value+1;
230 }
231
232 static void rna_Pose_active_bone_group_index_range(PointerRNA *ptr, int *min, int *max)
233 {
234         bPose *pose= (bPose*)ptr->data;
235
236         *min= 0;
237         *max= BLI_countlist(&pose->agroups)-1;
238         *max= MAX2(0, *max);
239 }
240
241 void rna_pose_bgroup_name_index_get(PointerRNA *ptr, char *value, int index)
242 {
243         bPose *pose= (bPose*)ptr->data;
244         bActionGroup *grp;
245
246         grp= BLI_findlink(&pose->agroups, index-1);
247
248         if(grp) BLI_strncpy(value, grp->name, sizeof(grp->name));
249         else BLI_strncpy(value, "", sizeof(grp->name)); // XXX if invalid pointer, won't this crash?
250 }
251
252 int rna_pose_bgroup_name_index_length(PointerRNA *ptr, int index)
253 {
254         bPose *pose= (bPose*)ptr->data;
255         bActionGroup *grp;
256
257         grp= BLI_findlink(&pose->agroups, index-1);
258         return (grp)? strlen(grp->name): 0;
259 }
260
261 void rna_pose_bgroup_name_index_set(PointerRNA *ptr, const char *value, short *index)
262 {
263         bPose *pose= (bPose*)ptr->data;
264         bActionGroup *grp;
265         int a;
266         
267         for (a=1, grp=pose->agroups.first; grp; grp=grp->next, a++) {
268                 if (strcmp(grp->name, value) == 0) {
269                         *index= a;
270                         return;
271                 }
272         }
273         
274         *index= 0;
275 }
276
277 void rna_pose_pgroup_name_set(PointerRNA *ptr, const char *value, char *result, int maxlen)
278 {
279         bPose *pose= (bPose*)ptr->data;
280         bActionGroup *grp;
281         
282         for (grp= pose->agroups.first; grp; grp= grp->next) {
283                 if (strcmp(grp->name, value) == 0) {
284                         BLI_strncpy(result, value, maxlen);
285                         return;
286                 }
287         }
288         
289         BLI_strncpy(result, "", maxlen);
290 }
291
292 #else
293
294 static void rna_def_bone_group(BlenderRNA *brna)
295 {
296         static EnumPropertyItem prop_colorSets_items[] = {
297                 {0, "DEFAULT", 0, "Default Colors", ""},
298                 {1, "THEME01", 0, "01 - Theme Color Set", ""},
299                 {2, "THEME02", 0, "02 - Theme Color Set", ""},
300                 {3, "THEME03", 0, "03 - Theme Color Set", ""},
301                 {4, "THEME04", 0, "04 - Theme Color Set", ""},
302                 {5, "THEME05", 0, "05 - Theme Color Set", ""},
303                 {6, "THEME06", 0, "06 - Theme Color Set", ""},
304                 {7, "THEME07", 0, "07 - Theme Color Set", ""},
305                 {8, "THEME08", 0, "08 - Theme Color Set", ""},
306                 {9, "THEME09", 0, "09 - Theme Color Set", ""},
307                 {10, "THEME10", 0, "10 - Theme Color Set", ""},
308                 {11, "THEME11", 0, "11 - Theme Color Set", ""},
309                 {12, "THEME12", 0, "12 - Theme Color Set", ""},
310                 {13, "THEME13", 0, "13 - Theme Color Set", ""},
311                 {14, "THEME14", 0, "14 - Theme Color Set", ""},
312                 {15, "THEME15", 0, "15 - Theme Color Set", ""},
313                 {16, "THEME16", 0, "16 - Theme Color Set", ""},
314                 {17, "THEME17", 0, "17 - Theme Color Set", ""},
315                 {18, "THEME18", 0, "18 - Theme Color Set", ""},
316                 {19, "THEME19", 0, "19 - Theme Color Set", ""},
317                 {20, "THEME20", 0, "20 - Theme Color Set", ""},
318                 {-1, "CUSTOM", 0, "Custom Color Set", ""},
319                 {0, NULL, 0, NULL, NULL}};
320         
321         StructRNA *srna;
322         PropertyRNA *prop;
323         
324         /* struct */
325         srna= RNA_def_struct(brna, "BoneGroup", NULL);
326         RNA_def_struct_sdna(srna, "bActionGroup");
327         RNA_def_struct_ui_text(srna, "Bone Group", "Groups of Pose Channels (Bones).");
328         
329         /* name */
330         prop= RNA_def_property(srna, "name", PROP_STRING, PROP_NONE);
331         RNA_def_property_ui_text(prop, "Name", "");
332         RNA_def_struct_name_property(srna, prop);
333         
334         // TODO: add some runtime-collections stuff to access grouped bones 
335         
336         /* color set + colors */
337         prop= RNA_def_property(srna, "color_set", PROP_ENUM, PROP_NONE);
338         RNA_def_property_enum_sdna(prop, NULL, "customCol");
339         RNA_def_property_enum_items(prop, prop_colorSets_items);
340         RNA_def_property_enum_funcs(prop, NULL, "rna_BoneGroup_color_set_set", NULL);
341         RNA_def_property_ui_text(prop, "Color Set", "Custom color set to use.");
342         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
343         
344                 // TODO: editing the colors for this should result in changes to the color type...
345         prop= RNA_def_property(srna, "colors", PROP_POINTER, PROP_NEVER_NULL);
346         RNA_def_property_struct_type(prop, "ThemeBoneColorSet");
347         RNA_def_property_pointer_sdna(prop, NULL, "cs"); /* NOTE: the DNA data is not really a pointer, but this code works :) */
348         RNA_def_property_ui_text(prop, "Colors", "Copy of the colors associated with the group's color set.");
349         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
350 }
351
352 static void rna_def_pose_channel(BlenderRNA *brna)
353 {
354         static EnumPropertyItem prop_rotmode_items[] = {
355                 {PCHAN_ROT_QUAT, "QUATERNION", 0, "Quaternion (WXYZ)", "No Gimbal Lock (default)"},
356                 {PCHAN_ROT_EUL, "EULER", 0, "Euler (XYZ)", "Prone to Gimbal Lock"},
357                 {0, NULL, 0, NULL, NULL}};
358         
359         StructRNA *srna;
360         PropertyRNA *prop;
361
362         srna= RNA_def_struct(brna, "PoseChannel", NULL);
363         RNA_def_struct_sdna(srna, "bPoseChannel");
364         RNA_def_struct_ui_text(srna, "Pose Channel", "Channel defining pose data for a bone in a Pose.");
365         RNA_def_struct_path_func(srna, "rna_PoseChannel_path");
366         RNA_def_struct_idproperties_func(srna, "rna_PoseChannel_idproperties");
367
368         /* Bone Constraints */
369         prop= RNA_def_property(srna, "constraints", PROP_COLLECTION, PROP_NONE);
370         RNA_def_property_struct_type(prop, "Constraint");
371         RNA_def_property_ui_text(prop, "Constraints", "Constraints that act on this PoseChannel."); 
372
373         /* Name + Selection Status */
374         prop= RNA_def_property(srna, "name", PROP_STRING, PROP_NONE);
375         RNA_def_property_string_funcs(prop, NULL, NULL, "rna_PoseChannel_name_set");
376         RNA_def_property_ui_text(prop, "Name", "");
377         RNA_def_struct_name_property(srna, prop);
378
379         prop= RNA_def_property(srna, "selected", PROP_BOOLEAN, PROP_NONE);
380         RNA_def_property_boolean_sdna(prop, NULL, "selectflag", BONE_SELECTED);
381         RNA_def_property_ui_text(prop, "Selected", "");
382         
383         /* Baked Bone Path cache data s*/
384         prop= RNA_def_property(srna, "path_start_frame", PROP_INT, PROP_TIME);
385         RNA_def_property_int_sdna(prop, NULL, "pathsf");
386         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
387         RNA_def_property_ui_text(prop, "Bone Paths Calculation Start Frame", "Starting frame of range of frames to use for Bone Path calculations.");
388         RNA_def_property_update(prop, NC_OBJECT|ND_POSE|ND_TRANSFORM, "rna_Pose_update");
389
390         prop= RNA_def_property(srna, "path_end_frame", PROP_INT, PROP_TIME);
391         RNA_def_property_int_sdna(prop, NULL, "pathef");
392         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
393         RNA_def_property_ui_text(prop, "Bone Paths Calculation End Frame", "End frame of range of frames to use for Bone Path calculations.");
394         RNA_def_property_update(prop, NC_OBJECT|ND_POSE|ND_TRANSFORM, "rna_Pose_update");
395
396         /* Relationships to other bones */
397         prop= RNA_def_property(srna, "bone", PROP_POINTER, PROP_NEVER_NULL);
398         RNA_def_property_struct_type(prop, "Bone");
399         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
400         RNA_def_property_ui_text(prop, "Bone", "Bone associated with this Pose Channel.");
401
402         prop= RNA_def_property(srna, "parent", PROP_POINTER, PROP_NONE);
403         RNA_def_property_struct_type(prop, "PoseChannel");
404         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
405         RNA_def_property_ui_text(prop, "Parent", "Parent of this pose channel.");
406
407         prop= RNA_def_property(srna, "child", PROP_POINTER, PROP_NONE);
408         RNA_def_property_struct_type(prop, "PoseChannel");
409         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
410         RNA_def_property_ui_text(prop, "Child", "Child of this pose channel.");
411
412         /* Transformation settings */
413         prop= RNA_def_property(srna, "location", PROP_FLOAT, PROP_TRANSLATION);
414         RNA_def_property_float_sdna(prop, NULL, "loc");
415         RNA_def_property_ui_text(prop, "Location", "");
416         RNA_def_property_update(prop, NC_OBJECT|ND_POSE|ND_TRANSFORM, "rna_Pose_update");
417
418         prop= RNA_def_property(srna, "scale", PROP_FLOAT, PROP_XYZ);
419         RNA_def_property_float_sdna(prop, NULL, "size");
420         RNA_def_property_ui_text(prop, "Scale", "");
421         RNA_def_property_update(prop, NC_OBJECT|ND_POSE|ND_TRANSFORM, "rna_Pose_update");
422
423         prop= RNA_def_property(srna, "rotation", PROP_FLOAT, PROP_QUATERNION);
424         RNA_def_property_float_sdna(prop, NULL, "quat");
425         RNA_def_property_ui_text(prop, "Rotation", "Rotation in Quaternions.");
426         RNA_def_property_update(prop, NC_OBJECT|ND_POSE|ND_TRANSFORM, "rna_Pose_update");
427         
428         prop= RNA_def_property(srna, "euler_rotation", PROP_FLOAT, PROP_EULER);
429         RNA_def_property_float_sdna(prop, NULL, "eul");
430         RNA_def_property_float_funcs(prop, "rna_PoseChannel_euler_rotation_get", "rna_PoseChannel_euler_rotation_set", NULL);
431         RNA_def_property_ui_text(prop, "Rotation (Euler)", "Rotation in Eulers.");
432         RNA_def_property_update(prop, NC_OBJECT|ND_POSE|ND_TRANSFORM, "rna_Pose_update");
433         
434         prop= RNA_def_property(srna, "rotation_mode", PROP_ENUM, PROP_NONE);
435         RNA_def_property_enum_sdna(prop, NULL, "rotmode");
436         RNA_def_property_enum_items(prop, prop_rotmode_items);
437         RNA_def_property_ui_text(prop, "Rotation Mode", "");
438         RNA_def_property_update(prop, NC_OBJECT|ND_POSE|ND_TRANSFORM, "rna_Pose_update");
439
440         /* These three matrix properties await an implementation of the PROP_MATRIX subtype, which currently doesn't exist. */
441 /*      prop= RNA_def_property(srna, "channel_matrix", PROP_FLOAT, PROP_MATRIX);
442         RNA_def_property_struct_type(prop, "chan_mat");
443         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
444         RNA_def_property_ui_text(prop, "Channel Matrix", "4x4 matrix, before constraints.");*/
445
446         /* kaito says this should be not user-editable; I disagree; power users should be able to force this in python; he's the boss. */
447 /*      prop= RNA_def_property(srna, "pose_matrix", PROP_FLOAT, PROP_MATRIX);
448         RNA_def_property_struct_type(prop, "pose_mat");
449         RNA_def_property_clear_flag(prop, PROP_EDITABLE); 
450         RNA_def_property_ui_text(prop, "Pose Matrix", "Final 4x4 matrix for this channel.");
451
452         prop= RNA_def_property(srna, "constraint_inverse_matrix", PROP_FLOAT, PROP_MATRIX);
453         RNA_def_property_struct_type(prop, "constinv");
454         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
455         RNA_def_property_ui_text(prop, "Constraint Inverse Matrix", "4x4 matrix, defines transform from final position to unconstrained position."); */
456         
457         /* Head/Tail Coordinates (in Pose Space) - Automatically calculated... */
458         prop= RNA_def_property(srna, "pose_head", PROP_FLOAT, PROP_TRANSLATION);
459         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
460         RNA_def_property_ui_text(prop, "Pose Head Position", "Location of head of the channel's bone.");
461
462         prop= RNA_def_property(srna, "pose_tail", PROP_FLOAT, PROP_TRANSLATION);
463         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
464         RNA_def_property_ui_text(prop, "Pose Tail Position", "Location of tail of the channel's bone.");
465
466         /* IK Settings */
467         prop= RNA_def_property(srna, "has_ik", PROP_BOOLEAN, PROP_NONE);
468         RNA_def_property_boolean_funcs(prop,  "rna_PoseChannel_has_ik_get", NULL);
469         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
470         RNA_def_property_ui_text(prop, "Has IK", "Is part of an IK chain.");
471         RNA_def_property_update(prop, NC_OBJECT|ND_POSE|ND_TRANSFORM, "rna_Pose_update");
472
473         prop= RNA_def_property(srna, "ik_dof_x", PROP_BOOLEAN, PROP_NONE);
474         RNA_def_property_boolean_negative_sdna(prop, NULL, "ikflag", BONE_IK_NO_XDOF);
475         RNA_def_property_ui_text(prop, "IK X DoF", "Allow movement around the X axis.");
476         RNA_def_property_update(prop, NC_OBJECT|ND_POSE|ND_TRANSFORM, "rna_Pose_update");
477
478         prop= RNA_def_property(srna, "ik_dof_y", PROP_BOOLEAN, PROP_NONE);
479         RNA_def_property_boolean_negative_sdna(prop, NULL, "ikflag", BONE_IK_NO_YDOF);
480         RNA_def_property_ui_text(prop, "IK Y DoF", "Allow movement around the Y axis.");
481         RNA_def_property_update(prop, NC_OBJECT|ND_POSE|ND_TRANSFORM, "rna_Pose_update");
482
483         prop= RNA_def_property(srna, "ik_dof_z", PROP_BOOLEAN, PROP_NONE);
484         RNA_def_property_boolean_negative_sdna(prop, NULL, "ikflag", BONE_IK_NO_ZDOF);
485         RNA_def_property_ui_text(prop, "IK Z DoF", "Allow movement around the Z axis.");
486         RNA_def_property_update(prop, NC_OBJECT|ND_POSE|ND_TRANSFORM, "rna_Pose_update");
487
488         prop= RNA_def_property(srna, "ik_limit_x", PROP_BOOLEAN, PROP_NONE);
489         RNA_def_property_boolean_sdna(prop, NULL, "ikflag", BONE_IK_XLIMIT);
490         RNA_def_property_ui_text(prop, "IK X Limit", "Limit movement around the X axis.");
491         RNA_def_property_update(prop, NC_OBJECT|ND_POSE|ND_TRANSFORM, "rna_Pose_update");
492
493         prop= RNA_def_property(srna, "ik_limit_y", PROP_BOOLEAN, PROP_NONE);
494         RNA_def_property_boolean_sdna(prop, NULL, "ikflag", BONE_IK_YLIMIT);
495         RNA_def_property_ui_text(prop, "IK Y Limit", "Limit movement around the Y axis.");
496         RNA_def_property_update(prop, NC_OBJECT|ND_POSE|ND_TRANSFORM, "rna_Pose_update");
497
498         prop= RNA_def_property(srna, "ik_limit_z", PROP_BOOLEAN, PROP_NONE);
499         RNA_def_property_boolean_sdna(prop, NULL, "ikflag", BONE_IK_ZLIMIT);
500         RNA_def_property_ui_text(prop, "IK Z Limit", "Limit movement around the Z axis.");
501         RNA_def_property_update(prop, NC_OBJECT|ND_POSE|ND_TRANSFORM, "rna_Pose_update");
502         
503         prop= RNA_def_property(srna, "ik_min_x", PROP_FLOAT, PROP_ANGLE);
504         RNA_def_property_float_sdna(prop, NULL, "limitmin[0]");
505         RNA_def_property_range(prop, -180.0f, 0.0f);
506         RNA_def_property_ui_text(prop, "IK X Minimum", "Minimum angles for IK Limit");
507         RNA_def_property_update(prop, NC_OBJECT|ND_POSE|ND_TRANSFORM, "rna_Pose_update");
508
509         prop= RNA_def_property(srna, "ik_max_x", PROP_FLOAT, PROP_ANGLE);
510         RNA_def_property_float_sdna(prop, NULL, "limitmax[0]");
511         RNA_def_property_range(prop, 0.0f, 180.0f);
512         RNA_def_property_ui_text(prop, "IK X Maximum", "Maximum angles for IK Limit");
513         RNA_def_property_update(prop, NC_OBJECT|ND_POSE|ND_TRANSFORM, "rna_Pose_update");
514
515         prop= RNA_def_property(srna, "ik_min_y", PROP_FLOAT, PROP_ANGLE);
516         RNA_def_property_float_sdna(prop, NULL, "limitmin[1]");
517         RNA_def_property_range(prop, -180.0f, 0.0f);
518         RNA_def_property_ui_text(prop, "IK Y Minimum", "Minimum angles for IK Limit");
519         RNA_def_property_update(prop, NC_OBJECT|ND_POSE|ND_TRANSFORM, "rna_Pose_update");
520
521         prop= RNA_def_property(srna, "ik_max_y", PROP_FLOAT, PROP_ANGLE);
522         RNA_def_property_float_sdna(prop, NULL, "limitmax[1]");
523         RNA_def_property_range(prop, 0.0f, 180.0f);
524         RNA_def_property_ui_text(prop, "IK Y Maximum", "Maximum angles for IK Limit");
525         RNA_def_property_update(prop, NC_OBJECT|ND_POSE|ND_TRANSFORM, "rna_Pose_update");
526
527         prop= RNA_def_property(srna, "ik_min_z", PROP_FLOAT, PROP_ANGLE);
528         RNA_def_property_float_sdna(prop, NULL, "limitmin[2]");
529         RNA_def_property_range(prop, -180.0f, 0.0f);
530         RNA_def_property_ui_text(prop, "IK Z Minimum", "Minimum angles for IK Limit");
531         RNA_def_property_update(prop, NC_OBJECT|ND_POSE|ND_TRANSFORM, "rna_Pose_update");
532
533         prop= RNA_def_property(srna, "ik_max_z", PROP_FLOAT, PROP_ANGLE);
534         RNA_def_property_float_sdna(prop, NULL, "limitmax[2]");
535         RNA_def_property_range(prop, 0.0f, 180.0f);
536         RNA_def_property_ui_text(prop, "IK Z Maximum", "Maximum angles for IK Limit");
537         RNA_def_property_update(prop, NC_OBJECT|ND_POSE|ND_TRANSFORM, "rna_Pose_update");
538
539         prop= RNA_def_property(srna, "ik_stiffness_x", PROP_FLOAT, PROP_NONE);
540         RNA_def_property_float_sdna(prop, NULL, "stiffness[0]");
541         RNA_def_property_range(prop, 0.0f, 0.99f);
542         RNA_def_property_ui_text(prop, "IK X Stiffness", "IK stiffness around the X axis.");
543         RNA_def_property_update(prop, NC_OBJECT|ND_POSE|ND_TRANSFORM, "rna_Pose_update");
544
545         prop= RNA_def_property(srna, "ik_stiffness_y", PROP_FLOAT, PROP_NONE);
546         RNA_def_property_float_sdna(prop, NULL, "stiffness[1]");
547         RNA_def_property_range(prop, 0.0f, 0.99f);
548         RNA_def_property_ui_text(prop, "IK Y Stiffness", "IK stiffness around the Y axis.");
549         RNA_def_property_update(prop, NC_OBJECT|ND_POSE|ND_TRANSFORM, "rna_Pose_update");
550
551         prop= RNA_def_property(srna, "ik_stiffness_z", PROP_FLOAT, PROP_NONE);
552         RNA_def_property_float_sdna(prop, NULL, "stiffness[2]");
553         RNA_def_property_range(prop, 0.0f, 0.99f);
554         RNA_def_property_ui_text(prop, "IK Z Stiffness", "IK stiffness around the Z axis.");
555         RNA_def_property_update(prop, NC_OBJECT|ND_POSE|ND_TRANSFORM, "rna_Pose_update");
556
557         prop= RNA_def_property(srna, "ik_stretch", PROP_FLOAT, PROP_NONE);
558         RNA_def_property_float_sdna(prop, NULL, "ikstretch");
559         RNA_def_property_range(prop, 0.0f,1.0f);
560         RNA_def_property_ui_text(prop, "IK Stretch", "Allow scaling of the bone for IK.");
561         RNA_def_property_update(prop, NC_OBJECT|ND_POSE|ND_TRANSFORM, "rna_Pose_update");
562
563         /* custom bone shapes */
564         prop= RNA_def_property(srna, "custom_shape", PROP_POINTER, PROP_NONE);
565         RNA_def_property_pointer_sdna(prop, NULL, "custom");
566         RNA_def_property_struct_type(prop, "Object");
567         RNA_def_property_flag(prop, PROP_EDITABLE);
568         RNA_def_property_ui_text(prop, "Custom Object", "Object that defines custom draw type for this bone.");
569         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
570
571         /* bone groups */
572         prop= RNA_def_property(srna, "bone_group_index", PROP_INT, PROP_NONE);
573         RNA_def_property_int_sdna(prop, NULL, "agrp_index");
574         RNA_def_property_flag(prop, PROP_EDITABLE);
575         RNA_def_property_int_funcs(prop, "rna_PoseChannel_bone_group_index_get", "rna_PoseChannel_bone_group_index_set", "rna_PoseChannel_bone_group_index_range");
576         RNA_def_property_ui_text(prop, "Bone Group Index", "Bone Group this pose channel belongs to (0=no group).");
577         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
578         
579         prop= RNA_def_property(srna, "bone_group", PROP_POINTER, PROP_NONE);
580         RNA_def_property_struct_type(prop, "BoneGroup");
581         RNA_def_property_flag(prop, PROP_EDITABLE);
582         RNA_def_property_pointer_funcs(prop, "rna_PoseChannel_bone_group_get", "rna_PoseChannel_bone_group_set", NULL);
583         RNA_def_property_ui_text(prop, "Bone Group", "Bone Group this pose channel belongs to");
584         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
585         
586         /* transform locks */
587         prop= RNA_def_property(srna, "lock_location", PROP_BOOLEAN, PROP_XYZ);
588         RNA_def_property_boolean_sdna(prop, NULL, "protectflag", OB_LOCK_LOCX);
589         RNA_def_property_array(prop, 3);
590         RNA_def_property_ui_text(prop, "Lock Location", "Lock editing of location in the interface.");
591
592         prop= RNA_def_property(srna, "lock_rotation", PROP_BOOLEAN, PROP_XYZ);
593         RNA_def_property_boolean_sdna(prop, NULL, "protectflag", OB_LOCK_ROTX);
594         RNA_def_property_array(prop, 3);
595         RNA_def_property_ui_text(prop, "Lock Rotation", "Lock editing of rotation in the interface.");
596
597         prop= RNA_def_property(srna, "lock_scale", PROP_BOOLEAN, PROP_XYZ);
598         RNA_def_property_boolean_sdna(prop, NULL, "protectflag", OB_LOCK_SCALEX);
599         RNA_def_property_array(prop, 3);
600         RNA_def_property_ui_text(prop, "Lock Scale", "Lock editing of scale in the interface.");
601 }
602
603 static void rna_def_pose(BlenderRNA *brna)
604 {
605         StructRNA *srna;
606         PropertyRNA *prop;
607         
608         /* struct definition */
609         srna= RNA_def_struct(brna, "Pose", NULL);
610         RNA_def_struct_sdna(srna, "bPose");
611         RNA_def_struct_ui_text(srna, "Pose", "A collection of pose channels, including settings for animating bones.");
612
613         /* pose channels */
614         prop= RNA_def_property(srna, "pose_channels", PROP_COLLECTION, PROP_NONE);
615         RNA_def_property_collection_sdna(prop, NULL, "chanbase", NULL);
616         RNA_def_property_struct_type(prop, "PoseChannel");
617         RNA_def_property_ui_text(prop, "Pose Channels", "Individual pose channels for the armature.");
618
619         /* bone groups */
620         prop= RNA_def_property(srna, "bone_groups", PROP_COLLECTION, PROP_NONE);
621         RNA_def_property_collection_sdna(prop, NULL, "agroups", NULL);
622         RNA_def_property_struct_type(prop, "BoneGroup");
623         RNA_def_property_ui_text(prop, "Bone Groups", "Groups of the bones.");
624
625         prop= RNA_def_property(srna, "active_bone_group", PROP_POINTER, PROP_NONE);
626         RNA_def_property_struct_type(prop, "BoneGroup");
627         RNA_def_property_flag(prop, PROP_EDITABLE);
628         RNA_def_property_pointer_funcs(prop, "rna_Pose_active_bone_group_get", "rna_Pose_active_bone_group_set", NULL);
629         RNA_def_property_ui_text(prop, "Active Bone Group", "Active bone group for this pose.");
630         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
631
632         prop= RNA_def_property(srna, "active_bone_group_index", PROP_INT, PROP_NONE);
633         RNA_def_property_int_sdna(prop, NULL, "active_group");
634         RNA_def_property_int_funcs(prop, "rna_Pose_active_bone_group_index_get", "rna_Pose_active_bone_group_index_set", "rna_Pose_active_bone_group_index_range");
635         RNA_def_property_ui_text(prop, "Active Bone Group Index", "Active index in bone groups array.");
636         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
637 }
638
639 void RNA_def_pose(BlenderRNA *brna)
640 {
641         rna_def_pose(brna);
642         rna_def_pose_channel(brna);
643         
644         rna_def_bone_group(brna);
645 }
646
647 #endif