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