keyframing.c

Go to the documentation of this file.

/*
 * ***** BEGIN GPL LICENSE BLOCK *****
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 *
 * The Original Code is Copyright (C) 2009 Blender Foundation, Joshua Leung
 * All rights reserved.
 *
 * The Original Code is: all of this file.
 *
 * Contributor(s): Joshua Leung (full recode)
 *
 * ***** END GPL LICENSE BLOCK *****
 */

#include <stdio.h>
#include <stddef.h>
#include <string.h>
#include <math.h>
#include <float.h>

#include "MEM_guardedalloc.h"

#include "BLI_blenlib.h"
#include "BLI_math.h"
#include "BLI_dynstr.h"
#include "BLI_utildefines.h"

#include "DNA_anim_types.h"
#include "DNA_armature_types.h"
#include "DNA_constraint_types.h"
#include "DNA_key_types.h"
#include "DNA_material_types.h"
#include "DNA_scene_types.h"
#include "DNA_object_types.h"

#include "BKE_animsys.h"
#include "BKE_action.h"
#include "BKE_armature.h"
#include "BKE_constraint.h"
#include "BKE_depsgraph.h"
#include "BKE_fcurve.h"
#include "BKE_main.h"
#include "BKE_nla.h"
#include "BKE_global.h"
#include "BKE_context.h"
#include "BKE_report.h"
#include "BKE_key.h"
#include "BKE_material.h"

#include "ED_anim_api.h"
#include "ED_keyframing.h"
#include "ED_keyframes_edit.h"
#include "ED_screen.h"

#include "UI_interface.h"

#include "WM_api.h"
#include "WM_types.h"

#include "RNA_access.h"
#include "RNA_define.h"
#include "RNA_enum_types.h"

#include "anim_intern.h"

/* ************************************************** */
/* Keyframing Setting Wrangling */

/* Get the active settings for keyframing settings from context (specifically the given scene) */
short ANIM_get_keyframing_flags (Scene *scene, short incl_mode)
{
    short flag = 0;
    
    /* standard flags */
    {
        /* visual keying */
        if (IS_AUTOKEY_FLAG(scene, AUTOMATKEY)) 
            flag |= INSERTKEY_MATRIX;
        
        /* only needed */
        if (IS_AUTOKEY_FLAG(scene, INSERTNEEDED)) 
            flag |= INSERTKEY_NEEDED;
        
        /* default F-Curve color mode - RGB from XYZ indices */
        if (IS_AUTOKEY_FLAG(scene, XYZ2RGB)) 
            flag |= INSERTKEY_XYZ2RGB;
    }
        
    /* only if including settings from the autokeying mode... */
    if (incl_mode) 
    { 
        /* keyframing mode - only replace existing keyframes */
        if (IS_AUTOKEY_MODE(scene, EDITKEYS)) 
            flag |= INSERTKEY_REPLACE;
    }
        
    return flag;
}

/* ******************************************* */
/* Animation Data Validation */

/* Get (or add relevant data to be able to do so) the Active Action for the given 
 * Animation Data block, given an ID block where the Animation Data should reside.
 */
bAction *verify_adt_action (ID *id, short add)
{
    AnimData *adt;
    
    /* init animdata if none available yet */
    adt= BKE_animdata_from_id(id);
    if ((adt == NULL) && (add))
        adt= BKE_id_add_animdata(id);
    if (adt == NULL) { 
        /* if still none (as not allowed to add, or ID doesn't have animdata for some reason) */
        printf("ERROR: Couldn't add AnimData (ID = %s) \n", (id) ? (id->name) : "<None>");
        return NULL;
    }
        
    /* init action if none available yet */
    // TODO: need some wizardry to handle NLA stuff correct
    if ((adt->action == NULL) && (add)) {
        char actname[sizeof(id->name)-2];
        BLI_snprintf(actname, sizeof(actname), "%sAction", id->name+2);
        adt->action= add_empty_action(actname);
    }
        
    /* return the action */
    return adt->action;
}

/* Get (or add relevant data to be able to do so) F-Curve from the Active Action, 
 * for the given Animation Data block. This assumes that all the destinations are valid.
 */
FCurve *verify_fcurve (bAction *act, const char group[], const char rna_path[], const int array_index, short add)
{
    bActionGroup *grp;
    FCurve *fcu;
    
    /* sanity checks */
    if ELEM(NULL, act, rna_path)
        return NULL;
        
    /* try to find f-curve matching for this setting 
     *  - add if not found and allowed to add one
     *      TODO: add auto-grouping support? how this works will need to be resolved
     */
    if (act)
        fcu= list_find_fcurve(&act->curves, rna_path, array_index);
    else
        fcu= NULL;
    
    if ((fcu == NULL) && (add)) {
        /* use default settings to make a F-Curve */
        fcu= MEM_callocN(sizeof(FCurve), "FCurve");
        
        fcu->flag = (FCURVE_VISIBLE|FCURVE_SELECTED);
        if (act->curves.first==NULL) 
            fcu->flag |= FCURVE_ACTIVE; /* first one added active */
            
        /* store path - make copy, and store that */
        fcu->rna_path= BLI_strdupn(rna_path, strlen(rna_path));
        fcu->array_index= array_index;
        
        /* if a group name has been provided, try to add or find a group, then add F-Curve to it */
        if (group) {
            /* try to find group */
            grp= action_groups_find_named(act, group);
            
            /* no matching groups, so add one */
            if (grp == NULL)
                grp= action_groups_add_new(act, group);
            
            /* add F-Curve to group */
            action_groups_add_channel(act, grp, fcu);
        }
        else {
            /* just add F-Curve to end of Action's list */
            BLI_addtail(&act->curves, fcu);
        }
    }
    
    /* return the F-Curve */
    return fcu;
}

/* ************************************************** */
/* KEYFRAME INSERTION */

/* -------------- BezTriple Insertion -------------------- */

/* This function adds a given BezTriple to an F-Curve. It will allocate 
 * memory for the array if needed, and will insert the BezTriple into a
 * suitable place in chronological order.
 * 
 * NOTE: any recalculate of the F-Curve that needs to be done will need to 
 *      be done by the caller.
 */
int insert_bezt_fcurve (FCurve *fcu, BezTriple *bezt, short flag)
{
    int i= 0;
    
    /* are there already keyframes? */
    if (fcu->bezt) {
        short replace = -1;
        i = binarysearch_bezt_index(fcu->bezt, bezt->vec[1][0], fcu->totvert, &replace);
        
        /* replace an existing keyframe? */
        if (replace) {          
            /* sanity check: 'i' may in rare cases exceed arraylen */
            if ((i >= 0) && (i < fcu->totvert)) {
                /* just change the values when replacing, so as to not overwrite handles */
                BezTriple *dst= (fcu->bezt + i);
                float dy= bezt->vec[1][1] - dst->vec[1][1];
                
                /* just apply delta value change to the handle values */
                dst->vec[0][1] += dy;
                dst->vec[1][1] += dy;
                dst->vec[2][1] += dy;
                
                dst->f1= bezt->f1;
                dst->f2= bezt->f2;
                dst->f3= bezt->f3;
                
                // TODO: perform some other operations?
            }
        }
        /* keyframing modes allow to not replace keyframe */
        else if ((flag & INSERTKEY_REPLACE) == 0) {
            /* insert new - if we're not restricted to replacing keyframes only */
            BezTriple *newb= MEM_callocN((fcu->totvert+1)*sizeof(BezTriple), "beztriple");
            
            /* add the beztriples that should occur before the beztriple to be pasted (originally in fcu) */
            if (i > 0)
                memcpy(newb, fcu->bezt, i*sizeof(BezTriple));
            
            /* add beztriple to paste at index i */
            *(newb + i)= *bezt;
            
            /* add the beztriples that occur after the beztriple to be pasted (originally in fcu) */
            if (i < fcu->totvert) 
                memcpy(newb+i+1, fcu->bezt+i, (fcu->totvert-i)*sizeof(BezTriple));
            
            /* replace (+ free) old with new, only if necessary to do so */
            MEM_freeN(fcu->bezt);
            fcu->bezt= newb;
            
            fcu->totvert++;
        }
    }
    /* no keyframes already, but can only add if...
     *  1) keyframing modes say that keyframes can only be replaced, so adding new ones won't know
     *  2) there are no samples on the curve
     *      // NOTE: maybe we may want to allow this later when doing samples -> bezt conversions, 
     *      // but for now, having both is asking for trouble
     */
    else if ((flag & INSERTKEY_REPLACE)==0 && (fcu->fpt==NULL)) {
        /* create new keyframes array */
        fcu->bezt= MEM_callocN(sizeof(BezTriple), "beztriple");
        *(fcu->bezt)= *bezt;
        fcu->totvert= 1;
    }
    /* cannot add anything */
    else {
        /* return error code -1 to prevent any misunderstandings */
        return -1;
    }
    
    
    /* we need to return the index, so that some tools which do post-processing can 
     * detect where we added the BezTriple in the array
     */
    return i;
}

