
Insert Keyframe function was not checking that an ID-block was given before trying to resolve the RNA-path using it.
1674 lines
50 KiB
C
1674 lines
50 KiB
C
/**
|
|
* $Id$
|
|
*
|
|
* ***** 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 "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 "BKE_animsys.h"
|
|
#include "BKE_action.h"
|
|
#include "BKE_constraint.h"
|
|
#include "BKE_depsgraph.h"
|
|
#include "BKE_fcurve.h"
|
|
#include "BKE_nla.h"
|
|
#include "BKE_global.h"
|
|
#include "BKE_utildefines.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 "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(AUTOMATKEY))
|
|
flag |= INSERTKEY_MATRIX;
|
|
|
|
/* only needed */
|
|
if (IS_AUTOKEY_FLAG(INSERTNEEDED))
|
|
flag |= INSERTKEY_NEEDED;
|
|
|
|
/* default F-Curve color mode - RGB from XYZ indicies */
|
|
if (IS_AUTOKEY_FLAG(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_AUTO_HANDLES|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;
|
|
|
|
// 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;
|
|
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...
|
|
*/
|
|
memset(&beztr, 0, sizeof(BezTriple));
|
|
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.ipo= U.ipo_new; /* use default interpolation mode here... */
|
|
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 */
|
|
|
|
/* 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) */
|
|
// FIXME: this doesn't work if user tweaked the interpolation specifically, and they were just overwriting some existing key in the process...
|
|
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_EQ(prevPosi, cFrame) && IS_EQ(beztPosi, cFrame) && IS_EQ(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_EQ(prevVal, nValue) && IS_EQ(beztVal, nValue) && IS_EQ(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_EQ(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_EQ(prevVal, nValue) && IS_EQ(beztVal, nValue) && IS_EQ(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_EQ(valA, nValue) && IS_EQ(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,
|
|
};
|
|
|
|
/* 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;
|
|
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= (char *)RNA_property_identifier(prop);
|
|
}
|
|
else if (ptr->type == &RNA_PoseBone) {
|
|
/* Pose Channel */
|
|
bPoseChannel *pchan= (bPoseChannel *)ptr->data;
|
|
|
|
con= pchan->constraints.first;
|
|
identifier= (char *)RNA_property_identifier(prop);
|
|
}
|
|
|
|
/* check if any data to search using */
|
|
if (ELEM(NULL, con, identifier))
|
|
return 0;
|
|
|
|
/* location or rotation identifiers only... */
|
|
if (strstr(identifier, "location"))
|
|
searchtype= VISUALKEY_LOC;
|
|
else if (strstr(identifier, "rotation"))
|
|
searchtype= VISUALKEY_ROT;
|
|
else {
|
|
printf("visualkey_can_use() failed: identifier - '%s' \n", identifier);
|
|
return 0;
|
|
}
|
|
|
|
|
|
/* only search if a searchtype and initial constraint are available */
|
|
if (searchtype && con) {
|
|
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:
|
|
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)
|
|
{
|
|
char *identifier= (char *)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;
|
|
|
|
/* parented objects are not supported, as the effects of the parent
|
|
* are included in the matrix, which kindof beats the point
|
|
*/
|
|
if (ob->parent == NULL) {
|
|
/* only Location or Rotation keyframes are supported now */
|
|
if (strstr(identifier, "location")) {
|
|
return ob->obmat[3][array_index];
|
|
}
|
|
else if (strstr(identifier, "rotation")) {
|
|
float eul[3];
|
|
|
|
mat4_to_eul( eul,ob->obmat);
|
|
return eul[array_index];
|
|
}
|
|
}
|
|
}
|
|
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];
|
|
|
|
/* euler-rotation test before standard rotation, as standard rotation does quats */
|
|
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];
|
|
}
|
|
// TODO: axis-angle...
|
|
}
|
|
|
|
/* 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 (PointerRNA ptr, PropertyRNA *prop, FCurve *fcu, float cfra, short flag)
|
|
{
|
|
float curval= 0.0f;
|
|
|
|
/* no F-Curve to add keyframe to? */
|
|
if (fcu == NULL) {
|
|
printf("ERROR: no F-Curve to add keyframes to \n");
|
|
return 0;
|
|
}
|
|
/* F-Curve not editable? */
|
|
if ( (fcu->flag & FCURVE_PROTECTED) || ((fcu->grp) && (fcu->grp->flag & AGRP_PROTECTED)) ) {
|
|
if (G.f & G_DEBUG)
|
|
printf("WARNING: not inserting keyframe for locked F-Curve \n");
|
|
return 0;
|
|
}
|
|
|
|
/* if no property given yet, try to validate from F-Curve info */
|
|
if ((ptr.id.data == NULL) && (ptr.data==NULL)) {
|
|
printf("ERROR: no RNA-pointer available to retrieve values for keyframing from\n");
|
|
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... */
|
|
char *idname= (ptr.id.data) ? ((ID *)ptr.id.data)->name : "<No ID-Pointer>";
|
|
|
|
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);
|
|
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 (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) {
|
|
printf("Insert Key: no ID-block to insert keyframe in (Path = %s) \n", rna_path);
|
|
return 0;
|
|
}
|
|
|
|
RNA_id_pointer_create(id, &id_ptr);
|
|
if ((RNA_path_resolve(&id_ptr, rna_path, &ptr, &prop) == 0) || (prop == NULL)) {
|
|
printf("Insert Key: Could not insert keyframe, as RNA Path is invalid for the given ID (ID = %s, Path = %s)\n",
|
|
(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) {
|
|
printf("Insert Key: Could not insert keyframe, as this type does not support animation data (ID = %s, Path = %s)\n", 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);
|
|
}
|
|
|
|
#if 0
|
|
/* apply special time tweaking */
|
|
// XXX check on this stuff...
|
|
if (GS(id->name) == ID_OB) {
|
|
//Object *ob= (Object *)id;
|
|
|
|
/* ancient time-offset cruft */
|
|
//if ( (ob->ipoflag & OB_OFFS_OB) && (give_timeoffset(ob)) ) {
|
|
// /* actually frametofloat calc again! */
|
|
// cfra-= give_timeoffset(ob)*scene->r.framelen;
|
|
//}
|
|
}
|
|
#endif
|
|
|
|
/* 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 (ELEM4(RNA_property_subtype(prop), PROP_TRANSLATION, PROP_XYZ, PROP_EULER, PROP_COLOR)) {
|
|
fcu->color_mode= FCURVE_COLOR_AUTO_RGB;
|
|
}
|
|
}
|
|
|
|
/* insert keyframe */
|
|
ret += insert_keyframe_direct(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 (ID *id, bAction *act, const char group[], const char rna_path[], int array_index, float cfra, short 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) {
|
|
printf("ERROR: no ID-block and/or AnimData to delete keyframe from \n");
|
|
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)) {
|
|
printf("Delete Key: Could not delete keyframe, as RNA Path is invalid for the given ID (ID = %s, Path = %s)\n", 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 {
|
|
printf("ERROR: no Action to delete keyframes from for ID = %s \n", id->name);
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
#if 0
|
|
/* apply special time tweaking */
|
|
// XXX check on this stuff...
|
|
if (GS(id->name) == ID_OB) {
|
|
//Object *ob= (Object *)id;
|
|
|
|
/* ancient time-offset cruft */
|
|
//if ( (ob->ipoflag & OB_OFFS_OB) && (give_timeoffset(ob)) ) {
|
|
// /* actually frametofloat calc again! */
|
|
// cfra-= give_timeoffset(ob)*scene->r.framelen;
|
|
//}
|
|
}
|
|
#endif
|
|
|
|
/* 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 inserting 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)
|
|
{
|
|
Scene *scene= CTX_data_scene(C);
|
|
KeyingSet *ks= NULL;
|
|
int type= RNA_int_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(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
|
|
* - here the type is int not enum, since many of the indicies here are determined dynamically
|
|
*/
|
|
prop= 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);
|
|
RNA_def_property_flag(prop, PROP_HIDDEN);
|
|
/* 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 *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_int_set(op->ptr, "type", 0);
|
|
RNA_boolean_set(op->ptr, "confirm_success", 1);
|
|
|
|
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";
|
|
|
|
/* 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
|
|
* - here the type is int not enum, since many of the indicies here are determined dynamically
|
|
*/
|
|
prop= 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);
|
|
RNA_def_property_flag(prop, PROP_HIDDEN);
|
|
/* 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)
|
|
{
|
|
Scene *scene= CTX_data_scene(C);
|
|
KeyingSet *ks= NULL;
|
|
int type= RNA_int_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(0);
|
|
|
|
return OPERATOR_FINISHED;
|
|
}
|
|
|
|
void ANIM_OT_keyframe_delete (wmOperatorType *ot)
|
|
{
|
|
/* identifiers */
|
|
ot->name= "Delete 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
|
|
* - here the type is int not enum, since many of the indicies here are determined dynamically
|
|
*/
|
|
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);
|
|
/* 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)
|
|
{
|
|
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) {
|
|
AnimData *adt= ob->adt;
|
|
bAction *act= adt->action;
|
|
|
|
for (fcu= act->curves.first; fcu; fcu= fcn) {
|
|
fcn= fcu->next;
|
|
success+= delete_keyframe(id, NULL, NULL, fcu->rna_path, fcu->array_index, cfra, 0);
|
|
}
|
|
}
|
|
|
|
BKE_reportf(op->reports, RPT_INFO, "Ob '%s' - Successfully removed %d keyframes \n", id->name+2, success);
|
|
|
|
ob->recalc |= OB_RECALC_OB;
|
|
}
|
|
CTX_DATA_END;
|
|
|
|
/* send updates */
|
|
DAG_ids_flush_update(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->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)
|
|
{
|
|
Scene *scene= CTX_data_scene(C);
|
|
PointerRNA ptr;
|
|
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 */
|
|
memset(&ptr, 0, sizeof(PointerRNA));
|
|
uiAnimContextProperty(C, &ptr, &prop, &index);
|
|
|
|
if ((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(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(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,", ptr.data, prop);
|
|
if(prop)
|
|
printf("animateable = %d \n", RNA_property_animateable(&ptr, prop));
|
|
else
|
|
printf("animateable = NULL \n");
|
|
}
|
|
|
|
if (success) {
|
|
/* send updates */
|
|
DAG_ids_flush_update(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)
|
|
{
|
|
Scene *scene= CTX_data_scene(C);
|
|
PointerRNA ptr;
|
|
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 */
|
|
memset(&ptr, 0, sizeof(PointerRNA));
|
|
uiAnimContextProperty(C, &ptr, &prop, &index);
|
|
|
|
if (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(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", ptr.data, prop);
|
|
}
|
|
|
|
|
|
if (success) {
|
|
/* send updates */
|
|
DAG_ids_flush_update(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...
|
|
*/
|
|
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 */
|
|
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;
|
|
}
|
|
|
|
/* ************************************************** */
|