2110 lines
		
	
	
		
			60 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			2110 lines
		
	
	
		
			60 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) 2001-2002 by NaN Holding BV.
 | 
						|
 * All rights reserved.
 | 
						|
 *
 | 
						|
 * The Original Code is: all of this file.
 | 
						|
 *
 | 
						|
 * Contributor(s): Joshua Leung
 | 
						|
 *
 | 
						|
 * ***** END GPL LICENSE BLOCK *****
 | 
						|
 */
 | 
						|
 | 
						|
#include <math.h>
 | 
						|
#include <stdlib.h>
 | 
						|
#include <string.h>
 | 
						|
#include <float.h>
 | 
						|
 | 
						|
#include "AUD_C-API.h"
 | 
						|
 | 
						|
#include "MEM_guardedalloc.h"
 | 
						|
 | 
						|
#include "BLI_blenlib.h"
 | 
						|
#include "BLI_math.h"
 | 
						|
 | 
						|
#include "DNA_anim_types.h"
 | 
						|
#include "DNA_object_types.h"
 | 
						|
#include "DNA_scene_types.h"
 | 
						|
 | 
						|
#include "RNA_access.h"
 | 
						|
#include "RNA_define.h"
 | 
						|
#include "RNA_enum_types.h"
 | 
						|
 | 
						|
#include "BKE_fcurve.h"
 | 
						|
#include "BKE_nla.h"
 | 
						|
#include "BKE_context.h"
 | 
						|
#include "BKE_report.h"
 | 
						|
 | 
						|
#include "UI_interface.h"
 | 
						|
#include "UI_view2d.h"
 | 
						|
 | 
						|
#include "ED_anim_api.h"
 | 
						|
#include "ED_keyframing.h"
 | 
						|
#include "ED_keyframes_edit.h"
 | 
						|
#include "ED_screen.h"
 | 
						|
#include "ED_transform.h"
 | 
						|
 | 
						|
#include "WM_api.h"
 | 
						|
#include "WM_types.h"
 | 
						|
 | 
						|
#include "graph_intern.h"
 | 
						|
 | 
						|
/* ************************************************************************** */
 | 
						|
/* KEYFRAME-RANGE STUFF */
 | 
						|
 | 
						|
/* *************************** Calculate Range ************************** */
 | 
						|
 | 
						|
/* Get the min/max keyframes*/
 | 
						|
/* note: it should return total boundbox, filter for selection only can be argument... */
 | 
						|
void get_graph_keyframe_extents (bAnimContext *ac, float *xmin, float *xmax, float *ymin, float *ymax)
 | 
						|
{
 | 
						|
	ListBase anim_data = {NULL, NULL};
 | 
						|
	bAnimListElem *ale;
 | 
						|
	int filter;
 | 
						|
	
 | 
						|
	/* get data to filter, from Dopesheet */
 | 
						|
	filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_CURVESONLY | ANIMFILTER_NODUPLIS);
 | 
						|
	ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
 | 
						|
	
 | 
						|
	/* set large values to try to override */
 | 
						|
	if (xmin) *xmin= 999999999.0f;
 | 
						|
	if (xmax) *xmax= -999999999.0f;
 | 
						|
	if (ymin) *ymin= 999999999.0f;
 | 
						|
	if (ymax) *ymax= -999999999.0f;
 | 
						|
	
 | 
						|
	/* check if any channels to set range with */
 | 
						|
	if (anim_data.first) {
 | 
						|
		/* go through channels, finding max extents */
 | 
						|
		for (ale= anim_data.first; ale; ale= ale->next) {
 | 
						|
			AnimData *adt= ANIM_nla_mapping_get(ac, ale);
 | 
						|
			FCurve *fcu= (FCurve *)ale->key_data;
 | 
						|
			float txmin, txmax, tymin, tymax;
 | 
						|
			float unitFac;
 | 
						|
			
 | 
						|
			/* get range */
 | 
						|
			calc_fcurve_bounds(fcu, &txmin, &txmax, &tymin, &tymax);
 | 
						|
			
 | 
						|
			/* apply NLA scaling */
 | 
						|
			if (adt) {
 | 
						|
				txmin= BKE_nla_tweakedit_remap(adt, txmin, NLATIME_CONVERT_MAP);
 | 
						|
				txmax= BKE_nla_tweakedit_remap(adt, txmax, NLATIME_CONVERT_MAP);
 | 
						|
			}
 | 
						|
			
 | 
						|
			/* apply unit corrections */
 | 
						|
			unitFac= ANIM_unit_mapping_get_factor(ac->scene, ale->id, fcu, 0);
 | 
						|
			tymin *= unitFac;
 | 
						|
			tymax *= unitFac;
 | 
						|
			
 | 
						|
			/* try to set cur using these values, if they're more extreme than previously set values */
 | 
						|
			if ((xmin) && (txmin < *xmin)) 		*xmin= txmin;
 | 
						|
			if ((xmax) && (txmax > *xmax)) 		*xmax= txmax;
 | 
						|
			if ((ymin) && (tymin < *ymin)) 		*ymin= tymin;
 | 
						|
			if ((ymax) && (tymax > *ymax)) 		*ymax= tymax;
 | 
						|
		}
 | 
						|
		
 | 
						|
		/* free memory */
 | 
						|
		BLI_freelistN(&anim_data);
 | 
						|
	}
 | 
						|
	else {
 | 
						|
		/* set default range */
 | 
						|
		if (ac->scene) {
 | 
						|
			if (xmin) *xmin= (float)ac->scene->r.sfra;
 | 
						|
			if (xmax) *xmax= (float)ac->scene->r.efra;
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			if (xmin) *xmin= -5;
 | 
						|
			if (xmax) *xmax= 100;
 | 
						|
		}
 | 
						|
		
 | 
						|
		if (ymin) *ymin= -5;
 | 
						|
		if (ymax) *ymax= 5;
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/* ****************** Automatic Preview-Range Operator ****************** */
 | 
						|
 | 
						|
static int graphkeys_previewrange_exec(bContext *C, wmOperator *UNUSED(op))
 | 
						|
{
 | 
						|
	bAnimContext ac;
 | 
						|
	Scene *scene;
 | 
						|
	float min, max;
 | 
						|
	
 | 
						|
	/* get editor data */
 | 
						|
	if (ANIM_animdata_get_context(C, &ac) == 0)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
	if (ac.scene == NULL)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
	else
 | 
						|
		scene= ac.scene;
 | 
						|
	
 | 
						|
	/* set the range directly */
 | 
						|
	get_graph_keyframe_extents(&ac, &min, &max, NULL, NULL);
 | 
						|
	scene->r.flag |= SCER_PRV_RANGE;
 | 
						|
	scene->r.psfra= (int)floor(min + 0.5f);
 | 
						|
	scene->r.pefra= (int)floor(max + 0.5f);
 | 
						|
	
 | 
						|
	/* set notifier that things have changed */
 | 
						|
	// XXX err... there's nothing for frame ranges yet, but this should do fine too
 | 
						|
	WM_event_add_notifier(C, NC_SCENE|ND_FRAME, ac.scene); 
 | 
						|
	
 | 
						|
	return OPERATOR_FINISHED;
 | 
						|
}
 | 
						|
 
 | 
						|
void GRAPH_OT_previewrange_set (wmOperatorType *ot)
 | 
						|
{
 | 
						|
	/* identifiers */
 | 
						|
	ot->name= "Auto-Set Preview Range";
 | 
						|
	ot->idname= "GRAPH_OT_previewrange_set";
 | 
						|
	
 | 
						|
	/* api callbacks */
 | 
						|
	ot->exec= graphkeys_previewrange_exec;
 | 
						|
	ot->poll= graphop_visible_keyframes_poll;
 | 
						|
	
 | 
						|
	/* flags */
 | 
						|
	ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 | 
						|
}
 | 
						|
 | 
						|
/* ****************** View-All Operator ****************** */
 | 
						|
 | 
						|
static int graphkeys_viewall_exec(bContext *C, wmOperator *UNUSED(op))
 | 
						|
{
 | 
						|
	bAnimContext ac;
 | 
						|
	View2D *v2d;
 | 
						|
	float extra;
 | 
						|
	
 | 
						|
	/* get editor data */
 | 
						|
	if (ANIM_animdata_get_context(C, &ac) == 0)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
	v2d= &ac.ar->v2d;
 | 
						|
	
 | 
						|
	/* set the horizontal range, with an extra offset so that the extreme keys will be in view */
 | 
						|
	get_graph_keyframe_extents(&ac, &v2d->cur.xmin, &v2d->cur.xmax, &v2d->cur.ymin, &v2d->cur.ymax);
 | 
						|
	
 | 
						|
	extra= 0.1f * (v2d->cur.xmax - v2d->cur.xmin);
 | 
						|
	v2d->cur.xmin -= extra;
 | 
						|
	v2d->cur.xmax += extra;
 | 
						|
	
 | 
						|
	extra= 0.1f * (v2d->cur.ymax - v2d->cur.ymin);
 | 
						|
	v2d->cur.ymin -= extra;
 | 
						|
	v2d->cur.ymax += extra;
 | 
						|
	
 | 
						|
	/* do View2D syncing */
 | 
						|
	UI_view2d_sync(CTX_wm_screen(C), CTX_wm_area(C), v2d, V2D_LOCK_COPY);
 | 
						|
	
 | 
						|
	/* set notifier that things have changed */
 | 
						|
	ED_area_tag_redraw(CTX_wm_area(C));
 | 
						|
	
 | 
						|
	return OPERATOR_FINISHED;
 | 
						|
}
 | 
						|
 
 | 
						|
void GRAPH_OT_view_all (wmOperatorType *ot)
 | 
						|
{
 | 
						|
	/* identifiers */
 | 
						|
	ot->name= "View All";
 | 
						|
	ot->idname= "GRAPH_OT_view_all";
 | 
						|
	ot->description= "Reset viewable area to show full keyframe range";
 | 
						|
	
 | 
						|
	/* api callbacks */
 | 
						|
	ot->exec= graphkeys_viewall_exec;
 | 
						|
	ot->poll= graphop_visible_keyframes_poll;
 | 
						|
	
 | 
						|
	/* flags */
 | 
						|
	ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 | 
						|
}
 | 
						|
 | 
						|
/* ******************** Create Ghost-Curves Operator *********************** */
 | 
						|
/* This operator samples the data of the selected F-Curves to F-Points, storing them
 | 
						|
 * as 'ghost curves' in the active Graph Editor
 | 
						|
 */
 | 
						|
 | 
						|
/* Bake each F-Curve into a set of samples, and store as a ghost curve */
 | 
						|
static void create_ghost_curves (bAnimContext *ac, int start, int end)
 | 
						|
{	
 | 
						|
	SpaceIpo *sipo= (SpaceIpo *)ac->sa->spacedata.first;
 | 
						|
	ListBase anim_data = {NULL, NULL};
 | 
						|
	bAnimListElem *ale;
 | 
						|
	int filter;
 | 
						|
	
 | 
						|
	/* free existing ghost curves */
 | 
						|
	free_fcurves(&sipo->ghostCurves);
 | 
						|
	
 | 
						|
	/* sanity check */
 | 
						|
	if (start >= end) {
 | 
						|
		printf("Error: Frame range for Ghost F-Curve creation is inappropriate \n");
 | 
						|
		return;
 | 
						|
	}
 | 
						|
	
 | 
						|
	/* filter data */
 | 
						|
	filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_SEL | ANIMFILTER_CURVESONLY | ANIMFILTER_NODUPLIS);
 | 
						|
	ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
 | 
						|
	
 | 
						|
	/* loop through filtered data and add keys between selected keyframes on every frame  */
 | 
						|
	for (ale= anim_data.first; ale; ale= ale->next) {
 | 
						|
		FCurve *fcu= (FCurve *)ale->key_data;
 | 
						|
		FCurve *gcu= MEM_callocN(sizeof(FCurve), "Ghost FCurve");
 | 
						|
		AnimData *adt= ANIM_nla_mapping_get(ac, ale);
 | 
						|
		ChannelDriver *driver= fcu->driver;
 | 
						|
		FPoint *fpt;
 | 
						|
		float unitFac;
 | 
						|
		int cfra;		
 | 
						|
		
 | 
						|
		/* disable driver so that it don't muck up the sampling process */
 | 
						|
		fcu->driver= NULL;
 | 
						|
		
 | 
						|
		/* calculate unit-mapping factor */
 | 
						|
		unitFac= ANIM_unit_mapping_get_factor(ac->scene, ale->id, fcu, 0);
 | 
						|
		
 | 
						|
		/* create samples, but store them in a new curve 
 | 
						|
		 *	- we cannot use fcurve_store_samples() as that will only overwrite the original curve 
 | 
						|
		 */
 | 
						|
		gcu->fpt= fpt= MEM_callocN(sizeof(FPoint)*(end-start+1), "Ghost FPoint Samples");
 | 
						|
		gcu->totvert= end - start + 1;
 | 
						|
		
 | 
						|
		/* use the sampling callback at 1-frame intervals from start to end frames */
 | 
						|
		for (cfra= start; cfra <= end; cfra++, fpt++) {
 | 
						|
			float cfrae= BKE_nla_tweakedit_remap(adt, cfra, NLATIME_CONVERT_UNMAP);
 | 
						|
			
 | 
						|
			fpt->vec[0]= cfrae;
 | 
						|
			fpt->vec[1]= fcurve_samplingcb_evalcurve(fcu, NULL, cfrae) * unitFac;
 | 
						|
		}
 | 
						|
		
 | 
						|
		/* set color of ghost curve 
 | 
						|
		 *	- make the color slightly darker
 | 
						|
		 */
 | 
						|
		gcu->color[0]= fcu->color[0] - 0.07f;
 | 
						|
		gcu->color[1]= fcu->color[1] - 0.07f;
 | 
						|
		gcu->color[2]= fcu->color[2] - 0.07f;
 | 
						|
		
 | 
						|
		/* store new ghost curve */
 | 
						|
		BLI_addtail(&sipo->ghostCurves, gcu);
 | 
						|
		
 | 
						|
		/* restore driver */
 | 
						|
		fcu->driver= driver;
 | 
						|
	}
 | 
						|
	
 | 
						|
	/* admin and redraws */
 | 
						|
	BLI_freelistN(&anim_data);
 | 
						|
}
 | 
						|
 | 
						|
