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