DopeSheet: Add menu entries and hotkeys to select all keyframes
[blender.git] / source / blender / editors / space_action / action_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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, 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 #include "MEM_guardedalloc.h"
33
34 #include "BLI_blenlib.h"
35 #include "BLI_math.h"
36 #include "BLI_dlrbTree.h"
37 #include "BLI_utildefines.h"
38
39 #include "DNA_anim_types.h"
40 #include "DNA_gpencil_types.h"
41 #include "DNA_object_types.h"
42 #include "DNA_scene_types.h"
43
44 #include "RNA_access.h"
45 #include "RNA_define.h"
46
47 #include "BKE_fcurve.h"
48 #include "BKE_nla.h"
49 #include "BKE_context.h"
50
51 #include "UI_view2d.h"
52
53 #include "ED_anim_api.h"
54 #include "ED_gpencil.h"
55 #include "ED_keyframes_draw.h"
56 #include "ED_keyframes_edit.h"
57 #include "ED_markers.h"
58 #include "ED_screen.h"
59
60 #include "WM_api.h"
61 #include "WM_types.h"
62
63 #include "action_intern.h"
64
65
66 /* ************************************************************************** */
67 /* KEYFRAMES STUFF */
68
69 /* ******************** Deselect All Operator ***************************** */
70 /* This operator works in one of three ways:
71  *      1) (de)select all (AKEY) - test if select all or deselect all
72  *      2) invert all (CTRL-IKEY) - invert selection of all keyframes
73  *      3) (de)select all - no testing is done; only for use internal tools as normal function...
74  */
75
76 /* Deselects keyframes in the action editor
77  *      - This is called by the deselect all operator, as well as other ones!
78  *
79  *      - test: check if select or deselect all
80  *      - sel: how to select keyframes (SELECT_*)
81  */
82 static void deselect_action_keys (bAnimContext *ac, short test, short sel)
83 {
84         ListBase anim_data = {NULL, NULL};
85         bAnimListElem *ale;
86         int filter;
87         
88         KeyframeEditData ked= {{0}};
89         KeyframeEditFunc test_cb, sel_cb;
90         
91         /* determine type-based settings */
92         if (ac->datatype == ANIMCONT_GPENCIL)
93                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_NODUPLIS);
94         else
95                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVESONLY | ANIMFILTER_NODUPLIS);
96         
97         /* filter data */
98         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
99         
100         /* init BezTriple looping data */
101         test_cb= ANIM_editkeyframes_ok(BEZT_OK_SELECTED);
102         
103         /* See if we should be selecting or deselecting */
104         if (test) {
105                 for (ale= anim_data.first; ale; ale= ale->next) {
106                         if (ale->type == ANIMTYPE_GPLAYER) {
107                                 if (is_gplayer_frame_selected(ale->data)) {
108                                         sel= SELECT_SUBTRACT;
109                                         break;
110                                 }
111                         }
112                         else {
113                                 if (ANIM_fcurve_keyframes_loop(&ked, ale->key_data, NULL, test_cb, NULL)) {
114                                         sel= SELECT_SUBTRACT;
115                                         break;
116                                 }
117                         }
118                 }
119         }
120         
121         /* convert sel to selectmode, and use that to get editor */
122         sel_cb= ANIM_editkeyframes_select(sel);
123         
124         /* Now set the flags */
125         for (ale= anim_data.first; ale; ale= ale->next) {
126                 if (ale->type == ANIMTYPE_GPLAYER)
127                         set_gplayer_frame_selection(ale->data, sel);
128                 else
129                         ANIM_fcurve_keyframes_loop(&ked, ale->key_data, NULL, sel_cb, NULL); 
130         }
131         
132         /* Cleanup */
133         BLI_freelistN(&anim_data);
134 }
135
136 /* ------------------- */
137
138 static int actkeys_deselectall_exec(bContext *C, wmOperator *op)
139 {
140         bAnimContext ac;
141         
142         /* get editor data */
143         if (ANIM_animdata_get_context(C, &ac) == 0)
144                 return OPERATOR_CANCELLED;
145                 
146         /* 'standard' behaviour - check if selected, then apply relevant selection */
147         if (RNA_boolean_get(op->ptr, "invert"))
148                 deselect_action_keys(&ac, 0, SELECT_INVERT);
149         else
150                 deselect_action_keys(&ac, 1, SELECT_ADD);
151         
152         /* set notifier that keyframe selection have changed */
153         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_SELECTED, NULL);
154         
155         return OPERATOR_FINISHED;
156 }
157  
158 void ACTION_OT_select_all_toggle (wmOperatorType *ot)
159 {
160         /* identifiers */
161         ot->name= "Select All";
162         ot->idname= "ACTION_OT_select_all_toggle";
163         ot->description= "Toggle selection of all keyframes";
164         
165         /* api callbacks */
166         ot->exec= actkeys_deselectall_exec;
167         ot->poll= ED_operator_action_active;
168         
169         /* flags */
170         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
171         
172         /* props */
173         ot->prop= RNA_def_boolean(ot->srna, "invert", 0, "Invert", "");
174 }
175
176 /* ******************** Border Select Operator **************************** */
177 /* This operator currently works in one of three ways:
178  *      -> BKEY         - 1) all keyframes within region are selected (ACTKEYS_BORDERSEL_ALLKEYS)
179  *      -> ALT-BKEY - depending on which axis of the region was larger...
180  *              -> 2) x-axis, so select all frames within frame range (ACTKEYS_BORDERSEL_FRAMERANGE)
181  *              -> 3) y-axis, so select all frames within channels that region included (ACTKEYS_BORDERSEL_CHANNELS)
182  */
183
184 /* defines for borderselect mode */
185 enum {
186         ACTKEYS_BORDERSEL_ALLKEYS       = 0,
187         ACTKEYS_BORDERSEL_FRAMERANGE,
188         ACTKEYS_BORDERSEL_CHANNELS,
189 } eActKeys_BorderSelect_Mode;
190
191
192 static void borderselect_action (bAnimContext *ac, rcti rect, short mode, short selectmode)
193 {
194         ListBase anim_data = {NULL, NULL};
195         bAnimListElem *ale;
196         int filter, filterflag;
197         
198         KeyframeEditData ked;
199         KeyframeEditFunc ok_cb, select_cb;
200         View2D *v2d= &ac->ar->v2d;
201         rctf rectf;
202         float ymin=0, ymax=(float)(-ACHANNEL_HEIGHT_HALF);
203         
204         /* convert mouse coordinates to frame ranges and channel coordinates corrected for view pan/zoom */
205         UI_view2d_region_to_view(v2d, rect.xmin, rect.ymin+2, &rectf.xmin, &rectf.ymin);
206         UI_view2d_region_to_view(v2d, rect.xmax, rect.ymax-2, &rectf.xmax, &rectf.ymax);
207         
208         /* filter data */
209         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CHANNELS | ANIMFILTER_NODUPLIS);
210         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
211         
212         /* get filtering flag for dopesheet data (if applicable) */
213         if (ac->datatype == ANIMCONT_DOPESHEET) {
214                 bDopeSheet *ads= (bDopeSheet *)ac->data;
215                 filterflag= ads->filterflag;
216         }
217         else
218                 filterflag= 0;
219         
220         /* get beztriple editing/validation funcs  */
221         select_cb= ANIM_editkeyframes_select(selectmode);
222         
223         if (ELEM(mode, ACTKEYS_BORDERSEL_FRAMERANGE, ACTKEYS_BORDERSEL_ALLKEYS))
224                 ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAMERANGE);
225         else
226                 ok_cb= NULL;
227                 
228         /* init editing data */
229         memset(&ked, 0, sizeof(KeyframeEditData));
230         
231         /* loop over data, doing border select */
232         for (ale= anim_data.first; ale; ale= ale->next) {
233                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
234                 
235                 /* get new vertical minimum extent of channel */
236                 ymin= ymax - ACHANNEL_STEP;
237                 
238                 /* set horizontal range (if applicable) */
239                 if (ELEM(mode, ACTKEYS_BORDERSEL_FRAMERANGE, ACTKEYS_BORDERSEL_ALLKEYS)) {
240                         /* if channel is mapped in NLA, apply correction */
241                         if (adt) {
242                                 ked.f1= BKE_nla_tweakedit_remap(adt, rectf.xmin, NLATIME_CONVERT_UNMAP);
243                                 ked.f2= BKE_nla_tweakedit_remap(adt, rectf.xmax, NLATIME_CONVERT_UNMAP);
244                         }
245                         else {
246                                 ked.f1= rectf.xmin;
247                                 ked.f2= rectf.xmax;
248                         }
249                 }
250                 
251                 /* perform vertical suitability check (if applicable) */
252                 if ( (mode == ACTKEYS_BORDERSEL_FRAMERANGE) || 
253                         !((ymax < rectf.ymin) || (ymin > rectf.ymax)) )
254                 {
255                         /* loop over data selecting */
256                         if (ale->type == ANIMTYPE_GPLAYER)
257                                 borderselect_gplayer_frames(ale->data, rectf.xmin, rectf.xmax, selectmode);
258                         else
259                                 ANIM_animchannel_keyframes_loop(&ked, ale, ok_cb, select_cb, NULL, filterflag);
260                 }
261                 
262                 /* set minimum extent to be the maximum of the next channel */
263                 ymax=ymin;
264         }
265         
266         /* cleanup */
267         BLI_freelistN(&anim_data);
268 }
269
270 /* ------------------- */
271
272 static int actkeys_borderselect_exec(bContext *C, wmOperator *op)
273 {
274         bAnimContext ac;
275         rcti rect;
276         short mode=0, selectmode=0;
277         int gesture_mode;
278         
279         /* get editor data */
280         if (ANIM_animdata_get_context(C, &ac) == 0)
281                 return OPERATOR_CANCELLED;
282         
283         /* get settings from operator */
284         rect.xmin= RNA_int_get(op->ptr, "xmin");
285         rect.ymin= RNA_int_get(op->ptr, "ymin");
286         rect.xmax= RNA_int_get(op->ptr, "xmax");
287         rect.ymax= RNA_int_get(op->ptr, "ymax");
288                 
289         gesture_mode= RNA_int_get(op->ptr, "gesture_mode");
290         if (gesture_mode == GESTURE_MODAL_SELECT)
291                 selectmode = SELECT_ADD;
292         else
293                 selectmode = SELECT_SUBTRACT;
294         
295         /* selection 'mode' depends on whether borderselect region only matters on one axis */
296         if (RNA_boolean_get(op->ptr, "axis_range")) {
297                 /* mode depends on which axis of the range is larger to determine which axis to use 
298                  *      - checking this in region-space is fine, as it's fundamentally still going to be a different rect size
299                  *      - the frame-range select option is favoured over the channel one (x over y), as frame-range one is often
300                  *        used for tweaking timing when "blocking", while channels is not that useful...
301                  */
302                 if ((rect.xmax - rect.xmin) >= (rect.ymax - rect.ymin))
303                         mode= ACTKEYS_BORDERSEL_FRAMERANGE;
304                 else
305                         mode= ACTKEYS_BORDERSEL_CHANNELS;
306         }
307         else 
308                 mode= ACTKEYS_BORDERSEL_ALLKEYS;
309         
310         /* apply borderselect action */
311         borderselect_action(&ac, rect, mode, selectmode);
312         
313         /* set notifier that keyframe selection have changed */
314         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_SELECTED, NULL);
315         
316         return OPERATOR_FINISHED;
317
318
319 void ACTION_OT_select_border(wmOperatorType *ot)
320 {
321         /* identifiers */
322         ot->name= "Border Select";
323         ot->idname= "ACTION_OT_select_border";
324         ot->description= "Select all keyframes within the specified region";
325         
326         /* api callbacks */
327         ot->invoke= WM_border_select_invoke;
328         ot->exec= actkeys_borderselect_exec;
329         ot->modal= WM_border_select_modal;
330         
331         ot->poll= ED_operator_action_active;
332         
333         /* flags */
334         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
335         
336         /* rna */
337         WM_operator_properties_gesture_border(ot, FALSE);
338         
339         ot->prop= RNA_def_boolean(ot->srna, "axis_range", 0, "Axis Range", "");
340 }
341
342 /* ******************** Column Select Operator **************************** */
343 /* This operator works in one of four ways:
344  *      - 1) select all keyframes in the same frame as a selected one  (KKEY)
345  *      - 2) select all keyframes in the same frame as the current frame marker (CTRL-KKEY)
346  *      - 3) select all keyframes in the same frame as a selected markers (SHIFT-KKEY)
347  *      - 4) select all keyframes that occur between selected markers (ALT-KKEY)
348  */
349
350 /* defines for column-select mode */
351 static EnumPropertyItem prop_column_select_types[] = {
352         {ACTKEYS_COLUMNSEL_KEYS, "KEYS", 0, "On Selected Keyframes", ""},
353         {ACTKEYS_COLUMNSEL_CFRA, "CFRA", 0, "On Current Frame", ""},
354         {ACTKEYS_COLUMNSEL_MARKERS_COLUMN, "MARKERS_COLUMN", 0, "On Selected Markers", ""},
355         {ACTKEYS_COLUMNSEL_MARKERS_BETWEEN, "MARKERS_BETWEEN", 0, "Between Min/Max Selected Markers", ""},
356         {0, NULL, 0, NULL, NULL}
357 };
358
359 /* ------------------- */ 
360
361 /* Selects all visible keyframes between the specified markers */
362 static void markers_selectkeys_between (bAnimContext *ac)
363 {
364         ListBase anim_data = {NULL, NULL};
365         bAnimListElem *ale;
366         int filter;
367         
368         KeyframeEditFunc ok_cb, select_cb;
369         KeyframeEditData ked= {{0}};
370         float min, max;
371         
372         /* get extreme markers */
373         ED_markers_get_minmax(ac->markers, 1, &min, &max);
374         min -= 0.5f;
375         max += 0.5f;
376         
377         /* get editing funcs + data */
378         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAMERANGE);
379         select_cb= ANIM_editkeyframes_select(SELECT_ADD);
380
381         ked.f1= min; 
382         ked.f2= max;
383         
384         /* filter data */
385         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVESONLY | ANIMFILTER_NODUPLIS);
386         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
387         
388         /* select keys in-between */
389         for (ale= anim_data.first; ale; ale= ale->next) {
390                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
391                 
392                 if (adt) {
393                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 0, 1);
394                         ANIM_fcurve_keyframes_loop(&ked, ale->key_data, ok_cb, select_cb, NULL);
395                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 1, 1);
396                 }
397                 else if (ale->type == ANIMTYPE_GPLAYER) {
398                         borderselect_gplayer_frames(ale->data, min, max, SELECT_ADD);
399                 }
400                 else {
401                         ANIM_fcurve_keyframes_loop(&ked, ale->key_data, ok_cb, select_cb, NULL);
402                 }
403         }
404         
405         /* Cleanup */
406         BLI_freelistN(&anim_data);
407 }
408
409
410 /* Selects all visible keyframes in the same frames as the specified elements */
411 static void columnselect_action_keys (bAnimContext *ac, short mode)
412 {
413         ListBase anim_data= {NULL, NULL};
414         bAnimListElem *ale;
415         int filter;
416         
417         Scene *scene= ac->scene;
418         CfraElem *ce;
419         KeyframeEditFunc select_cb, ok_cb;
420         KeyframeEditData ked= {{0}};
421         
422         /* initialise keyframe editing data */
423         
424         /* build list of columns */
425         switch (mode) {
426                 case ACTKEYS_COLUMNSEL_KEYS: /* list of selected keys */
427                         if (ac->datatype == ANIMCONT_GPENCIL) {
428                                 filter= (ANIMFILTER_VISIBLE);
429                                 ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
430                                 
431                                 for (ale= anim_data.first; ale; ale= ale->next)
432                                         gplayer_make_cfra_list(ale->data, &ked.list, 1);
433                         }
434                         else {
435                                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVESONLY);
436                                 ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
437                                 
438                                 for (ale= anim_data.first; ale; ale= ale->next)
439                                         ANIM_fcurve_keyframes_loop(&ked, ale->key_data, NULL, bezt_to_cfraelem, NULL);
440                         }
441                         BLI_freelistN(&anim_data);
442                         break;
443                         
444                 case ACTKEYS_COLUMNSEL_CFRA: /* current frame */
445                         /* make a single CfraElem for storing this */
446                         ce= MEM_callocN(sizeof(CfraElem), "cfraElem");
447                         BLI_addtail(&ked.list, ce);
448                         
449                         ce->cfra= (float)CFRA;
450                         break;
451                         
452                 case ACTKEYS_COLUMNSEL_MARKERS_COLUMN: /* list of selected markers */
453                         ED_markers_make_cfra_list(ac->markers, &ked.list, 1);
454                         break;
455                         
456                 default: /* invalid option */
457                         return;
458         }
459         
460         /* set up BezTriple edit callbacks */
461         select_cb= ANIM_editkeyframes_select(SELECT_ADD);
462         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAME);
463         
464         /* loop through all of the keys and select additional keyframes
465          * based on the keys found to be selected above
466          */
467         if (ac->datatype == ANIMCONT_GPENCIL)
468                 filter= (ANIMFILTER_VISIBLE);
469         else
470                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVESONLY);
471         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
472         
473         for (ale= anim_data.first; ale; ale= ale->next) {
474                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
475                 
476                 /* loop over cfraelems (stored in the KeyframeEditData->list)
477                  *      - we need to do this here, as we can apply fewer NLA-mapping conversions
478                  */
479                 for (ce= ked.list.first; ce; ce= ce->next) {
480                         /* set frame for validation callback to refer to */
481                         if (adt)
482                                 ked.f1= BKE_nla_tweakedit_remap(adt, ce->cfra, NLATIME_CONVERT_UNMAP);
483                         else
484                                 ked.f1= ce->cfra;
485                         
486                         /* select elements with frame number matching cfraelem */
487                         if (ale->type == ANIMTYPE_GPLAYER)
488                                 select_gpencil_frame(ale->data, ce->cfra, SELECT_ADD);
489                         else
490                                 ANIM_fcurve_keyframes_loop(&ked, ale->key_data, ok_cb, select_cb, NULL);
491                 }
492         }
493         
494         /* free elements */
495         BLI_freelistN(&ked.list);
496         BLI_freelistN(&anim_data);
497 }
498
499 /* ------------------- */
500
501 static int actkeys_columnselect_exec(bContext *C, wmOperator *op)
502 {
503         bAnimContext ac;
504         short mode;
505         
506         /* get editor data */
507         if (ANIM_animdata_get_context(C, &ac) == 0)
508                 return OPERATOR_CANCELLED;
509                 
510         /* action to take depends on the mode */
511         mode= RNA_enum_get(op->ptr, "mode");
512         
513         if (mode == ACTKEYS_COLUMNSEL_MARKERS_BETWEEN)
514                 markers_selectkeys_between(&ac);
515         else
516                 columnselect_action_keys(&ac, mode);
517         
518         /* set notifier that keyframe selection have changed */
519         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_SELECTED, NULL);
520         
521         return OPERATOR_FINISHED;
522 }
523  
524 void ACTION_OT_select_column (wmOperatorType *ot)
525 {
526         /* identifiers */
527         ot->name= "Select All";
528         ot->idname= "ACTION_OT_select_column";
529         ot->description= "Select all keyframes on the specified frame(s)";
530         
531         /* api callbacks */
532         ot->exec= actkeys_columnselect_exec;
533         ot->poll= ED_operator_action_active;
534         
535         /* flags */
536         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
537         
538         /* props */
539         ot->prop= RNA_def_enum(ot->srna, "mode", prop_column_select_types, 0, "Mode", "");
540 }
541
542 /* ******************** Select Linked Operator *********************** */
543
544 static int actkeys_select_linked_exec (bContext *C, wmOperator *UNUSED(op))
545 {
546         bAnimContext ac;
547         
548         ListBase anim_data= {NULL, NULL};
549         bAnimListElem *ale;
550         int filter;
551         
552         KeyframeEditFunc ok_cb = ANIM_editkeyframes_ok(BEZT_OK_SELECTED);
553         KeyframeEditFunc sel_cb = ANIM_editkeyframes_select(SELECT_ADD);
554         
555         /* get editor data */
556         if (ANIM_animdata_get_context(C, &ac) == 0)
557                 return OPERATOR_CANCELLED;
558         
559         /* loop through all of the keys and select additional keyframes based on these */
560         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_CURVESONLY | ANIMFILTER_NODUPLIS);
561         ANIM_animdata_filter(&ac, &anim_data, filter, ac.data, ac.datatype);
562         
563         for (ale= anim_data.first; ale; ale= ale->next) {
564                 FCurve *fcu= (FCurve *)ale->key_data;
565                 
566                 /* check if anything selected? */
567                 if (ANIM_fcurve_keyframes_loop(NULL, fcu, NULL, ok_cb, NULL)) {
568                         /* select every keyframe in this curve then */
569                         ANIM_fcurve_keyframes_loop(NULL, fcu, NULL, sel_cb, NULL);
570                 }
571         }
572         
573         /* Cleanup */
574         BLI_freelistN(&anim_data);
575         
576         /* set notifier that keyframe selection has changed */
577         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_SELECTED, NULL);
578         
579         return OPERATOR_FINISHED;
580 }
581
582 void ACTION_OT_select_linked (wmOperatorType *ot)
583 {
584         /* identifiers */
585         ot->name = "Select Linked";
586         ot->idname= "ACTION_OT_select_linked";
587         ot->description = "Select keyframes occurring the same F-Curves as selected ones";
588         
589         /* api callbacks */
590         ot->exec= actkeys_select_linked_exec;
591         ot->poll= ED_operator_action_active;
592         
593         /* flags */
594         ot->flag= OPTYPE_REGISTER/*|OPTYPE_UNDO*/;
595 }
596
597 /* ******************** Select More/Less Operators *********************** */
598
599 /* Common code to perform selection */
600 static void select_moreless_action_keys (bAnimContext *ac, short mode)
601 {
602         ListBase anim_data= {NULL, NULL};
603         bAnimListElem *ale;
604         int filter;
605         
606         KeyframeEditData ked= {{0}};
607         KeyframeEditFunc build_cb;
608         
609         
610         /* init selmap building data */
611         build_cb= ANIM_editkeyframes_buildselmap(mode);
612         
613         /* loop through all of the keys and select additional keyframes based on these */
614         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVESONLY | ANIMFILTER_NODUPLIS);
615         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
616         
617         for (ale= anim_data.first; ale; ale= ale->next) {
618                 FCurve *fcu= (FCurve *)ale->key_data;
619                 
620                 /* only continue if F-Curve has keyframes */
621                 if (fcu->bezt == NULL)
622                         continue;
623                 
624                 /* build up map of whether F-Curve's keyframes should be selected or not */
625                 ked.data= MEM_callocN(fcu->totvert, "selmap actEdit more");
626                 ANIM_fcurve_keyframes_loop(&ked, fcu, NULL, build_cb, NULL);
627                 
628                 /* based on this map, adjust the selection status of the keyframes */
629                 ANIM_fcurve_keyframes_loop(&ked, fcu, NULL, bezt_selmap_flush, NULL);
630                 
631                 /* free the selmap used here */
632                 MEM_freeN(ked.data);
633                 ked.data= NULL;
634         }
635         
636         /* Cleanup */
637         BLI_freelistN(&anim_data);
638 }
639
640 /* ----------------- */
641
642 static int actkeys_select_more_exec (bContext *C, wmOperator *UNUSED(op))
643 {
644         bAnimContext ac;
645         
646         /* get editor data */
647         if (ANIM_animdata_get_context(C, &ac) == 0)
648                 return OPERATOR_CANCELLED;
649         
650         /* perform select changes */
651         select_moreless_action_keys(&ac, SELMAP_MORE);
652         
653         /* set notifier that keyframe selection has changed */
654         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_SELECTED, NULL);
655         
656         return OPERATOR_FINISHED;
657 }
658
659 void ACTION_OT_select_more (wmOperatorType *ot)
660 {
661         /* identifiers */
662         ot->name = "Select More";
663         ot->idname= "ACTION_OT_select_more";
664         ot->description = "Select keyframes beside already selected ones";
665         
666         /* api callbacks */
667         ot->exec= actkeys_select_more_exec;
668         ot->poll= ED_operator_action_active;
669         
670         /* flags */
671         ot->flag= OPTYPE_REGISTER/*|OPTYPE_UNDO*/;
672 }
673
674 /* ----------------- */
675
676 static int actkeys_select_less_exec (bContext *C, wmOperator *UNUSED(op))
677 {
678         bAnimContext ac;
679         
680         /* get editor data */
681         if (ANIM_animdata_get_context(C, &ac) == 0)
682                 return OPERATOR_CANCELLED;
683         
684         /* perform select changes */
685         select_moreless_action_keys(&ac, SELMAP_LESS);
686         
687         /* set notifier that keyframe selection has changed */
688         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_SELECTED, NULL);
689         
690         return OPERATOR_FINISHED;
691 }
692
693 void ACTION_OT_select_less (wmOperatorType *ot)
694 {
695         /* identifiers */
696         ot->name = "Select Less";
697         ot->idname= "ACTION_OT_select_less";
698         ot->description = "Deselect keyframes on ends of selection islands";
699         
700         /* api callbacks */
701         ot->exec= actkeys_select_less_exec;
702         ot->poll= ED_operator_action_active;
703         
704         /* flags */
705         ot->flag= OPTYPE_REGISTER/*|OPTYPE_UNDO*/;
706 }
707
708 /* ******************** Select Left/Right Operator ************************* */
709 /* Select keyframes left/right of the current frame indicator */
710
711 /* defines for left-right select tool */
712 static EnumPropertyItem prop_actkeys_leftright_select_types[] = {
713         {ACTKEYS_LRSEL_TEST, "CHECK", 0, "Check if Select Left or Right", ""},
714         {ACTKEYS_LRSEL_LEFT, "LEFT", 0, "Before current frame", ""},
715         {ACTKEYS_LRSEL_RIGHT, "RIGHT", 0, "After current frame", ""},
716         {0, NULL, 0, NULL, NULL}
717 };
718
719 /* --------------------------------- */
720
721 static void actkeys_select_leftright (bAnimContext *ac, short leftright, short select_mode)
722 {
723         ListBase anim_data = {NULL, NULL};
724         bAnimListElem *ale;
725         int filter;
726         
727         KeyframeEditFunc ok_cb, select_cb;
728         KeyframeEditData ked= {{0}};
729         Scene *scene= ac->scene;
730         
731         /* if select mode is replace, deselect all keyframes (and channels) first */
732         if (select_mode==SELECT_REPLACE) {
733                 select_mode= SELECT_ADD;
734                 
735                 /* deselect all other channels and keyframes */
736                 ANIM_deselect_anim_channels(ac, ac->data, ac->datatype, 0, ACHANNEL_SETFLAG_CLEAR);
737                 deselect_action_keys(ac, 0, SELECT_SUBTRACT);
738         }
739         
740         /* set callbacks and editing data */
741         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAMERANGE);
742         select_cb= ANIM_editkeyframes_select(select_mode);
743         
744         /* NOTE: we used to include FRAME_CLICK_THRESH in the tolerance for the current frame, 
745          * but this has now been remove, since it is not so relevant anymore
746          */
747         if (leftright == ACTKEYS_LRSEL_LEFT) {
748                 ked.f1 = MINAFRAMEF;
749                 ked.f2 = (float)(CFRA);
750         } 
751         else {
752                 ked.f1 = (float)(CFRA);
753                 ked.f2 = MAXFRAMEF;
754         }
755         
756         /* filter data */
757         if (ac->datatype == ANIMCONT_GPENCIL)
758                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_NODUPLIS);
759         else
760                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVESONLY | ANIMFILTER_NODUPLIS);
761         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
762                 
763         /* select keys */
764         for (ale= anim_data.first; ale; ale= ale->next) {
765                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
766                 
767                 if (adt) {
768                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 0, 1);
769                         ANIM_fcurve_keyframes_loop(&ked, ale->key_data, ok_cb, select_cb, NULL);
770                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 1, 1);
771                 }
772                 else if (ale->type == ANIMTYPE_GPLAYER) 
773                         borderselect_gplayer_frames(ale->data, ked.f1, ked.f2, select_mode);
774                 else
775                         ANIM_fcurve_keyframes_loop(&ked, ale->key_data, ok_cb, select_cb, NULL);
776         }
777         
778         /* Sync marker support */
779         if (select_mode==SELECT_ADD) {
780                 SpaceAction *saction= ac->sa->spacedata.first;
781                 
782                 if ((saction) && (saction->flag & SACTION_MARKERS_MOVE)) {
783                         ListBase *markers = ED_animcontext_get_markers(ac);
784                         TimeMarker *marker;
785                         
786                         for (marker= markers->first; marker; marker= marker->next) {
787                                 if(     ((leftright == ACTKEYS_LRSEL_LEFT) && (marker->frame < CFRA)) ||
788                                         ((leftright == ACTKEYS_LRSEL_RIGHT) && (marker->frame >= CFRA)) ) 
789                                 {
790                                         marker->flag |= SELECT;
791                                 }
792                                 else {
793                                         marker->flag &= ~SELECT;
794                                 }
795                         }
796                 }
797         }
798
799         /* Cleanup */
800         BLI_freelistN(&anim_data);
801 }
802
803 /* ----------------- */
804
805 static int actkeys_select_leftright_exec (bContext *C, wmOperator *op)
806 {
807         bAnimContext ac;
808         short leftright = RNA_enum_get(op->ptr, "mode");
809         short selectmode;
810         
811         /* get editor data */
812         if (ANIM_animdata_get_context(C, &ac) == 0)
813                 return OPERATOR_CANCELLED;
814         
815         /* select mode is either replace (deselect all, then add) or add/extend */
816         if (RNA_boolean_get(op->ptr, "extend"))
817                 selectmode= SELECT_INVERT;
818         else
819                 selectmode= SELECT_REPLACE;
820                 
821         /* if "test" mode is set, we don't have any info to set this with */
822         if (leftright == ACTKEYS_LRSEL_TEST)
823                 return OPERATOR_CANCELLED;
824         
825         /* do the selecting now */
826         actkeys_select_leftright(&ac, leftright, selectmode);
827         
828         /* set notifier that keyframe selection (and channels too) have changed */
829         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|ND_ANIMCHAN|NA_SELECTED, NULL);
830         
831         return OPERATOR_FINISHED;
832 }
833
834 static int actkeys_select_leftright_invoke (bContext *C, wmOperator *op, wmEvent *event)
835 {
836         bAnimContext ac;
837         short leftright = RNA_enum_get(op->ptr, "mode");
838         
839         /* get editor data */
840         if (ANIM_animdata_get_context(C, &ac) == 0)
841                 return OPERATOR_CANCELLED;
842                 
843         /* handle mode-based testing */
844         if (leftright == ACTKEYS_LRSEL_TEST) {
845                 Scene *scene= ac.scene;
846                 ARegion *ar= ac.ar;
847                 View2D *v2d= &ar->v2d;
848                 
849                 short mval[2];
850                 float x;
851                 
852                 /* get mouse coordinates (in region coordinates) */
853                 mval[0]= (event->x - ar->winrct.xmin);
854                 mval[1]= (event->y - ar->winrct.ymin);
855                 
856                 /* determine which side of the current frame mouse is on */
857                 UI_view2d_region_to_view(v2d, mval[0], mval[1], &x, NULL);
858                 if (x < CFRA)
859                         RNA_int_set(op->ptr, "mode", ACTKEYS_LRSEL_LEFT);
860                 else    
861                         RNA_int_set(op->ptr, "mode", ACTKEYS_LRSEL_RIGHT);
862         }
863         
864         /* perform selection */
865         return actkeys_select_leftright_exec(C, op);
866 }
867
868 void ACTION_OT_select_leftright (wmOperatorType *ot)
869 {
870         /* identifiers */
871         ot->name= "Select Left/Right";
872         ot->idname= "ACTION_OT_select_leftright";
873         ot->description= "Select keyframes to the left or the right of the current keyframe ";
874         
875         /* api callbacks  */
876         ot->invoke= actkeys_select_leftright_invoke;
877         ot->exec= actkeys_select_leftright_exec;
878         ot->poll= ED_operator_action_active;
879         
880         /* flags */
881         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
882         
883         /* id-props */
884         ot->prop= RNA_def_enum(ot->srna, "mode", prop_actkeys_leftright_select_types, ACTKEYS_LRSEL_TEST, "Mode", "");
885         RNA_def_boolean(ot->srna, "extend", 0, "Extend Select", "");
886 }
887
888 /* ******************** Mouse-Click Select Operator *********************** */
889 /* This operator works in one of three ways:
890  *      - 1) keyframe under mouse - no special modifiers
891  *      - 2) all keyframes on the same side of current frame indicator as mouse - ALT modifier
892  *      - 3) column select all keyframes in frame under mouse - CTRL modifier
893  *
894  * In addition to these basic options, the SHIFT modifier can be used to toggle the 
895  * selection mode between replacing the selection (without) and inverting the selection (with).
896  */
897
898 /* sensitivity factor for frame-selections */
899 #define FRAME_CLICK_THRESH              0.1f
900
901 /* ------------------- */
902  
903 /* option 1) select keyframe directly under mouse */
904 static void actkeys_mselect_single (bAnimContext *ac, bAnimListElem *ale, short select_mode, float selx)
905 {
906         bDopeSheet *ads= (ac->datatype == ANIMCONT_DOPESHEET) ? ac->data : NULL;
907         int ds_filter = ((ads) ? (ads->filterflag) : (0));
908         
909         KeyframeEditData ked= {{0}};
910         KeyframeEditFunc select_cb, ok_cb;
911         
912         /* get functions for selecting keyframes */
913         select_cb= ANIM_editkeyframes_select(select_mode);
914         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAME);
915         ked.f1= selx;
916         
917         /* select the nominated keyframe on the given frame */
918         if (ale->type == ANIMTYPE_GPLAYER)
919                 select_gpencil_frame(ale->data, selx, select_mode);
920         else
921                 ANIM_animchannel_keyframes_loop(&ked, ale, ok_cb, select_cb, NULL, ds_filter);
922 }
923
924 /* Option 2) Selects all the keyframes on either side of the current frame (depends on which side the mouse is on) */
925 /* (see actkeys_select_leftright) */
926
927 /* Option 3) Selects all visible keyframes in the same frame as the mouse click */
928 static void actkeys_mselect_column(bAnimContext *ac, short select_mode, float selx)
929 {
930         ListBase anim_data= {NULL, NULL};
931         bAnimListElem *ale;
932         int filter;
933         
934         KeyframeEditFunc select_cb, ok_cb;
935         KeyframeEditData ked= {{0}};
936         
937         /* initialise keyframe editing data */
938         
939         /* set up BezTriple edit callbacks */
940         select_cb= ANIM_editkeyframes_select(select_mode);
941         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAME);
942         
943         /* loop through all of the keys and select additional keyframes
944          * based on the keys found to be selected above
945          */
946         if (ac->datatype == ANIMCONT_GPENCIL)
947                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_NODUPLIS);
948         else
949                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVESONLY | ANIMFILTER_NODUPLIS);
950         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
951         
952         for (ale= anim_data.first; ale; ale= ale->next) {
953                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
954                 
955                 /* set frame for validation callback to refer to */
956                 if (adt)
957                         ked.f1= BKE_nla_tweakedit_remap(adt, selx, NLATIME_CONVERT_UNMAP);
958                 else
959                         ked.f1= selx;
960                 
961                 /* select elements with frame number matching cfra */
962                 if (ale->type == ANIMTYPE_GPLAYER)
963                         select_gpencil_frame(ale->key_data, selx, select_mode);
964                 else 
965                         ANIM_fcurve_keyframes_loop(&ked, ale->key_data, ok_cb, select_cb, NULL);
966         }
967         
968         /* free elements */
969         BLI_freelistN(&ked.list);
970         BLI_freelistN(&anim_data);
971 }
972  
973 /* ------------------- */
974
975 static void mouse_action_keys (bAnimContext *ac, int mval[2], short select_mode, short column)
976 {
977         ListBase anim_data = {NULL, NULL};
978         DLRBT_Tree anim_keys;
979         bAnimListElem *ale;
980         int filter;
981         
982         View2D *v2d= &ac->ar->v2d;
983         bDopeSheet *ads = NULL;
984         int channel_index;
985         short found = 0;
986         float selx = 0.0f;
987         float x, y;
988         rctf rectf;
989         
990         /* get dopesheet info */
991         if (ac->datatype == ANIMCONT_DOPESHEET)
992                 ads= ac->data;
993         
994         /* use View2D to determine the index of the channel (i.e a row in the list) where keyframe was */
995         UI_view2d_region_to_view(v2d, mval[0], mval[1], &x, &y);
996         UI_view2d_listview_view_to_cell(v2d, 0, ACHANNEL_STEP, 0, (float)ACHANNEL_HEIGHT_HALF, x, y, NULL, &channel_index);
997         
998         /* x-range to check is +/- 7 (in screen/region-space) on either side of mouse click (size of keyframe icon) */
999         UI_view2d_region_to_view(v2d, mval[0]-7, mval[1], &rectf.xmin, &rectf.ymin);
1000         UI_view2d_region_to_view(v2d, mval[0]+7, mval[1], &rectf.xmax, &rectf.ymax);
1001         
1002         /* filter data */
1003         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CHANNELS);
1004         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
1005         
1006         /* try to get channel */
1007         ale= BLI_findlink(&anim_data, channel_index);
1008         if (ale == NULL) {
1009                 /* channel not found */
1010                 printf("Error: animation channel (index = %d) not found in mouse_action_keys() \n", channel_index);
1011                 BLI_freelistN(&anim_data);
1012                 return;
1013         }
1014         else {
1015                 /* found match - must return here... */
1016                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
1017                 ActKeyColumn *ak, *akn=NULL;
1018                 
1019                 /* make list of keyframes */
1020                 BLI_dlrbTree_init(&anim_keys);
1021                 
1022                 if (ale->key_data) {
1023                         switch (ale->datatype) {
1024                                 case ALE_SCE:
1025                                 {
1026                                         Scene *scene= (Scene *)ale->key_data;
1027                                         scene_to_keylist(ads, scene, &anim_keys, NULL);
1028                                 }
1029                                         break;
1030                                 case ALE_OB:
1031                                 {
1032                                         Object *ob= (Object *)ale->key_data;
1033                                         ob_to_keylist(ads, ob, &anim_keys, NULL);
1034                                 }
1035                                         break;
1036                                 case ALE_ACT:
1037                                 {
1038                                         bAction *act= (bAction *)ale->key_data;
1039                                         action_to_keylist(adt, act, &anim_keys, NULL);
1040                                 }
1041                                         break;
1042                                 case ALE_FCURVE:
1043                                 {
1044                                         FCurve *fcu= (FCurve *)ale->key_data;
1045                                         fcurve_to_keylist(adt, fcu, &anim_keys, NULL);
1046                                 }
1047                                         break;
1048                         }
1049                 }
1050                 else if (ale->type == ANIMTYPE_SUMMARY) {
1051                         /* dopesheet summary covers everything */
1052                         summary_to_keylist(ac, &anim_keys, NULL);
1053                 }
1054                 else if (ale->type == ANIMTYPE_GROUP) { 
1055                         // TODO: why don't we just give groups key_data too?
1056                         bActionGroup *agrp= (bActionGroup *)ale->data;
1057                         agroup_to_keylist(adt, agrp, &anim_keys, NULL);
1058                 }
1059                 else if (ale->type == ANIMTYPE_GPLAYER) {
1060                         // TODO: why don't we just give gplayers key_data too?
1061                         bGPDlayer *gpl = (bGPDlayer *)ale->data;
1062                         gpl_to_keylist(ads, gpl, &anim_keys);
1063                 }
1064                 
1065                 /* start from keyframe at root of BST, traversing until we find one within the range that was clicked on */
1066                 for (ak= anim_keys.root; ak; ak= akn) {
1067                         if (IN_RANGE(ak->cfra, rectf.xmin, rectf.xmax)) {
1068                                 /* set the frame to use, and apply inverse-correction for NLA-mapping 
1069                                  * so that the frame will get selected by the selection functions without
1070                                  * requiring to map each frame once again...
1071                                  */
1072                                 selx= BKE_nla_tweakedit_remap(adt, ak->cfra, NLATIME_CONVERT_UNMAP);
1073                                 found= 1;
1074                                 break;
1075                         }
1076                         else if (ak->cfra < rectf.xmin)
1077                                 akn= ak->right;
1078                         else
1079                                 akn= ak->left;
1080                 }
1081                 
1082                 /* remove active channel from list of channels for separate treatment (since it's needed later on) */
1083                 BLI_remlink(&anim_data, ale);
1084                 
1085                 /* cleanup temporary lists */
1086                 BLI_dlrbTree_free(&anim_keys);
1087                 
1088                 /* free list of channels, since it's not used anymore */
1089                 BLI_freelistN(&anim_data);
1090         }
1091         
1092         /* for replacing selection, firstly need to clear existing selection */
1093         if (select_mode == SELECT_REPLACE) {
1094                 /* reset selection mode for next steps */
1095                 select_mode = SELECT_ADD;
1096                 
1097                 /* deselect all keyframes */
1098                 deselect_action_keys(ac, 0, SELECT_SUBTRACT);
1099                 
1100                 /* highlight channel clicked on */
1101                 if (ELEM(ac->datatype, ANIMCONT_ACTION, ANIMCONT_DOPESHEET)) {
1102                         /* deselect all other channels first */
1103                         ANIM_deselect_anim_channels(ac, ac->data, ac->datatype, 0, ACHANNEL_SETFLAG_CLEAR);
1104                         
1105                         /* Highlight Action-Group or F-Curve? */
1106                         if (ale && ale->data) {
1107                                 if (ale->type == ANIMTYPE_GROUP) {
1108                                         bActionGroup *agrp= ale->data;
1109                                         
1110                                         agrp->flag |= AGRP_SELECTED;
1111                                         ANIM_set_active_channel(ac, ac->data, ac->datatype, filter, agrp, ANIMTYPE_GROUP);
1112                                 }       
1113                                 else if (ale->type == ANIMTYPE_FCURVE) {
1114                                         FCurve *fcu= ale->data;
1115                                         
1116                                         fcu->flag |= FCURVE_SELECTED;
1117                                         ANIM_set_active_channel(ac, ac->data, ac->datatype, filter, fcu, ANIMTYPE_FCURVE);
1118                                 }
1119                         }
1120                 }
1121                 else if (ac->datatype == ANIMCONT_GPENCIL) {
1122                         /* deselect all other channels first */
1123                         ANIM_deselect_anim_channels(ac, ac->data, ac->datatype, 0, ACHANNEL_SETFLAG_CLEAR);
1124                         
1125                         /* Highlight GPencil Layer */
1126                         if ((ale && ale->data) && (ale->type == ANIMTYPE_GPLAYER)) {
1127                                 bGPDlayer *gpl = ale->data;
1128                                 
1129                                 gpl->flag |= GP_LAYER_SELECT;
1130                                 //gpencil_layer_setactive(gpd, gpl);
1131                         }
1132                 }
1133         }
1134         
1135         /* only select keyframes if we clicked on a valid channel and hit something */
1136         if (ale) {
1137                 if (found) {
1138                         /* apply selection to keyframes */
1139                         if (column) {
1140                                 /* select all keyframes in the same frame as the one we hit on the active channel */
1141                                 actkeys_mselect_column(ac, select_mode, selx);
1142                         }
1143                         else {
1144                                 /* select the nominated keyframe on the given frame */
1145                                 actkeys_mselect_single(ac, ale, select_mode, selx);
1146                         }
1147                 }
1148                 
1149                 /* free this channel */
1150                 MEM_freeN(ale);
1151         }
1152 }
1153
1154 /* handle clicking */
1155 static int actkeys_clickselect_invoke(bContext *C, wmOperator *op, wmEvent *event)
1156 {
1157         bAnimContext ac;
1158         Scene *scene;
1159         ARegion *ar;
1160         View2D *v2d;
1161         short selectmode, column;
1162         int mval[2];
1163         
1164         /* get editor data */
1165         if (ANIM_animdata_get_context(C, &ac) == 0)
1166                 return OPERATOR_CANCELLED;
1167                 
1168         /* get useful pointers from animation context data */
1169         scene= ac.scene;
1170         ar= ac.ar;
1171         v2d= &ar->v2d;
1172         
1173         /* get mouse coordinates (in region coordinates) */
1174         mval[0]= (event->x - ar->winrct.xmin);
1175         mval[1]= (event->y - ar->winrct.ymin);
1176         
1177         /* select mode is either replace (deselect all, then add) or add/extend */
1178         if (RNA_boolean_get(op->ptr, "extend"))
1179                 selectmode= SELECT_INVERT;
1180         else
1181                 selectmode= SELECT_REPLACE;
1182                 
1183         /* column selection */
1184         column= RNA_boolean_get(op->ptr, "column");
1185         
1186         /* select keyframe(s) based upon mouse position*/
1187         mouse_action_keys(&ac, mval, selectmode, column);
1188         
1189         /* set notifier that keyframe selection (and channels too) have changed */
1190         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|ND_ANIMCHAN|NA_SELECTED, NULL);
1191         
1192         /* for tweak grab to work */
1193         return OPERATOR_FINISHED|OPERATOR_PASS_THROUGH;
1194 }
1195  
1196 void ACTION_OT_clickselect (wmOperatorType *ot)
1197 {
1198         /* identifiers */
1199         ot->name= "Mouse Select Keys";
1200         ot->idname= "ACTION_OT_clickselect";
1201         ot->description= "Select keyframes by clicking on them";
1202         
1203         /* api callbacks - absolutely no exec() this yet... */
1204         ot->invoke= actkeys_clickselect_invoke;
1205         ot->poll= ED_operator_action_active;
1206         
1207         /* flags */
1208         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
1209         
1210         /* id-props */
1211         RNA_def_boolean(ot->srna, "extend", 0, "Extend Select", ""); // SHIFTKEY
1212         RNA_def_boolean(ot->srna, "column", 0, "Column Select", ""); // ALTKEY
1213 }
1214
1215 /* ************************************************************************** */