/* ------------------- */
 | 
						|
 | 
						|
static int graphkeys_create_ghostcurves_exec(bContext *C, wmOperator *UNUSED(op))
 | 
						|
{
 | 
						|
	bAnimContext ac;
 | 
						|
	View2D *v2d;
 | 
						|
	int start, end;
 | 
						|
	
 | 
						|
	/* get editor data */
 | 
						|
	if (ANIM_animdata_get_context(C, &ac) == 0)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
		
 | 
						|
	/* ghost curves are snapshots of the visible portions of the curves, so set range to be the visible range */
 | 
						|
	v2d= &ac.ar->v2d;
 | 
						|
	start= (int)v2d->cur.xmin;
 | 
						|
	end= (int)v2d->cur.xmax;
 | 
						|
	
 | 
						|
	/* bake selected curves into a ghost curve */
 | 
						|
	create_ghost_curves(&ac, start, end);
 | 
						|
	
 | 
						|
	/* update this editor only */
 | 
						|
	ED_area_tag_redraw(CTX_wm_area(C));
 | 
						|
	
 | 
						|
	return OPERATOR_FINISHED;
 | 
						|
}
 | 
						|
 
 | 
						|
void GRAPH_OT_ghost_curves_create (wmOperatorType *ot)
 | 
						|
{
 | 
						|
	/* identifiers */
 | 
						|
	ot->name= "Create Ghost Curves";
 | 
						|
	ot->idname= "GRAPH_OT_ghost_curves_create";
 | 
						|
	ot->description= "Create snapshot (Ghosts) of selected F-Curves as background aid for active Graph Editor";
 | 
						|
	
 | 
						|
	/* api callbacks */
 | 
						|
	ot->exec= graphkeys_create_ghostcurves_exec;
 | 
						|
	ot->poll= graphop_visible_keyframes_poll;
 | 
						|
	
 | 
						|
	/* flags */
 | 
						|
	ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 | 
						|
	
 | 
						|
	// todo: add props for start/end frames
 | 
						|
}
 | 
						|
 | 
						|
/* ******************** Clear Ghost-Curves Operator *********************** */
 | 
						|
/* This operator clears the 'ghost curves' for the active Graph Editor */
 | 
						|
 | 
						|
static int graphkeys_clear_ghostcurves_exec(bContext *C, wmOperator *UNUSED(op))
 | 
						|
{
 | 
						|
	bAnimContext ac;
 | 
						|
	SpaceIpo *sipo;
 | 
						|
	
 | 
						|
	/* get editor data */
 | 
						|
	if (ANIM_animdata_get_context(C, &ac) == 0)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
	sipo= (SpaceIpo *)ac.sa->spacedata.first;
 | 
						|
		
 | 
						|
	/* if no ghost curves, don't do anything */
 | 
						|
	if (sipo->ghostCurves.first == NULL)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
	
 | 
						|
	/* free ghost curves */
 | 
						|
	free_fcurves(&sipo->ghostCurves);
 | 
						|
	
 | 
						|
	/* update this editor only */
 | 
						|
	ED_area_tag_redraw(CTX_wm_area(C));
 | 
						|
	
 | 
						|
	return OPERATOR_FINISHED;
 | 
						|
}
 | 
						|
 
 | 
						|
void GRAPH_OT_ghost_curves_clear (wmOperatorType *ot)
 | 
						|
{
 | 
						|
	/* identifiers */
 | 
						|
	ot->name= "Clear Ghost Curves";
 | 
						|
	ot->idname= "GRAPH_OT_ghost_curves_clear";
 | 
						|
	ot->description= "Clear F-Curve snapshots (Ghosts) for active Graph Editor";
 | 
						|
	
 | 
						|
	/* api callbacks */
 | 
						|
	ot->exec= graphkeys_clear_ghostcurves_exec;
 | 
						|
	ot->poll= ED_operator_ipo_active;
 | 
						|
	
 | 
						|
	/* flags */
 | 
						|
	ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 | 
						|
}
 | 
						|
 | 
						|
/* ************************************************************************** */
 | 
						|
/* GENERAL STUFF */
 | 
						|
 | 
						|
/* ******************** Insert Keyframes Operator ************************* */
 | 
						|
 | 
						|
/* defines for insert keyframes tool */
 | 
						|
EnumPropertyItem prop_graphkeys_insertkey_types[] = {
 | 
						|
	{1, "ALL", 0, "All Channels", ""},
 | 
						|
	{2, "SEL", 0, "Only Selected Channels", ""},
 | 
						|
	{0, NULL, 0, NULL, NULL}
 | 
						|
};
 | 
						|
 | 
						|
/* this function is responsible for snapping keyframes to frame-times */
 | 
						|
static void insert_graph_keys(bAnimContext *ac, short mode) 
 | 
						|
{
 | 
						|
	ListBase anim_data = {NULL, NULL};
 | 
						|
	bAnimListElem *ale;
 | 
						|
	int filter;
 | 
						|
	
 | 
						|
	Scene *scene= ac->scene;
 | 
						|
	float cfra= (float)CFRA;
 | 
						|
	short flag = 0;
 | 
						|
	
 | 
						|
	/* filter data */
 | 
						|
	filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_FOREDIT | ANIMFILTER_CURVESONLY | ANIMFILTER_NODUPLIS);
 | 
						|
	if (mode == 2) filter |= ANIMFILTER_SEL;
 | 
						|
	
 | 
						|
	ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
 | 
						|
	
 | 
						|
	/* init keyframing flag */
 | 
						|
	flag = ANIM_get_keyframing_flags(scene, 1);
 | 
						|
	
 | 
						|
	/* insert keyframes */
 | 
						|
	for (ale= anim_data.first; ale; ale= ale->next) {
 | 
						|
		AnimData *adt= ANIM_nla_mapping_get(ac, ale);
 | 
						|
		FCurve *fcu= (FCurve *)ale->key_data;
 | 
						|
		
 | 
						|
		/* adjust current frame for NLA-mapping */
 | 
						|
		if (adt)
 | 
						|
			cfra= BKE_nla_tweakedit_remap(adt, (float)CFRA, NLATIME_CONVERT_UNMAP);
 | 
						|
		else 
 | 
						|
			cfra= (float)CFRA;
 | 
						|
			
 | 
						|
		/* if there's an id */
 | 
						|
		if (ale->id)
 | 
						|
			insert_keyframe(ale->id, NULL, ((fcu->grp)?(fcu->grp->name):(NULL)), fcu->rna_path, fcu->array_index, cfra, flag);
 | 
						|
		else
 | 
						|
			insert_vert_fcurve(fcu, cfra, fcu->curval, 0);
 | 
						|
	}
 | 
						|
	
 | 
						|
	BLI_freelistN(&anim_data);
 | 
						|
}
 | 
						|
 | 
						|
/* ------------------- */
 | 
						|
 | 
						|
static int graphkeys_insertkey_exec(bContext *C, wmOperator *op)
 | 
						|
{
 | 
						|
	bAnimContext ac;
 | 
						|
	short mode;
 | 
						|
	
 | 
						|
	/* get editor data */
 | 
						|
	if (ANIM_animdata_get_context(C, &ac) == 0)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
		
 | 
						|
	/* which channels to affect? */
 | 
						|
	mode= RNA_enum_get(op->ptr, "type");
 | 
						|
	
 | 
						|
	/* insert keyframes */
 | 
						|
	insert_graph_keys(&ac, mode);
 | 
						|
	
 | 
						|
	/* validate keyframes after editing */
 | 
						|
	ANIM_editkeyframes_refresh(&ac);
 | 
						|
	
 | 
						|
	/* set notifier that keyframes have changed */
 | 
						|
	WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_EDITED, NULL);
 | 
						|
	
 | 
						|
	return OPERATOR_FINISHED;
 | 
						|
}
 | 
						|
 | 
						|
void GRAPH_OT_keyframe_insert (wmOperatorType *ot)
 | 
						|
{
 | 
						|
	/* identifiers */
 | 
						|
	ot->name= "Insert Keyframes";
 | 
						|
	ot->idname= "GRAPH_OT_keyframe_insert";
 | 
						|
	ot->description= "Insert keyframes for the specified channels";
 | 
						|
	
 | 
						|
	/* api callbacks */
 | 
						|
	ot->invoke= WM_menu_invoke;
 | 
						|
	ot->exec= graphkeys_insertkey_exec;
 | 
						|
	ot->poll= graphop_editable_keyframes_poll;
 | 
						|
	
 | 
						|
	/* flags */
 | 
						|
	ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 | 
						|
	
 | 
						|
	/* id-props */
 | 
						|
	ot->prop= RNA_def_enum(ot->srna, "type", prop_graphkeys_insertkey_types, 0, "Type", "");
 | 
						|
}
 | 
						|
 | 
						|
/* ******************** Click-Insert Keyframes Operator ************************* */
 | 
						|
 | 
						|
static int graphkeys_click_insert_exec (bContext *C, wmOperator *op)
 | 
						|
{
 | 
						|
	bAnimContext ac;
 | 
						|
	bAnimListElem *ale;
 | 
						|
	AnimData *adt;
 | 
						|
	FCurve *fcu;
 | 
						|
	float frame, val;
 | 
						|
	
 | 
						|
	/* get animation context */
 | 
						|
	if (ANIM_animdata_get_context(C, &ac) == 0)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
	
 | 
						|
	/* get active F-Curve 'anim-list-element' */
 | 
						|
	ale= get_active_fcurve_channel(&ac);
 | 
						|
	if (ELEM(NULL, ale, ale->data)) {
 | 
						|
		if (ale) MEM_freeN(ale);
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
	}
 | 
						|
	fcu = ale->data;
 | 
						|
	
 | 
						|
	/* get frame and value from props */
 | 
						|
	frame= RNA_float_get(op->ptr, "frame");
 | 
						|
	val= RNA_float_get(op->ptr, "value");
 | 
						|
	
 | 
						|
	/* apply inverse NLA-mapping to frame to get correct time in un-scaled action */
 | 
						|
	adt= ANIM_nla_mapping_get(&ac, ale);
 | 
						|
	frame= BKE_nla_tweakedit_remap(adt, frame, NLATIME_CONVERT_UNMAP);
 | 
						|
	
 | 
						|
	/* apply inverse unit-mapping to value to get correct value for F-Curves */
 | 
						|
	val *= ANIM_unit_mapping_get_factor(ac.scene, ale->id, fcu, 1);
 | 
						|
	
 | 
						|
	/* insert keyframe on the specified frame + value */
 | 
						|
	insert_vert_fcurve(fcu, frame, val, 0);
 | 
						|
	
 | 
						|
	/* free temp data */
 | 
						|
	MEM_freeN(ale);
 | 
						|
	
 | 
						|
	/* set notifier that keyframes have changed */
 | 
						|
	WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_EDITED, NULL);
 | 
						|
	
 | 
						|
	/* done */
 | 
						|
	return OPERATOR_FINISHED;
 | 
						|
}
 | 
						|
 | 
						|
static int graphkeys_click_insert_invoke (bContext *C, wmOperator *op, wmEvent *evt)
 | 
						|
{
 | 
						|
	bAnimContext ac;
 | 
						|
	ARegion *ar;
 | 
						|
	View2D *v2d;
 | 
						|
	int mval[2];
 | 
						|
	float x, y;
 | 
						|
	
 | 
						|
	/* get animation context */
 | 
						|
	if (ANIM_animdata_get_context(C, &ac) == 0)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
	
 | 
						|
	/* store mouse coordinates in View2D space, into the operator's properties */
 | 
						|
	ar= ac.ar;
 | 
						|
	v2d= &ar->v2d;
 | 
						|
	
 | 
						|
	mval[0]= (evt->x - ar->winrct.xmin);
 | 
						|
	mval[1]= (evt->y - ar->winrct.ymin);
 | 
						|
	
 | 
						|
	UI_view2d_region_to_view(v2d, mval[0], mval[1], &x, &y);
 | 
						|
	
 | 
						|
	RNA_float_set(op->ptr, "frame", x);
 | 
						|
	RNA_float_set(op->ptr, "value", y);
 | 
						|
	
 | 
						|
	/* run exec now */
 | 
						|
	return graphkeys_click_insert_exec(C, op);
 | 
						|
}
 | 
						|
 | 
						|