/* This function is a wrapper for insert_bezt_fcurve_internal(), and should be used when
 * adding a new keyframe to a curve, when the keyframe doesn't exist anywhere else yet. 
 * It returns the index at which the keyframe was added.
 */
int insert_vert_fcurve (FCurve *fcu, float x, float y, short flag)
{
    BezTriple beztr= {{{0}}};
    unsigned int oldTot = fcu->totvert;
    int a;
    
    /* set all three points, for nicer start position 
     * NOTE: +/- 1 on vec.x for left and right handles is so that 'free' handles work ok...
     */
    beztr.vec[0][0]= x-1.0f; 
    beztr.vec[0][1]= y;
    beztr.vec[1][0]= x;
    beztr.vec[1][1]= y;
    beztr.vec[2][0]= x+1.0f;
    beztr.vec[2][1]= y;
    beztr.f1= beztr.f2= beztr.f3= SELECT;
    beztr.h1= beztr.h2= U.keyhandles_new; /* use default handle type here */
    //BEZKEYTYPE(&beztr)= scene->keytype; /* default keyframe type */

    /* use default interpolation mode, with exceptions for int/discrete values */
    beztr.ipo= U.ipo_new;

    if(fcu->flag & FCURVE_DISCRETE_VALUES)
        beztr.ipo = BEZT_IPO_CONST;
    else if(beztr.ipo == BEZT_IPO_BEZ && (fcu->flag & FCURVE_INT_VALUES))
        beztr.ipo = BEZT_IPO_LIN;
    
    /* add temp beztriple to keyframes */
    a= insert_bezt_fcurve(fcu, &beztr, flag);
    
    /* what if 'a' is a negative index? 
     * for now, just exit to prevent any segfaults
     */
    if (a < 0) return -1;
    
    /* don't recalculate handles if fast is set
     *  - this is a hack to make importers faster
     *  - we may calculate twice (due to autohandle needing to be calculated twice)
     */
    if ((flag & INSERTKEY_FAST) == 0) 
        calchandles_fcurve(fcu);
    
    /* set handletype and interpolation */
    if ((fcu->totvert > 2) && (flag & INSERTKEY_REPLACE)==0) {
        BezTriple *bezt= (fcu->bezt + a);
        
        /* set interpolation from previous (if available), but only if we didn't just replace some keyframe 
         *  - replacement is indicated by no-change in number of verts
         *  - when replacing, the user may have specified some interpolation that should be kept
         */
        if (fcu->totvert > oldTot) {
            if (a > 0) 
                bezt->ipo= (bezt-1)->ipo;
            else if (a < fcu->totvert-1) 
                bezt->ipo= (bezt+1)->ipo;
        }
            
        /* don't recalculate handles if fast is set
         *  - this is a hack to make importers faster
         *  - we may calculate twice (due to autohandle needing to be calculated twice)
         */
        if ((flag & INSERTKEY_FAST) == 0) 
            calchandles_fcurve(fcu);
    }
    
    /* return the index at which the keyframe was added */
    return a;
}

/* -------------- 'Smarter' Keyframing Functions -------------------- */
/* return codes for new_key_needed */
enum {
    KEYNEEDED_DONTADD = 0,
    KEYNEEDED_JUSTADD,
    KEYNEEDED_DELPREV,
    KEYNEEDED_DELNEXT
} /*eKeyNeededStatus*/;

/* This helper function determines whether a new keyframe is needed */
/* Cases where keyframes should not be added:
 *  1. Keyframe to be added bewteen two keyframes with similar values
 *  2. Keyframe to be added on frame where two keyframes are already situated
 *  3. Keyframe lies at point that intersects the linear line between two keyframes
 */
static short new_key_needed (FCurve *fcu, float cFrame, float nValue) 
{
    BezTriple *bezt=NULL, *prev=NULL;
    int totCount, i;
    float valA = 0.0f, valB = 0.0f;
    
    /* safety checking */
    if (fcu == NULL) return KEYNEEDED_JUSTADD;
    totCount= fcu->totvert;
    if (totCount == 0) return KEYNEEDED_JUSTADD;
    
    /* loop through checking if any are the same */
    bezt= fcu->bezt;
    for (i=0; i<totCount; i++) {
        float prevPosi=0.0f, prevVal=0.0f;
        float beztPosi=0.0f, beztVal=0.0f;
            
        /* get current time+value */    
        beztPosi= bezt->vec[1][0];
        beztVal= bezt->vec[1][1];
            
        if (prev) {
            /* there is a keyframe before the one currently being examined */       
            
            /* get previous time+value */
            prevPosi= prev->vec[1][0];
            prevVal= prev->vec[1][1];
            
            /* keyframe to be added at point where there are already two similar points? */
            if (IS_EQF(prevPosi, cFrame) && IS_EQF(beztPosi, cFrame) && IS_EQF(beztPosi, prevPosi)) {
                return KEYNEEDED_DONTADD;
            }
            
            /* keyframe between prev+current points ? */
            if ((prevPosi <= cFrame) && (cFrame <= beztPosi)) {
                /* is the value of keyframe to be added the same as keyframes on either side ? */
                if (IS_EQF(prevVal, nValue) && IS_EQF(beztVal, nValue) && IS_EQF(prevVal, beztVal)) {
                    return KEYNEEDED_DONTADD;
                }
                else {
                    float realVal;
                    
                    /* get real value of curve at that point */
                    realVal= evaluate_fcurve(fcu, cFrame);
                    
                    /* compare whether it's the same as proposed */
                    if (IS_EQF(realVal, nValue))
                        return KEYNEEDED_DONTADD;
                    else 
                        return KEYNEEDED_JUSTADD;
                }
            }
            
            /* new keyframe before prev beztriple? */
            if (cFrame < prevPosi) {
                /* A new keyframe will be added. However, whether the previous beztriple
                 * stays around or not depends on whether the values of previous/current
                 * beztriples and new keyframe are the same.
                 */
                if (IS_EQF(prevVal, nValue) && IS_EQF(beztVal, nValue) && IS_EQF(prevVal, beztVal))
                    return KEYNEEDED_DELNEXT;
                else 
                    return KEYNEEDED_JUSTADD;
            }
        }
        else {
            /* just add a keyframe if there's only one keyframe 
             * and the new one occurs before the exisiting one does.
             */
            if ((cFrame < beztPosi) && (totCount==1))
                return KEYNEEDED_JUSTADD;
        }
        
        /* continue. frame to do not yet passed (or other conditions not met) */
        if (i < (totCount-1)) {
            prev= bezt;
            bezt++;
        }
        else
            break;
    }
    
    /* Frame in which to add a new-keyframe occurs after all other keys
     * -> If there are at least two existing keyframes, then if the values of the
     *   last two keyframes and the new-keyframe match, the last existing keyframe
     *   gets deleted as it is no longer required.
     * -> Otherwise, a keyframe is just added. 1.0 is added so that fake-2nd-to-last
     *   keyframe is not equal to last keyframe.
     */
    bezt= (fcu->bezt + (fcu->totvert - 1));
    valA= bezt->vec[1][1];
    
    if (prev)
        valB= prev->vec[1][1];
    else 
        valB= bezt->vec[1][1] + 1.0f; 
        
    if (IS_EQF(valA, nValue) && IS_EQF(valA, valB))
        return KEYNEEDED_DELPREV;
    else 
        return KEYNEEDED_JUSTADD;
}

