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