void GRAPH_OT_click_insert (wmOperatorType *ot)
 | 
						|
{
 | 
						|
	/* identifiers */
 | 
						|
	ot->name= "Click-Insert Keyframes";
 | 
						|
	ot->idname= "GRAPH_OT_click_insert";
 | 
						|
	ot->description= "Insert new keyframe at the cursor position for the active F-Curve";
 | 
						|
	
 | 
						|
	/* api callbacks */
 | 
						|
	ot->invoke= graphkeys_click_insert_invoke;
 | 
						|
	ot->exec= graphkeys_click_insert_exec;
 | 
						|
	ot->poll= graphop_active_fcurve_poll;
 | 
						|
	
 | 
						|
	/* flags */
 | 
						|
	ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 | 
						|
	
 | 
						|
	/* properties */
 | 
						|
	RNA_def_float(ot->srna, "frame", 1.0f, -FLT_MAX, FLT_MAX, "Frame Number", "Frame to insert keyframe on", 0, 100);
 | 
						|
	RNA_def_float(ot->srna, "value", 1.0f, -FLT_MAX, FLT_MAX, "Value", "Value for keyframe on", 0, 100);
 | 
						|
}
 | 
						|
 | 
						|
/* ******************** Copy/Paste Keyframes Operator ************************* */
 | 
						|
/* NOTE: the backend code for this is shared with the dopesheet editor */
 | 
						|
 | 
						|
static short copy_graph_keys (bAnimContext *ac)
 | 
						|
{	
 | 
						|
	ListBase anim_data = {NULL, NULL};
 | 
						|
	int filter, ok=0;
 | 
						|
	
 | 
						|
	/* clear buffer first */
 | 
						|
	free_anim_copybuf();
 | 
						|
	
 | 
						|
	/* filter data */
 | 
						|
	filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_CURVESONLY | ANIMFILTER_NODUPLIS);
 | 
						|
	ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
 | 
						|
	
 | 
						|
	/* copy keyframes */
 | 
						|
	ok= copy_animedit_keys(ac, &anim_data);
 | 
						|
	
 | 
						|
	/* clean up */
 | 
						|
	BLI_freelistN(&anim_data);
 | 
						|
 | 
						|
	return ok;
 | 
						|
}
 | 
						|
 | 
						|
static short paste_graph_keys (bAnimContext *ac)
 | 
						|
{	
 | 
						|
	ListBase anim_data = {NULL, NULL};
 | 
						|
	int filter, ok=0;
 | 
						|
	
 | 
						|
	/* filter data */
 | 
						|
	filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_SEL | ANIMFILTER_FOREDIT | ANIMFILTER_CURVESONLY | ANIMFILTER_NODUPLIS);
 | 
						|
	ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
 | 
						|
	
 | 
						|
	/* paste keyframes */
 | 
						|
	ok= paste_animedit_keys(ac, &anim_data);
 | 
						|
	
 | 
						|
	/* clean up */
 | 
						|
	BLI_freelistN(&anim_data);
 | 
						|
 | 
						|
	return ok;
 | 
						|
}
 | 
						|
 | 
						|
/* ------------------- */
 | 
						|
 | 
						|
static int graphkeys_copy_exec(bContext *C, wmOperator *op)
 | 
						|
{
 | 
						|
	bAnimContext ac;
 | 
						|
	
 | 
						|
	/* get editor data */
 | 
						|
	if (ANIM_animdata_get_context(C, &ac) == 0)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
	
 | 
						|
	/* copy keyframes */
 | 
						|
	if (copy_graph_keys(&ac)) {	
 | 
						|
		BKE_report(op->reports, RPT_ERROR, "No keyframes copied to keyframes copy/paste buffer");
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
	}
 | 
						|
	
 | 
						|
	/* just return - no operator needed here (no changes) */
 | 
						|
	return OPERATOR_FINISHED;
 | 
						|
}
 | 
						|
 
 | 
						|
void GRAPH_OT_copy (wmOperatorType *ot)
 | 
						|
{
 | 
						|
	/* identifiers */
 | 
						|
	ot->name= "Copy Keyframes";
 | 
						|
	ot->idname= "GRAPH_OT_copy";
 | 
						|
	ot->description= "Copy selected keyframes to the copy/paste buffer";
 | 
						|
	
 | 
						|
	/* api callbacks */
 | 
						|
	ot->exec= graphkeys_copy_exec;
 | 
						|
	ot->poll= graphop_editable_keyframes_poll;
 | 
						|
	
 | 
						|
	/* flags */
 | 
						|
	ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
static int graphkeys_paste_exec(bContext *C, wmOperator *op)
 | 
						|
{
 | 
						|
	bAnimContext ac;
 | 
						|
	
 | 
						|
	/* get editor data */
 | 
						|
	if (ANIM_animdata_get_context(C, &ac) == 0)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
	
 | 
						|
	/* paste keyframes */
 | 
						|
	if (paste_graph_keys(&ac)) {
 | 
						|
		BKE_report(op->reports, RPT_ERROR, "No keyframes to paste");
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
	}
 | 
						|
	
 | 
						|
	/* validate keyframes after editing */
 | 
						|
	ANIM_editkeyframes_refresh(&ac);
 | 
						|
	
 | 
						|
	/* set notifier that keyframes have changed */
 | 
						|
	WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_EDITED, NULL);
 | 
						|
	
 | 
						|
	return OPERATOR_FINISHED;
 | 
						|
}
 | 
						|
 
 | 
						|
void GRAPH_OT_paste (wmOperatorType *ot)
 | 
						|
{
 | 
						|
	/* identifiers */
 | 
						|
	ot->name= "Paste Keyframes";
 | 
						|
	ot->idname= "GRAPH_OT_paste";
 | 
						|
	ot->description= "Paste keyframes from copy/paste buffer for the selected channels, starting on the current frame";
 | 
						|
	
 | 
						|
	/* api callbacks */
 | 
						|
	ot->exec= graphkeys_paste_exec;
 | 
						|
	ot->poll= graphop_editable_keyframes_poll;
 | 
						|
	
 | 
						|
	/* flags */
 | 
						|
	ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 | 
						|
}
 | 
						|
 | 
						|
/* ******************** Duplicate Keyframes Operator ************************* */
 | 
						|
 | 
						|
static void duplicate_graph_keys (bAnimContext *ac)
 | 
						|
{
 | 
						|
	ListBase anim_data = {NULL, NULL};
 | 
						|
	bAnimListElem *ale;
 | 
						|
	int filter;
 | 
						|
	
 | 
						|
	/* filter data */
 | 
						|
	filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_FOREDIT | ANIMFILTER_CURVESONLY | ANIMFILTER_NODUPLIS);
 | 
						|
	ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
 | 
						|
	
 | 
						|
	/* loop through filtered data and delete selected keys */
 | 
						|
	for (ale= anim_data.first; ale; ale= ale->next) {
 | 
						|
		duplicate_fcurve_keys((FCurve *)ale->key_data);
 | 
						|
	}
 | 
						|
	
 | 
						|
	/* free filtered list */
 | 
						|
	BLI_freelistN(&anim_data);
 | 
						|
}
 | 
						|
 | 
						|
/* ------------------- */
 | 
						|
 | 
						|
static int graphkeys_duplicate_exec(bContext *C, wmOperator *UNUSED(op))
 | 
						|
{
 | 
						|
	bAnimContext ac;
 | 
						|
	
 | 
						|
	/* get editor data */
 | 
						|
	if (ANIM_animdata_get_context(C, &ac) == 0)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
		
 | 
						|
	/* duplicate keyframes */
 | 
						|
	duplicate_graph_keys(&ac);
 | 
						|
	
 | 
						|
	/* validate keyframes after editing */
 | 
						|
	ANIM_editkeyframes_refresh(&ac);
 | 
						|
	
 | 
						|
	/* set notifier that keyframes have changed */
 | 
						|
	WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_EDITED, NULL);
 | 
						|
	
 | 
						|
	return OPERATOR_FINISHED;
 | 
						|
}
 | 
						|
 | 
						|
static int graphkeys_duplicate_invoke(bContext *C, wmOperator *op, wmEvent *UNUSED(event))
 | 
						|
{
 | 
						|
	graphkeys_duplicate_exec(C, op);
 | 
						|
	
 | 
						|
	RNA_int_set(op->ptr, "mode", TFM_TRANSLATION);
 | 
						|
	WM_operator_name_call(C, "TRANSFORM_OT_transform", WM_OP_INVOKE_REGION_WIN, op->ptr);
 | 
						|
 | 
						|
	return OPERATOR_FINISHED;
 | 
						|
}
 | 
						|
 
 | 
						|
void GRAPH_OT_duplicate (wmOperatorType *ot)
 | 
						|
{
 | 
						|
	/* identifiers */
 | 
						|
	ot->name= "Duplicate Keyframes";
 | 
						|
	ot->idname= "GRAPH_OT_duplicate";
 | 
						|
	ot->description= "Make a copy of all selected keyframes";
 | 
						|
	
 | 
						|
	/* api callbacks */
 | 
						|
	ot->invoke= graphkeys_duplicate_invoke;
 | 
						|
	ot->exec= graphkeys_duplicate_exec;
 | 
						|
	ot->poll= graphop_editable_keyframes_poll;
 | 
						|
	
 | 
						|
	/* flags */
 | 
						|
	ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 | 
						|
	
 | 
						|
	/* to give to transform */
 | 
						|
	RNA_def_int(ot->srna, "mode", TFM_TRANSLATION, 0, INT_MAX, "Mode", "", 0, INT_MAX);
 | 
						|
}
 | 
						|
 | 
						|
/* ******************** Delete Keyframes Operator ************************* */
 | 
						|
 | 
						|
static void delete_graph_keys (bAnimContext *ac)
 | 
						|
{
 | 
						|
	ListBase anim_data = {NULL, NULL};
 | 
						|
	bAnimListElem *ale;
 | 
						|
	int filter;
 | 
						|
	
 | 
						|
	/* filter data */
 | 
						|
	filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_FOREDIT | ANIMFILTER_CURVESONLY | ANIMFILTER_NODUPLIS);
 | 
						|
	ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
 | 
						|
	
 | 
						|
	/* loop through filtered data and delete selected keys */
 | 
						|
	for (ale= anim_data.first; ale; ale= ale->next) {
 | 
						|
		FCurve *fcu= (FCurve *)ale->key_data;
 | 
						|
		AnimData *adt= ale->adt;
 | 
						|
		
 | 
						|
		/* delete selected keyframes only */
 | 
						|
		delete_fcurve_keys(fcu); 
 | 
						|
		
 | 
						|
		/* 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(ac, adt, fcu);
 | 
						|
	}
 | 
						|
	
 | 
						|
	/* free filtered list */
 | 
						|
	BLI_freelistN(&anim_data);
 | 
						|
}
 | 
						|
 | 
						|
/* ------------------- */
 | 
						|
 | 
						|
static int graphkeys_delete_exec(bContext *C, wmOperator *UNUSED(op))
 | 
						|
{
 | 
						|
	bAnimContext ac;
 | 
						|
	
 | 
						|
	/* get editor data */
 | 
						|
	if (ANIM_animdata_get_context(C, &ac) == 0)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
		
 | 
						|
	/* delete keyframes */
 | 
						|
	delete_graph_keys(&ac);
 | 
						|
	
 | 
						|
	/* validate keyframes after editing */
 | 
						|
	ANIM_editkeyframes_refresh(&ac);
 | 
						|
	
 | 
						|
	/* set notifier that keyframes have changed */
 | 
						|
	WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_EDITED, NULL);
 | 
						|
	
 | 
						|
	return OPERATOR_FINISHED;
 | 
						|
}
 | 
						|
 
 | 
						|
void GRAPH_OT_delete (wmOperatorType *ot)
 | 
						|
{
 | 
						|
	/* identifiers */
 | 
						|
	ot->name= "Delete Keyframes";
 | 
						|
	ot->idname= "GRAPH_OT_delete";
 | 
						|
	ot->description= "Remove all selected keyframes";
 | 
						|
	
 | 
						|
	/* api callbacks */
 | 
						|
	ot->invoke= WM_operator_confirm;
 | 
						|
	ot->exec= graphkeys_delete_exec;
 | 
						|
	ot->poll= graphop_editable_keyframes_poll;
 | 
						|
	
 | 
						|
	/* flags */
 | 
						|
	ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 | 
						|
}
 | 
						|
 | 
						|
/* ******************** Clean Keyframes Operator ************************* */
 | 
						|
 | 
						|
