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