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