static void clean_graph_keys (bAnimContext *ac, float thresh)
 | 
						|
{	
 | 
						|
	ListBase anim_data = {NULL, NULL};
 | 
						|
	bAnimListElem *ale;
 | 
						|
	int filter;
 | 
						|
	
 | 
						|
	/* filter data */
 | 
						|
	filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_FOREDIT | ANIMFILTER_SEL | ANIMFILTER_CURVESONLY | ANIMFILTER_NODUPLIS);
 | 
						|
	ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
 | 
						|
	
 | 
						|
	/* loop through filtered data and clean curves */
 | 
						|
	for (ale= anim_data.first; ale; ale= ale->next)
 | 
						|
		clean_fcurve((FCurve *)ale->key_data, thresh);
 | 
						|
	
 | 
						|
	/* free temp data */
 | 
						|
	BLI_freelistN(&anim_data);
 | 
						|
}
 | 
						|
 | 
						|
/* ------------------- */
 | 
						|
 | 
						|
static int graphkeys_clean_exec(bContext *C, wmOperator *op)
 | 
						|
{
 | 
						|
	bAnimContext ac;
 | 
						|
	float thresh;
 | 
						|
	
 | 
						|
	/* get editor data */
 | 
						|
	if (ANIM_animdata_get_context(C, &ac) == 0)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
		
 | 
						|
	/* get cleaning threshold */
 | 
						|
	thresh= RNA_float_get(op->ptr, "threshold");
 | 
						|
	
 | 
						|
	/* clean keyframes */
 | 
						|
	clean_graph_keys(&ac, thresh);
 | 
						|
	
 | 
						|
	/* validate keyframes after editing */
 | 
						|
	ANIM_editkeyframes_refresh(&ac);
 | 
						|
	
 | 
						|
	/* set notifier that keyframes have changed */
 | 
						|
	WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_EDITED, NULL);
 | 
						|
	
 | 
						|
	return OPERATOR_FINISHED;
 | 
						|
}
 | 
						|
 
 | 
						|
void GRAPH_OT_clean (wmOperatorType *ot)
 | 
						|
{
 | 
						|
	/* identifiers */
 | 
						|
	ot->name= "Clean Keyframes";
 | 
						|
	ot->idname= "GRAPH_OT_clean";
 | 
						|
	ot->description= "Simplify F-Curves by removing closely spaced keyframes";
 | 
						|
	
 | 
						|
	/* api callbacks */
 | 
						|
	//ot->invoke=  // XXX we need that number popup for this! 
 | 
						|
	ot->exec= graphkeys_clean_exec;
 | 
						|
	ot->poll= graphop_editable_keyframes_poll;
 | 
						|
	
 | 
						|
	/* flags */
 | 
						|
	ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 | 
						|
	
 | 
						|
	/* properties */
 | 
						|
	ot->prop= RNA_def_float(ot->srna, "threshold", 0.001f, 0.0f, FLT_MAX, "Threshold", "", 0.0f, 1000.0f);
 | 
						|
}
 | 
						|
 | 
						|
/* ******************** Bake F-Curve Operator *********************** */
 | 
						|
/* This operator bakes the data of the selected F-Curves to F-Points */
 | 
						|
 | 
						|
/* Bake each F-Curve into a set of samples */
 | 
						|
static void bake_graph_curves (bAnimContext *ac, int start, int end)
 | 
						|
{	
 | 
						|
	ListBase anim_data = {NULL, NULL};
 | 
						|
	bAnimListElem *ale;
 | 
						|
	int filter;
 | 
						|
	
 | 
						|
	/* filter data */
 | 
						|
	filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_SEL | ANIMFILTER_FOREDIT | ANIMFILTER_CURVESONLY | ANIMFILTER_NODUPLIS);
 | 
						|
	ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
 | 
						|
	
 | 
						|
	/* loop through filtered data and add keys between selected keyframes on every frame  */
 | 
						|
	for (ale= anim_data.first; ale; ale= ale->next) {
 | 
						|
		FCurve *fcu= (FCurve *)ale->key_data;
 | 
						|
		ChannelDriver *driver= fcu->driver;
 | 
						|
		
 | 
						|
		/* disable driver so that it don't muck up the sampling process */
 | 
						|
		fcu->driver= NULL;
 | 
						|
		
 | 
						|
		/* create samples */
 | 
						|
		fcurve_store_samples(fcu, NULL, start, end, fcurve_samplingcb_evalcurve);
 | 
						|
		
 | 
						|
		/* restore driver */
 | 
						|
		fcu->driver= driver;
 | 
						|
	}
 | 
						|
	
 | 
						|
	/* admin and redraws */
 | 
						|
	BLI_freelistN(&anim_data);
 | 
						|
}
 | 
						|
 | 
						|
/* ------------------- */
 | 
						|
 | 
						|
static int graphkeys_bake_exec(bContext *C, wmOperator *UNUSED(op))
 | 
						|
{
 | 
						|
	bAnimContext ac;
 | 
						|
	Scene *scene= NULL;
 | 
						|
	int start, end;
 | 
						|
	
 | 
						|
	/* get editor data */
 | 
						|
	if (ANIM_animdata_get_context(C, &ac) == 0)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
		
 | 
						|
	/* for now, init start/end from preview-range extents */
 | 
						|
	// TODO: add properties for this 
 | 
						|
	scene= ac.scene;
 | 
						|
	start= PSFRA;
 | 
						|
	end= PEFRA;
 | 
						|
	
 | 
						|
	/* bake keyframes */
 | 
						|
	bake_graph_curves(&ac, start, end);
 | 
						|
	
 | 
						|
	/* validate keyframes after editing */
 | 
						|
	ANIM_editkeyframes_refresh(&ac);
 | 
						|
	
 | 
						|
	/* set notifier that keyframes have changed */
 | 
						|
	// NOTE: some distinction between order/number of keyframes and type should be made?
 | 
						|
	WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_EDITED, NULL);
 | 
						|
	
 | 
						|
	return OPERATOR_FINISHED;
 | 
						|
}
 | 
						|
 
 | 
						|
void GRAPH_OT_bake (wmOperatorType *ot)
 | 
						|
{
 | 
						|
	/* identifiers */
 | 
						|
	ot->name= "Bake Curve";
 | 
						|
	ot->idname= "GRAPH_OT_bake";
 | 
						|
	ot->description= "Bake selected F-Curves to a set of sampled points defining a similar curve";
 | 
						|
	
 | 
						|
	/* api callbacks */
 | 
						|
	ot->invoke= WM_operator_confirm; // FIXME...
 | 
						|
	ot->exec= graphkeys_bake_exec;
 | 
						|
	ot->poll= graphop_selected_fcurve_poll; 
 | 
						|
	
 | 
						|
	/* flags */
 | 
						|
	ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 | 
						|
	
 | 
						|
	// todo: add props for start/end frames
 | 
						|
}
 | 
						|
 | 
						|
/* ******************** Sound Bake F-Curve Operator *********************** */
 | 
						|
/* This operator bakes the given sound to the selected F-Curves */
 | 
						|
 | 
						|
/* ------------------- */
 | 
						|
 | 
						|
/* Custom data storage passed to the F-Sample-ing function,
 | 
						|
 * which provides the necessary info for baking the sound
 | 
						|
 */
 | 
						|
typedef struct tSoundBakeInfo {
 | 
						|
	float *samples;
 | 
						|
	int length;
 | 
						|
	int cfra;
 | 
						|
} tSoundBakeInfo;
 | 
						|
 | 
						|
/* ------------------- */
 | 
						|
 | 
						|
/* Sampling callback used to determine the value from the sound to
 | 
						|
 * save in the F-Curve at the specified frame
 | 
						|
 */
 | 
						|
static float fcurve_samplingcb_sound (FCurve *fcu, void *data, float evaltime)
 | 
						|
{
 | 
						|
	tSoundBakeInfo *sbi= (tSoundBakeInfo *)data;
 | 
						|
 | 
						|
	int position = evaltime - sbi->cfra;
 | 
						|
	if((position < 0) || (position >= sbi->length))
 | 
						|
		return 0.0f;
 | 
						|
 | 
						|
	return sbi->samples[position];
 | 
						|
}
 | 
						|
 | 
						|
/* ------------------- */
 | 
						|
 | 
						|
static int graphkeys_sound_bake_exec(bContext *C, wmOperator *op)
 | 
						|
{
 | 
						|
	bAnimContext ac;
 | 
						|
	ListBase anim_data = {NULL, NULL};
 | 
						|
	bAnimListElem *ale;
 | 
						|
	int filter;
 | 
						|
 | 
						|
	tSoundBakeInfo sbi;
 | 
						|
	Scene *scene= NULL;
 | 
						|
	int start, end;
 | 
						|
 | 
						|
	char path[FILE_MAX];
 | 
						|
 | 
						|
	/* get editor data */
 | 
						|
	if (ANIM_animdata_get_context(C, &ac) == 0)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
 | 
						|
	RNA_string_get(op->ptr, "filepath", path);
 | 
						|
 | 
						|
	scene= ac.scene;	/* current scene */
 | 
						|
 | 
						|
	/* store necessary data for the baking steps */
 | 
						|
	sbi.samples = AUD_readSoundBuffer(path,
 | 
						|
									  RNA_float_get(op->ptr, "low"),
 | 
						|
									  RNA_float_get(op->ptr, "high"),
 | 
						|
									  RNA_float_get(op->ptr, "attack"),
 | 
						|
									  RNA_float_get(op->ptr, "release"),
 | 
						|
									  RNA_float_get(op->ptr, "threshold"),
 | 
						|
									  RNA_boolean_get(op->ptr, "accumulate"),
 | 
						|
									  RNA_boolean_get(op->ptr, "use_additive"),
 | 
						|
									  RNA_boolean_get(op->ptr, "square"),
 | 
						|
									  RNA_float_get(op->ptr, "sthreshold"),
 | 
						|
									  FPS, &sbi.length);
 | 
						|
 | 
						|
	if (sbi.samples == NULL) {
 | 
						|
		BKE_report(op->reports, RPT_ERROR, "Unsupported audio format");
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
	}
 | 
						|
 | 
						|
	/* determine extents of the baking */
 | 
						|
	sbi.cfra = start = CFRA;
 | 
						|
	end = CFRA + sbi.length - 1;
 | 
						|
 | 
						|
	/* filter anim channels */
 | 
						|
	filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_SEL | ANIMFILTER_FOREDIT | ANIMFILTER_CURVESONLY | ANIMFILTER_NODUPLIS);
 | 
						|
	ANIM_animdata_filter(&ac, &anim_data, filter, ac.data, ac.datatype);
 | 
						|
 | 
						|
	/* loop through all selected F-Curves, replacing its data with the sound samples */
 | 
						|
	for (ale= anim_data.first; ale; ale= ale->next) {
 | 
						|
		FCurve *fcu= (FCurve *)ale->key_data;
 | 
						|
		
 | 
						|
		/* sample the sound */
 | 
						|
		fcurve_store_samples(fcu, &sbi, start, end, fcurve_samplingcb_sound);
 | 
						|
	}
 | 
						|
 | 
						|
	/* free sample data */
 | 
						|
	free(sbi.samples);
 | 
						|
 | 
						|
	/* admin and redraws */
 | 
						|
	BLI_freelistN(&anim_data);
 | 
						|
 | 
						|
	/* validate keyframes after editing */
 | 
						|
	ANIM_editkeyframes_refresh(&ac);
 | 
						|
 | 
						|
	/* set notifier that 'keyframes' have changed */
 | 
						|
	WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_EDITED, NULL);
 | 
						|
 | 
						|
	return OPERATOR_FINISHED;
 | 
						|
}
 | 
						|
 | 
						|
static int graphkeys_sound_bake_invoke (bContext *C, wmOperator *op, wmEvent *event)
 | 
						|
{
 | 
						|
	bAnimContext ac;
 | 
						|
 | 
						|
	/* verify editor data */
 | 
						|
	if (ANIM_animdata_get_context(C, &ac) == 0)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
 | 
						|
	return WM_operator_filesel(C, op, event);
 | 
						|
}
 | 
						|
 | 
						|
