e44f1abd262261b968b38ff708a2b4c5521ede1c
[blender.git] / source / blender / editors / space_graph / graph_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_graph/graph_select.c
26  *  \ingroup spgraph
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_utildefines.h"
40
41 #include "DNA_anim_types.h"
42 #include "DNA_object_types.h"
43 #include "DNA_screen_types.h"
44 #include "DNA_scene_types.h"
45 #include "DNA_space_types.h"
46
47 #include "RNA_access.h"
48 #include "RNA_define.h"
49
50 #include "BKE_fcurve.h"
51 #include "BKE_nla.h"
52 #include "BKE_context.h"
53
54 #include "UI_view2d.h"
55
56 #include "ED_anim_api.h"
57 #include "ED_keyframes_edit.h"
58 #include "ED_markers.h"
59
60 #include "WM_api.h"
61 #include "WM_types.h"
62
63 #include "graph_intern.h"
64
65
66 /* ************************************************************************** */
67 /* KEYFRAMES STUFF */
68
69 /* ******************** Deselect All Operator ***************************** */
70 /* This operator works in one of three ways:
71  *      1) (de)select all (AKEY) - test if select all or deselect all
72  *      2) invert all (CTRL-IKEY) - invert selection of all keyframes
73  *      3) (de)select all - no testing is done; only for use internal tools as normal function...
74  */
75
76 /* Deselects keyframes in the Graph Editor
77  *      - This is called by the deselect all operator, as well as other ones!
78  *
79  *      - test: check if select or deselect all
80  *      - sel: how to select keyframes 
81  *              0 = deselect
82  *              1 = select
83  *              2 = invert
84  *      - do_channels: whether to affect selection status of channels
85  */
86 static void deselect_graph_keys (bAnimContext *ac, short test, short sel, short do_channels)
87 {
88         ListBase anim_data = {NULL, NULL};
89         bAnimListElem *ale;
90         int filter;
91         
92         SpaceIpo *sipo= (SpaceIpo *)ac->sl;
93         KeyframeEditData ked= {{NULL}};
94         KeyframeEditFunc test_cb, sel_cb;
95         
96         /* determine type-based settings */
97         filter= (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_CURVE_VISIBLE | ANIMFILTER_NODUPLIS);
98         
99         /* filter data */
100         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
101         
102         /* init BezTriple looping data */
103         test_cb= ANIM_editkeyframes_ok(BEZT_OK_SELECTED);
104         
105         /* See if we should be selecting or deselecting */
106         if (test) {
107                 for (ale= anim_data.first; ale; ale= ale->next) {
108                         if (ANIM_fcurve_keyframes_loop(&ked, ale->key_data, NULL, test_cb, NULL)) {
109                                 sel= SELECT_SUBTRACT;
110                                 break;
111                         }
112                 }
113         }
114         
115         /* convert sel to selectmode, and use that to get editor */
116         sel_cb= ANIM_editkeyframes_select(sel);
117         
118         /* Now set the flags */
119         for (ale= anim_data.first; ale; ale= ale->next) {
120                 FCurve *fcu= (FCurve *)ale->key_data;
121                 
122                 /* Keyframes First */
123                 ANIM_fcurve_keyframes_loop(&ked, ale->key_data, NULL, sel_cb, NULL);
124                 
125                 /* affect channel selection status? */
126                 if (do_channels) {
127                         /* only change selection of channel when the visibility of keyframes doesn't depend on this */
128                         if ((sipo->flag & SIPO_SELCUVERTSONLY) == 0) {
129                                 /* deactivate the F-Curve, and deselect if deselecting keyframes.
130                                  * otherwise select the F-Curve too since we've selected all the keyframes
131                                  */
132                                 if (sel == SELECT_SUBTRACT) 
133                                         fcu->flag &= ~FCURVE_SELECTED;
134                                 else
135                                         fcu->flag |= FCURVE_SELECTED;
136                         }
137                         
138                         /* always deactivate all F-Curves if we perform batch ops for selection */
139                         fcu->flag &= ~FCURVE_ACTIVE;
140                 }
141         }
142         
143         /* Cleanup */
144         BLI_freelistN(&anim_data);
145 }
146
147 /* ------------------- */
148
149 static int graphkeys_deselectall_exec(bContext *C, wmOperator *op)
150 {
151         bAnimContext ac;
152         
153         /* get editor data */
154         if (ANIM_animdata_get_context(C, &ac) == 0)
155                 return OPERATOR_CANCELLED;
156                 
157         /* 'standard' behavior - check if selected, then apply relevant selection */
158         if (RNA_boolean_get(op->ptr, "invert"))
159                 deselect_graph_keys(&ac, 0, SELECT_INVERT, TRUE);
160         else
161                 deselect_graph_keys(&ac, 1, SELECT_ADD, TRUE);
162         
163         /* set notifier that things have changed */
164         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_SELECTED, NULL);
165         
166         return OPERATOR_FINISHED;
167 }
168  
169 void GRAPH_OT_select_all_toggle (wmOperatorType *ot)
170 {
171         /* identifiers */
172         ot->name= "Select All";
173         ot->idname= "GRAPH_OT_select_all_toggle";
174         ot->description= "Toggle selection of all keyframes";
175         
176         /* api callbacks */
177         ot->exec= graphkeys_deselectall_exec;
178         ot->poll= graphop_visible_keyframes_poll;
179         
180         /* flags */
181         ot->flag= OPTYPE_REGISTER/*|OPTYPE_UNDO*/;
182         
183         /* props */
184         ot->prop= RNA_def_boolean(ot->srna, "invert", 0, "Invert", "");
185 }
186
187 /* ******************** Border Select Operator **************************** */
188 /* This operator currently works in one of three ways:
189  *      -> BKEY         - 1) all keyframes within region are selected (validation with BEZT_OK_REGION)
190  *      -> ALT-BKEY - depending on which axis of the region was larger...
191  *              -> 2) x-axis, so select all frames within frame range (validation with BEZT_OK_FRAMERANGE)
192  *              -> 3) y-axis, so select all frames within channels that region included (validation with BEZT_OK_VALUERANGE)
193  */
194
195 /* Borderselect only selects keyframes now, as overshooting handles often get caught too,
196  * which means that they may be inadvertently moved as well. However, incl_handles overrides
197  * this, and allow handles to be considered independently too.
198  * Also, for convenience, handles should get same status as keyframe (if it was within bounds).
199  */
200 static void borderselect_graphkeys (bAnimContext *ac, rcti rect, short mode, short selectmode, short incl_handles)
201 {
202         ListBase anim_data = {NULL, NULL};
203         bAnimListElem *ale;
204         int filter, mapping_flag;
205         
206         SpaceIpo *sipo= (SpaceIpo *)ac->sl;
207         KeyframeEditData ked;
208         KeyframeEditFunc ok_cb, select_cb;
209         View2D *v2d= &ac->ar->v2d;
210         rctf rectf;
211         
212         /* convert mouse coordinates to frame ranges and channel coordinates corrected for view pan/zoom */
213         UI_view2d_region_to_view(v2d, rect.xmin, rect.ymin, &rectf.xmin, &rectf.ymin);
214         UI_view2d_region_to_view(v2d, rect.xmax, rect.ymax, &rectf.xmax, &rectf.ymax);
215         
216         /* filter data */
217         filter= (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_CURVE_VISIBLE | ANIMFILTER_NODUPLIS);
218         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
219         
220         /* get beztriple editing/validation funcs  */
221         select_cb= ANIM_editkeyframes_select(selectmode);
222         ok_cb= ANIM_editkeyframes_ok(mode);
223         
224         /* init editing data */
225         memset(&ked, 0, sizeof(KeyframeEditData));
226         ked.data= &rectf;
227         
228         /* treat handles separately? */
229         if (incl_handles) {
230                 ked.iterflags |= KEYFRAME_ITER_INCL_HANDLES;
231                 mapping_flag= 0;
232         }
233         else
234                 mapping_flag= ANIM_UNITCONV_ONLYKEYS;
235         
236         /* loop over data, doing border select */
237         for (ale= anim_data.first; ale; ale= ale->next) {
238                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
239                 FCurve *fcu= (FCurve *)ale->key_data;
240                 
241                 /* apply unit corrections */
242                 ANIM_unit_mapping_apply_fcurve(ac->scene, ale->id, ale->key_data, mapping_flag);
243                 
244                 /* apply NLA mapping to all the keyframes, since it's easier than trying to
245                  * guess when a callback might use something different
246                  */
247                 if (adt)
248                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 0, incl_handles==0);
249                 
250                 /* set horizontal range (if applicable) 
251                  * NOTE: these values are only used for x-range and y-range but not region 
252                  *              (which uses ked.data, i.e. rectf)
253                  */
254                 if (mode != BEZT_OK_VALUERANGE) {
255                         ked.f1= rectf.xmin;
256                         ked.f2= rectf.xmax;
257                 }
258                 else {
259                         ked.f1= rectf.ymin;
260                         ked.f2= rectf.ymax;
261                 }
262                 
263                 /* firstly, check if any keyframes will be hit by this */
264                 if (ANIM_fcurve_keyframes_loop(&ked, fcu, NULL, ok_cb, NULL)) {
265                         /* select keyframes that are in the appropriate places */
266                         ANIM_fcurve_keyframes_loop(&ked, fcu, ok_cb, select_cb, NULL);
267                         
268                         /* only change selection of channel when the visibility of keyframes doesn't depend on this */
269                         if ((sipo->flag & SIPO_SELCUVERTSONLY) == 0) {
270                                 /* select the curve too now that curve will be touched */
271                                 if (selectmode == SELECT_ADD)
272                                         fcu->flag |= FCURVE_SELECTED;
273                         }
274                 }
275                 
276                 /* un-apply NLA mapping from all the keyframes */
277                 if (adt)
278                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 1, incl_handles==0);
279                         
280                 /* unapply unit corrections */
281                 ANIM_unit_mapping_apply_fcurve(ac->scene, ale->id, ale->key_data, ANIM_UNITCONV_RESTORE|mapping_flag);
282         }
283         
284         /* cleanup */
285         BLI_freelistN(&anim_data);
286 }
287
288 /* ------------------- */
289
290 static int graphkeys_borderselect_exec(bContext *C, wmOperator *op)
291 {
292         bAnimContext ac;
293         rcti rect;
294         short mode=0, selectmode=0;
295         short incl_handles;
296         int extend;
297         
298         /* get editor data */
299         if (ANIM_animdata_get_context(C, &ac) == 0)
300                 return OPERATOR_CANCELLED;
301
302         /* clear all selection if not extending selection */
303         extend= RNA_boolean_get(op->ptr, "extend");
304         if (!extend)
305                 deselect_graph_keys(&ac, 1, SELECT_SUBTRACT, TRUE);
306
307         /* get select mode 
308          *      - 'gesture_mode' from the operator specifies how to select
309          *      - 'include_handles' from the operator specifies whether to include handles in the selection
310          */
311         if (RNA_int_get(op->ptr, "gesture_mode")==GESTURE_MODAL_SELECT)
312                 selectmode= SELECT_ADD;
313         else
314                 selectmode= SELECT_SUBTRACT;
315                 
316         incl_handles = RNA_boolean_get(op->ptr, "include_handles");
317         
318         /* get settings from operator */
319         rect.xmin= RNA_int_get(op->ptr, "xmin");
320         rect.ymin= RNA_int_get(op->ptr, "ymin");
321         rect.xmax= RNA_int_get(op->ptr, "xmax");
322         rect.ymax= RNA_int_get(op->ptr, "ymax");
323         
324         /* selection 'mode' depends on whether borderselect region only matters on one axis */
325         if (RNA_boolean_get(op->ptr, "axis_range")) {
326                 /* mode depends on which axis of the range is larger to determine which axis to use 
327                  *      - checking this in region-space is fine, as it's fundamentally still going to be a different rect size
328                  *      - the frame-range select option is favored over the channel one (x over y), as frame-range one is often
329                  *        used for tweaking timing when "blocking", while channels is not that useful...
330                  */
331                 if ((rect.xmax - rect.xmin) >= (rect.ymax - rect.ymin))
332                         mode= BEZT_OK_FRAMERANGE;
333                 else
334                         mode= BEZT_OK_VALUERANGE;
335         }
336         else 
337                 mode= BEZT_OK_REGION;
338         
339         /* apply borderselect action */
340         borderselect_graphkeys(&ac, rect, mode, selectmode, incl_handles);
341         
342         /* send notifier that keyframe selection has changed */
343         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_SELECTED, NULL);
344         
345         return OPERATOR_FINISHED;
346
347
348 void GRAPH_OT_select_border(wmOperatorType *ot)
349 {
350         /* identifiers */
351         ot->name= "Border Select";
352         ot->idname= "GRAPH_OT_select_border";
353         ot->description= "Select all keyframes within the specified region";
354         
355         /* api callbacks */
356         ot->invoke= WM_border_select_invoke;
357         ot->exec= graphkeys_borderselect_exec;
358         ot->modal= WM_border_select_modal;
359         ot->cancel= WM_border_select_cancel;
360         
361         ot->poll= graphop_visible_keyframes_poll;
362         
363         /* flags */
364         ot->flag= OPTYPE_REGISTER/*|OPTYPE_UNDO*/;
365         
366         /* rna */
367         WM_operator_properties_gesture_border(ot, TRUE);
368         
369         ot->prop= RNA_def_boolean(ot->srna, "axis_range", 0, "Axis Range", "");
370         RNA_def_boolean(ot->srna, "include_handles", 0, "Include Handles", "Are handles tested individually against the selection criteria");
371 }
372
373 /* ******************** Column Select Operator **************************** */
374 /* This operator works in one of four ways:
375  *      - 1) select all keyframes in the same frame as a selected one  (KKEY)
376  *      - 2) select all keyframes in the same frame as the current frame marker (CTRL-KKEY)
377  *      - 3) select all keyframes in the same frame as a selected markers (SHIFT-KKEY)
378  *      - 4) select all keyframes that occur between selected markers (ALT-KKEY)
379  */
380
381 /* defines for column-select mode */
382 static EnumPropertyItem prop_column_select_types[] = {
383         {GRAPHKEYS_COLUMNSEL_KEYS, "KEYS", 0, "On Selected Keyframes", ""},
384         {GRAPHKEYS_COLUMNSEL_CFRA, "CFRA", 0, "On Current Frame", ""},
385         {GRAPHKEYS_COLUMNSEL_MARKERS_COLUMN, "MARKERS_COLUMN", 0, "On Selected Markers", ""},
386         {GRAPHKEYS_COLUMNSEL_MARKERS_BETWEEN, "MARKERS_BETWEEN", 0, "Between Min/Max Selected Markers", ""},
387         {0, NULL, 0, NULL, NULL}
388 };
389
390 /* ------------------- */ 
391
392 /* Selects all visible keyframes between the specified markers */
393 /* TODO, this is almost an _exact_ duplicate of a function of the same name in action_select.c
394  * should de-duplicate - campbell */
395 static void markers_selectkeys_between (bAnimContext *ac)
396 {
397         ListBase anim_data = {NULL, NULL};
398         bAnimListElem *ale;
399         int filter;
400         
401         KeyframeEditFunc ok_cb, select_cb;
402         KeyframeEditData ked= {{NULL}};
403         float min, max;
404         
405         /* get extreme markers */
406         ED_markers_get_minmax(ac->markers, 1, &min, &max);
407         min -= 0.5f;
408         max += 0.5f;
409         
410         /* get editing funcs + data */
411         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAMERANGE);
412         select_cb= ANIM_editkeyframes_select(SELECT_ADD);
413
414         ked.f1= min;
415         ked.f2= max;
416         
417         /* filter data */
418         filter= (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_CURVE_VISIBLE | ANIMFILTER_NODUPLIS);
419         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
420         
421         /* select keys in-between */
422         for (ale= anim_data.first; ale; ale= ale->next) {
423                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
424
425                 if (adt) {
426                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 0, 1);
427                         ANIM_fcurve_keyframes_loop(&ked, ale->key_data, ok_cb, select_cb, NULL);
428                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 1, 1);
429                 }
430                 else {
431                         ANIM_fcurve_keyframes_loop(&ked, ale->key_data, ok_cb, select_cb, NULL);
432                 }
433         }
434         
435         /* Cleanup */
436         BLI_freelistN(&anim_data);
437 }
438
439
440 /* Selects all visible keyframes in the same frames as the specified elements */
441 static void columnselect_graph_keys (bAnimContext *ac, short mode)
442 {
443         ListBase anim_data= {NULL, NULL};
444         bAnimListElem *ale;
445         int filter;
446         
447         Scene *scene= ac->scene;
448         CfraElem *ce;
449         KeyframeEditFunc select_cb, ok_cb;
450         KeyframeEditData ked;
451         
452         /* initialize keyframe editing data */
453         memset(&ked, 0, sizeof(KeyframeEditData));
454         
455         /* build list of columns */
456         switch (mode) {
457                 case GRAPHKEYS_COLUMNSEL_KEYS: /* list of selected keys */
458                         filter= (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_CURVE_VISIBLE | ANIMFILTER_NODUPLIS);
459                         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
460                         
461                         for (ale= anim_data.first; ale; ale= ale->next)
462                                 ANIM_fcurve_keyframes_loop(&ked, ale->key_data, NULL, bezt_to_cfraelem, NULL);
463                         
464                         BLI_freelistN(&anim_data);
465                         break;
466                         
467                 case GRAPHKEYS_COLUMNSEL_CFRA: /* current frame */
468                         /* make a single CfraElem for storing this */
469                         ce= MEM_callocN(sizeof(CfraElem), "cfraElem");
470                         BLI_addtail(&ked.list, ce);
471                         
472                         ce->cfra= (float)CFRA;
473                         break;
474                         
475                 case GRAPHKEYS_COLUMNSEL_MARKERS_COLUMN: /* list of selected markers */
476                         ED_markers_make_cfra_list(ac->markers, &ked.list, SELECT);
477                         break;
478                         
479                 default: /* invalid option */
480                         return;
481         }
482         
483         /* set up BezTriple edit callbacks */
484         select_cb= ANIM_editkeyframes_select(SELECT_ADD);
485         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAME);
486         
487         /* loop through all of the keys and select additional keyframes
488          * based on the keys found to be selected above
489          */
490         filter= (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_CURVE_VISIBLE | ANIMFILTER_NODUPLIS);
491         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
492         
493         for (ale= anim_data.first; ale; ale= ale->next) {
494                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
495                 
496                 /* loop over cfraelems (stored in the KeyframeEditData->list)
497                  *      - we need to do this here, as we can apply fewer NLA-mapping conversions
498                  */
499                 for (ce= ked.list.first; ce; ce= ce->next) {
500                         /* set frame for validation callback to refer to */
501                         ked.f1= BKE_nla_tweakedit_remap(adt, ce->cfra, NLATIME_CONVERT_UNMAP);
502
503                         /* select elements with frame number matching cfraelem */
504                         ANIM_fcurve_keyframes_loop(&ked, ale->key_data, ok_cb, select_cb, NULL);
505                 }
506         }
507         
508         /* free elements */
509         BLI_freelistN(&ked.list);
510         BLI_freelistN(&anim_data);
511 }
512
513 /* ------------------- */
514
515 static int graphkeys_columnselect_exec(bContext *C, wmOperator *op)
516 {
517         bAnimContext ac;
518         short mode;
519         
520         /* get editor data */
521         if (ANIM_animdata_get_context(C, &ac) == 0)
522                 return OPERATOR_CANCELLED;
523                 
524         /* action to take depends on the mode */
525         mode= RNA_enum_get(op->ptr, "mode");
526         
527         if (mode == GRAPHKEYS_COLUMNSEL_MARKERS_BETWEEN)
528                 markers_selectkeys_between(&ac);
529         else
530                 columnselect_graph_keys(&ac, mode);
531         
532         /* set notifier that keyframe selection has changed */
533         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_SELECTED, NULL);
534         
535         return OPERATOR_FINISHED;
536 }
537  
538 void GRAPH_OT_select_column (wmOperatorType *ot)
539 {
540         /* identifiers */
541         ot->name= "Select All";
542         ot->idname= "GRAPH_OT_select_column";
543         ot->description= "Select all keyframes on the specified frame(s)";
544         
545         /* api callbacks */
546         ot->exec= graphkeys_columnselect_exec;
547         ot->poll= graphop_visible_keyframes_poll;
548         
549         /* flags */
550         ot->flag= OPTYPE_REGISTER/*|OPTYPE_UNDO*/;
551         
552         /* props */
553         ot->prop= RNA_def_enum(ot->srna, "mode", prop_column_select_types, 0, "Mode", "");
554 }
555
556 /* ******************** Select Linked Operator *********************** */
557
558 static int graphkeys_select_linked_exec (bContext *C, wmOperator *UNUSED(op))
559 {
560         bAnimContext ac;
561         
562         ListBase anim_data= {NULL, NULL};
563         bAnimListElem *ale;
564         int filter;
565         
566         KeyframeEditFunc ok_cb = ANIM_editkeyframes_ok(BEZT_OK_SELECTED);
567         KeyframeEditFunc sel_cb = ANIM_editkeyframes_select(SELECT_ADD);
568         
569         /* get editor data */
570         if (ANIM_animdata_get_context(C, &ac) == 0)
571                 return OPERATOR_CANCELLED;
572         
573         /* loop through all of the keys and select additional keyframes based on these */
574         filter= (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_CURVE_VISIBLE | ANIMFILTER_NODUPLIS);
575         ANIM_animdata_filter(&ac, &anim_data, filter, ac.data, ac.datatype);
576         
577         for (ale= anim_data.first; ale; ale= ale->next) {
578                 FCurve *fcu= (FCurve *)ale->key_data;
579                 
580                 /* check if anything selected? */
581                 if (ANIM_fcurve_keyframes_loop(NULL, fcu, NULL, ok_cb, NULL)) {
582                         /* select every keyframe in this curve then */
583                         ANIM_fcurve_keyframes_loop(NULL, fcu, NULL, sel_cb, NULL);
584                 }
585         }
586         
587         /* Cleanup */
588         BLI_freelistN(&anim_data);
589         
590         /* set notifier that keyframe selection has changed */
591         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_SELECTED, NULL);
592         
593         return OPERATOR_FINISHED;
594 }
595
596 void GRAPH_OT_select_linked (wmOperatorType *ot)
597 {
598         /* identifiers */
599         ot->name = "Select Linked";
600         ot->idname= "GRAPH_OT_select_linked";
601         ot->description = "Select keyframes occurring in the same F-Curves as selected ones";
602         
603         /* api callbacks */
604         ot->exec= graphkeys_select_linked_exec;
605         ot->poll= graphop_visible_keyframes_poll;
606         
607         /* flags */
608         ot->flag= OPTYPE_REGISTER/*|OPTYPE_UNDO*/;
609 }
610
611 /* ******************** Select More/Less Operators *********************** */
612
613 /* Common code to perform selection */
614 static void select_moreless_graph_keys (bAnimContext *ac, short mode)
615 {
616         ListBase anim_data= {NULL, NULL};
617         bAnimListElem *ale;
618         int filter;
619         
620         KeyframeEditData ked;
621         KeyframeEditFunc build_cb;
622         
623         
624         /* init selmap building data */
625         build_cb= ANIM_editkeyframes_buildselmap(mode);
626         memset(&ked, 0, sizeof(KeyframeEditData)); 
627         
628         /* loop through all of the keys and select additional keyframes based on these */
629         filter= (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_CURVE_VISIBLE | ANIMFILTER_NODUPLIS);
630         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
631         
632         for (ale= anim_data.first; ale; ale= ale->next) {
633                 FCurve *fcu= (FCurve *)ale->key_data;
634                 
635                 /* only continue if F-Curve has keyframes */
636                 if (fcu->bezt == NULL)
637                         continue;
638                 
639                 /* build up map of whether F-Curve's keyframes should be selected or not */
640                 ked.data= MEM_callocN(fcu->totvert, "selmap graphEdit");
641                 ANIM_fcurve_keyframes_loop(&ked, fcu, NULL, build_cb, NULL);
642                 
643                 /* based on this map, adjust the selection status of the keyframes */
644                 ANIM_fcurve_keyframes_loop(&ked, fcu, NULL, bezt_selmap_flush, NULL);
645                 
646                 /* free the selmap used here */
647                 MEM_freeN(ked.data);
648                 ked.data= NULL;
649         }
650         
651         /* Cleanup */
652         BLI_freelistN(&anim_data);
653 }
654
655 /* ----------------- */
656
657 static int graphkeys_select_more_exec (bContext *C, wmOperator *UNUSED(op))
658 {
659         bAnimContext ac;
660         
661         /* get editor data */
662         if (ANIM_animdata_get_context(C, &ac) == 0)
663                 return OPERATOR_CANCELLED;
664         
665         /* perform select changes */
666         select_moreless_graph_keys(&ac, SELMAP_MORE);
667         
668         /* set notifier that keyframe selection has changed */
669         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_SELECTED, NULL);
670         
671         return OPERATOR_FINISHED;
672 }
673
674 void GRAPH_OT_select_more (wmOperatorType *ot)
675 {
676         /* identifiers */
677         ot->name = "Select More";
678         ot->idname= "GRAPH_OT_select_more";
679         ot->description = "Select keyframes beside already selected ones";
680         
681         /* api callbacks */
682         ot->exec= graphkeys_select_more_exec;
683         ot->poll= graphop_visible_keyframes_poll;
684         
685         /* flags */
686         ot->flag= OPTYPE_REGISTER/*|OPTYPE_UNDO*/;
687 }
688
689 /* ----------------- */
690
691 static int graphkeys_select_less_exec (bContext *C, wmOperator *UNUSED(op))
692 {
693         bAnimContext ac;
694         
695         /* get editor data */
696         if (ANIM_animdata_get_context(C, &ac) == 0)
697                 return OPERATOR_CANCELLED;
698         
699         /* perform select changes */
700         select_moreless_graph_keys(&ac, SELMAP_LESS);
701         
702         /* set notifier that keyframe selection has changed */
703         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_SELECTED, NULL);
704         
705         return OPERATOR_FINISHED;
706 }
707
708 void GRAPH_OT_select_less (wmOperatorType *ot)
709 {
710         /* identifiers */
711         ot->name = "Select Less";
712         ot->idname= "GRAPH_OT_select_less";
713         ot->description = "Deselect keyframes on ends of selection islands";
714         
715         /* api callbacks */
716         ot->exec= graphkeys_select_less_exec;
717         ot->poll= graphop_visible_keyframes_poll;
718         
719         /* flags */
720         ot->flag= OPTYPE_REGISTER/*|OPTYPE_UNDO*/;
721 }
722
723 /* ******************** Select Left/Right Operator ************************* */
724 /* Select keyframes left/right of the current frame indicator */
725
726 /* defines for left-right select tool */
727 static EnumPropertyItem prop_graphkeys_leftright_select_types[] = {
728         {GRAPHKEYS_LRSEL_TEST, "CHECK", 0, "Check if Select Left or Right", ""},
729         {GRAPHKEYS_LRSEL_LEFT, "LEFT", 0, "Before current frame", ""},
730         {GRAPHKEYS_LRSEL_RIGHT, "RIGHT", 0, "After current frame", ""},
731         {0, NULL, 0, NULL, NULL}
732 };
733
734 /* --------------------------------- */
735
736 static void graphkeys_select_leftright (bAnimContext *ac, short leftright, short select_mode)
737 {
738         ListBase anim_data = {NULL, NULL};
739         bAnimListElem *ale;
740         int filter;
741         
742         KeyframeEditFunc ok_cb, select_cb;
743         KeyframeEditData ked= {{NULL}};
744         Scene *scene= ac->scene;
745         
746         /* if select mode is replace, deselect all keyframes (and channels) first */
747         if (select_mode==SELECT_REPLACE) {
748                 select_mode= SELECT_ADD;
749                 
750                 /* - deselect all other keyframes, so that just the newly selected remain
751                  * - channels aren't deselected, since we don't re-select any as a consequence
752                  */
753                 deselect_graph_keys(ac, 0, SELECT_SUBTRACT, FALSE);
754         }
755         
756         /* set callbacks and editing data */
757         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAMERANGE);
758         select_cb= ANIM_editkeyframes_select(select_mode);
759         
760         if (leftright == GRAPHKEYS_LRSEL_LEFT) {
761                 ked.f1 = MINAFRAMEF;
762                 ked.f2 = (float)(CFRA + 0.1f);
763         } 
764         else {
765                 ked.f1 = (float)(CFRA - 0.1f);
766                 ked.f2 = MAXFRAMEF;
767         }
768         
769         /* filter data */
770         filter= (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_NODUPLIS);
771         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
772                 
773         /* select keys */
774         for (ale= anim_data.first; ale; ale= ale->next) {
775                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
776                 
777                 if (adt) {
778                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 0, 1);
779                         ANIM_fcurve_keyframes_loop(&ked, ale->key_data, ok_cb, select_cb, NULL);
780                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 1, 1);
781                 }
782                 else
783                         ANIM_fcurve_keyframes_loop(&ked, ale->key_data, ok_cb, select_cb, NULL);
784         }
785
786         /* Cleanup */
787         BLI_freelistN(&anim_data);
788 }
789
790 /* ----------------- */
791
792 static int graphkeys_select_leftright_exec (bContext *C, wmOperator *op)
793 {
794         bAnimContext ac;
795         short leftright = RNA_enum_get(op->ptr, "mode");
796         short selectmode;
797         
798         /* get editor data */
799         if (ANIM_animdata_get_context(C, &ac) == 0)
800                 return OPERATOR_CANCELLED;
801         
802         /* select mode is either replace (deselect all, then add) or add/extend */
803         if (RNA_boolean_get(op->ptr, "extend"))
804                 selectmode= SELECT_INVERT;
805         else
806                 selectmode= SELECT_REPLACE;
807                 
808         /* if "test" mode is set, we don't have any info to set this with */
809         if (leftright == GRAPHKEYS_LRSEL_TEST)
810                 return OPERATOR_CANCELLED;
811         
812         /* do the selecting now */
813         graphkeys_select_leftright(&ac, leftright, selectmode);
814         
815         /* set notifier that keyframe selection (and channels too) have changed */
816         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|ND_ANIMCHAN|NA_SELECTED, NULL);
817         
818         return OPERATOR_FINISHED;
819 }
820
821 static int graphkeys_select_leftright_invoke (bContext *C, wmOperator *op, wmEvent *event)
822 {
823         bAnimContext ac;
824         short leftright = RNA_enum_get(op->ptr, "mode");
825         
826         /* get editor data */
827         if (ANIM_animdata_get_context(C, &ac) == 0)
828                 return OPERATOR_CANCELLED;
829                 
830         /* handle mode-based testing */
831         if (leftright == GRAPHKEYS_LRSEL_TEST) {
832                 Scene *scene= ac.scene;
833                 ARegion *ar= ac.ar;
834                 View2D *v2d= &ar->v2d;
835                 float x;
836
837                 /* determine which side of the current frame mouse is on */
838                 UI_view2d_region_to_view(v2d, event->mval[0], event->mval[1], &x, NULL);
839                 if (x < CFRA)
840                         RNA_enum_set(op->ptr, "mode", GRAPHKEYS_LRSEL_LEFT);
841                 else    
842                         RNA_enum_set(op->ptr, "mode", GRAPHKEYS_LRSEL_RIGHT);
843         }
844         
845         /* perform selection */
846         return graphkeys_select_leftright_exec(C, op);
847 }
848
849 void GRAPH_OT_select_leftright (wmOperatorType *ot)
850 {
851         /* identifiers */
852         ot->name= "Select Left/Right";
853         ot->idname= "GRAPH_OT_select_leftright";
854         ot->description= "Select keyframes to the left or the right of the current frame";
855         
856         /* api callbacks  */
857         ot->invoke=     graphkeys_select_leftright_invoke;
858         ot->exec= graphkeys_select_leftright_exec;
859         ot->poll= graphop_visible_keyframes_poll;
860         
861         /* flags */
862         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
863         
864         /* id-props */
865         ot->prop= RNA_def_enum(ot->srna, "mode", prop_graphkeys_leftright_select_types, GRAPHKEYS_LRSEL_TEST, "Mode", "");
866         RNA_def_boolean(ot->srna, "extend", 0, "Extend Select", "");
867 }
868
869 /* ******************** Mouse-Click Select Operator *********************** */
870 /* This operator works in one of three ways:
871  *      - 1) keyframe under mouse - no special modifiers
872  *      - 2) all keyframes on the same side of current frame indicator as mouse - ALT modifier
873  *      - 3) column select all keyframes in frame under mouse - CTRL modifier
874  *
875  * In addition to these basic options, the SHIFT modifier can be used to toggle the 
876  * selection mode between replacing the selection (without) and inverting the selection (with).
877  */
878
879 /* temp info for caching handle vertices close */
880 typedef struct tNearestVertInfo {
881         struct tNearestVertInfo *next, *prev;
882         
883         FCurve *fcu;            /* F-Curve that keyframe comes from */
884         
885         BezTriple *bezt;        /* keyframe to consider */
886         FPoint *fpt;            /* sample point to consider */
887         
888         short hpoint;           /* the handle index that we hit (eHandleIndex) */
889         short sel;                      /* whether the handle is selected or not */
890         int dist;                       /* distance from mouse to vert */
891 } tNearestVertInfo;
892
893 /* Tags for the type of graph vert that we have */
894 typedef enum eGraphVertIndex {
895         NEAREST_HANDLE_LEFT     = -1,
896         NEAREST_HANDLE_KEY,
897         NEAREST_HANDLE_RIGHT
898 } eGraphVertIndex; 
899
900 /* Tolerance for absolute radius (in pixels) of the vert from the cursor to use */
901 // TODO: perhaps this should depend a bit on the size that the user set the vertices to be?
902 #define GVERTSEL_TOL    10
903
904 /* ....... */
905
906 /* check if its ok to select a handle */
907 // XXX also need to check for int-values only?
908 static int fcurve_handle_sel_check(SpaceIpo *sipo, BezTriple *bezt)
909 {
910         if (sipo->flag & SIPO_NOHANDLES) return 0;
911         if ((sipo->flag & SIPO_SELVHANDLESONLY) && BEZSELECTED(bezt)==0) return 0;
912         return 1;
913 }
914
915 /* check if the given vertex is within bounds or not */
916 // TODO: should we return if we hit something?
917 static void nearest_fcurve_vert_store (ListBase *matches, View2D *v2d, FCurve *fcu, BezTriple *bezt, FPoint *fpt, short hpoint, const int mval[2])
918 {
919         /* Keyframes or Samples? */
920         if (bezt) {
921                 int screen_co[2], dist;
922                 
923                 /* convert from data-space to screen coordinates 
924                  * NOTE: hpoint+1 gives us 0,1,2 respectively for each handle, 
925                  *      needed to access the relevant vertex coordinates in the 3x3 
926                  *      'vec' matrix
927                  */
928                 UI_view2d_view_to_region(v2d, bezt->vec[hpoint+1][0], bezt->vec[hpoint+1][1], &screen_co[0], &screen_co[1]);
929                 
930                 /* check if distance from mouse cursor to vert in screen space is within tolerance */
931                         // XXX: inlined distance calculation, since we cannot do this on ints using the math lib...
932                 //dist = len_v2v2(mval, screen_co);
933                 dist = sqrt((mval[0] - screen_co[0])*(mval[0] - screen_co[0]) + 
934                                         (mval[1] - screen_co[1])*(mval[1] - screen_co[1]));
935                 
936                 if (dist <= GVERTSEL_TOL) {
937                         tNearestVertInfo *nvi = (tNearestVertInfo *)matches->last;
938                         short replace = 0;
939                         
940                         /* if there is already a point for the F-Curve, check if this point is closer than that was */
941                         if ((nvi) && (nvi->fcu == fcu)) {
942                                 /* replace if we are closer, or if equal and that one wasn't selected but we are... */
943                                 if ( (nvi->dist > dist) || ((nvi->sel == 0) && BEZSELECTED(bezt)) )
944                                         replace= 1;
945                         }
946                         /* add new if not replacing... */
947                         if (replace == 0)
948                                 nvi = MEM_callocN(sizeof(tNearestVertInfo), "Nearest Graph Vert Info - Bezt");
949                         
950                         /* store values */
951                         nvi->fcu = fcu;
952                         nvi->bezt = bezt;
953                         nvi->hpoint = hpoint;
954                         nvi->dist = dist;
955                         
956                         nvi->sel= BEZSELECTED(bezt); // XXX... should this use the individual verts instead?
957                         
958                         /* add to list of matches if appropriate... */
959                         if (replace == 0)
960                                 BLI_addtail(matches, nvi);
961                 }
962         }
963         else if (fpt) {
964                 // TODO...
965         }
966
967
968 /* helper for find_nearest_fcurve_vert() - build the list of nearest matches */
969 static void get_nearest_fcurve_verts_list (bAnimContext *ac, const int mval[2], ListBase *matches)
970 {
971         ListBase anim_data = {NULL, NULL};
972         bAnimListElem *ale;
973         int filter;
974         
975         SpaceIpo *sipo= (SpaceIpo *)ac->sl;
976         View2D *v2d= &ac->ar->v2d;
977         
978         /* get curves to search through 
979          *      - if the option to only show keyframes that belong to selected F-Curves is enabled,
980          *        include the 'only selected' flag...
981          */
982         filter= (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_CURVE_VISIBLE | ANIMFILTER_NODUPLIS);
983         if (sipo->flag & SIPO_SELCUVERTSONLY)   // FIXME: this should really be check for by the filtering code...
984                 filter |= ANIMFILTER_SEL;
985         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
986         
987         for (ale= anim_data.first; ale; ale= ale->next) {
988                 FCurve *fcu= (FCurve *)ale->key_data;
989                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
990                 
991                 /* apply unit corrections */
992                 ANIM_unit_mapping_apply_fcurve(ac->scene, ale->id, ale->key_data, 0);
993                 
994                 /* apply NLA mapping to all the keyframes */
995                 if (adt)
996                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 0, 0);
997                 
998                 if (fcu->bezt) {
999                         BezTriple *bezt1=fcu->bezt, *prevbezt=NULL;
1000                         int i;
1001                         
1002                         for (i=0; i < fcu->totvert; i++, prevbezt=bezt1, bezt1++) {
1003                                 /* keyframe */
1004                                 nearest_fcurve_vert_store(matches, v2d, fcu, bezt1, NULL, NEAREST_HANDLE_KEY, mval);
1005                                 
1006                                 /* handles - only do them if they're visible */
1007                                 if (fcurve_handle_sel_check(sipo, bezt1) && (fcu->totvert > 1)) {
1008                                         /* first handle only visible if previous segment had handles */
1009                                         if ( (!prevbezt && (bezt1->ipo==BEZT_IPO_BEZ)) || (prevbezt && (prevbezt->ipo==BEZT_IPO_BEZ)) )
1010                                         {
1011                                                 nearest_fcurve_vert_store(matches, v2d, fcu, bezt1, NULL, NEAREST_HANDLE_LEFT, mval);
1012                                         }
1013                                         
1014                                         /* second handle only visible if this segment is bezier */
1015                                         if (bezt1->ipo == BEZT_IPO_BEZ) 
1016                                         {
1017                                                 nearest_fcurve_vert_store(matches, v2d, fcu, bezt1, NULL, NEAREST_HANDLE_RIGHT, mval);
1018                                         }
1019                                 }
1020                         }
1021                 }
1022                 else if (fcu->fpt) {
1023                         // TODO; do this for samples too
1024                         
1025                 }
1026                 
1027                 /* un-apply NLA mapping from all the keyframes */
1028                 if (adt)
1029                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 1, 0);
1030                 
1031                 /* unapply unit corrections */
1032                 ANIM_unit_mapping_apply_fcurve(ac->scene, ale->id, ale->key_data, ANIM_UNITCONV_RESTORE);
1033         }
1034         
1035         /* free channels */
1036         BLI_freelistN(&anim_data);
1037 }
1038
1039 /* helper for find_nearest_fcurve_vert() - get the best match to use */
1040 static tNearestVertInfo *get_best_nearest_fcurve_vert (ListBase *matches)
1041 {
1042         tNearestVertInfo *nvi = NULL;
1043         short found = 0;
1044         
1045         /* abort if list is empty */
1046         if (matches->first == NULL) 
1047                 return NULL;
1048                 
1049         /* if list only has 1 item, remove it from the list and return */
1050         if (matches->first == matches->last) {
1051                 /* need to remove from the list, otherwise it gets freed and then we can't return it */
1052                 nvi= matches->first;
1053                 BLI_remlink(matches, nvi);
1054                 
1055                 return nvi;
1056         }
1057         
1058         /* try to find the first selected F-Curve vert, then take the one after it */
1059         for (nvi = matches->first; nvi; nvi = nvi->next) {
1060                 /* which mode of search are we in: find first selected, or find vert? */
1061                 if (found) {
1062                         /* just take this vert now that we've found the selected one 
1063                          *      - we'll need to remove this from the list so that it can be returned to the original caller
1064                          */
1065                         BLI_remlink(matches, nvi);
1066                         return nvi;
1067                 }
1068                 else {
1069                         /* if vert is selected, we've got what we want... */
1070                         if (nvi->sel)
1071                                 found= 1;
1072                 }
1073         }
1074         
1075         /* if we're still here, this means that we failed to find anything appropriate in the first pass,
1076          * so just take the first item now...
1077          */
1078         nvi = matches->first;
1079         BLI_remlink(matches, nvi);
1080         return nvi;
1081 }
1082
1083 /* Find the nearest vertices (either a handle or the keyframe) that are nearest to the mouse cursor (in area coordinates) 
1084  * NOTE: the match info found must still be freed 
1085  */
1086 static tNearestVertInfo *find_nearest_fcurve_vert (bAnimContext *ac, const int mval[2])
1087 {
1088         ListBase matches = {NULL, NULL};
1089         tNearestVertInfo *nvi;
1090         
1091         /* step 1: get the nearest verts */
1092         get_nearest_fcurve_verts_list(ac, mval, &matches);
1093         
1094         /* step 2: find the best vert */
1095         nvi= get_best_nearest_fcurve_vert(&matches);
1096         
1097         BLI_freelistN(&matches);
1098         
1099         /* return the best vert found */
1100         return nvi;
1101 }
1102
1103 /* ------------------- */
1104
1105 /* option 1) select keyframe directly under mouse */
1106 static void mouse_graph_keys (bAnimContext *ac, const int mval[2], short select_mode, short curves_only)
1107 {
1108         SpaceIpo *sipo= (SpaceIpo *)ac->sl;
1109         tNearestVertInfo *nvi;
1110         BezTriple *bezt= NULL;
1111         
1112         /* find the beztriple that we're selecting, and the handle that was clicked on */
1113         nvi = find_nearest_fcurve_vert(ac, mval);
1114         
1115         /* check if anything to select */
1116         if (nvi == NULL)        
1117                 return;
1118         
1119         /* deselect all other curves? */
1120         if (select_mode == SELECT_REPLACE) {
1121                 /* reset selection mode */
1122                 select_mode= SELECT_ADD;
1123                 
1124                 /* deselect all other keyframes (+ F-Curves too) */
1125                 deselect_graph_keys(ac, 0, SELECT_SUBTRACT, TRUE);
1126                 
1127                 /* deselect other channels too, but only only do this if 
1128                  * selection of channel when the visibility of keyframes 
1129                  * doesn't depend on this 
1130                  */
1131                 if ((sipo->flag & SIPO_SELCUVERTSONLY) == 0)
1132                         ANIM_deselect_anim_channels(ac, ac->data, ac->datatype, 0, ACHANNEL_SETFLAG_CLEAR);
1133         }
1134         
1135         /* if points can be selected on this F-Curve */
1136         // TODO: what about those with no keyframes?
1137         if ((curves_only == 0) && ((nvi->fcu->flag & FCURVE_PROTECTED)==0)) {
1138                 /* only if there's keyframe */
1139                 if (nvi->bezt) {
1140                         bezt= nvi->bezt; /* used to check bezt seletion is set */
1141                         /* depends on selection mode */
1142                         if (select_mode == SELECT_INVERT) {
1143                                 /* keyframe - invert select of all */
1144                                 if (nvi->hpoint == NEAREST_HANDLE_KEY) {
1145                                         if (BEZSELECTED(bezt)) {
1146                                                 BEZ_DESEL(bezt);
1147                                         }
1148                                         else {
1149                                                 BEZ_SEL(bezt);
1150                                         }
1151                                 }
1152                                 
1153                                 /* handles - toggle selection of relevant handle */
1154                                 else if (nvi->hpoint == NEAREST_HANDLE_LEFT) {
1155                                         /* toggle selection */
1156                                         bezt->f1 ^= SELECT;
1157                                 }
1158                                 else {
1159                                         /* toggle selection */
1160                                         bezt->f3 ^= SELECT;
1161                                 }
1162                         }
1163                         else {
1164                                 /* if the keyframe was clicked on, select all verts of given beztriple */
1165                                 if (nvi->hpoint == NEAREST_HANDLE_KEY) {
1166                                         BEZ_SEL(bezt);
1167                                 }
1168                                 /* otherwise, select the handle that applied */
1169                                 else if (nvi->hpoint == NEAREST_HANDLE_LEFT) 
1170                                         bezt->f1 |= SELECT;
1171                                 else 
1172                                         bezt->f3 |= SELECT;
1173                         }
1174                 }
1175                 else if (nvi->fpt) {
1176                         // TODO: need to handle sample points
1177                 }
1178         }
1179         else {
1180                 KeyframeEditFunc select_cb;
1181                 KeyframeEditData ked;
1182                 
1183                 /* initialize keyframe editing data */
1184                 memset(&ked, 0, sizeof(KeyframeEditData));
1185                 
1186                 /* set up BezTriple edit callbacks */
1187                 select_cb= ANIM_editkeyframes_select(select_mode);
1188                 
1189                 /* select all keyframes */
1190                 ANIM_fcurve_keyframes_loop(&ked, nvi->fcu, NULL, select_cb, NULL);
1191         }
1192         
1193         /* only change selection of channel when the visibility of keyframes doesn't depend on this */
1194         if ((sipo->flag & SIPO_SELCUVERTSONLY) == 0) {
1195                 /* select or deselect curve? */
1196                 if (bezt) {
1197                         /* take selection status from item that got hit, to prevent flip/flop on channel 
1198                          * selection status when shift-selecting (i.e. "SELECT_INVERT") points
1199                          */
1200                         if (BEZSELECTED(bezt))
1201                                 nvi->fcu->flag |= FCURVE_SELECTED;
1202                         else
1203                                 nvi->fcu->flag &= ~FCURVE_SELECTED;
1204                 }
1205                 else {
1206                         /* didn't hit any channel, so just apply that selection mode to the curve's selection status */
1207                         if (select_mode == SELECT_INVERT)
1208                                 nvi->fcu->flag ^= FCURVE_SELECTED;
1209                         else if (select_mode == SELECT_ADD)
1210                                 nvi->fcu->flag |= FCURVE_SELECTED;
1211                 }
1212         }
1213
1214         /* set active F-Curve (NOTE: sync the filter flags with findnearest_fcurve_vert) */
1215         /* needs to be called with (sipo->flag & SIPO_SELCUVERTSONLY) otherwise the active flag won't be set [#26452] */
1216         if (nvi->fcu->flag & FCURVE_SELECTED) {
1217                 int filter= (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_CURVE_VISIBLE | ANIMFILTER_NODUPLIS);
1218                 ANIM_set_active_channel(ac, ac->data, ac->datatype, filter, nvi->fcu, ANIMTYPE_FCURVE);
1219         }
1220
1221         /* free temp sample data for filtering */
1222         MEM_freeN(nvi);
1223 }
1224
1225 /* Option 2) Selects all the keyframes on either side of the current frame (depends on which side the mouse is on) */
1226 /* (see graphkeys_select_leftright) */
1227
1228 /* Option 3) Selects all visible keyframes in the same frame as the mouse click */
1229 static void graphkeys_mselect_column (bAnimContext *ac, const int mval[2], short select_mode)
1230 {
1231         ListBase anim_data= {NULL, NULL};
1232         bAnimListElem *ale;
1233         int filter;
1234         
1235         KeyframeEditFunc select_cb, ok_cb;
1236         KeyframeEditData ked;
1237         tNearestVertInfo *nvi;
1238         float selx = (float)ac->scene->r.cfra;
1239         
1240         /* find the beztriple that we're selecting, and the handle that was clicked on */
1241         nvi = find_nearest_fcurve_vert(ac, mval);
1242         
1243         /* check if anything to select */
1244         if (nvi == NULL)        
1245                 return;
1246         
1247         /* get frame number on which elements should be selected */
1248         // TODO: should we restrict to integer frames only?
1249         if (nvi->bezt)
1250                 selx= nvi->bezt->vec[1][0];
1251         else if (nvi->fpt)
1252                 selx= nvi->fpt->vec[0];
1253         
1254         /* if select mode is replace, deselect all keyframes first */
1255         if (select_mode==SELECT_REPLACE) {
1256                 /* reset selection mode to add to selection */
1257                 select_mode= SELECT_ADD;
1258                 
1259                 /* - deselect all other keyframes, so that just the newly selected remain
1260                  * - channels aren't deselected, since we don't re-select any as a consequence
1261                  */
1262                 deselect_graph_keys(ac, 0, SELECT_SUBTRACT, FALSE);
1263         }
1264         
1265         /* initialize keyframe editing data */
1266         memset(&ked, 0, sizeof(KeyframeEditData));
1267         
1268         /* set up BezTriple edit callbacks */
1269         select_cb= ANIM_editkeyframes_select(select_mode);
1270         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAME);
1271         
1272         /* loop through all of the keys and select additional keyframes
1273          * based on the keys found to be selected above
1274          */
1275         filter= (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_CURVE_VISIBLE | ANIMFILTER_NODUPLIS);
1276         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
1277         
1278         for (ale= anim_data.first; ale; ale= ale->next) {
1279                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
1280                 
1281                 /* set frame for validation callback to refer to */
1282                 if (adt)
1283                         ked.f1= BKE_nla_tweakedit_remap(adt, selx, NLATIME_CONVERT_UNMAP);
1284                 else
1285                         ked.f1= selx;
1286                 
1287                 /* select elements with frame number matching cfra */
1288                 ANIM_fcurve_keyframes_loop(&ked, ale->key_data, ok_cb, select_cb, NULL);
1289         }
1290         
1291         /* free elements */
1292         MEM_freeN(nvi);
1293         BLI_freelistN(&ked.list);
1294         BLI_freelistN(&anim_data);
1295 }
1296  
1297 /* ------------------- */
1298
1299 /* handle clicking */
1300 static int graphkeys_clickselect_invoke(bContext *C, wmOperator *op, wmEvent *event)
1301 {
1302         bAnimContext ac;
1303         short selectmode;
1304
1305         /* get editor data */
1306         if (ANIM_animdata_get_context(C, &ac) == 0)
1307                 return OPERATOR_CANCELLED;
1308
1309         /* select mode is either replace (deselect all, then add) or add/extend */
1310         if (RNA_boolean_get(op->ptr, "extend"))
1311                 selectmode= SELECT_INVERT;
1312         else
1313                 selectmode= SELECT_REPLACE;
1314         
1315         /* figure out action to take */
1316         if (RNA_boolean_get(op->ptr, "column")) {
1317                 /* select all keyframes in the same frame as the one that was under the mouse */
1318                 graphkeys_mselect_column(&ac, event->mval, selectmode);
1319         }
1320         else if (RNA_boolean_get(op->ptr, "curves")) {
1321                 /* select all keyframes in the same F-Curve as the one under the mouse */
1322                 mouse_graph_keys(&ac, event->mval, selectmode, 1);
1323         }
1324         else {
1325                 /* select keyframe under mouse */
1326                 mouse_graph_keys(&ac, event->mval, selectmode, 0);
1327         }
1328         
1329         /* set notifier that keyframe selection (and also channel selection in some cases) has changed */
1330         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|ND_ANIMCHAN|NA_SELECTED, NULL);
1331         
1332         /* for tweak grab to work */
1333         return OPERATOR_FINISHED|OPERATOR_PASS_THROUGH;
1334 }
1335  
1336 void GRAPH_OT_clickselect (wmOperatorType *ot)
1337 {
1338         /* identifiers */
1339         ot->name= "Mouse Select Keys";
1340         ot->idname= "GRAPH_OT_clickselect";
1341         ot->description= "Select keyframes by clicking on them";
1342         
1343         /* api callbacks */
1344         ot->invoke= graphkeys_clickselect_invoke;
1345         ot->poll= graphop_visible_keyframes_poll;
1346         
1347         /* id-props */
1348         RNA_def_boolean(ot->srna, "extend", 0, "Extend Select", ""); // SHIFTKEY
1349         RNA_def_boolean(ot->srna, "column", 0, "Column Select", "Select all keyframes that occur on the same frame as the one under the mouse"); // ALTKEY
1350         RNA_def_boolean(ot->srna, "curves", 0, "Only Curves", "Select all the keyframes in the curve"); // CTRLKEY + ALTKEY
1351 }
1352
1353 /* ************************************************************************** */