- added GCC warning -Wstrict-prototypes
[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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, 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 #include <string.h>
27
28 #include "RNA_define.h"
29 #include "RNA_enum_types.h"
30
31 #include "rna_internal.h"
32
33 #include "DNA_action_types.h"
34 #include "DNA_armature_types.h"
35 #include "DNA_constraint_types.h"
36 #include "DNA_object_types.h"
37 #include "DNA_scene_types.h"
38
39 #include "BLI_math.h"
40
41 #include "WM_types.h"
42
43 #ifdef RNA_RUNTIME
44
45 #include "BIK_api.h"
46 #include "BKE_action.h"
47 #include "BKE_armature.h"
48
49 #include "DNA_userdef_types.h"
50
51 #include "MEM_guardedalloc.h"
52
53 #include "BLI_ghash.h"
54
55 #include "BKE_context.h"
56 #include "BKE_constraint.h"
57 #include "BKE_depsgraph.h"
58 #include "BKE_idprop.h"
59
60 #include "ED_object.h"
61 #include "ED_armature.h"
62
63 #include "MEM_guardedalloc.h"
64
65 #include "WM_api.h"
66
67 #include "RNA_access.h"
68
69 static void rna_Pose_update(Main *bmain, Scene *scene, PointerRNA *ptr)
70 {
71         // XXX when to use this? ob->pose->flag |= (POSE_LOCKED|POSE_DO_UNLOCK);
72
73         DAG_id_flush_update(ptr->id.data, OB_RECALC_DATA);
74 }
75
76 static void rna_Pose_IK_update(Main *bmain, Scene *scene, PointerRNA *ptr)
77 {
78         // XXX when to use this? ob->pose->flag |= (POSE_LOCKED|POSE_DO_UNLOCK);
79         Object *ob= ptr->id.data;
80
81         DAG_id_flush_update(&ob->id, OB_RECALC_DATA);
82         BIK_clear_data(ob->pose);
83 }
84
85 static char *rna_PoseBone_path(PointerRNA *ptr)
86 {
87         return BLI_sprintfN("pose.bones[\"%s\"]", ((bPoseChannel*)ptr->data)->name);
88 }
89
90 static void rna_BoneGroup_color_set_set(PointerRNA *ptr, int value)
91 {
92         bActionGroup *grp= ptr->data;
93         
94         /* if valid value, set the new enum value, then copy the relevant colors? */
95         if ((value >= -1) && (value < 21))
96                 grp->customCol= value;
97         else
98                 return;
99         
100         /* only do color copying if using a custom color (i.e. not default color)  */
101         if (grp->customCol) {
102                 if (grp->customCol > 0) {
103                         /* copy theme colors on-to group's custom color in case user tries to edit color */
104                         bTheme *btheme= U.themes.first;
105                         ThemeWireColor *col_set= &btheme->tarm[(grp->customCol - 1)];
106                         
107                         memcpy(&grp->cs, col_set, sizeof(ThemeWireColor));
108                 }
109                 else {
110                         /* init custom colors with a generic multi-color rgb set, if not initialised already (for custom color set) */
111                         if (grp->cs.solid[0] == 0) {
112                                 /* define for setting colors in theme below */
113                                 #define SETCOL(col, r, g, b, a)  col[0]=r; col[1]=g; col[2]= b; col[3]= a;
114                                 
115                                 SETCOL(grp->cs.solid, 0xff, 0x00, 0x00, 255);
116                                 SETCOL(grp->cs.select, 0x81, 0xe6, 0x14, 255);
117                                 SETCOL(grp->cs.active, 0x18, 0xb6, 0xe0, 255);
118                                 
119                                 #undef SETCOL
120                         }
121                 }
122         }
123 }
124
125 static IDProperty *rna_PoseBone_idprops(PointerRNA *ptr, int create)
126 {
127         bPoseChannel *pchan= ptr->data;
128
129         if(create && !pchan->prop) {
130                 IDPropertyTemplate val = {0};
131                 pchan->prop= IDP_New(IDP_GROUP, val, "RNA_PoseBone group");
132         }
133
134         return pchan->prop;
135 }
136
137 static void rna_Pose_ik_solver_set(struct PointerRNA *ptr, int value)
138 {
139         bPose *pose= (bPose*)ptr->data;
140
141         if (pose->iksolver != value) {
142                 // the solver has changed, must clean any temporary structures
143                 BIK_clear_data(pose);
144                 if (pose->ikparam) {
145                         MEM_freeN(pose->ikparam);
146                         pose->ikparam = NULL;
147                 }
148                 pose->iksolver = value;
149                 init_pose_ikparam(pose);
150         }
151 }
152
153 static void rna_Pose_ik_solver_update(Main *bmain, Scene *scene, PointerRNA *ptr)
154 {
155         Object *ob= ptr->id.data;
156         bPose *pose = ptr->data;
157
158         pose->flag |= POSE_RECALC;      // checks & sorts pose channels
159         DAG_scene_sort(bmain, scene);
160         
161         update_pose_constraint_flags(pose);
162         
163         object_test_constraints(ob);
164
165         DAG_id_flush_update(&ob->id, OB_RECALC_DATA|OB_RECALC_OB);
166 }
167
168 /* rotation - axis-angle */
169 static void rna_PoseChannel_rotation_axis_angle_get(PointerRNA *ptr, float *value)
170 {
171         bPoseChannel *pchan= ptr->data;
172         
173         /* for now, assume that rotation mode is axis-angle */
174         value[0]= pchan->rotAngle;
175         copy_v3_v3(&value[1], pchan->rotAxis);
176 }
177
178 /* rotation - axis-angle */
179 static void rna_PoseChannel_rotation_axis_angle_set(PointerRNA *ptr, const float *value)
180 {
181         bPoseChannel *pchan= ptr->data;
182         
183         /* for now, assume that rotation mode is axis-angle */
184         pchan->rotAngle= value[0];
185         copy_v3_v3(pchan->rotAxis, (float *)&value[1]);
186         
187         // TODO: validate axis?
188 }
189
190 static void rna_PoseChannel_rotation_mode_set(PointerRNA *ptr, int value)
191 {
192         bPoseChannel *pchan= ptr->data;
193         
194         /* use API Method for conversions... */
195         BKE_rotMode_change_values(pchan->quat, pchan->eul, pchan->rotAxis, &pchan->rotAngle, pchan->rotmode, (short)value);
196         
197         /* finally, set the new rotation type */
198         pchan->rotmode= value;
199 }
200
201 static void rna_PoseChannel_name_set(PointerRNA *ptr, const char *value)
202 {
203         Object *ob= (Object*)ptr->id.data;
204         bPoseChannel *pchan= (bPoseChannel*)ptr->data;
205         char oldname[sizeof(pchan->name)], newname[sizeof(pchan->name)];
206
207         /* need to be on the stack */
208         BLI_strncpy(newname, value, sizeof(pchan->name));
209         BLI_strncpy(oldname, pchan->name, sizeof(pchan->name));
210
211         ED_armature_bone_rename(ob->data, oldname, newname);
212 }
213
214 static int rna_PoseChannel_has_ik_get(PointerRNA *ptr)
215 {
216         Object *ob= (Object*)ptr->id.data;
217         bPoseChannel *pchan= (bPoseChannel*)ptr->data;
218
219         return ED_pose_channel_in_IK_chain(ob, pchan);
220 }
221
222 StructRNA *rna_IKParam_refine(PointerRNA *ptr)
223 {
224         bIKParam *param = (bIKParam *)ptr->data;
225
226         switch (param->iksolver) {
227         case IKSOLVER_ITASC:
228                 return &RNA_Itasc;
229         default:
230                 return &RNA_IKParam;
231         }
232 }
233
234 PointerRNA rna_Pose_ikparam_get(struct PointerRNA *ptr)
235 {
236         bPose *pose= (bPose*)ptr->data;
237         return rna_pointer_inherit_refine(ptr, &RNA_IKParam, pose->ikparam);
238 }
239
240 static StructRNA *rna_Pose_ikparam_typef(PointerRNA *ptr)
241 {
242         bPose *pose= (bPose*)ptr->data;
243
244         switch (pose->iksolver) {
245         case IKSOLVER_ITASC:
246                 return &RNA_Itasc;
247         default:
248                 return &RNA_IKParam;
249         }
250 }
251
252 static void rna_Itasc_update(Main *bmain, Scene *scene, PointerRNA *ptr)
253 {
254         Object *ob = ptr->id.data;
255         bItasc *itasc = ptr->data;
256
257         /* verify values */
258         if (itasc->precision < 0.0001f)
259                 itasc->precision = 0.0001f;
260         if (itasc->minstep < 0.001f)
261                 itasc->minstep = 0.001f;
262         if (itasc->maxstep < itasc->minstep)
263                 itasc->maxstep = itasc->minstep;
264         if (itasc->feedback < 0.01f)
265                 itasc->feedback = 0.01f;
266         if (itasc->feedback > 100.f)
267                 itasc->feedback = 100.f;
268         if (itasc->maxvel < 0.01f)
269                 itasc->maxvel = 0.01f;
270         if (itasc->maxvel > 100.f)
271                 itasc->maxvel = 100.f;
272         BIK_update_param(ob->pose);
273
274         DAG_id_flush_update(&ob->id, OB_RECALC_DATA);
275 }
276
277 static void rna_Itasc_update_rebuild(Main *bmain, Scene *scene, PointerRNA *ptr)
278 {
279         Object *ob= ptr->id.data;
280         bPose *pose = ob->pose;
281
282         pose->flag |= POSE_RECALC;      // checks & sorts pose channels
283         rna_Itasc_update(bmain, scene, ptr);
284 }
285
286 static PointerRNA rna_PoseChannel_bone_group_get(PointerRNA *ptr)
287 {
288         Object *ob= (Object*)ptr->id.data;
289         bPose *pose= (ob) ? ob->pose : NULL;
290         bPoseChannel *pchan= (bPoseChannel*)ptr->data;
291         bActionGroup *grp;
292         
293         if (pose)
294                 grp= BLI_findlink(&pose->agroups, pchan->agrp_index-1);
295         else
296                 grp= NULL;
297         
298         return rna_pointer_inherit_refine(ptr, &RNA_BoneGroup, grp);
299 }
300
301 static void rna_PoseChannel_bone_group_set(PointerRNA *ptr, PointerRNA value)
302 {
303         Object *ob= (Object*)ptr->id.data;
304         bPose *pose= (ob) ? ob->pose : NULL;
305         bPoseChannel *pchan= (bPoseChannel*)ptr->data;
306         
307         if (pose)
308                 pchan->agrp_index= BLI_findindex(&pose->agroups, value.data) + 1;
309         else
310                 pchan->agrp_index= 0;
311 }
312
313 static int rna_PoseChannel_bone_group_index_get(PointerRNA *ptr)
314 {
315         bPoseChannel *pchan= (bPoseChannel*)ptr->data;
316         return MAX2(pchan->agrp_index-1, 0);
317 }
318
319 static void rna_PoseChannel_bone_group_index_set(PointerRNA *ptr, int value)
320 {
321         bPoseChannel *pchan= (bPoseChannel*)ptr->data;
322         pchan->agrp_index= value+1;
323 }
324
325 static void rna_PoseChannel_bone_group_index_range(PointerRNA *ptr, int *min, int *max)
326 {
327         Object *ob= (Object*)ptr->id.data;
328         bPose *pose= (ob) ? ob->pose : NULL;
329         
330         *min= 0;
331         
332         if (pose) {
333                 *max= BLI_countlist(&pose->agroups)-1;
334                 *max= MAX2(0, *max);
335         }
336         else
337                 *max= 0;
338 }
339
340 static PointerRNA rna_Pose_active_bone_group_get(PointerRNA *ptr)
341 {
342         bPose *pose= (bPose*)ptr->data;
343         return rna_pointer_inherit_refine(ptr, &RNA_BoneGroup, BLI_findlink(&pose->agroups, pose->active_group-1));
344 }
345
346 static void rna_Pose_active_bone_group_set(PointerRNA *ptr, PointerRNA value)
347 {
348         bPose *pose= (bPose*)ptr->data;
349         pose->active_group= BLI_findindex(&pose->agroups, value.data) + 1;
350 }
351
352 static int rna_Pose_active_bone_group_index_get(PointerRNA *ptr)
353 {
354         bPose *pose= (bPose*)ptr->data;
355         return MAX2(pose->active_group-1, 0);
356 }
357
358 static void rna_Pose_active_bone_group_index_set(PointerRNA *ptr, int value)
359 {
360         bPose *pose= (bPose*)ptr->data;
361         pose->active_group= value+1;
362 }
363
364 static void rna_Pose_active_bone_group_index_range(PointerRNA *ptr, int *min, int *max)
365 {
366         bPose *pose= (bPose*)ptr->data;
367
368         *min= 0;
369         *max= BLI_countlist(&pose->agroups)-1;
370         *max= MAX2(0, *max);
371 }
372
373 #if 0
374 static void rna_pose_bgroup_name_index_get(PointerRNA *ptr, char *value, int index)
375 {
376         bPose *pose= (bPose*)ptr->data;
377         bActionGroup *grp;
378
379         grp= BLI_findlink(&pose->agroups, index-1);
380
381         if(grp) BLI_strncpy(value, grp->name, sizeof(grp->name));
382         else BLI_strncpy(value, "", sizeof(grp->name)); // XXX if invalid pointer, won't this crash?
383 }
384
385 static int rna_pose_bgroup_name_index_length(PointerRNA *ptr, int index)
386 {
387         bPose *pose= (bPose*)ptr->data;
388         bActionGroup *grp;
389
390         grp= BLI_findlink(&pose->agroups, index-1);
391         return (grp)? strlen(grp->name): 0;
392 }
393
394 static void rna_pose_bgroup_name_index_set(PointerRNA *ptr, const char *value, short *index)
395 {
396         bPose *pose= (bPose*)ptr->data;
397         bActionGroup *grp;
398         int a;
399         
400         for (a=1, grp=pose->agroups.first; grp; grp=grp->next, a++) {
401                 if (strcmp(grp->name, value) == 0) {
402                         *index= a;
403                         return;
404                 }
405         }
406         
407         *index= 0;
408 }
409
410 static void rna_pose_pgroup_name_set(PointerRNA *ptr, const char *value, char *result, int maxlen)
411 {
412         bPose *pose= (bPose*)ptr->data;
413         bActionGroup *grp;
414         
415         for (grp= pose->agroups.first; grp; grp= grp->next) {
416                 if (strcmp(grp->name, value) == 0) {
417                         BLI_strncpy(result, value, maxlen);
418                         return;
419                 }
420         }
421         
422         BLI_strncpy(result, "", maxlen);
423 }
424 #endif
425
426 static PointerRNA rna_PoseChannel_active_constraint_get(PointerRNA *ptr)
427 {
428         bPoseChannel *pchan= (bPoseChannel*)ptr->data;
429         bConstraint *con= constraints_get_active(&pchan->constraints);
430         return rna_pointer_inherit_refine(ptr, &RNA_Constraint, con);
431 }
432
433 static void rna_PoseChannel_active_constraint_set(PointerRNA *ptr, PointerRNA value)
434 {
435         bPoseChannel *pchan= (bPoseChannel*)ptr->data;
436         constraints_set_active(&pchan->constraints, (bConstraint *)value.data);
437 }
438
439 static bConstraint *rna_PoseChannel_constraints_new(bPoseChannel *pchan, int type)
440 {
441         //WM_main_add_notifier(NC_OBJECT|ND_CONSTRAINT|NA_ADDED, object);
442         // TODO, pass object also
443         // TODO, new pose bones don't have updated draw flags
444         return add_pose_constraint(NULL, pchan, NULL, type);
445 }
446
447 static void rna_PoseChannel_constraints_remove(ID *id, bPoseChannel *pchan, ReportList *reports, bConstraint *con)
448 {
449         if(BLI_findindex(&pchan->constraints, con) == -1) {
450                 BKE_reportf(reports, RPT_ERROR, "Constraint '%s' not found in pose bone '%s'.", con->name, pchan->name);
451                 return;
452         }
453
454         // TODO
455         //ED_object_constraint_set_active(id, NULL);
456         WM_main_add_notifier(NC_OBJECT|ND_CONSTRAINT, id);
457         remove_constraint(&pchan->constraints, con);
458 }
459
460 static int rna_PoseChannel_proxy_editable(PointerRNA *ptr)
461 {
462         Object *ob= (Object*)ptr->id.data;
463         bArmature *arm= ob->data;
464         bPoseChannel *pchan= (bPoseChannel*)ptr->data;
465         
466         return (ob->proxy && pchan->bone && (pchan->bone->layer & arm->layer_protected))? 0: PROP_EDITABLE;
467 }
468
469 static int rna_PoseChannel_location_editable(PointerRNA *ptr, int index)
470 {
471         bPoseChannel *pchan= (bPoseChannel*)ptr->data;
472
473         /* only if the axis in question is locked, not editable... */
474         if ((index == 0) && (pchan->protectflag & OB_LOCK_LOCX))
475                 return 0;
476         else if ((index == 1) && (pchan->protectflag & OB_LOCK_LOCY))
477                 return 0;
478         else if ((index == 2) && (pchan->protectflag & OB_LOCK_LOCZ))
479                 return 0;
480         else
481                 return PROP_EDITABLE;
482 }
483
484 static int rna_PoseChannel_scale_editable(PointerRNA *ptr, int index)
485 {
486         bPoseChannel *pchan= (bPoseChannel*)ptr->data;
487         
488         /* only if the axis in question is locked, not editable... */
489         if ((index == 0) && (pchan->protectflag & OB_LOCK_SCALEX))
490                 return 0;
491         else if ((index == 1) && (pchan->protectflag & OB_LOCK_SCALEY))
492                 return 0;
493         else if ((index == 2) && (pchan->protectflag & OB_LOCK_SCALEZ))
494                 return 0;
495         else
496                 return PROP_EDITABLE;
497 }
498
499 static int rna_PoseChannel_rotation_euler_editable(PointerRNA *ptr, int index)
500 {
501         bPoseChannel *pchan= (bPoseChannel*)ptr->data;
502         
503         /* only if the axis in question is locked, not editable... */
504         if ((index == 0) && (pchan->protectflag & OB_LOCK_ROTX))
505                 return 0;
506         else if ((index == 1) && (pchan->protectflag & OB_LOCK_ROTY))
507                 return 0;
508         else if ((index == 2) && (pchan->protectflag & OB_LOCK_ROTZ))
509                 return 0;
510         else
511                 return PROP_EDITABLE;
512 }
513
514 static int rna_PoseChannel_rotation_4d_editable(PointerRNA *ptr, int index)
515 {
516         bPoseChannel *pchan= (bPoseChannel*)ptr->data;
517         
518         /* only consider locks if locking components individually... */
519         if (pchan->protectflag & OB_LOCK_ROT4D) {
520                 /* only if the axis in question is locked, not editable... */
521                 if ((index == 0) && (pchan->protectflag & OB_LOCK_ROTW))
522                         return 0;
523                 else if ((index == 1) && (pchan->protectflag & OB_LOCK_ROTX))
524                         return 0;
525                 else if ((index == 2) && (pchan->protectflag & OB_LOCK_ROTY))
526                         return 0;
527                 else if ((index == 3) && (pchan->protectflag & OB_LOCK_ROTZ))
528                         return 0;
529         }
530                 
531         return PROP_EDITABLE;
532 }
533
534 /* not essential, but much faster then the default lookup function */
535 PointerRNA rna_PoseBones_lookup_string(PointerRNA *ptr, const char *key)
536 {
537         PointerRNA rptr;
538         bPose *pose= (bPose*)ptr->data;
539         bPoseChannel *pchan= get_pose_channel(pose, key);
540         RNA_pointer_create(ptr->id.data, &RNA_PoseBone, pchan, &rptr);
541         return rptr;
542 }
543
544 static void rna_PoseChannel_matrix_basis_get(PointerRNA *ptr, float *values)
545 {
546         bPoseChannel *pchan= (bPoseChannel*)ptr->data;
547         pchan_to_mat4(pchan, (float (*)[4])values);
548 }
549
550 static void rna_PoseChannel_matrix_basis_set(PointerRNA *ptr, const float *values)
551 {
552         bPoseChannel *pchan= (bPoseChannel*)ptr->data;
553         pchan_apply_mat4(pchan, (float (*)[4])values, FALSE); /* no compat for pradictable result */
554 }
555
556 #else
557
558 static void rna_def_bone_group(BlenderRNA *brna)
559 {
560         static EnumPropertyItem prop_colorSets_items[] = {
561                 {0, "DEFAULT", 0, "Default Colors", ""},
562                 {1, "THEME01", 0, "01 - Theme Color Set", ""},
563                 {2, "THEME02", 0, "02 - Theme Color Set", ""},
564                 {3, "THEME03", 0, "03 - Theme Color Set", ""},
565                 {4, "THEME04", 0, "04 - Theme Color Set", ""},
566                 {5, "THEME05", 0, "05 - Theme Color Set", ""},
567                 {6, "THEME06", 0, "06 - Theme Color Set", ""},
568                 {7, "THEME07", 0, "07 - Theme Color Set", ""},
569                 {8, "THEME08", 0, "08 - Theme Color Set", ""},
570                 {9, "THEME09", 0, "09 - Theme Color Set", ""},
571                 {10, "THEME10", 0, "10 - Theme Color Set", ""},
572                 {11, "THEME11", 0, "11 - Theme Color Set", ""},
573                 {12, "THEME12", 0, "12 - Theme Color Set", ""},
574                 {13, "THEME13", 0, "13 - Theme Color Set", ""},
575                 {14, "THEME14", 0, "14 - Theme Color Set", ""},
576                 {15, "THEME15", 0, "15 - Theme Color Set", ""},
577                 {16, "THEME16", 0, "16 - Theme Color Set", ""},
578                 {17, "THEME17", 0, "17 - Theme Color Set", ""},
579                 {18, "THEME18", 0, "18 - Theme Color Set", ""},
580                 {19, "THEME19", 0, "19 - Theme Color Set", ""},
581                 {20, "THEME20", 0, "20 - Theme Color Set", ""},
582                 {-1, "CUSTOM", 0, "Custom Color Set", ""},
583                 {0, NULL, 0, NULL, NULL}};
584         
585         StructRNA *srna;
586         PropertyRNA *prop;
587         
588         /* struct */
589         srna= RNA_def_struct(brna, "BoneGroup", NULL);
590         RNA_def_struct_sdna(srna, "bActionGroup");
591         RNA_def_struct_ui_text(srna, "Bone Group", "Groups of Pose Channels (Bones)");
592         RNA_def_struct_ui_icon(srna, ICON_GROUP_BONE);
593         
594         /* name */
595         prop= RNA_def_property(srna, "name", PROP_STRING, PROP_NONE);
596         RNA_def_property_ui_text(prop, "Name", "");
597         RNA_def_struct_name_property(srna, prop);
598         
599         // TODO: add some runtime-collections stuff to access grouped bones 
600         
601         /* color set + colors */
602         prop= RNA_def_property(srna, "color_set", PROP_ENUM, PROP_NONE);
603         RNA_def_property_enum_sdna(prop, NULL, "customCol");
604         RNA_def_property_enum_items(prop, prop_colorSets_items);
605         RNA_def_property_enum_funcs(prop, NULL, "rna_BoneGroup_color_set_set", NULL);
606         RNA_def_property_ui_text(prop, "Color Set", "Custom color set to use");
607         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
608         
609                 // TODO: editing the colors for this should result in changes to the color type...
610         prop= RNA_def_property(srna, "colors", PROP_POINTER, PROP_NONE);
611         RNA_def_property_flag(prop, PROP_NEVER_NULL);
612         RNA_def_property_struct_type(prop, "ThemeBoneColorSet");
613         RNA_def_property_pointer_sdna(prop, NULL, "cs"); /* NOTE: the DNA data is not really a pointer, but this code works :) */
614         RNA_def_property_ui_text(prop, "Colors", "Copy of the colors associated with the group's color set");
615         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
616 }
617
618 static EnumPropertyItem prop_iksolver_items[] = {
619         {IKSOLVER_LEGACY, "LEGACY", 0, "Legacy", "Original IK solver"},
620         {IKSOLVER_ITASC, "ITASC", 0, "iTaSC", "Multi constraint, stateful IK solver"},
621         {0, NULL, 0, NULL, NULL}};
622
623 static EnumPropertyItem prop_solver_items[] = {
624         {ITASC_SOLVER_SDLS, "SDLS", 0, "SDLS", "Selective Damped Least Square"},
625         {ITASC_SOLVER_DLS, "DLS", 0, "DLS", "Damped Least Square with Numerical Filtering"},
626         {0, NULL, 0, NULL, NULL}};
627
628
629 static void rna_def_pose_channel_constraints(BlenderRNA *brna, PropertyRNA *cprop)
630 {
631         StructRNA *srna;
632         PropertyRNA *prop;
633
634         FunctionRNA *func;
635         PropertyRNA *parm;
636
637         RNA_def_property_srna(cprop, "PoseBoneConstraints");
638         srna= RNA_def_struct(brna, "PoseBoneConstraints", NULL);
639         RNA_def_struct_sdna(srna, "bPoseChannel");
640         RNA_def_struct_ui_text(srna, "PoseBone Constraints", "Collection of pose bone constraints");
641
642         /* Collection active property */
643         prop= RNA_def_property(srna, "active", PROP_POINTER, PROP_NONE);
644         RNA_def_property_struct_type(prop, "Constraint");
645         RNA_def_property_pointer_funcs(prop, "rna_PoseChannel_active_constraint_get", "rna_PoseChannel_active_constraint_set", NULL, NULL);
646         RNA_def_property_flag(prop, PROP_EDITABLE);
647         RNA_def_property_ui_text(prop, "Active Constraint", "Active PoseChannel constraint");
648
649
650         /* Constraint collection */
651         func= RNA_def_function(srna, "new", "rna_PoseChannel_constraints_new");
652         RNA_def_function_ui_description(func, "Add a constraint to this object");
653         /* return type */
654         parm= RNA_def_pointer(func, "constraint", "Constraint", "", "New constraint.");
655         RNA_def_function_return(func, parm);
656         /* constraint to add */
657         parm= RNA_def_enum(func, "type", constraint_type_items, 1, "", "Constraint type to add.");
658         RNA_def_property_flag(parm, PROP_REQUIRED);
659
660         func= RNA_def_function(srna, "remove", "rna_PoseChannel_constraints_remove");
661         RNA_def_function_ui_description(func, "Remove a constraint from this object.");
662         RNA_def_function_flag(func, FUNC_USE_REPORTS|FUNC_USE_SELF_ID); /* ID needed for refresh */
663         /* constraint to remove */
664         parm= RNA_def_pointer(func, "constraint", "Constraint", "", "Removed constraint.");
665         RNA_def_property_flag(parm, PROP_REQUIRED|PROP_NEVER_NULL);
666 }
667
668 static void rna_def_pose_channel(BlenderRNA *brna)
669 {
670         // XXX: this RNA enum define is currently duplicated for objects, since there is some text here which is not applicable
671         static EnumPropertyItem prop_rotmode_items[] = {
672                 {ROT_MODE_QUAT, "QUATERNION", 0, "Quaternion (WXYZ)", "No Gimbal Lock (default)"},
673                 {ROT_MODE_XYZ, "XYZ", 0, "XYZ Euler", "XYZ Rotation Order. Prone to Gimbal Lock"},
674                 {ROT_MODE_XZY, "XZY", 0, "XZY Euler", "XZY Rotation Order. Prone to Gimbal Lock"},
675                 {ROT_MODE_YXZ, "YXZ", 0, "YXZ Euler", "YXZ Rotation Order. Prone to Gimbal Lock"},
676                 {ROT_MODE_YZX, "YZX", 0, "YZX Euler", "YZX Rotation Order. Prone to Gimbal Lock"},
677                 {ROT_MODE_ZXY, "ZXY", 0, "ZXY Euler", "ZXY Rotation Order. Prone to Gimbal Lock"},
678                 {ROT_MODE_ZYX, "ZYX", 0, "ZYX Euler", "ZYX Rotation Order. Prone to Gimbal Lock"},
679                 {ROT_MODE_AXISANGLE, "AXIS_ANGLE", 0, "Axis Angle", "Axis Angle (W+XYZ). Defines a rotation around some axis defined by 3D-Vector"},
680                 {0, NULL, 0, NULL, NULL}};
681                 
682         static float default_quat[4] = {1,0,0,0};       /* default quaternion values */
683         static float default_axisAngle[4] = {0,0,1,0};  /* default axis-angle rotation values */
684         static float default_scale[3] = {1,1,1}; /* default scale values */
685         
686         const int matrix_dimsize[]= {4, 4};
687         
688         StructRNA *srna;
689         PropertyRNA *prop;
690
691         srna= RNA_def_struct(brna, "PoseBone", NULL);
692         RNA_def_struct_sdna(srna, "bPoseChannel");
693         RNA_def_struct_ui_text(srna, "Pose Bone", "Channel defining pose data for a bone in a Pose");
694         RNA_def_struct_path_func(srna, "rna_PoseBone_path");
695         RNA_def_struct_idprops_func(srna, "rna_PoseBone_idprops");
696         
697         /* Bone Constraints */
698         prop= RNA_def_property(srna, "constraints", PROP_COLLECTION, PROP_NONE);
699         RNA_def_property_struct_type(prop, "Constraint");
700         RNA_def_property_ui_text(prop, "Constraints", "Constraints that act on this PoseChannel");
701
702         rna_def_pose_channel_constraints(brna, prop);
703
704         /* Name + Selection Status */
705         prop= RNA_def_property(srna, "name", PROP_STRING, PROP_NONE);
706         RNA_def_property_string_funcs(prop, NULL, NULL, "rna_PoseChannel_name_set");
707         RNA_def_property_ui_text(prop, "Name", "");
708         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
709         RNA_def_struct_name_property(srna, prop);
710
711         prop= RNA_def_property(srna, "select", PROP_BOOLEAN, PROP_NONE);
712         RNA_def_property_boolean_sdna(prop, NULL, "selectflag", BONE_SELECTED);
713         RNA_def_property_ui_text(prop, "Select", "");
714
715         /* Baked Bone Path cache data */
716         rna_def_motionpath_common(srna);
717         
718         /* Relationships to other bones */
719         prop= RNA_def_property(srna, "bone", PROP_POINTER, PROP_NONE);
720         RNA_def_property_flag(prop, PROP_NEVER_NULL);
721         RNA_def_property_struct_type(prop, "Bone");
722         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
723         RNA_def_property_ui_text(prop, "Bone", "Bone associated with this PoseBone");
724
725         prop= RNA_def_property(srna, "parent", PROP_POINTER, PROP_NONE);
726         RNA_def_property_struct_type(prop, "PoseBone");
727         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
728         RNA_def_property_ui_text(prop, "Parent", "Parent of this pose bone");
729
730         prop= RNA_def_property(srna, "child", PROP_POINTER, PROP_NONE);
731         RNA_def_property_struct_type(prop, "PoseBone");
732         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
733         RNA_def_property_ui_text(prop, "Child", "Child of this pose bone");
734         
735         /* Transformation settings */
736         prop= RNA_def_property(srna, "location", PROP_FLOAT, PROP_TRANSLATION);
737         RNA_def_property_float_sdna(prop, NULL, "loc");
738         RNA_def_property_editable_array_func(prop, "rna_PoseChannel_location_editable");
739         RNA_def_property_ui_text(prop, "Location", "");
740         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable"); // XXX... disabled, since proxy-locked layers are currently used for ensuring proxy-syncing too
741         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
742
743         prop= RNA_def_property(srna, "scale", PROP_FLOAT, PROP_XYZ);
744         RNA_def_property_float_sdna(prop, NULL, "size");
745         RNA_def_property_editable_array_func(prop, "rna_PoseChannel_scale_editable");
746         RNA_def_property_float_array_default(prop, default_scale);
747         RNA_def_property_ui_text(prop, "Scale", "");
748         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable"); // XXX... disabled, since proxy-locked layers are currently used for ensuring proxy-syncing too
749         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
750
751         prop= RNA_def_property(srna, "rotation_quaternion", PROP_FLOAT, PROP_QUATERNION);
752         RNA_def_property_float_sdna(prop, NULL, "quat");
753         RNA_def_property_editable_array_func(prop, "rna_PoseChannel_rotation_4d_editable");
754         RNA_def_property_float_array_default(prop, default_quat);
755         RNA_def_property_ui_text(prop, "Quaternion Rotation", "Rotation in Quaternions");
756         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable"); // XXX... disabled, since proxy-locked layers are currently used for ensuring proxy-syncing too
757         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
758         
759                 /* XXX: for axis-angle, it would have been nice to have 2 separate fields for UI purposes, but
760                  * having a single one is better for Keyframing and other property-management situations...
761                  */
762         prop= RNA_def_property(srna, "rotation_axis_angle", PROP_FLOAT, PROP_AXISANGLE);
763         RNA_def_property_array(prop, 4); // TODO: maybe we'll need to define the 'default value' getter too...
764         RNA_def_property_float_funcs(prop, "rna_PoseChannel_rotation_axis_angle_get", "rna_PoseChannel_rotation_axis_angle_set", NULL);
765         RNA_def_property_editable_array_func(prop, "rna_PoseChannel_rotation_4d_editable");
766         RNA_def_property_float_array_default(prop, default_axisAngle);
767         RNA_def_property_ui_text(prop, "Axis-Angle Rotation", "Angle of Rotation for Axis-Angle rotation representation");
768         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable"); // XXX... disabled, since proxy-locked layers are currently used for ensuring proxy-syncing too
769         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
770         
771         prop= RNA_def_property(srna, "rotation_euler", PROP_FLOAT, PROP_EULER);
772         RNA_def_property_float_sdna(prop, NULL, "eul");
773         RNA_def_property_editable_array_func(prop, "rna_PoseChannel_rotation_euler_editable");
774         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable"); // XXX... disabled, since proxy-locked layers are currently used for ensuring proxy-syncing too
775         RNA_def_property_ui_text(prop, "Euler Rotation", "Rotation in Eulers");
776         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
777         
778         prop= RNA_def_property(srna, "rotation_mode", PROP_ENUM, PROP_NONE);
779         RNA_def_property_enum_sdna(prop, NULL, "rotmode");
780         RNA_def_property_enum_items(prop, prop_rotmode_items); // XXX move to using a single define of this someday
781         RNA_def_property_enum_funcs(prop, NULL, "rna_PoseChannel_rotation_mode_set", NULL);
782         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable"); // XXX... disabled, since proxy-locked layers are currently used for ensuring proxy-syncing too
783         RNA_def_property_ui_text(prop, "Rotation Mode", "");
784         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
785         
786         /* transform matrices - should be read-only since these are set directly by AnimSys evaluation */
787         prop= RNA_def_property(srna, "matrix_channel", PROP_FLOAT, PROP_MATRIX);
788         RNA_def_property_float_sdna(prop, NULL, "chan_mat");
789         RNA_def_property_multi_array(prop, 2, matrix_dimsize);
790         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
791         RNA_def_property_ui_text(prop, "Channel Matrix", "4x4 matrix, before constraints");
792
793         /* writable because it touches loc/scale/rot directly */
794         prop= RNA_def_property(srna, "matrix_basis", PROP_FLOAT, PROP_MATRIX);
795         RNA_def_property_multi_array(prop, 2, matrix_dimsize);
796         RNA_def_property_ui_text(prop, "Basis Matrix", "Provides an alternative access to loc/scale/rotation relative to the parent and own rest bone.");
797         RNA_def_property_float_funcs(prop, "rna_PoseChannel_matrix_basis_get", "rna_PoseChannel_matrix_basis_set", NULL);
798         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
799         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
800
801         /* final matrix */
802         prop= RNA_def_property(srna, "matrix", PROP_FLOAT, PROP_MATRIX);
803         RNA_def_property_float_sdna(prop, NULL, "pose_mat");
804         RNA_def_property_multi_array(prop, 2, matrix_dimsize);
805         RNA_def_property_clear_flag(prop, PROP_EDITABLE); 
806         RNA_def_property_ui_text(prop, "Pose Matrix", "Final 4x4 matrix for this channel");
807
808         /* Head/Tail Coordinates (in Pose Space) - Automatically calculated... */
809         prop= RNA_def_property(srna, "head", PROP_FLOAT, PROP_TRANSLATION);
810         RNA_def_property_float_sdna(prop, NULL, "pose_head");
811         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
812         RNA_def_property_ui_text(prop, "Pose Head Position", "Location of head of the channel's bone");
813
814         prop= RNA_def_property(srna, "tail", PROP_FLOAT, PROP_TRANSLATION);
815         RNA_def_property_float_sdna(prop, NULL, "pose_tail");
816         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
817         RNA_def_property_ui_text(prop, "Pose Tail Position", "Location of tail of the channel's bone");
818         
819         /* IK Settings */
820         prop= RNA_def_property(srna, "is_in_ik_chain", PROP_BOOLEAN, PROP_NONE);
821         RNA_def_property_boolean_funcs(prop,  "rna_PoseChannel_has_ik_get", NULL);
822         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
823         RNA_def_property_ui_text(prop, "Has IK", "Is part of an IK chain");
824         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_IK_update");
825
826         prop= RNA_def_property(srna, "lock_ik_x", PROP_BOOLEAN, PROP_NONE);
827         RNA_def_property_boolean_sdna(prop, NULL, "ikflag", BONE_IK_NO_XDOF);
828         RNA_def_property_ui_text(prop, "IK X Lock", "Disallow movement around the X axis");
829         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
830         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_IK_update");
831
832         prop= RNA_def_property(srna, "lock_ik_y", PROP_BOOLEAN, PROP_NONE);
833         RNA_def_property_boolean_sdna(prop, NULL, "ikflag", BONE_IK_NO_YDOF);
834         RNA_def_property_ui_text(prop, "IK Y Lock", "Disallow movement around the Y axis");
835         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
836         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_IK_update");
837
838         prop= RNA_def_property(srna, "lock_ik_z", PROP_BOOLEAN, PROP_NONE);
839         RNA_def_property_boolean_sdna(prop, NULL, "ikflag", BONE_IK_NO_ZDOF);
840         RNA_def_property_ui_text(prop, "IK Z Lock", "Disallow movement around the Z axis");
841         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
842         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_IK_update");
843
844         prop= RNA_def_property(srna, "use_ik_limit_x", PROP_BOOLEAN, PROP_NONE);
845         RNA_def_property_boolean_sdna(prop, NULL, "ikflag", BONE_IK_XLIMIT);
846         RNA_def_property_ui_text(prop, "IK X Limit", "Limit movement around the X axis");
847         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
848         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_IK_update");
849
850         prop= RNA_def_property(srna, "use_ik_limit_y", PROP_BOOLEAN, PROP_NONE);
851         RNA_def_property_boolean_sdna(prop, NULL, "ikflag", BONE_IK_YLIMIT);
852         RNA_def_property_ui_text(prop, "IK Y Limit", "Limit movement around the Y axis");
853         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
854         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_IK_update");
855
856         prop= RNA_def_property(srna, "use_ik_limit_z", PROP_BOOLEAN, PROP_NONE);
857         RNA_def_property_boolean_sdna(prop, NULL, "ikflag", BONE_IK_ZLIMIT);
858         RNA_def_property_ui_text(prop, "IK Z Limit", "Limit movement around the Z axis");
859         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
860         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_IK_update");
861         
862         prop= RNA_def_property(srna, "use_ik_rotation_control", PROP_BOOLEAN, PROP_NONE);
863         RNA_def_property_boolean_sdna(prop, NULL, "ikflag", BONE_IK_ROTCTL);
864         RNA_def_property_ui_text(prop, "IK rot control", "Apply channel rotation as IK constraint");
865         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
866         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_IK_update");
867         
868         prop= RNA_def_property(srna, "use_ik_linear_control", PROP_BOOLEAN, PROP_NONE);
869         RNA_def_property_boolean_sdna(prop, NULL, "ikflag", BONE_IK_LINCTL);
870         RNA_def_property_ui_text(prop, "IK rot control", "Apply channel size as IK constraint if stretching is enabled");
871         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
872         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_IK_update");
873         
874         prop= RNA_def_property(srna, "ik_min_x", PROP_FLOAT, PROP_ANGLE);
875         RNA_def_property_float_sdna(prop, NULL, "limitmin[0]");
876         RNA_def_property_range(prop, -M_PI, 0.0f);
877         RNA_def_property_ui_text(prop, "IK X Minimum", "Minimum angles for IK Limit");
878         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
879         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_IK_update");
880
881         prop= RNA_def_property(srna, "ik_max_x", PROP_FLOAT, PROP_ANGLE);
882         RNA_def_property_float_sdna(prop, NULL, "limitmax[0]");
883         RNA_def_property_range(prop, 0.0f, M_PI);
884         RNA_def_property_ui_text(prop, "IK X Maximum", "Maximum angles for IK Limit");
885         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
886         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_IK_update");
887
888         prop= RNA_def_property(srna, "ik_min_y", PROP_FLOAT, PROP_ANGLE);
889         RNA_def_property_float_sdna(prop, NULL, "limitmin[1]");
890         RNA_def_property_range(prop, -M_PI, 0.0f);
891         RNA_def_property_ui_text(prop, "IK Y Minimum", "Minimum angles for IK Limit");
892         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
893         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_IK_update");
894
895         prop= RNA_def_property(srna, "ik_max_y", PROP_FLOAT, PROP_ANGLE);
896         RNA_def_property_float_sdna(prop, NULL, "limitmax[1]");
897         RNA_def_property_range(prop, 0.0f, M_PI);
898         RNA_def_property_ui_text(prop, "IK Y Maximum", "Maximum angles for IK Limit");
899         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
900         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_IK_update");
901
902         prop= RNA_def_property(srna, "ik_min_z", PROP_FLOAT, PROP_ANGLE);
903         RNA_def_property_float_sdna(prop, NULL, "limitmin[2]");
904         RNA_def_property_range(prop, -M_PI, 0.0f);
905         RNA_def_property_ui_text(prop, "IK Z Minimum", "Minimum angles for IK Limit");
906         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
907         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_IK_update");
908
909         prop= RNA_def_property(srna, "ik_max_z", PROP_FLOAT, PROP_ANGLE);
910         RNA_def_property_float_sdna(prop, NULL, "limitmax[2]");
911         RNA_def_property_range(prop, 0.0f, M_PI);
912         RNA_def_property_ui_text(prop, "IK Z Maximum", "Maximum angles for IK Limit");
913         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
914         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_IK_update");
915
916         prop= RNA_def_property(srna, "ik_stiffness_x", PROP_FLOAT, PROP_NONE);
917         RNA_def_property_float_sdna(prop, NULL, "stiffness[0]");
918         RNA_def_property_range(prop, 0.0f, 0.99f);
919         RNA_def_property_ui_text(prop, "IK X Stiffness", "IK stiffness around the X axis");
920         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
921         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_IK_update");
922
923         prop= RNA_def_property(srna, "ik_stiffness_y", PROP_FLOAT, PROP_NONE);
924         RNA_def_property_float_sdna(prop, NULL, "stiffness[1]");
925         RNA_def_property_range(prop, 0.0f, 0.99f);
926         RNA_def_property_ui_text(prop, "IK Y Stiffness", "IK stiffness around the Y axis");
927         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
928         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_IK_update");
929
930         prop= RNA_def_property(srna, "ik_stiffness_z", PROP_FLOAT, PROP_NONE);
931         RNA_def_property_float_sdna(prop, NULL, "stiffness[2]");
932         RNA_def_property_range(prop, 0.0f, 0.99f);
933         RNA_def_property_ui_text(prop, "IK Z Stiffness", "IK stiffness around the Z axis");
934         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
935         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_IK_update");
936
937         prop= RNA_def_property(srna, "ik_stretch", PROP_FLOAT, PROP_NONE);
938         RNA_def_property_float_sdna(prop, NULL, "ikstretch");
939         RNA_def_property_range(prop, 0.0f,1.0f);
940         RNA_def_property_ui_text(prop, "IK Stretch", "Allow scaling of the bone for IK");
941         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
942         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_IK_update");
943         
944         prop= RNA_def_property(srna, "ik_rotation_weight", PROP_FLOAT, PROP_NONE);
945         RNA_def_property_float_sdna(prop, NULL, "ikrotweight");
946         RNA_def_property_range(prop, 0.0f,1.0f);
947         RNA_def_property_ui_text(prop, "IK Rot Weight", "Weight of rotation constraint for IK");
948         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
949         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
950         
951         prop= RNA_def_property(srna, "ik_linear_weight", PROP_FLOAT, PROP_NONE);
952         RNA_def_property_float_sdna(prop, NULL, "iklinweight");
953         RNA_def_property_range(prop, 0.0f,1.0f);
954         RNA_def_property_ui_text(prop, "IK Lin Weight", "Weight of scale constraint for IK");
955         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
956         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
957         
958         /* custom bone shapes */
959         prop= RNA_def_property(srna, "custom_shape", PROP_POINTER, PROP_NONE);
960         RNA_def_property_pointer_sdna(prop, NULL, "custom");
961         RNA_def_property_struct_type(prop, "Object");
962         RNA_def_property_flag(prop, PROP_EDITABLE);
963         RNA_def_property_ui_text(prop, "Custom Object", "Object that defines custom draw type for this bone");
964         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
965         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
966         
967         prop= RNA_def_property(srna, "custom_shape_transform", PROP_POINTER, PROP_NONE);
968         RNA_def_property_pointer_sdna(prop, NULL, "custom_tx");
969         RNA_def_property_struct_type(prop, "PoseBone");
970         RNA_def_property_flag(prop, PROP_EDITABLE);
971         RNA_def_property_ui_text(prop, "Custom Shape Transform", "Bone that defines the display transform of this custom shape");
972         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
973         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
974         
975         /* bone groups */
976         prop= RNA_def_property(srna, "bone_group_index", PROP_INT, PROP_NONE);
977         RNA_def_property_int_sdna(prop, NULL, "agrp_index");
978         RNA_def_property_flag(prop, PROP_EDITABLE);
979         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
980         RNA_def_property_int_funcs(prop, "rna_PoseChannel_bone_group_index_get", "rna_PoseChannel_bone_group_index_set", "rna_PoseChannel_bone_group_index_range");
981         RNA_def_property_ui_text(prop, "Bone Group Index", "Bone Group this pose channel belongs to (0=no group)");
982         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
983         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
984         
985         prop= RNA_def_property(srna, "bone_group", PROP_POINTER, PROP_NONE);
986         RNA_def_property_struct_type(prop, "BoneGroup");
987         RNA_def_property_flag(prop, PROP_EDITABLE);
988         RNA_def_property_pointer_funcs(prop, "rna_PoseChannel_bone_group_get", "rna_PoseChannel_bone_group_set", NULL, NULL);
989         RNA_def_property_ui_text(prop, "Bone Group", "Bone Group this pose channel belongs to");
990         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
991         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
992         
993         /* transform locks */
994         prop= RNA_def_property(srna, "lock_location", PROP_BOOLEAN, PROP_XYZ);
995         RNA_def_property_boolean_sdna(prop, NULL, "protectflag", OB_LOCK_LOCX);
996         RNA_def_property_array(prop, 3);
997         RNA_def_property_ui_text(prop, "Lock Location", "Lock editing of location in the interface");
998         RNA_def_property_ui_icon(prop, ICON_UNLOCKED, 1);
999         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
1000         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
1001
1002         prop= RNA_def_property(srna, "lock_rotation", PROP_BOOLEAN, PROP_XYZ);
1003         RNA_def_property_boolean_sdna(prop, NULL, "protectflag", OB_LOCK_ROTX);
1004         RNA_def_property_array(prop, 3);
1005         RNA_def_property_ui_text(prop, "Lock Rotation", "Lock editing of rotation in the interface");
1006         RNA_def_property_ui_icon(prop, ICON_UNLOCKED, 1);
1007         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
1008         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
1009         
1010                 // XXX this is sub-optimal - it really should be included above, but due to technical reasons we can't do this!
1011         prop= RNA_def_property(srna, "lock_rotation_w", PROP_BOOLEAN, PROP_NONE);
1012         RNA_def_property_boolean_sdna(prop, NULL, "protectflag", OB_LOCK_ROTW);
1013         RNA_def_property_ui_text(prop, "Lock Rotation (4D Angle)", "Lock editing of 'angle' component of four-component rotations in the interface");
1014         RNA_def_property_ui_icon(prop, ICON_UNLOCKED, 1);
1015         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
1016         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
1017
1018                 // XXX this needs a better name
1019         prop= RNA_def_property(srna, "lock_rotations_4d", PROP_BOOLEAN, PROP_NONE);
1020         RNA_def_property_boolean_sdna(prop, NULL, "protectflag", OB_LOCK_ROT4D);
1021         RNA_def_property_ui_text(prop, "Lock Rotations (4D)", "Lock editing of four component rotations by components (instead of as Eulers)");
1022         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
1023         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
1024
1025         prop= RNA_def_property(srna, "lock_scale", PROP_BOOLEAN, PROP_XYZ);
1026         RNA_def_property_boolean_sdna(prop, NULL, "protectflag", OB_LOCK_SCALEX);
1027         RNA_def_property_array(prop, 3);
1028         RNA_def_property_ui_text(prop, "Lock Scale", "Lock editing of scale in the interface");
1029         RNA_def_property_ui_icon(prop, ICON_UNLOCKED, 1);
1030         RNA_def_property_editable_func(prop, "rna_PoseChannel_proxy_editable");
1031         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
1032
1033         RNA_api_pose_channel(srna);
1034 }
1035
1036 static void rna_def_pose_itasc(BlenderRNA *brna)
1037 {
1038         static const EnumPropertyItem prop_itasc_mode_items[]= {
1039                 {0, "ANIMATION", 0, "Animation", "Stateless solver computing pose starting from current action and non-IK constraints"},
1040                 {ITASC_SIMULATION, "SIMULATION", 0, "Simulation", "Statefull solver running in real-time context and ignoring actions and non-IK constraints"},
1041                 {0, NULL, 0, NULL, NULL}};
1042         static const EnumPropertyItem prop_itasc_reiteration_items[]= {
1043                 {0, "NEVER", 0, "Never", "The solver does not reiterate, not even on first frame (starts from rest pose)"},
1044                 {ITASC_INITIAL_REITERATION, "INITIAL", 0, "Initial", "The solver reiterates (converges) on the first frame but not on subsequent frame"},
1045                 {ITASC_INITIAL_REITERATION|ITASC_REITERATION, "ALWAYS", 0, "Always", "The solver reiterates (converges) on all frames"},
1046                 {0, NULL, 0, NULL, NULL}};
1047
1048         StructRNA *srna;
1049         PropertyRNA *prop;
1050
1051         srna= RNA_def_struct(brna, "Itasc", "IKParam");
1052         RNA_def_struct_sdna(srna, "bItasc");
1053         RNA_def_struct_ui_text(srna, "bItasc", "Parameters for the iTaSC IK solver");
1054
1055         prop= RNA_def_property(srna, "precision", PROP_FLOAT, PROP_NONE);
1056         RNA_def_property_float_sdna(prop, NULL, "precision");
1057         RNA_def_property_range(prop, 0.0f,0.1f);
1058         RNA_def_property_ui_text(prop, "Precision", "Precision of convergence in case of reiteration");
1059         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Itasc_update");
1060
1061         prop= RNA_def_property(srna, "iterations", PROP_INT, PROP_NONE);
1062         RNA_def_property_int_sdna(prop, NULL, "numiter");
1063         RNA_def_property_range(prop, 1.f,1000.f);
1064         RNA_def_property_ui_text(prop, "Iterations", "Maximum number of iterations for convergence in case of reiteration");
1065         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Itasc_update");
1066
1067         prop= RNA_def_property(srna, "step_count", PROP_INT, PROP_NONE);
1068         RNA_def_property_int_sdna(prop, NULL, "numstep");
1069         RNA_def_property_range(prop, 1.f, 50.f);
1070         RNA_def_property_ui_text(prop, "Num steps", "Divides the frame interval into this many steps");
1071         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Itasc_update");
1072
1073         prop= RNA_def_property(srna, "mode", PROP_ENUM, PROP_NONE);
1074         RNA_def_property_enum_bitflag_sdna(prop, NULL, "flag");
1075         RNA_def_property_enum_items(prop, prop_itasc_mode_items);
1076         RNA_def_property_ui_text(prop, "Mode", NULL);
1077         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Itasc_update_rebuild");
1078
1079         prop= RNA_def_property(srna, "reiteration_method", PROP_ENUM, PROP_NONE);
1080         RNA_def_property_enum_bitflag_sdna(prop, NULL, "flag");
1081         RNA_def_property_enum_items(prop, prop_itasc_reiteration_items);
1082         RNA_def_property_ui_text(prop, "Reiteration", "Defines if the solver is allowed to reiterate (converges until precision is met) on none, first or all frames");
1083         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Itasc_update");
1084
1085         prop= RNA_def_property(srna, "use_auto_step", PROP_BOOLEAN, PROP_NONE);
1086         RNA_def_property_boolean_sdna(prop, NULL, "flag", ITASC_AUTO_STEP);
1087         RNA_def_property_ui_text(prop, "Auto step", "Automatically determine the optimal number of steps for best performance/accuracy trade off");
1088         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Itasc_update");
1089
1090         prop= RNA_def_property(srna, "step_min", PROP_FLOAT, PROP_NONE);
1091         RNA_def_property_float_sdna(prop, NULL, "minstep");
1092         RNA_def_property_range(prop, 0.0f,0.1f);
1093         RNA_def_property_ui_text(prop, "Min step", "Lower bound for timestep in second in case of automatic substeps");
1094         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Itasc_update");
1095
1096         prop= RNA_def_property(srna, "step_max", PROP_FLOAT, PROP_NONE);
1097         RNA_def_property_float_sdna(prop, NULL, "maxstep");
1098         RNA_def_property_range(prop, 0.0f,1.0f);
1099         RNA_def_property_ui_text(prop, "Max step", "Higher bound for timestep in second in case of automatic substeps");
1100         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Itasc_update");
1101
1102         prop= RNA_def_property(srna, "feedback", PROP_FLOAT, PROP_NONE);
1103         RNA_def_property_float_sdna(prop, NULL, "feedback");
1104         RNA_def_property_range(prop, 0.0f,100.0f);
1105         RNA_def_property_ui_text(prop, "Feedback", "Feedback coefficient for error correction. Average response time=1/feedback. Default=20");
1106         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Itasc_update");
1107
1108         prop= RNA_def_property(srna, "velocity_max", PROP_FLOAT, PROP_NONE);
1109         RNA_def_property_float_sdna(prop, NULL, "maxvel");
1110         RNA_def_property_range(prop, 0.0f,100.0f);
1111         RNA_def_property_ui_text(prop, "Max Velocity", "Maximum joint velocity in rad/s. Default=50");
1112         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Itasc_update");
1113
1114         prop= RNA_def_property(srna, "solver", PROP_ENUM, PROP_NONE);
1115         RNA_def_property_enum_sdna(prop, NULL, "solver");
1116         RNA_def_property_enum_items(prop, prop_solver_items);
1117         RNA_def_property_ui_text(prop, "Solver", "Solving method selection: Automatic damping or manual damping");
1118         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Itasc_update_rebuild");
1119
1120         prop= RNA_def_property(srna, "damping_max", PROP_FLOAT, PROP_NONE);
1121         RNA_def_property_float_sdna(prop, NULL, "dampmax");
1122         RNA_def_property_range(prop, 0.0f,1.0f);
1123         RNA_def_property_ui_text(prop, "Damp", "Maximum damping coefficient when singular value is nearly 0. Higher values=more stability, less reactivity. Default=0.5");
1124         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Itasc_update");
1125
1126         prop= RNA_def_property(srna, "damping_epsilon", PROP_FLOAT, PROP_NONE);
1127         RNA_def_property_float_sdna(prop, NULL, "dampeps");
1128         RNA_def_property_range(prop, 0.0f,1.0f);
1129         RNA_def_property_ui_text(prop, "Epsilon", "Singular value under which damping is progressively applied. Higher values=more stability, less reactivity. Default=0.1");
1130         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Itasc_update");
1131 }
1132
1133 static void rna_def_pose_ikparam(BlenderRNA *brna)
1134 {
1135         StructRNA *srna;
1136         PropertyRNA *prop;
1137
1138         srna= RNA_def_struct(brna, "IKParam", NULL);
1139         RNA_def_struct_sdna(srna, "bIKParam");
1140         RNA_def_struct_ui_text(srna, "IKParam", "Base type for IK solver parameters");
1141         RNA_def_struct_refine_func(srna, "rna_IKParam_refine");
1142
1143         prop= RNA_def_property(srna, "ik_solver", PROP_ENUM, PROP_NONE);
1144         RNA_def_property_enum_sdna(prop, NULL, "iksolver");
1145         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
1146         RNA_def_property_enum_items(prop, prop_iksolver_items);
1147         RNA_def_property_ui_text(prop, "IK Solver", "IK solver for which these parameters are defined, 0 for Legacy, 1 for iTaSC");
1148 }
1149
1150 /* pose.bone_groups */
1151 static void rna_def_bone_groups(BlenderRNA *brna, PropertyRNA *cprop)
1152 {
1153         StructRNA *srna;
1154         PropertyRNA *prop;
1155
1156 //      FunctionRNA *func;
1157 //      PropertyRNA *parm;
1158
1159         RNA_def_property_srna(cprop, "BoneGroups");
1160         srna= RNA_def_struct(brna, "BoneGroups", NULL);
1161         RNA_def_struct_sdna(srna, "bPose");
1162         RNA_def_struct_ui_text(srna, "Bone Groups", "Collection of bone groups");
1163
1164         prop= RNA_def_property(srna, "active", PROP_POINTER, PROP_NONE);
1165         RNA_def_property_struct_type(prop, "BoneGroup");
1166         RNA_def_property_flag(prop, PROP_EDITABLE);
1167         RNA_def_property_pointer_funcs(prop, "rna_Pose_active_bone_group_get", "rna_Pose_active_bone_group_set", NULL, NULL);
1168         RNA_def_property_ui_text(prop, "Active Bone Group", "Active bone group for this pose");
1169         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
1170         
1171         prop= RNA_def_property(srna, "active_index", PROP_INT, PROP_NONE);
1172         RNA_def_property_int_sdna(prop, NULL, "active_group");
1173         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");
1174         RNA_def_property_ui_text(prop, "Active Bone Group Index", "Active index in bone groups array");
1175         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_update");
1176 }
1177
1178 static void rna_def_pose(BlenderRNA *brna)
1179 {
1180         StructRNA *srna;
1181         PropertyRNA *prop;
1182         
1183         /* struct definition */
1184         srna= RNA_def_struct(brna, "Pose", NULL);
1185         RNA_def_struct_sdna(srna, "bPose");
1186         RNA_def_struct_ui_text(srna, "Pose", "A collection of pose channels, including settings for animating bones");
1187
1188         /* pose channels */
1189         prop= RNA_def_property(srna, "bones", PROP_COLLECTION, PROP_NONE);
1190         RNA_def_property_collection_sdna(prop, NULL, "chanbase", NULL);
1191         RNA_def_property_struct_type(prop, "PoseBone");
1192         RNA_def_property_ui_text(prop, "Pose Bones", "Individual pose bones for the armature");
1193         RNA_def_property_collection_funcs(prop, 0, 0, 0, 0, 0, 0, "rna_PoseBones_lookup_string"); /* can be removed, only for fast lookup */
1194         /* bone groups */
1195         prop= RNA_def_property(srna, "bone_groups", PROP_COLLECTION, PROP_NONE);
1196         RNA_def_property_collection_sdna(prop, NULL, "agroups", NULL);
1197         RNA_def_property_struct_type(prop, "BoneGroup");
1198         RNA_def_property_ui_text(prop, "Bone Groups", "Groups of the bones");
1199         rna_def_bone_groups(brna, prop);
1200         
1201         /* ik solvers */
1202         prop= RNA_def_property(srna, "ik_solver", PROP_ENUM, PROP_NONE);
1203         RNA_def_property_enum_sdna(prop, NULL, "iksolver");
1204         RNA_def_property_enum_funcs(prop, NULL, "rna_Pose_ik_solver_set", NULL);
1205         RNA_def_property_enum_items(prop, prop_iksolver_items);
1206         RNA_def_property_ui_text(prop, "IK Solver", "Selection of IK solver for IK chain, current choice is 0 for Legacy, 1 for iTaSC");
1207         RNA_def_property_update(prop, NC_OBJECT|ND_POSE, "rna_Pose_ik_solver_update");
1208
1209         prop= RNA_def_property(srna, "ik_param", PROP_POINTER, PROP_NONE);
1210         RNA_def_property_struct_type(prop, "IKParam");
1211         RNA_def_property_pointer_funcs(prop, "rna_Pose_ikparam_get", NULL, "rna_Pose_ikparam_typef", NULL);
1212         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
1213         RNA_def_property_ui_text(prop, "IK Param", "Parameters for IK solver");
1214         
1215         /* animviz */
1216         rna_def_animviz_common(srna);
1217         
1218         /* RNA_api_pose(srna); */
1219 }
1220
1221 void RNA_def_pose(BlenderRNA *brna)
1222 {
1223         rna_def_pose(brna);
1224         rna_def_pose_channel(brna);
1225         rna_def_pose_ikparam(brna);
1226         rna_def_pose_itasc(brna);
1227         rna_def_bone_group(brna);
1228 }
1229
1230 #endif