svn merge -r 21041:21301 https://svn.blender.org/svnroot/bf-blender/branches/blender2...
[blender.git] / source / blender / editors / space_action / action_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_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 #include "DNA_world_types.h"
58
59 #include "RNA_access.h"
60 #include "RNA_define.h"
61
62 #include "BKE_action.h"
63 #include "BKE_depsgraph.h"
64 #include "BKE_fcurve.h"
65 #include "BKE_key.h"
66 #include "BKE_material.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 "action_intern.h"
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 action 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_action_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         if (ac->datatype == ANIMCONT_GPENCIL)
116                 filter= (ANIMFILTER_VISIBLE);
117         else
118                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVESONLY);
119         
120         /* filter data */
121         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
122         
123         /* init BezTriple looping data */
124         memset(&bed, 0, sizeof(BeztEditData));
125         test_cb= ANIM_editkeyframes_ok(BEZT_OK_SELECTED);
126         
127         /* See if we should be selecting or deselecting */
128         if (test) {
129                 for (ale= anim_data.first; ale; ale= ale->next) {
130                         if (ale->type == ANIMTYPE_GPLAYER) {
131                                 //if (is_gplayer_frame_selected(ale->data)) {
132                                 //      sel= 0;
133                                 //      break;
134                                 //}
135                         }
136                         else {
137                                 if (ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, NULL, test_cb, NULL)) {
138                                         sel= SELECT_SUBTRACT;
139                                         break;
140                                 }
141                         }
142                 }
143         }
144         
145         /* convert sel to selectmode, and use that to get editor */
146         sel_cb= ANIM_editkeyframes_select(sel);
147         
148         /* Now set the flags */
149         for (ale= anim_data.first; ale; ale= ale->next) {
150                 //if (ale->type == ACTTYPE_GPLAYER)
151                 //      set_gplayer_frame_selection(ale->data, sel);
152                 //else
153                         ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, NULL, sel_cb, NULL);
154         }
155         
156         /* Cleanup */
157         BLI_freelistN(&anim_data);
158 }
159
160 /* ------------------- */
161
162 static int actkeys_deselectall_exec(bContext *C, wmOperator *op)
163 {
164         bAnimContext ac;
165         
166         /* get editor data */
167         if (ANIM_animdata_get_context(C, &ac) == 0)
168                 return OPERATOR_CANCELLED;
169                 
170         /* 'standard' behaviour - check if selected, then apply relevant selection */
171         if (RNA_boolean_get(op->ptr, "invert"))
172                 deselect_action_keys(&ac, 0, SELECT_INVERT);
173         else
174                 deselect_action_keys(&ac, 1, SELECT_ADD);
175         
176         /* set notifier that things have changed */
177         ED_area_tag_redraw(CTX_wm_area(C)); // FIXME... should be updating 'keyframes' data context or so instead!
178         
179         return OPERATOR_FINISHED;
180 }
181  
182 void ACT_OT_keyframes_select_all_toggle (wmOperatorType *ot)
183 {
184         /* identifiers */
185         ot->name= "Select All";
186         ot->idname= "ACT_OT_keyframes_select_all_toggle";
187         
188         /* api callbacks */
189         ot->exec= actkeys_deselectall_exec;
190         ot->poll= ED_operator_areaactive;
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 (ACTKEYS_BORDERSEL_ALLKEYS)
202  *      -> ALT-BKEY - depending on which axis of the region was larger...
203  *              -> 2) x-axis, so select all frames within frame range (ACTKEYS_BORDERSEL_FRAMERANGE)
204  *              -> 3) y-axis, so select all frames within channels that region included (ACTKEYS_BORDERSEL_CHANNELS)
205  */
206
207 /* defines for borderselect mode */
208 enum {
209         ACTKEYS_BORDERSEL_ALLKEYS       = 0,
210         ACTKEYS_BORDERSEL_FRAMERANGE,
211         ACTKEYS_BORDERSEL_CHANNELS,
212 } eActKeys_BorderSelect_Mode;
213
214
215 static void borderselect_action (bAnimContext *ac, rcti rect, short mode, short selectmode)
216 {
217         ListBase anim_data = {NULL, NULL};
218         bAnimListElem *ale;
219         int filter;
220         
221         BeztEditData bed;
222         BeztEditFunc ok_cb, select_cb;
223         View2D *v2d= &ac->ar->v2d;
224         rctf rectf;
225         float ymin=0, ymax=(float)(-ACHANNEL_HEIGHT);
226         
227         /* convert mouse coordinates to frame ranges and channel coordinates corrected for view pan/zoom */
228         UI_view2d_region_to_view(v2d, rect.xmin, rect.ymin+2, &rectf.xmin, &rectf.ymin);
229         UI_view2d_region_to_view(v2d, rect.xmax, rect.ymax-2, &rectf.xmax, &rectf.ymax);
230         
231         /* filter data */
232         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CHANNELS);
233         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
234         
235         /* get beztriple editing/validation funcs  */
236         select_cb= ANIM_editkeyframes_select(selectmode);
237         
238         if (ELEM(mode, ACTKEYS_BORDERSEL_FRAMERANGE, ACTKEYS_BORDERSEL_ALLKEYS))
239                 ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAMERANGE);
240         else
241                 ok_cb= NULL;
242                 
243         /* init editing data */
244         memset(&bed, 0, sizeof(BeztEditData));
245         
246         /* loop over data, doing border select */
247         for (ale= anim_data.first; ale; ale= ale->next) {
248                 Object *nob= ANIM_nla_mapping_get(ac, ale);
249                 
250                 /* get new vertical minimum extent of channel */
251                 ymin= ymax - ACHANNEL_STEP;
252                 
253                 /* set horizontal range (if applicable) */
254                 if (ELEM(mode, ACTKEYS_BORDERSEL_FRAMERANGE, ACTKEYS_BORDERSEL_ALLKEYS)) {
255                         /* if channel is mapped in NLA, apply correction */
256                         if (nob) {
257                                 bed.f1= get_action_frame(nob, rectf.xmin);
258                                 bed.f2= get_action_frame(nob, rectf.xmax);
259                         }
260                         else {
261                                 bed.f1= rectf.xmin;
262                                 bed.f2= rectf.xmax;
263                         }
264                 }
265                 
266                 /* perform vertical suitability check (if applicable) */
267                 if ( (mode == ACTKEYS_BORDERSEL_FRAMERANGE) || 
268                         !((ymax < rectf.ymin) || (ymin > rectf.ymax)) )
269                 {
270                         /* loop over data selecting */
271                         if (ale->key_data) {
272                                 if (ale->datatype == ALE_FCURVE)
273                                         ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, ok_cb, select_cb, NULL);
274                         }
275                         else if (ale->type == ANIMTYPE_GROUP) {
276                                 bActionGroup *agrp= ale->data;
277                                 FCurve *fcu;
278                                 
279                                 for (fcu= agrp->channels.first; fcu && fcu->grp==agrp; fcu= fcu->next) 
280                                         ANIM_fcurve_keys_bezier_loop(&bed, fcu, ok_cb, select_cb, NULL);
281                         }
282                         //else if (ale->type == ANIMTYPE_GPLAYER) {
283                         //      borderselect_gplayer_frames(ale->data, rectf.xmin, rectf.xmax, selectmode);
284                         //}
285                 }
286                 
287                 /* set minimum extent to be the maximum of the next channel */
288                 ymax=ymin;
289         }
290         
291         /* cleanup */
292         BLI_freelistN(&anim_data);
293 }
294
295 /* ------------------- */
296
297 static int actkeys_borderselect_exec(bContext *C, wmOperator *op)
298 {
299         bAnimContext ac;
300         rcti rect;
301         short mode=0, selectmode=0;
302         int event;
303         
304         /* get editor data */
305         if (ANIM_animdata_get_context(C, &ac) == 0)
306                 return OPERATOR_CANCELLED;
307         
308         /* get settings from operator */
309         rect.xmin= RNA_int_get(op->ptr, "xmin");
310         rect.ymin= RNA_int_get(op->ptr, "ymin");
311         rect.xmax= RNA_int_get(op->ptr, "xmax");
312         rect.ymax= RNA_int_get(op->ptr, "ymax");
313                 
314         event= RNA_int_get(op->ptr, "event_type");
315         if (event == LEFTMOUSE) // FIXME... hardcoded
316                 selectmode = SELECT_ADD;
317         else
318                 selectmode = SELECT_SUBTRACT;
319         
320         /* selection 'mode' depends on whether borderselect region only matters on one axis */
321         if (RNA_boolean_get(op->ptr, "axis_range")) {
322                 /* mode depends on which axis of the range is larger to determine which axis to use 
323                  *      - checking this in region-space is fine, as it's fundamentally still going to be a different rect size
324                  *      - the frame-range select option is favoured over the channel one (x over y), as frame-range one is often
325                  *        used for tweaking timing when "blocking", while channels is not that useful...
326                  */
327                 if ((rect.xmax - rect.xmin) >= (rect.ymax - rect.ymin))
328                         mode= ACTKEYS_BORDERSEL_FRAMERANGE;
329                 else
330                         mode= ACTKEYS_BORDERSEL_CHANNELS;
331         }
332         else 
333                 mode= ACTKEYS_BORDERSEL_ALLKEYS;
334         
335         /* apply borderselect action */
336         borderselect_action(&ac, rect, mode, selectmode);
337         
338         return OPERATOR_FINISHED;
339
340
341 void ACT_OT_keyframes_select_border(wmOperatorType *ot)
342 {
343         /* identifiers */
344         ot->name= "Border Select";
345         ot->idname= "ACT_OT_keyframes_select_border";
346         
347         /* api callbacks */
348         ot->invoke= WM_border_select_invoke;
349         ot->exec= actkeys_borderselect_exec;
350         ot->modal= WM_border_select_modal;
351         
352         ot->poll= ED_operator_areaactive;
353         
354         /* flags */
355         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
356         
357         /* rna */
358         RNA_def_int(ot->srna, "event_type", 0, INT_MIN, INT_MAX, "Event Type", "", INT_MIN, INT_MAX);
359         RNA_def_int(ot->srna, "xmin", 0, INT_MIN, INT_MAX, "X Min", "", INT_MIN, INT_MAX);
360         RNA_def_int(ot->srna, "xmax", 0, INT_MIN, INT_MAX, "X Max", "", INT_MIN, INT_MAX);
361         RNA_def_int(ot->srna, "ymin", 0, INT_MIN, INT_MAX, "Y Min", "", INT_MIN, INT_MAX);
362         RNA_def_int(ot->srna, "ymax", 0, INT_MIN, INT_MAX, "Y Max", "", INT_MIN, INT_MAX);
363         
364         RNA_def_boolean(ot->srna, "axis_range", 0, "Axis Range", "");
365 }
366
367 /* ******************** Column Select Operator **************************** */
368 /* This operator works in one of four ways:
369  *      - 1) select all keyframes in the same frame as a selected one  (KKEY)
370  *      - 2) select all keyframes in the same frame as the current frame marker (CTRL-KKEY)
371  *      - 3) select all keyframes in the same frame as a selected markers (SHIFT-KKEY)
372  *      - 4) select all keyframes that occur between selected markers (ALT-KKEY)
373  */
374
375 /* defines for column-select mode */
376 static EnumPropertyItem prop_column_select_types[] = {
377         {ACTKEYS_COLUMNSEL_KEYS, "KEYS", 0, "On Selected Keyframes", ""},
378         {ACTKEYS_COLUMNSEL_CFRA, "CFRA", 0, "On Current Frame", ""},
379         {ACTKEYS_COLUMNSEL_MARKERS_COLUMN, "MARKERS_COLUMN", 0, "On Selected Markers", ""},
380         {ACTKEYS_COLUMNSEL_MARKERS_BETWEEN, "MARKERS_BETWEEN", 0, "Between Min/Max Selected Markers", ""},
381         {0, NULL, 0, NULL, NULL}
382 };
383
384 /* ------------------- */ 
385
386 /* Selects all visible keyframes between the specified markers */
387 static void markers_selectkeys_between (bAnimContext *ac)
388 {
389         ListBase anim_data = {NULL, NULL};
390         bAnimListElem *ale;
391         int filter;
392         
393         BeztEditFunc ok_cb, select_cb;
394         BeztEditData bed;
395         float min, max;
396         
397         /* get extreme markers */
398         ED_markers_get_minmax(ac->markers, 1, &min, &max);
399         min -= 0.5f;
400         max += 0.5f;
401         
402         /* get editing funcs + data */
403         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAMERANGE);
404         select_cb= ANIM_editkeyframes_select(SELECT_ADD);
405         
406         memset(&bed, 0, sizeof(BeztEditData));
407         bed.f1= min; 
408         bed.f2= max;
409         
410         /* filter data */
411         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVESONLY);
412         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
413         
414         /* select keys in-between */
415         for (ale= anim_data.first; ale; ale= ale->next) {
416                 Object *nob= ANIM_nla_mapping_get(ac, ale);
417                 
418                 if (nob) {      
419                         ANIM_nla_mapping_apply_fcurve(nob, ale->key_data, 0, 1);
420                         ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, ok_cb, select_cb, NULL);
421                         ANIM_nla_mapping_apply_fcurve(nob, ale->key_data, 1, 1);
422                 }
423                 else {
424                         ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, ok_cb, select_cb, NULL);
425                 }
426         }
427         
428         /* Cleanup */
429         BLI_freelistN(&anim_data);
430 }
431
432
433 /* Selects all visible keyframes in the same frames as the specified elements */
434 static void columnselect_action_keys (bAnimContext *ac, short mode)
435 {
436         ListBase anim_data= {NULL, NULL};
437         bAnimListElem *ale;
438         int filter;
439         
440         Scene *scene= ac->scene;
441         CfraElem *ce;
442         BeztEditFunc select_cb, ok_cb;
443         BeztEditData bed;
444         
445         /* initialise keyframe editing data */
446         memset(&bed, 0, sizeof(BeztEditData));
447         
448         /* build list of columns */
449         switch (mode) {
450                 case ACTKEYS_COLUMNSEL_KEYS: /* list of selected keys */
451                         if (ac->datatype == ANIMCONT_GPENCIL) {
452                                 filter= (ANIMFILTER_VISIBLE);
453                                 ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
454                                 
455                                 //for (ale= anim_data.first; ale; ale= ale->next)
456                                 //      gplayer_make_cfra_list(ale->data, &elems, 1);
457                         }
458                         else {
459                                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVESONLY);
460                                 ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
461                                 
462                                 for (ale= anim_data.first; ale; ale= ale->next)
463                                         ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, NULL, bezt_to_cfraelem, NULL);
464                         }
465                         BLI_freelistN(&anim_data);
466                         break;
467                         
468                 case ACTKEYS_COLUMNSEL_CFRA: /* current frame */
469                         /* make a single CfraElem for storing this */
470                         ce= MEM_callocN(sizeof(CfraElem), "cfraElem");
471                         BLI_addtail(&bed.list, ce);
472                         
473                         ce->cfra= (float)CFRA;
474                         break;
475                         
476                 case ACTKEYS_COLUMNSEL_MARKERS_COLUMN: /* list of selected markers */
477                         ED_markers_make_cfra_list(ac->markers, &bed.list, 1);
478                         break;
479                         
480                 default: /* invalid option */
481                         return;
482         }
483         
484         /* set up BezTriple edit callbacks */
485         select_cb= ANIM_editkeyframes_select(SELECT_ADD);
486         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAME);
487         
488         /* loop through all of the keys and select additional keyframes
489          * based on the keys found to be selected above
490          */
491         if (ac->datatype == ANIMCONT_GPENCIL)
492                 filter= (ANIMFILTER_VISIBLE);
493         else
494                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVESONLY);
495         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
496         
497         for (ale= anim_data.first; ale; ale= ale->next) {
498                 Object *nob= ANIM_nla_mapping_get(ac, ale);
499                 
500                 /* loop over cfraelems (stored in the BeztEditData->list)
501                  *      - we need to do this here, as we can apply fewer NLA-mapping conversions
502                  */
503                 for (ce= bed.list.first; ce; ce= ce->next) {
504                         /* set frame for validation callback to refer to */
505                         if (nob)
506                                 bed.f1= get_action_frame(nob, ce->cfra);
507                         else
508                                 bed.f1= ce->cfra;
509                         
510                         /* select elements with frame number matching cfraelem */
511                         ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, ok_cb, select_cb, NULL);
512                         
513 #if 0 // XXX reenable when Grease Pencil stuff is back
514                         if (ale->type == ANIMTYPE_GPLAYER) {
515                                 bGPDlayer *gpl= (bGPDlayer *)ale->data;
516                                 bGPDframe *gpf;
517                                 
518                                 for (gpf= gpl->frames.first; gpf; gpf= gpf->next) {
519                                         if (ecfra == gpf->framenum) 
520                                                 gpf->flag |= GP_FRAME_SELECT;
521                                 }
522                         }
523                         //else... 
524 #endif // XXX reenable when Grease Pencil stuff is back
525                 }
526         }
527         
528         /* free elements */
529         BLI_freelistN(&bed.list);
530         BLI_freelistN(&anim_data);
531 }
532
533 /* ------------------- */
534
535 static int actkeys_columnselect_exec(bContext *C, wmOperator *op)
536 {
537         bAnimContext ac;
538         short mode;
539         
540         /* get editor data */
541         if (ANIM_animdata_get_context(C, &ac) == 0)
542                 return OPERATOR_CANCELLED;
543                 
544         /* action to take depends on the mode */
545         mode= RNA_enum_get(op->ptr, "mode");
546         
547         if (mode == ACTKEYS_COLUMNSEL_MARKERS_BETWEEN)
548                 markers_selectkeys_between(&ac);
549         else
550                 columnselect_action_keys(&ac, mode);
551         
552         /* set notifier that things have changed */
553         ANIM_animdata_send_notifiers(C, &ac, ANIM_CHANGED_KEYFRAMES_SELECT);
554         
555         return OPERATOR_FINISHED;
556 }
557  
558 void ACT_OT_keyframes_select_column (wmOperatorType *ot)
559 {
560         /* identifiers */
561         ot->name= "Select All";
562         ot->idname= "ACT_OT_keyframes_select_column";
563         
564         /* api callbacks */
565         ot->exec= actkeys_columnselect_exec;
566         ot->poll= ED_operator_areaactive;
567         
568         /* flags */
569         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
570         
571         /* props */
572         RNA_def_enum(ot->srna, "mode", prop_column_select_types, 0, "Mode", "");
573 }
574
575 /* ******************** Mouse-Click Select Operator *********************** */
576 /* This operator works in one of three ways:
577  *      - 1) keyframe under mouse - no special modifiers
578  *      - 2) all keyframes on the same side of current frame indicator as mouse - ALT modifier
579  *      - 3) column select all keyframes in frame under mouse - CTRL modifier
580  *
581  * In addition to these basic options, the SHIFT modifier can be used to toggle the 
582  * selection mode between replacing the selection (without) and inverting the selection (with).
583  */
584
585 /* defines for left-right select tool */
586 static EnumPropertyItem prop_actkeys_leftright_select_types[] = {
587         {ACTKEYS_LRSEL_TEST, "CHECK", 0, "Check if Select Left or Right", ""},
588         {ACTKEYS_LRSEL_NONE, "OFF", 0, "Don't select", ""},
589         {ACTKEYS_LRSEL_LEFT, "LEFT", 0, "Before current frame", ""},
590         {ACTKEYS_LRSEL_RIGHT, "RIGHT", 0, "After current frame", ""},
591         {0, NULL, 0, NULL, NULL}
592 };
593
594 /* sensitivity factor for frame-selections */
595 #define FRAME_CLICK_THRESH              0.1f
596
597 /* ------------------- */
598  
599 /* option 1) select keyframe directly under mouse */
600 static void actkeys_mselect_single (bAnimContext *ac, bAnimListElem *ale, short select_mode, float selx)
601 {
602         bDopeSheet *ads= (ac->datatype == ANIMCONT_DOPESHEET) ? ac->data : NULL;
603         int ds_filter = ((ads) ? (ads->filterflag) : (0));
604         
605         BeztEditData bed;
606         BeztEditFunc select_cb, ok_cb;
607         
608         /* get functions for selecting keyframes */
609         select_cb= ANIM_editkeyframes_select(select_mode);
610         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAME);
611         memset(&bed, 0, sizeof(BeztEditData)); 
612         bed.f1= selx;
613         
614         /* select the nominated keyframe on the given frame */
615         ANIM_animchannel_keys_bezier_loop(&bed, ale, ok_cb, select_cb, NULL, ds_filter);
616 }
617
618 /* Option 2) Selects all the keyframes on either side of the current frame (depends on which side the mouse is on) */
619 static void actkeys_mselect_leftright (bAnimContext *ac, short leftright, short select_mode)
620 {
621         ListBase anim_data = {NULL, NULL};
622         bAnimListElem *ale;
623         int filter;
624         
625         BeztEditFunc ok_cb, select_cb;
626         BeztEditData bed;
627         Scene *scene= ac->scene;
628         
629         /* if select mode is replace, deselect all keyframes (and channels) first */
630         if (select_mode==SELECT_REPLACE) {
631                 select_mode= SELECT_ADD;
632                 
633                 /* deselect all other channels and keyframes */
634                 ANIM_deselect_anim_channels(ac->data, ac->datatype, 0, ACHANNEL_SETFLAG_CLEAR);
635                 deselect_action_keys(ac, 0, SELECT_SUBTRACT);
636         }
637         
638         /* set callbacks and editing data */
639         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAMERANGE);
640         select_cb= ANIM_editkeyframes_select(select_mode);
641         
642         memset(&bed, 0, sizeof(BeztEditFunc));
643         if (leftright == ACTKEYS_LRSEL_LEFT) {
644                 bed.f1 = -MAXFRAMEF;
645                 bed.f2 = (float)(CFRA + FRAME_CLICK_THRESH);
646         } 
647         else {
648                 bed.f1 = (float)(CFRA - FRAME_CLICK_THRESH);
649                 bed.f2 = MAXFRAMEF;
650         }
651         
652         /* filter data */
653         if (ac->datatype == ANIMCONT_GPENCIL)
654                 filter= (ANIMFILTER_VISIBLE);
655         else
656                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVESONLY);
657         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
658                 
659         /* select keys on the side where most data occurs */
660         for (ale= anim_data.first; ale; ale= ale->next) {
661                 Object *nob= ANIM_nla_mapping_get(ac, ale);
662                 
663                 if (nob) {
664                         ANIM_nla_mapping_apply_fcurve(nob, ale->key_data, 0, 1);
665                         ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, ok_cb, select_cb, NULL);
666                         ANIM_nla_mapping_apply_fcurve(nob, ale->key_data, 1, 1);
667                 }
668                 //else if (ale->type == ANIMTYPE_GPLAYER)
669                 //      borderselect_gplayer_frames(ale->data, min, max, SELECT_ADD);
670                 else
671                         ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, ok_cb, select_cb, NULL);
672         }
673         
674         /* Cleanup */
675         BLI_freelistN(&anim_data);
676 }
677
678 /* Option 3) Selects all visible keyframes in the same frame as the mouse click */
679 static void actkeys_mselect_column(bAnimContext *ac, short select_mode, float selx)
680 {
681         ListBase anim_data= {NULL, NULL};
682         bAnimListElem *ale;
683         int filter;
684         
685         BeztEditFunc select_cb, ok_cb;
686         BeztEditData bed;
687         
688         /* initialise keyframe editing data */
689         memset(&bed, 0, sizeof(BeztEditData));
690         
691         /* set up BezTriple edit callbacks */
692         select_cb= ANIM_editkeyframes_select(select_mode);
693         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAME);
694         
695         /* loop through all of the keys and select additional keyframes
696          * based on the keys found to be selected above
697          */
698         if (ac->datatype == ANIMCONT_GPENCIL)
699                 filter= (ANIMFILTER_VISIBLE);
700         else
701                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CURVESONLY);
702         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
703         
704         for (ale= anim_data.first; ale; ale= ale->next) {
705                 Object *nob= ANIM_nla_mapping_get(ac, ale);
706                 
707                 /* set frame for validation callback to refer to */
708                 if (nob)
709                         bed.f1= get_action_frame(nob, selx);
710                 else
711                         bed.f1= selx;
712                 
713                 /* select elements with frame number matching cfra */
714                 ANIM_fcurve_keys_bezier_loop(&bed, ale->key_data, ok_cb, select_cb, NULL);
715                         
716 #if 0 // XXX reenable when Grease Pencil stuff is back
717                         if (ale->type == ANIMTYPE_GPLAYER) {
718                                 bGPDlayer *gpl= (bGPDlayer *)ale->data;
719                                 bGPDframe *gpf;
720                                 
721                                 for (gpf= gpl->frames.first; gpf; gpf= gpf->next) {
722                                         if (ecfra == gpf->framenum) 
723                                                 gpf->flag |= GP_FRAME_SELECT;
724                                 }
725                         }
726                         //else... 
727 #endif // XXX reenable when Grease Pencil stuff is back
728         }
729         
730         /* free elements */
731         BLI_freelistN(&bed.list);
732         BLI_freelistN(&anim_data);
733 }
734  
735 /* ------------------- */
736
737 static void mouse_action_keys (bAnimContext *ac, int mval[2], short select_mode, short column)
738 {
739         ListBase anim_data = {NULL, NULL};
740         ListBase anim_keys = {NULL, NULL};
741         bAnimListElem *ale;
742         int filter;
743         
744         View2D *v2d= &ac->ar->v2d;
745         int channel_index;
746         short found = 0;
747         float selx = 0.0f;
748         float x, y;
749         rctf rectf;
750         
751         
752         /* use View2D to determine the index of the channel (i.e a row in the list) where keyframe was */
753         UI_view2d_region_to_view(v2d, mval[0], mval[1], &x, &y);
754         UI_view2d_listview_view_to_cell(v2d, 0, ACHANNEL_STEP, 0, (float)ACHANNEL_HEIGHT_HALF, x, y, NULL, &channel_index);
755         
756         /* x-range to check is +/- 7 (in screen/region-space) on either side of mouse click (size of keyframe icon) */
757         UI_view2d_region_to_view(v2d, mval[0]-7, mval[1], &rectf.xmin, &rectf.ymin);
758         UI_view2d_region_to_view(v2d, mval[0]+7, mval[1], &rectf.xmax, &rectf.ymax);
759         
760         /* filter data */
761         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CHANNELS);
762         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
763         
764         /* try to get channel */
765         ale= BLI_findlink(&anim_data, channel_index);
766         if (ale == NULL) {
767                 /* channel not found */
768                 printf("Error: animation channel (index = %d) not found in mouse_action_keys() \n", channel_index);
769                 return;
770         }
771         else {
772                 /* found match - must return here... */
773                 Object *nob= ANIM_nla_mapping_get(ac, ale);
774                 ActKeysInc *aki= init_aki_data(ac, ale);
775                 ActKeyColumn *ak;
776                 float xmin, xmax;
777                 
778                 /* apply NLA-scaling correction? */
779                 if (nob) {
780                         xmin= get_action_frame(nob, rectf.xmin);
781                         xmax= get_action_frame(nob, rectf.xmax);
782                 }
783                 else {
784                         xmin= rectf.xmin;
785                         xmax= rectf.xmax;
786                 }
787                 
788                 /* make list of keyframes */
789                 if (ale->key_data) {
790                         switch (ale->datatype) {
791                                 case ALE_OB:
792                                 {
793                                         Object *ob= (Object *)ale->key_data;
794                                         ob_to_keylist(ob, &anim_keys, NULL, aki);
795                                 }
796                                         break;
797                                 case ALE_ACT:
798                                 {
799                                         bAction *act= (bAction *)ale->key_data;
800                                         action_to_keylist(act, &anim_keys, NULL, aki);
801                                 }
802                                         break;
803                                 case ALE_FCURVE:
804                                 {
805                                         FCurve *fcu= (FCurve *)ale->key_data;
806                                         fcurve_to_keylist(fcu, &anim_keys, NULL, aki);
807                                 }
808                                         break;
809                         }
810                 }
811                 else if (ale->type == ANIMTYPE_GROUP) {
812                         bActionGroup *agrp= (bActionGroup *)ale->data;
813                         agroup_to_keylist(agrp, &anim_keys, NULL, aki);
814                 }
815                 else if (ale->type == ANIMTYPE_GPDATABLOCK) {
816                         /* cleanup */
817                         // FIXME:...
818                         BLI_freelistN(&anim_data);
819                         return;
820                 }
821                 else if (ale->type == ANIMTYPE_GPLAYER) {
822                         bGPDlayer *gpl= (bGPDlayer *)ale->data;
823                         gpl_to_keylist(gpl, &anim_keys, NULL, aki);
824                 }
825                 
826                 /* loop through keyframes, finding one that was clicked on */
827                 for (ak= anim_keys.first; ak; ak= ak->next) {
828                         if (IN_RANGE(ak->cfra, xmin, xmax)) {
829                                 selx= ak->cfra;
830                                 found= 1;
831                                 break;
832                         }
833                 }
834                 
835                 /* remove active channel from list of channels for separate treatment (since it's needed later on) */
836                 BLI_remlink(&anim_data, ale);
837                 
838                 /* cleanup temporary lists */
839                 BLI_freelistN(&anim_keys);
840                 anim_keys.first = anim_keys.last = NULL;
841                 
842                 /* free list of channels, since it's not used anymore */
843                 BLI_freelistN(&anim_data);
844         }
845         
846         /* for replacing selection, firstly need to clear existing selection */
847         if (select_mode == SELECT_REPLACE) {
848                 /* reset selection mode for next steps */
849                 select_mode = SELECT_ADD;
850                 
851                 /* deselect all keyframes */
852                 deselect_action_keys(ac, 0, SELECT_SUBTRACT);
853                 
854                 /* highlight channel clicked on */
855                 if (ELEM(ac->datatype, ANIMCONT_ACTION, ANIMCONT_DOPESHEET)) {
856                         /* deselect all other channels first */
857                         ANIM_deselect_anim_channels(ac->data, ac->datatype, 0, ACHANNEL_SETFLAG_CLEAR);
858                         
859                         /* Highlight Action-Group or F-Curve? */
860                         if (ale->type == ANIMTYPE_GROUP) {
861                                 bActionGroup *agrp= ale->data;
862                                 
863                                 agrp->flag |= AGRP_SELECTED;
864                                 ANIM_set_active_channel(ac->data, ac->datatype, filter, agrp, ANIMTYPE_GROUP);
865                         }       
866                         else if (ale->type == ANIMTYPE_FCURVE) {
867                                 FCurve *fcu= ale->data;
868                                 
869                                 fcu->flag |= FCURVE_SELECTED;
870                                 ANIM_set_active_channel(ac->data, ac->datatype, filter, fcu, ANIMTYPE_FCURVE);
871                         }
872                 }
873                 else if (ac->datatype == ANIMCONT_GPENCIL) {
874                         ANIM_deselect_anim_channels(ac->data, ac->datatype, 0, ACHANNEL_SETFLAG_CLEAR);
875                         
876                         /* Highlight gpencil layer */
877                         //gpl->flag |= GP_LAYER_SELECT;
878                         //gpencil_layer_setactive(gpd, gpl);
879                 }
880         }
881         
882         /* only select keyframes if we clicked on a valid channel and hit something */
883         if (ale) {
884                 /* apply selection to keyframes */
885                 if (/*gpl*/0) {
886                         /* grease pencil */
887                         //select_gpencil_frame(gpl, (int)selx, selectmode);
888                 }
889                 else if (column) {
890                         /* select all keyframes in the same frame as the one we hit on the active channel */
891                         actkeys_mselect_column(ac, select_mode, selx);
892                 }
893                 else {
894                         /* select the nominated keyframe on the given frame */
895                         actkeys_mselect_single(ac, ale, select_mode, selx);
896                 }
897                 
898                 /* free this channel */
899                 MEM_freeN(ale);
900         }
901 }
902
903 /* handle clicking */
904 static int actkeys_clickselect_invoke(bContext *C, wmOperator *op, wmEvent *event)
905 {
906         bAnimContext ac;
907         Scene *scene;
908         ARegion *ar;
909         View2D *v2d;
910         short selectmode, column;
911         int mval[2];
912         
913         /* get editor data */
914         if (ANIM_animdata_get_context(C, &ac) == 0)
915                 return OPERATOR_CANCELLED;
916                 
917         /* get useful pointers from animation context data */
918         scene= ac.scene;
919         ar= ac.ar;
920         v2d= &ar->v2d;
921         
922         /* get mouse coordinates (in region coordinates) */
923         mval[0]= (event->x - ar->winrct.xmin);
924         mval[1]= (event->y - ar->winrct.ymin);
925         
926         /* select mode is either replace (deselect all, then add) or add/extend */
927         if (RNA_boolean_get(op->ptr, "extend"))
928                 selectmode= SELECT_INVERT;
929         else
930                 selectmode= SELECT_REPLACE;
931                 
932         /* column selection */
933         column= RNA_boolean_get(op->ptr, "column");
934         
935         /* figure out action to take */
936         if (RNA_enum_get(op->ptr, "left_right")) {
937                 /* select all keys on same side of current frame as mouse */
938                 float x;
939                 
940                 UI_view2d_region_to_view(v2d, mval[0], mval[1], &x, NULL);
941                 if (x < CFRA)
942                         RNA_int_set(op->ptr, "left_right", ACTKEYS_LRSEL_LEFT);
943                 else    
944                         RNA_int_set(op->ptr, "left_right", ACTKEYS_LRSEL_RIGHT);
945                 
946                 actkeys_mselect_leftright(&ac, RNA_enum_get(op->ptr, "left_right"), selectmode);
947         }
948         else {
949                 /* select keyframe(s) based upon mouse position*/
950                 mouse_action_keys(&ac, mval, selectmode, column);
951         }
952         
953         /* set notifier that things have changed */
954         ANIM_animdata_send_notifiers(C, &ac, ANIM_CHANGED_BOTH);
955         
956         /* for tweak grab to work */
957         return OPERATOR_FINISHED|OPERATOR_PASS_THROUGH;
958 }
959  
960 void ACT_OT_keyframes_clickselect (wmOperatorType *ot)
961 {
962         /* identifiers */
963         ot->name= "Mouse Select Keys";
964         ot->idname= "ACT_OT_keyframes_clickselect";
965         
966         /* api callbacks - absolutely no exec() this yet... */
967         ot->invoke= actkeys_clickselect_invoke;
968         ot->poll= ED_operator_areaactive;
969         
970         /* flags */
971         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
972         
973         /* id-props */
974         // XXX should we make this into separate operators?
975         RNA_def_enum(ot->srna, "left_right", prop_actkeys_leftright_select_types, 0, "Left Right", ""); // CTRLKEY
976         RNA_def_boolean(ot->srna, "extend", 0, "Extend Select", ""); // SHIFTKEY
977         RNA_def_boolean(ot->srna, "column", 0, "Column Select", ""); // ALTKEY
978 }
979
980 /* ************************************************************************** */