Operator Execution Contexts Bugfix:
[blender.git] / source / blender / editors / space_graph / graph_select.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) 2008 Blender Foundation
21  *
22  * Contributor(s): Joshua Leung
23  *
24  * ***** END GPL LICENSE BLOCK *****
25  */
26
27 #include <math.h>
28 #include <stdlib.h>
29 #include <string.h>
30 #include <float.h>
31
32 #ifdef HAVE_CONFIG_H
33 #include <config.h>
34 #endif
35
36 #include "MEM_guardedalloc.h"
37
38 #include "BLI_blenlib.h"
39 #include "BLI_math.h"
40
41 #include "DNA_anim_types.h"
42 #include "DNA_action_types.h"
43 #include "DNA_armature_types.h"
44 #include "DNA_camera_types.h"
45 #include "DNA_curve_types.h"
46 #include "DNA_object_types.h"
47 #include "DNA_screen_types.h"
48 #include "DNA_scene_types.h"
49 #include "DNA_space_types.h"
50 #include "DNA_constraint_types.h"
51 #include "DNA_key_types.h"
52 #include "DNA_lamp_types.h"
53 #include "DNA_material_types.h"
54 #include "DNA_userdef_types.h"
55 #include "DNA_gpencil_types.h"
56 #include "DNA_windowmanager_types.h"
57
58 #include "RNA_access.h"
59 #include "RNA_define.h"
60
61 #include "BKE_action.h"
62 #include "BKE_depsgraph.h"
63 #include "BKE_fcurve.h"
64 #include "BKE_key.h"
65 #include "BKE_material.h"
66 #include "BKE_nla.h"
67 #include "BKE_object.h"
68 #include "BKE_context.h"
69 #include "BKE_utildefines.h"
70
71 #include "UI_view2d.h"
72
73 #include "ED_anim_api.h"
74 #include "ED_keyframing.h"
75 #include "ED_keyframes_draw.h"
76 #include "ED_keyframes_edit.h"
77 #include "ED_markers.h"
78 #include "ED_screen.h"
79 #include "ED_space_api.h"
80
81 #include "WM_api.h"
82 #include "WM_types.h"
83
84 #include "graph_intern.h"
85
86
87 /* ************************************************************************** */
88 /* KEYFRAMES STUFF */
89
90 /* ******************** Deselect All Operator ***************************** */
91 /* This operator works in one of three ways:
92  *      1) (de)select all (AKEY) - test if select all or deselect all
93  *      2) invert all (CTRL-IKEY) - invert selection of all keyframes
94  *      3) (de)select all - no testing is done; only for use internal tools as normal function...
95  */
96
97 /* Deselects keyframes in the Graph Editor
98  *      - This is called by the deselect all operator, as well as other ones!
99  *
100  *      - test: check if select or deselect all
101  *      - sel: how to select keyframes 
102  *              0 = deselect
103  *              1 = select
104  *              2 = invert
105  */
106 static void deselect_graph_keys (bAnimContext *ac, short test, short sel)
107 {
108         ListBase anim_data = {NULL, NULL};
109         bAnimListElem *ale;
110         int filter;
111         
112         SpaceIpo *sipo= (SpaceIpo *)ac->sa->spacedata.first;
113         BeztEditData bed;
114         BeztEditFunc test_cb, sel_cb;
115         
116         /* determine type-based settings */
117         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_CURVESONLY);
118         
119         /* filter data */
120         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
121         
122         /* init BezTriple looping data */
123         memset(&bed, 0, sizeof(BeztEditData));
124         test_cb= ANIM_editkeyframes_ok(BEZT_OK_SELECTED);
125         
126         /* See if we should be selecting or deselecting */
127         if (test) {
128                 for (ale= anim_data.first; ale; ale= ale->next) {
129                         if (ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, NULL, test_cb, NULL)) {
130                                 sel= SELECT_SUBTRACT;
131                                 break;
132                         }
133                 }
134         }
135         
136         /* convert sel to selectmode, and use that to get editor */
137         sel_cb= ANIM_editkeyframes_select(sel);
138         
139         /* Now set the flags */
140         for (ale= anim_data.first; ale; ale= ale->next) {
141                 FCurve *fcu= (FCurve *)ale->key_data;
142                 
143                 /* Keyframes First */
144                 ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, NULL, sel_cb, NULL);
145                 
146                 /* only change selection of channel when the visibility of keyframes doesn't depend on this */
147                 if ((sipo->flag & SIPO_SELCUVERTSONLY) == 0) {
148                         /* deactivate the F-Curve, and deselect if deselecting keyframes.
149                          * otherwise select the F-Curve too since we've selected all the keyframes
150                          */
151                         if (sel == SELECT_SUBTRACT) 
152                                 fcu->flag &= ~FCURVE_SELECTED;
153                         else
154                                 fcu->flag |= FCURVE_SELECTED;
155                 }
156                 
157                 /* always deactivate all F-Curves if we perform batch ops for selection */
158                 fcu->flag &= ~FCURVE_ACTIVE;
159         }
160         
161         /* Cleanup */
162         BLI_freelistN(&anim_data);
163 }
164
165 /* ------------------- */
166
167 static int graphkeys_deselectall_exec(bContext *C, wmOperator *op)
168 {
169         bAnimContext ac;
170         
171         /* get editor data */
172         if (ANIM_animdata_get_context(C, &ac) == 0)
173                 return OPERATOR_CANCELLED;
174                 
175         /* 'standard' behaviour - check if selected, then apply relevant selection */
176         if (RNA_boolean_get(op->ptr, "invert"))
177                 deselect_graph_keys(&ac, 0, SELECT_INVERT);
178         else
179                 deselect_graph_keys(&ac, 1, SELECT_ADD);
180         
181         /* set notifier that things have changed */
182         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME_SELECT, NULL);
183         
184         return OPERATOR_FINISHED;
185 }
186  
187 void GRAPH_OT_select_all_toggle (wmOperatorType *ot)
188 {
189         /* identifiers */
190         ot->name= "Select All";
191         ot->idname= "GRAPH_OT_select_all_toggle";
192         ot->description= "Toggle selection of all keyframes.";
193         
194         /* api callbacks */
195         ot->exec= graphkeys_deselectall_exec;
196         ot->poll= graphop_visible_keyframes_poll;
197         
198         /* flags */
199         ot->flag= OPTYPE_REGISTER/*|OPTYPE_UNDO*/;
200         
201         /* props */
202         ot->prop= RNA_def_boolean(ot->srna, "invert", 0, "Invert", "");
203 }
204
205 /* ******************** Border Select Operator **************************** */
206 /* This operator currently works in one of three ways:
207  *      -> BKEY         - 1) all keyframes within region are selected (validation with BEZT_OK_REGION)
208  *      -> ALT-BKEY - depending on which axis of the region was larger...
209  *              -> 2) x-axis, so select all frames within frame range (validation with BEZT_OK_FRAMERANGE)
210  *              -> 3) y-axis, so select all frames within channels that region included (validation with BEZT_OK_VALUERANGE)
211  */
212
213 /* Borderselect only selects keyframes now, as overshooting handles often get caught too,
214  * which means that they may be inadvertantly moved as well.
215  * Also, for convenience, handles should get same status as keyframe (if it was within bounds)
216  */
217 static void borderselect_graphkeys (bAnimContext *ac, rcti rect, short mode, short selectmode)
218 {
219         ListBase anim_data = {NULL, NULL};
220         bAnimListElem *ale;
221         int filter;
222         
223         SpaceIpo *sipo= (SpaceIpo *)ac->sa->spacedata.first;
224         BeztEditData bed;
225         BeztEditFunc ok_cb, select_cb;
226         View2D *v2d= &ac->ar->v2d;
227         rctf rectf;
228         
229         /* convert mouse coordinates to frame ranges and channel coordinates corrected for view pan/zoom */
230         UI_view2d_region_to_view(v2d, rect.xmin, rect.ymin, &rectf.xmin, &rectf.ymin);
231         UI_view2d_region_to_view(v2d, rect.xmax, rect.ymax, &rectf.xmax, &rectf.ymax);
232         
233         /* filter data */
234         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVESONLY | ANIMFILTER_CURVEVISIBLE);
235         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
236         
237         /* get beztriple editing/validation funcs  */
238         select_cb= ANIM_editkeyframes_select(selectmode);
239         ok_cb= ANIM_editkeyframes_ok(mode);
240         
241         /* init editing data */
242         memset(&bed, 0, sizeof(BeztEditData));
243         bed.data= &rectf;
244         
245         /* loop over data, doing border select */
246         for (ale= anim_data.first; ale; ale= ale->next) {
247                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
248                 FCurve *fcu= (FCurve *)ale->key_data;
249                 
250                 /* apply unit corrections */
251                 ANIM_unit_mapping_apply_fcurve(ac->scene, ale->id, ale->key_data, ANIM_UNITCONV_ONLYKEYS);
252                 
253                 /* apply NLA mapping to all the keyframes, since it's easier than trying to
254                  * guess when a callback might use something different
255                  */
256                 if (adt)
257                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 0, 1);
258                 
259                 /* set horizontal range (if applicable) 
260                  * NOTE: these values are only used for x-range and y-range but not region 
261                  *              (which uses bed.data, i.e. rectf)
262                  */
263                 if (mode != BEZT_OK_VALUERANGE) {
264                         bed.f1= rectf.xmin;
265                         bed.f2= rectf.xmax;
266                 }
267                 else {
268                         bed.f1= rectf.ymin;
269                         bed.f2= rectf.ymax;
270                 }
271                 
272                 /* firstly, check if any keyframes will be hit by this */
273                 if (ANIM_fcurve_keys_bezier_loop(&bed, fcu, NULL, ok_cb, NULL)) {
274                         /* select keyframes that are in the appropriate places */
275                         ANIM_fcurve_keys_bezier_loop(&bed, fcu, ok_cb, select_cb, NULL);
276                         
277                         /* only change selection of channel when the visibility of keyframes doesn't depend on this */
278                         if ((sipo->flag & SIPO_SELCUVERTSONLY) == 0) {
279                                 /* select the curve too now that curve will be touched */
280                                 if (selectmode == SELECT_ADD)
281                                         fcu->flag |= FCURVE_SELECTED;
282                         }
283                 }
284                 
285                 /* un-apply NLA mapping from all the keyframes */
286                 if (adt)
287                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 1, 1);
288                         
289                 /* unapply unit corrections */
290                 ANIM_unit_mapping_apply_fcurve(ac->scene, ale->id, ale->key_data, ANIM_UNITCONV_RESTORE|ANIM_UNITCONV_ONLYKEYS);
291         }
292         
293         /* cleanup */
294         BLI_freelistN(&anim_data);
295 }
296
297 /* ------------------- */
298
299 static int graphkeys_borderselect_exec(bContext *C, wmOperator *op)
300 {
301         bAnimContext ac;
302         rcti rect;
303         short mode=0, selectmode=0;
304         
305         /* get editor data */
306         if (ANIM_animdata_get_context(C, &ac) == 0)
307                 return OPERATOR_CANCELLED;
308
309         if (RNA_int_get(op->ptr, "gesture_mode")==GESTURE_MODAL_SELECT)
310                 selectmode= SELECT_ADD;
311         else
312                 selectmode= SELECT_SUBTRACT;
313
314         /* get settings from operator */
315         rect.xmin= RNA_int_get(op->ptr, "xmin");
316         rect.ymin= RNA_int_get(op->ptr, "ymin");
317         rect.xmax= RNA_int_get(op->ptr, "xmax");
318         rect.ymax= RNA_int_get(op->ptr, "ymax");
319         
320         /* selection 'mode' depends on whether borderselect region only matters on one axis */
321         if (RNA_boolean_get(op->ptr, "axis_range")) {
322                 /* mode depends on which axis of the range is larger to determine which axis to use 
323                  *      - checking this in region-space is fine, as it's fundamentally still going to be a different rect size
324                  *      - the frame-range select option is favoured over the channel one (x over y), as frame-range one is often
325                  *        used for tweaking timing when "blocking", while channels is not that useful...
326                  */
327                 if ((rect.xmax - rect.xmin) >= (rect.ymax - rect.ymin))
328                         mode= BEZT_OK_FRAMERANGE;
329                 else
330                         mode= BEZT_OK_VALUERANGE;
331         }
332         else 
333                 mode= BEZT_OK_REGION;
334         
335         /* apply borderselect action */
336         borderselect_graphkeys(&ac, rect, mode, selectmode);
337         
338         /* send notifier that keyframe selection has changed */
339         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME_SELECT, NULL);
340         
341         return OPERATOR_FINISHED;
342
343
344 void GRAPH_OT_select_border(wmOperatorType *ot)
345 {
346         /* identifiers */
347         ot->name= "Border Select";
348         ot->idname= "GRAPH_OT_select_border";
349         ot->description= "Select all keyframes within the specified region.";
350         
351         /* api callbacks */
352         ot->invoke= WM_border_select_invoke;
353         ot->exec= graphkeys_borderselect_exec;
354         ot->modal= WM_border_select_modal;
355         
356         ot->poll= graphop_visible_keyframes_poll;
357         
358         /* flags */
359         ot->flag= OPTYPE_REGISTER/*|OPTYPE_UNDO*/;
360         
361         /* rna */
362         WM_operator_properties_gesture_border(ot, FALSE);
363         
364         ot->prop= RNA_def_boolean(ot->srna, "axis_range", 0, "Axis Range", "");
365 }
366
367 /* ******************** Column Select Operator **************************** */
368 /* This operator works in one of four ways:
369  *      - 1) select all keyframes in the same frame as a selected one  (KKEY)
370  *      - 2) select all keyframes in the same frame as the current frame marker (CTRL-KKEY)
371  *      - 3) select all keyframes in the same frame as a selected markers (SHIFT-KKEY)
372  *      - 4) select all keyframes that occur between selected markers (ALT-KKEY)
373  */
374
375 /* defines for column-select mode */
376 static EnumPropertyItem prop_column_select_types[] = {
377         {GRAPHKEYS_COLUMNSEL_KEYS, "KEYS", 0, "On Selected Keyframes", ""},
378         {GRAPHKEYS_COLUMNSEL_CFRA, "CFRA", 0, "On Current Frame", ""},
379         {GRAPHKEYS_COLUMNSEL_MARKERS_COLUMN, "MARKERS_COLUMN", 0, "On Selected Markers", ""},
380         {GRAPHKEYS_COLUMNSEL_MARKERS_BETWEEN, "MARKERS_BETWEEN", 0, "Between Min/Max Selected Markers", ""},
381         {0, NULL, 0, NULL, NULL}
382 };
383
384 /* ------------------- */ 
385
386 /* Selects all visible keyframes between the specified markers */
387 static void markers_selectkeys_between (bAnimContext *ac)
388 {
389         ListBase anim_data = {NULL, NULL};
390         bAnimListElem *ale;
391         int filter;
392         
393         BeztEditFunc ok_cb, select_cb;
394         BeztEditData bed;
395         float min, max;
396         
397         /* get extreme markers */
398         ED_markers_get_minmax(ac->markers, 1, &min, &max);
399         min -= 0.5f;
400         max += 0.5f;
401         
402         /* get editing funcs + data */
403         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAMERANGE);
404         select_cb= ANIM_editkeyframes_select(SELECT_ADD);
405         
406         memset(&bed, 0, sizeof(BeztEditData));
407         bed.f1= min; 
408         bed.f2= max;
409         
410         /* filter data */
411         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_CURVESONLY);
412         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
413         
414         /* select keys in-between */
415         for (ale= anim_data.first; ale; ale= ale->next) {
416                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
417                 
418                 if (adt) {      
419                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 0, 1);
420                         ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, ok_cb, select_cb, NULL);
421                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 1, 1);
422                 }
423                 else {
424                         ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, ok_cb, select_cb, NULL);
425                 }
426         }
427         
428         /* Cleanup */
429         BLI_freelistN(&anim_data);
430 }
431
432
433 /* Selects all visible keyframes in the same frames as the specified elements */
434 static void columnselect_graph_keys (bAnimContext *ac, short mode)
435 {
436         ListBase anim_data= {NULL, NULL};
437         bAnimListElem *ale;
438         int filter;
439         
440         Scene *scene= ac->scene;
441         CfraElem *ce;
442         BeztEditFunc select_cb, ok_cb;
443         BeztEditData bed;
444         
445         /* initialise keyframe editing data */
446         memset(&bed, 0, sizeof(BeztEditData));
447         
448         /* build list of columns */
449         switch (mode) {
450                 case GRAPHKEYS_COLUMNSEL_KEYS: /* list of selected keys */
451                         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_CURVESONLY);
452                         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
453                         
454                         for (ale= anim_data.first; ale; ale= ale->next)
455                                 ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, NULL, bezt_to_cfraelem, NULL);
456                         
457                         BLI_freelistN(&anim_data);
458                         break;
459                         
460                 case GRAPHKEYS_COLUMNSEL_CFRA: /* current frame */
461                         /* make a single CfraElem for storing this */
462                         ce= MEM_callocN(sizeof(CfraElem), "cfraElem");
463                         BLI_addtail(&bed.list, ce);
464                         
465                         ce->cfra= (float)CFRA;
466                         break;
467                         
468                 case GRAPHKEYS_COLUMNSEL_MARKERS_COLUMN: /* list of selected markers */
469                         ED_markers_make_cfra_list(ac->markers, &bed.list, 1);
470                         break;
471                         
472                 default: /* invalid option */
473                         return;
474         }
475         
476         /* set up BezTriple edit callbacks */
477         select_cb= ANIM_editkeyframes_select(SELECT_ADD);
478         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAME);
479         
480         /* loop through all of the keys and select additional keyframes
481          * based on the keys found to be selected above
482          */
483         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_CURVESONLY);
484         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
485         
486         for (ale= anim_data.first; ale; ale= ale->next) {
487                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
488                 
489                 /* loop over cfraelems (stored in the BeztEditData->list)
490                  *      - we need to do this here, as we can apply fewer NLA-mapping conversions
491                  */
492                 for (ce= bed.list.first; ce; ce= ce->next) {
493                         /* set frame for validation callback to refer to */
494                         if (ale)
495                                 bed.f1= BKE_nla_tweakedit_remap(adt, ce->cfra, NLATIME_CONVERT_UNMAP);
496                         else
497                                 bed.f1= ce->cfra;
498                         
499                         /* select elements with frame number matching cfraelem */
500                         ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, ok_cb, select_cb, NULL);
501                 }
502         }
503         
504         /* free elements */
505         BLI_freelistN(&bed.list);
506         BLI_freelistN(&anim_data);
507 }
508
509 /* ------------------- */
510
511 static int graphkeys_columnselect_exec(bContext *C, wmOperator *op)
512 {
513         bAnimContext ac;
514         short mode;
515         
516         /* get editor data */
517         if (ANIM_animdata_get_context(C, &ac) == 0)
518                 return OPERATOR_CANCELLED;
519                 
520         /* action to take depends on the mode */
521         mode= RNA_enum_get(op->ptr, "mode");
522         
523         if (mode == GRAPHKEYS_COLUMNSEL_MARKERS_BETWEEN)
524                 markers_selectkeys_between(&ac);
525         else
526                 columnselect_graph_keys(&ac, mode);
527         
528         /* set notifier that keyframe selection has changed */
529         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME_SELECT, NULL);
530         
531         return OPERATOR_FINISHED;
532 }
533  
534 void GRAPH_OT_select_column (wmOperatorType *ot)
535 {
536         /* identifiers */
537         ot->name= "Select All";
538         ot->idname= "GRAPH_OT_select_column";
539         ot->description= "Select all keyframes on the specified frame(s).";
540         
541         /* api callbacks */
542         ot->exec= graphkeys_columnselect_exec;
543         ot->poll= graphop_visible_keyframes_poll;
544         
545         /* flags */
546         ot->flag= OPTYPE_REGISTER/*|OPTYPE_UNDO*/;
547         
548         /* props */
549         ot->prop= RNA_def_enum(ot->srna, "mode", prop_column_select_types, 0, "Mode", "");
550 }
551
552 /* ******************** Select More/Less Operators *********************** */
553
554 /* Common code to perform selection */
555 static void select_moreless_graph_keys (bAnimContext *ac, short mode)
556 {
557         ListBase anim_data= {NULL, NULL};
558         bAnimListElem *ale;
559         int filter;
560         
561         BeztEditData bed;
562         BeztEditFunc build_cb;
563         
564         
565         /* init selmap building data */
566         build_cb= ANIM_editkeyframes_buildselmap(mode);
567         memset(&bed, 0, sizeof(BeztEditData)); 
568         
569         /* loop through all of the keys and select additional keyframes based on these */
570         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_CURVESONLY);
571         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
572         
573         for (ale= anim_data.first; ale; ale= ale->next) {
574                 FCurve *fcu= (FCurve *)ale->key_data;
575                 
576                 /* only continue if F-Curve has keyframes */
577                 if (fcu->bezt == NULL)
578                         continue;
579                 
580                 /* build up map of whether F-Curve's keyframes should be selected or not */
581                 bed.data= MEM_callocN(fcu->totvert, "selmap graphEdit");
582                 ANIM_fcurve_keys_bezier_loop(&bed, fcu, NULL, build_cb, NULL);
583                 
584                 /* based on this map, adjust the selection status of the keyframes */
585                 ANIM_fcurve_keys_bezier_loop(&bed, fcu, NULL, bezt_selmap_flush, NULL);
586                 
587                 /* free the selmap used here */
588                 MEM_freeN(bed.data);
589                 bed.data= NULL;
590         }
591         
592         /* Cleanup */
593         BLI_freelistN(&anim_data);
594 }
595
596 /* ----------------- */
597
598 static int graphkeys_select_more_exec (bContext *C, wmOperator *op)
599 {
600         bAnimContext ac;
601         
602         /* get editor data */
603         if (ANIM_animdata_get_context(C, &ac) == 0)
604                 return OPERATOR_CANCELLED;
605         
606         /* perform select changes */
607         select_moreless_graph_keys(&ac, SELMAP_MORE);
608         
609         /* set notifier that keyframe selection has changed */
610         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME_SELECT, NULL);
611         
612         return OPERATOR_FINISHED;
613 }
614
615 void GRAPH_OT_select_more (wmOperatorType *ot)
616 {
617         /* identifiers */
618         ot->name = "Select More";
619         ot->idname= "GRAPH_OT_select_more";
620         ot->description = "Select keyframes beside already selected ones.";
621         
622         /* api callbacks */
623         ot->exec= graphkeys_select_more_exec;
624         ot->poll= graphop_visible_keyframes_poll;
625         
626         /* flags */
627         ot->flag= OPTYPE_REGISTER/*|OPTYPE_UNDO*/;
628 }
629
630 /* ----------------- */
631
632 static int graphkeys_select_less_exec (bContext *C, wmOperator *op)
633 {
634         bAnimContext ac;
635         
636         /* get editor data */
637         if (ANIM_animdata_get_context(C, &ac) == 0)
638                 return OPERATOR_CANCELLED;
639         
640         /* perform select changes */
641         select_moreless_graph_keys(&ac, SELMAP_LESS);
642         
643         /* set notifier that keyframe selection has changed */
644         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME_SELECT, NULL);
645         
646         return OPERATOR_FINISHED;
647 }
648
649 void GRAPH_OT_select_less (wmOperatorType *ot)
650 {
651         /* identifiers */
652         ot->name = "Select Less";
653         ot->idname= "GRAPH_OT_select_less";
654         ot->description = "Deselect keyframes on ends of selection islands.";
655         
656         /* api callbacks */
657         ot->exec= graphkeys_select_less_exec;
658         ot->poll= graphop_visible_keyframes_poll;
659         
660         /* flags */
661         ot->flag= OPTYPE_REGISTER/*|OPTYPE_UNDO*/;
662 }
663
664 /* ******************** Mouse-Click Select Operator *********************** */
665 /* This operator works in one of three ways:
666  *      - 1) keyframe under mouse - no special modifiers
667  *      - 2) all keyframes on the same side of current frame indicator as mouse - ALT modifier
668  *      - 3) column select all keyframes in frame under mouse - CTRL modifier
669  *
670  * In addition to these basic options, the SHIFT modifier can be used to toggle the 
671  * selection mode between replacing the selection (without) and inverting the selection (with).
672  */
673
674 /* defines for left-right select tool */
675 static EnumPropertyItem prop_graphkeys_leftright_select_types[] = {
676         {GRAPHKEYS_LRSEL_TEST, "CHECK", 0, "Check if Select Left or Right", ""},
677         {GRAPHKEYS_LRSEL_NONE, "OFF", 0, "Don't select", ""},
678         {GRAPHKEYS_LRSEL_LEFT, "LEFT", 0, "Before current frame", ""},
679         {GRAPHKEYS_LRSEL_RIGHT, "RIGHT", 0, "After current frame", ""},
680         {0, NULL, 0, NULL, NULL}
681 };
682
683 /* ------------------- */
684
685 enum {
686         NEAREST_HANDLE_LEFT     = 0,
687         NEAREST_HANDLE_KEY,
688         NEAREST_HANDLE_RIGHT
689 } eHandleIndex; 
690  
691 /* Find the vertex (either handle (0/2) or the keyframe (1)) that is nearest to the mouse cursor (in area coordinates)  
692  * Selected verts get a disadvantage, to make it easier to select handles behind.
693  * Returns eHandleIndex
694  */
695 static short findnearest_fcurve_vert (bAnimContext *ac, int mval[2], FCurve **fcurve, BezTriple **bezt)
696 {
697         ListBase anim_data = {NULL, NULL};
698         bAnimListElem *ale;
699         int filter;
700         
701         SpaceIpo *sipo= (SpaceIpo *)ac->sa->spacedata.first;
702         View2D *v2d= &ac->ar->v2d;
703         int hpoint=0, sco[3][2];
704         int dist= 100, temp, i;
705         
706         /* clear pointers first */
707         *fcurve= 0;
708         *bezt= 0;
709         
710         /* get curves to search through 
711          *      - if the option to only show keyframes that belong to selected F-Curves is enabled,
712          *        include the 'only selected' flag...
713          */
714         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_CURVESONLY);
715         if (sipo->flag & SIPO_SELCUVERTSONLY)   // FIXME: this should really be check for by the filtering code...
716                 filter |= ANIMFILTER_SEL;
717         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
718         
719         for (ale= anim_data.first; ale; ale= ale->next) {
720                 FCurve *fcu= (FCurve *)ale->key_data;
721                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
722                 
723                 /* apply unit corrections */
724                 ANIM_unit_mapping_apply_fcurve(ac->scene, ale->id, ale->key_data, 0);
725                 
726                 /* try to progressively get closer to the right point... */
727                 if (fcu->bezt) {
728                         BezTriple *bezt1=fcu->bezt, *prevbezt=NULL;
729                         
730                         /* apply NLA mapping to all the keyframes */
731                         if (adt)
732                                 ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 0, 1);
733                         
734                         for (i=0; i < fcu->totvert; i++, prevbezt=bezt1, bezt1++) {
735                                 /* convert beztriple points to screen-space */
736                                 UI_view2d_to_region_no_clip(v2d, bezt1->vec[0][0], bezt1->vec[0][1], &sco[0][0], &sco[0][1]);
737                                 UI_view2d_to_region_no_clip(v2d, bezt1->vec[1][0], bezt1->vec[1][1], &sco[1][0], &sco[1][1]);
738                                 UI_view2d_to_region_no_clip(v2d, bezt1->vec[2][0], bezt1->vec[2][1], &sco[2][0], &sco[2][1]);
739                                 
740                                 /* keyframe - do select? */
741                                 temp= abs(mval[0] - sco[1][0]) + abs(mval[1] - sco[1][1]);
742                                 
743                                 if (bezt1->f2 & SELECT) 
744                                         temp += 5;
745                                 
746                                 if (temp < dist) { 
747                                         hpoint= NEAREST_HANDLE_KEY; 
748                                         *bezt= bezt1; 
749                                         dist= temp; 
750                                         *fcurve= fcu; 
751                                 }
752                                 
753                                 /* handles - only do them if they're visible */
754                                 // XXX also need to check for int-values only?
755                                 if ((sipo->flag & SIPO_NOHANDLES)==0) {
756                                         /* first handle only visible if previous segment had handles */
757                                         if ( (!prevbezt && (bezt1->ipo==BEZT_IPO_BEZ)) || (prevbezt && (prevbezt->ipo==BEZT_IPO_BEZ)) )
758                                         {
759                                                 temp= -3 + abs(mval[0] - sco[0][0]) + abs(mval[1] - sco[0][1]);
760                                                 if (bezt1->f1 & SELECT) 
761                                                         temp += 5;
762                                                         
763                                                 if (temp < dist) { 
764                                                         hpoint= NEAREST_HANDLE_LEFT; 
765                                                         *bezt= bezt1; 
766                                                         dist= temp; 
767                                                         *fcurve= fcu; 
768                                                 }
769                                         }
770                                         
771                                         /* second handle only visible if this segment is bezier */
772                                         if (bezt1->ipo == BEZT_IPO_BEZ) 
773                                         {
774                                                 temp= abs(mval[0] - sco[2][0]) + abs(mval[1] - sco[2][1]);
775                                                 if (bezt1->f3 & SELECT) 
776                                                         temp += 5;
777                                                 
778                                                 if (temp < dist) { 
779                                                         hpoint= NEAREST_HANDLE_RIGHT; 
780                                                         *bezt=bezt1; 
781                                                         dist= temp; 
782                                                         *fcurve= fcu; 
783                                                 }
784                                         }
785                                 }
786                         }
787                         
788                         /* un-apply NLA mapping from all the keyframes */
789                         if (adt)
790                                 ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 1, 1);
791                 }
792                 
793                 /* unapply unit corrections */
794                 ANIM_unit_mapping_apply_fcurve(ac->scene, ale->id, ale->key_data, ANIM_UNITCONV_RESTORE);
795         }
796         
797         /* free channels */
798         BLI_freelistN(&anim_data);
799         
800         /* return handle */
801         return hpoint;
802 }
803  
804 /* option 1) select keyframe directly under mouse */
805 static void mouse_graph_keys (bAnimContext *ac, int mval[], short select_mode, short curves_only)
806 {
807         SpaceIpo *sipo= (SpaceIpo *)ac->sa->spacedata.first;
808         FCurve *fcu;
809         BezTriple *bezt;
810         short handle;
811         int filter;
812         
813         /* find the beztriple that we're selecting, and the handle that was clicked on */
814         handle= findnearest_fcurve_vert(ac, mval, &fcu, &bezt);
815         
816         /* check if anything to select */
817         if (fcu == NULL)        
818                 return;
819         
820         /* deselect all other curves? */
821         if (select_mode == SELECT_REPLACE) {
822                 /* reset selection mode */
823                 select_mode= SELECT_ADD;
824                 
825                 /* deselect all other keyframes */
826                 deselect_graph_keys(ac, 0, SELECT_SUBTRACT);
827                 
828                 /* deselect other channels too, but only only do this if 
829                  * selection of channel when the visibility of keyframes 
830                  * doesn't depend on this 
831                  */
832                 if ((sipo->flag & SIPO_SELCUVERTSONLY) == 0)
833                         ANIM_deselect_anim_channels(ac, ac->data, ac->datatype, 0, ACHANNEL_SETFLAG_CLEAR);
834         }
835         
836         /* if points can be selected on this F-Curve */
837         // TODO: what about those with no keyframes?
838         if ((curves_only == 0) && ((fcu->flag & FCURVE_PROTECTED)==0)) {
839                 /* only if there's keyframe */
840                 if (bezt) {
841                         /* depends on selection mode */
842                         if (select_mode == SELECT_INVERT) {
843                                 /* keyframe - invert select of all */
844                                 if (handle == NEAREST_HANDLE_KEY) {
845                                         if (BEZSELECTED(bezt)) {
846                                                 BEZ_DESEL(bezt);
847                                         }
848                                         else {
849                                                 BEZ_SEL(bezt);
850                                         }
851                                 }
852                                 
853                                 /* handles - toggle selection of relevant handle */
854                                 else if (handle == NEAREST_HANDLE_LEFT) {
855                                         /* toggle selection */
856                                         bezt->f1 ^= SELECT;
857                                 }
858                                 else {
859                                         /* toggle selection */
860                                         bezt->f3 ^= SELECT;
861                                 }
862                         }
863                         else {
864                                 /* if the keyframe was clicked on, select all verts of given beztriple */
865                                 if (handle == NEAREST_HANDLE_KEY) {
866                                         BEZ_SEL(bezt);
867                                 }
868                                 /* otherwise, select the handle that applied */
869                                 else if (handle == NEAREST_HANDLE_LEFT) 
870                                         bezt->f1 |= SELECT;
871                                 else 
872                                         bezt->f3 |= SELECT;
873                         }
874                 }
875         }
876         else {
877                 BeztEditFunc select_cb;
878                 BeztEditData bed;
879                 
880                 /* initialise keyframe editing data */
881                 memset(&bed, 0, sizeof(BeztEditData));
882                 
883                 /* set up BezTriple edit callbacks */
884                 select_cb= ANIM_editkeyframes_select(select_mode);
885                 
886                 /* select all keyframes */
887                 ANIM_fcurve_keys_bezier_loop(&bed, fcu, NULL, select_cb, NULL);
888         }
889         
890         /* only change selection of channel when the visibility of keyframes doesn't depend on this */
891         if ((sipo->flag & SIPO_SELCUVERTSONLY) == 0) {
892                 /* select or deselect curve? */
893                 if (select_mode == SELECT_INVERT)
894                         fcu->flag ^= FCURVE_SELECTED;
895                 else if (select_mode == SELECT_ADD)
896                         fcu->flag |= FCURVE_SELECTED;
897                         
898                 /* set active F-Curve (NOTE: sync the filter flags with findnearest_fcurve_vert) */
899                 if (fcu->flag & FCURVE_SELECTED) {
900                         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_CURVESONLY);
901                         ANIM_set_active_channel(ac, ac->data, ac->datatype, filter, fcu, ANIMTYPE_FCURVE);
902                 }
903         }
904 }
905
906 /* Option 2) Selects all the keyframes on either side of the current frame (depends on which side the mouse is on) */
907 static void graphkeys_mselect_leftright (bAnimContext *ac, short leftright, short select_mode)
908 {
909         ListBase anim_data = {NULL, NULL};
910         bAnimListElem *ale;
911         int filter;
912         
913         SpaceIpo *sipo= (SpaceIpo *)ac->sa->spacedata.first;
914         BeztEditFunc ok_cb, select_cb;
915         BeztEditData bed;
916         Scene *scene= ac->scene;
917         
918         /* if select mode is replace, deselect all keyframes (and channels) first */
919         if (select_mode==SELECT_REPLACE) {
920                 /* reset selection mode to add to selection */
921                 select_mode= SELECT_ADD;
922                 
923                 /* deselect all other keyframes */
924                 deselect_graph_keys(ac, 0, SELECT_SUBTRACT);
925                 
926                 /* deselect other channels too, but only only do this if 
927                  * selection of channel when the visibility of keyframes 
928                  * doesn't depend on this 
929                  */
930                 if ((sipo->flag & SIPO_SELCUVERTSONLY) == 0)
931                         ANIM_deselect_anim_channels(ac, ac->data, ac->datatype, 0, ACHANNEL_SETFLAG_CLEAR);
932         }
933         
934         /* set callbacks and editing data */
935         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAMERANGE);
936         select_cb= ANIM_editkeyframes_select(select_mode);
937         
938         memset(&bed, 0, sizeof(BeztEditFunc));
939         if (leftright == GRAPHKEYS_LRSEL_LEFT) {
940                 bed.f1 = MINAFRAMEF;
941                 bed.f2 = (float)(CFRA + 0.1f);
942         } 
943         else {
944                 bed.f1 = (float)(CFRA - 0.1f);
945                 bed.f2 = MAXFRAMEF;
946         }
947         
948         /* filter data */
949         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_CURVESONLY);
950         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
951                 
952         /* select keys on the side where most data occurs */
953         for (ale= anim_data.first; ale; ale= ale->next) {
954                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
955                 
956                 if (adt) {
957                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 0, 1);
958                         ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, ok_cb, select_cb, NULL);
959                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 1, 1);
960                 }
961                 else
962                         ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, ok_cb, select_cb, NULL);
963         }
964         
965         /* Cleanup */
966         BLI_freelistN(&anim_data);
967 }
968
969 /* Option 3) Selects all visible keyframes in the same frame as the mouse click */
970 static void graphkeys_mselect_column (bAnimContext *ac, int mval[2], short select_mode)
971 {
972         ListBase anim_data= {NULL, NULL};
973         bAnimListElem *ale;
974         int filter;
975         
976         SpaceIpo *sipo= (SpaceIpo *)ac->sa->spacedata.first;
977         BeztEditFunc select_cb, ok_cb;
978         BeztEditData bed;
979         FCurve *fcu;
980         BezTriple *bezt;
981         float selx = (float)ac->scene->r.cfra;
982         
983         /* find the beztriple that occurs on this frame, and use his as the frame number we're using */
984         findnearest_fcurve_vert(ac, mval, &fcu, &bezt);
985         
986         /* check if anything to select */
987         if (ELEM(NULL, fcu, bezt))      
988                 return;
989         selx= bezt->vec[1][0];
990         
991         /* if select mode is replace, deselect all keyframes (and channels) first */
992         if (select_mode==SELECT_REPLACE) {
993                 /* reset selection mode to add to selection */
994                 select_mode= SELECT_ADD;
995                 
996                 /* deselect all other keyframes */
997                 deselect_graph_keys(ac, 0, SELECT_SUBTRACT);
998                 
999                 /* deselect other channels too, but only only do this if 
1000                  * selection of channel when the visibility of keyframes 
1001                  * doesn't depend on this 
1002                  */
1003                 if ((sipo->flag & SIPO_SELCUVERTSONLY) == 0)
1004                         ANIM_deselect_anim_channels(ac, ac->data, ac->datatype, 0, ACHANNEL_SETFLAG_CLEAR);
1005         }
1006         
1007         /* initialise keyframe editing data */
1008         memset(&bed, 0, sizeof(BeztEditData));
1009         
1010         /* set up BezTriple edit callbacks */
1011         select_cb= ANIM_editkeyframes_select(select_mode);
1012         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAME);
1013         
1014         /* loop through all of the keys and select additional keyframes
1015          * based on the keys found to be selected above
1016          */
1017         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_CURVESONLY);
1018         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
1019         
1020         for (ale= anim_data.first; ale; ale= ale->next) {
1021                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
1022                 
1023                 /* set frame for validation callback to refer to */
1024                 if (adt)
1025                         bed.f1= BKE_nla_tweakedit_remap(adt, selx, NLATIME_CONVERT_UNMAP);
1026                 else
1027                         bed.f1= selx;
1028                 
1029                 /* select elements with frame number matching cfra */
1030                 ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, ok_cb, select_cb, NULL);
1031         }
1032         
1033         /* free elements */
1034         BLI_freelistN(&bed.list);
1035         BLI_freelistN(&anim_data);
1036 }
1037  
1038 /* ------------------- */
1039
1040 /* handle clicking */
1041 static int graphkeys_clickselect_invoke(bContext *C, wmOperator *op, wmEvent *event)
1042 {
1043         bAnimContext ac;
1044         Scene *scene;
1045         ARegion *ar;
1046         View2D *v2d;
1047         short selectmode;
1048         int mval[2];
1049         
1050         /* get editor data */
1051         if (ANIM_animdata_get_context(C, &ac) == 0)
1052                 return OPERATOR_CANCELLED;
1053         
1054         /* get useful pointers from animation context data */
1055         scene= ac.scene;
1056         ar= ac.ar;
1057         v2d= &ar->v2d;
1058         
1059         /* get mouse coordinates (in region coordinates) */
1060         mval[0]= (event->x - ar->winrct.xmin);
1061         mval[1]= (event->y - ar->winrct.ymin);
1062         
1063         /* select mode is either replace (deselect all, then add) or add/extend */
1064         if (RNA_boolean_get(op->ptr, "extend"))
1065                 selectmode= SELECT_INVERT;
1066         else
1067                 selectmode= SELECT_REPLACE;
1068         
1069         /* figure out action to take */
1070         if (RNA_enum_get(op->ptr, "left_right")) {
1071                 /* select all keys on same side of current frame as mouse */
1072                 float x;
1073                 
1074                 UI_view2d_region_to_view(v2d, mval[0], mval[1], &x, NULL);
1075                 if (x < CFRA)
1076                         RNA_int_set(op->ptr, "left_right", GRAPHKEYS_LRSEL_LEFT);
1077                 else    
1078                         RNA_int_set(op->ptr, "left_right", GRAPHKEYS_LRSEL_RIGHT);
1079                 
1080                 graphkeys_mselect_leftright(&ac, RNA_enum_get(op->ptr, "left_right"), selectmode);
1081         }
1082         else if (RNA_boolean_get(op->ptr, "column")) {
1083                 /* select all keyframes in the same frame as the one that was under the mouse */
1084                 graphkeys_mselect_column(&ac, mval, selectmode);
1085         }
1086         else if (RNA_boolean_get(op->ptr, "curves")) {
1087                 /* select all keyframes in F-Curve under mouse */
1088                 mouse_graph_keys(&ac, mval, selectmode, 1);
1089         }
1090         else {
1091                 /* select keyframe under mouse */
1092                 mouse_graph_keys(&ac, mval, selectmode, 0);
1093         }
1094         
1095         /* set notifier that keyframe selection (and also channel selection in some cases) has changed */
1096         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME_SELECT|ND_ANIMCHAN_SELECT, NULL);
1097         
1098         /* for tweak grab to work */
1099         return OPERATOR_FINISHED|OPERATOR_PASS_THROUGH;
1100 }
1101  
1102 void GRAPH_OT_clickselect (wmOperatorType *ot)
1103 {
1104         /* identifiers */
1105         ot->name= "Mouse Select Keys";
1106         ot->idname= "GRAPH_OT_clickselect";
1107         ot->description= "Select keyframes by clicking on them.";
1108         
1109         /* api callbacks */
1110         ot->invoke= graphkeys_clickselect_invoke;
1111         ot->poll= graphop_visible_keyframes_poll;
1112         
1113         /* id-props */
1114         // XXX should we make this into separate operators?
1115         RNA_def_enum(ot->srna, "left_right", prop_graphkeys_leftright_select_types, 0, "Left Right", ""); // CTRLKEY
1116         RNA_def_boolean(ot->srna, "extend", 0, "Extend Select", ""); // SHIFTKEY
1117         RNA_def_boolean(ot->srna, "column", 0, "Column Select", "Select all keyframes that occur on the same frame as the one under the mouse"); // ALTKEY
1118         RNA_def_boolean(ot->srna, "curves", 0, "Only Curves", "Select all the keyframes in the curve"); // CTRLKEY + ALTKEY
1119 }
1120
1121 /* ************************************************************************** */