void GRAPH_OT_sound_bake (wmOperatorType *ot)
 | 
						|
{
 | 
						|
	/* identifiers */
 | 
						|
	ot->name= "Bake Sound to F-Curves";
 | 
						|
	ot->idname= "GRAPH_OT_sound_bake";
 | 
						|
	ot->description= "Bakes a sound wave to selected F-Curves";
 | 
						|
 | 
						|
	/* api callbacks */
 | 
						|
	ot->invoke= graphkeys_sound_bake_invoke;
 | 
						|
	ot->exec= graphkeys_sound_bake_exec;
 | 
						|
	ot->poll= graphop_selected_fcurve_poll;
 | 
						|
 | 
						|
	/* flags */
 | 
						|
	ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 | 
						|
 | 
						|
	/* properties */
 | 
						|
	WM_operator_properties_filesel(ot, FOLDERFILE|SOUNDFILE|MOVIEFILE, FILE_SPECIAL, FILE_OPENFILE, WM_FILESEL_FILEPATH);
 | 
						|
	RNA_def_float(ot->srna, "low", 0.0f, 0.0, 100000.0, "Lowest frequency", "", 0.1, 1000.00);
 | 
						|
	RNA_def_float(ot->srna, "high", 100000.0, 0.0, 100000.0, "Highest frequency", "", 0.1, 1000.00);
 | 
						|
	RNA_def_float(ot->srna, "attack", 0.005, 0.0, 2.0, "Attack time", "", 0.01, 0.1);
 | 
						|
	RNA_def_float(ot->srna, "release", 0.2, 0.0, 5.0, "Release time", "", 0.01, 0.2);
 | 
						|
	RNA_def_float(ot->srna, "threshold", 0.0, 0.0, 1.0, "Threshold", "", 0.01, 0.1);
 | 
						|
	RNA_def_boolean(ot->srna, "accumulate", 0, "Accumulate", "");
 | 
						|
	RNA_def_boolean(ot->srna, "use_additive", 0, "Additive", "");
 | 
						|
	RNA_def_boolean(ot->srna, "square", 0, "Square", "");
 | 
						|
	RNA_def_float(ot->srna, "sthreshold", 0.1, 0.0, 1.0, "Square Threshold", "", 0.01, 0.1);
 | 
						|
}
 | 
						|
 | 
						|
/* ******************** Sample Keyframes Operator *********************** */
 | 
						|
/* This operator 'bakes' the values of the curve into new keyframes between pairs
 | 
						|
 * of selected keyframes. It is useful for creating keyframes for tweaking overlap.
 | 
						|
 */
 | 
						|
 | 
						|
/* Evaluates the curves between each selected keyframe on each frame, and keys the value  */
 | 
						|
static void sample_graph_keys (bAnimContext *ac)
 | 
						|
{	
 | 
						|
	ListBase anim_data = {NULL, NULL};
 | 
						|
	bAnimListElem *ale;
 | 
						|
	int filter;
 | 
						|
	
 | 
						|
	/* filter data */
 | 
						|
	filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE| ANIMFILTER_FOREDIT | ANIMFILTER_CURVESONLY | ANIMFILTER_NODUPLIS);
 | 
						|
	ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
 | 
						|
	
 | 
						|
	/* loop through filtered data and add keys between selected keyframes on every frame  */
 | 
						|
	for (ale= anim_data.first; ale; ale= ale->next)
 | 
						|
		sample_fcurve((FCurve *)ale->key_data);
 | 
						|
	
 | 
						|
	/* admin and redraws */
 | 
						|
	BLI_freelistN(&anim_data);
 | 
						|
}
 | 
						|
 | 
						|
/* ------------------- */
 | 
						|
 | 
						|
static int graphkeys_sample_exec(bContext *C, wmOperator *UNUSED(op))
 | 
						|
{
 | 
						|
	bAnimContext ac;
 | 
						|
	
 | 
						|
	/* get editor data */
 | 
						|
	if (ANIM_animdata_get_context(C, &ac) == 0)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
	
 | 
						|
	/* sample keyframes */
 | 
						|
	sample_graph_keys(&ac);
 | 
						|
	
 | 
						|
	/* validate keyframes after editing */
 | 
						|
	ANIM_editkeyframes_refresh(&ac);
 | 
						|
	
 | 
						|
	/* set notifier that keyframes have changed */
 | 
						|
	WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_EDITED, NULL);
 | 
						|
	
 | 
						|
	return OPERATOR_FINISHED;
 | 
						|
}
 | 
						|
 
 | 
						|
void GRAPH_OT_sample (wmOperatorType *ot)
 | 
						|
{
 | 
						|
	/* identifiers */
 | 
						|
	ot->name= "Sample Keyframes";
 | 
						|
	ot->idname= "GRAPH_OT_sample";
 | 
						|
	ot->description= "Add keyframes on every frame between the selected keyframes";
 | 
						|
	
 | 
						|
	/* api callbacks */
 | 
						|
	ot->exec= graphkeys_sample_exec;
 | 
						|
	ot->poll= graphop_editable_keyframes_poll;
 | 
						|
	
 | 
						|
	/* flags */
 | 
						|
	ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/* ************************************************************************** */
 | 
						|
/* SETTINGS STUFF */
 | 
						|
 | 
						|
/* ******************** Set Extrapolation-Type Operator *********************** */
 | 
						|
 | 
						|
/* defines for set extrapolation-type for selected keyframes tool */
 | 
						|
EnumPropertyItem prop_graphkeys_expo_types[] = {
 | 
						|
	{FCURVE_EXTRAPOLATE_CONSTANT, "CONSTANT", 0, "Constant Extrapolation", ""},
 | 
						|
	{FCURVE_EXTRAPOLATE_LINEAR, "LINEAR", 0, "Linear Extrapolation", ""},
 | 
						|
	{0, NULL, 0, NULL, NULL}
 | 
						|
};
 | 
						|
 | 
						|
/* this function is responsible for setting extrapolation mode for keyframes */
 | 
						|
static void setexpo_graph_keys(bAnimContext *ac, short mode) 
 | 
						|
{
 | 
						|
	ListBase anim_data = {NULL, NULL};
 | 
						|
	bAnimListElem *ale;
 | 
						|
	int filter;
 | 
						|
	
 | 
						|
	/* filter data */
 | 
						|
	filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_SEL | ANIMFILTER_FOREDIT | ANIMFILTER_CURVESONLY | ANIMFILTER_NODUPLIS);
 | 
						|
	ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
 | 
						|
	
 | 
						|
	/* loop through setting mode per F-Curve */
 | 
						|
	for (ale= anim_data.first; ale; ale= ale->next) {
 | 
						|
		FCurve *fcu= (FCurve *)ale->data;
 | 
						|
		fcu->extend= mode;
 | 
						|
	}
 | 
						|
	
 | 
						|
	/* cleanup */
 | 
						|
	BLI_freelistN(&anim_data);
 | 
						|
}
 | 
						|
 | 
						|
/* ------------------- */
 | 
						|
 | 
						|
static int graphkeys_expo_exec(bContext *C, wmOperator *op)
 | 
						|
{
 | 
						|
	bAnimContext ac;
 | 
						|
	short mode;
 | 
						|
	
 | 
						|
	/* get editor data */
 | 
						|
	if (ANIM_animdata_get_context(C, &ac) == 0)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
		
 | 
						|
	/* get handle setting mode */
 | 
						|
	mode= RNA_enum_get(op->ptr, "type");
 | 
						|
	
 | 
						|
	/* set handle type */
 | 
						|
	setexpo_graph_keys(&ac, mode);
 | 
						|
	
 | 
						|
	/* validate keyframes after editing */
 | 
						|
	ANIM_editkeyframes_refresh(&ac);
 | 
						|
	
 | 
						|
	/* set notifier that keyframe properties have changed */
 | 
						|
	WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME_PROP, NULL);
 | 
						|
	
 | 
						|
	return OPERATOR_FINISHED;
 | 
						|
}
 | 
						|
 
 | 
						|
void GRAPH_OT_extrapolation_type (wmOperatorType *ot)
 | 
						|
{
 | 
						|
	/* identifiers */
 | 
						|
	ot->name= "Set Keyframe Extrapolation";
 | 
						|
	ot->idname= "GRAPH_OT_extrapolation_type";
 | 
						|
	ot->description= "Set extrapolation mode for selected F-Curves";
 | 
						|
	
 | 
						|
	/* api callbacks */
 | 
						|
	ot->invoke= WM_menu_invoke;
 | 
						|
	ot->exec= graphkeys_expo_exec;
 | 
						|
	ot->poll= graphop_editable_keyframes_poll;
 | 
						|
	
 | 
						|
	/* flags */
 | 
						|
	ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 | 
						|
	
 | 
						|
	/* id-props */
 | 
						|
	ot->prop= RNA_def_enum(ot->srna, "type", prop_graphkeys_expo_types, 0, "Type", "");
 | 
						|
}
 | 
						|
 | 
						|
/* ******************** Set Interpolation-Type Operator *********************** */
 | 
						|
 | 
						|
/* this function is responsible for setting interpolation mode for keyframes */
 | 
						|
static void setipo_graph_keys(bAnimContext *ac, short mode) 
 | 
						|
{
 | 
						|
	ListBase anim_data = {NULL, NULL};
 | 
						|
	bAnimListElem *ale;
 | 
						|
	int filter;
 | 
						|
	KeyframeEditFunc set_cb= ANIM_editkeyframes_ipo(mode);
 | 
						|
	
 | 
						|
	/* filter data */
 | 
						|
	filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_FOREDIT | ANIMFILTER_CURVESONLY | ANIMFILTER_NODUPLIS);
 | 
						|
	ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
 | 
						|
	
 | 
						|
	/* loop through setting BezTriple interpolation
 | 
						|
	 * Note: we do not supply KeyframeEditData to the looper yet. Currently that's not necessary here...
 | 
						|
	 */
 | 
						|
	for (ale= anim_data.first; ale; ale= ale->next)
 | 
						|
		ANIM_fcurve_keyframes_loop(NULL, ale->key_data, NULL, set_cb, calchandles_fcurve);
 | 
						|
	
 | 
						|
	/* cleanup */
 | 
						|
	BLI_freelistN(&anim_data);
 | 
						|
}
 | 
						|
 | 
						|
/* ------------------- */
 | 
						|
 | 
						|
static int graphkeys_ipo_exec(bContext *C, wmOperator *op)
 | 
						|
{
 | 
						|
	bAnimContext ac;
 | 
						|
	short mode;
 | 
						|
	
 | 
						|
	/* get editor data */
 | 
						|
	if (ANIM_animdata_get_context(C, &ac) == 0)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
		
 | 
						|
	/* get handle setting mode */
 | 
						|
	mode= RNA_enum_get(op->ptr, "type");
 | 
						|
	
 | 
						|
	/* set handle type */
 | 
						|
	setipo_graph_keys(&ac, mode);
 | 
						|
	
 | 
						|
	/* validate keyframes after editing */
 | 
						|
	ANIM_editkeyframes_refresh(&ac);
 | 
						|
	
 | 
						|
	/* set notifier that keyframe properties have changed */
 | 
						|
	WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME_PROP, NULL);
 | 
						|
	
 | 
						|
	return OPERATOR_FINISHED;
 | 
						|
}
 | 
						|
 
 | 
						|
void GRAPH_OT_interpolation_type (wmOperatorType *ot)
 | 
						|
{
 | 
						|
	/* identifiers */
 | 
						|
	ot->name= "Set Keyframe Interpolation";
 | 
						|
	ot->idname= "GRAPH_OT_interpolation_type";
 | 
						|
	ot->description= "Set interpolation mode for the F-Curve segments starting from the selected keyframes";
 | 
						|
	
 | 
						|
	/* api callbacks */
 | 
						|
	ot->invoke= WM_menu_invoke;
 | 
						|
	ot->exec= graphkeys_ipo_exec;
 | 
						|
	ot->poll= graphop_editable_keyframes_poll;
 | 
						|
	
 | 
						|
	/* flags */
 | 
						|
	ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 | 
						|
	
 | 
						|
	/* id-props */
 | 
						|
	ot->prop= RNA_def_enum(ot->srna, "type", beztriple_interpolation_mode_items, 0, "Type", "");
 | 
						|
}
 | 
						|
 | 
						|
/* ******************** Set Handle-Type Operator *********************** */
 | 
						|
 | 
						|
EnumPropertyItem graphkeys_handle_type_items[] = {
 | 
						|
	{HD_FREE, "FREE", 0, "Free", ""},
 | 
						|
	{HD_VECT, "VECTOR", 0, "Vector", ""},
 | 
						|
	{HD_ALIGN, "ALIGNED", 0, "Aligned", ""},
 | 
						|
	{0, "", 0, "", ""},
 | 
						|
	{HD_AUTO, "AUTO", 0, "Auto", "Handles that are automatically adjusted upon moving the keyframe. Whole curve"},
 | 
						|
	{HD_AUTO_ANIM, "ANIM_CLAMPED", 0, "Auto Clamped", "Auto handles clamped to not overshoot. Whole curve"},
 | 
						|
	{0, NULL, 0, NULL, NULL}};
 | 
						|
 | 
						|
/* ------------------- */
 | 
						|
 | 
						|
/* this function is responsible for setting handle-type of selected keyframes */
 | 
						|