/* ------------------ RNA Data-Access Functions ------------------ */

/* Try to read value using RNA-properties obtained already */
static float setting_get_rna_value (PointerRNA *ptr, PropertyRNA *prop, int index)
{
    float value= 0.0f;
    
    switch (RNA_property_type(prop)) {
        case PROP_BOOLEAN:
            if (RNA_property_array_length(ptr, prop))
                value= (float)RNA_property_boolean_get_index(ptr, prop, index);
            else
                value= (float)RNA_property_boolean_get(ptr, prop);
            break;
        case PROP_INT:
            if (RNA_property_array_length(ptr, prop))
                value= (float)RNA_property_int_get_index(ptr, prop, index);
            else
                value= (float)RNA_property_int_get(ptr, prop);
            break;
        case PROP_FLOAT:
            if (RNA_property_array_length(ptr, prop))
                value= RNA_property_float_get_index(ptr, prop, index);
            else
                value= RNA_property_float_get(ptr, prop);
            break;
        case PROP_ENUM:
            value= (float)RNA_property_enum_get(ptr, prop);
            break;
        default:
            break;
    }
    
    return value;
}

/* ------------------ 'Visual' Keyframing Functions ------------------ */

/* internal status codes for visualkey_can_use */
enum {
    VISUALKEY_NONE = 0,
    VISUALKEY_LOC,
    VISUALKEY_ROT
    /* VISUALKEY_SCA */ /* TODO - looks like support can be added now */
};

/* This helper function determines if visual-keyframing should be used when  
 * inserting keyframes for the given channel. As visual-keyframing only works
 * on Object and Pose-Channel blocks, this should only get called for those 
 * blocktypes, when using "standard" keying but 'Visual Keying' option in Auto-Keying 
 * settings is on.
 */
static short visualkey_can_use (PointerRNA *ptr, PropertyRNA *prop)
{
    bConstraint *con= NULL;
    short searchtype= VISUALKEY_NONE;
    short has_parent = FALSE;
    const char *identifier= NULL;
    
    /* validate data */
    // TODO: this check is probably not needed, but it won't hurt
    if (ELEM3(NULL, ptr, ptr->data, prop))
        return 0;
        
    /* get first constraint and determine type of keyframe constraints to check for 
     *  - constraints can be on either Objects or PoseChannels, so we only check if the
     *    ptr->type is RNA_Object or RNA_PoseBone, which are the RNA wrapping-info for
     *    those structs, allowing us to identify the owner of the data
     */
    if (ptr->type == &RNA_Object) {
        /* Object */
        Object *ob= (Object *)ptr->data;
        
        con= ob->constraints.first;
        identifier= RNA_property_identifier(prop);
        has_parent= (ob->parent != NULL);
    }
    else if (ptr->type == &RNA_PoseBone) {
        /* Pose Channel */
        bPoseChannel *pchan= (bPoseChannel *)ptr->data;
        
        con= pchan->constraints.first;
        identifier= RNA_property_identifier(prop);
        has_parent= (pchan->parent != NULL);
    }
    
    /* check if any data to search using */
    if (ELEM(NULL, con, identifier) && (has_parent == FALSE))
        return 0;
        
    /* location or rotation identifiers only... */
    if(identifier == NULL) {
        printf("%s failed: NULL identifier\n", __func__);
        return 0;
    }
    else if (strstr(identifier, "location")) {
        searchtype= VISUALKEY_LOC;
    }
    else if (strstr(identifier, "rotation")) {
        searchtype= VISUALKEY_ROT;
    }
    else {
        printf("%s failed: identifier - '%s' \n", __func__, identifier);
        return 0;
    }
    
    
    /* only search if a searchtype and initial constraint are available */
    if (searchtype) {
        /* parent is always matching */
        if (has_parent)
            return 1;
        
        /* constraints */
        for (; con; con= con->next) {
            /* only consider constraint if it is not disabled, and has influence */
            if (con->flag & CONSTRAINT_DISABLE) continue;
            if (con->enforce == 0.0f) continue;
            
            /* some constraints may alter these transforms */
            switch (con->type) {
                /* multi-transform constraints */
                case CONSTRAINT_TYPE_CHILDOF:
                    return 1;
                case CONSTRAINT_TYPE_TRANSFORM:
                case CONSTRAINT_TYPE_TRANSLIKE:
                    return 1;
                case CONSTRAINT_TYPE_FOLLOWPATH:
                    return 1;
                case CONSTRAINT_TYPE_KINEMATIC:
                    return 1;
                    
                /* single-transform constraits  */
                case CONSTRAINT_TYPE_TRACKTO:
                    if (searchtype==VISUALKEY_ROT) return 1;
                    break;
                case CONSTRAINT_TYPE_ROTLIMIT:
                    if (searchtype==VISUALKEY_ROT) return 1;
                    break;
                case CONSTRAINT_TYPE_LOCLIMIT:
                    if (searchtype==VISUALKEY_LOC) return 1;
                    break;
                case CONSTRAINT_TYPE_ROTLIKE:
                    if (searchtype==VISUALKEY_ROT) return 1;
                    break;
                case CONSTRAINT_TYPE_DISTLIMIT:
                    if (searchtype==VISUALKEY_LOC) return 1;
                    break;
                case CONSTRAINT_TYPE_LOCLIKE:
                    if (searchtype==VISUALKEY_LOC) return 1;
                    break;
                case CONSTRAINT_TYPE_LOCKTRACK:
                    if (searchtype==VISUALKEY_ROT) return 1;
                    break;
                case CONSTRAINT_TYPE_MINMAX:
                    if (searchtype==VISUALKEY_LOC) return 1;
                    break;
                
                default:
                    break;
            }
        }
    }
    
    /* when some condition is met, this function returns, so here it can be 0 */
    return 0;
}

/* This helper function extracts the value to use for visual-keyframing 
 * In the event that it is not possible to perform visual keying, try to fall-back
 * to using the default method. Assumes that all data it has been passed is valid.
 */
