2.5 - Action Editor / Animation Stuff:
[blender-staging.git] / source / blender / editors / space_action / action_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) 2001-2002 by NaN Holding BV.
21  * All rights reserved.
22  *
23  * The Original Code is: all of this file.
24  *
25  * Contributor(s): Joshua Leung
26  *
27  * ***** END GPL LICENSE BLOCK *****
28  */
29
30 #include <math.h>
31 #include <stdlib.h>
32 #include <string.h>
33 #include <float.h>
34
35 #ifdef HAVE_CONFIG_H
36 #include <config.h>
37 #endif
38
39 #include "MEM_guardedalloc.h"
40
41 #include "BLI_blenlib.h"
42 #include "BLI_arithb.h"
43
44 #include "DNA_listBase.h"
45 #include "DNA_action_types.h"
46 #include "DNA_armature_types.h"
47 #include "DNA_camera_types.h"
48 #include "DNA_curve_types.h"
49 #include "DNA_ipo_types.h"
50 #include "DNA_object_types.h"
51 #include "DNA_screen_types.h"
52 #include "DNA_scene_types.h"
53 #include "DNA_space_types.h"
54 #include "DNA_constraint_types.h"
55 #include "DNA_key_types.h"
56 #include "DNA_lamp_types.h"
57 #include "DNA_material_types.h"
58 #include "DNA_userdef_types.h"
59 #include "DNA_gpencil_types.h"
60 #include "DNA_windowmanager_types.h"
61
62 #include "RNA_access.h"
63 #include "RNA_define.h"
64
65 #include "BKE_action.h"
66 #include "BKE_depsgraph.h"
67 #include "BKE_ipo.h"
68 #include "BKE_key.h"
69 #include "BKE_material.h"
70 #include "BKE_object.h"
71 #include "BKE_context.h"
72 #include "BKE_utildefines.h"
73
74 #include "UI_view2d.h"
75
76 #include "ED_anim_api.h"
77 #include "ED_keyframing.h"
78 #include "ED_keyframes_draw.h"
79 #include "ED_keyframes_edit.h"
80 #include "ED_screen.h"
81 #include "ED_space_api.h"
82
83 #include "WM_api.h"
84 #include "WM_types.h"
85
86 #include "action_intern.h"
87
88 /* ************************************************************************** */
89 /* GENERAL STUFF */
90
91 /* this function finds the channel that mouse is floating over */
92 void *get_nearest_act_channel (bAnimContext *ac, int mval[2], short *ret_type, void **owner)
93 {
94         ListBase anim_data = {NULL, NULL};
95         bAnimListElem *ale;
96         int filter;
97         
98         View2D *v2d= &ac->ar->v2d;
99         int clickmin, clickmax;
100         float x, y;
101         void *data= NULL;
102         
103         /* init 'owner' return val */
104         *owner= NULL;
105         
106         UI_view2d_region_to_view(v2d, mval[0], mval[1], &x, &y);
107         clickmin = (int) ((ACHANNEL_HEIGHT_HALF - y) / (ACHANNEL_STEP));
108         clickmax = clickmin;
109         
110         if (clickmax < 0) {
111                 *ret_type= ANIMTYPE_NONE;
112                 return NULL;
113         }
114         
115         /* filter data */
116         filter= (ANIMFILTER_FORDRAWING | ANIMFILTER_VISIBLE | ANIMFILTER_CHANNELS);
117         ANIM_animdata_filter(&anim_data, filter, ac->data, ac->datatype);
118         
119         for (ale= anim_data.first; ale; ale= ale->next) {
120                 if (clickmax < 0) 
121                         break;
122                 if (clickmin <= 0) {
123                         /* found match */
124                         *ret_type= ale->type;
125                         data= ale->data;
126                         
127                         /* if an 'ID' has been set, this takes presidence as owner (for dopesheet) */
128                         if (ac->datatype == ANIMCONT_DOPESHEET) {
129                                 /* return pointer to ID as owner instead */
130                                 if (ale->id) 
131                                         *owner= ale->id;
132                                 else
133                                         *owner= ale->owner;
134                         }
135                         else {
136                                 /* just use own owner */
137                                 *owner= ale->owner;
138                         }
139                         
140                         BLI_freelistN(&anim_data);
141                         
142                         return data;
143                 }
144                 --clickmin;
145                 --clickmax;
146         }
147         
148         /* cleanup */
149         BLI_freelistN(&anim_data);
150         
151         *ret_type= ANIMTYPE_NONE;
152         return NULL;
153 }
154
155
156 /* used only by mouse_action. It is used to find the location of the nearest 
157  * keyframe to where the mouse clicked, 
158  */
159 static void *get_nearest_action_key (bAnimContext *ac, int mval[2], float *selx, short *sel, short *ret_type, bActionChannel **par)
160 {
161         ListBase anim_data = {NULL, NULL};
162         ListBase anim_keys = {NULL, NULL};
163         bAnimListElem *ale;
164         ActKeyColumn *ak;
165         View2D *v2d= &ac->ar->v2d;
166         int filter;
167         
168         rctf rectf;
169         void *data = NULL;
170         float xmin, xmax, x, y;
171         int clickmin, clickmax;
172         short found = 0;
173         
174         /* action-channel */
175         *par= NULL;
176         
177         UI_view2d_region_to_view(v2d, mval[0], mval[1], &x, &y);
178     clickmin = (int) ((-y) / (ACHANNEL_STEP));
179         clickmax = clickmin;
180         
181         /* x-range to check is +/- 7 on either side of mouse click (size of keyframe icon) */
182         UI_view2d_region_to_view(v2d, mval[0]-7, mval[1], &rectf.xmin, &rectf.ymin);
183         UI_view2d_region_to_view(v2d, mval[0]+7, mval[1], &rectf.xmax, &rectf.ymax);
184         
185         if (clickmax < 0) {
186                 *ret_type= ANIMTYPE_NONE;
187                 return NULL;
188         }
189         
190         /* filter data */
191         filter= (ANIMFILTER_FORDRAWING | ANIMFILTER_VISIBLE | ANIMFILTER_CHANNELS);
192         ANIM_animdata_filter(&anim_data, filter, ac->data, ac->datatype);
193         
194         for (ale= anim_data.first; ale; ale= ale->next) {
195                 if (clickmax < 0) 
196                         break;
197                 if (clickmin <= 0) {
198                         /* found match - must return here... */
199                         Object *nob= ANIM_nla_mapping_get(ac, ale);
200                         
201                         /* apply NLA-scaling correction? */
202                         if (nob) {
203                                 xmin= get_action_frame(nob, rectf.xmin);
204                                 xmax= get_action_frame(nob, rectf.xmax);
205                         }
206                         else {
207                                 xmin= rectf.xmin;
208                                 xmax= rectf.xmax;
209                         }
210                         
211                         /* make list of keyframes */
212                         if (ale->key_data) {
213                                 switch (ale->datatype) {
214                                         case ALE_OB:
215                                         {
216                                                 Object *ob= (Object *)ale->key_data;
217                                                 ob_to_keylist(ob, &anim_keys, NULL, NULL);
218                                         }
219                                                 break;
220                                         case ALE_ACT:
221                                         {
222                                                 bAction *act= (bAction *)ale->key_data;
223                                                 action_to_keylist(act, &anim_keys, NULL, NULL);
224                                         }
225                                                 break;
226                                         case ALE_IPO:
227                                         {
228                                                 Ipo *ipo= (Ipo *)ale->key_data;
229                                                 ipo_to_keylist(ipo, &anim_keys, NULL, NULL);
230                                         }
231                                                 break;
232                                         case ALE_ICU:
233                                         {
234                                                 IpoCurve *icu= (IpoCurve *)ale->key_data;
235                                                 icu_to_keylist(icu, &anim_keys, NULL, NULL);
236                                         }
237                                                 break;
238                                 }
239                         }
240                         else if (ale->type == ANIMTYPE_GROUP) {
241                                 bActionGroup *agrp= (bActionGroup *)ale->data;
242                                 agroup_to_keylist(agrp, &anim_keys, NULL, NULL);
243                         }
244                         else if (ale->type == ANIMTYPE_GPDATABLOCK) {
245                                 /* cleanup */
246                                 BLI_freelistN(&anim_data);
247                                 
248                                 /* this channel currently doens't have any keyframes... must ignore! */
249                                 *ret_type= ANIMTYPE_NONE;
250                                 return NULL;
251                         }
252                         else if (ale->type == ANIMTYPE_GPLAYER) {
253                                 bGPDlayer *gpl= (bGPDlayer *)ale->data;
254                                 gpl_to_keylist(gpl, &anim_keys, NULL, NULL);
255                         }
256                         
257                         /* loop through keyframes, finding one that was clicked on */
258                         for (ak= anim_keys.first; ak; ak= ak->next) {
259                                 if (IN_RANGE(ak->cfra, xmin, xmax)) {
260                                         *selx= ak->cfra;
261                                         found= 1;
262                                         break;
263                                 }
264                         }
265                         /* no matching keyframe found - set to mean frame value so it doesn't actually select anything */
266                         if (found == 0)
267                                 *selx= ((xmax+xmin) / 2);
268                         
269                         /* figure out what to return */
270                         if (ac->datatype == ANIMCONT_ACTION) {
271                                 *par= ale->owner; /* assume that this is an action channel */
272                                 *ret_type= ale->type;
273                                 data = ale->data;
274                         }
275                         else if (ac->datatype == ANIMCONT_SHAPEKEY) {
276                                 data = ale->key_data;
277                                 *ret_type= ANIMTYPE_ICU;
278                         }
279                         else if (ac->datatype == ANIMCONT_DOPESHEET) {
280                                 data = ale->data;
281                                 *ret_type= ale->type;
282                         }
283                         else if (ac->datatype == ANIMCONT_GPENCIL) {
284                                 data = ale->data;
285                                 *ret_type= ANIMTYPE_GPLAYER;
286                         }
287                         
288                         /* cleanup tempolary lists */
289                         BLI_freelistN(&anim_keys);
290                         anim_keys.first = anim_keys.last = NULL;
291                         
292                         BLI_freelistN(&anim_data);
293                         
294                         return data;
295                 }
296                 --clickmin;
297                 --clickmax;
298         }
299         
300         /* cleanup */
301         BLI_freelistN(&anim_data);
302         
303         *ret_type= ANIMTYPE_NONE;
304         return NULL;
305 }
306
307 /* ************************************************************************** */
308 /* CHANNEL STUFF */
309
310 /* ************************************************************************** */
311 /* KEYFRAMES STUFF */
312
313 /* ******************** Deselect All Operator ***************************** */
314 /* This operator works in one of three ways:
315  *      1) (de)select all (AKEY) - test if select all or deselect all
316  *      2) invert all (CTRL-IKEY) - invert selection of all keyframes
317  *      3) (de)select all - no testing is done; only for use internal tools as normal function...
318  */
319
320 /* Deselects keyframes in the action editor
321  *      - This is called by the deselect all operator, as well as other ones!
322  *
323  *      - test: check if select or deselect all
324  *      - sel: how to select keyframes 
325  *              0 = deselect
326  *              1 = select
327  *              2 = invert
328  */
329 static void deselect_action_keys (bAnimContext *ac, short test, short sel)
330 {
331         ListBase anim_data = {NULL, NULL};
332         bAnimListElem *ale;
333         int filter;
334         
335         /* determine type-based settings */
336         if (ac->datatype == ANIMCONT_GPENCIL)
337                 filter= (ANIMFILTER_VISIBLE);
338         else
339                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_IPOKEYS);
340         
341         /* filter data */
342         ANIM_animdata_filter(&anim_data, filter, ac->data, ac->datatype);
343         
344         /* See if we should be selecting or deselecting */
345         if (test) {
346                 for (ale= anim_data.first; ale; ale= ale->next) {
347                         if (ale->type == ANIMTYPE_GPLAYER) {
348                                 //if (is_gplayer_frame_selected(ale->data)) {
349                                 //      sel= 0;
350                                 //      break;
351                                 //}
352                         }
353                         else {
354                                 if (is_ipo_key_selected(ale->key_data)) {
355                                         sel= 0;
356                                         break;
357                                 }
358                         }
359                 }
360         }
361                 
362         /* Now set the flags */
363         for (ale= anim_data.first; ale; ale= ale->next) {
364                 //if (ale->type == ACTTYPE_GPLAYER)
365                 //      set_gplayer_frame_selection(ale->data, sel);
366                 //else
367                         set_ipo_key_selection(ale->key_data, sel);
368         }
369         
370         /* Cleanup */
371         BLI_freelistN(&anim_data);
372 }
373
374 /* ------------------- */
375
376 static int actkeys_deselectall_exec(bContext *C, wmOperator *op)
377 {
378         bAnimContext ac;
379         
380         /* get editor data */
381         if (ANIM_animdata_get_context(C, &ac) == 0)
382                 return OPERATOR_CANCELLED;
383                 
384         /* 'standard' behaviour - check if selected, then apply relevant selection */
385         if (RNA_boolean_get(op->ptr, "invert"))
386                 deselect_action_keys(&ac, 0, 2);
387         else
388                 deselect_action_keys(&ac, 1, 1);
389         
390         /* set notifier tha things have changed */
391         ED_area_tag_redraw(CTX_wm_area(C)); // FIXME... should be updating 'keyframes' data context or so instead!
392         
393         return OPERATOR_FINISHED;
394 }
395  
396 void ACT_OT_keyframes_deselectall (wmOperatorType *ot)
397 {
398         /* identifiers */
399         ot->name= "Select All";
400         ot->idname= "ACT_OT_keyframes_deselectall";
401         
402         /* api callbacks */
403         ot->exec= actkeys_deselectall_exec;
404         ot->poll= ED_operator_areaactive;
405         
406         /* flags */
407         ot->flag= OPTYPE_REGISTER/*|OPTYPE_UNDO*/;
408         
409         /* props */
410         RNA_def_property(ot->srna, "invert", PROP_BOOLEAN, PROP_NONE);
411 }
412
413 /* ******************** Border Select Operator **************************** */
414 /* This operator currently works in one of three ways:
415  *      -> BKEY         - 1) all keyframes within region are selected (ACTKEYS_BORDERSEL_ALLKEYS)
416  *      -> ALT-BKEY - depending on which axis of the region was larger...
417  *              -> 2) x-axis, so select all frames within frame range (ACTKEYS_BORDERSEL_FRAMERANGE)
418  *              -> 3) y-axis, so select all frames within channels that region included (ACTKEYS_BORDERSEL_CHANNELS)
419  */
420
421 /* defines for borderselect mode */
422 enum {
423         ACTKEYS_BORDERSEL_ALLKEYS       = 0,
424         ACTKEYS_BORDERSEL_FRAMERANGE,
425         ACTKEYS_BORDERSEL_CHANNELS,
426 } eActKeys_BorderSelect_Mode;
427
428
429 static void borderselect_action (bAnimContext *ac, rcti rect, short mode, short selectmode)
430 {
431         ListBase anim_data = {NULL, NULL};
432         bAnimListElem *ale;
433         int filter;
434         
435         BeztEditData bed;
436         BeztEditFunc ok_cb, select_cb;
437         View2D *v2d= &ac->ar->v2d;
438         rctf rectf;
439         float ymin=0, ymax=(float)(-ACHANNEL_HEIGHT);
440         
441         /* convert mouse coordinates to frame ranges and channel coordinates corrected for view pan/zoom */
442         UI_view2d_region_to_view(v2d, rect.xmin, rect.ymin+2, &rectf.xmin, &rectf.ymin);
443         UI_view2d_region_to_view(v2d, rect.xmax, rect.ymax-2, &rectf.xmax, &rectf.ymax);
444         
445         /* filter data */
446         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_CHANNELS);
447         ANIM_animdata_filter(&anim_data, filter, ac->data, ac->datatype);
448         
449         /* get beztriple editing/validation funcs  */
450         select_cb= ANIM_editkeyframes_select(selectmode);
451         
452         if (ELEM(mode, ACTKEYS_BORDERSEL_FRAMERANGE, ACTKEYS_BORDERSEL_ALLKEYS))
453                 ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAMERANGE);
454         else
455                 ok_cb= NULL;
456                 
457         /* init editing data */
458         memset(&bed, 0, sizeof(BeztEditData));
459         
460         /* loop over data, doing border select */
461         for (ale= anim_data.first; ale; ale= ale->next) {
462                 Object *nob= ANIM_nla_mapping_get(ac, ale);
463                 
464                 /* get new vertical minimum extent of channel */
465                 ymin= ymax - ACHANNEL_STEP;
466                 
467                 /* set horizontal range (if applicable) */
468                 if (ELEM(mode, ACTKEYS_BORDERSEL_FRAMERANGE, ACTKEYS_BORDERSEL_ALLKEYS)) {
469                         /* if channel is mapped in NLA, apply correction */
470                         if (nob) {
471                                 bed.f1= get_action_frame(nob, rectf.xmin);
472                                 bed.f2= get_action_frame(nob, rectf.xmax);
473                         }
474                         else {
475                                 bed.f1= rectf.xmin;
476                                 bed.f2= rectf.xmax;
477                         }
478                 }
479                 
480                 /* perform vertical suitability check (if applicable) */
481                 if ( (mode == ACTKEYS_BORDERSEL_FRAMERANGE) || 
482                         !((ymax < rectf.ymin) || (ymin > rectf.ymax)) )
483                 {
484                         /* loop over data selecting */
485                         if (ale->key_data) {
486                                 if (ale->datatype == ALE_IPO)
487                                         ipo_keys_bezier_loop(&bed, ale->key_data, ok_cb, select_cb, NULL);
488                                 else if (ale->datatype == ALE_ICU)
489                                         icu_keys_bezier_loop(&bed, ale->key_data, ok_cb, select_cb, NULL);
490                         }
491                         else if (ale->type == ANIMTYPE_GROUP) {
492                                 bActionGroup *agrp= ale->data;
493                                 bActionChannel *achan;
494                                 bConstraintChannel *conchan;
495                                 
496                                 for (achan= agrp->channels.first; achan && achan->grp==agrp; achan= achan->next) {
497                                         ipo_keys_bezier_loop(&bed, achan->ipo, ok_cb, select_cb, NULL);
498                                         
499                                         for (conchan=achan->constraintChannels.first; conchan; conchan=conchan->next)
500                                                 ipo_keys_bezier_loop(&bed, conchan->ipo, ok_cb, select_cb, NULL);
501                                 }
502                         }
503                         //else if (ale->type == ANIMTYPE_GPLAYER) {
504                         //      borderselect_gplayer_frames(ale->data, rectf.xmin, rectf.xmax, selectmode);
505                         //}
506                 }
507                 
508                 /* set minimum extent to be the maximum of the next channel */
509                 ymax=ymin;
510         }
511         
512         /* cleanup */
513         BLI_freelistN(&anim_data);
514 }
515
516 /* ------------------- */
517
518 static int actkeys_borderselect_exec(bContext *C, wmOperator *op)
519 {
520         bAnimContext ac;
521         rcti rect;
522         short mode=0, selectmode=0;
523         int event;
524         
525         /* get editor data */
526         if (ANIM_animdata_get_context(C, &ac) == 0)
527                 return OPERATOR_CANCELLED;
528         
529         /* get settings from operator */
530         rect.xmin= RNA_int_get(op->ptr, "xmin");
531         rect.ymin= RNA_int_get(op->ptr, "ymin");
532         rect.xmax= RNA_int_get(op->ptr, "xmax");
533         rect.ymax= RNA_int_get(op->ptr, "ymax");
534                 
535         event= RNA_int_get(op->ptr, "event_type");
536         if (event == LEFTMOUSE) // FIXME... hardcoded
537                 selectmode = SELECT_ADD;
538         else
539                 selectmode = SELECT_SUBTRACT;
540         
541         /* selection 'mode' depends on whether borderselect region only matters on one axis */
542         if (RNA_boolean_get(op->ptr, "axis_range")) {
543                 /* mode depends on which axis of the range is larger to determine which axis to use 
544                  *      - checking this in region-space is fine, as it's fundamentally still going to be a different rect size
545                  *      - the frame-range select option is favoured over the channel one (x over y), as frame-range one is often
546                  *        used for tweaking timing when "blocking", while channels is not that useful...
547                  */
548                 if ((rect.xmax - rect.xmin) >= (rect.ymax - rect.ymin))
549                         mode= ACTKEYS_BORDERSEL_FRAMERANGE;
550                 else
551                         mode= ACTKEYS_BORDERSEL_CHANNELS;
552         }
553         else 
554                 mode= ACTKEYS_BORDERSEL_ALLKEYS;
555         
556         /* apply borderselect action */
557         borderselect_action(&ac, rect, mode, selectmode);
558         
559         return OPERATOR_FINISHED;
560
561
562 void ACT_OT_keyframes_borderselect(wmOperatorType *ot)
563 {
564         /* identifiers */
565         ot->name= "Border Select";
566         ot->idname= "ACT_OT_keyframes_borderselect";
567         
568         /* api callbacks */
569         ot->invoke= WM_border_select_invoke;
570         ot->exec= actkeys_borderselect_exec;
571         ot->modal= WM_border_select_modal;
572         
573         ot->poll= ED_operator_areaactive;
574         
575         /* flags */
576         // XXX er...
577         ot->flag= OPTYPE_REGISTER/*|OPTYPE_UNDO*/;
578         
579         /* rna */
580         RNA_def_property(ot->srna, "event_type", PROP_INT, PROP_NONE);
581         RNA_def_property(ot->srna, "xmin", PROP_INT, PROP_NONE);
582         RNA_def_property(ot->srna, "xmax", PROP_INT, PROP_NONE);
583         RNA_def_property(ot->srna, "ymin", PROP_INT, PROP_NONE);
584         RNA_def_property(ot->srna, "ymax", PROP_INT, PROP_NONE);
585         
586         RNA_def_property(ot->srna, "axis_range", PROP_BOOLEAN, PROP_NONE);
587 }
588
589 /* ******************** Column Select Operator **************************** */
590 /* This operator works in one of four ways:
591  *      - 1) select all keyframes in the same frame as a selected one  (KKEY)
592  *      - 2) select all keyframes in the same frame as the current frame marker (CTRL-KKEY)
593  *      - 3) select all keyframes in the same frame as a selected markers (SHIFT-KKEY)
594  *      - 4) select all keyframes that occur between selected markers (ALT-KKEY)
595  */
596
597 /* defines for column-select mode */
598 EnumPropertyItem prop_column_select_types[] = {
599         {ACTKEYS_COLUMNSEL_KEYS, "KEYS", "On Selected Keyframes", ""},
600         {ACTKEYS_COLUMNSEL_CFRA, "CFRA", "On Current Frame", ""},
601         {ACTKEYS_COLUMNSEL_MARKERS_COLUMN, "MARKERS_COLUMN", "On Selected Markers", ""},
602         {ACTKEYS_COLUMNSEL_MARKERS_BETWEEN, "MARKERS_BETWEEN", "Between Min/Max Selected Markers", ""},
603         {0, NULL, NULL, NULL}
604 };
605
606 /* ------------------- */ 
607
608 /* Selects all visible keyframes between the specified markers */
609 static void markers_selectkeys_between (bAnimContext *ac)
610 {
611         ListBase anim_data = {NULL, NULL};
612         bAnimListElem *ale;
613         int filter;
614         
615         BeztEditFunc select_cb;
616         BeztEditData bed;
617         float min, max;
618         
619         /* get extreme markers */
620         //get_minmax_markers(1, &min, &max); // FIXME... add back markers api!
621         min= ac->scene->r.sfra; // xxx temp code
622         max= ac->scene->r.efra; // xxx temp code
623         
624         if (min==max) return;
625         min -= 0.5f;
626         max += 0.5f;
627         
628         /* get editing funcs + data */
629         select_cb= ANIM_editkeyframes_select(SELECT_ADD);
630         memset(&bed, 0, sizeof(BeztEditData));
631         bed.f1= min; 
632         bed.f2= max;
633         
634         /* filter data */
635         filter= (ANIMFILTER_VISIBLE | ANIMFILTER_IPOKEYS);
636         ANIM_animdata_filter(&anim_data, filter, ac->data, ac->datatype);
637         
638         /* select keys in-between */
639         for (ale= anim_data.first; ale; ale= ale->next) {
640                 Object *nob= ANIM_nla_mapping_get(ac, ale);
641                 
642                 if (nob) {      
643                         ANIM_nla_mapping_apply(nob, ale->key_data, 0, 1);
644                         ipo_keys_bezier_loop(&bed, ale->key_data, NULL, select_cb, NULL);
645                         ANIM_nla_mapping_apply(nob, ale->key_data, 1, 1);
646                 }
647                 else {
648                         ipo_keys_bezier_loop(&bed, ale->key_data, NULL, select_cb, NULL);
649                 }
650         }
651         
652         /* Cleanup */
653         BLI_freelistN(&anim_data);
654 }
655
656
657 /* helper callback for columnselect_action_keys() -> populate list CfraElems with frame numbers from selected beztriples */
658 // TODO: if some other code somewhere needs this, it'll be time to port this over to keyframes_edit.c!!!
659 static short bezt_to_cfraelem(BeztEditData *bed, BezTriple *bezt)
660 {
661         /* only if selected */
662         if (bezt->f2 & SELECT) {
663                 CfraElem *ce= MEM_callocN(sizeof(CfraElem), "cfraElem");
664                 BLI_addtail(&bed->list, ce);
665                 
666                 ce->cfra= bezt->vec[1][0];
667         }
668         
669         return 0;
670 }
671
672 /* Selects all visible keyframes in the same frames as the specified elements */
673 static void columnselect_action_keys (bAnimContext *ac, short mode)
674 {
675         ListBase anim_data= {NULL, NULL};
676         bAnimListElem *ale;
677         int filter;
678         
679         Scene *scene= ac->scene;
680         CfraElem *ce;
681         BeztEditFunc select_cb, ok_cb;
682         BeztEditData bed;
683         
684         /* initialise keyframe editing data */
685         memset(&bed, 0, sizeof(BeztEditData));
686         
687         /* build list of columns */
688         switch (mode) {
689                 case ACTKEYS_COLUMNSEL_KEYS: /* list of selected keys */
690                         if (ac->datatype == ANIMCONT_GPENCIL) {
691                                 filter= (ANIMFILTER_VISIBLE);
692                                 ANIM_animdata_filter(&anim_data, filter, ac->data, ac->datatype);
693                                 
694                                 //for (ale= anim_data.first; ale; ale= ale->next)
695                                 //      gplayer_make_cfra_list(ale->data, &elems, 1);
696                         }
697                         else {
698                                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_IPOKEYS);
699                                 ANIM_animdata_filter(&anim_data, filter, ac->data, ac->datatype);
700                                 
701                                 for (ale= anim_data.first; ale; ale= ale->next)
702                                         ipo_keys_bezier_loop(&bed, ale->key_data, NULL, bezt_to_cfraelem, NULL);
703                         }
704                         BLI_freelistN(&anim_data);
705                         break;
706                         
707                 case ACTKEYS_COLUMNSEL_CFRA: /* current frame */
708                         /* make a single CfraElem for storing this */
709                         ce= MEM_callocN(sizeof(CfraElem), "cfraElem");
710                         BLI_addtail(&bed.list, ce);
711                         
712                         ce->cfra= (float)CFRA;
713                         break;
714                         
715                 case ACTKEYS_COLUMNSEL_MARKERS_COLUMN: /* list of selected markers */
716                         // FIXME: markers api needs to be improved for this first!
717                         //make_marker_cfra_list(&elems, 1);
718                         return; // XXX currently, this does nothing!
719                         break;
720                         
721                 default: /* invalid option */
722                         return;
723         }
724         
725         /* set up BezTriple edit callbacks */
726         select_cb= ANIM_editkeyframes_select(SELECT_ADD);
727         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAME);
728         
729         /* loop through all of the keys and select additional keyframes
730          * based on the keys found to be selected above
731          */
732         if (ac->datatype == ANIMCONT_GPENCIL)
733                 filter= (ANIMFILTER_VISIBLE);
734         else
735                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_ONLYICU);
736         ANIM_animdata_filter(&anim_data, filter, ac->data, ac->datatype);
737         
738         for (ale= anim_data.first; ale; ale= ale->next) {
739                 Object *nob= ANIM_nla_mapping_get(ac, ale);
740                 
741                 /* loop over cfraelems (stored in the BeztEditData->list)
742                  *      - we need to do this here, as we can apply fewer NLA-mapping conversions
743                  */
744                 for (ce= bed.list.first; ce; ce= ce->next) {
745                         /* set frame for validation callback to refer to */
746                         if (nob)
747                                 bed.f1= get_action_frame(nob, ce->cfra);
748                         else
749                                 bed.f1= ce->cfra;
750                         
751                         /* select elements with frame number matching cfraelem */
752                         icu_keys_bezier_loop(&bed, ale->key_data, ok_cb, select_cb, NULL);
753                         
754 #if 0 // XXX reenable when Grease Pencil stuff is back
755                         if (ale->type == ANIMTYPE_GPLAYER) {
756                                 bGPDlayer *gpl= (bGPDlayer *)ale->data;
757                                 bGPDframe *gpf;
758                                 
759                                 for (gpf= gpl->frames.first; gpf; gpf= gpf->next) {
760                                         if (ecfra == gpf->framenum) 
761                                                 gpf->flag |= GP_FRAME_SELECT;
762                                 }
763                         }
764                         //else... 
765 #endif // XXX reenable when Grease Pencil stuff is back
766                 }
767         }
768         
769         /* free elements */
770         BLI_freelistN(&bed.list);
771         BLI_freelistN(&anim_data);
772 }
773
774 /* ------------------- */
775
776 static int actkeys_columnselect_exec(bContext *C, wmOperator *op)
777 {
778         bAnimContext ac;
779         short mode;
780         
781         /* get editor data */
782         if (ANIM_animdata_get_context(C, &ac) == 0)
783                 return OPERATOR_CANCELLED;
784                 
785         /* action to take depends on the mode */
786         mode= RNA_enum_get(op->ptr, "mode");
787         
788         if (mode == ACTKEYS_COLUMNSEL_MARKERS_BETWEEN)
789                 markers_selectkeys_between(&ac);
790         else
791                 columnselect_action_keys(&ac, mode);
792         
793         /* set notifier tha things have changed */
794         ED_area_tag_redraw(CTX_wm_area(C)); // FIXME... should be updating 'keyframes' data context or so instead!
795         
796         return OPERATOR_FINISHED;
797 }
798  
799 void ACT_OT_keyframes_columnselect (wmOperatorType *ot)
800 {
801         PropertyRNA *prop;
802         
803         /* identifiers */
804         ot->name= "Select All";
805         ot->idname= "ACT_OT_keyframes_columnselect";
806         
807         /* api callbacks */
808         ot->exec= actkeys_columnselect_exec;
809         ot->poll= ED_operator_areaactive;
810         
811         /* flags */
812         ot->flag= OPTYPE_REGISTER/*|OPTYPE_UNDO*/;
813         
814         /* props */
815         prop= RNA_def_property(ot->srna, "mode", PROP_ENUM, PROP_NONE);
816         RNA_def_property_enum_items(prop, prop_column_select_types);
817 }
818
819 /* ******************** Mouse-Click Select Operator *********************** */
820 /* This operator works in one of three ways:
821  *      - 1) keyframe under mouse - no special modifiers
822  *      - 2) all keyframes on the same side of current frame indicator as mouse - ALT modifier
823  *      - 3) column select all keyframes in frame under mouse - CTRL modifier
824  *
825  * In addition to these basic options, the SHIFT modifier can be used to toggle the 
826  * selection mode between replacing the selection (without) and inverting the selection (with).
827  */
828
829 /* defines for left-right select tool */
830 EnumPropertyItem prop_leftright_select_types[] = {
831         {ACTKEYS_LRSEL_TEST, "CHECK", "Check if Select Left or Right", ""},
832         {ACTKEYS_LRSEL_NONE, "OFF", "Don't select", ""},
833         {ACTKEYS_LRSEL_LEFT, "LEFT", "Before current frame", ""},
834         {ACTKEYS_LRSEL_RIGHT, "RIGHT", "After current frame", ""},
835         {0, NULL, NULL, NULL}
836 };
837
838 /* ------------------- */
839  
840 /* option 1) select keyframe directly under mouse */
841 static void mouse_action_keys (bAnimContext *ac, int mval[2], short selectmode)
842 {
843         Object *ob= NULL;
844         bDopeSheet *ads= NULL;
845         bAction *act= NULL;
846         bActionGroup *agrp= NULL;
847         bActionChannel *achan= NULL;
848         bConstraintChannel *conchan= NULL;
849         Ipo *ipo= NULL;
850         IpoCurve *icu= NULL;
851         bGPdata *gpd = NULL;
852         bGPDlayer *gpl = NULL;
853         
854         BeztEditData bed;
855         BeztEditFunc select_cb, ok_cb;
856         void *anim_channel;
857         short sel, chan_type = 0;
858         float selx = 0.0f, selxa;
859         
860         /* determine what type of data we are operating on */
861         if (ac->datatype == ANIMCONT_ACTION) 
862                 act= (bAction *)ac->data;
863         else if (ac->datatype == ANIMCONT_DOPESHEET) 
864                 ads= (bDopeSheet *)ac->data;
865         else if (ac->datatype == ANIMCONT_GPENCIL) 
866                 gpd= (bGPdata *)ac->data;
867         
868         /* get channel and selection info */
869         anim_channel= get_nearest_action_key(ac, mval, &selx, &sel, &chan_type, &achan);
870         if (anim_channel == NULL) 
871                 return;
872         
873         switch (chan_type) {
874                 case ANIMTYPE_ICU:
875                         icu= (IpoCurve *)anim_channel;
876                         break;
877                 case ANIMTYPE_CONCHAN:
878                         conchan= (bConstraintChannel *)anim_channel;
879                         break;
880                 case ANIMTYPE_ACHAN:
881                         achan= (bActionChannel *)anim_channel;
882                         break;
883                 case ANIMTYPE_GROUP:
884                         agrp= (bActionGroup *)anim_channel;
885                         break;
886                 case ANIMTYPE_DSMAT:
887                         ipo= ((Material *)anim_channel)->ipo;
888                         break;
889                 case ANIMTYPE_DSLAM:
890                         ipo= ((Lamp *)anim_channel)->ipo;
891                         break;
892                 case ANIMTYPE_DSCAM:
893                         ipo= ((Camera *)anim_channel)->ipo;
894                         break;
895                 case ANIMTYPE_DSCUR:
896                         ipo= ((Curve *)anim_channel)->ipo;
897                         break;
898                 case ANIMTYPE_DSSKEY:
899                         ipo= ((Key *)anim_channel)->ipo;
900                         break;
901                 case ANIMTYPE_FILLACTD:
902                         act= (bAction *)anim_channel;
903                         break;
904                 case ANIMTYPE_FILLIPOD:
905                         ipo= ((Object *)anim_channel)->ipo;
906                         break;
907                 case ANIMTYPE_OBJECT:
908                         ob= ((Base *)anim_channel)->object;
909                         break;
910                 case ANIMTYPE_GPLAYER:
911                         gpl= (bGPDlayer *)anim_channel;
912                         break;
913                 default:
914                         return;
915         }
916         
917         /* for replacing selection, firstly need to clear existing selection */
918         if (selectmode == SELECT_REPLACE) {
919                 selectmode = SELECT_ADD;
920                 
921                 deselect_action_keys(ac, 0, 0);
922                 
923                 if (ELEM(ac->datatype, ANIMCONT_ACTION, ANIMCONT_DOPESHEET)) {
924                         //deselect_action_channels(0);
925                         
926                         /* Highlight either an Action-Channel or Action-Group */
927                         if (achan) {
928                                 achan->flag |= ACHAN_SELECTED;
929                                 //hilight_channel(act, achan, 1);
930                                 //select_poseelement_by_name(achan->name, 2);   /* 2 is activate */
931                         }
932                         else if (agrp) {
933                                 agrp->flag |= AGRP_SELECTED;
934                                 //set_active_actiongroup(act, agrp, 1);
935                         }
936                 }
937                 else if (ac->datatype == ANIMCONT_GPENCIL) {
938                         //deselect_action_channels(0);
939                         
940                         /* Highlight gpencil layer */
941                         gpl->flag |= GP_LAYER_SELECT;
942                         //gpencil_layer_setactive(gpd, gpl);
943                 }
944         }
945         
946         /* get functions for selecting keyframes */
947         select_cb= ANIM_editkeyframes_select(selectmode);
948         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAME);
949         memset(&bed, 0, sizeof(BeztEditData)); 
950         bed.f1= selx;
951         
952         /* apply selection to keyframes */
953         if (icu)
954                 icu_keys_bezier_loop(&bed, icu, ok_cb, select_cb, NULL);
955         else if (ipo)
956                 ipo_keys_bezier_loop(&bed, ipo, ok_cb, select_cb, NULL);
957         else if (conchan)
958                 ipo_keys_bezier_loop(&bed, conchan->ipo, ok_cb, select_cb, NULL);
959         else if (achan)
960                 ipo_keys_bezier_loop(&bed, achan->ipo, ok_cb, select_cb, NULL);
961         else if (agrp) {
962                 for (achan= agrp->channels.first; achan && achan->grp==agrp; achan= achan->next) {
963                         ipo_keys_bezier_loop(&bed, achan->ipo, ok_cb, select_cb, NULL);
964                         
965                         for (conchan=achan->constraintChannels.first; conchan; conchan=conchan->next)
966                                 ipo_keys_bezier_loop(&bed, conchan->ipo, ok_cb, select_cb, NULL);
967                 }
968         }
969         else if (act) {
970                 for (achan= act->chanbase.first; achan; achan= achan->next) {
971                         ipo_keys_bezier_loop(&bed, achan->ipo, ok_cb, select_cb, NULL);
972                         
973                         for (conchan=achan->constraintChannels.first; conchan; conchan=conchan->next)
974                                 ipo_keys_bezier_loop(&bed, conchan->ipo, ok_cb, select_cb, NULL);
975                 }
976         }
977         else if (ob) {
978                 if (ob->ipo) {
979                         bed.f1= selx;
980                         ipo_keys_bezier_loop(&bed, ob->ipo, ok_cb, select_cb, NULL);
981                 }
982                 
983                 if (ob->action) {
984                         selxa= get_action_frame(ob, selx);
985                         bed.f1= selxa;
986                         
987                         for (achan= ob->action->chanbase.first; achan; achan= achan->next) {
988                                 ipo_keys_bezier_loop(&bed, achan->ipo, ok_cb, select_cb, NULL);
989                                 
990                                 for (conchan=achan->constraintChannels.first; conchan; conchan=conchan->next)
991                                         ipo_keys_bezier_loop(&bed, conchan->ipo, ok_cb, select_cb, NULL);
992                         }
993                 }
994                 
995                 if (ob->constraintChannels.first) {
996                         bed.f1= selx;
997                         
998                         for (conchan=ob->constraintChannels.first; conchan; conchan=conchan->next)
999                                 ipo_keys_bezier_loop(&bed, conchan->ipo, ok_cb, select_cb, NULL);
1000                 }
1001         }
1002         //else if (gpl)
1003         //      select_gpencil_frame(gpl, (int)selx, selectmode);
1004 }
1005
1006 /* Option 2) Selects all the keyframes on either side of the current frame (depends on which side the mouse is on) */
1007 static void selectkeys_leftright (bAnimContext *ac, short leftright, short select_mode)
1008 {
1009         ListBase anim_data = {NULL, NULL};
1010         bAnimListElem *ale;
1011         int filter;
1012         
1013         BeztEditFunc ok_cb, select_cb;
1014         BeztEditData bed;
1015         Scene *scene= ac->scene;
1016         
1017         /* if select mode is replace, deselect all keyframes first */
1018         if (select_mode==SELECT_REPLACE) {
1019                 select_mode=SELECT_ADD;
1020                 deselect_action_keys(ac, 0, 0);
1021         }
1022         
1023         /* set callbacks and editing data */
1024         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAMERANGE);
1025         select_cb= ANIM_editkeyframes_select(select_mode);
1026         
1027         memset(&bed, 0, sizeof(BeztEditFunc));
1028         if (leftright == ACTKEYS_LRSEL_LEFT) {
1029                 bed.f1 = -MAXFRAMEF;
1030                 bed.f2 = (float)(CFRA + 0.1f);
1031         } 
1032         else {
1033                 bed.f1 = (float)(CFRA - 0.1f);
1034                 bed.f2 = MAXFRAMEF;
1035         }
1036         
1037         /* filter data */
1038         if (ac->datatype == ANIMCONT_GPENCIL)
1039                 filter= (ANIMFILTER_VISIBLE);
1040         else
1041                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_IPOKEYS);
1042         ANIM_animdata_filter(&anim_data, filter, ac->data, ac->datatype);
1043                 
1044         /* select keys on the side where most data occurs */
1045         for (ale= anim_data.first; ale; ale= ale->next) {
1046                 Object *nob= ANIM_nla_mapping_get(ac, ale);
1047                 
1048                 if (nob) {
1049                         ANIM_nla_mapping_apply(nob, ale->key_data, 0, 1);
1050                         ipo_keys_bezier_loop(&bed, ale->key_data, ok_cb, select_cb, NULL);
1051                         ANIM_nla_mapping_apply(nob, ale->key_data, 1, 1);
1052                 }
1053                 //else if (ale->type == ANIMTYPE_GPLAYER)
1054                 //      borderselect_gplayer_frames(ale->data, min, max, SELECT_ADD);
1055                 else
1056                         ipo_keys_bezier_loop(&bed, ale->key_data, ok_cb, select_cb, NULL);
1057         }
1058         
1059         /* Cleanup */
1060         BLI_freelistN(&anim_data);
1061 }
1062
1063 /* Option 3) Selects all visible keyframes in the same frame as the mouse click */
1064 static void mouse_columnselect_action_keys (bAnimContext *ac, float selx)
1065 {
1066         ListBase anim_data= {NULL, NULL};
1067         bAnimListElem *ale;
1068         int filter;
1069         
1070         BeztEditFunc select_cb, ok_cb;
1071         BeztEditData bed;
1072         
1073         /* initialise keyframe editing data */
1074         memset(&bed, 0, sizeof(BeztEditData));
1075         
1076         /* set up BezTriple edit callbacks */
1077         select_cb= ANIM_editkeyframes_select(SELECT_ADD);
1078         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAME);
1079         
1080         /* loop through all of the keys and select additional keyframes
1081          * based on the keys found to be selected above
1082          */
1083         if (ac->datatype == ANIMCONT_GPENCIL)
1084                 filter= (ANIMFILTER_VISIBLE);
1085         else
1086                 filter= (ANIMFILTER_VISIBLE | ANIMFILTER_ONLYICU);
1087         ANIM_animdata_filter(&anim_data, filter, ac->data, ac->datatype);
1088         
1089         for (ale= anim_data.first; ale; ale= ale->next) {
1090                 Object *nob= ANIM_nla_mapping_get(ac, ale);
1091                 
1092                 /* set frame for validation callback to refer to */
1093                 if (nob)
1094                         bed.f1= get_action_frame(nob, selx);
1095                 else
1096                         bed.f1= selx;
1097                 
1098                 /* select elements with frame number matching cfraelem */
1099                 icu_keys_bezier_loop(&bed, ale->key_data, ok_cb, select_cb, NULL);
1100                         
1101 #if 0 // XXX reenable when Grease Pencil stuff is back
1102                         if (ale->type == ANIMTYPE_GPLAYER) {
1103                                 bGPDlayer *gpl= (bGPDlayer *)ale->data;
1104                                 bGPDframe *gpf;
1105                                 
1106                                 for (gpf= gpl->frames.first; gpf; gpf= gpf->next) {
1107                                         if (ecfra == gpf->framenum) 
1108                                                 gpf->flag |= GP_FRAME_SELECT;
1109                                 }
1110                         }
1111                         //else... 
1112 #endif // XXX reenable when Grease Pencil stuff is back
1113         }
1114         
1115         /* free elements */
1116         BLI_freelistN(&bed.list);
1117         BLI_freelistN(&anim_data);
1118 }
1119  
1120 /* ------------------- */
1121
1122 /* handle clicking */
1123 static int actkeys_clickselect_invoke(bContext *C, wmOperator *op, wmEvent *event)
1124 {
1125         bAnimContext ac;
1126         Scene *scene;
1127         ARegion *ar;
1128         View2D *v2d;
1129         short selectmode;
1130         int mval[2];
1131         
1132         /* get editor data */
1133         if (ANIM_animdata_get_context(C, &ac) == 0)
1134                 return OPERATOR_CANCELLED;
1135                 
1136         /* get useful pointers from animation context data */
1137         scene= ac.scene;
1138         ar= ac.ar;
1139         v2d= &ar->v2d;
1140         
1141         /* get mouse coordinates (in region coordinates) */
1142         mval[0]= (event->x - ar->winrct.xmin);
1143         mval[1]= (event->y - ar->winrct.ymin);
1144         
1145         /* select mode is either replace (deselect all, then add) or add/extend */
1146         // XXX this is currently only available for normal select only
1147         if (RNA_boolean_get(op->ptr, "extend_select"))
1148                 selectmode= SELECT_INVERT;
1149         else
1150                 selectmode= SELECT_REPLACE;
1151         
1152         /* figure out action to take */
1153         if (RNA_enum_get(op->ptr, "left_right")) {
1154                 /* select all keys on same side of current frame as mouse */
1155                 float x;
1156                 
1157                 UI_view2d_region_to_view(v2d, mval[0], mval[1], &x, NULL);
1158                 if (x < CFRA)
1159                         RNA_int_set(op->ptr, "left_right", ACTKEYS_LRSEL_LEFT);
1160                 else    
1161                         RNA_int_set(op->ptr, "left_right", ACTKEYS_LRSEL_RIGHT);
1162                 
1163                 selectkeys_leftright(&ac, RNA_enum_get(op->ptr, "left_right"), selectmode);
1164         }
1165         else if (RNA_boolean_get(op->ptr, "column_select")) {
1166                 /* select all the keyframes that occur on the same frame as where the mouse clicked */
1167                 float x;
1168                 
1169                 /* figure out where (the frame) the mouse clicked, and set all keyframes in that frame */
1170                 UI_view2d_region_to_view(v2d, mval[0], mval[1], &x, NULL);
1171                 mouse_columnselect_action_keys(&ac, x);
1172         }
1173         else {
1174                 /* select keyframe under mouse */
1175                 mouse_action_keys(&ac, mval, selectmode);
1176                 // XXX activate transform...
1177         }
1178         
1179         /* set notifier tha things have changed */
1180         ED_area_tag_redraw(CTX_wm_area(C)); // FIXME... should be updating 'keyframes' data context or so instead!
1181         
1182         return OPERATOR_FINISHED;
1183 }
1184  
1185 void ACT_OT_keyframes_clickselect (wmOperatorType *ot)
1186 {
1187         PropertyRNA *prop;
1188         
1189         /* identifiers */
1190         ot->name= "Mouse Select Keys";
1191         ot->idname= "ACT_OT_keyframes_clickselect";
1192         
1193         /* api callbacks */
1194         ot->invoke= actkeys_clickselect_invoke;
1195         ot->poll= ED_operator_areaactive;
1196         
1197         /* id-props */
1198         prop= RNA_def_property(ot->srna, "left_right", PROP_ENUM, PROP_NONE); // ALTKEY
1199                 //RNA_def_property_enum_items(prop, prop_actkeys_clickselect_items);
1200         prop= RNA_def_property(ot->srna, "extend_select", PROP_BOOLEAN, PROP_NONE); // SHIFTKEY
1201         prop= RNA_def_property(ot->srna, "column_select", PROP_BOOLEAN, PROP_NONE); // CTRLKEY
1202 }
1203
1204 /* ************************************************************************** */