static void sethandles_graph_keys(bAnimContext *ac, short mode) 
 | 
						|
{
 | 
						|
	ListBase anim_data = {NULL, NULL};
 | 
						|
	bAnimListElem *ale;
 | 
						|
	int filter;
 | 
						|
	
 | 
						|
	KeyframeEditFunc edit_cb= ANIM_editkeyframes_handles(mode);
 | 
						|
	KeyframeEditFunc sel_cb= ANIM_editkeyframes_ok(BEZT_OK_SELECTED);
 | 
						|
	
 | 
						|
	/* filter data */
 | 
						|
	filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_FOREDIT | ANIMFILTER_CURVESONLY | ANIMFILTER_NODUPLIS);
 | 
						|
	ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
 | 
						|
	
 | 
						|
	/* loop through setting flags for handles 
 | 
						|
	 * Note: we do not supply KeyframeEditData to the looper yet. Currently that's not necessary here...
 | 
						|
	 */
 | 
						|
	for (ale= anim_data.first; ale; ale= ale->next) {
 | 
						|
		FCurve *fcu= (FCurve *)ale->key_data;
 | 
						|
		
 | 
						|
		/* any selected keyframes for editing? */
 | 
						|
		if (ANIM_fcurve_keyframes_loop(NULL, fcu, NULL, sel_cb, NULL)) {
 | 
						|
			/* for auto/auto-clamped, toggle the auto-handles flag on the F-Curve */
 | 
						|
			if (mode == HD_AUTO_ANIM)
 | 
						|
				fcu->flag |= FCURVE_AUTO_HANDLES;
 | 
						|
			else if (mode == HD_AUTO)
 | 
						|
				fcu->flag &= ~FCURVE_AUTO_HANDLES;
 | 
						|
			
 | 
						|
			/* change type of selected handles */
 | 
						|
			ANIM_fcurve_keyframes_loop(NULL, fcu, NULL, edit_cb, calchandles_fcurve);
 | 
						|
		}
 | 
						|
	}
 | 
						|
	
 | 
						|
	/* cleanup */
 | 
						|
	BLI_freelistN(&anim_data);
 | 
						|
}
 | 
						|
/* ------------------- */
 | 
						|
 | 
						|
static int graphkeys_handletype_exec(bContext *C, wmOperator *op)
 | 
						|
{
 | 
						|
	bAnimContext ac;
 | 
						|
	short mode;
 | 
						|
	
 | 
						|
	/* get editor data */
 | 
						|
	if (ANIM_animdata_get_context(C, &ac) == 0)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
		
 | 
						|
	/* get handle setting mode */
 | 
						|
	mode= RNA_enum_get(op->ptr, "type");
 | 
						|
	
 | 
						|
	/* set handle type */
 | 
						|
	sethandles_graph_keys(&ac, mode);
 | 
						|
	
 | 
						|
	/* validate keyframes after editing */
 | 
						|
	ANIM_editkeyframes_refresh(&ac);
 | 
						|
	
 | 
						|
	/* set notifier that keyframe properties have changed */
 | 
						|
	WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME_PROP, NULL);
 | 
						|
	
 | 
						|
	return OPERATOR_FINISHED;
 | 
						|
}
 | 
						|
 
 | 
						|
 void GRAPH_OT_handle_type (wmOperatorType *ot)
 | 
						|
{
 | 
						|
	/* identifiers */
 | 
						|
	ot->name= "Set Keyframe Handle Type";
 | 
						|
	ot->idname= "GRAPH_OT_handle_type";
 | 
						|
	ot->description= "Set type of handle for selected keyframes";
 | 
						|
	
 | 
						|
	/* api callbacks */
 | 
						|
	ot->invoke= WM_menu_invoke;
 | 
						|
	ot->exec= graphkeys_handletype_exec;
 | 
						|
	ot->poll= graphop_editable_keyframes_poll;
 | 
						|
	
 | 
						|
	/* flags */
 | 
						|
	ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 | 
						|
	
 | 
						|
	/* id-props */
 | 
						|
	ot->prop= RNA_def_enum(ot->srna, "type", graphkeys_handle_type_items, 0, "Type", "");
 | 
						|
}
 | 
						|
 | 
						|
/* ************************************************************************** */
 | 
						|
/* TRANSFORM STUFF */
 | 
						|
 | 
						|
/* ***************** 'Euler Filter' Operator **************************** */
 | 
						|
/* Euler filter tools (as seen in Maya), are necessary for working with 'baked'
 | 
						|
 * rotation curves (with Euler rotations). The main purpose of such tools is to
 | 
						|
 * resolve any discontinuities that may arise in the curves due to the clamping
 | 
						|
 * of values to -180 degrees to 180 degrees.
 | 
						|
 */
 | 
						|
 | 
						|
#if 0 // XXX this is not ready for the primetime yet
 | 
						|
 
 | 
						|
/* set of three euler-rotation F-Curves */
 | 
						|
typedef struct tEulerFilter {
 | 
						|
	ID *id;							/* ID-block which owns the channels */
 | 
						|
	FCurve (*fcurves)[3];			/* 3 Pointers to F-Curves */				
 | 
						|
} tEulerFilter;
 | 
						|
 
 | 
						|
static int graphkeys_euler_filter_exec (bContext *C, wmOperator *op)
 | 
						|
{
 | 
						|
	bAnimContext ac;
 | 
						|
	
 | 
						|
	ListBase anim_data= {NULL, NULL};
 | 
						|
	bAnimListElem *ale;
 | 
						|
	int filter;
 | 
						|
	
 | 
						|
	ListBase eulers = {NULL, NULL};
 | 
						|
	tEulerFilter *euf= NULL;	
 | 
						|
	
 | 
						|
	/* get editor data */
 | 
						|
	if (ANIM_animdata_get_context(C, &ac) == 0)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
		
 | 
						|
	/* The process is done in two passes:
 | 
						|
	 * 	 1) Sets of three related rotation curves are identified from the selected channels,
 | 
						|
	 *		and are stored as a single 'operation unit' for the next step
 | 
						|
	 *	 2) Each set of three F-Curves is processed for each keyframe, with the values being
 | 
						|
	 * 		processed according to one of several ways.
 | 
						|
	 */
 | 
						|
	 
 | 
						|
	/* step 1: extract only the rotation f-curves */
 | 
						|
	filter= (ANIMFILTER_VISIBLE | ANIMFILTER_SEL | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_CURVESONLY | ANIMFILTER_FOREDIT | ANIMFILTER_NODUPLIS);
 | 
						|
	ANIM_animdata_filter(&ac, &anim_data, filter, ac.data, ac.datatype);
 | 
						|
	
 | 
						|
	for (ale= anim_data.first; ale; ale= ale->next) {
 | 
						|
		FCurve *fcu = (FCurve *)ale->data;
 | 
						|
		
 | 
						|
		/* check if this is an appropriate F-Curve 
 | 
						|
		 *	- only rotation curves
 | 
						|
		 *	- for pchan curves, make sure we're only using the euler curves
 | 
						|
		 */
 | 
						|
		if (strstr(fcu->rna_path, "rotation_euler") == 0)
 | 
						|
			continue;
 | 
						|
		
 | 
						|
		/* check if current set of 3-curves is suitable to add this curve to 
 | 
						|
		 *	- things like whether the current set of curves is 'full' should be checked later only
 | 
						|
		 *	- first check if id-blocks are compatible
 | 
						|
		 */
 | 
						|
		if ((euf) && (ale->id != euf->id)) {
 | 
						|
			/* if the paths match, add this curve to the set of curves */
 | 
						|
			// NOTE: simple string compare for now... could be a bit more fancy...
 | 
						|
			
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			/* just add to a new block */
 | 
						|
			euf= MEM_callocN(sizeof(tEulerFilter), "tEulerFilter");
 | 
						|
			BLI_addtail(&eulers, euf);
 | 
						|
			
 | 
						|
			euf->id= ale->id;
 | 
						|
			euf->fcurves[fcu->array_index]= fcu;
 | 
						|
		}
 | 
						|
	}
 | 
						|
	BLI_freelistN(&anim_data);
 | 
						|
	
 | 
						|
	/* step 2: go through each set of curves, processing the values at each keyframe 
 | 
						|
	 *	- it is assumed that there must be a full set of keyframes at each keyframe position
 | 
						|
	 */
 | 
						|
	for (euf= eulers.first; euf; euf= euf->next) {
 | 
						|
		
 | 
						|
	}
 | 
						|
	BLI_freelistN(&eulers);
 | 
						|
	
 | 
						|
	return OPERATOR_FINISHED;
 | 
						|
}
 | 
						|
 
 | 
						|
void GRAPH_OT_euler_filter (wmOperatorType *ot)
 | 
						|
{
 | 
						|
	/* identifiers */
 | 
						|
	ot->name= "Euler Filter";
 | 
						|
	ot->idname= "GRAPH_OT_euler_filter";
 | 
						|
	
 | 
						|
	/* api callbacks */
 | 
						|
	ot->exec= graphkeys_euler_filter_exec;
 | 
						|
	ot->poll= graphop_editable_keyframes_poll;
 | 
						|
	
 | 
						|
	/* flags */
 | 
						|
	ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 | 
						|
}
 | 
						|
 | 
						|
#endif // XXX this is not ready for the primetime yet
 | 
						|
 | 
						|
/* ***************** Jump to Selected Frames Operator *********************** */
 | 
						|
 | 
						|
/* snap current-frame indicator to 'average time' of selected keyframe */
 | 
						|
static int graphkeys_framejump_exec(bContext *C, wmOperator *UNUSED(op))
 | 
						|
{
 | 
						|
	bAnimContext ac;
 | 
						|
	ListBase anim_data= {NULL, NULL};
 | 
						|
	bAnimListElem *ale;
 | 
						|
	int filter;
 | 
						|
	KeyframeEditData ked;
 | 
						|
	
 | 
						|
	/* get editor data */
 | 
						|
	if (ANIM_animdata_get_context(C, &ac) == 0)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
	
 | 
						|
	/* init edit data */
 | 
						|
	memset(&ked, 0, sizeof(KeyframeEditData));
 | 
						|
	
 | 
						|
	/* loop over action data, averaging values */
 | 
						|
	filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_CURVESONLY | ANIMFILTER_NODUPLIS);
 | 
						|
	ANIM_animdata_filter(&ac, &anim_data, filter, ac.data, ac.datatype);
 | 
						|
	
 | 
						|
	for (ale= anim_data.first; ale; ale= ale->next) {
 | 
						|
		AnimData *adt= ANIM_nla_mapping_get(&ac, ale);
 | 
						|
		
 | 
						|
		/* apply unit corrections */
 | 
						|
		ANIM_unit_mapping_apply_fcurve(ac.scene, ale->id, ale->key_data, ANIM_UNITCONV_ONLYKEYS);
 | 
						|
		
 | 
						|
		if (adt) {
 | 
						|
			ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 0, 1); 
 | 
						|
			ANIM_fcurve_keyframes_loop(&ked, ale->key_data, NULL, bezt_calc_average, NULL);
 | 
						|
			ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 1, 1); 
 | 
						|
		}
 | 
						|
		else
 | 
						|
			ANIM_fcurve_keyframes_loop(&ked, ale->key_data, NULL, bezt_calc_average, NULL);
 | 
						|
		
 | 
						|
		/* unapply unit corrections */
 | 
						|
		ANIM_unit_mapping_apply_fcurve(ac.scene, ale->id, ale->key_data, ANIM_UNITCONV_RESTORE|ANIM_UNITCONV_ONLYKEYS);
 | 
						|
	}
 | 
						|
	
 | 
						|
	BLI_freelistN(&anim_data);
 | 
						|
	
 | 
						|
	/* set the new current frame and cursor values, based on the average time and value */
 | 
						|
	if (ked.i1) {
 | 
						|
		SpaceIpo *sipo= ac.sa->spacedata.first;
 | 
						|
		Scene *scene= ac.scene;
 | 
						|
		
 | 
						|
		/* take the average values, rounding to the nearest int for the current frame */
 | 
						|
		CFRA= (int)floor((ked.f1 / ked.i1) + 0.5f);
 | 
						|
		SUBFRA= 0.f;
 | 
						|
		sipo->cursorVal= ked.f2 / (float)ked.i1;
 | 
						|
	}
 | 
						|
	
 | 
						|
	/* set notifier that things have changed */
 | 
						|
	WM_event_add_notifier(C, NC_SCENE|ND_FRAME, ac.scene);
 | 
						|
	
 | 
						|
	return OPERATOR_FINISHED;
 | 
						|
}
 | 
						|
 | 
						|
void GRAPH_OT_frame_jump (wmOperatorType *ot)
 | 
						|
{
 | 
						|
	/* identifiers */
 | 
						|
	ot->name= "Jump to Frame";
 | 
						|
	ot->idname= "GRAPH_OT_frame_jump";
 | 
						|
	ot->description= "Set the current frame to the average frame of the selected keyframes";
 | 
						|
	
 | 
						|
	/* api callbacks */
 | 
						|
	ot->exec= graphkeys_framejump_exec;
 | 
						|
	ot->poll= graphop_visible_keyframes_poll;
 | 
						|
	
 | 
						|
	/* flags */
 | 
						|
	ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 | 
						|
}
 | 
						|
 | 
						|
/* ******************** Snap Keyframes Operator *********************** */
 | 
						|
 | 
						|
/* defines for snap keyframes tool */
 | 
						|