static float visualkey_get_value (PointerRNA *ptr, PropertyRNA *prop, int array_index)
{
    const char *identifier= RNA_property_identifier(prop);
    
    /* handle for Objects or PoseChannels only 
     *  - constraints can be on either Objects or PoseChannels, so we only check if the
     *    ptr->type is RNA_Object or RNA_PoseBone, which are the RNA wrapping-info for
     *        those structs, allowing us to identify the owner of the data 
     *  - assume that array_index will be sane
     */
    if (ptr->type == &RNA_Object) {
        Object *ob= (Object *)ptr->data;
        
        /* only Location or Rotation keyframes are supported now */
        if (strstr(identifier, "location")) {
            return ob->obmat[3][array_index];
        }
        else if (strstr(identifier, "rotation_euler")) {
            float eul[3];
            
            mat4_to_eulO(eul, ob->rotmode, ob->obmat);
            return eul[array_index];
        }
        else if (strstr(identifier, "rotation_quaternion")) {
            float trimat[3][3], quat[4];
            
            copy_m3_m4(trimat, ob->obmat);
            mat3_to_quat_is_ok(quat, trimat);
            
            return quat[array_index];
        }
        else if (strstr(identifier, "rotation_axis_angle")) {
            float axis[3], angle;
            
            mat4_to_axis_angle(axis, &angle, ob->obmat);
            
            /* w = 0, x,y,z = 1,2,3 */
            if (array_index == 0)
                return angle;
            else
                return axis[array_index - 1];
        }
    }
    else if (ptr->type == &RNA_PoseBone) {
        Object *ob = (Object *)ptr->id.data; /* we assume that this is always set, and is an object */
        bPoseChannel *pchan= (bPoseChannel *)ptr->data;
        float tmat[4][4];
        
        /* Although it is not strictly required for this particular space conversion, 
         * arg1 must not be null, as there is a null check for the other conversions to
         * be safe. Therefore, the active object is passed here, and in many cases, this
         * will be what owns the pose-channel that is getting this anyway.
         */
        copy_m4_m4(tmat, pchan->pose_mat);
        constraint_mat_convertspace(ob, pchan, tmat, CONSTRAINT_SPACE_POSE, CONSTRAINT_SPACE_LOCAL);
        
        /* Loc, Rot/Quat keyframes are supported... */
        if (strstr(identifier, "location")) {
            /* only use for non-connected bones */
            if ((pchan->bone->parent) && !(pchan->bone->flag & BONE_CONNECTED))
                return tmat[3][array_index];
            else if (pchan->bone->parent == NULL)
                return tmat[3][array_index];
        }
        else if (strstr(identifier, "rotation_euler")) {
            float eul[3];
            
            mat4_to_eulO(eul, pchan->rotmode, tmat);
            return eul[array_index];
        }
        else if (strstr(identifier, "rotation_quaternion")) {
            float trimat[3][3], quat[4];
            
            copy_m3_m4(trimat, tmat);
            mat3_to_quat_is_ok(quat, trimat);
            
            return quat[array_index];
        }
        else if (strstr(identifier, "rotation_axis_angle")) {
            float axis[3], angle;
            
            mat4_to_axis_angle(axis, &angle, tmat);
            
            /* w = 0, x,y,z = 1,2,3 */
            if (array_index == 0)
                return angle;
            else
                return axis[array_index - 1];
        }
    }
    
    /* as the function hasn't returned yet, read value from system in the default way */
    return setting_get_rna_value(ptr, prop, array_index);
}

/* ------------------------- Insert Key API ------------------------- */

/* Secondary Keyframing API call: 
 *  Use this when validation of necessary animation data is not necessary, since an RNA-pointer to the necessary
 *  data being keyframed, and a pointer to the F-Curve to use have both been provided.
 *
 *  The flag argument is used for special settings that alter the behaviour of
 *  the keyframe insertion. These include the 'visual' keyframing modes, quick refresh,
 *  and extra keyframe filtering.
 */
short insert_keyframe_direct (ReportList *reports, PointerRNA ptr, PropertyRNA *prop, FCurve *fcu, float cfra, short flag)
{
    float curval= 0.0f;
    
    /* no F-Curve to add keyframe to? */
    if (fcu == NULL) {
        BKE_report(reports, RPT_ERROR, "No F-Curve to add keyframes to");
        return 0;
    }
    /* F-Curve not editable? */
    if (fcurve_is_keyframable(fcu) == 0) {
        BKE_reportf(reports, RPT_ERROR, 
            "F-Curve with path = '%s' [%d] cannot be keyframed. Ensure that it is not locked or sampled. Also, try removing F-Modifiers",
            fcu->rna_path, fcu->array_index);
        return 0;
    }
    
    /* if no property given yet, try to validate from F-Curve info */
    if ((ptr.id.data == NULL) && (ptr.data==NULL)) {
        BKE_report(reports, RPT_ERROR, "No RNA-pointer available to retrieve values for keyframing from");
        return 0;
    }
    if (prop == NULL) {
        PointerRNA tmp_ptr;
        
        /* try to get property we should be affecting */
        if ((RNA_path_resolve(&ptr, fcu->rna_path, &tmp_ptr, &prop) == 0) || (prop == NULL)) {
            /* property not found... */
            const char *idname= (ptr.id.data) ? ((ID *)ptr.id.data)->name : "<No ID-Pointer>";
            
            BKE_reportf(reports, RPT_ERROR,
                "Could not insert keyframe, as RNA Path is invalid for the given ID (ID = %s, Path = %s)", 
                idname, fcu->rna_path);
            return 0;
        }
        else {
            /* property found, so overwrite 'ptr' to make later code easier */
            ptr= tmp_ptr;
        }
    }
    
    /* set additional flags for the F-Curve (i.e. only integer values) */
    fcu->flag &= ~(FCURVE_INT_VALUES|FCURVE_DISCRETE_VALUES);
    switch (RNA_property_type(prop)) {
        case PROP_FLOAT:
            /* do nothing */
            break;
        case PROP_INT:
            /* do integer (only 'whole' numbers) interpolation between all points */
            fcu->flag |= FCURVE_INT_VALUES;
            break;
        default:
            /* do 'discrete' (i.e. enum, boolean values which cannot take any intermediate
             * values at all) interpolation between all points
             *  - however, we must also ensure that evaluated values are only integers still
             */
            fcu->flag |= (FCURVE_DISCRETE_VALUES|FCURVE_INT_VALUES);
            break;
    }
    
    /* obtain value to give keyframe */
    if ( (flag & INSERTKEY_MATRIX) && 
         (visualkey_can_use(&ptr, prop)) ) 
    {
        /* visual-keying is only available for object and pchan datablocks, as 
         * it works by keyframing using a value extracted from the final matrix 
         * instead of using the kt system to extract a value.
         */
        curval= visualkey_get_value(&ptr, prop, fcu->array_index);
    }
    else {
        /* read value from system */
        curval= setting_get_rna_value(&ptr, prop, fcu->array_index);
    }
    
    /* only insert keyframes where they are needed */
    if (flag & INSERTKEY_NEEDED) {
        short insert_mode;
        
        /* check whether this curve really needs a new keyframe */
        insert_mode= new_key_needed(fcu, cfra, curval);
        
        /* insert new keyframe at current frame */
        if (insert_mode)
            insert_vert_fcurve(fcu, cfra, curval, flag);
        
        /* delete keyframe immediately before/after newly added */
        switch (insert_mode) {
            case KEYNEEDED_DELPREV:
                delete_fcurve_key(fcu, fcu->totvert-2, 1);
                break;
            case KEYNEEDED_DELNEXT:
                delete_fcurve_key(fcu, 1, 1);
                break;
        }
        
        /* only return success if keyframe added */
        if (insert_mode)
            return 1;
    }
    else {
        /* just insert keyframe */
        insert_vert_fcurve(fcu, cfra, curval, flag);
        
        /* return success */
        return 1;
    }
    
    /* failed */
    return 0;
}

/* Main Keyframing API call:
 *  Use this when validation of necessary animation data is necessary, since it may not exist yet.
 *  
 *  The flag argument is used for special settings that alter the behaviour of
 *  the keyframe insertion. These include the 'visual' keyframing modes, quick refresh,
 *  and extra keyframe filtering.
 *
 *  index of -1 keys all array indices
 */
