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