EnumPropertyItem prop_graphkeys_snap_types[] = {
 | 
						|
	{GRAPHKEYS_SNAP_CFRA, "CFRA", 0, "Current Frame", ""},
 | 
						|
	{GRAPHKEYS_SNAP_VALUE, "VALUE", 0, "Cursor Value", ""},
 | 
						|
	{GRAPHKEYS_SNAP_NEAREST_FRAME, "NEAREST_FRAME", 0, "Nearest Frame", ""}, // XXX as single entry?
 | 
						|
	{GRAPHKEYS_SNAP_NEAREST_SECOND, "NEAREST_SECOND", 0, "Nearest Second", ""}, // XXX as single entry?
 | 
						|
	{GRAPHKEYS_SNAP_NEAREST_MARKER, "NEAREST_MARKER", 0, "Nearest Marker", ""},
 | 
						|
	{GRAPHKEYS_SNAP_HORIZONTAL, "HORIZONTAL", 0, "Flatten Handles", ""},
 | 
						|
	{0, NULL, 0, NULL, NULL}
 | 
						|
};
 | 
						|
 | 
						|
/* this function is responsible for snapping keyframes to frame-times */
 | 
						|
static void snap_graph_keys(bAnimContext *ac, short mode) 
 | 
						|
{
 | 
						|
	ListBase anim_data = {NULL, NULL};
 | 
						|
	bAnimListElem *ale;
 | 
						|
	int filter;
 | 
						|
	
 | 
						|
	KeyframeEditData ked;
 | 
						|
	KeyframeEditFunc edit_cb;
 | 
						|
	
 | 
						|
	/* filter data */
 | 
						|
	filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_FOREDIT | ANIMFILTER_CURVESONLY | ANIMFILTER_NODUPLIS);
 | 
						|
	ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
 | 
						|
	
 | 
						|
	/* get beztriple editing callbacks */
 | 
						|
	edit_cb= ANIM_editkeyframes_snap(mode);
 | 
						|
	
 | 
						|
	memset(&ked, 0, sizeof(KeyframeEditData)); 
 | 
						|
	ked.scene= ac->scene;
 | 
						|
	if (mode == GRAPHKEYS_SNAP_NEAREST_MARKER) {
 | 
						|
		ked.list.first= (ac->markers) ? ac->markers->first : NULL;
 | 
						|
		ked.list.last= (ac->markers) ? ac->markers->last : NULL;
 | 
						|
	}
 | 
						|
	else if (mode == GRAPHKEYS_SNAP_VALUE) {
 | 
						|
		SpaceIpo *sipo= (SpaceIpo *)ac->sa->spacedata.first;
 | 
						|
		ked.f1= (sipo) ? sipo->cursorVal : 0.0f;
 | 
						|
	}
 | 
						|
	
 | 
						|
	/* snap keyframes */
 | 
						|
	for (ale= anim_data.first; ale; ale= ale->next) {
 | 
						|
		AnimData *adt= ANIM_nla_mapping_get(ac, ale);
 | 
						|
		
 | 
						|
		/* apply unit corrections */
 | 
						|
		ANIM_unit_mapping_apply_fcurve(ac->scene, ale->id, ale->key_data, 0);
 | 
						|
		
 | 
						|
		if (adt) {
 | 
						|
			ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 0, 1); 
 | 
						|
			ANIM_fcurve_keyframes_loop(&ked, ale->key_data, NULL, edit_cb, calchandles_fcurve);
 | 
						|
			ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 1, 1);
 | 
						|
		}
 | 
						|
		else 
 | 
						|
			ANIM_fcurve_keyframes_loop(&ked, ale->key_data, NULL, edit_cb, calchandles_fcurve);
 | 
						|
			
 | 
						|
		/* apply unit corrections */
 | 
						|
		ANIM_unit_mapping_apply_fcurve(ac->scene, ale->id, ale->key_data, ANIM_UNITCONV_RESTORE);
 | 
						|
	}
 | 
						|
	
 | 
						|
	BLI_freelistN(&anim_data);
 | 
						|
}
 | 
						|
 | 
						|
/* ------------------- */
 | 
						|
 | 
						|
static int graphkeys_snap_exec(bContext *C, wmOperator *op)
 | 
						|
{
 | 
						|
	bAnimContext ac;
 | 
						|
	short mode;
 | 
						|
	
 | 
						|
	/* get editor data */
 | 
						|
	if (ANIM_animdata_get_context(C, &ac) == 0)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
		
 | 
						|
	/* get snapping mode */
 | 
						|
	mode= RNA_enum_get(op->ptr, "type");
 | 
						|
	
 | 
						|
	/* snap keyframes */
 | 
						|
	snap_graph_keys(&ac, mode);
 | 
						|
	
 | 
						|
	/* validate keyframes after editing */
 | 
						|
	ANIM_editkeyframes_refresh(&ac);
 | 
						|
	
 | 
						|
	/* set notifier that keyframes have changed */
 | 
						|
	WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_EDITED, NULL);
 | 
						|
	
 | 
						|
	return OPERATOR_FINISHED;
 | 
						|
}
 | 
						|
 
 | 
						|
void GRAPH_OT_snap (wmOperatorType *ot)
 | 
						|
{
 | 
						|
	/* identifiers */
 | 
						|
	ot->name= "Snap Keys";
 | 
						|
	ot->idname= "GRAPH_OT_snap";
 | 
						|
	ot->description= "Snap selected keyframes to the chosen times/values";
 | 
						|
	
 | 
						|
	/* api callbacks */
 | 
						|
	ot->invoke= WM_menu_invoke;
 | 
						|
	ot->exec= graphkeys_snap_exec;
 | 
						|
	ot->poll= graphop_editable_keyframes_poll;
 | 
						|
	
 | 
						|
	/* flags */
 | 
						|
	ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 | 
						|
	
 | 
						|
	/* id-props */
 | 
						|
	ot->prop= RNA_def_enum(ot->srna, "type", prop_graphkeys_snap_types, 0, "Type", "");
 | 
						|
}
 | 
						|
 | 
						|
/* ******************** Mirror Keyframes Operator *********************** */
 | 
						|
 | 
						|
/* defines for mirror keyframes tool */
 | 
						|
EnumPropertyItem prop_graphkeys_mirror_types[] = {
 | 
						|
	{GRAPHKEYS_MIRROR_CFRA, "CFRA", 0, "By Times over Current Frame", ""},
 | 
						|
	{GRAPHKEYS_MIRROR_VALUE, "VALUE", 0, "By Values over Cursor Value", ""},
 | 
						|
	{GRAPHKEYS_MIRROR_YAXIS, "YAXIS", 0, "By Times over Time=0", ""},
 | 
						|
	{GRAPHKEYS_MIRROR_XAXIS, "XAXIS", 0, "By Values over Value=0", ""},
 | 
						|
	{GRAPHKEYS_MIRROR_MARKER, "MARKER", 0, "By Times over First Selected Marker", ""},
 | 
						|
	{0, NULL, 0, NULL, NULL}
 | 
						|
};
 | 
						|
 | 
						|
/* this function is responsible for mirroring keyframes */
 | 
						|
static void mirror_graph_keys(bAnimContext *ac, short mode) 
 | 
						|
{
 | 
						|
	ListBase anim_data = {NULL, NULL};
 | 
						|
	bAnimListElem *ale;
 | 
						|
	int filter;
 | 
						|
	
 | 
						|
	KeyframeEditData ked;
 | 
						|
	KeyframeEditFunc edit_cb;
 | 
						|
	
 | 
						|
	/* get beztriple editing callbacks */
 | 
						|
	edit_cb= ANIM_editkeyframes_mirror(mode);
 | 
						|
	
 | 
						|
	memset(&ked, 0, sizeof(KeyframeEditData)); 
 | 
						|
	ked.scene= ac->scene;
 | 
						|
	
 | 
						|
	/* for 'first selected marker' mode, need to find first selected marker first! */
 | 
						|
	// XXX should this be made into a helper func in the API?
 | 
						|
	if (mode == GRAPHKEYS_MIRROR_MARKER) {
 | 
						|
		TimeMarker *marker= NULL;
 | 
						|
		
 | 
						|
		/* find first selected marker */
 | 
						|
		if (ac->markers) {
 | 
						|
			for (marker= ac->markers->first; marker; marker=marker->next) {
 | 
						|
				if (marker->flag & SELECT) {
 | 
						|
					break;
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
		
 | 
						|
		/* store marker's time (if available) */
 | 
						|
		if (marker)
 | 
						|
			ked.f1= (float)marker->frame;
 | 
						|
		else
 | 
						|
			return;
 | 
						|
	}
 | 
						|
	else if (mode == GRAPHKEYS_MIRROR_VALUE) {
 | 
						|
		SpaceIpo *sipo= (SpaceIpo *)ac->sa->spacedata.first;
 | 
						|
		ked.f1= (sipo) ? sipo->cursorVal : 0.0f;
 | 
						|
	}
 | 
						|
	
 | 
						|
	/* filter data */
 | 
						|
	filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE| ANIMFILTER_FOREDIT | ANIMFILTER_CURVESONLY | ANIMFILTER_NODUPLIS);
 | 
						|
	ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
 | 
						|
	
 | 
						|
	/* mirror keyframes */
 | 
						|
	for (ale= anim_data.first; ale; ale= ale->next) {
 | 
						|
		AnimData *adt= ANIM_nla_mapping_get(ac, ale);
 | 
						|
		
 | 
						|
		/* apply unit corrections */
 | 
						|
		ANIM_unit_mapping_apply_fcurve(ac->scene, ale->id, ale->key_data, ANIM_UNITCONV_ONLYKEYS);
 | 
						|
		
 | 
						|
		if (adt) {
 | 
						|
			ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 0, 1); 
 | 
						|
			ANIM_fcurve_keyframes_loop(&ked, ale->key_data, NULL, edit_cb, calchandles_fcurve);
 | 
						|
			ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 1, 1);
 | 
						|
		}
 | 
						|
		else 
 | 
						|
			ANIM_fcurve_keyframes_loop(&ked, ale->key_data, NULL, edit_cb, calchandles_fcurve);
 | 
						|
			
 | 
						|
		/* unapply unit corrections */
 | 
						|
		ANIM_unit_mapping_apply_fcurve(ac->scene, ale->id, ale->key_data, ANIM_UNITCONV_ONLYKEYS|ANIM_UNITCONV_RESTORE);
 | 
						|
	}
 | 
						|
	
 | 
						|
	BLI_freelistN(&anim_data);
 | 
						|
}
 | 
						|
 | 
						|
/* ------------------- */
 | 
						|
 | 
						|
static int graphkeys_mirror_exec(bContext *C, wmOperator *op)
 | 
						|
{
 | 
						|
	bAnimContext ac;
 | 
						|
	short mode;
 | 
						|
	
 | 
						|
	/* get editor data */
 | 
						|
	if (ANIM_animdata_get_context(C, &ac) == 0)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
		
 | 
						|
	/* get mirroring mode */
 | 
						|
	mode= RNA_enum_get(op->ptr, "type");
 | 
						|
	
 | 
						|
	/* mirror keyframes */
 | 
						|
	mirror_graph_keys(&ac, mode);
 | 
						|
	
 | 
						|
	/* validate keyframes after editing */
 | 
						|
	ANIM_editkeyframes_refresh(&ac);
 | 
						|
	
 | 
						|
	/* set notifier that keyframes have changed */
 | 
						|
	WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_EDITED, NULL);
 | 
						|
	
 | 
						|
	return OPERATOR_FINISHED;
 | 
						|
}
 | 
						|
 
 | 
						|
void GRAPH_OT_mirror (wmOperatorType *ot)
 | 
						|
{
 | 
						|
	/* identifiers */
 | 
						|
	ot->name= "Mirror Keys";
 | 
						|
	ot->idname= "GRAPH_OT_mirror";
 | 
						|
	ot->description= "Flip selected keyframes over the selected mirror line";
 | 
						|
	
 | 
						|
	/* api callbacks */
 | 
						|
	ot->invoke= WM_menu_invoke;
 | 
						|
	ot->exec= graphkeys_mirror_exec;
 | 
						|
	ot->poll= graphop_editable_keyframes_poll;
 | 
						|
	
 | 
						|
	/* flags */
 | 
						|
	ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 | 
						|
	
 | 
						|
	/* id-props */
 | 
						|
	ot->prop= RNA_def_enum(ot->srna, "type", prop_graphkeys_mirror_types, 0, "Type", "");
 | 
						|
}
 | 
						|
 | 
						|
/* ******************** Smooth Keyframes Operator *********************** */
 | 
						|
 | 
						|
static int graphkeys_smooth_exec(bContext *C, wmOperator *UNUSED(op))
 | 
						|
{
 | 
						|
	bAnimContext ac;
 | 
						|
	ListBase anim_data = {NULL, NULL};
 | 
						|
	bAnimListElem *ale;
 | 
						|
	int filter;
 | 
						|
	
 | 
						|
	/* get editor data */
 | 
						|
	if (ANIM_animdata_get_context(C, &ac) == 0)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
	
 | 
						|
	/* filter data */
 | 
						|
	filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE| ANIMFILTER_FOREDIT | ANIMFILTER_CURVESONLY | ANIMFILTER_NODUPLIS);
 | 
						|
	ANIM_animdata_filter(&ac, &anim_data, filter, ac.data, ac.datatype);
 | 
						|
	
 | 
						|
	/* smooth keyframes */
 | 
						|
	for (ale= anim_data.first; ale; ale= ale->next) {
 | 
						|
		/* For now, we can only smooth by flattening handles AND smoothing curve values.
 | 
						|
		 * Perhaps the mode argument could be removed, as that functionality is offerred through 
 | 
						|
		 * Snap->Flatten Handles anyway.
 | 
						|
		 */
 | 
						|
		smooth_fcurve(ale->key_data);
 | 
						|
	}
 | 
						|
	BLI_freelistN(&anim_data);
 | 
						|
	
 | 
						|
	/* validate keyframes after editing */
 | 
						|
	ANIM_editkeyframes_refresh(&ac);
 | 
						|
	
 | 
						|
	/* set notifier that keyframes have changed */
 | 
						|
	WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_EDITED, NULL);
 | 
						|
	
 | 
						|
	return OPERATOR_FINISHED;
 | 
						|
}
 | 
						|
 
 | 
						|