short insert_keyframe (ReportList *reports, ID *id, bAction *act, const char group[], const char rna_path[], int array_index, float cfra, short flag)
{   
    PointerRNA id_ptr, ptr;
    PropertyRNA *prop = NULL;
    FCurve *fcu;
    int array_index_max= array_index+1;
    int ret= 0;
    
    /* validate pointer first - exit if failure */
    if (id == NULL) {
        BKE_reportf(reports, RPT_ERROR, "No ID-block to insert keyframe in (Path = %s)", rna_path);
        return 0;
    }
    
    RNA_id_pointer_create(id, &id_ptr);
    if ((RNA_path_resolve(&id_ptr, rna_path, &ptr, &prop) == 0) || (prop == NULL)) {
        BKE_reportf(reports, RPT_ERROR,
            "Could not insert keyframe, as RNA Path is invalid for the given ID (ID = %s, Path = %s)", 
            (id)? id->name : "<Missing ID-Block>", rna_path);
        return 0;
    }
    
    /* if no action is provided, keyframe to the default one attached to this ID-block */
    if (act == NULL) {
        AnimData *adt= BKE_animdata_from_id(id);
        
        /* get action to add F-Curve+keyframe to */
        act= verify_adt_action(id, 1);
        
        if (act == NULL) {
            BKE_reportf(reports, RPT_ERROR, 
                "Could not insert keyframe, as this type does not support animation data (ID = %s, Path = %s)", 
                id->name, rna_path);
            return 0;
        }
        
        /* apply NLA-mapping to frame to use (if applicable) */
        cfra= BKE_nla_tweakedit_remap(adt, cfra, NLATIME_CONVERT_UNMAP);
    }
    
    /* key entire array convenience method */
    if (array_index == -1) { 
        array_index= 0;
        array_index_max= RNA_property_array_length(&ptr, prop);
        
        /* for single properties, increase max_index so that the property itself gets included,
         * but don't do this for standard arrays since that can cause corruption issues 
         * (extra unused curves)
         */
        if (array_index_max == array_index)
            array_index_max++;
    }
    
    /* will only loop once unless the array index was -1 */
    for (; array_index < array_index_max; array_index++) {
        /* make sure the F-Curve exists 
         *  - if we're replacing keyframes only, DO NOT create new F-Curves if they do not exist yet
         *    but still try to get the F-Curve if it exists...
         */
        fcu= verify_fcurve(act, group, rna_path, array_index, (flag & INSERTKEY_REPLACE)==0);
        
        /* we may not have a F-Curve when we're replacing only... */
        if (fcu) {
            /* set color mode if the F-Curve is new (i.e. without any keyframes) */
            if ((fcu->totvert == 0) && (flag & INSERTKEY_XYZ2RGB)) {
                /* for Loc/Rot/Scale and also Color F-Curves, the color of the F-Curve in the Graph Editor,
                 * is determined by the array index for the F-Curve
                 */
                if (ELEM5(RNA_property_subtype(prop), PROP_TRANSLATION, PROP_XYZ, PROP_EULER, PROP_COLOR, PROP_COORDS)) {
                    fcu->color_mode= FCURVE_COLOR_AUTO_RGB;
                }
            }
            
            /* insert keyframe */
            ret += insert_keyframe_direct(reports, ptr, prop, fcu, cfra, flag);
        }
    }
    
    return ret;
}

/* ************************************************** */
/* KEYFRAME DELETION */

/* Main Keyframing API call:
 *  Use this when validation of necessary animation data isn't necessary as it
 *  already exists. It will delete a keyframe at the current frame.
 *  
 *  The flag argument is used for special settings that alter the behaviour of
 *  the keyframe deletion. These include the quick refresh options.
 */
short delete_keyframe (ReportList *reports, ID *id, bAction *act, const char group[], const char rna_path[], int array_index, float cfra, short UNUSED(flag))
{
    AnimData *adt= BKE_animdata_from_id(id);
    PointerRNA id_ptr, ptr;
    PropertyRNA *prop;
    int array_index_max= array_index+1;
    int ret= 0;
    
    /* sanity checks */
    if ELEM(NULL, id, adt) {
        BKE_report(reports, RPT_ERROR, "No ID-Block and/Or AnimData to delete keyframe from");
        return 0;
    }
    
    /* validate pointer first - exit if failure */
    RNA_id_pointer_create(id, &id_ptr);
    if ((RNA_path_resolve(&id_ptr, rna_path, &ptr, &prop) == 0) || (prop == NULL)) {
        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);
        return 0;
    }
    
    /* get F-Curve
     * Note: here is one of the places where we don't want new Action + F-Curve added!
     *      so 'add' var must be 0
     */
    if (act == NULL) {
        /* if no action is provided, use the default one attached to this ID-block 
         *  - if it doesn't exist, then we're out of options...
         */
        if (adt->action) {
            act= adt->action;
            
            /* apply NLA-mapping to frame to use (if applicable) */
            cfra= BKE_nla_tweakedit_remap(adt, cfra, NLATIME_CONVERT_UNMAP); 
        }
        else {
            BKE_reportf(reports, RPT_ERROR, "No Action to delete keyframes from for ID = %s \n", id->name);
            return 0;
        }
    }
    
    /* key entire array convenience method */
    if (array_index == -1) { 
        array_index= 0;
        array_index_max= RNA_property_array_length(&ptr, prop);
        
        /* for single properties, increase max_index so that the property itself gets included,
         * but don't do this for standard arrays since that can cause corruption issues 
         * (extra unused curves)
         */
        if (array_index_max == array_index)
            array_index_max++;
    }
    
    /* will only loop once unless the array index was -1 */
    for (; array_index < array_index_max; array_index++) {
        FCurve *fcu= verify_fcurve(act, group, rna_path, array_index, 0);
        short found = -1;
        int i;
        
        /* check if F-Curve exists and/or whether it can be edited */
        if (fcu == NULL)
            continue;
            
        if ( (fcu->flag & FCURVE_PROTECTED) || ((fcu->grp) && (fcu->grp->flag & AGRP_PROTECTED)) ) {
            if (G.f & G_DEBUG)
                printf("WARNING: not deleting keyframe for locked F-Curve \n");
            continue;
        }
        
        /* try to find index of beztriple to get rid of */
        i = binarysearch_bezt_index(fcu->bezt, cfra, fcu->totvert, &found);
        if (found) {            
            /* delete the key at the index (will sanity check + do recalc afterwards) */
            delete_fcurve_key(fcu, i, 1);
            
            /* Only delete curve too if it won't be doing anything anymore */
            if ((fcu->totvert == 0) && (list_has_suitable_fmodifier(&fcu->modifiers, 0, FMI_TYPE_GENERATE_CURVE) == 0))
                ANIM_fcurve_delete_from_animdata(NULL, adt, fcu);
            
            /* return success */
            ret++;
        }
    }
    
    /* return success/failure */
    return ret;
}

/* ******************************************* */
/* KEYFRAME MODIFICATION */

/* mode for commonkey_modifykey */
enum {
    COMMONKEY_MODE_INSERT = 0,
    COMMONKEY_MODE_DELETE,
} /*eCommonModifyKey_Modes*/;

/* Polling callback for use with ANIM_*_keyframe() operators
 * This is based on the standard ED_operator_areaactive callback,
 * except that it does special checks for a few spacetypes too...
 */
static int modify_key_op_poll(bContext *C)
{
    ScrArea *sa= CTX_wm_area(C);
    Scene *scene= CTX_data_scene(C);
    SpaceOops *so= CTX_wm_space_outliner(C);
    
    /* if no area or active scene */
    if (ELEM(NULL, sa, scene)) 
        return 0;
    
    /* if Outliner, don't allow in some views */
    if (so) {
        if (ELEM4(so->outlinevis, SO_GROUPS, SO_LIBRARIES, SO_VERSE_SESSION, SO_VERSE_SESSION))
            return 0;
        if (ELEM3(so->outlinevis, SO_SEQUENCE, SO_USERDEF, SO_KEYMAP))
            return 0;
    }
    
    /* TODO: checks for other space types can be added here */
    
    /* should be fine */
    return 1;
}

/* Insert Key Operator ------------------------ */

