operator renaming for more consistent word ordering (_add/_remmove shold be last...
[blender.git] / source / blender / editors / space_action / action_edit.c
1 /**
2  * $Id$
3  *
4  * ***** BEGIN GPL LICENSE BLOCK *****
5  *
6  * This program is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU General Public License
8  * as published by the Free Software Foundation; either version 2
9  * of the License, or (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software Foundation,
18  * Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
19  *
20  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
21  * All rights reserved.
22  *
23  * The Original Code is: all of this file.
24  *
25  * Contributor(s): Joshua Leung
26  *
27  * ***** END GPL LICENSE BLOCK *****
28  */
29
30 #include <math.h>
31 #include <stdlib.h>
32 #include <string.h>
33 #include <float.h>
34
35 #ifdef HAVE_CONFIG_H
36 #include <config.h>
37 #endif
38
39 #include "MEM_guardedalloc.h"
40
41 #include "BLI_blenlib.h"
42 #include "BLI_math.h"
43
44 #include "DNA_anim_types.h"
45 #include "DNA_action_types.h"
46 #include "DNA_armature_types.h"
47 #include "DNA_camera_types.h"
48 #include "DNA_curve_types.h"
49 #include "DNA_object_types.h"
50 #include "DNA_screen_types.h"
51 #include "DNA_scene_types.h"
52 #include "DNA_space_types.h"
53 #include "DNA_constraint_types.h"
54 #include "DNA_key_types.h"
55 #include "DNA_lamp_types.h"
56 #include "DNA_material_types.h"
57 #include "DNA_userdef_types.h"
58 #include "DNA_gpencil_types.h"
59 #include "DNA_windowmanager_types.h"
60
61 #include "RNA_access.h"
62 #include "RNA_define.h"
63 #include "RNA_enum_types.h"
64
65 #include "BKE_action.h"
66 #include "BKE_depsgraph.h"
67 #include "BKE_fcurve.h"
68 #include "BKE_key.h"
69 #include "BKE_material.h"
70 #include "BKE_nla.h"
71 #include "BKE_object.h"
72 #include "BKE_context.h"
73 #include "BKE_report.h"
74 #include "BKE_utildefines.h"
75
76 #include "UI_view2d.h"
77
78 #include "ED_anim_api.h"
79 #include "ED_keyframing.h"
80 #include "ED_keyframes_draw.h"
81 #include "ED_keyframes_edit.h"
82 #include "ED_screen.h"
83 #include "ED_space_api.h"
84 #include "ED_transform.h"
85
86 #include "WM_api.h"
87 #include "WM_types.h"
88
89 #include "UI_interface.h"
90
91 #include "action_intern.h"
92
93 /* ************************************************************************** */
94 /* ACTION MANAGEMENT */
95
96 /* ******************** New Action Operator *********************** */
97
98 static int act_new_exec(bContext *C, wmOperator *op)
99 {
100         bAction *action;
101         PointerRNA ptr, idptr;
102         PropertyRNA *prop;
103
104         // XXX need to restore behaviour to copy old actions...
105         action= add_empty_action("Action");
106
107         /* hook into UI */
108         uiIDContextProperty(C, &ptr, &prop);
109
110         if(prop) {
111                 /* when creating new ID blocks, use is already 1, but RNA
112                  * pointer se also increases user, so this compensates it */
113                 action->id.us--;
114
115                 RNA_id_pointer_create(&action->id, &idptr);
116                 RNA_property_pointer_set(&ptr, prop, idptr);
117                 RNA_property_update(C, &ptr, prop);
118         }
119
120         /* set notifier that keyframes have changed */
121         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME_EDIT, NULL);
122         
123         return OPERATOR_FINISHED;
124 }
125  
126 void ACTION_OT_new (wmOperatorType *ot)
127 {
128         /* identifiers */
129         ot->name= "New";
130         ot->idname= "ACTION_OT_new";
131         ot->description= "Create new action.";
132         
133         /* api callbacks */
134         ot->exec= act_new_exec;
135                 // NOTE: this is used in the NLA too...
136         //ot->poll= ED_operator_action_active;
137         
138         /* flags */
139         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
140 }
141
142 /* ************************************************************************** */
143 /* KEYFRAME-RANGE STUFF */
144
145 /* *************************** Calculate Range ************************** */
146
147 /* Get the min/max keyframes*/
148 static void get_keyframe_extents (bAnimContext *ac, float *min, float *max)
149 {
150         ListBase anim_data = {NULL, NULL};
151         bAnimListElem *ale;
152         int filter;
153         
154         /* get data to filter, from Action or Dopesheet */
155         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_SEL | ANIMFILTER_CURVESONLY);
156         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
157         
158         /* set large values to try to override */
159         *min= 999999999.0f;
160         *max= -999999999.0f;
161         
162         /* check if any channels to set range with */
163         if (anim_data.first) {
164                 /* go through channels, finding max extents */
165                 for (ale= anim_data.first; ale; ale= ale->next) {
166                         AnimData *adt= ANIM_nla_mapping_get(ac, ale);
167                         FCurve *fcu= (FCurve *)ale->key_data;
168                         float tmin, tmax;
169                         
170                         /* get range and apply necessary scaling before */
171                         calc_fcurve_range(fcu, &tmin, &tmax);
172                         
173                         if (adt) {
174                                 tmin= BKE_nla_tweakedit_remap(adt, tmin, NLATIME_CONVERT_MAP);
175                                 tmax= BKE_nla_tweakedit_remap(adt, tmax, NLATIME_CONVERT_MAP);
176                         }
177                         
178                         /* try to set cur using these values, if they're more extreme than previously set values */
179                         *min= MIN2(*min, tmin);
180                         *max= MAX2(*max, tmax);
181                 }
182                 
183                 /* free memory */
184                 BLI_freelistN(&anim_data);
185         }
186         else {
187                 /* set default range */
188                 if (ac->scene) {
189                         *min= (float)ac->scene->r.sfra;
190                         *max= (float)ac->scene->r.efra;
191                 }
192                 else {
193                         *min= -5;
194                         *max= 100;
195                 }
196         }
197 }
198
199 /* ****************** Automatic Preview-Range Operator ****************** */
200
201 static int actkeys_previewrange_exec(bContext *C, wmOperator *op)
202 {
203         bAnimContext ac;
204         Scene *scene;
205         float min, max;
206         
207         /* get editor data */
208         if (ANIM_animdata_get_context(C, &ac) == 0)
209                 return OPERATOR_CANCELLED;
210         if (ac.scene == NULL)
211                 return OPERATOR_CANCELLED;
212         else
213                 scene= ac.scene;
214         
215         /* set the range directly */
216         get_keyframe_extents(&ac, &min, &max);
217         scene->r.psfra= (int)floor(min + 0.5f);
218         scene->r.pefra= (int)floor(max + 0.5f);
219         
220         /* set notifier that things have changed */
221         // XXX err... there's nothing for frame ranges yet, but this should do fine too
222         WM_event_add_notifier(C, NC_SCENE|ND_FRAME, ac.scene); 
223         
224         return OPERATOR_FINISHED;
225 }
226  
227 void ACTION_OT_previewrange_set (wmOperatorType *ot)
228 {
229         /* identifiers */
230         ot->name= "Auto-Set Preview Range";
231         ot->idname= "ACTION_OT_previewrange_set";
232         ot->description= "Set Preview Range based on extents of selected Keyframes.";
233         
234         /* api callbacks */
235         ot->exec= actkeys_previewrange_exec;
236         ot->poll= ED_operator_action_active;
237         
238         /* flags */
239         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
240 }
241
242 /* ****************** View-All Operator ****************** */
243
244 static int actkeys_viewall_exec(bContext *C, wmOperator *op)
245 {
246         bAnimContext ac;
247         View2D *v2d;
248         float extra;
249         
250         /* get editor data */
251         if (ANIM_animdata_get_context(C, &ac) == 0)
252                 return OPERATOR_CANCELLED;
253         v2d= &ac.ar->v2d;
254         
255         /* set the horizontal range, with an extra offset so that the extreme keys will be in view */
256         get_keyframe_extents(&ac, &v2d->cur.xmin, &v2d->cur.xmax);
257         
258         extra= 0.1f * (v2d->cur.xmax - v2d->cur.xmin);
259         v2d->cur.xmin -= extra;
260         v2d->cur.xmax += extra;
261         
262         /* set vertical range */
263         v2d->cur.ymax= 0.0f;
264         v2d->cur.ymin= (float)-(v2d->mask.ymax - v2d->mask.ymin);
265         
266         /* do View2D syncing */
267         UI_view2d_sync(CTX_wm_screen(C), CTX_wm_area(C), v2d, V2D_LOCK_COPY);
268         
269         /* just redraw this view */
270         ED_area_tag_redraw(CTX_wm_area(C));
271         
272         return OPERATOR_FINISHED;
273 }
274  
275 void ACTION_OT_view_all (wmOperatorType *ot)
276 {
277         /* identifiers */
278         ot->name= "View All";
279         ot->idname= "ACTION_OT_view_all";
280         ot->description= "Reset viewable area to show full keyframe range.";
281         
282         /* api callbacks */
283         ot->exec= actkeys_viewall_exec;
284         ot->poll= ED_operator_action_active;
285         
286         /* flags */
287         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
288 }
289
290 /* ************************************************************************** */
291 /* GENERAL STUFF */
292
293 /* ******************** Copy/Paste Keyframes Operator ************************* */
294 /* NOTE: the backend code for this is shared with the graph editor */
295
296 static short copy_action_keys (bAnimContext *ac)
297 {       
298         ListBase anim_data = {NULL, NULL};
299         int filter, ok=0;
300         
301         /* clear buffer first */
302         free_anim_copybuf();
303         
304         /* filter data */
305         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVESONLY);
306         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
307         
308         /* copy keyframes */
309         ok= copy_animedit_keys(ac, &anim_data);
310         
311         /* clean up */
312         BLI_freelistN(&anim_data);
313
314         return ok;
315 }
316
317
318 static short paste_action_keys (bAnimContext *ac)
319 {       
320         ListBase anim_data = {NULL, NULL};
321         int filter, ok=0;
322         
323         /* filter data */
324         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_SEL | ANIMFILTER_FOREDIT | ANIMFILTER_CURVESONLY);
325         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
326         
327         /* paste keyframes */
328         ok= paste_animedit_keys(ac, &anim_data);
329         
330         /* clean up */
331         BLI_freelistN(&anim_data);
332
333         return ok;
334 }
335
336 /* ------------------- */
337
338 static int actkeys_copy_exec(bContext *C, wmOperator *op)
339 {
340         bAnimContext ac;
341         
342         /* get editor data */
343         if (ANIM_animdata_get_context(C, &ac) == 0)
344                 return OPERATOR_CANCELLED;
345         
346         /* copy keyframes */
347         if (ac.datatype == ANIMCONT_GPENCIL) {
348                 // FIXME...
349         }
350         else {
351                 if (copy_action_keys(&ac)) {    
352                         BKE_report(op->reports, RPT_ERROR, "No keyframes copied to keyframes copy/paste buffer");
353                         return OPERATOR_CANCELLED;
354                 }
355         }
356         
357         return OPERATOR_FINISHED;
358 }
359  
360 void ACTION_OT_copy (wmOperatorType *ot)
361 {
362         /* identifiers */
363         ot->name= "Copy Keyframes";
364         ot->idname= "ACTION_OT_copy";
365         ot->description= "Copy selected keyframes to the copy/paste buffer.";
366         
367         /* api callbacks */
368         ot->exec= actkeys_copy_exec;
369         ot->poll= ED_operator_action_active;
370         
371         /* flags */
372         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
373 }
374
375
376
377 static int actkeys_paste_exec(bContext *C, wmOperator *op)
378 {
379         bAnimContext ac;
380         
381         /* get editor data */
382         if (ANIM_animdata_get_context(C, &ac) == 0)
383                 return OPERATOR_CANCELLED;
384         
385         /* paste keyframes */
386         if (ac.datatype == ANIMCONT_GPENCIL) {
387                 // FIXME...
388         }
389         else {
390                 if (paste_action_keys(&ac)) {
391                         BKE_report(op->reports, RPT_ERROR, "No keyframes to paste");
392                         return OPERATOR_CANCELLED;
393                 }
394         }
395         
396         /* validate keyframes after editing */
397         ANIM_editkeyframes_refresh(&ac);
398         
399         /* set notifier that keyframes have changed */
400         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME_EDIT, NULL);
401         
402         return OPERATOR_FINISHED;
403 }
404  
405 void ACTION_OT_paste (wmOperatorType *ot)
406 {
407         /* identifiers */
408         ot->name= "Paste Keyframes";
409         ot->idname= "ACTION_OT_paste";
410         ot->description= "Paste keyframes from copy/paste buffer for the selected channels, starting on the current frame.";
411         
412         /* api callbacks */
413         ot->exec= actkeys_paste_exec;
414         ot->poll= ED_operator_action_active;
415         
416         /* flags */
417         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
418 }
419
420 /* ******************** Insert Keyframes Operator ************************* */
421
422 /* defines for insert keyframes tool */
423 EnumPropertyItem prop_actkeys_insertkey_types[] = {
424         {1, "ALL", 0, "All Channels", ""},
425         {2, "SEL", 0, "Only Selected Channels", ""},
426         {3, "GROUP", 0, "In Active Group", ""}, // xxx not in all cases
427         {0, NULL, 0, NULL, NULL}
428 };
429
430 /* this function is responsible for snapping keyframes to frame-times */
431 static void insert_action_keys(bAnimContext *ac, short mode) 
432 {
433         ListBase anim_data = {NULL, NULL};
434         bAnimListElem *ale;
435         int filter;
436         
437         Scene *scene= ac->scene;
438         float cfra= (float)CFRA;
439         short flag = 0;
440         
441         /* filter data */
442         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_FOREDIT | ANIMFILTER_CURVESONLY);
443         if (mode == 2)                  filter |= ANIMFILTER_SEL;
444         else if (mode == 3)     filter |= ANIMFILTER_ACTGROUPED;
445         
446         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
447         
448         /* init keyframing flag */
449         if (IS_AUTOKEY_FLAG(AUTOMATKEY)) flag |= INSERTKEY_MATRIX;
450         if (IS_AUTOKEY_FLAG(INSERTNEEDED)) flag |= INSERTKEY_NEEDED;
451         if (IS_AUTOKEY_MODE(scene, EDITKEYS)) flag |= INSERTKEY_REPLACE;
452         
453         /* insert keyframes */
454         for (ale= anim_data.first; ale; ale= ale->next) {
455                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
456                 FCurve *fcu= (FCurve *)ale->key_data;
457                 
458                 /* adjust current frame for NLA-scaling */
459                 if (adt)
460                         cfra= BKE_nla_tweakedit_remap(adt, (float)CFRA, NLATIME_CONVERT_UNMAP);
461                 else 
462                         cfra= (float)CFRA;
463                         
464                 /* if there's an id */
465                 if (ale->id)
466                         insert_keyframe(ale->id, NULL, ((fcu->grp)?(fcu->grp->name):(NULL)), fcu->rna_path, fcu->array_index, cfra, flag);
467                 else
468                         insert_vert_fcurve(fcu, cfra, fcu->curval, 0);
469         }
470         
471         BLI_freelistN(&anim_data);
472 }
473
474 /* ------------------- */
475
476 static int actkeys_insertkey_exec(bContext *C, wmOperator *op)
477 {
478         bAnimContext ac;
479         short mode;
480         
481         /* get editor data */
482         if (ANIM_animdata_get_context(C, &ac) == 0)
483                 return OPERATOR_CANCELLED;
484         if (ac.datatype == ANIMCONT_GPENCIL)
485                 return OPERATOR_CANCELLED;
486                 
487         /* what channels to affect? */
488         mode= RNA_enum_get(op->ptr, "type");
489         
490         /* insert keyframes */
491         insert_action_keys(&ac, mode);
492         
493         /* validate keyframes after editing */
494         ANIM_editkeyframes_refresh(&ac);
495         
496         /* set notifier that keyframes have changed */
497         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME_EDIT, NULL);
498         
499         return OPERATOR_FINISHED;
500 }
501
502 void ACT_OT_keyframe_insert (wmOperatorType *ot)
503 {
504         /* identifiers */
505         ot->name= "Insert Keyframes";
506         ot->idname= "ACT_OT_keyframe_insert";
507         ot->description= "Insert keyframes for the specified channels.";
508         
509         /* api callbacks */
510         ot->invoke= WM_menu_invoke;
511         ot->exec= actkeys_insertkey_exec;
512         ot->poll= ED_operator_action_active;
513         
514         /* flags */
515         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
516         
517         /* id-props */
518         RNA_def_enum(ot->srna, "type", prop_actkeys_insertkey_types, 0, "Type", "");
519 }
520
521 /* ******************** Duplicate Keyframes Operator ************************* */
522
523 static void duplicate_action_keys (bAnimContext *ac)
524 {
525         ListBase anim_data = {NULL, NULL};
526         bAnimListElem *ale;
527         int filter;
528         
529         /* filter data */
530         if (ac->datatype == ANIMCONT_GPENCIL)
531                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_FOREDIT);
532         else
533                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_FOREDIT | ANIMFILTER_CURVESONLY);
534         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
535         
536         /* loop through filtered data and delete selected keys */
537         for (ale= anim_data.first; ale; ale= ale->next) {
538                 //if (ale->type == ANIMTYPE_GPLAYER)
539                 //      delete_gplayer_frames((bGPDlayer *)ale->data);
540                 //else
541                         duplicate_fcurve_keys((FCurve *)ale->key_data);
542         }
543         
544         /* free filtered list */
545         BLI_freelistN(&anim_data);
546 }
547
548 /* ------------------- */
549
550 static int actkeys_duplicate_exec(bContext *C, wmOperator *op)
551 {
552         bAnimContext ac;
553         
554         /* get editor data */
555         if (ANIM_animdata_get_context(C, &ac) == 0)
556                 return OPERATOR_CANCELLED;
557                 
558         /* duplicate keyframes */
559         duplicate_action_keys(&ac);
560         
561         /* validate keyframes after editing */
562         ANIM_editkeyframes_refresh(&ac);
563         
564         /* set notifier that keyframes have changed */
565         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME_EDIT, NULL);
566         
567         return OPERATOR_FINISHED; // xxx - start transform
568 }
569
570 static int actkeys_duplicate_invoke(bContext *C, wmOperator *op, wmEvent *event)
571 {
572         actkeys_duplicate_exec(C, op);
573         
574         RNA_int_set(op->ptr, "mode", TFM_TIME_TRANSLATE);
575         WM_operator_name_call(C, "TFM_OT_transform", WM_OP_INVOKE_REGION_WIN, op->ptr);
576
577         return OPERATOR_FINISHED;
578 }
579  
580 void ACTION_OT_duplicate (wmOperatorType *ot)
581 {
582         /* identifiers */
583         ot->name= "Duplicate Keyframes";
584         ot->idname= "ACTION_OT_duplicate";
585         ot->description= "Make a copy of all selected keyframes.";
586         
587         /* api callbacks */
588         ot->invoke= actkeys_duplicate_invoke;
589         ot->exec= actkeys_duplicate_exec;
590         ot->poll= ED_operator_action_active;
591         
592         /* flags */
593         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
594         
595         /* to give to transform */
596         RNA_def_int(ot->srna, "mode", TFM_TIME_TRANSLATE, 0, INT_MAX, "Mode", "", 0, INT_MAX);
597 }
598
599 /* ******************** Delete Keyframes Operator ************************* */
600
601 static void delete_action_keys (bAnimContext *ac)
602 {
603         ListBase anim_data = {NULL, NULL};
604         bAnimListElem *ale;
605         int filter;
606         
607         /* filter data */
608         if (ac->datatype == ANIMCONT_GPENCIL)
609                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_FOREDIT);
610         else
611                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_FOREDIT | ANIMFILTER_CURVESONLY);
612         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
613         
614         /* loop through filtered data and delete selected keys */
615         for (ale= anim_data.first; ale; ale= ale->next) {
616                 //if (ale->type == ANIMTYPE_GPLAYER)
617                 //      delete_gplayer_frames((bGPDlayer *)ale->data);
618                 //else
619                         delete_fcurve_keys((FCurve *)ale->key_data); // XXX... this doesn't delete empty curves anymore
620         }
621         
622         /* free filtered list */
623         BLI_freelistN(&anim_data);
624 }
625
626 /* ------------------- */
627
628 static int actkeys_delete_exec(bContext *C, wmOperator *op)
629 {
630         bAnimContext ac;
631         
632         /* get editor data */
633         if (ANIM_animdata_get_context(C, &ac) == 0)
634                 return OPERATOR_CANCELLED;
635                 
636         /* delete keyframes */
637         delete_action_keys(&ac);
638         
639         /* validate keyframes after editing */
640         ANIM_editkeyframes_refresh(&ac);
641         
642         /* set notifier that keyframes have changed */
643         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME_EDIT, NULL);
644         
645         return OPERATOR_FINISHED;
646 }
647  
648 void ACTION_OT_delete (wmOperatorType *ot)
649 {
650         /* identifiers */
651         ot->name= "Delete Keyframes";
652         ot->idname= "ACTION_OT_delete";
653         ot->description= "Remove all selected keyframes.";
654         
655         /* api callbacks */
656         ot->invoke= WM_operator_confirm;
657         ot->exec= actkeys_delete_exec;
658         ot->poll= ED_operator_action_active;
659         
660         /* flags */
661         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
662 }
663
664 /* ******************** Clean Keyframes Operator ************************* */
665
666 static void clean_action_keys (bAnimContext *ac, float thresh)
667 {       
668         ListBase anim_data = {NULL, NULL};
669         bAnimListElem *ale;
670         int filter;
671         
672         /* filter data */
673         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_FOREDIT | ANIMFILTER_SEL | ANIMFILTER_CURVESONLY);
674         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
675         
676         /* loop through filtered data and clean curves */
677         for (ale= anim_data.first; ale; ale= ale->next)
678                 clean_fcurve((FCurve *)ale->key_data, thresh);
679         
680         /* free temp data */
681         BLI_freelistN(&anim_data);
682 }
683
684 /* ------------------- */
685
686 static int actkeys_clean_exec(bContext *C, wmOperator *op)
687 {
688         bAnimContext ac;
689         float thresh;
690         
691         /* get editor data */
692         if (ANIM_animdata_get_context(C, &ac) == 0)
693                 return OPERATOR_CANCELLED;
694         if (ac.datatype == ANIMCONT_GPENCIL)
695                 return OPERATOR_PASS_THROUGH;
696                 
697         /* get cleaning threshold */
698         thresh= RNA_float_get(op->ptr, "threshold");
699         
700         /* clean keyframes */
701         clean_action_keys(&ac, thresh);
702         
703         /* validate keyframes after editing */
704         ANIM_editkeyframes_refresh(&ac);
705         
706         /* set notifier that keyframes have changed */
707         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME_EDIT, NULL);
708         
709         return OPERATOR_FINISHED;
710 }
711  
712 void ACTION_OT_clean (wmOperatorType *ot)
713 {
714         /* identifiers */
715         ot->name= "Clean Keyframes";
716         ot->idname= "ACTION_OT_clean";
717         ot->description= "Simplify F-Curves by removing closely spaced keyframes.";
718         
719         /* api callbacks */
720         //ot->invoke=  // XXX we need that number popup for this! 
721         ot->exec= actkeys_clean_exec;
722         ot->poll= ED_operator_action_active;
723         
724         /* flags */
725         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
726         
727         /* properties */
728         RNA_def_float(ot->srna, "threshold", 0.001f, 0.0f, FLT_MAX, "Threshold", "", 0.0f, 1000.0f);
729 }
730
731 /* ******************** Sample Keyframes Operator *********************** */
732
733 /* Evaluates the curves between each selected keyframe on each frame, and keys the value  */
734 static void sample_action_keys (bAnimContext *ac)
735 {       
736         ListBase anim_data = {NULL, NULL};
737         bAnimListElem *ale;
738         int filter;
739         
740         /* filter data */
741         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_FOREDIT | ANIMFILTER_CURVESONLY);
742         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
743         
744         /* loop through filtered data and add keys between selected keyframes on every frame  */
745         for (ale= anim_data.first; ale; ale= ale->next)
746                 sample_fcurve((FCurve *)ale->key_data);
747         
748         /* admin and redraws */
749         BLI_freelistN(&anim_data);
750 }
751
752 /* ------------------- */
753
754 static int actkeys_sample_exec(bContext *C, wmOperator *op)
755 {
756         bAnimContext ac;
757         
758         /* get editor data */
759         if (ANIM_animdata_get_context(C, &ac) == 0)
760                 return OPERATOR_CANCELLED;
761         if (ac.datatype == ANIMCONT_GPENCIL)
762                 return OPERATOR_PASS_THROUGH;
763         
764         /* sample keyframes */
765         sample_action_keys(&ac);
766         
767         /* validate keyframes after editing */
768         ANIM_editkeyframes_refresh(&ac);
769         
770         /* set notifier that keyframes have changed */
771         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME_EDIT, NULL);
772         
773         return OPERATOR_FINISHED;
774 }
775  
776 void ACTION_OT_sample (wmOperatorType *ot)
777 {
778         /* identifiers */
779         ot->name= "Sample Keyframes";
780         ot->idname= "ACTION_OT_sample";
781         ot->description= "Add keyframes on every frame between the selected keyframes.";
782         
783         /* api callbacks */
784         ot->exec= actkeys_sample_exec;
785         ot->poll= ED_operator_action_active;
786         
787         /* flags */
788         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
789 }
790
791 /* ************************************************************************** */
792 /* SETTINGS STUFF */
793
794 /* ******************** Set Extrapolation-Type Operator *********************** */
795
796 /* defines for set extrapolation-type for selected keyframes tool */
797 EnumPropertyItem prop_actkeys_expo_types[] = {
798         {FCURVE_EXTRAPOLATE_CONSTANT, "CONSTANT", 0, "Constant Extrapolation", ""},
799         {FCURVE_EXTRAPOLATE_LINEAR, "LINEAR", 0, "Linear Extrapolation", ""},
800         {0, NULL, 0, NULL, NULL}
801 };
802
803 /* this function is responsible for setting extrapolation mode for keyframes */
804 static void setexpo_action_keys(bAnimContext *ac, short mode) 
805 {
806         ListBase anim_data = {NULL, NULL};
807         bAnimListElem *ale;
808         int filter;
809         
810         /* filter data */
811         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_FOREDIT | ANIMFILTER_CURVESONLY);
812         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
813         
814         /* loop through setting mode per F-Curve */
815         for (ale= anim_data.first; ale; ale= ale->next) {
816                 FCurve *fcu= (FCurve *)ale->data;
817                 fcu->extend= mode;
818         }
819         
820         /* cleanup */
821         BLI_freelistN(&anim_data);
822 }
823
824 /* ------------------- */
825
826 static int actkeys_expo_exec(bContext *C, wmOperator *op)
827 {
828         bAnimContext ac;
829         short mode;
830         
831         /* get editor data */
832         if (ANIM_animdata_get_context(C, &ac) == 0)
833                 return OPERATOR_CANCELLED;
834         if (ac.datatype == ANIMCONT_GPENCIL) 
835                 return OPERATOR_PASS_THROUGH;
836                 
837         /* get handle setting mode */
838         mode= RNA_enum_get(op->ptr, "type");
839         
840         /* set handle type */
841         setexpo_action_keys(&ac, mode);
842         
843         /* validate keyframes after editing */
844         ANIM_editkeyframes_refresh(&ac);
845         
846         /* set notifier that keyframe properties have changed */
847         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME_PROP, NULL);
848         
849         return OPERATOR_FINISHED;
850 }
851  
852 void ACTION_OT_extrapolation_type (wmOperatorType *ot)
853 {
854         /* identifiers */
855         ot->name= "Set Keyframe Extrapolation";
856         ot->idname= "ACTION_OT_extrapolation_type";
857         ot->description= "Set extrapolation mode for selected F-Curves.";
858         
859         /* api callbacks */
860         ot->invoke= WM_menu_invoke;
861         ot->exec= actkeys_expo_exec;
862         ot->poll= ED_operator_action_active;
863         
864         /* flags */
865         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
866         
867         /* id-props */
868         RNA_def_enum(ot->srna, "type", prop_actkeys_expo_types, 0, "Type", "");
869 }
870
871 /* ******************** Set Interpolation-Type Operator *********************** */
872
873 /* this function is responsible for setting interpolation mode for keyframes */
874 static void setipo_action_keys(bAnimContext *ac, short mode) 
875 {
876         ListBase anim_data = {NULL, NULL};
877         bAnimListElem *ale;
878         int filter;
879         BeztEditFunc set_cb= ANIM_editkeyframes_ipo(mode);
880         
881         /* filter data */
882         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_FOREDIT | ANIMFILTER_CURVESONLY);
883         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
884         
885         /* loop through setting BezTriple interpolation
886          * Note: we do not supply BeztEditData to the looper yet. Currently that's not necessary here...
887          */
888         for (ale= anim_data.first; ale; ale= ale->next)
889                 ANIM_fcurve_keys_bezier_loop(NULL, ale->key_data, NULL, set_cb, calchandles_fcurve);
890         
891         /* cleanup */
892         BLI_freelistN(&anim_data);
893 }
894
895 /* ------------------- */
896
897 static int actkeys_ipo_exec(bContext *C, wmOperator *op)
898 {
899         bAnimContext ac;
900         short mode;
901         
902         /* get editor data */
903         if (ANIM_animdata_get_context(C, &ac) == 0)
904                 return OPERATOR_CANCELLED;
905         if (ac.datatype == ANIMCONT_GPENCIL) 
906                 return OPERATOR_PASS_THROUGH;
907                 
908         /* get handle setting mode */
909         mode= RNA_enum_get(op->ptr, "type");
910         
911         /* set handle type */
912         setipo_action_keys(&ac, mode);
913         
914         /* validate keyframes after editing */
915         ANIM_editkeyframes_refresh(&ac);
916         
917         /* set notifier that keyframe properties have changed */
918         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME_PROP, NULL);
919         
920         return OPERATOR_FINISHED;
921 }
922  
923 void ACTION_OT_interpolation_type (wmOperatorType *ot)
924 {
925         /* identifiers */
926         ot->name= "Set Keyframe Interpolation";
927         ot->idname= "ACTION_OT_interpolation_type";
928         ot->description= "Set interpolation mode for the F-Curve segments starting from the selected keyframes.";
929         
930         /* api callbacks */
931         ot->invoke= WM_menu_invoke;
932         ot->exec= actkeys_ipo_exec;
933         ot->poll= ED_operator_action_active;
934         
935         /* flags */
936         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
937         
938         /* id-props */
939         RNA_def_enum(ot->srna, "type", beztriple_interpolation_mode_items, 0, "Type", "");
940 }
941
942 /* ******************** Set Handle-Type Operator *********************** */
943
944 /* this function is responsible for setting handle-type of selected keyframes */
945 static void sethandles_action_keys(bAnimContext *ac, short mode) 
946 {
947         ListBase anim_data = {NULL, NULL};
948         bAnimListElem *ale;
949         int filter;
950         BeztEditFunc set_cb= ANIM_editkeyframes_handles(mode);
951         
952         /* filter data */
953         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_FOREDIT | ANIMFILTER_CURVESONLY);
954         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
955         
956         /* loop through setting flags for handles 
957          * Note: we do not supply BeztEditData to the looper yet. Currently that's not necessary here...
958          */
959         for (ale= anim_data.first; ale; ale= ale->next) {
960                 if (mode == -1) {       
961                         BeztEditFunc toggle_cb;
962                         
963                         /* check which type of handle to set (free or aligned) 
964                          *      - check here checks for handles with free alignment already
965                          */
966                         if (ANIM_fcurve_keys_bezier_loop(NULL, ale->key_data, NULL, set_cb, NULL))
967                                 toggle_cb= ANIM_editkeyframes_handles(HD_FREE);
968                         else
969                                 toggle_cb= ANIM_editkeyframes_handles(HD_ALIGN);
970                                 
971                         /* set handle-type */
972                         ANIM_fcurve_keys_bezier_loop(NULL, ale->key_data, NULL, toggle_cb, calchandles_fcurve);
973                 }
974                 else {
975                         /* directly set handle-type */
976                         ANIM_fcurve_keys_bezier_loop(NULL, ale->key_data, NULL, set_cb, calchandles_fcurve);
977                 }
978         }
979         
980         /* cleanup */
981         BLI_freelistN(&anim_data);
982 }
983
984 /* ------------------- */
985
986 static int actkeys_handletype_exec(bContext *C, wmOperator *op)
987 {
988         bAnimContext ac;
989         short mode;
990         
991         /* get editor data */
992         if (ANIM_animdata_get_context(C, &ac) == 0)
993                 return OPERATOR_CANCELLED;
994         if (ac.datatype == ANIMCONT_GPENCIL) 
995                 return OPERATOR_PASS_THROUGH;
996                 
997         /* get handle setting mode */
998         mode= RNA_enum_get(op->ptr, "type");
999         
1000         /* set handle type */
1001         sethandles_action_keys(&ac, mode);
1002         
1003         /* validate keyframes after editing */
1004         ANIM_editkeyframes_refresh(&ac);
1005         
1006         /* set notifier that keyframe properties have changed */
1007         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME_PROP, NULL);
1008         
1009         return OPERATOR_FINISHED;
1010 }
1011  
1012 void ACTION_OT_handle_type (wmOperatorType *ot)
1013 {
1014         /* identifiers */
1015         ot->name= "Set Keyframe Handle Type";
1016         ot->idname= "ACTION_OT_handle_type";
1017         ot->description= "Set type of handle for selected keyframes.";
1018         
1019         /* api callbacks */
1020         ot->invoke= WM_menu_invoke;
1021         ot->exec= actkeys_handletype_exec;
1022         ot->poll= ED_operator_action_active;
1023         
1024         /* flags */
1025         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
1026         
1027         /* id-props */
1028         RNA_def_enum(ot->srna, "type", beztriple_handle_type_items, 0, "Type", "");
1029 }
1030
1031 /* ******************** Set Keyframe-Type Operator *********************** */
1032
1033 /* this function is responsible for setting interpolation mode for keyframes */
1034 static void setkeytype_action_keys(bAnimContext *ac, short mode) 
1035 {
1036         ListBase anim_data = {NULL, NULL};
1037         bAnimListElem *ale;
1038         int filter;
1039         BeztEditFunc set_cb= ANIM_editkeyframes_keytype(mode);
1040         
1041         /* filter data */
1042         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_FOREDIT | ANIMFILTER_CURVESONLY);
1043         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
1044         
1045         /* loop through setting BezTriple interpolation
1046          * Note: we do not supply BeztEditData to the looper yet. Currently that's not necessary here...
1047          */
1048         for (ale= anim_data.first; ale; ale= ale->next)
1049                 ANIM_fcurve_keys_bezier_loop(NULL, ale->key_data, NULL, set_cb, NULL);
1050         
1051         /* cleanup */
1052         BLI_freelistN(&anim_data);
1053 }
1054
1055 /* ------------------- */
1056
1057 static int actkeys_keytype_exec(bContext *C, wmOperator *op)
1058 {
1059         bAnimContext ac;
1060         short mode;
1061         
1062         /* get editor data */
1063         if (ANIM_animdata_get_context(C, &ac) == 0)
1064                 return OPERATOR_CANCELLED;
1065         if (ac.datatype == ANIMCONT_GPENCIL) 
1066                 return OPERATOR_PASS_THROUGH;
1067                 
1068         /* get handle setting mode */
1069         mode= RNA_enum_get(op->ptr, "type");
1070         
1071         /* set handle type */
1072         setkeytype_action_keys(&ac, mode);
1073         
1074         /* validate keyframes after editing */
1075         ANIM_editkeyframes_refresh(&ac);
1076         
1077         /* set notifier that keyframe properties have changed */
1078         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME_PROP, NULL);
1079         
1080         return OPERATOR_FINISHED;
1081 }
1082  
1083 void ACTION_OT_keyframe_type (wmOperatorType *ot)
1084 {
1085         /* identifiers */
1086         ot->name= "Set Keyframe Type";
1087         ot->idname= "ACTION_OT_keyframe_type";
1088         ot->description= "Set type of keyframe for the seleced keyframes.";
1089         
1090         /* api callbacks */
1091         ot->invoke= WM_menu_invoke;
1092         ot->exec= actkeys_keytype_exec;
1093         ot->poll= ED_operator_action_active;
1094         
1095         /* flags */
1096         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
1097         
1098         /* id-props */
1099         RNA_def_enum(ot->srna, "type", beztriple_keyframe_type_items, 0, "Type", "");
1100 }
1101
1102 /* ************************************************************************** */
1103 /* TRANSFORM STUFF */
1104
1105 /* ***************** Jump to Selected Frames Operator *********************** */
1106
1107 /* snap current-frame indicator to 'average time' of selected keyframe */
1108 static int actkeys_framejump_exec(bContext *C, wmOperator *op)
1109 {
1110         bAnimContext ac;
1111         ListBase anim_data= {NULL, NULL};
1112         bAnimListElem *ale;
1113         int filter;
1114         BeztEditData bed;
1115         
1116         /* get editor data */
1117         if (ANIM_animdata_get_context(C, &ac) == 0)
1118                 return OPERATOR_CANCELLED;
1119         
1120         /* init edit data */
1121         memset(&bed, 0, sizeof(BeztEditData));
1122         
1123         /* loop over action data, averaging values */
1124         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVESONLY);
1125         ANIM_animdata_filter(&ac, &anim_data, filter, ac.data, ac.datatype);
1126         
1127         for (ale= anim_data.first; ale; ale= ale->next) {
1128                 AnimData *adt= ANIM_nla_mapping_get(&ac, ale);
1129                 
1130                 if (adt) {
1131                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 0, 1); 
1132                         ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, NULL, bezt_calc_average, NULL);
1133                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 1, 1);
1134                 }
1135                 else
1136                         ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, NULL, bezt_calc_average, NULL);
1137         }
1138         
1139         BLI_freelistN(&anim_data);
1140         
1141         /* set the new current frame value, based on the average time */
1142         if (bed.i1) {
1143                 Scene *scene= ac.scene;
1144                 CFRA= (int)floor((bed.f1 / bed.i1) + 0.5f);
1145         }
1146         
1147         /* set notifier that things have changed */
1148         WM_event_add_notifier(C, NC_SCENE|ND_FRAME, ac.scene);
1149         
1150         return OPERATOR_FINISHED;
1151 }
1152
1153 void ACTION_OT_frame_jump (wmOperatorType *ot)
1154 {
1155         /* identifiers */
1156         ot->name= "Jump to Frame";
1157         ot->idname= "ACTION_OT_frame_jump";
1158         ot->description= "Set the current frame to the average frame of the selected keyframes.";
1159         
1160         /* api callbacks */
1161         ot->exec= actkeys_framejump_exec;
1162         ot->poll= ED_operator_action_active;
1163         
1164         /* flags */
1165         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
1166 }
1167
1168 /* ******************** Snap Keyframes Operator *********************** */
1169
1170 /* defines for snap keyframes tool */
1171 EnumPropertyItem prop_actkeys_snap_types[] = {
1172         {ACTKEYS_SNAP_CFRA, "CFRA", 0, "Current frame", ""},
1173         {ACTKEYS_SNAP_NEAREST_FRAME, "NEAREST_FRAME", 0, "Nearest Frame", ""}, // XXX as single entry?
1174         {ACTKEYS_SNAP_NEAREST_SECOND, "NEAREST_SECOND", 0, "Nearest Second", ""}, // XXX as single entry?
1175         {ACTKEYS_SNAP_NEAREST_MARKER, "NEAREST_MARKER", 0, "Nearest Marker", ""},
1176         {0, NULL, 0, NULL, NULL}
1177 };
1178
1179 /* this function is responsible for snapping keyframes to frame-times */
1180 static void snap_action_keys(bAnimContext *ac, short mode) 
1181 {
1182         ListBase anim_data = {NULL, NULL};
1183         bAnimListElem *ale;
1184         int filter;
1185         
1186         BeztEditData bed;
1187         BeztEditFunc edit_cb;
1188         
1189         /* filter data */
1190         if (ac->datatype == ANIMCONT_GPENCIL)
1191                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_FOREDIT);
1192         else
1193                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_FOREDIT | ANIMFILTER_CURVESONLY);
1194         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
1195         
1196         /* get beztriple editing callbacks */
1197         edit_cb= ANIM_editkeyframes_snap(mode);
1198         
1199         memset(&bed, 0, sizeof(BeztEditData)); 
1200         bed.scene= ac->scene;
1201         if (mode == ACTKEYS_SNAP_NEAREST_MARKER) {
1202                 bed.list.first= (ac->markers) ? ac->markers->first : NULL;
1203                 bed.list.last= (ac->markers) ? ac->markers->last : NULL;
1204         }
1205         
1206         /* snap keyframes */
1207         for (ale= anim_data.first; ale; ale= ale->next) {
1208                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
1209                 
1210                 if (adt) {
1211                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 0, 1); 
1212                         ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, NULL, edit_cb, calchandles_fcurve);
1213                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 1, 1);
1214                 }
1215                 //else if (ale->type == ACTTYPE_GPLAYER)
1216                 //      snap_gplayer_frames(ale->data, mode);
1217                 else 
1218                         ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, NULL, edit_cb, calchandles_fcurve);
1219         }
1220         BLI_freelistN(&anim_data);
1221 }
1222
1223 /* ------------------- */
1224
1225 static int actkeys_snap_exec(bContext *C, wmOperator *op)
1226 {
1227         bAnimContext ac;
1228         short mode;
1229         
1230         /* get editor data */
1231         if (ANIM_animdata_get_context(C, &ac) == 0)
1232                 return OPERATOR_CANCELLED;
1233                 
1234         /* get snapping mode */
1235         mode= RNA_enum_get(op->ptr, "type");
1236         
1237         /* snap keyframes */
1238         snap_action_keys(&ac, mode);
1239         
1240         /* validate keyframes after editing */
1241         ANIM_editkeyframes_refresh(&ac);
1242         
1243         /* set notifier that keyframes have changed */
1244         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME_EDIT, NULL);
1245         
1246         return OPERATOR_FINISHED;
1247 }
1248  
1249 void ACTION_OT_snap (wmOperatorType *ot)
1250 {
1251         /* identifiers */
1252         ot->name= "Snap Keys";
1253         ot->idname= "ACTION_OT_snap";
1254         ot->description= "Snap selected keyframes to the times specified.";
1255         
1256         /* api callbacks */
1257         ot->invoke= WM_menu_invoke;
1258         ot->exec= actkeys_snap_exec;
1259         ot->poll= ED_operator_action_active;
1260         
1261         /* flags */
1262         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
1263         
1264         /* id-props */
1265         RNA_def_enum(ot->srna, "type", prop_actkeys_snap_types, 0, "Type", "");
1266 }
1267
1268 /* ******************** Mirror Keyframes Operator *********************** */
1269
1270 /* defines for mirror keyframes tool */
1271 EnumPropertyItem prop_actkeys_mirror_types[] = {
1272         {ACTKEYS_MIRROR_CFRA, "CFRA", 0, "By Times over Current frame", ""},
1273         {ACTKEYS_MIRROR_YAXIS, "YAXIS", 0, "By Times over Time=0", ""},
1274         {ACTKEYS_MIRROR_XAXIS, "XAXIS", 0, "By Values over Value=0", ""},
1275         {ACTKEYS_MIRROR_MARKER, "MARKER", 0, "By Times over First Selected Marker", ""},
1276         {0, NULL, 0, NULL, NULL}
1277 };
1278
1279 /* this function is responsible for mirroring keyframes */
1280 static void mirror_action_keys(bAnimContext *ac, short mode) 
1281 {
1282         ListBase anim_data = {NULL, NULL};
1283         bAnimListElem *ale;
1284         int filter;
1285         
1286         BeztEditData bed;
1287         BeztEditFunc edit_cb;
1288         
1289         /* get beztriple editing callbacks */
1290         edit_cb= ANIM_editkeyframes_mirror(mode);
1291         
1292         memset(&bed, 0, sizeof(BeztEditData)); 
1293         bed.scene= ac->scene;
1294         
1295         /* for 'first selected marker' mode, need to find first selected marker first! */
1296         // XXX should this be made into a helper func in the API?
1297         if (mode == ACTKEYS_MIRROR_MARKER) {
1298                 TimeMarker *marker= NULL;
1299                 
1300                 /* find first selected marker */
1301                 if (ac->markers) {
1302                         for (marker= ac->markers->first; marker; marker=marker->next) {
1303                                 if (marker->flag & SELECT) {
1304                                         break;
1305                                 }
1306                         }
1307                 }
1308                 
1309                 /* store marker's time (if available) */
1310                 if (marker)
1311                         bed.f1= (float)marker->frame;
1312                 else
1313                         return;
1314         }
1315         
1316         /* filter data */
1317         if (ac->datatype == ANIMCONT_GPENCIL)
1318                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_FOREDIT);
1319         else
1320                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_FOREDIT | ANIMFILTER_CURVESONLY);
1321         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
1322         
1323         /* mirror keyframes */
1324         for (ale= anim_data.first; ale; ale= ale->next) {
1325                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
1326                 
1327                 if (adt) {
1328                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 0, 1); 
1329                         ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, NULL, edit_cb, calchandles_fcurve);
1330                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 1, 1);
1331                 }
1332                 //else if (ale->type == ACTTYPE_GPLAYER)
1333                 //      snap_gplayer_frames(ale->data, mode);
1334                 else 
1335                         ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, NULL, edit_cb, calchandles_fcurve);
1336         }
1337         BLI_freelistN(&anim_data);
1338 }
1339
1340 /* ------------------- */
1341
1342 static int actkeys_mirror_exec(bContext *C, wmOperator *op)
1343 {
1344         bAnimContext ac;
1345         short mode;
1346         
1347         /* get editor data */
1348         if (ANIM_animdata_get_context(C, &ac) == 0)
1349                 return OPERATOR_CANCELLED;
1350                 
1351         /* get mirroring mode */
1352         mode= RNA_enum_get(op->ptr, "type");
1353         
1354         /* mirror keyframes */
1355         mirror_action_keys(&ac, mode);
1356         
1357         /* validate keyframes after editing */
1358         ANIM_editkeyframes_refresh(&ac);
1359         
1360         /* set notifier that keyframes have changed */
1361         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME_EDIT, NULL);
1362         
1363         return OPERATOR_FINISHED;
1364 }
1365  
1366 void ACTION_OT_mirror (wmOperatorType *ot)
1367 {
1368         /* identifiers */
1369         ot->name= "Mirror Keys";
1370         ot->idname= "ACTION_OT_mirror";
1371         ot->description= "Flip selected keyframes over the selected mirror line.";
1372         
1373         /* api callbacks */
1374         ot->invoke= WM_menu_invoke;
1375         ot->exec= actkeys_mirror_exec;
1376         ot->poll= ED_operator_action_active;
1377         
1378         /* flags */
1379         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
1380         
1381         /* id-props */
1382         RNA_def_enum(ot->srna, "type", prop_actkeys_mirror_types, 0, "Type", "");
1383 }
1384
1385 /* ************************************************************************** */