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