2.50: svn merge https://svn.blender.org/svnroot/bf-blender/trunk/blender -r19323...
[blender.git] / source / blender / editors / space_graph / graph_select.c
1 /**
2  * $Id: editaction.c 17746 2008-12-08 11:19:44Z aligorith $
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_arithb.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_object.h"
67 #include "BKE_context.h"
68 #include "BKE_utildefines.h"
69
70 #include "UI_view2d.h"
71
72 #include "ED_anim_api.h"
73 #include "ED_keyframing.h"
74 #include "ED_keyframes_draw.h"
75 #include "ED_keyframes_edit.h"
76 #include "ED_screen.h"
77 #include "ED_space_api.h"
78
79 #include "WM_api.h"
80 #include "WM_types.h"
81
82 #include "graph_intern.h"
83
84
85 /* ************************************************************************** */
86 /* KEYFRAMES STUFF */
87
88 /* ******************** Deselect All Operator ***************************** */
89 /* This operator works in one of three ways:
90  *      1) (de)select all (AKEY) - test if select all or deselect all
91  *      2) invert all (CTRL-IKEY) - invert selection of all keyframes
92  *      3) (de)select all - no testing is done; only for use internal tools as normal function...
93  */
94
95 /* Deselects keyframes in the Graph Editor
96  *      - This is called by the deselect all operator, as well as other ones!
97  *
98  *      - test: check if select or deselect all
99  *      - sel: how to select keyframes 
100  *              0 = deselect
101  *              1 = select
102  *              2 = invert
103  */
104 static void deselect_graph_keys (bAnimContext *ac, short test, short sel)
105 {
106         ListBase anim_data = {NULL, NULL};
107         bAnimListElem *ale;
108         int filter;
109         
110         BeztEditData bed;
111         BeztEditFunc test_cb, sel_cb;
112         
113         /* determine type-based settings */
114         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_CURVESONLY);
115         
116         /* filter data */
117         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
118         
119         /* init BezTriple looping data */
120         memset(&bed, 0, sizeof(BeztEditData));
121         test_cb= ANIM_editkeyframes_ok(BEZT_OK_SELECTED);
122         
123         /* See if we should be selecting or deselecting */
124         if (test) {
125                 for (ale= anim_data.first; ale; ale= ale->next) {
126                         if (ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, NULL, test_cb, NULL)) {
127                                 sel= SELECT_SUBTRACT;
128                                 break;
129                         }
130                 }
131         }
132         
133         /* convert sel to selectmode, and use that to get editor */
134         sel_cb= ANIM_editkeyframes_select(sel);
135         
136         /* Now set the flags */
137         for (ale= anim_data.first; ale; ale= ale->next) {
138                 FCurve *fcu= (FCurve *)ale->key_data;
139                 
140                 /* Keyframes First */
141                 ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, NULL, sel_cb, NULL);
142                 
143                 /* deactivate the F-Curve */
144                 fcu->flag &= ~FCURVE_ACTIVE;
145         }
146         
147         /* Cleanup */
148         BLI_freelistN(&anim_data);
149 }
150
151 /* ------------------- */
152
153 static int graphkeys_deselectall_exec(bContext *C, wmOperator *op)
154 {
155         bAnimContext ac;
156         
157         /* get editor data */
158         if (ANIM_animdata_get_context(C, &ac) == 0)
159                 return OPERATOR_CANCELLED;
160                 
161         /* 'standard' behaviour - check if selected, then apply relevant selection */
162         if (RNA_boolean_get(op->ptr, "invert"))
163                 deselect_graph_keys(&ac, 0, SELECT_INVERT);
164         else
165                 deselect_graph_keys(&ac, 1, SELECT_ADD);
166         
167         /* set notifier that things have changed */
168         ED_area_tag_redraw(CTX_wm_area(C)); // FIXME... should be updating 'keyframes' data context or so instead!
169         
170         return OPERATOR_FINISHED;
171 }
172  
173 void GRAPHEDIT_OT_keyframes_select_all_toggle (wmOperatorType *ot)
174 {
175         /* identifiers */
176         ot->name= "Select All";
177         ot->idname= "GRAPHEDIT_OT_keyframes_select_all_toggle";
178         
179         /* api callbacks */
180         ot->exec= graphkeys_deselectall_exec;
181         ot->poll= ED_operator_areaactive;
182         
183         /* flags */
184         ot->flag= OPTYPE_REGISTER/*|OPTYPE_UNDO*/;
185         
186         /* props */
187         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         BeztEditData bed;
209         BeztEditFunc ok_cb, select_cb;
210         View2D *v2d= &ac->ar->v2d;
211         rctf rectf;
212         
213         /* convert mouse coordinates to frame ranges and channel coordinates corrected for view pan/zoom */
214         UI_view2d_region_to_view(v2d, rect.xmin, rect.ymin, &rectf.xmin, &rectf.ymin);
215         UI_view2d_region_to_view(v2d, rect.xmax, rect.ymax, &rectf.xmax, &rectf.ymax);
216         
217         /* filter data */
218         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVESONLY | ANIMFILTER_CURVEVISIBLE);
219         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
220         
221         /* get beztriple editing/validation funcs  */
222         select_cb= ANIM_editkeyframes_select(selectmode);
223         ok_cb= ANIM_editkeyframes_ok(mode);
224         
225         /* init editing data */
226         memset(&bed, 0, sizeof(BeztEditData));
227         bed.data= &rectf;
228         
229         /* loop over data, doing border select */
230         for (ale= anim_data.first; ale; ale= ale->next) {
231                 Object *nob= ANIM_nla_mapping_get(ac, ale);
232                 
233                 /* set horizontal range (if applicable) */
234                 if (mode != BEZT_OK_VALUERANGE) {
235                         /* if channel is mapped in NLA, apply correction */
236                         if (nob) {
237                                 bed.f1= get_action_frame(nob, rectf.xmin);
238                                 bed.f2= get_action_frame(nob, rectf.xmax);
239                         }
240                         else {
241                                 bed.f1= rectf.xmin;
242                                 bed.f2= rectf.xmax;
243                         }
244                 }
245                 else {
246                         bed.f1= rectf.ymin;
247                         bed.f2= rectf.ymax;
248                 }
249                 
250                 /* select keyframes that are in the appropriate places */
251                 ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, ok_cb, select_cb, NULL);
252         }
253         
254         /* cleanup */
255         BLI_freelistN(&anim_data);
256 }
257
258 /* ------------------- */
259
260 static int graphkeys_borderselect_exec(bContext *C, wmOperator *op)
261 {
262         bAnimContext ac;
263         rcti rect;
264         short mode=0, selectmode=0;
265         int event;
266         
267         /* get editor data */
268         if (ANIM_animdata_get_context(C, &ac) == 0)
269                 return OPERATOR_CANCELLED;
270         
271         /* get settings from operator */
272         rect.xmin= RNA_int_get(op->ptr, "xmin");
273         rect.ymin= RNA_int_get(op->ptr, "ymin");
274         rect.xmax= RNA_int_get(op->ptr, "xmax");
275         rect.ymax= RNA_int_get(op->ptr, "ymax");
276                 
277         event= RNA_int_get(op->ptr, "event_type");
278         if (event == LEFTMOUSE) // FIXME... hardcoded
279                 selectmode = SELECT_ADD;
280         else
281                 selectmode = SELECT_SUBTRACT;
282         
283         /* selection 'mode' depends on whether borderselect region only matters on one axis */
284         if (RNA_boolean_get(op->ptr, "axis_range")) {
285                 /* mode depends on which axis of the range is larger to determine which axis to use 
286                  *      - checking this in region-space is fine, as it's fundamentally still going to be a different rect size
287                  *      - the frame-range select option is favoured over the channel one (x over y), as frame-range one is often
288                  *        used for tweaking timing when "blocking", while channels is not that useful...
289                  */
290                 if ((rect.xmax - rect.xmin) >= (rect.ymax - rect.ymin))
291                         mode= BEZT_OK_FRAMERANGE;
292                 else
293                         mode= BEZT_OK_VALUERANGE;
294         }
295         else 
296                 mode= BEZT_OK_REGION;
297         
298         /* apply borderselect action */
299         borderselect_graphkeys(&ac, rect, mode, selectmode);
300         
301         return OPERATOR_FINISHED;
302
303
304 void GRAPHEDIT_OT_keyframes_select_border(wmOperatorType *ot)
305 {
306         /* identifiers */
307         ot->name= "Border Select";
308         ot->idname= "GRAPHEDIT_OT_keyframes_select_border";
309         
310         /* api callbacks */
311         ot->invoke= WM_border_select_invoke;
312         ot->exec= graphkeys_borderselect_exec;
313         ot->modal= WM_border_select_modal;
314         
315         ot->poll= ED_operator_areaactive;
316         
317         /* flags */
318         ot->flag= OPTYPE_REGISTER/*|OPTYPE_UNDO*/;
319         
320         /* rna */
321         RNA_def_int(ot->srna, "event_type", 0, INT_MIN, INT_MAX, "Event Type", "", INT_MIN, INT_MAX);
322         RNA_def_int(ot->srna, "xmin", 0, INT_MIN, INT_MAX, "X Min", "", INT_MIN, INT_MAX);
323         RNA_def_int(ot->srna, "xmax", 0, INT_MIN, INT_MAX, "X Max", "", INT_MIN, INT_MAX);
324         RNA_def_int(ot->srna, "ymin", 0, INT_MIN, INT_MAX, "Y Min", "", INT_MIN, INT_MAX);
325         RNA_def_int(ot->srna, "ymax", 0, INT_MIN, INT_MAX, "Y Max", "", INT_MIN, INT_MAX);
326         
327         RNA_def_boolean(ot->srna, "axis_range", 0, "Axis Range", "");
328 }
329
330 /* ******************** Column Select Operator **************************** */
331 /* This operator works in one of four ways:
332  *      - 1) select all keyframes in the same frame as a selected one  (KKEY)
333  *      - 2) select all keyframes in the same frame as the current frame marker (CTRL-KKEY)
334  *      - 3) select all keyframes in the same frame as a selected markers (SHIFT-KKEY)
335  *      - 4) select all keyframes that occur between selected markers (ALT-KKEY)
336  */
337
338 /* defines for column-select mode */
339 static EnumPropertyItem prop_column_select_types[] = {
340         {GRAPHKEYS_COLUMNSEL_KEYS, "KEYS", "On Selected Keyframes", ""},
341         {GRAPHKEYS_COLUMNSEL_CFRA, "CFRA", "On Current Frame", ""},
342         {GRAPHKEYS_COLUMNSEL_MARKERS_COLUMN, "MARKERS_COLUMN", "On Selected Markers", ""},
343         {GRAPHKEYS_COLUMNSEL_MARKERS_BETWEEN, "MARKERS_BETWEEN", "Between Min/Max Selected Markers", ""},
344         {0, NULL, NULL, NULL}
345 };
346
347 /* ------------------- */ 
348
349 /* Selects all visible keyframes between the specified markers */
350 static void markers_selectkeys_between (bAnimContext *ac)
351 {
352         ListBase anim_data = {NULL, NULL};
353         bAnimListElem *ale;
354         int filter;
355         
356         BeztEditFunc select_cb;
357         BeztEditData bed;
358         float min, max;
359         
360         /* get extreme markers */
361         //get_minmax_markers(1, &min, &max); // FIXME... add back markers api!
362         min= (float)ac->scene->r.sfra; // xxx temp code
363         max= (float)ac->scene->r.efra; // xxx temp code
364         
365         if (min==max) return;
366         min -= 0.5f;
367         max += 0.5f;
368         
369         /* get editing funcs + data */
370         select_cb= ANIM_editkeyframes_select(SELECT_ADD);
371         memset(&bed, 0, sizeof(BeztEditData));
372         bed.f1= min; 
373         bed.f2= max;
374         
375         /* filter data */
376         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_CURVESONLY);
377         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
378         
379         /* select keys in-between */
380         for (ale= anim_data.first; ale; ale= ale->next) {
381                 Object *nob= ANIM_nla_mapping_get(ac, ale);
382                 
383                 if (nob) {      
384                         ANIM_nla_mapping_apply_fcurve(nob, ale->key_data, 0, 1);
385                         ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, NULL, select_cb, NULL);
386                         ANIM_nla_mapping_apply_fcurve(nob, ale->key_data, 1, 1);
387                 }
388                 else {
389                         ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, NULL, select_cb, NULL);
390                 }
391         }
392         
393         /* Cleanup */
394         BLI_freelistN(&anim_data);
395 }
396
397
398 /* helper callback for columnselect_graph_keys() -> populate list CfraElems with frame numbers from selected beztriples */
399 // TODO: if some other code somewhere needs this, it'll be time to port this over to keyframes_edit.c!!!
400 static short bezt_to_cfraelem(BeztEditData *bed, BezTriple *bezt)
401 {
402         /* only if selected */
403         if (bezt->f2 & SELECT) {
404                 CfraElem *ce= MEM_callocN(sizeof(CfraElem), "cfraElem");
405                 BLI_addtail(&bed->list, ce);
406                 
407                 ce->cfra= bezt->vec[1][0];
408         }
409         
410         return 0;
411 }
412
413 /* Selects all visible keyframes in the same frames as the specified elements */
414 static void columnselect_graph_keys (bAnimContext *ac, short mode)
415 {
416         ListBase anim_data= {NULL, NULL};
417         bAnimListElem *ale;
418         int filter;
419         
420         Scene *scene= ac->scene;
421         CfraElem *ce;
422         BeztEditFunc select_cb, ok_cb;
423         BeztEditData bed;
424         
425         /* initialise keyframe editing data */
426         memset(&bed, 0, sizeof(BeztEditData));
427         
428         /* build list of columns */
429         switch (mode) {
430                 case GRAPHKEYS_COLUMNSEL_KEYS: /* list of selected keys */
431                         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_CURVESONLY);
432                         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
433                         
434                         for (ale= anim_data.first; ale; ale= ale->next)
435                                 ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, NULL, bezt_to_cfraelem, NULL);
436                         
437                         BLI_freelistN(&anim_data);
438                         break;
439                         
440                 case GRAPHKEYS_COLUMNSEL_CFRA: /* current frame */
441                         /* make a single CfraElem for storing this */
442                         ce= MEM_callocN(sizeof(CfraElem), "cfraElem");
443                         BLI_addtail(&bed.list, ce);
444                         
445                         ce->cfra= (float)CFRA;
446                         break;
447                         
448                 case GRAPHKEYS_COLUMNSEL_MARKERS_COLUMN: /* list of selected markers */
449                         // FIXME: markers api needs to be improved for this first!
450                         //make_marker_cfra_list(&elems, 1);
451                         return; // XXX currently, this does nothing!
452                         break;
453                         
454                 default: /* invalid option */
455                         return;
456         }
457         
458         /* set up BezTriple edit callbacks */
459         select_cb= ANIM_editkeyframes_select(SELECT_ADD);
460         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAME);
461         
462         /* loop through all of the keys and select additional keyframes
463          * based on the keys found to be selected above
464          */
465         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_CURVESONLY);
466         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
467         
468         for (ale= anim_data.first; ale; ale= ale->next) {
469                 Object *nob= ANIM_nla_mapping_get(ac, ale);
470                 
471                 /* loop over cfraelems (stored in the BeztEditData->list)
472                  *      - we need to do this here, as we can apply fewer NLA-mapping conversions
473                  */
474                 for (ce= bed.list.first; ce; ce= ce->next) {
475                         /* set frame for validation callback to refer to */
476                         if (nob)
477                                 bed.f1= get_action_frame(nob, ce->cfra);
478                         else
479                                 bed.f1= ce->cfra;
480                         
481                         /* select elements with frame number matching cfraelem */
482                         ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, ok_cb, select_cb, NULL);
483                 }
484         }
485         
486         /* free elements */
487         BLI_freelistN(&bed.list);
488         BLI_freelistN(&anim_data);
489 }
490
491 /* ------------------- */
492
493 static int graphkeys_columnselect_exec(bContext *C, wmOperator *op)
494 {
495         bAnimContext ac;
496         short mode;
497         
498         /* get editor data */
499         if (ANIM_animdata_get_context(C, &ac) == 0)
500                 return OPERATOR_CANCELLED;
501                 
502         /* action to take depends on the mode */
503         mode= RNA_enum_get(op->ptr, "mode");
504         
505         if (mode == GRAPHKEYS_COLUMNSEL_MARKERS_BETWEEN)
506                 markers_selectkeys_between(&ac);
507         else
508                 columnselect_graph_keys(&ac, mode);
509         
510         /* set notifier that things have changed */
511         ANIM_animdata_send_notifiers(C, &ac, ANIM_CHANGED_KEYFRAMES_SELECT);
512         
513         return OPERATOR_FINISHED;
514 }
515  
516 void GRAPHEDIT_OT_keyframes_columnselect (wmOperatorType *ot)
517 {
518         /* identifiers */
519         ot->name= "Select All";
520         ot->idname= "GRAPHEDIT_OT_keyframes_columnselect";
521         
522         /* api callbacks */
523         ot->exec= graphkeys_columnselect_exec;
524         ot->poll= ED_operator_areaactive;
525         
526         /* flags */
527         ot->flag= OPTYPE_REGISTER/*|OPTYPE_UNDO*/;
528         
529         /* props */
530         RNA_def_enum(ot->srna, "mode", prop_column_select_types, 0, "Mode", "");
531 }
532
533 /* ******************** Mouse-Click Select Operator *********************** */
534 /* This operator works in one of three ways:
535  *      - 1) keyframe under mouse - no special modifiers
536  *      - 2) all keyframes on the same side of current frame indicator as mouse - ALT modifier
537  *      - 3) column select all keyframes in frame under mouse - CTRL modifier
538  *
539  * In addition to these basic options, the SHIFT modifier can be used to toggle the 
540  * selection mode between replacing the selection (without) and inverting the selection (with).
541  */
542
543 /* defines for left-right select tool */
544 static EnumPropertyItem prop_graphkeys_leftright_select_types[] = {
545         {GRAPHKEYS_LRSEL_TEST, "CHECK", "Check if Select Left or Right", ""},
546         {GRAPHKEYS_LRSEL_NONE, "OFF", "Don't select", ""},
547         {GRAPHKEYS_LRSEL_LEFT, "LEFT", "Before current frame", ""},
548         {GRAPHKEYS_LRSEL_RIGHT, "RIGHT", "After current frame", ""},
549         {0, NULL, NULL, NULL}
550 };
551
552 /* ------------------- */
553
554 enum {
555         NEAREST_HANDLE_LEFT     = 0,
556         NEAREST_HANDLE_KEY,
557         NEAREST_HANDLE_RIGHT
558 } eHandleIndex; 
559  
560 /* Find the vertex (either handle (0/2) or the keyframe (1)) that is nearest to the mouse cursor (in area coordinates)  
561  * Selected verts get a disadvantage, to make it easier to select handles behind.
562  * Returns eHandleIndex
563  */
564 static short findnearest_fcurve_vert (bAnimContext *ac, int mval[2], FCurve **fcurve, BezTriple **bezt)
565 {
566         ListBase anim_data = {NULL, NULL};
567         bAnimListElem *ale;
568         int filter;
569         
570         SpaceIpo *sipo= (SpaceIpo *)ac->sa->spacedata.first;
571         View2D *v2d= &ac->ar->v2d;
572         int hpoint=0, sco[3][2];
573         int dist= 100, temp, i;
574         
575         /* clear pointers first */
576         *fcurve= 0;
577         *bezt= 0;
578         
579         /* get curves to search through */
580         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_CURVESONLY);
581         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
582         
583         for (ale= anim_data.first; ale; ale= ale->next) {
584                 FCurve *fcu= (FCurve *)ale->key_data;
585                 
586                 /* try to progressively get closer to the right point... */
587                 if (fcu->bezt) {
588                         BezTriple *bezt1=fcu->bezt, *prevbezt=NULL;
589                         
590                         for (i=0; i < fcu->totvert; i++, prevbezt=bezt1, bezt1++) {
591                                 /* convert beztriple points to screen-space */
592                                 UI_view2d_to_region_no_clip(v2d, bezt1->vec[0][0], bezt1->vec[0][1], &sco[0][0], &sco[0][1]);
593                                 UI_view2d_to_region_no_clip(v2d, bezt1->vec[1][0], bezt1->vec[1][1], &sco[1][0], &sco[1][1]);
594                                 UI_view2d_to_region_no_clip(v2d, bezt1->vec[2][0], bezt1->vec[2][1], &sco[2][0], &sco[2][1]);
595                                 
596                                 /* keyframe - do select? */
597                                 temp= abs(mval[0] - sco[1][0]) + abs(mval[1] - sco[1][1]);
598                                 
599                                 if (bezt1->f2 & SELECT) 
600                                         temp += 5;
601                                 
602                                 if (temp < dist) { 
603                                         hpoint= NEAREST_HANDLE_KEY; 
604                                         *bezt= bezt1; 
605                                         dist= temp; 
606                                         *fcurve= fcu; 
607                                 }
608                                 
609                                 /* handles - only do them if they're visible */
610                                 // XXX also need to check for int-values only?
611                                 if ((sipo->flag & SIPO_NOHANDLES)==0) {
612                                         /* first handle only visible if previous segment had handles */
613                                         if ( (!prevbezt && (bezt1->ipo==BEZT_IPO_BEZ)) || (prevbezt && (prevbezt->ipo==BEZT_IPO_BEZ)) )
614                                         {
615                                                 temp= -3 + abs(mval[0] - sco[0][0]) + abs(mval[1] - sco[0][1]);
616                                                 if (bezt1->f1 & SELECT) 
617                                                         temp += 5;
618                                                         
619                                                 if (temp < dist) { 
620                                                         hpoint= NEAREST_HANDLE_LEFT; 
621                                                         *bezt= bezt1; 
622                                                         dist= temp; 
623                                                         *fcurve= fcu; 
624                                                 }
625                                         }
626                                         
627                                         /* second handle only visible if this segment is bezier */
628                                         if (bezt1->ipo == BEZT_IPO_BEZ) 
629                                         {
630                                                 temp= abs(mval[0] - sco[2][0]) + abs(mval[1] - sco[2][1]);
631                                                 if (bezt1->f3 & SELECT) 
632                                                         temp += 5;
633                                                 
634                                                 if (temp < dist) { 
635                                                         hpoint= NEAREST_HANDLE_RIGHT; 
636                                                         *bezt=bezt1; 
637                                                         dist= temp; 
638                                                         *fcurve= fcu; 
639                                                 }
640                                         }
641                                 }
642                         }
643                 }
644         }
645         
646         /* free channels */
647         BLI_freelistN(&anim_data);
648         
649         /* return handle */
650         return hpoint;
651 }
652  
653 /* option 1) select keyframe directly under mouse */
654 static void mouse_graph_keys (bAnimContext *ac, int mval[], short select_mode, short curves_only)
655 {
656         FCurve *fcu;
657         BezTriple *bezt;
658         short handle;
659         int filter;
660         
661         /* find the beztriple that we're selecting, and the handle that was clicked on */
662         handle= findnearest_fcurve_vert(ac, mval, &fcu, &bezt);
663         
664         /* check if anything to select */
665         if (fcu == NULL)        
666                 return;
667         
668         /* deselect all other curves? */
669         if (select_mode == SELECT_REPLACE) {
670                 /* reset selection mode */
671                 select_mode= SELECT_ADD;
672                 
673                 /* deselect all other channels and keyframes */
674                 //ANIM_deselect_anim_channels(ac->data, ac->datatype, 0, ACHANNEL_SETFLAG_CLEAR);
675                 deselect_graph_keys(ac, 0, SELECT_SUBTRACT);
676         }
677         
678         /* if points can be selected on this F-Curve */
679         // TODO: what about those with no keyframes?
680         if ((curves_only == 0) && ((fcu->flag & FCURVE_PROTECTED)==0)) {
681                 /* only if there's keyframe */
682                 if (bezt) {
683                         /* depends on selection mode */
684                         if (select_mode == SELECT_INVERT) {
685                                 /* keyframe - invert select of all */
686                                 if (handle == NEAREST_HANDLE_KEY) {
687                                         if (BEZSELECTED(bezt)) {
688                                                 BEZ_DESEL(bezt);
689                                         }
690                                         else {
691                                                 BEZ_SEL(bezt);
692                                         }
693                                 }
694                                 
695                                 /* handles - toggle selection of relevant handle */
696                                 else if (handle == NEAREST_HANDLE_LEFT) {
697                                         /* toggle selection */
698                                         bezt->f1 ^= SELECT;
699                                 }
700                                 else {
701                                         /* toggle selection */
702                                         bezt->f3 ^= SELECT;
703                                 }
704                         }
705                         else {
706                                 /* if the keyframe was clicked on, select all verts of given beztriple */
707                                 if (handle == NEAREST_HANDLE_KEY) {
708                                         BEZ_SEL(bezt);
709                                 }
710                                 /* otherwise, select the handle that applied */
711                                 else if (handle == NEAREST_HANDLE_LEFT) 
712                                         bezt->f1 |= SELECT;
713                                 else 
714                                         bezt->f3 |= SELECT;
715                         }
716                 }
717         }
718         else {
719                 BeztEditFunc select_cb;
720                 BeztEditData bed;
721         
722                 /* initialise keyframe editing data */
723                 memset(&bed, 0, sizeof(BeztEditData));
724                 
725                 /* set up BezTriple edit callbacks */
726                 select_cb= ANIM_editkeyframes_select(select_mode);
727                 
728                 /* select all keyframes */
729                 ANIM_fcurve_keys_bezier_loop(&bed, fcu, NULL, select_cb, NULL);
730         }
731         
732         /* select or deselect curve? */
733         if (select_mode == SELECT_INVERT)
734                 fcu->flag ^= FCURVE_SELECTED;
735         else if (select_mode == SELECT_ADD)
736                 fcu->flag |= FCURVE_SELECTED;
737                 
738         /* set active F-Curve (NOTE: sync the filter flags with findnearest_fcurve_vert) */
739         if (fcu->flag & FCURVE_SELECTED) {
740                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_CURVESONLY);
741                 ANIM_set_active_channel(ac->data, ac->datatype, filter, fcu, ANIMTYPE_FCURVE);
742         }
743 }
744
745 /* Option 2) Selects all the keyframes on either side of the current frame (depends on which side the mouse is on) */
746 static void graphkeys_mselect_leftright (bAnimContext *ac, short leftright, short select_mode)
747 {
748         ListBase anim_data = {NULL, NULL};
749         bAnimListElem *ale;
750         int filter;
751         
752         BeztEditFunc ok_cb, select_cb;
753         BeztEditData bed;
754         Scene *scene= ac->scene;
755         
756         /* if select mode is replace, deselect all keyframes (and channels) first */
757         if (select_mode==SELECT_REPLACE) {
758                 /* reset selection mode to add to selection */
759                 select_mode= SELECT_ADD;
760                 
761                 /* deselect all other channels and keyframes */
762                 ANIM_deselect_anim_channels(ac->data, ac->datatype, 0, ACHANNEL_SETFLAG_CLEAR);
763                 deselect_graph_keys(ac, 0, SELECT_SUBTRACT);
764         }
765         
766         /* set callbacks and editing data */
767         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAMERANGE);
768         select_cb= ANIM_editkeyframes_select(select_mode);
769         
770         memset(&bed, 0, sizeof(BeztEditFunc));
771         if (leftright == GRAPHKEYS_LRSEL_LEFT) {
772                 bed.f1 = -MAXFRAMEF;
773                 bed.f2 = (float)(CFRA + 0.1f);
774         } 
775         else {
776                 bed.f1 = (float)(CFRA - 0.1f);
777                 bed.f2 = MAXFRAMEF;
778         }
779         
780         /* filter data */
781         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_CURVESONLY);
782         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
783                 
784         /* select keys on the side where most data occurs */
785         for (ale= anim_data.first; ale; ale= ale->next) {
786                 Object *nob= ANIM_nla_mapping_get(ac, ale);
787                 
788                 if (nob) {
789                         ANIM_nla_mapping_apply_fcurve(nob, ale->key_data, 0, 1);
790                         ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, ok_cb, select_cb, NULL);
791                         ANIM_nla_mapping_apply_fcurve(nob, ale->key_data, 1, 1);
792                 }
793                 else
794                         ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, ok_cb, select_cb, NULL);
795         }
796         
797         /* Cleanup */
798         BLI_freelistN(&anim_data);
799 }
800
801 /* Option 3) Selects all visible keyframes in the same frame as the mouse click */
802 static void graphkeys_mselect_column (bAnimContext *ac, int mval[2], short select_mode)
803 {
804         ListBase anim_data= {NULL, NULL};
805         bAnimListElem *ale;
806         int filter;
807         
808         BeztEditFunc select_cb, ok_cb;
809         BeztEditData bed;
810         FCurve *fcu;
811         BezTriple *bezt;
812         float selx = (float)ac->scene->r.cfra;
813         
814         /* find the beztriple that occurs on this frame, and use his as the frame number we're using */
815         findnearest_fcurve_vert(ac, mval, &fcu, &bezt);
816         
817         /* check if anything to select */
818         if (ELEM(NULL, fcu, bezt))      
819                 return;
820         selx= bezt->vec[1][0];
821         
822         /* if select mode is replace, deselect all keyframes (and channels) first */
823         if (select_mode==SELECT_REPLACE) {
824                 /* reset selection mode to add to selection */
825                 select_mode= SELECT_ADD;
826                 
827                 /* deselect all other channels and keyframes */
828                 ANIM_deselect_anim_channels(ac->data, ac->datatype, 0, ACHANNEL_SETFLAG_CLEAR);
829                 deselect_graph_keys(ac, 0, SELECT_SUBTRACT);
830         }
831         
832         /* initialise keyframe editing data */
833         memset(&bed, 0, sizeof(BeztEditData));
834         
835         /* set up BezTriple edit callbacks */
836         select_cb= ANIM_editkeyframes_select(select_mode);
837         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAME);
838         
839         /* loop through all of the keys and select additional keyframes
840          * based on the keys found to be selected above
841          */
842         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVEVISIBLE | ANIMFILTER_CURVESONLY);
843         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
844         
845         for (ale= anim_data.first; ale; ale= ale->next) {
846                 Object *nob= ANIM_nla_mapping_get(ac, ale);
847                 
848                 /* set frame for validation callback to refer to */
849                 if (nob)
850                         bed.f1= get_action_frame(nob, selx);
851                 else
852                         bed.f1= selx;
853                 
854                 /* select elements with frame number matching cfra */
855                 ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, ok_cb, select_cb, NULL);
856         }
857         
858         /* free elements */
859         BLI_freelistN(&bed.list);
860         BLI_freelistN(&anim_data);
861 }
862  
863 /* ------------------- */
864
865 /* handle clicking */
866 static int graphkeys_clickselect_invoke(bContext *C, wmOperator *op, wmEvent *event)
867 {
868         bAnimContext ac;
869         Scene *scene;
870         ARegion *ar;
871         View2D *v2d;
872         short selectmode;
873         int mval[2];
874         
875         /* get editor data */
876         if (ANIM_animdata_get_context(C, &ac) == 0)
877                 return OPERATOR_CANCELLED;
878         
879         /* get useful pointers from animation context data */
880         scene= ac.scene;
881         ar= ac.ar;
882         v2d= &ar->v2d;
883         
884         /* get mouse coordinates (in region coordinates) */
885         mval[0]= (event->x - ar->winrct.xmin);
886         mval[1]= (event->y - ar->winrct.ymin);
887         
888         /* select mode is either replace (deselect all, then add) or add/extend */
889         if (RNA_boolean_get(op->ptr, "extend"))
890                 selectmode= SELECT_INVERT;
891         else
892                 selectmode= SELECT_REPLACE;
893         
894         /* figure out action to take */
895         if (RNA_enum_get(op->ptr, "left_right")) {
896                 /* select all keys on same side of current frame as mouse */
897                 float x;
898                 
899                 UI_view2d_region_to_view(v2d, mval[0], mval[1], &x, NULL);
900                 if (x < CFRA)
901                         RNA_int_set(op->ptr, "left_right", GRAPHKEYS_LRSEL_LEFT);
902                 else    
903                         RNA_int_set(op->ptr, "left_right", GRAPHKEYS_LRSEL_RIGHT);
904                 
905                 graphkeys_mselect_leftright(&ac, RNA_enum_get(op->ptr, "left_right"), selectmode);
906         }
907         else if (RNA_boolean_get(op->ptr, "column")) {
908                 /* select all keyframes in the same frame as the one that was under the mouse */
909                 graphkeys_mselect_column(&ac, mval, selectmode);
910         }
911         else if (RNA_boolean_get(op->ptr, "curves")) {
912                 /* select all keyframes in F-Curve under mouse */
913                 mouse_graph_keys(&ac, mval, selectmode, 1);
914         }
915         else {
916                 /* select keyframe under mouse */
917                 mouse_graph_keys(&ac, mval, selectmode, 0);
918         }
919         
920         /* set notifier that things have changed */
921         ANIM_animdata_send_notifiers(C, &ac, ANIM_CHANGED_BOTH);
922         
923         /* for tweak grab to work */
924         return OPERATOR_FINISHED|OPERATOR_PASS_THROUGH;
925 }
926  
927 void GRAPHEDIT_OT_keyframes_clickselect (wmOperatorType *ot)
928 {
929         /* identifiers */
930         ot->name= "Mouse Select Keys";
931         ot->idname= "GRAPHEDIT_OT_keyframes_clickselect";
932         
933         /* api callbacks */
934         ot->invoke= graphkeys_clickselect_invoke;
935         ot->poll= ED_operator_areaactive;
936         
937         /* id-props */
938         // XXX should we make this into separate operators?
939         RNA_def_enum(ot->srna, "left_right", prop_graphkeys_leftright_select_types, 0, "Left Right", ""); // CTRLKEY
940         RNA_def_boolean(ot->srna, "extend", 0, "Extend Select", ""); // SHIFTKEY
941         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
942         RNA_def_boolean(ot->srna, "curves", 0, "Only Curves", "Select all the keyframes in the curve"); // CTRLKEY + ALTKEY
943 }
944
945 /* ************************************************************************** */