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