Bugfix [#27157] keyframing a constrained bone does not work as before
[blender.git] / source / blender / editors / animation / keyframing.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  * The Original Code is Copyright (C) 2009 Blender Foundation, Joshua Leung
21  * All rights reserved.
22  *
23  * The Original Code is: all of this file.
24  *
25  * Contributor(s): Joshua Leung (full recode)
26  *
27  * ***** END GPL LICENSE BLOCK *****
28  */
29
30 /** \file blender/editors/animation/keyframing.c
31  *  \ingroup edanimation
32  */
33
34  
35 #include <stdio.h>
36 #include <stddef.h>
37 #include <string.h>
38 #include <math.h>
39 #include <float.h>
40
41 #include "MEM_guardedalloc.h"
42
43 #include "BLI_blenlib.h"
44 #include "BLI_math.h"
45 #include "BLI_dynstr.h"
46 #include "BLI_utildefines.h"
47
48 #include "DNA_anim_types.h"
49 #include "DNA_armature_types.h"
50 #include "DNA_constraint_types.h"
51 #include "DNA_key_types.h"
52 #include "DNA_material_types.h"
53 #include "DNA_scene_types.h"
54 #include "DNA_object_types.h"
55
56 #include "BKE_animsys.h"
57 #include "BKE_action.h"
58 #include "BKE_armature.h"
59 #include "BKE_constraint.h"
60 #include "BKE_depsgraph.h"
61 #include "BKE_fcurve.h"
62 #include "BKE_main.h"
63 #include "BKE_nla.h"
64 #include "BKE_global.h"
65 #include "BKE_context.h"
66 #include "BKE_report.h"
67 #include "BKE_key.h"
68 #include "BKE_material.h"
69
70 #include "ED_anim_api.h"
71 #include "ED_keyframing.h"
72 #include "ED_keyframes_edit.h"
73 #include "ED_screen.h"
74
75 #include "UI_interface.h"
76
77 #include "WM_api.h"
78 #include "WM_types.h"
79
80 #include "RNA_access.h"
81 #include "RNA_define.h"
82 #include "RNA_enum_types.h"
83
84 #include "anim_intern.h"
85
86 /* ************************************************** */
87 /* Keyframing Setting Wrangling */
88
89 /* Get the active settings for keyframing settings from context (specifically the given scene) */
90 short ANIM_get_keyframing_flags (Scene *scene, short incl_mode)
91 {
92         short flag = 0;
93         
94         /* standard flags */
95         {
96                 /* visual keying */
97                 if (IS_AUTOKEY_FLAG(scene, AUTOMATKEY)) 
98                         flag |= INSERTKEY_MATRIX;
99                 
100                 /* only needed */
101                 if (IS_AUTOKEY_FLAG(scene, INSERTNEEDED)) 
102                         flag |= INSERTKEY_NEEDED;
103                 
104                 /* default F-Curve color mode - RGB from XYZ indices */
105                 if (IS_AUTOKEY_FLAG(scene, XYZ2RGB)) 
106                         flag |= INSERTKEY_XYZ2RGB;
107         }
108                 
109         /* only if including settings from the autokeying mode... */
110         if (incl_mode) 
111         { 
112                 /* keyframing mode - only replace existing keyframes */
113                 if (IS_AUTOKEY_MODE(scene, EDITKEYS)) 
114                         flag |= INSERTKEY_REPLACE;
115         }
116                 
117         return flag;
118 }
119
120 /* ******************************************* */
121 /* Animation Data Validation */
122
123 /* Get (or add relevant data to be able to do so) the Active Action for the given 
124  * Animation Data block, given an ID block where the Animation Data should reside.
125  */
126 bAction *verify_adt_action (ID *id, short add)
127 {
128         AnimData *adt;
129         
130         /* init animdata if none available yet */
131         adt= BKE_animdata_from_id(id);
132         if ((adt == NULL) && (add))
133                 adt= BKE_id_add_animdata(id);
134         if (adt == NULL) { 
135                 /* if still none (as not allowed to add, or ID doesn't have animdata for some reason) */
136                 printf("ERROR: Couldn't add AnimData (ID = %s) \n", (id) ? (id->name) : "<None>");
137                 return NULL;
138         }
139                 
140         /* init action if none available yet */
141         // TODO: need some wizardry to handle NLA stuff correct
142         if ((adt->action == NULL) && (add)) {
143                 char actname[sizeof(id->name)-2];
144                 BLI_snprintf(actname, sizeof(actname), "%sAction", id->name+2);
145                 adt->action= add_empty_action(actname);
146         }
147                 
148         /* return the action */
149         return adt->action;
150 }
151
152 /* Get (or add relevant data to be able to do so) F-Curve from the Active Action, 
153  * for the given Animation Data block. This assumes that all the destinations are valid.
154  */
155 FCurve *verify_fcurve (bAction *act, const char group[], const char rna_path[], const int array_index, short add)
156 {
157         bActionGroup *grp;
158         FCurve *fcu;
159         
160         /* sanity checks */
161         if ELEM(NULL, act, rna_path)
162                 return NULL;
163                 
164         /* try to find f-curve matching for this setting 
165          *      - add if not found and allowed to add one
166          *              TODO: add auto-grouping support? how this works will need to be resolved
167          */
168         if (act)
169                 fcu= list_find_fcurve(&act->curves, rna_path, array_index);
170         else
171                 fcu= NULL;
172         
173         if ((fcu == NULL) && (add)) {
174                 /* use default settings to make a F-Curve */
175                 fcu= MEM_callocN(sizeof(FCurve), "FCurve");
176                 
177                 fcu->flag = (FCURVE_VISIBLE|FCURVE_AUTO_HANDLES|FCURVE_SELECTED);
178                 if (act->curves.first==NULL) 
179                         fcu->flag |= FCURVE_ACTIVE;     /* first one added active */
180                         
181                 /* store path - make copy, and store that */
182                 fcu->rna_path= BLI_strdupn(rna_path, strlen(rna_path));
183                 fcu->array_index= array_index;
184                 
185                 /* if a group name has been provided, try to add or find a group, then add F-Curve to it */
186                 if (group) {
187                         /* try to find group */
188                         grp= action_groups_find_named(act, group);
189                         
190                         /* no matching groups, so add one */
191                         if (grp == NULL)
192                                 grp= action_groups_add_new(act, group);
193                         
194                         /* add F-Curve to group */
195                         action_groups_add_channel(act, grp, fcu);
196                 }
197                 else {
198                         /* just add F-Curve to end of Action's list */
199                         BLI_addtail(&act->curves, fcu);
200                 }
201         }
202         
203         /* return the F-Curve */
204         return fcu;
205 }
206
207 /* ************************************************** */
208 /* KEYFRAME INSERTION */
209
210 /* -------------- BezTriple Insertion -------------------- */
211
212 /* This function adds a given BezTriple to an F-Curve. It will allocate 
213  * memory for the array if needed, and will insert the BezTriple into a
214  * suitable place in chronological order.
215  * 
216  * NOTE: any recalculate of the F-Curve that needs to be done will need to 
217  *              be done by the caller.
218  */
219 int insert_bezt_fcurve (FCurve *fcu, BezTriple *bezt, short flag)
220 {
221         int i= 0;
222         
223         /* are there already keyframes? */
224         if (fcu->bezt) {
225                 short replace = -1;
226                 i = binarysearch_bezt_index(fcu->bezt, bezt->vec[1][0], fcu->totvert, &replace);
227                 
228                 /* replace an existing keyframe? */
229                 if (replace) {                  
230                         /* sanity check: 'i' may in rare cases exceed arraylen */
231                         if ((i >= 0) && (i < fcu->totvert)) {
232                                 /* just change the values when replacing, so as to not overwrite handles */
233                                 BezTriple *dst= (fcu->bezt + i);
234                                 float dy= bezt->vec[1][1] - dst->vec[1][1];
235                                 
236                                 /* just apply delta value change to the handle values */
237                                 dst->vec[0][1] += dy;
238                                 dst->vec[1][1] += dy;
239                                 dst->vec[2][1] += dy;
240                                 
241                                 dst->f1= bezt->f1;
242                                 dst->f2= bezt->f2;
243                                 dst->f3= bezt->f3;
244                                 
245                                 // TODO: perform some other operations?
246                         }
247                 }
248                 /* keyframing modes allow to not replace keyframe */
249                 else if ((flag & INSERTKEY_REPLACE) == 0) {
250                         /* insert new - if we're not restricted to replacing keyframes only */
251                         BezTriple *newb= MEM_callocN((fcu->totvert+1)*sizeof(BezTriple), "beztriple");
252                         
253                         /* add the beztriples that should occur before the beztriple to be pasted (originally in fcu) */
254                         if (i > 0)
255                                 memcpy(newb, fcu->bezt, i*sizeof(BezTriple));
256                         
257                         /* add beztriple to paste at index i */
258                         *(newb + i)= *bezt;
259                         
260                         /* add the beztriples that occur after the beztriple to be pasted (originally in fcu) */
261                         if (i < fcu->totvert) 
262                                 memcpy(newb+i+1, fcu->bezt+i, (fcu->totvert-i)*sizeof(BezTriple));
263                         
264                         /* replace (+ free) old with new, only if necessary to do so */
265                         MEM_freeN(fcu->bezt);
266                         fcu->bezt= newb;
267                         
268                         fcu->totvert++;
269                 }
270         }
271         /* no keyframes already, but can only add if...
272          *      1) keyframing modes say that keyframes can only be replaced, so adding new ones won't know
273          *      2) there are no samples on the curve
274          *              // NOTE: maybe we may want to allow this later when doing samples -> bezt conversions, 
275          *              // but for now, having both is asking for trouble
276          */
277         else if ((flag & INSERTKEY_REPLACE)==0 && (fcu->fpt==NULL)) {
278                 /* create new keyframes array */
279                 fcu->bezt= MEM_callocN(sizeof(BezTriple), "beztriple");
280                 *(fcu->bezt)= *bezt;
281                 fcu->totvert= 1;
282         }
283         /* cannot add anything */
284         else {
285                 /* return error code -1 to prevent any misunderstandings */
286                 return -1;
287         }
288         
289         
290         /* we need to return the index, so that some tools which do post-processing can 
291          * detect where we added the BezTriple in the array
292          */
293         return i;
294 }
295
296 /* This function is a wrapper for insert_bezt_fcurve_internal(), and should be used when
297  * adding a new keyframe to a curve, when the keyframe doesn't exist anywhere else yet. 
298  * It returns the index at which the keyframe was added.
299  */
300 int insert_vert_fcurve (FCurve *fcu, float x, float y, short flag)
301 {
302         BezTriple beztr= {{{0}}};
303         unsigned int oldTot = fcu->totvert;
304         int a;
305         
306         /* set all three points, for nicer start position 
307          * NOTE: +/- 1 on vec.x for left and right handles is so that 'free' handles work ok...
308          */
309         beztr.vec[0][0]= x-1.0f; 
310         beztr.vec[0][1]= y;
311         beztr.vec[1][0]= x;
312         beztr.vec[1][1]= y;
313         beztr.vec[2][0]= x+1.0f;
314         beztr.vec[2][1]= y;
315         beztr.ipo= U.ipo_new; /* use default interpolation mode here... */
316         beztr.f1= beztr.f2= beztr.f3= SELECT;
317         beztr.h1= beztr.h2= U.keyhandles_new; /* use default handle type here */
318         //BEZKEYTYPE(&beztr)= scene->keytype; /* default keyframe type */
319         
320         /* add temp beztriple to keyframes */
321         a= insert_bezt_fcurve(fcu, &beztr, flag);
322         
323         /* what if 'a' is a negative index? 
324          * for now, just exit to prevent any segfaults
325          */
326         if (a < 0) return -1;
327         
328         /* don't recalculate handles if fast is set
329          *      - this is a hack to make importers faster
330          *      - we may calculate twice (due to autohandle needing to be calculated twice)
331          */
332         if ((flag & INSERTKEY_FAST) == 0) 
333                 calchandles_fcurve(fcu);
334         
335         /* set handletype and interpolation */
336         if ((fcu->totvert > 2) && (flag & INSERTKEY_REPLACE)==0) {
337                 BezTriple *bezt= (fcu->bezt + a);
338                 
339                 /* set interpolation from previous (if available), but only if we didn't just replace some keyframe 
340                  *      - replacement is indicated by no-change in number of verts
341                  *      - when replacing, the user may have specified some interpolation that should be kept
342                  */
343                 if (fcu->totvert > oldTot) {
344                         if (a > 0) 
345                                 bezt->ipo= (bezt-1)->ipo;
346                         else if (a < fcu->totvert-1) 
347                                 bezt->ipo= (bezt+1)->ipo;
348                 }
349                         
350                 /* don't recalculate handles if fast is set
351                  *      - this is a hack to make importers faster
352                  *      - we may calculate twice (due to autohandle needing to be calculated twice)
353                  */
354                 if ((flag & INSERTKEY_FAST) == 0) 
355                         calchandles_fcurve(fcu);
356         }
357         
358         /* return the index at which the keyframe was added */
359         return a;
360 }
361
362 /* -------------- 'Smarter' Keyframing Functions -------------------- */
363 /* return codes for new_key_needed */
364 enum {
365         KEYNEEDED_DONTADD = 0,
366         KEYNEEDED_JUSTADD,
367         KEYNEEDED_DELPREV,
368         KEYNEEDED_DELNEXT
369 } /*eKeyNeededStatus*/;
370
371 /* This helper function determines whether a new keyframe is needed */
372 /* Cases where keyframes should not be added:
373  *      1. Keyframe to be added bewteen two keyframes with similar values
374  *      2. Keyframe to be added on frame where two keyframes are already situated
375  *      3. Keyframe lies at point that intersects the linear line between two keyframes
376  */
377 static short new_key_needed (FCurve *fcu, float cFrame, float nValue) 
378 {
379         BezTriple *bezt=NULL, *prev=NULL;
380         int totCount, i;
381         float valA = 0.0f, valB = 0.0f;
382         
383         /* safety checking */
384         if (fcu == NULL) return KEYNEEDED_JUSTADD;
385         totCount= fcu->totvert;
386         if (totCount == 0) return KEYNEEDED_JUSTADD;
387         
388         /* loop through checking if any are the same */
389         bezt= fcu->bezt;
390         for (i=0; i<totCount; i++) {
391                 float prevPosi=0.0f, prevVal=0.0f;
392                 float beztPosi=0.0f, beztVal=0.0f;
393                         
394                 /* get current time+value */    
395                 beztPosi= bezt->vec[1][0];
396                 beztVal= bezt->vec[1][1];
397                         
398                 if (prev) {
399                         /* there is a keyframe before the one currently being examined */               
400                         
401                         /* get previous time+value */
402                         prevPosi= prev->vec[1][0];
403                         prevVal= prev->vec[1][1];
404                         
405                         /* keyframe to be added at point where there are already two similar points? */
406                         if (IS_EQF(prevPosi, cFrame) && IS_EQF(beztPosi, cFrame) && IS_EQF(beztPosi, prevPosi)) {
407                                 return KEYNEEDED_DONTADD;
408                         }
409                         
410                         /* keyframe between prev+current points ? */
411                         if ((prevPosi <= cFrame) && (cFrame <= beztPosi)) {
412                                 /* is the value of keyframe to be added the same as keyframes on either side ? */
413                                 if (IS_EQF(prevVal, nValue) && IS_EQF(beztVal, nValue) && IS_EQF(prevVal, beztVal)) {
414                                         return KEYNEEDED_DONTADD;
415                                 }
416                                 else {
417                                         float realVal;
418                                         
419                                         /* get real value of curve at that point */
420                                         realVal= evaluate_fcurve(fcu, cFrame);
421                                         
422                                         /* compare whether it's the same as proposed */
423                                         if (IS_EQF(realVal, nValue))
424                                                 return KEYNEEDED_DONTADD;
425                                         else 
426                                                 return KEYNEEDED_JUSTADD;
427                                 }
428                         }
429                         
430                         /* new keyframe before prev beztriple? */
431                         if (cFrame < prevPosi) {
432                                 /* A new keyframe will be added. However, whether the previous beztriple
433                                  * stays around or not depends on whether the values of previous/current
434                                  * beztriples and new keyframe are the same.
435                                  */
436                                 if (IS_EQF(prevVal, nValue) && IS_EQF(beztVal, nValue) && IS_EQF(prevVal, beztVal))
437                                         return KEYNEEDED_DELNEXT;
438                                 else 
439                                         return KEYNEEDED_JUSTADD;
440                         }
441                 }
442                 else {
443                         /* just add a keyframe if there's only one keyframe 
444                          * and the new one occurs before the exisiting one does.
445                          */
446                         if ((cFrame < beztPosi) && (totCount==1))
447                                 return KEYNEEDED_JUSTADD;
448                 }
449                 
450                 /* continue. frame to do not yet passed (or other conditions not met) */
451                 if (i < (totCount-1)) {
452                         prev= bezt;
453                         bezt++;
454                 }
455                 else
456                         break;
457         }
458         
459         /* Frame in which to add a new-keyframe occurs after all other keys
460          * -> If there are at least two existing keyframes, then if the values of the
461          *       last two keyframes and the new-keyframe match, the last existing keyframe
462          *       gets deleted as it is no longer required.
463          * -> Otherwise, a keyframe is just added. 1.0 is added so that fake-2nd-to-last
464          *       keyframe is not equal to last keyframe.
465          */
466         bezt= (fcu->bezt + (fcu->totvert - 1));
467         valA= bezt->vec[1][1];
468         
469         if (prev)
470                 valB= prev->vec[1][1];
471         else 
472                 valB= bezt->vec[1][1] + 1.0f; 
473                 
474         if (IS_EQF(valA, nValue) && IS_EQF(valA, valB))
475                 return KEYNEEDED_DELPREV;
476         else 
477                 return KEYNEEDED_JUSTADD;
478 }
479
480 /* ------------------ RNA Data-Access Functions ------------------ */
481
482 /* Try to read value using RNA-properties obtained already */
483 static float setting_get_rna_value (PointerRNA *ptr, PropertyRNA *prop, int index)
484 {
485         float value= 0.0f;
486         
487         switch (RNA_property_type(prop)) {
488                 case PROP_BOOLEAN:
489                         if (RNA_property_array_length(ptr, prop))
490                                 value= (float)RNA_property_boolean_get_index(ptr, prop, index);
491                         else
492                                 value= (float)RNA_property_boolean_get(ptr, prop);
493                         break;
494                 case PROP_INT:
495                         if (RNA_property_array_length(ptr, prop))
496                                 value= (float)RNA_property_int_get_index(ptr, prop, index);
497                         else
498                                 value= (float)RNA_property_int_get(ptr, prop);
499                         break;
500                 case PROP_FLOAT:
501                         if (RNA_property_array_length(ptr, prop))
502                                 value= RNA_property_float_get_index(ptr, prop, index);
503                         else
504                                 value= RNA_property_float_get(ptr, prop);
505                         break;
506                 case PROP_ENUM:
507                         value= (float)RNA_property_enum_get(ptr, prop);
508                         break;
509                 default:
510                         break;
511         }
512         
513         return value;
514 }
515
516 /* ------------------ 'Visual' Keyframing Functions ------------------ */
517
518 /* internal status codes for visualkey_can_use */
519 enum {
520         VISUALKEY_NONE = 0,
521         VISUALKEY_LOC,
522         VISUALKEY_ROT,
523 };
524
525 /* This helper function determines if visual-keyframing should be used when  
526  * inserting keyframes for the given channel. As visual-keyframing only works
527  * on Object and Pose-Channel blocks, this should only get called for those 
528  * blocktypes, when using "standard" keying but 'Visual Keying' option in Auto-Keying 
529  * settings is on.
530  */
531 static short visualkey_can_use (PointerRNA *ptr, PropertyRNA *prop)
532 {
533         bConstraint *con= NULL;
534         short searchtype= VISUALKEY_NONE;
535         char *identifier= NULL;
536         
537         /* validate data */
538         // TODO: this check is probably not needed, but it won't hurt
539         if (ELEM3(NULL, ptr, ptr->data, prop))
540                 return 0;
541                 
542         /* get first constraint and determine type of keyframe constraints to check for 
543          *      - constraints can be on either Objects or PoseChannels, so we only check if the
544          *        ptr->type is RNA_Object or RNA_PoseBone, which are the RNA wrapping-info for
545          *        those structs, allowing us to identify the owner of the data 
546          */
547         if (ptr->type == &RNA_Object) {
548                 /* Object */
549                 Object *ob= (Object *)ptr->data;
550                 
551                 con= ob->constraints.first;
552                 identifier= (char *)RNA_property_identifier(prop);
553         }
554         else if (ptr->type == &RNA_PoseBone) {
555                 /* Pose Channel */
556                 bPoseChannel *pchan= (bPoseChannel *)ptr->data;
557                 
558                 con= pchan->constraints.first;
559                 identifier= (char *)RNA_property_identifier(prop);
560         }
561         
562         /* check if any data to search using */
563         if (ELEM(NULL, con, identifier))
564                 return 0;
565                 
566         /* location or rotation identifiers only... */
567         if (strstr(identifier, "location"))
568                 searchtype= VISUALKEY_LOC;
569         else if (strstr(identifier, "rotation"))
570                 searchtype= VISUALKEY_ROT;
571         else {
572                 printf("visualkey_can_use() failed: identifier - '%s' \n", identifier);
573                 return 0;
574         }
575         
576         
577         /* only search if a searchtype and initial constraint are available */
578         if (searchtype && con) {
579                 for (; con; con= con->next) {
580                         /* only consider constraint if it is not disabled, and has influence */
581                         if (con->flag & CONSTRAINT_DISABLE) continue;
582                         if (con->enforce == 0.0f) continue;
583                         
584                         /* some constraints may alter these transforms */
585                         switch (con->type) {
586                                 /* multi-transform constraints */
587                                 case CONSTRAINT_TYPE_CHILDOF:
588                                         return 1;
589                                 case CONSTRAINT_TYPE_TRANSFORM:
590                                 case CONSTRAINT_TYPE_TRANSLIKE:
591                                         return 1;
592                                 case CONSTRAINT_TYPE_FOLLOWPATH:
593                                         return 1;
594                                 case CONSTRAINT_TYPE_KINEMATIC:
595                                         return 1;
596                                         
597                                 /* single-transform constraits  */
598                                 case CONSTRAINT_TYPE_TRACKTO:
599                                         if (searchtype==VISUALKEY_ROT) return 1;
600                                         break;
601                                 case CONSTRAINT_TYPE_ROTLIMIT:
602                                         if (searchtype==VISUALKEY_ROT) return 1;
603                                         break;
604                                 case CONSTRAINT_TYPE_LOCLIMIT:
605                                         if (searchtype==VISUALKEY_LOC) return 1;
606                                         break;
607                                 case CONSTRAINT_TYPE_ROTLIKE:
608                                         if (searchtype==VISUALKEY_ROT) return 1;
609                                         break;
610                                 case CONSTRAINT_TYPE_DISTLIMIT:
611                                         if (searchtype==VISUALKEY_LOC) return 1;
612                                         break;
613                                 case CONSTRAINT_TYPE_LOCLIKE:
614                                         if (searchtype==VISUALKEY_LOC) return 1;
615                                         break;
616                                 case CONSTRAINT_TYPE_LOCKTRACK:
617                                         if (searchtype==VISUALKEY_ROT) return 1;
618                                         break;
619                                 case CONSTRAINT_TYPE_MINMAX:
620                                         if (searchtype==VISUALKEY_LOC) return 1;
621                                         break;
622                                 
623                                 default:
624                                         break;
625                         }
626                 }
627         }
628         
629         /* when some condition is met, this function returns, so here it can be 0 */
630         return 0;
631 }
632
633 /* This helper function extracts the value to use for visual-keyframing 
634  * In the event that it is not possible to perform visual keying, try to fall-back
635  * to using the default method. Assumes that all data it has been passed is valid.
636  */
637 static float visualkey_get_value (PointerRNA *ptr, PropertyRNA *prop, int array_index)
638 {
639         char *identifier= (char *)RNA_property_identifier(prop);
640         
641         /* handle for Objects or PoseChannels only 
642          *      - constraints can be on either Objects or PoseChannels, so we only check if the
643          *        ptr->type is RNA_Object or RNA_PoseBone, which are the RNA wrapping-info for
644          *        those structs, allowing us to identify the owner of the data 
645          *      - assume that array_index will be sane
646          */
647         if (ptr->type == &RNA_Object) {
648                 Object *ob= (Object *)ptr->data;
649                 
650                 /* parented objects are not supported, as the effects of the parent
651                  * are included in the matrix, which kindof beats the point
652                  */
653                 if (ob->parent == NULL) {
654                         /* only Location or Rotation keyframes are supported now */
655                         if (strstr(identifier, "location")) {
656                                 return ob->obmat[3][array_index];
657                         }
658                         else if (strstr(identifier, "rotation_euler")) {
659                                 float eul[3];
660                                 
661                                 mat4_to_eulO(eul, ob->rotmode, ob->obmat);
662                                 return eul[array_index];
663                         }
664                         else if (strstr(identifier, "rotation_quaternion")) {
665                                 float trimat[3][3], quat[4];
666                                 
667                                 copy_m3_m4(trimat, ob->obmat);
668                                 mat3_to_quat_is_ok(quat, trimat);
669                                 
670                                 return quat[array_index];
671                         }
672                         else if (strstr(identifier, "rotation_axis_angle")) {
673                                 float axis[3], angle;
674                                 
675                                 mat4_to_axis_angle(axis, &angle, ob->obmat);
676                                 
677                                 /* w = 0, x,y,z = 1,2,3 */
678                                 if (array_index == 0)
679                                         return angle;
680                                 else
681                                         return axis[array_index - 1];
682                         }
683                 }
684         }
685         else if (ptr->type == &RNA_PoseBone) {
686                 Object *ob = (Object *)ptr->id.data; /* we assume that this is always set, and is an object */
687                 bPoseChannel *pchan= (bPoseChannel *)ptr->data;
688                 float tmat[4][4];
689                 
690                 /* Although it is not strictly required for this particular space conversion, 
691                  * arg1 must not be null, as there is a null check for the other conversions to
692                  * be safe. Therefore, the active object is passed here, and in many cases, this
693                  * will be what owns the pose-channel that is getting this anyway.
694                  */
695                 copy_m4_m4(tmat, pchan->pose_mat);
696                 constraint_mat_convertspace(ob, pchan, tmat, CONSTRAINT_SPACE_POSE, CONSTRAINT_SPACE_LOCAL);
697                 
698                 /* Loc, Rot/Quat keyframes are supported... */
699                 if (strstr(identifier, "location")) {
700                         /* only use for non-connected bones */
701                         if ((pchan->bone->parent) && !(pchan->bone->flag & BONE_CONNECTED))
702                                 return tmat[3][array_index];
703                         else if (pchan->bone->parent == NULL)
704                                 return tmat[3][array_index];
705                 }
706                 else if (strstr(identifier, "rotation_euler")) {
707                         float eul[3];
708                         
709                         mat4_to_eulO(eul, pchan->rotmode, tmat);
710                         return eul[array_index];
711                 }
712                 else if (strstr(identifier, "rotation_quaternion")) {
713                         float trimat[3][3], quat[4];
714                         
715                         copy_m3_m4(trimat, tmat);
716                         mat3_to_quat_is_ok(quat, trimat);
717                         
718                         return quat[array_index];
719                 }
720                 else if (strstr(identifier, "rotation_axis_angle")) {
721                         float axis[3], angle;
722                         
723                         mat4_to_axis_angle(axis, &angle, tmat);
724                         
725                         /* w = 0, x,y,z = 1,2,3 */
726                         if (array_index == 0)
727                                 return angle;
728                         else
729                                 return axis[array_index - 1];
730                 }
731         }
732         
733         /* as the function hasn't returned yet, read value from system in the default way */
734         return setting_get_rna_value(ptr, prop, array_index);
735 }
736
737 /* ------------------------- Insert Key API ------------------------- */
738
739 /* Secondary Keyframing API call: 
740  *      Use this when validation of necessary animation data is not necessary, since an RNA-pointer to the necessary
741  *      data being keyframed, and a pointer to the F-Curve to use have both been provided.
742  *
743  *      The flag argument is used for special settings that alter the behaviour of
744  *      the keyframe insertion. These include the 'visual' keyframing modes, quick refresh,
745  *      and extra keyframe filtering.
746  */
747 short insert_keyframe_direct (ReportList *reports, PointerRNA ptr, PropertyRNA *prop, FCurve *fcu, float cfra, short flag)
748 {
749         float curval= 0.0f;
750         
751         /* no F-Curve to add keyframe to? */
752         if (fcu == NULL) {
753                 BKE_report(reports, RPT_ERROR, "No F-Curve to add keyframes to");
754                 return 0;
755         }
756         /* F-Curve not editable? */
757         if (fcurve_is_keyframable(fcu) == 0) {
758                 BKE_reportf(reports, RPT_ERROR, 
759                         "F-Curve with path = '%s' [%d] cannot be keyframed. Ensure that it is not locked or sampled. Also, try removing F-Modifiers.",
760                         fcu->rna_path, fcu->array_index);
761                 return 0;
762         }
763         
764         /* if no property given yet, try to validate from F-Curve info */
765         if ((ptr.id.data == NULL) && (ptr.data==NULL)) {
766                 BKE_report(reports, RPT_ERROR, "No RNA-pointer available to retrieve values for keyframing from");
767                 return 0;
768         }
769         if (prop == NULL) {
770                 PointerRNA tmp_ptr;
771                 
772                 /* try to get property we should be affecting */
773                 if ((RNA_path_resolve(&ptr, fcu->rna_path, &tmp_ptr, &prop) == 0) || (prop == NULL)) {
774                         /* property not found... */
775                         const char *idname= (ptr.id.data) ? ((ID *)ptr.id.data)->name : "<No ID-Pointer>";
776                         
777                         BKE_reportf(reports, RPT_ERROR,
778                                 "Could not insert keyframe, as RNA Path is invalid for the given ID (ID = %s, Path = %s)", 
779                                 idname, fcu->rna_path);
780                         return 0;
781                 }
782                 else {
783                         /* property found, so overwrite 'ptr' to make later code easier */
784                         ptr= tmp_ptr;
785                 }
786         }
787         
788         /* set additional flags for the F-Curve (i.e. only integer values) */
789         fcu->flag &= ~(FCURVE_INT_VALUES|FCURVE_DISCRETE_VALUES);
790         switch (RNA_property_type(prop)) {
791                 case PROP_FLOAT:
792                         /* do nothing */
793                         break;
794                 case PROP_INT:
795                         /* do integer (only 'whole' numbers) interpolation between all points */
796                         fcu->flag |= FCURVE_INT_VALUES;
797                         break;
798                 default:
799                         /* do 'discrete' (i.e. enum, boolean values which cannot take any intermediate
800                          * values at all) interpolation between all points
801                          *      - however, we must also ensure that evaluated values are only integers still
802                          */
803                         fcu->flag |= (FCURVE_DISCRETE_VALUES|FCURVE_INT_VALUES);
804                         break;
805         }
806         
807         /* obtain value to give keyframe */
808         if ( (flag & INSERTKEY_MATRIX) && 
809                  (visualkey_can_use(&ptr, prop)) ) 
810         {
811                 /* visual-keying is only available for object and pchan datablocks, as 
812                  * it works by keyframing using a value extracted from the final matrix 
813                  * instead of using the kt system to extract a value.
814                  */
815                 curval= visualkey_get_value(&ptr, prop, fcu->array_index);
816         }
817         else {
818                 /* read value from system */
819                 curval= setting_get_rna_value(&ptr, prop, fcu->array_index);
820         }
821         
822         /* only insert keyframes where they are needed */
823         if (flag & INSERTKEY_NEEDED) {
824                 short insert_mode;
825                 
826                 /* check whether this curve really needs a new keyframe */
827                 insert_mode= new_key_needed(fcu, cfra, curval);
828                 
829                 /* insert new keyframe at current frame */
830                 if (insert_mode)
831                         insert_vert_fcurve(fcu, cfra, curval, flag);
832                 
833                 /* delete keyframe immediately before/after newly added */
834                 switch (insert_mode) {
835                         case KEYNEEDED_DELPREV:
836                                 delete_fcurve_key(fcu, fcu->totvert-2, 1);
837                                 break;
838                         case KEYNEEDED_DELNEXT:
839                                 delete_fcurve_key(fcu, 1, 1);
840                                 break;
841                 }
842                 
843                 /* only return success if keyframe added */
844                 if (insert_mode)
845                         return 1;
846         }
847         else {
848                 /* just insert keyframe */
849                 insert_vert_fcurve(fcu, cfra, curval, flag);
850                 
851                 /* return success */
852                 return 1;
853         }
854         
855         /* failed */
856         return 0;
857 }
858
859 /* Main Keyframing API call:
860  *      Use this when validation of necessary animation data is necessary, since it may not exist yet.
861  *      
862  *      The flag argument is used for special settings that alter the behaviour of
863  *      the keyframe insertion. These include the 'visual' keyframing modes, quick refresh,
864  *      and extra keyframe filtering.
865  *
866  *      index of -1 keys all array indices
867  */
868 short insert_keyframe (ReportList *reports, ID *id, bAction *act, const char group[], const char rna_path[], int array_index, float cfra, short flag)
869 {       
870         PointerRNA id_ptr, ptr;
871         PropertyRNA *prop = NULL;
872         FCurve *fcu;
873         int array_index_max= array_index+1;
874         int ret= 0;
875         
876         /* validate pointer first - exit if failure */
877         if (id == NULL) {
878                 BKE_reportf(reports, RPT_ERROR, "No ID-block to insert keyframe in (Path = %s)", rna_path);
879                 return 0;
880         }
881         
882         RNA_id_pointer_create(id, &id_ptr);
883         if ((RNA_path_resolve(&id_ptr, rna_path, &ptr, &prop) == 0) || (prop == NULL)) {
884                 BKE_reportf(reports, RPT_ERROR,
885                         "Could not insert keyframe, as RNA Path is invalid for the given ID (ID = %s, Path = %s)", 
886                         (id)? id->name : "<Missing ID-Block>", rna_path);
887                 return 0;
888         }
889         
890         /* if no action is provided, keyframe to the default one attached to this ID-block */
891         if (act == NULL) {
892                 AnimData *adt= BKE_animdata_from_id(id);
893                 
894                 /* get action to add F-Curve+keyframe to */
895                 act= verify_adt_action(id, 1);
896                 
897                 if (act == NULL) {
898                         BKE_reportf(reports, RPT_ERROR, 
899                                 "Could not insert keyframe, as this type does not support animation data (ID = %s, Path = %s)", 
900                                 id->name, rna_path);
901                         return 0;
902                 }
903                 
904                 /* apply NLA-mapping to frame to use (if applicable) */
905                 cfra= BKE_nla_tweakedit_remap(adt, cfra, NLATIME_CONVERT_UNMAP);
906         }
907
908 #if 0
909         /* apply special time tweaking */
910                 // XXX check on this stuff...
911         if (GS(id->name) == ID_OB) {
912                 //Object *ob= (Object *)id;
913                 
914                 /* ancient time-offset cruft */
915                 //if ( (ob->ipoflag & OB_OFFS_OB) && (give_timeoffset(ob)) ) {
916                 //      /* actually frametofloat calc again! */
917                 //      cfra-= give_timeoffset(ob)*scene->r.framelen;
918                 //}
919         }
920 #endif
921         
922         /* key entire array convenience method */
923         if (array_index == -1) { 
924                 array_index= 0;
925                 array_index_max= RNA_property_array_length(&ptr, prop);
926                 
927                 /* for single properties, increase max_index so that the property itself gets included,
928                  * but don't do this for standard arrays since that can cause corruption issues 
929                  * (extra unused curves)
930                  */
931                 if (array_index_max == array_index)
932                         array_index_max++;
933         }
934         
935         /* will only loop once unless the array index was -1 */
936         for (; array_index < array_index_max; array_index++) {
937                 /* make sure the F-Curve exists 
938                  *      - if we're replacing keyframes only, DO NOT create new F-Curves if they do not exist yet
939                  *        but still try to get the F-Curve if it exists...
940                  */
941                 fcu= verify_fcurve(act, group, rna_path, array_index, (flag & INSERTKEY_REPLACE)==0);
942                 
943                 /* we may not have a F-Curve when we're replacing only... */
944                 if (fcu) {
945                         /* set color mode if the F-Curve is new (i.e. without any keyframes) */
946                         if ((fcu->totvert == 0) && (flag & INSERTKEY_XYZ2RGB)) {
947                                 /* for Loc/Rot/Scale and also Color F-Curves, the color of the F-Curve in the Graph Editor,
948                                  * is determined by the array index for the F-Curve
949                                  */
950                                 if (ELEM5(RNA_property_subtype(prop), PROP_TRANSLATION, PROP_XYZ, PROP_EULER, PROP_COLOR, PROP_COORDS)) {
951                                         fcu->color_mode= FCURVE_COLOR_AUTO_RGB;
952                                 }
953                         }
954                         
955                         /* insert keyframe */
956                         ret += insert_keyframe_direct(reports, ptr, prop, fcu, cfra, flag);
957                 }
958         }
959         
960         return ret;
961 }
962
963 /* ************************************************** */
964 /* KEYFRAME DELETION */
965
966 /* Main Keyframing API call:
967  *      Use this when validation of necessary animation data isn't necessary as it
968  *      already exists. It will delete a keyframe at the current frame.
969  *      
970  *      The flag argument is used for special settings that alter the behaviour of
971  *      the keyframe deletion. These include the quick refresh options.
972  */
973 short delete_keyframe (ReportList *reports, ID *id, bAction *act, const char group[], const char rna_path[], int array_index, float cfra, short UNUSED(flag))
974 {
975         AnimData *adt= BKE_animdata_from_id(id);
976         PointerRNA id_ptr, ptr;
977         PropertyRNA *prop;
978         int array_index_max= array_index+1;
979         int ret= 0;
980         
981         /* sanity checks */
982         if ELEM(NULL, id, adt) {
983                 BKE_report(reports, RPT_ERROR, "No ID-Block and/Or AnimData to delete keyframe from");
984                 return 0;
985         }
986         
987         /* validate pointer first - exit if failure */
988         RNA_id_pointer_create(id, &id_ptr);
989         if ((RNA_path_resolve(&id_ptr, rna_path, &ptr, &prop) == 0) || (prop == NULL)) {
990                 BKE_reportf(reports, RPT_ERROR, "Could not delete keyframe, as RNA Path is invalid for the given ID (ID = %s, Path = %s)", id->name, rna_path);
991                 return 0;
992         }
993         
994         /* get F-Curve
995          * Note: here is one of the places where we don't want new Action + F-Curve added!
996          *              so 'add' var must be 0
997          */
998         if (act == NULL) {
999                 /* if no action is provided, use the default one attached to this ID-block 
1000                  *      - if it doesn't exist, then we're out of options...
1001                  */
1002                 if (adt->action) {
1003                         act= adt->action;
1004                         
1005                         /* apply NLA-mapping to frame to use (if applicable) */
1006                         cfra= BKE_nla_tweakedit_remap(adt, cfra, NLATIME_CONVERT_UNMAP); 
1007                 }
1008                 else {
1009                         BKE_reportf(reports, RPT_ERROR, "No Action to delete keyframes from for ID = %s \n", id->name);
1010                         return 0;
1011                 }
1012         }
1013         
1014 #if 0
1015         /* apply special time tweaking */
1016                 // XXX check on this stuff...
1017         if (GS(id->name) == ID_OB) {
1018                 //Object *ob= (Object *)id;
1019                 
1020                 /* ancient time-offset cruft */
1021                 //if ( (ob->ipoflag & OB_OFFS_OB) && (give_timeoffset(ob)) ) {
1022                 //      /* actually frametofloat calc again! */
1023                 //      cfra-= give_timeoffset(ob)*scene->r.framelen;
1024                 //}
1025         }
1026 #endif
1027         
1028         /* key entire array convenience method */
1029         if (array_index == -1) { 
1030                 array_index= 0;
1031                 array_index_max= RNA_property_array_length(&ptr, prop);
1032                 
1033                 /* for single properties, increase max_index so that the property itself gets included,
1034                  * but don't do this for standard arrays since that can cause corruption issues 
1035                  * (extra unused curves)
1036                  */
1037                 if (array_index_max == array_index)
1038                         array_index_max++;
1039         }
1040         
1041         /* will only loop once unless the array index was -1 */
1042         for (; array_index < array_index_max; array_index++) {
1043                 FCurve *fcu= verify_fcurve(act, group, rna_path, array_index, 0);
1044                 short found = -1;
1045                 int i;
1046                 
1047                 /* check if F-Curve exists and/or whether it can be edited */
1048                 if (fcu == NULL)
1049                         continue;
1050                         
1051                 if ( (fcu->flag & FCURVE_PROTECTED) || ((fcu->grp) && (fcu->grp->flag & AGRP_PROTECTED)) ) {
1052                         if (G.f & G_DEBUG)
1053                                 printf("WARNING: not deleting keyframe for locked F-Curve \n");
1054                         continue;
1055                 }
1056                 
1057                 /* try to find index of beztriple to get rid of */
1058                 i = binarysearch_bezt_index(fcu->bezt, cfra, fcu->totvert, &found);
1059                 if (found) {                    
1060                         /* delete the key at the index (will sanity check + do recalc afterwards) */
1061                         delete_fcurve_key(fcu, i, 1);
1062                         
1063                         /* Only delete curve too if it won't be doing anything anymore */
1064                         if ((fcu->totvert == 0) && (list_has_suitable_fmodifier(&fcu->modifiers, 0, FMI_TYPE_GENERATE_CURVE) == 0))
1065                                 ANIM_fcurve_delete_from_animdata(NULL, adt, fcu);
1066                         
1067                         /* return success */
1068                         ret++;
1069                 }
1070         }
1071         
1072         /* return success/failure */
1073         return ret;
1074 }
1075
1076 /* ******************************************* */
1077 /* KEYFRAME MODIFICATION */
1078
1079 /* mode for commonkey_modifykey */
1080 enum {
1081         COMMONKEY_MODE_INSERT = 0,
1082         COMMONKEY_MODE_DELETE,
1083 } /*eCommonModifyKey_Modes*/;
1084
1085 /* Polling callback for use with ANIM_*_keyframe() operators
1086  * This is based on the standard ED_operator_areaactive callback,
1087  * except that it does special checks for a few spacetypes too...
1088  */
1089 static int modify_key_op_poll(bContext *C)
1090 {
1091         ScrArea *sa= CTX_wm_area(C);
1092         Scene *scene= CTX_data_scene(C);
1093         SpaceOops *so= CTX_wm_space_outliner(C);
1094         
1095         /* if no area or active scene */
1096         if (ELEM(NULL, sa, scene)) 
1097                 return 0;
1098         
1099         /* if Outliner, don't allow in some views */
1100         if (so) {
1101                 if (ELEM4(so->outlinevis, SO_GROUPS, SO_LIBRARIES, SO_VERSE_SESSION, SO_VERSE_SESSION))
1102                         return 0;
1103                 if (ELEM3(so->outlinevis, SO_SEQUENCE, SO_USERDEF, SO_KEYMAP))
1104                         return 0;
1105         }
1106         
1107         /* TODO: checks for other space types can be added here */
1108         
1109         /* should be fine */
1110         return 1;
1111 }
1112
1113 /* Insert Key Operator ------------------------ */
1114
1115 static int insert_key_exec (bContext *C, wmOperator *op)
1116 {
1117         Main *bmain= CTX_data_main(C);
1118         Scene *scene= CTX_data_scene(C);
1119         KeyingSet *ks= NULL;
1120         int type= RNA_enum_get(op->ptr, "type");
1121         float cfra= (float)CFRA; // XXX for now, don't bother about all the yucky offset crap
1122         short success;
1123         
1124         /* type is the Keying Set the user specified to use when calling the operator:
1125          *      - type == 0: use scene's active Keying Set
1126          *      - type > 0: use a user-defined Keying Set from the active scene
1127          *      - type < 0: use a builtin Keying Set
1128          */
1129         if (type == 0) 
1130                 type= scene->active_keyingset;
1131         if (type > 0)
1132                 ks= BLI_findlink(&scene->keyingsets, scene->active_keyingset-1);
1133         else
1134                 ks= BLI_findlink(&builtin_keyingsets, -type-1);
1135                 
1136         /* report failures */
1137         if (ks == NULL) {
1138                 BKE_report(op->reports, RPT_ERROR, "No active Keying Set");
1139                 return OPERATOR_CANCELLED;
1140         }
1141         
1142         /* try to insert keyframes for the channels specified by KeyingSet */
1143         success= ANIM_apply_keyingset(C, NULL, NULL, ks, MODIFYKEY_MODE_INSERT, cfra);
1144         if (G.f & G_DEBUG)
1145                 BKE_reportf(op->reports, RPT_INFO, "KeyingSet '%s' - Successfully added %d Keyframes \n", ks->name, success);
1146         
1147         /* report failure or do updates? */
1148         if (success == MODIFYKEY_INVALID_CONTEXT) {
1149                 BKE_report(op->reports, RPT_ERROR, "No suitable context info for active Keying Set");
1150                 return OPERATOR_CANCELLED;
1151         }
1152         else if (success) {
1153                 /* if the appropriate properties have been set, make a note that we've inserted something */
1154                 if (RNA_boolean_get(op->ptr, "confirm_success"))
1155                         BKE_reportf(op->reports, RPT_INFO, "Successfully added %d Keyframes for KeyingSet '%s'", success, ks->name);
1156                 
1157                 /* send notifiers that keyframes have been changed */
1158                 WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_EDITED, NULL);
1159         }
1160         else
1161                 BKE_report(op->reports, RPT_WARNING, "Keying Set failed to insert any keyframes");
1162         
1163         /* send updates */
1164         DAG_ids_flush_update(bmain, 0);
1165         
1166         return OPERATOR_FINISHED;
1167 }
1168
1169 void ANIM_OT_keyframe_insert (wmOperatorType *ot)
1170 {
1171         PropertyRNA *prop;
1172         
1173         /* identifiers */
1174         ot->name= "Insert Keyframe";
1175         ot->idname= "ANIM_OT_keyframe_insert";
1176         ot->description= "Insert keyframes on the current frame for all properties in the specified Keying Set";
1177         
1178         /* callbacks */
1179         ot->exec= insert_key_exec; 
1180         ot->poll= modify_key_op_poll;
1181         
1182         /* flags */
1183         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
1184         
1185         /* keyingset to use (dynamic enum) */
1186         prop= RNA_def_enum(ot->srna, "type", DummyRNA_DEFAULT_items, 0, "Keying Set", "The Keying Set to use");
1187         RNA_def_enum_funcs(prop, ANIM_keying_sets_enum_itemf);
1188         RNA_def_property_flag(prop, PROP_HIDDEN);
1189         ot->prop= prop;
1190         
1191         /* confirm whether a keyframe was added by showing a popup 
1192          *      - by default, this is enabled, since this operator is assumed to be called independently
1193          */
1194         prop= RNA_def_boolean(ot->srna, "confirm_success", 1, "Confirm Successful Insert", "Show a popup when the keyframes get successfully added");
1195         RNA_def_property_flag(prop, PROP_HIDDEN);
1196 }
1197
1198 /* Insert Key Operator (With Menu) ------------------------ */
1199 /* This operator checks if a menu should be shown for choosing the KeyingSet to use, 
1200  * then calls the menu if necessary before 
1201  */
1202
1203 static int insert_key_menu_invoke (bContext *C, wmOperator *op, wmEvent *UNUSED(event))
1204 {
1205         Scene *scene= CTX_data_scene(C);
1206         
1207         /* if prompting or no active Keying Set, show the menu */
1208         if ((scene->active_keyingset == 0) || RNA_boolean_get(op->ptr, "always_prompt")) {
1209                 /* call the menu, which will call this operator again, hence the cancelled */
1210                 ANIM_keying_sets_menu_setup(C, op->type->name, "ANIM_OT_keyframe_insert_menu");
1211                 return OPERATOR_CANCELLED;
1212         }
1213         else {
1214                 /* just call the exec() on the active keyingset */
1215                 RNA_enum_set(op->ptr, "type", 0);
1216                 RNA_boolean_set(op->ptr, "confirm_success", 1);
1217                 
1218                 return op->type->exec(C, op);
1219         }
1220 }
1221  
1222 void ANIM_OT_keyframe_insert_menu (wmOperatorType *ot)
1223 {
1224         PropertyRNA *prop;
1225         
1226         /* identifiers */
1227         ot->name= "Insert Keyframe Menu";
1228         ot->idname= "ANIM_OT_keyframe_insert_menu";
1229         ot->description= "Insert Keyframes for specified Keying Set, with menu of available Keying Sets if undefined";
1230         
1231         /* callbacks */
1232         ot->invoke= insert_key_menu_invoke;
1233         ot->exec= insert_key_exec; 
1234         ot->poll= ED_operator_areaactive;
1235         
1236         /* flags */
1237         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
1238         
1239         /* keyingset to use (dynamic enum) */
1240         prop= RNA_def_enum(ot->srna, "type", DummyRNA_DEFAULT_items, 0, "Keying Set", "The Keying Set to use");
1241         RNA_def_enum_funcs(prop, ANIM_keying_sets_enum_itemf);
1242         RNA_def_property_flag(prop, PROP_HIDDEN);
1243         ot->prop= prop;
1244         
1245         /* confirm whether a keyframe was added by showing a popup 
1246          *      - by default, this is disabled so that if a menu is shown, this doesn't come up too
1247          */
1248         // XXX should this just be always on?
1249         prop= RNA_def_boolean(ot->srna, "confirm_success", 0, "Confirm Successful Insert", "Show a popup when the keyframes get successfully added");
1250         RNA_def_property_flag(prop, PROP_HIDDEN);
1251         
1252         /* whether the menu should always be shown 
1253          *      - by default, the menu should only be shown when there is no active Keying Set (2.5 behaviour),
1254          *        although in some cases it might be useful to always shown (pre 2.5 behaviour)
1255          */
1256         prop= RNA_def_boolean(ot->srna, "always_prompt", 0, "Always Show Menu", "");
1257         RNA_def_property_flag(prop, PROP_HIDDEN);
1258 }
1259
1260 /* Delete Key Operator ------------------------ */
1261
1262 static int delete_key_exec (bContext *C, wmOperator *op)
1263 {
1264         Main *bmain= CTX_data_main(C);
1265         Scene *scene= CTX_data_scene(C);
1266         KeyingSet *ks= NULL;    
1267         int type= RNA_enum_get(op->ptr, "type");
1268         float cfra= (float)CFRA; // XXX for now, don't bother about all the yucky offset crap
1269         short success;
1270         
1271         /* type is the Keying Set the user specified to use when calling the operator:
1272          *      - type == 0: use scene's active Keying Set
1273          *      - type > 0: use a user-defined Keying Set from the active scene
1274          *      - type < 0: use a builtin Keying Set
1275          */
1276         if (type == 0) 
1277                 type= scene->active_keyingset;
1278         if (type > 0)
1279                 ks= BLI_findlink(&scene->keyingsets, scene->active_keyingset-1);
1280         else
1281                 ks= BLI_findlink(&builtin_keyingsets, -type-1);
1282         
1283         /* report failure */
1284         if (ks == NULL) {
1285                 BKE_report(op->reports, RPT_ERROR, "No active Keying Set");
1286                 return OPERATOR_CANCELLED;
1287         }
1288         
1289         /* try to insert keyframes for the channels specified by KeyingSet */
1290         success= ANIM_apply_keyingset(C, NULL, NULL, ks, MODIFYKEY_MODE_DELETE, cfra);
1291         if (G.f & G_DEBUG)
1292                 printf("KeyingSet '%s' - Successfully removed %d Keyframes \n", ks->name, success);
1293         
1294         /* report failure or do updates? */
1295         if (success == MODIFYKEY_INVALID_CONTEXT) {
1296                 BKE_report(op->reports, RPT_ERROR, "No suitable context info for active Keying Set");
1297                 return OPERATOR_CANCELLED;
1298         }
1299         else if (success) {
1300                 /* if the appropriate properties have been set, make a note that we've inserted something */
1301                 if (RNA_boolean_get(op->ptr, "confirm_success"))
1302                         BKE_reportf(op->reports, RPT_INFO, "Successfully removed %d Keyframes for KeyingSet '%s'", success, ks->name);
1303                 
1304                 /* send notifiers that keyframes have been changed */
1305                 WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_EDITED, NULL);
1306         }
1307         else
1308                 BKE_report(op->reports, RPT_WARNING, "Keying Set failed to remove any keyframes");
1309         
1310         /* send updates */
1311         DAG_ids_flush_update(bmain, 0);
1312         
1313         return OPERATOR_FINISHED;
1314 }
1315
1316 void ANIM_OT_keyframe_delete (wmOperatorType *ot)
1317 {
1318         PropertyRNA *prop;
1319         
1320         /* identifiers */
1321         ot->name= "Delete Keying-Set Keyframe";
1322         ot->idname= "ANIM_OT_keyframe_delete";
1323         ot->description= "Delete keyframes on the current frame for all properties in the specified Keying Set";
1324         
1325         /* callbacks */
1326         ot->exec= delete_key_exec; 
1327         ot->poll= modify_key_op_poll;
1328         
1329         /* flags */
1330         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
1331         
1332         /* keyingset to use (dynamic enum) */
1333         prop= RNA_def_enum(ot->srna, "type", DummyRNA_DEFAULT_items, 0, "Keying Set", "The Keying Set to use");
1334         RNA_def_enum_funcs(prop, ANIM_keying_sets_enum_itemf);
1335         RNA_def_property_flag(prop, PROP_HIDDEN);
1336         ot->prop= prop;
1337         
1338         /* confirm whether a keyframe was added by showing a popup 
1339          *      - by default, this is enabled, since this operator is assumed to be called independently
1340          */
1341         RNA_def_boolean(ot->srna, "confirm_success", 1, "Confirm Successful Insert", "Show a popup when the keyframes get successfully added");
1342 }
1343
1344 /* Delete Key Operator ------------------------ */
1345
1346 /* XXX WARNING:
1347  * This is currently just a basic operator, which work in 3d-view context on objects only. 
1348  * Should this be kept? It does have advantages over a version which requires selecting a keyingset to use...
1349  * -- Joshua Leung, Jan 2009
1350  */
1351  
1352 static int delete_key_v3d_exec (bContext *C, wmOperator *op)
1353 {
1354         Main *bmain= CTX_data_main(C);
1355         Scene *scene= CTX_data_scene(C);
1356         float cfra= (float)CFRA; // XXX for now, don't bother about all the yucky offset crap
1357         
1358         // XXX more comprehensive tests will be needed
1359         CTX_DATA_BEGIN(C, Object*, ob, selected_objects) 
1360         {
1361                 ID *id= (ID *)ob;
1362                 FCurve *fcu, *fcn;
1363                 short success= 0;
1364                 
1365                 /* loop through all curves in animdata and delete keys on this frame */
1366                 if ((ob->adt) && (ob->adt->action)) {
1367                         AnimData *adt= ob->adt;
1368                         bAction *act= adt->action;
1369                         
1370                         for (fcu= act->curves.first; fcu; fcu= fcn) {
1371                                 fcn= fcu->next;
1372                                 success+= delete_keyframe(op->reports, id, NULL, NULL, fcu->rna_path, fcu->array_index, cfra, 0);
1373                         }
1374                 }
1375                 
1376                 BKE_reportf(op->reports, RPT_INFO, "Ob '%s' - Successfully had %d keyframes removed", id->name+2, success);
1377                 
1378                 ob->recalc |= OB_RECALC_OB;
1379         }
1380         CTX_DATA_END;
1381         
1382         /* send updates */
1383         DAG_ids_flush_update(bmain, 0);
1384         
1385         WM_event_add_notifier(C, NC_OBJECT|ND_KEYS, NULL);
1386         
1387         return OPERATOR_FINISHED;
1388 }
1389
1390 void ANIM_OT_keyframe_delete_v3d (wmOperatorType *ot)
1391 {
1392         /* identifiers */
1393         ot->name= "Delete Keyframe";
1394         ot->description= "Remove keyframes on current frame for selected object";
1395         ot->idname= "ANIM_OT_keyframe_delete_v3d";
1396         
1397         /* callbacks */
1398         ot->invoke= WM_operator_confirm;
1399         ot->exec= delete_key_v3d_exec; 
1400         
1401         ot->poll= ED_operator_areaactive;
1402         
1403         /* flags */
1404         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
1405 }
1406
1407
1408 /* Insert Key Button Operator ------------------------ */
1409
1410 static int insert_key_button_exec (bContext *C, wmOperator *op)
1411 {
1412         Main *bmain= CTX_data_main(C);
1413         Scene *scene= CTX_data_scene(C);
1414         PointerRNA ptr= {{NULL}};
1415         PropertyRNA *prop= NULL;
1416         char *path;
1417         float cfra= (float)CFRA; // XXX for now, don't bother about all the yucky offset crap
1418         short success= 0;
1419         int a, index, length, all= RNA_boolean_get(op->ptr, "all");
1420         short flag = 0;
1421         
1422         /* flags for inserting keyframes */
1423         flag = ANIM_get_keyframing_flags(scene, 1);
1424         
1425         /* try to insert keyframe using property retrieved from UI */
1426         uiContextActiveProperty(C, &ptr, &prop, &index);
1427         
1428         if ((ptr.id.data && ptr.data && prop) && RNA_property_animateable(&ptr, prop)) {
1429                 path= RNA_path_from_ID_to_property(&ptr, prop);
1430                 
1431                 if (path) {
1432                         if (all) {
1433                                 length= RNA_property_array_length(&ptr, prop);
1434                                 
1435                                 if(length) index= 0;
1436                                 else length= 1;
1437                         }
1438                         else
1439                                 length= 1;
1440                         
1441                         for (a=0; a<length; a++)
1442                                 success+= insert_keyframe(op->reports, ptr.id.data, NULL, NULL, path, index+a, cfra, flag);
1443                         
1444                         MEM_freeN(path);
1445                 }
1446                 else if (ptr.type == &RNA_NlaStrip) {
1447                         /* handle special vars for NLA-strips */
1448                         NlaStrip *strip= (NlaStrip *)ptr.data;
1449                         FCurve *fcu= list_find_fcurve(&strip->fcurves, RNA_property_identifier(prop), flag);
1450                         
1451                         success+= insert_keyframe_direct(op->reports, ptr, prop, fcu, cfra, 0);
1452                 }
1453                 else {
1454                         if (G.f & G_DEBUG)
1455                                 printf("Button Insert-Key: no path to property \n");
1456                         BKE_report(op->reports, RPT_WARNING, "Failed to resolve path to property. Try using a Keying Set instead.");
1457                 }
1458         }
1459         else if (G.f & G_DEBUG) {
1460                 printf("ptr.data = %p, prop = %p,", (void *)ptr.data, (void *)prop);
1461                 if (prop)
1462                         printf("animateable = %d \n", RNA_property_animateable(&ptr, prop));
1463                 else
1464                         printf("animateable = NULL \n");
1465         }
1466         
1467         if (success) {
1468                 /* send updates */
1469                 uiContextAnimUpdate(C);
1470                 
1471                 DAG_ids_flush_update(bmain, 0);
1472                 
1473                 /* send notifiers that keyframes have been changed */
1474                 WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_EDITED, NULL);
1475         }
1476         
1477         return (success)? OPERATOR_FINISHED: OPERATOR_CANCELLED;
1478 }
1479
1480 void ANIM_OT_keyframe_insert_button (wmOperatorType *ot)
1481 {
1482         /* identifiers */
1483         ot->name= "Insert Keyframe (Buttons)";
1484         ot->idname= "ANIM_OT_keyframe_insert_button";
1485         
1486         /* callbacks */
1487         ot->exec= insert_key_button_exec; 
1488         ot->poll= modify_key_op_poll;
1489         
1490         /* flags */
1491         ot->flag= OPTYPE_UNDO;
1492
1493         /* properties */
1494         RNA_def_boolean(ot->srna, "all", 1, "All", "Insert a keyframe for all element of the array.");
1495 }
1496
1497 /* Delete Key Button Operator ------------------------ */
1498
1499 static int delete_key_button_exec (bContext *C, wmOperator *op)
1500 {
1501         Main *bmain= CTX_data_main(C);
1502         Scene *scene= CTX_data_scene(C);
1503         PointerRNA ptr= {{NULL}};
1504         PropertyRNA *prop= NULL;
1505         char *path;
1506         float cfra= (float)CFRA; // XXX for now, don't bother about all the yucky offset crap
1507         short success= 0;
1508         int a, index, length, all= RNA_boolean_get(op->ptr, "all");
1509         
1510         /* try to insert keyframe using property retrieved from UI */
1511         uiContextActiveProperty(C, &ptr, &prop, &index);
1512
1513         if (ptr.id.data && ptr.data && prop) {
1514                 path= RNA_path_from_ID_to_property(&ptr, prop);
1515                 
1516                 if (path) {
1517                         if (all) {
1518                                 length= RNA_property_array_length(&ptr, prop);
1519                                 
1520                                 if (length) index= 0;
1521                                 else length= 1;
1522                         }
1523                         else
1524                                 length= 1;
1525                         
1526                         for (a=0; a<length; a++)
1527                                 success+= delete_keyframe(op->reports, ptr.id.data, NULL, NULL, path, index+a, cfra, 0);
1528                         
1529                         MEM_freeN(path);
1530                 }
1531                 else if (G.f & G_DEBUG)
1532                         printf("Button Delete-Key: no path to property \n");
1533         }
1534         else if (G.f & G_DEBUG) {
1535                 printf("ptr.data = %p, prop = %p \n", (void *)ptr.data, (void *)prop);
1536         }
1537         
1538         
1539         if (success) {
1540                 /* send updates */
1541                 uiContextAnimUpdate(C);
1542                 
1543                 DAG_ids_flush_update(bmain, 0);
1544                 
1545                 /* send notifiers that keyframes have been changed */
1546                 WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_EDITED, NULL);
1547         }
1548         
1549         return (success)? OPERATOR_FINISHED: OPERATOR_CANCELLED;
1550 }
1551
1552 void ANIM_OT_keyframe_delete_button (wmOperatorType *ot)
1553 {
1554         /* identifiers */
1555         ot->name= "Delete Keyframe (Buttons)";
1556         ot->idname= "ANIM_OT_keyframe_delete_button";
1557         
1558         /* callbacks */
1559         ot->exec= delete_key_button_exec; 
1560         ot->poll= modify_key_op_poll;
1561         
1562         /* flags */
1563         ot->flag= OPTYPE_UNDO;
1564
1565         /* properties */
1566         RNA_def_boolean(ot->srna, "all", 1, "All", "Delete keyfames from all elements of the array.");
1567 }
1568
1569 /* ******************************************* */
1570 /* AUTO KEYFRAME */
1571
1572 int autokeyframe_cfra_can_key(Scene *scene, ID *id)
1573 {
1574         float cfra= (float)CFRA; // XXX for now, this will do
1575         
1576         /* only filter if auto-key mode requires this */
1577         if (IS_AUTOKEY_ON(scene) == 0)
1578                 return 0;
1579                 
1580         if (IS_AUTOKEY_MODE(scene, NORMAL)) {
1581                 /* can insert anytime we like... */
1582                 return 1;
1583         }
1584         else /* REPLACE */ {
1585                 /* for whole block - only key if there's a keyframe on that frame already
1586                  *      this is a valid assumption when we're blocking + tweaking
1587                  */
1588                 return id_frame_has_keyframe(id, cfra, ANIMFILTER_KEYS_LOCAL);
1589         }
1590 }
1591
1592 /* ******************************************* */
1593 /* KEYFRAME DETECTION */
1594
1595 /* --------------- API/Per-Datablock Handling ------------------- */
1596
1597 /* Checks if some F-Curve has a keyframe for a given frame */
1598 short fcurve_frame_has_keyframe (FCurve *fcu, float frame, short filter)
1599 {
1600         /* quick sanity check */
1601         if (ELEM(NULL, fcu, fcu->bezt))
1602                 return 0;
1603         
1604         /* we either include all regardless of muting, or only non-muted  */
1605         if ((filter & ANIMFILTER_KEYS_MUTED) || (fcu->flag & FCURVE_MUTED)==0) {
1606                 short replace = -1;
1607                 int i = binarysearch_bezt_index(fcu->bezt, frame, fcu->totvert, &replace);
1608                 
1609                 /* binarysearch_bezt_index will set replace to be 0 or 1
1610                  *      - obviously, 1 represents a match
1611                  */
1612                 if (replace) {                  
1613                         /* sanity check: 'i' may in rare cases exceed arraylen */
1614                         if ((i >= 0) && (i < fcu->totvert))
1615                                 return 1;
1616                 }
1617         }
1618         
1619         return 0;
1620 }
1621
1622 /* Checks whether an Action has a keyframe for a given frame 
1623  * Since we're only concerned whether a keyframe exists, we can simply loop until a match is found...
1624  */
1625 static short action_frame_has_keyframe (bAction *act, float frame, short filter)
1626 {
1627         FCurve *fcu;
1628         
1629         /* can only find if there is data */
1630         if (act == NULL)
1631                 return 0;
1632                 
1633         /* if only check non-muted, check if muted */
1634         if ((filter & ANIMFILTER_KEYS_MUTED) || (act->flag & ACT_MUTED))
1635                 return 0;
1636         
1637         /* loop over F-Curves, using binary-search to try to find matches 
1638          *      - this assumes that keyframes are only beztriples
1639          */
1640         for (fcu= act->curves.first; fcu; fcu= fcu->next) {
1641                 /* only check if there are keyframes (currently only of type BezTriple) */
1642                 if (fcu->bezt && fcu->totvert) {
1643                         if (fcurve_frame_has_keyframe(fcu, frame, filter))
1644                                 return 1;
1645                 }
1646         }
1647         
1648         /* nothing found */
1649         return 0;
1650 }
1651
1652 /* Checks whether an Object has a keyframe for a given frame */
1653 static short object_frame_has_keyframe (Object *ob, float frame, short filter)
1654 {
1655         /* error checking */
1656         if (ob == NULL)
1657                 return 0;
1658         
1659         /* check own animation data - specifically, the action it contains */
1660         if ((ob->adt) && (ob->adt->action)) {
1661                 if (action_frame_has_keyframe(ob->adt->action, frame, filter))
1662                         return 1;
1663         }
1664         
1665         /* try shapekey keyframes (if available, and allowed by filter) */
1666         if ( !(filter & ANIMFILTER_KEYS_LOCAL) && !(filter & ANIMFILTER_KEYS_NOSKEY) ) {
1667                 Key *key= ob_get_key(ob);
1668                 
1669                 /* shapekeys can have keyframes ('Relative Shape Keys') 
1670                  * or depend on time (old 'Absolute Shape Keys') 
1671                  */
1672                  
1673                         /* 1. test for relative (with keyframes) */
1674                 if (id_frame_has_keyframe((ID *)key, frame, filter))
1675                         return 1;
1676                         
1677                         /* 2. test for time */
1678                 // TODO... yet to be implemented (this feature may evolve before then anyway)
1679         }
1680         
1681         /* try materials */
1682         if ( !(filter & ANIMFILTER_KEYS_LOCAL) && !(filter & ANIMFILTER_KEYS_NOMAT) ) {
1683                 /* if only active, then we can skip a lot of looping */
1684                 if (filter & ANIMFILTER_KEYS_ACTIVE) {
1685                         Material *ma= give_current_material(ob, (ob->actcol + 1));
1686                         
1687                         /* we only retrieve the active material... */
1688                         if (id_frame_has_keyframe((ID *)ma, frame, filter))
1689                                 return 1;
1690                 }
1691                 else {
1692                         int a;
1693                         
1694                         /* loop over materials */
1695                         for (a=0; a<ob->totcol; a++) {
1696                                 Material *ma= give_current_material(ob, a+1);
1697                                 
1698                                 if (id_frame_has_keyframe((ID *)ma, frame, filter))
1699                                         return 1;
1700                         }
1701                 }
1702         }
1703         
1704         /* nothing found */
1705         return 0;
1706 }
1707
1708 /* --------------- API ------------------- */
1709
1710 /* Checks whether a keyframe exists for the given ID-block one the given frame */
1711 short id_frame_has_keyframe (ID *id, float frame, short filter)
1712 {
1713         /* sanity checks */
1714         if (id == NULL)
1715                 return 0;
1716         
1717         /* perform special checks for 'macro' types */
1718         switch (GS(id->name)) {
1719                 case ID_OB: /* object */
1720                         return object_frame_has_keyframe((Object *)id, frame, filter);
1721                         break;
1722                         
1723                 case ID_SCE: /* scene */
1724                 // XXX TODO... for now, just use 'normal' behaviour
1725                 //      break;
1726                 
1727                 default:        /* 'normal type' */
1728                 {
1729                         AnimData *adt= BKE_animdata_from_id(id);
1730                         
1731                         /* only check keyframes in active action */
1732                         if (adt)
1733                                 return action_frame_has_keyframe(adt->action, frame, filter);
1734                 }
1735                         break;
1736         }
1737         
1738         
1739         /* no keyframe found */
1740         return 0;
1741 }
1742
1743 /* ************************************************** */