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