void GRAPH_OT_smooth (wmOperatorType *ot)
 | 
						|
{
 | 
						|
	/* identifiers */
 | 
						|
	ot->name= "Smooth Keys";
 | 
						|
	ot->idname= "GRAPH_OT_smooth";
 | 
						|
	ot->description= "Apply weighted moving means to make selected F-Curves less bumpy";
 | 
						|
	
 | 
						|
	/* api callbacks */
 | 
						|
	ot->exec= graphkeys_smooth_exec;
 | 
						|
	ot->poll= graphop_editable_keyframes_poll;
 | 
						|
	
 | 
						|
	/* flags */
 | 
						|
	ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 | 
						|
}
 | 
						|
 | 
						|
/* ************************************************************************** */
 | 
						|
/* F-CURVE MODIFIERS */
 | 
						|
 | 
						|
/* ******************** Add F-Modifier Operator *********************** */
 | 
						|
 | 
						|
/* present a special customised popup menu for this, with some filtering */
 | 
						|
static int graph_fmodifier_add_invoke (bContext *C, wmOperator *op, wmEvent *UNUSED(event))
 | 
						|
{
 | 
						|
	uiPopupMenu *pup;
 | 
						|
	uiLayout *layout;
 | 
						|
	int i;
 | 
						|
	
 | 
						|
	pup= uiPupMenuBegin(C, "Add F-Curve Modifier", 0);
 | 
						|
	layout= uiPupMenuLayout(pup);
 | 
						|
	
 | 
						|
	/* start from 1 to skip the 'Invalid' modifier type */
 | 
						|
	for (i = 1; i < FMODIFIER_NUM_TYPES; i++) {
 | 
						|
		FModifierTypeInfo *fmi= get_fmodifier_typeinfo(i);
 | 
						|
		PointerRNA props_ptr;
 | 
						|
		
 | 
						|
		/* check if modifier is valid for this context */
 | 
						|
		if (fmi == NULL)
 | 
						|
			continue;
 | 
						|
		
 | 
						|
		/* create operator menu item with relevant properties filled in */
 | 
						|
		props_ptr= uiItemFullO(layout, "GRAPH_OT_fmodifier_add", fmi->name, 0, NULL, WM_OP_EXEC_REGION_WIN, UI_ITEM_O_RETURN_PROPS);
 | 
						|
			/* the only thing that gets set from the menu is the type of F-Modifier to add */
 | 
						|
		RNA_enum_set(&props_ptr, "type", i);
 | 
						|
			/* the following properties are just repeats of existing ones... */
 | 
						|
		RNA_boolean_set(&props_ptr, "only_active", RNA_boolean_get(op->ptr, "only_active"));
 | 
						|
	}
 | 
						|
	uiItemS(layout);
 | 
						|
	
 | 
						|
	uiPupMenuEnd(C, pup);
 | 
						|
	
 | 
						|
	return OPERATOR_CANCELLED;
 | 
						|
}
 | 
						|
 | 
						|
static int graph_fmodifier_add_exec(bContext *C, wmOperator *op)
 | 
						|
{
 | 
						|
	bAnimContext ac;
 | 
						|
	ListBase anim_data = {NULL, NULL};
 | 
						|
	bAnimListElem *ale;
 | 
						|
	int filter;
 | 
						|
	short type;
 | 
						|
	
 | 
						|
	/* get editor data */
 | 
						|
	if (ANIM_animdata_get_context(C, &ac) == 0)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
	
 | 
						|
	/* get type of modifier to add */
 | 
						|
	type= RNA_enum_get(op->ptr, "type");
 | 
						|
	
 | 
						|
	/* filter data */
 | 
						|
	filter= (ANIMFILTER_VISIBLE | ANIMFILTER_FOREDIT | ANIMFILTER_CURVESONLY | ANIMFILTER_NODUPLIS);
 | 
						|
	if (RNA_boolean_get(op->ptr, "only_active"))
 | 
						|
		filter |= ANIMFILTER_ACTIVE;
 | 
						|
	else
 | 
						|
		filter |= (ANIMFILTER_SEL|ANIMFILTER_CURVEVISIBLE);
 | 
						|
	ANIM_animdata_filter(&ac, &anim_data, filter, ac.data, ac.datatype);
 | 
						|
	
 | 
						|
	/* add f-modifier to each curve */
 | 
						|
	for (ale= anim_data.first; ale; ale= ale->next) {
 | 
						|
		FCurve *fcu= (FCurve *)ale->data;
 | 
						|
		FModifier *fcm;
 | 
						|
		
 | 
						|
		/* add F-Modifier of specified type to active F-Curve, and make it the active one */
 | 
						|
		fcm= add_fmodifier(&fcu->modifiers, type);
 | 
						|
		if (fcm)
 | 
						|
			set_active_fmodifier(&fcu->modifiers, fcm);
 | 
						|
		else { // TODO: stop when this happens?
 | 
						|
			BKE_report(op->reports, RPT_ERROR, "Modifier couldn't be added. See console for details.");
 | 
						|
			break;
 | 
						|
		}
 | 
						|
	}
 | 
						|
	BLI_freelistN(&anim_data);
 | 
						|
	
 | 
						|
	/* validate keyframes after editing */
 | 
						|
	ANIM_editkeyframes_refresh(&ac);
 | 
						|
	
 | 
						|
	/* set notifier that things have changed */
 | 
						|
	// FIXME: this really isn't the best description for it...
 | 
						|
	WM_event_add_notifier(C, NC_ANIMATION, NULL);
 | 
						|
	
 | 
						|
	return OPERATOR_FINISHED;
 | 
						|
}
 | 
						|
 
 | 
						|
void GRAPH_OT_fmodifier_add (wmOperatorType *ot)
 | 
						|
{
 | 
						|
	/* identifiers */
 | 
						|
	ot->name= "Add F-Curve Modifier";
 | 
						|
	ot->idname= "GRAPH_OT_fmodifier_add";
 | 
						|
	ot->description= "Add F-Modifiers to the selected F-Curves";
 | 
						|
	
 | 
						|
	/* api callbacks */
 | 
						|
	ot->invoke= graph_fmodifier_add_invoke;
 | 
						|
	ot->exec= graph_fmodifier_add_exec;
 | 
						|
	ot->poll= graphop_selected_fcurve_poll; 
 | 
						|
	
 | 
						|
	/* flags */
 | 
						|
	ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 | 
						|
	
 | 
						|
	/* id-props */
 | 
						|
	ot->prop= RNA_def_enum(ot->srna, "type", fmodifier_type_items, 0, "Type", "");
 | 
						|
	RNA_def_boolean(ot->srna, "only_active", 1, "Only Active", "Only add F-Modifier to active F-Curve.");
 | 
						|
}
 | 
						|
 | 
						|
/* ******************** Copy F-Modifiers Operator *********************** */
 | 
						|
 | 
						|
static int graph_fmodifier_copy_exec(bContext *C, wmOperator *op)
 | 
						|
{
 | 
						|
	bAnimContext ac;
 | 
						|
	bAnimListElem *ale;
 | 
						|
	short ok = 0;
 | 
						|
	
 | 
						|
	/* get editor data */
 | 
						|
	if (ANIM_animdata_get_context(C, &ac) == 0)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
	
 | 
						|
	/* clear buffer first */
 | 
						|
	free_fmodifiers_copybuf();
 | 
						|
	
 | 
						|
	/* get the active F-Curve */
 | 
						|
	ale= get_active_fcurve_channel(&ac);
 | 
						|
	
 | 
						|
	/* if this exists, call the copy F-Modifiers API function */
 | 
						|
	if (ale && ale->data) {
 | 
						|
		FCurve *fcu= (FCurve *)ale->data;
 | 
						|
		
 | 
						|
		// TODO: when 'active' vs 'all' boolean is added, change last param!
 | 
						|
		ok= ANIM_fmodifiers_copy_to_buf(&fcu->modifiers, 0);
 | 
						|
		
 | 
						|
		/* free temp data now */
 | 
						|
		MEM_freeN(ale);
 | 
						|
	}
 | 
						|
	
 | 
						|
	/* successful or not? */
 | 
						|
	if (ok == 0) {
 | 
						|
		BKE_report(op->reports, RPT_ERROR, "No F-Modifiers available to be copied");
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
	}
 | 
						|
	else
 | 
						|
		return OPERATOR_FINISHED;
 | 
						|
}
 | 
						|
 
 | 
						|
void GRAPH_OT_fmodifier_copy (wmOperatorType *ot)
 | 
						|
{
 | 
						|
	/* identifiers */
 | 
						|
	ot->name= "Copy F-Modifiers";
 | 
						|
	ot->idname= "GRAPH_OT_fmodifier_copy";
 | 
						|
	ot->description= "Copy the F-Modifier(s) of the active F-Curve.";
 | 
						|
	
 | 
						|
	/* api callbacks */
 | 
						|
	ot->exec= graph_fmodifier_copy_exec;
 | 
						|
	ot->poll= graphop_active_fcurve_poll; 
 | 
						|
	
 | 
						|
	/* flags */
 | 
						|
	ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 | 
						|
	
 | 
						|
	/* id-props */
 | 
						|
	//ot->prop = RNA_def_boolean(ot->srna, "all", 1, "All F-Modifiers", "Copy all the F-Modifiers, instead of just the active one");
 | 
						|
}
 | 
						|
 | 
						|
/* ******************** Paste F-Modifiers Operator *********************** */
 | 
						|
 | 
						|
static int graph_fmodifier_paste_exec(bContext *C, wmOperator *op)
 | 
						|
{
 | 
						|
	bAnimContext ac;
 | 
						|
	ListBase anim_data = {NULL, NULL};
 | 
						|
	bAnimListElem *ale;
 | 
						|
	int filter, ok=0;
 | 
						|
	
 | 
						|
	/* get editor data */
 | 
						|
	if (ANIM_animdata_get_context(C, &ac) == 0)
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
	
 | 
						|
	/* filter data */
 | 
						|
	filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_SEL | ANIMFILTER_FOREDIT | ANIMFILTER_CURVESONLY);
 | 
						|
	ANIM_animdata_filter(&ac, &anim_data, filter, ac.data, ac.datatype);
 | 
						|
	
 | 
						|
	/* paste modifiers */
 | 
						|
	for (ale = anim_data.first; ale; ale = ale->next) {
 | 
						|
		FCurve *fcu= (FCurve *)ale->data;
 | 
						|
		
 | 
						|
		// TODO: do we want to replace existing modifiers? add user pref for that!
 | 
						|
		ok += ANIM_fmodifiers_paste_from_buf(&fcu->modifiers, 0);
 | 
						|
	}
 | 
						|
	
 | 
						|
	/* clean up */
 | 
						|
	BLI_freelistN(&anim_data);
 | 
						|
	
 | 
						|
	/* successful or not? */
 | 
						|
	if (ok) {
 | 
						|
		/* validate keyframes after editing */
 | 
						|
		ANIM_editkeyframes_refresh(&ac);
 | 
						|
		
 | 
						|
		/* set notifier that keyframes have changed */
 | 
						|
		WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_EDITED, NULL);
 | 
						|
		
 | 
						|
		return OPERATOR_FINISHED;
 | 
						|
	}
 | 
						|
	else {
 | 
						|
		BKE_report(op->reports, RPT_ERROR, "No F-Modifiers to paste");
 | 
						|
		return OPERATOR_CANCELLED;
 | 
						|
	}
 | 
						|
}
 | 
						|
 
 | 
						|
void GRAPH_OT_fmodifier_paste (wmOperatorType *ot)
 | 
						|
{
 | 
						|
	/* identifiers */
 | 
						|
	ot->name= "Paste F-Modifiers";
 | 
						|
	ot->idname= "GRAPH_OT_fmodifier_paste";
 | 
						|
	ot->description= "Add copied F-Modifiers to the selected F-Curves";
 | 
						|
	
 | 
						|
	/* api callbacks */
 | 
						|
	ot->exec= graph_fmodifier_paste_exec;
 | 
						|
	ot->poll= graphop_editable_keyframes_poll;
 | 
						|
	
 | 
						|
	/* flags */
 | 
						|
	ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 | 
						|
}
 | 
						|
 | 
						|
/* ************************************************************************** */
 |