static int insert_key_exec (bContext *C, wmOperator *op)
{
    Main *bmain= CTX_data_main(C);
    Scene *scene= CTX_data_scene(C);
    KeyingSet *ks= NULL;
    int type= RNA_enum_get(op->ptr, "type");
    float cfra= (float)CFRA; // XXX for now, don't bother about all the yucky offset crap
    short success;
    
    /* type is the Keying Set the user specified to use when calling the operator:
     *  - type == 0: use scene's active Keying Set
     *  - type > 0: use a user-defined Keying Set from the active scene
     *  - type < 0: use a builtin Keying Set
     */
    if (type == 0) 
        type= scene->active_keyingset;
    if (type > 0)
        ks= BLI_findlink(&scene->keyingsets, scene->active_keyingset-1);
    else
        ks= BLI_findlink(&builtin_keyingsets, -type-1);
        
    /* report failures */
    if (ks == NULL) {
        BKE_report(op->reports, RPT_ERROR, "No active Keying Set");
        return OPERATOR_CANCELLED;
    }
    
    /* try to insert keyframes for the channels specified by KeyingSet */
    success= ANIM_apply_keyingset(C, NULL, NULL, ks, MODIFYKEY_MODE_INSERT, cfra);
    if (G.f & G_DEBUG)
        BKE_reportf(op->reports, RPT_INFO, "KeyingSet '%s' - Successfully added %d Keyframes \n", ks->name, success);
    
    /* report failure or do updates? */
    if (success == MODIFYKEY_INVALID_CONTEXT) {
        BKE_report(op->reports, RPT_ERROR, "No suitable context info for active Keying Set");
        return OPERATOR_CANCELLED;
    }
    else if (success) {
        /* if the appropriate properties have been set, make a note that we've inserted something */
        if (RNA_boolean_get(op->ptr, "confirm_success"))
            BKE_reportf(op->reports, RPT_INFO, "Successfully added %d Keyframes for KeyingSet '%s'", success, ks->name);
        
        /* send notifiers that keyframes have been changed */
        WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_EDITED, NULL);
    }
    else
        BKE_report(op->reports, RPT_WARNING, "Keying Set failed to insert any keyframes");
    
    /* send updates */
    DAG_ids_flush_update(bmain, 0);
    
    return OPERATOR_FINISHED;
}

void ANIM_OT_keyframe_insert (wmOperatorType *ot)
{
    PropertyRNA *prop;
    
    /* identifiers */
    ot->name= "Insert Keyframe";
    ot->idname= "ANIM_OT_keyframe_insert";
    ot->description= "Insert keyframes on the current frame for all properties in the specified Keying Set";
    
    /* callbacks */
    ot->exec= insert_key_exec; 
    ot->poll= modify_key_op_poll;
    
    /* flags */
    ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
    
    /* keyingset to use (dynamic enum) */
    prop= RNA_def_enum(ot->srna, "type", DummyRNA_DEFAULT_items, 0, "Keying Set", "The Keying Set to use");
    RNA_def_enum_funcs(prop, ANIM_keying_sets_enum_itemf);
    RNA_def_property_flag(prop, PROP_HIDDEN);
    ot->prop= prop;
    
    /* confirm whether a keyframe was added by showing a popup 
     *  - by default, this is enabled, since this operator is assumed to be called independently
     */
    prop= RNA_def_boolean(ot->srna, "confirm_success", 1, "Confirm Successful Insert", "Show a popup when the keyframes get successfully added");
    RNA_def_property_flag(prop, PROP_HIDDEN);
}

/* Insert Key Operator (With Menu) ------------------------ */
/* This operator checks if a menu should be shown for choosing the KeyingSet to use, 
 * then calls the menu if necessary before 
 */

static int insert_key_menu_invoke (bContext *C, wmOperator *op, wmEvent *UNUSED(event))
{
    Scene *scene= CTX_data_scene(C);
    
    /* if prompting or no active Keying Set, show the menu */
    if ((scene->active_keyingset == 0) || RNA_boolean_get(op->ptr, "always_prompt")) {
        /* call the menu, which will call this operator again, hence the cancelled */
        ANIM_keying_sets_menu_setup(C, op->type->name, "ANIM_OT_keyframe_insert_menu");
        return OPERATOR_CANCELLED;
    }
    else {
        /* just call the exec() on the active keyingset */
        RNA_enum_set(op->ptr, "type", 0);
        RNA_boolean_set(op->ptr, "confirm_success", TRUE);
        
        return op->type->exec(C, op);
    }
}
 
void ANIM_OT_keyframe_insert_menu (wmOperatorType *ot)
{
    PropertyRNA *prop;
    
    /* identifiers */
    ot->name= "Insert Keyframe Menu";
    ot->idname= "ANIM_OT_keyframe_insert_menu";
    ot->description= "Insert Keyframes for specified Keying Set, with menu of available Keying Sets if undefined";
    
    /* callbacks */
    ot->invoke= insert_key_menu_invoke;
    ot->exec= insert_key_exec; 
    ot->poll= ED_operator_areaactive;
    
    /* flags */
    ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
    
    /* keyingset to use (dynamic enum) */
    prop= RNA_def_enum(ot->srna, "type", DummyRNA_DEFAULT_items, 0, "Keying Set", "The Keying Set to use");
    RNA_def_enum_funcs(prop, ANIM_keying_sets_enum_itemf);
    RNA_def_property_flag(prop, PROP_HIDDEN);
    ot->prop= prop;
    
    /* confirm whether a keyframe was added by showing a popup 
     *  - by default, this is disabled so that if a menu is shown, this doesn't come up too
     */
    // XXX should this just be always on?
    prop= RNA_def_boolean(ot->srna, "confirm_success", 0, "Confirm Successful Insert", "Show a popup when the keyframes get successfully added");
    RNA_def_property_flag(prop, PROP_HIDDEN);
    
    /* whether the menu should always be shown 
     *  - by default, the menu should only be shown when there is no active Keying Set (2.5 behaviour),
     *    although in some cases it might be useful to always shown (pre 2.5 behaviour)
     */
    prop= RNA_def_boolean(ot->srna, "always_prompt", 0, "Always Show Menu", "");
    RNA_def_property_flag(prop, PROP_HIDDEN);
}

/* Delete Key Operator ------------------------ */

static int delete_key_exec (bContext *C, wmOperator *op)
{
    Main *bmain= CTX_data_main(C);
    Scene *scene= CTX_data_scene(C);
    KeyingSet *ks= NULL;    
    int type= RNA_enum_get(op->ptr, "type");
    float cfra= (float)CFRA; // XXX for now, don't bother about all the yucky offset crap
    short success;
    
    /* type is the Keying Set the user specified to use when calling the operator:
     *  - type == 0: use scene's active Keying Set
     *  - type > 0: use a user-defined Keying Set from the active scene
     *  - type < 0: use a builtin Keying Set
     */
    if (type == 0) 
        type= scene->active_keyingset;
    if (type > 0)
        ks= BLI_findlink(&scene->keyingsets, scene->active_keyingset-1);
    else
        ks= BLI_findlink(&builtin_keyingsets, -type-1);
    
    /* report failure */
    if (ks == NULL) {
        BKE_report(op->reports, RPT_ERROR, "No active Keying Set");
        return OPERATOR_CANCELLED;
    }
    
    /* try to insert keyframes for the channels specified by KeyingSet */
    success= ANIM_apply_keyingset(C, NULL, NULL, ks, MODIFYKEY_MODE_DELETE, cfra);
    if (G.f & G_DEBUG)
        printf("KeyingSet '%s' - Successfully removed %d Keyframes \n", ks->name, success);
    
    /* report failure or do updates? */
    if (success == MODIFYKEY_INVALID_CONTEXT) {
        BKE_report(op->reports, RPT_ERROR, "No suitable context info for active Keying Set");
        return OPERATOR_CANCELLED;
    }
    else if (success) {
        /* if the appropriate properties have been set, make a note that we've inserted something */
        if (RNA_boolean_get(op->ptr, "confirm_success"))
            BKE_reportf(op->reports, RPT_INFO, "Successfully removed %d Keyframes for KeyingSet '%s'", success, ks->name);
        
        /* send notifiers that keyframes have been changed */
        WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_EDITED, NULL);
    }
    else
        BKE_report(op->reports, RPT_WARNING, "Keying Set failed to remove any keyframes");
    
    /* send updates */
    DAG_ids_flush_update(bmain, 0);
    
    return OPERATOR_FINISHED;
}

