Aligorith: please check. this was copied from mouse select. however BKE_nla_tweakedit...
[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., 59 Temple Place - Suite 330, Boston, MA  02111-1307, 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_action_types.h"
43 #include "DNA_armature_types.h"
44 #include "DNA_camera_types.h"
45 #include "DNA_curve_types.h"
46 #include "DNA_object_types.h"
47 #include "DNA_screen_types.h"
48 #include "DNA_scene_types.h"
49 #include "DNA_space_types.h"
50 #include "DNA_constraint_types.h"
51 #include "DNA_key_types.h"
52 #include "DNA_lamp_types.h"
53 #include "DNA_material_types.h"
54 #include "DNA_userdef_types.h"
55 #include "DNA_gpencil_types.h"
56 #include "DNA_windowmanager_types.h"
57
58 #include "RNA_access.h"
59 #include "RNA_define.h"
60
61 #include "BKE_action.h"
62 #include "BKE_depsgraph.h"
63 #include "BKE_fcurve.h"
64 #include "BKE_key.h"
65 #include "BKE_material.h"
66 #include "BKE_nla.h"
67 #include "BKE_object.h"
68 #include "BKE_context.h"
69 #include "BKE_utildefines.h"
70
71 #include "UI_view2d.h"
72
73 #include "ED_anim_api.h"
74 #include "ED_keyframing.h"
75 #include "ED_keyframes_draw.h"
76 #include "ED_keyframes_edit.h"
77 #include "ED_markers.h"
78 #include "ED_screen.h"
79 #include "ED_space_api.h"
80
81 #include "WM_api.h"
82 #include "WM_types.h"
83
84 #include "graph_intern.h"
85
86
87 /* ************************************************************************** */
88 /* KEYFRAMES STUFF */
89
90 /* ******************** Deselect All Operator ***************************** */
91 /* This operator works in one of three ways:
92  *      1) (de)select all (AKEY) - test if select all or deselect all
93  *      2) invert all (CTRL-IKEY) - invert selection of all keyframes
94  *      3) (de)select all - no testing is done; only for use internal tools as normal function...
95  */
96
97 /* Deselects keyframes in the Graph Editor
98  *      - This is called by the deselect all operator, as well as other ones!
99  *
100  *      - test: check if select or deselect all
101  *      - sel: how to select keyframes 
102  *              0 = deselect
103  *              1 = select
104  *              2 = invert
105  */
106 static void deselect_graph_keys (bAnimContext *ac, short test, short sel)
107 {
108         ListBase anim_data = {NULL, NULL};
109         bAnimListElem *ale;
110         int filter;
111         
112         BeztEditData bed;
113         BeztEditFunc test_cb, sel_cb;
114         
115         /* determine type-based settings */
116         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_CURVESONLY);
117         
118         /* filter data */
119         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
120         
121         /* init BezTriple looping data */
122         memset(&bed, 0, sizeof(BeztEditData));
123         test_cb= ANIM_editkeyframes_ok(BEZT_OK_SELECTED);
124         
125         /* See if we should be selecting or deselecting */
126         if (test) {
127                 for (ale= anim_data.first; ale; ale= ale->next) {
128                         if (ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, NULL, test_cb, NULL)) {
129                                 sel= SELECT_SUBTRACT;
130                                 break;
131                         }
132                 }
133         }
134         
135         /* convert sel to selectmode, and use that to get editor */
136         sel_cb= ANIM_editkeyframes_select(sel);
137         
138         /* Now set the flags */
139         for (ale= anim_data.first; ale; ale= ale->next) {
140                 FCurve *fcu= (FCurve *)ale->key_data;
141                 
142                 /* Keyframes First */
143                 ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, NULL, sel_cb, NULL);
144                 
145                 /* deactivate the F-Curve, and deselect if deselecting keyframes.
146                  * otherwise select the F-Curve too since we've selected all the keyframes
147                  */
148                 if (sel == SELECT_SUBTRACT) 
149                         fcu->flag &= ~FCURVE_SELECTED;
150                 else
151                         fcu->flag |= FCURVE_SELECTED;
152                 fcu->flag &= ~FCURVE_ACTIVE;
153         }
154         
155         /* Cleanup */
156         BLI_freelistN(&anim_data);
157 }
158
159 /* ------------------- */
160
161 static int graphkeys_deselectall_exec(bContext *C, wmOperator *op)
162 {
163         bAnimContext ac;
164         
165         /* get editor data */
166         if (ANIM_animdata_get_context(C, &ac) == 0)
167                 return OPERATOR_CANCELLED;
168                 
169         /* 'standard' behaviour - check if selected, then apply relevant selection */
170         if (RNA_boolean_get(op->ptr, "invert"))
171                 deselect_graph_keys(&ac, 0, SELECT_INVERT);
172         else
173                 deselect_graph_keys(&ac, 1, SELECT_ADD);
174         
175         /* set notifier that things have changed */
176         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME_SELECT, NULL);
177         
178         return OPERATOR_FINISHED;
179 }
180  
181 void GRAPH_OT_select_all_toggle (wmOperatorType *ot)
182 {
183         /* identifiers */
184         ot->name= "Select All";
185         ot->idname= "GRAPH_OT_select_all_toggle";
186         ot->description= "Toggle selection of all keyframes.";
187         
188         /* api callbacks */
189         ot->exec= graphkeys_deselectall_exec;
190         ot->poll= graphop_visible_keyframes_poll;
191         
192         /* flags */
193         ot->flag= OPTYPE_REGISTER/*|OPTYPE_UNDO*/;
194         
195         /* props */
196         RNA_def_boolean(ot->srna, "invert", 0, "Invert", "");
197 }
198
199 /* ******************** Border Select Operator **************************** */
200 /* This operator currently works in one of three ways:
201  *      -> BKEY         - 1) all keyframes within region are selected (validation with BEZT_OK_REGION)
202  *      -> ALT-BKEY - depending on which axis of the region was larger...
203  *              -> 2) x-axis, so select all frames within frame range (validation with BEZT_OK_FRAMERANGE)
204  *              -> 3) y-axis, so select all frames within channels that region included (validation with BEZT_OK_VALUERANGE)
205  */
206
207 /* Borderselect only selects keyframes now, as overshooting handles often get caught too,
208  * which means that they may be inadvertantly moved as well.
209  * Also, for convenience, handles should get same status as keyframe (if it was within bounds)
210  */
211 static void borderselect_graphkeys (bAnimContext *ac, rcti rect, short mode, short selectmode)
212 {
213         ListBase anim_data = {NULL, NULL};
214         bAnimListElem *ale;
215         int filter;
216         
217         BeztEditData bed;
218         BeztEditFunc ok_cb, select_cb;
219         View2D *v2d= &ac->ar->v2d;
220         rctf rectf;
221         
222         /* convert mouse coordinates to frame ranges and channel coordinates corrected for view pan/zoom */
223         UI_view2d_region_to_view(v2d, rect.xmin, rect.ymin, &rectf.xmin, &rectf.ymin);
224         UI_view2d_region_to_view(v2d, rect.xmax, rect.ymax, &rectf.xmax, &rectf.ymax);
225         
226         /* filter data */
227         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVESONLY | ANIMFILTER_CURVEVISIBLE);
228         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
229         
230         /* get beztriple editing/validation funcs  */
231         select_cb= ANIM_editkeyframes_select(selectmode);
232         ok_cb= ANIM_editkeyframes_ok(mode);
233         
234         /* init editing data */
235         memset(&bed, 0, sizeof(BeztEditData));
236         bed.data= &rectf;
237         
238         /* loop over data, doing border select */
239         for (ale= anim_data.first; ale; ale= ale->next) {
240                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
241                 FCurve *fcu= (FCurve *)ale->key_data;
242
243                 if (adt)
244                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 0, 1);
245
246                 /* set horizontal range (if applicable) */
247                 if (mode != BEZT_OK_VALUERANGE) {
248                         /* if channel is mapped in NLA, apply correction */
249                         if (adt) {
250                                 bed.f1= BKE_nla_tweakedit_remap(adt, rectf.xmin, NLATIME_CONVERT_UNMAP);
251                                 bed.f2= BKE_nla_tweakedit_remap(adt, rectf.xmax, NLATIME_CONVERT_UNMAP);
252                         }
253                         else {
254                                 bed.f1= rectf.xmin;
255                                 bed.f2= rectf.xmax;
256                         }
257                 }
258                 else {
259                         bed.f1= rectf.ymin;
260                         bed.f2= rectf.ymax;
261                 }
262                 
263                 /* select keyframes that are in the appropriate places */
264                 ANIM_fcurve_keys_bezier_loop(&bed, fcu, ok_cb, select_cb, NULL);
265                 
266                 /* select the curve too 
267                  * NOTE: this should really only happen if the curve got touched...
268                  */
269                 if (selectmode == SELECT_ADD) {
270                         fcu->flag |= FCURVE_SELECTED;
271                 }
272
273                 /* un-apply NLA mapping from all the keyframes */
274                 if (adt)
275                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 1, 1);
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         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         RNA_def_enum(ot->srna, "mode", prop_column_select_types, 0, "Mode", "");
535 }
536
537 /* ******************** Mouse-Click Select Operator *********************** */
538 /* This operator works in one of three ways:
539  *      - 1) keyframe under mouse - no special modifiers
540  *      - 2) all keyframes on the same side of current frame indicator as mouse - ALT modifier
541  *      - 3) column select all keyframes in frame under mouse - CTRL modifier
542  *
543  * In addition to these basic options, the SHIFT modifier can be used to toggle the 
544  * selection mode between replacing the selection (without) and inverting the selection (with).
545  */
546
547 /* defines for left-right select tool */
548 static EnumPropertyItem prop_graphkeys_leftright_select_types[] = {
549         {GRAPHKEYS_LRSEL_TEST, "CHECK", 0, "Check if Select Left or Right", ""},
550         {GRAPHKEYS_LRSEL_NONE, "OFF", 0, "Don't select", ""},
551         {GRAPHKEYS_LRSEL_LEFT, "LEFT", 0, "Before current frame", ""},
552         {GRAPHKEYS_LRSEL_RIGHT, "RIGHT", 0, "After current frame", ""},
553         {0, NULL, 0, NULL, NULL}
554 };
555
556 /* ------------------- */
557
558 enum {
559         NEAREST_HANDLE_LEFT     = 0,
560         NEAREST_HANDLE_KEY,
561         NEAREST_HANDLE_RIGHT
562 } eHandleIndex; 
563  
564 /* Find the vertex (either handle (0/2) or the keyframe (1)) that is nearest to the mouse cursor (in area coordinates)  
565  * Selected verts get a disadvantage, to make it easier to select handles behind.
566  * Returns eHandleIndex
567  */
568 static short findnearest_fcurve_vert (bAnimContext *ac, int mval[2], FCurve **fcurve, BezTriple **bezt)
569 {
570         ListBase anim_data = {NULL, NULL};
571         bAnimListElem *ale;
572         int filter;
573         
574         SpaceIpo *sipo= (SpaceIpo *)ac->sa->spacedata.first;
575         View2D *v2d= &ac->ar->v2d;
576         int hpoint=0, sco[3][2];
577         int dist= 100, temp, i;
578         
579         /* clear pointers first */
580         *fcurve= 0;
581         *bezt= 0;
582         
583         /* get curves to search through 
584          *      - if the option to only show keyframes that belong to selected F-Curves is enabled,
585          *        include the 'only selected' flag...
586          */
587         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_CURVESONLY);
588         if (sipo->flag & SIPO_SELCUVERTSONLY) 
589                 filter |= ANIMFILTER_SEL;
590         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
591         
592         for (ale= anim_data.first; ale; ale= ale->next) {
593                 FCurve *fcu= (FCurve *)ale->key_data;
594                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
595                 
596                 /* try to progressively get closer to the right point... */
597                 if (fcu->bezt) {
598                         BezTriple *bezt1=fcu->bezt, *prevbezt=NULL;
599                         
600                         /* apply NLA mapping to all the keyframes */
601                         if (adt)
602                                 ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 0, 1);
603                         
604                         for (i=0; i < fcu->totvert; i++, prevbezt=bezt1, bezt1++) {
605                                 /* convert beztriple points to screen-space */
606                                 UI_view2d_to_region_no_clip(v2d, bezt1->vec[0][0], bezt1->vec[0][1], &sco[0][0], &sco[0][1]);
607                                 UI_view2d_to_region_no_clip(v2d, bezt1->vec[1][0], bezt1->vec[1][1], &sco[1][0], &sco[1][1]);
608                                 UI_view2d_to_region_no_clip(v2d, bezt1->vec[2][0], bezt1->vec[2][1], &sco[2][0], &sco[2][1]);
609                                 
610                                 /* keyframe - do select? */
611                                 temp= abs(mval[0] - sco[1][0]) + abs(mval[1] - sco[1][1]);
612                                 
613                                 if (bezt1->f2 & SELECT) 
614                                         temp += 5;
615                                 
616                                 if (temp < dist) { 
617                                         hpoint= NEAREST_HANDLE_KEY; 
618                                         *bezt= bezt1; 
619                                         dist= temp; 
620                                         *fcurve= fcu; 
621                                 }
622                                 
623                                 /* handles - only do them if they're visible */
624                                 // XXX also need to check for int-values only?
625                                 if ((sipo->flag & SIPO_NOHANDLES)==0) {
626                                         /* first handle only visible if previous segment had handles */
627                                         if ( (!prevbezt && (bezt1->ipo==BEZT_IPO_BEZ)) || (prevbezt && (prevbezt->ipo==BEZT_IPO_BEZ)) )
628                                         {
629                                                 temp= -3 + abs(mval[0] - sco[0][0]) + abs(mval[1] - sco[0][1]);
630                                                 if (bezt1->f1 & SELECT) 
631                                                         temp += 5;
632                                                         
633                                                 if (temp < dist) { 
634                                                         hpoint= NEAREST_HANDLE_LEFT; 
635                                                         *bezt= bezt1; 
636                                                         dist= temp; 
637                                                         *fcurve= fcu; 
638                                                 }
639                                         }
640                                         
641                                         /* second handle only visible if this segment is bezier */
642                                         if (bezt1->ipo == BEZT_IPO_BEZ) 
643                                         {
644                                                 temp= abs(mval[0] - sco[2][0]) + abs(mval[1] - sco[2][1]);
645                                                 if (bezt1->f3 & SELECT) 
646                                                         temp += 5;
647                                                 
648                                                 if (temp < dist) { 
649                                                         hpoint= NEAREST_HANDLE_RIGHT; 
650                                                         *bezt=bezt1; 
651                                                         dist= temp; 
652                                                         *fcurve= fcu; 
653                                                 }
654                                         }
655                                 }
656                         }
657                         
658                         /* un-apply NLA mapping from all the keyframes */
659                         if (adt)
660                                 ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 1, 1);
661                 }
662         }
663         
664         /* free channels */
665         BLI_freelistN(&anim_data);
666         
667         /* return handle */
668         return hpoint;
669 }
670  
671 /* option 1) select keyframe directly under mouse */
672 static void mouse_graph_keys (bAnimContext *ac, int mval[], short select_mode, short curves_only)
673 {
674         FCurve *fcu;
675         BezTriple *bezt;
676         short handle;
677         int filter;
678         
679         /* find the beztriple that we're selecting, and the handle that was clicked on */
680         handle= findnearest_fcurve_vert(ac, mval, &fcu, &bezt);
681         
682         /* check if anything to select */
683         if (fcu == NULL)        
684                 return;
685         
686         /* deselect all other curves? */
687         if (select_mode == SELECT_REPLACE) {
688                 /* reset selection mode */
689                 select_mode= SELECT_ADD;
690                 
691                 /* deselect all other channels and keyframes */
692                 //ANIM_deselect_anim_channels(ac->data, ac->datatype, 0, ACHANNEL_SETFLAG_CLEAR);
693                 deselect_graph_keys(ac, 0, SELECT_SUBTRACT);
694         }
695         
696         /* if points can be selected on this F-Curve */
697         // TODO: what about those with no keyframes?
698         if ((curves_only == 0) && ((fcu->flag & FCURVE_PROTECTED)==0)) {
699                 /* only if there's keyframe */
700                 if (bezt) {
701                         /* depends on selection mode */
702                         if (select_mode == SELECT_INVERT) {
703                                 /* keyframe - invert select of all */
704                                 if (handle == NEAREST_HANDLE_KEY) {
705                                         if (BEZSELECTED(bezt)) {
706                                                 BEZ_DESEL(bezt);
707                                         }
708                                         else {
709                                                 BEZ_SEL(bezt);
710                                         }
711                                 }
712                                 
713                                 /* handles - toggle selection of relevant handle */
714                                 else if (handle == NEAREST_HANDLE_LEFT) {
715                                         /* toggle selection */
716                                         bezt->f1 ^= SELECT;
717                                 }
718                                 else {
719                                         /* toggle selection */
720                                         bezt->f3 ^= SELECT;
721                                 }
722                         }
723                         else {
724                                 /* if the keyframe was clicked on, select all verts of given beztriple */
725                                 if (handle == NEAREST_HANDLE_KEY) {
726                                         BEZ_SEL(bezt);
727                                 }
728                                 /* otherwise, select the handle that applied */
729                                 else if (handle == NEAREST_HANDLE_LEFT) 
730                                         bezt->f1 |= SELECT;
731                                 else 
732                                         bezt->f3 |= SELECT;
733                         }
734                 }
735         }
736         else {
737                 BeztEditFunc select_cb;
738                 BeztEditData bed;
739                 
740                 /* initialise keyframe editing data */
741                 memset(&bed, 0, sizeof(BeztEditData));
742                 
743                 /* set up BezTriple edit callbacks */
744                 select_cb= ANIM_editkeyframes_select(select_mode);
745                 
746                 /* select all keyframes */
747                 ANIM_fcurve_keys_bezier_loop(&bed, fcu, NULL, select_cb, NULL);
748         }
749         
750         /* select or deselect curve? */
751         if (select_mode == SELECT_INVERT)
752                 fcu->flag ^= FCURVE_SELECTED;
753         else if (select_mode == SELECT_ADD)
754                 fcu->flag |= FCURVE_SELECTED;
755                 
756         /* set active F-Curve (NOTE: sync the filter flags with findnearest_fcurve_vert) */
757         if (fcu->flag & FCURVE_SELECTED) {
758                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_CURVESONLY);
759                 ANIM_set_active_channel(ac, ac->data, ac->datatype, filter, fcu, ANIMTYPE_FCURVE);
760         }
761 }
762
763 /* Option 2) Selects all the keyframes on either side of the current frame (depends on which side the mouse is on) */
764 static void graphkeys_mselect_leftright (bAnimContext *ac, short leftright, short select_mode)
765 {
766         ListBase anim_data = {NULL, NULL};
767         bAnimListElem *ale;
768         int filter;
769         
770         BeztEditFunc ok_cb, select_cb;
771         BeztEditData bed;
772         Scene *scene= ac->scene;
773         
774         /* if select mode is replace, deselect all keyframes (and channels) first */
775         if (select_mode==SELECT_REPLACE) {
776                 /* reset selection mode to add to selection */
777                 select_mode= SELECT_ADD;
778                 
779                 /* deselect all other channels and keyframes */
780                 ANIM_deselect_anim_channels(ac->data, ac->datatype, 0, ACHANNEL_SETFLAG_CLEAR);
781                 deselect_graph_keys(ac, 0, SELECT_SUBTRACT);
782         }
783         
784         /* set callbacks and editing data */
785         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAMERANGE);
786         select_cb= ANIM_editkeyframes_select(select_mode);
787         
788         memset(&bed, 0, sizeof(BeztEditFunc));
789         if (leftright == GRAPHKEYS_LRSEL_LEFT) {
790                 bed.f1 = MINAFRAMEF;
791                 bed.f2 = (float)(CFRA + 0.1f);
792         } 
793         else {
794                 bed.f1 = (float)(CFRA - 0.1f);
795                 bed.f2 = MAXFRAMEF;
796         }
797         
798         /* filter data */
799         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_CURVESONLY);
800         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
801                 
802         /* select keys on the side where most data occurs */
803         for (ale= anim_data.first; ale; ale= ale->next) {
804                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
805                 
806                 if (adt) {
807                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 0, 1);
808                         ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, ok_cb, select_cb, NULL);
809                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 1, 1);
810                 }
811                 else
812                         ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, ok_cb, select_cb, NULL);
813         }
814         
815         /* Cleanup */
816         BLI_freelistN(&anim_data);
817 }
818
819 /* Option 3) Selects all visible keyframes in the same frame as the mouse click */
820 static void graphkeys_mselect_column (bAnimContext *ac, int mval[2], short select_mode)
821 {
822         ListBase anim_data= {NULL, NULL};
823         bAnimListElem *ale;
824         int filter;
825         
826         BeztEditFunc select_cb, ok_cb;
827         BeztEditData bed;
828         FCurve *fcu;
829         BezTriple *bezt;
830         float selx = (float)ac->scene->r.cfra;
831         
832         /* find the beztriple that occurs on this frame, and use his as the frame number we're using */
833         findnearest_fcurve_vert(ac, mval, &fcu, &bezt);
834         
835         /* check if anything to select */
836         if (ELEM(NULL, fcu, bezt))      
837                 return;
838         selx= bezt->vec[1][0];
839         
840         /* if select mode is replace, deselect all keyframes (and channels) first */
841         if (select_mode==SELECT_REPLACE) {
842                 /* reset selection mode to add to selection */
843                 select_mode= SELECT_ADD;
844                 
845                 /* deselect all other channels and keyframes */
846                 ANIM_deselect_anim_channels(ac->data, ac->datatype, 0, ACHANNEL_SETFLAG_CLEAR);
847                 deselect_graph_keys(ac, 0, SELECT_SUBTRACT);
848         }
849         
850         /* initialise keyframe editing data */
851         memset(&bed, 0, sizeof(BeztEditData));
852         
853         /* set up BezTriple edit callbacks */
854         select_cb= ANIM_editkeyframes_select(select_mode);
855         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAME);
856         
857         /* loop through all of the keys and select additional keyframes
858          * based on the keys found to be selected above
859          */
860         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_CURVESONLY);
861         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
862         
863         for (ale= anim_data.first; ale; ale= ale->next) {
864                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
865                 
866                 /* set frame for validation callback to refer to */
867                 if (adt)
868                         bed.f1= BKE_nla_tweakedit_remap(adt, selx, NLATIME_CONVERT_UNMAP);
869                 else
870                         bed.f1= selx;
871                 
872                 /* select elements with frame number matching cfra */
873                 ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, ok_cb, select_cb, NULL);
874         }
875         
876         /* free elements */
877         BLI_freelistN(&bed.list);
878         BLI_freelistN(&anim_data);
879 }
880  
881 /* ------------------- */
882
883 /* handle clicking */
884 static int graphkeys_clickselect_invoke(bContext *C, wmOperator *op, wmEvent *event)
885 {
886         bAnimContext ac;
887         Scene *scene;
888         ARegion *ar;
889         View2D *v2d;
890         short selectmode;
891         int mval[2];
892         
893         /* get editor data */
894         if (ANIM_animdata_get_context(C, &ac) == 0)
895                 return OPERATOR_CANCELLED;
896         
897         /* get useful pointers from animation context data */
898         scene= ac.scene;
899         ar= ac.ar;
900         v2d= &ar->v2d;
901         
902         /* get mouse coordinates (in region coordinates) */
903         mval[0]= (event->x - ar->winrct.xmin);
904         mval[1]= (event->y - ar->winrct.ymin);
905         
906         /* select mode is either replace (deselect all, then add) or add/extend */
907         if (RNA_boolean_get(op->ptr, "extend"))
908                 selectmode= SELECT_INVERT;
909         else
910                 selectmode= SELECT_REPLACE;
911         
912         /* figure out action to take */
913         if (RNA_enum_get(op->ptr, "left_right")) {
914                 /* select all keys on same side of current frame as mouse */
915                 float x;
916                 
917                 UI_view2d_region_to_view(v2d, mval[0], mval[1], &x, NULL);
918                 if (x < CFRA)
919                         RNA_int_set(op->ptr, "left_right", GRAPHKEYS_LRSEL_LEFT);
920                 else    
921                         RNA_int_set(op->ptr, "left_right", GRAPHKEYS_LRSEL_RIGHT);
922                 
923                 graphkeys_mselect_leftright(&ac, RNA_enum_get(op->ptr, "left_right"), selectmode);
924         }
925         else if (RNA_boolean_get(op->ptr, "column")) {
926                 /* select all keyframes in the same frame as the one that was under the mouse */
927                 graphkeys_mselect_column(&ac, mval, selectmode);
928         }
929         else if (RNA_boolean_get(op->ptr, "curves")) {
930                 /* select all keyframes in F-Curve under mouse */
931                 mouse_graph_keys(&ac, mval, selectmode, 1);
932         }
933         else {
934                 /* select keyframe under mouse */
935                 mouse_graph_keys(&ac, mval, selectmode, 0);
936         }
937         
938         /* set notifier that keyframe selection (and also channel selection in some cases) has changed */
939         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME_SELECT|ND_ANIMCHAN_SELECT, NULL);
940         
941         /* for tweak grab to work */
942         return OPERATOR_FINISHED|OPERATOR_PASS_THROUGH;
943 }
944  
945 void GRAPH_OT_clickselect (wmOperatorType *ot)
946 {
947         /* identifiers */
948         ot->name= "Mouse Select Keys";
949         ot->idname= "GRAPH_OT_clickselect";
950         ot->description= "Select keyframes by clicking on them.";
951         
952         /* api callbacks */
953         ot->invoke= graphkeys_clickselect_invoke;
954         ot->poll= graphop_visible_keyframes_poll;
955         
956         /* id-props */
957         // XXX should we make this into separate operators?
958         RNA_def_enum(ot->srna, "left_right", prop_graphkeys_leftright_select_types, 0, "Left Right", ""); // CTRLKEY
959         RNA_def_boolean(ot->srna, "extend", 0, "Extend Select", ""); // SHIFTKEY
960         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
961         RNA_def_boolean(ot->srna, "curves", 0, "Only Curves", "Select all the keyframes in the curve"); // CTRLKEY + ALTKEY
962 }
963
964 /* ************************************************************************** */