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