void ANIM_OT_keyframe_delete (wmOperatorType *ot)
{
    PropertyRNA *prop;
    
    /* identifiers */
    ot->name= "Delete Keying-Set Keyframe";
    ot->idname= "ANIM_OT_keyframe_delete";
    ot->description= "Delete keyframes on the current frame for all properties in the specified Keying Set";
    
    /* callbacks */
    ot->exec= delete_key_exec; 
    ot->poll= modify_key_op_poll;
    
    /* flags */
    ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
    
    /* keyingset to use (dynamic enum) */
    prop= RNA_def_enum(ot->srna, "type", DummyRNA_DEFAULT_items, 0, "Keying Set", "The Keying Set to use");
    RNA_def_enum_funcs(prop, ANIM_keying_sets_enum_itemf);
    RNA_def_property_flag(prop, PROP_HIDDEN);
    ot->prop= prop;
    
    /* confirm whether a keyframe was added by showing a popup 
     *  - by default, this is enabled, since this operator is assumed to be called independently
     */
    RNA_def_boolean(ot->srna, "confirm_success", 1, "Confirm Successful Insert", "Show a popup when the keyframes get successfully added");
}

/* Delete Key Operator ------------------------ */

/* XXX WARNING:
 * This is currently just a basic operator, which work in 3d-view context on objects only. 
 * Should this be kept? It does have advantages over a version which requires selecting a keyingset to use...
 * -- Joshua Leung, Jan 2009
 */
 
static int delete_key_v3d_exec (bContext *C, wmOperator *op)
{
    Main *bmain= CTX_data_main(C);
    Scene *scene= CTX_data_scene(C);
    float cfra= (float)CFRA; // XXX for now, don't bother about all the yucky offset crap
    
    // XXX more comprehensive tests will be needed
    CTX_DATA_BEGIN(C, Object*, ob, selected_objects) 
    {
        ID *id= (ID *)ob;
        FCurve *fcu, *fcn;
        short success= 0;
        
        /* loop through all curves in animdata and delete keys on this frame */
        if ((ob->adt) && (ob->adt->action)) {
            AnimData *adt= ob->adt;
            bAction *act= adt->action;
            
            for (fcu= act->curves.first; fcu; fcu= fcn) {
                fcn= fcu->next;
                success+= delete_keyframe(op->reports, id, NULL, NULL, fcu->rna_path, fcu->array_index, cfra, 0);
            }
        }
        
        BKE_reportf(op->reports, RPT_INFO, "Ob '%s' - Successfully had %d keyframes removed", id->name+2, success);
        
        ob->recalc |= OB_RECALC_OB;
    }
    CTX_DATA_END;
    
    /* send updates */
    DAG_ids_flush_update(bmain, 0);
    
    WM_event_add_notifier(C, NC_OBJECT|ND_KEYS, NULL);
    
    return OPERATOR_FINISHED;
}

void ANIM_OT_keyframe_delete_v3d (wmOperatorType *ot)
{
    /* identifiers */
    ot->name= "Delete Keyframe";
    ot->description= "Remove keyframes on current frame for selected object";
    ot->idname= "ANIM_OT_keyframe_delete_v3d";
    
    /* callbacks */
    ot->invoke= WM_operator_confirm;
    ot->exec= delete_key_v3d_exec; 
    
    ot->poll= ED_operator_areaactive;
    
    /* flags */
    ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
}


/* Insert Key Button Operator ------------------------ */

static int insert_key_button_exec (bContext *C, wmOperator *op)
{
    Main *bmain= CTX_data_main(C);
    Scene *scene= CTX_data_scene(C);
    PointerRNA ptr= {{NULL}};
    PropertyRNA *prop= NULL;
    char *path;
    float cfra= (float)CFRA; // XXX for now, don't bother about all the yucky offset crap
    short success= 0;
    int a, index, length, all= RNA_boolean_get(op->ptr, "all");
    short flag = 0;
    
    /* flags for inserting keyframes */
    flag = ANIM_get_keyframing_flags(scene, 1);
    
    /* try to insert keyframe using property retrieved from UI */
    uiContextActiveProperty(C, &ptr, &prop, &index);
    
    if ((ptr.id.data && ptr.data && prop) && RNA_property_animateable(&ptr, prop)) {
        path= RNA_path_from_ID_to_property(&ptr, prop);
        
        if (path) {
            if (all) {
                length= RNA_property_array_length(&ptr, prop);
                
                if(length) index= 0;
                else length= 1;
            }
            else
                length= 1;
            
            for (a=0; a<length; a++)
                success+= insert_keyframe(op->reports, ptr.id.data, NULL, NULL, path, index+a, cfra, flag);
            
            MEM_freeN(path);
        }
        else if (ptr.type == &RNA_NlaStrip) {
            /* handle special vars for NLA-strips */
            NlaStrip *strip= (NlaStrip *)ptr.data;
            FCurve *fcu= list_find_fcurve(&strip->fcurves, RNA_property_identifier(prop), flag);
            
            success+= insert_keyframe_direct(op->reports, ptr, prop, fcu, cfra, 0);
        }
        else {
            if (G.f & G_DEBUG)
                printf("Button Insert-Key: no path to property \n");
            BKE_report(op->reports, RPT_WARNING, "Failed to resolve path to property. Try using a Keying Set instead");
        }
    }
    else if (G.f & G_DEBUG) {
        printf("ptr.data = %p, prop = %p,", (void *)ptr.data, (void *)prop);
        if (prop)
            printf("animateable = %d \n", RNA_property_animateable(&ptr, prop));
        else
            printf("animateable = NULL \n");
    }
    
    if (success) {
        /* send updates */
        uiContextAnimUpdate(C);
        
        DAG_ids_flush_update(bmain, 0);
        
        /* send notifiers that keyframes have been changed */
        WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_EDITED, NULL);
    }
    
    return (success)? OPERATOR_FINISHED: OPERATOR_CANCELLED;
}

void ANIM_OT_keyframe_insert_button (wmOperatorType *ot)
{
    /* identifiers */
    ot->name= "Insert Keyframe (Buttons)";
    ot->idname= "ANIM_OT_keyframe_insert_button";
    
    /* callbacks */
    ot->exec= insert_key_button_exec; 
    ot->poll= modify_key_op_poll;
    
    /* flags */
    ot->flag= OPTYPE_UNDO;

    /* properties */
    RNA_def_boolean(ot->srna, "all", 1, "All", "Insert a keyframe for all element of the array");
}

/* Delete Key Button Operator ------------------------ */

static int delete_key_button_exec (bContext *C, wmOperator *op)
{
    Main *bmain= CTX_data_main(C);
    Scene *scene= CTX_data_scene(C);
    PointerRNA ptr= {{NULL}};
    PropertyRNA *prop= NULL;
    char *path;
    float cfra= (float)CFRA; // XXX for now, don't bother about all the yucky offset crap
    short success= 0;
    int a, index, length, all= RNA_boolean_get(op->ptr, "all");
    
    /* try to insert keyframe using property retrieved from UI */
    uiContextActiveProperty(C, &ptr, &prop, &index);

    if (ptr.id.data && ptr.data && prop) {
        path= RNA_path_from_ID_to_property(&ptr, prop);
        
        if (path) {
            if (all) {
                length= RNA_property_array_length(&ptr, prop);
                
                if (length) index= 0;
                else length= 1;
            }
            else
                length= 1;
            
            for (a=0; a<length; a++)
                success+= delete_keyframe(op->reports, ptr.id.data, NULL, NULL, path, index+a, cfra, 0);
            
            MEM_freeN(path);
        }
        else if (G.f & G_DEBUG)
            printf("Button Delete-Key: no path to property \n");
    }
    else if (G.f & G_DEBUG) {
        printf("ptr.data = %p, prop = %p \n", (void *)ptr.data, (void *)prop);
    }
    
    
    if (success) {
        /* send updates */
        uiContextAnimUpdate(C);
        
        DAG_ids_flush_update(bmain, 0);
        
        /* send notifiers that keyframes have been changed */
        WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_EDITED, NULL);
    }
    
    return (success)? OPERATOR_FINISHED: OPERATOR_CANCELLED;
}

void ANIM_OT_keyframe_delete_button (wmOperatorType *ot)
{
    /* identifiers */
    ot->name= "Delete Keyframe (Buttons)";
    ot->idname= "ANIM_OT_keyframe_delete_button";
    
    /* callbacks */
    ot->exec= delete_key_button_exec; 
    ot->poll= modify_key_op_poll;
    
    /* flags */
    ot->flag= OPTYPE_UNDO;

    /* properties */
    RNA_def_boolean(ot->srna, "all", 1, "All", "Delete keyfames from all elements of the array");
}

/* ******************************************* */
/* AUTO KEYFRAME */

int autokeyframe_cfra_can_key(Scene *scene, ID *id)
{
    float cfra= (float)CFRA; // XXX for now, this will do
    
    /* only filter if auto-key mode requires this */
    if (IS_AUTOKEY_ON(scene) == 0)
        return 0;
        
    if (IS_AUTOKEY_MODE(scene, NORMAL)) {
        /* can insert anytime we like... */
        return 1;
    }
    else /* REPLACE */ {
        /* for whole block - only key if there's a keyframe on that frame already
         *  this is a valid assumption when we're blocking + tweaking
         */
        return id_frame_has_keyframe(id, cfra, ANIMFILTER_KEYS_LOCAL);
    }
}

/* ******************************************* */
/* KEYFRAME DETECTION */

/* --------------- API/Per-Datablock Handling ------------------- */

/* Checks if some F-Curve has a keyframe for a given frame */
short fcurve_frame_has_keyframe (FCurve *fcu, float frame, short filter)
{
    /* quick sanity check */
    if (ELEM(NULL, fcu, fcu->bezt))
        return 0;
    
    /* we either include all regardless of muting, or only non-muted  */
    if ((filter & ANIMFILTER_KEYS_MUTED) || (fcu->flag & FCURVE_MUTED)==0) {
        short replace = -1;
        int i = binarysearch_bezt_index(fcu->bezt, frame, fcu->totvert, &replace);
        
        /* binarysearch_bezt_index will set replace to be 0 or 1
         *  - obviously, 1 represents a match
         */
        if (replace) {          
            /* sanity check: 'i' may in rare cases exceed arraylen */
            if ((i >= 0) && (i < fcu->totvert))
                return 1;
        }
    }
    
    return 0;
}

/* Checks whether an Action has a keyframe for a given frame 
 * Since we're only concerned whether a keyframe exists, we can simply loop until a match is found...
 */
static short action_frame_has_keyframe (bAction *act, float frame, short filter)
{
    FCurve *fcu;
    
    /* can only find if there is data */
    if (act == NULL)
        return 0;
        
    /* if only check non-muted, check if muted */
    if ((filter & ANIMFILTER_KEYS_MUTED) || (act->flag & ACT_MUTED))
        return 0;
    
    /* loop over F-Curves, using binary-search to try to find matches 
     *  - this assumes that keyframes are only beztriples
     */
    for (fcu= act->curves.first; fcu; fcu= fcu->next) {
        /* only check if there are keyframes (currently only of type BezTriple) */
        if (fcu->bezt && fcu->totvert) {
            if (fcurve_frame_has_keyframe(fcu, frame, filter))
                return 1;
        }
    }
    
    /* nothing found */
    return 0;
}

/* Checks whether an Object has a keyframe for a given frame */
static short object_frame_has_keyframe (Object *ob, float frame, short filter)
{
    /* error checking */
    if (ob == NULL)
        return 0;
    
    /* check own animation data - specifically, the action it contains */
    if ((ob->adt) && (ob->adt->action)) {
        if (action_frame_has_keyframe(ob->adt->action, frame, filter))
            return 1;
    }
    
    /* try shapekey keyframes (if available, and allowed by filter) */
    if ( !(filter & ANIMFILTER_KEYS_LOCAL) && !(filter & ANIMFILTER_KEYS_NOSKEY) ) {
        Key *key= ob_get_key(ob);
        
        /* shapekeys can have keyframes ('Relative Shape Keys') 
         * or depend on time (old 'Absolute Shape Keys') 
         */
         
            /* 1. test for relative (with keyframes) */
        if (id_frame_has_keyframe((ID *)key, frame, filter))
            return 1;
            
            /* 2. test for time */
        // TODO... yet to be implemented (this feature may evolve before then anyway)
    }
    
    /* try materials */
    if ( !(filter & ANIMFILTER_KEYS_LOCAL) && !(filter & ANIMFILTER_KEYS_NOMAT) ) {
        /* if only active, then we can skip a lot of looping */
        if (filter & ANIMFILTER_KEYS_ACTIVE) {
            Material *ma= give_current_material(ob, (ob->actcol + 1));
            
            /* we only retrieve the active material... */
            if (id_frame_has_keyframe((ID *)ma, frame, filter))
                return 1;
        }
        else {
            int a;
            
            /* loop over materials */
            for (a=0; a<ob->totcol; a++) {
                Material *ma= give_current_material(ob, a+1);
                
                if (id_frame_has_keyframe((ID *)ma, frame, filter))
                    return 1;
            }
        }
    }
    
    /* nothing found */
    return 0;
}

/* --------------- API ------------------- */

/* Checks whether a keyframe exists for the given ID-block one the given frame */
short id_frame_has_keyframe (ID *id, float frame, short filter)
{
    /* sanity checks */
    if (id == NULL)
        return 0;
    
    /* perform special checks for 'macro' types */
    switch (GS(id->name)) {
        case ID_OB: /* object */
            return object_frame_has_keyframe((Object *)id, frame, filter);
            break;
            
        case ID_SCE: /* scene */
        // XXX TODO... for now, just use 'normal' behaviour
        //  break;
        
        default:    /* 'normal type' */
        {
            AnimData *adt= BKE_animdata_from_id(id);
            
            /* only check keyframes in active action */
            if (adt)
                return action_frame_has_keyframe(adt->action, frame, filter);
        }
            break;
    }
    
    
    /* no keyframe found */
    return 0;
}

/* ************************************************** */

int ED_autokeyframe_object(bContext *C, Scene *scene, Object *ob, KeyingSet *ks)
{
    /* auto keyframing */
    if (autokeyframe_cfra_can_key(scene, &ob->id)) {
        ListBase dsources = {NULL, NULL};

        /* now insert the keyframe(s) using the Keying Set
         *  1) add datasource override for the Object
         *  2) insert keyframes
         *  3) free the extra info
         */
        ANIM_relative_keyingset_add_source(&dsources, &ob->id, NULL, NULL);
        ANIM_apply_keyingset(C, &dsources, NULL, ks, MODIFYKEY_MODE_INSERT, (float)CFRA);
        BLI_freelistN(&dsources);

        return TRUE;
    }
    else {
        return FALSE;
    }
}

int ED_autokeyframe_pchan(bContext *C, Scene *scene, Object *ob, bPoseChannel *pchan, KeyingSet *ks)
{
    if (autokeyframe_cfra_can_key(scene, &ob->id)) {
        ListBase dsources = {NULL, NULL};

        /* now insert the keyframe(s) using the Keying Set
         *  1) add datasource override for the PoseChannel
         *  2) insert keyframes
         *  3) free the extra info
         */
        ANIM_relative_keyingset_add_source(&dsources, &ob->id, &RNA_PoseBone, pchan);
        ANIM_apply_keyingset(C, &dsources, NULL, ks, MODIFYKEY_MODE_INSERT, (float)CFRA);
        BLI_freelistN(&dsources);

        /* clear any unkeyed tags */
        if (pchan->bone) {
            pchan->bone->flag &= ~BONE_UNKEYED;
        }

        return TRUE;
    }
    else {
        /* add unkeyed tags */
        if (pchan->bone) {
            pchan->bone->flag |= BONE_UNKEYED;
        }

        return FALSE;
    }
}