source/blender/editors/space_action/action_select.c

   1 /*
   2  * ***** BEGIN GPL LICENSE BLOCK *****
   3  *
   4  * This program is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU General Public License
   6  * as published by the Free Software Foundation; either version 2
   7  * of the License, or (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, write to the Free Software Foundation,
  16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  17  *
  18  * The Original Code is Copyright (C) 2008 Blender Foundation
  19  *
  20  * Contributor(s): Joshua Leung
  21  *
  22  * ***** END GPL LICENSE BLOCK *****
  23  */
  24
  25 /** \file blender/editors/space_action/action_select.c
  26  *  \ingroup spaction
  27  */
  28
  29
  30 #include <math.h>
  31 #include <stdlib.h>
  32 #include <string.h>
  33 #include <float.h>
  34
  35 #include "MEM_guardedalloc.h"
  36
  37 #include "BLI_blenlib.h"
  38 #include "BLI_math.h"
  39 #include "BLI_dlrbTree.h"
  40 #include "BLI_utildefines.h"
  41
  42 #include "DNA_anim_types.h"
  43 #include "DNA_gpencil_types.h"
  44 #include "DNA_object_types.h"
  45 #include "DNA_scene_types.h"
  46
  47 #include "RNA_access.h"
  48 #include "RNA_define.h"
  49
  50 #include "BKE_fcurve.h"
  51 #include "BKE_nla.h"
  52 #include "BKE_context.h"
  53
  54 #include "UI_view2d.h"
  55
  56 #include "ED_anim_api.h"
  57 #include "ED_gpencil.h"
  58 #include "ED_keyframes_draw.h"
  59 #include "ED_keyframes_edit.h"
  60 #include "ED_markers.h"
  61 #include "ED_screen.h"
  62
  63 #include "WM_api.h"
  64 #include "WM_types.h"
  65
  66 #include "action_intern.h"
  67
  68
  69 /* ************************************************************************** */
  70 /* KEYFRAMES STUFF */
  71
  72 /* ******************** Deselect All Operator ***************************** */
  73 /* This operator works in one of three ways:
  74  *      1) (de)select all (AKEY) - test if select all or deselect all
  75  *      2) invert all (CTRL-IKEY) - invert selection of all keyframes
  76  *      3) (de)select all - no testing is done; only for use internal tools as normal function...
  77  */
  78
  79 /* Deselects keyframes in the action editor
  80  *      - This is called by the deselect all operator, as well as other ones!
  81  *
  82  *      - test: check if select or deselect all
  83  *      - sel: how to select keyframes (SELECT_*)
  84  */
  85 static void deselect_action_keys (bAnimContext *ac, short test, short sel)
  86 {
  87         ListBase anim_data = {NULL, NULL};
  88         bAnimListElem *ale;
  89         int filter;
  90
  91         KeyframeEditData ked= {{NULL}};
  92         KeyframeEditFunc test_cb, sel_cb;
  93
  94         /* determine type-based settings */
  95         if (ac->datatype == ANIMCONT_GPENCIL)
  96                 filter= (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_LIST_VISIBLE | ANIMFILTER_NODUPLIS);
  97         else
  98                 filter= (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_LIST_VISIBLE /*| ANIMFILTER_CURVESONLY*/ | ANIMFILTER_NODUPLIS);
  99
 100         /* filter data */
 101         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
 102
 103         /* init BezTriple looping data */
 104         test_cb= ANIM_editkeyframes_ok(BEZT_OK_SELECTED);
 105
 106         /* See if we should be selecting or deselecting */
 107         if (test) {
 108                 for (ale= anim_data.first; ale; ale= ale->next) {
 109                         if (ale->type == ANIMTYPE_GPLAYER) {
 110                                 if (is_gplayer_frame_selected(ale->data)) {
 111                                         sel= SELECT_SUBTRACT;
 112                                         break;
 113                                 }
 114                         }
 115                         else {
 116                                 if (ANIM_fcurve_keyframes_loop(&ked, ale->key_data, NULL, test_cb, NULL)) {
 117                                         sel= SELECT_SUBTRACT;
 118                                         break;
 119                                 }
 120                         }
 121                 }
 122         }
 123
 124         /* convert sel to selectmode, and use that to get editor */
 125         sel_cb= ANIM_editkeyframes_select(sel);
 126
 127         /* Now set the flags */
 128         for (ale= anim_data.first; ale; ale= ale->next) {
 129                 if (ale->type == ANIMTYPE_GPLAYER)
 130                         set_gplayer_frame_selection(ale->data, sel);
 131                 else
 132                         ANIM_fcurve_keyframes_loop(&ked, ale->key_data, NULL, sel_cb, NULL);
 133         }
 134
 135         /* Cleanup */
 136         BLI_freelistN(&anim_data);
 137 }
 138
 139 /* ------------------- */
 140
 141 static int actkeys_deselectall_exec(bContext *C, wmOperator *op)
 142 {
 143         bAnimContext ac;
 144
 145         /* get editor data */
 146         if (ANIM_animdata_get_context(C, &ac) == 0)
 147                 return OPERATOR_CANCELLED;
 148
 149         /* 'standard' behaviour - check if selected, then apply relevant selection */
 150         if (RNA_boolean_get(op->ptr, "invert"))
 151                 deselect_action_keys(&ac, 0, SELECT_INVERT);
 152         else
 153                 deselect_action_keys(&ac, 1, SELECT_ADD);
 154
 155         /* set notifier that keyframe selection have changed */
 156         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_SELECTED, NULL);
 157
 158         return OPERATOR_FINISHED;
 159 }
 160
 161 void ACTION_OT_select_all_toggle (wmOperatorType *ot)
 162 {
 163         /* identifiers */
 164         ot->name= "Select All";
 165         ot->idname= "ACTION_OT_select_all_toggle";
 166         ot->description= "Toggle selection of all keyframes";
 167
 168         /* api callbacks */
 169         ot->exec= actkeys_deselectall_exec;
 170         ot->poll= ED_operator_action_active;
 171
 172         /* flags */
 173         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 174
 175         /* props */
 176         ot->prop= RNA_def_boolean(ot->srna, "invert", 0, "Invert", "");
 177 }
 178
 179 /* ******************** Border Select Operator **************************** */
 180 /* This operator currently works in one of three ways:
 181  *      -> BKEY         - 1) all keyframes within region are selected (ACTKEYS_BORDERSEL_ALLKEYS)
 182  *      -> ALT-BKEY - depending on which axis of the region was larger...
 183  *              -> 2) x-axis, so select all frames within frame range (ACTKEYS_BORDERSEL_FRAMERANGE)
 184  *              -> 3) y-axis, so select all frames within channels that region included (ACTKEYS_BORDERSEL_CHANNELS)
 185  */
 186
 187 /* defines for borderselect mode */
 188 enum {
 189         ACTKEYS_BORDERSEL_ALLKEYS       = 0,
 190         ACTKEYS_BORDERSEL_FRAMERANGE,
 191         ACTKEYS_BORDERSEL_CHANNELS,
 192 } /*eActKeys_BorderSelect_Mode*/;
 193
 194
 195 static void borderselect_action (bAnimContext *ac, rcti rect, short mode, short selectmode)
 196 {
 197         ListBase anim_data = {NULL, NULL};
 198         bAnimListElem *ale;
 199         int filter;
 200
 201         KeyframeEditData ked;
 202         KeyframeEditFunc ok_cb, select_cb;
 203         View2D *v2d= &ac->ar->v2d;
 204         rctf rectf;
 205         float ymin=0, ymax=(float)(-ACHANNEL_HEIGHT_HALF);
 206
 207         /* convert mouse coordinates to frame ranges and channel coordinates corrected for view pan/zoom */
 208         UI_view2d_region_to_view(v2d, rect.xmin, rect.ymin+2, &rectf.xmin, &rectf.ymin);
 209         UI_view2d_region_to_view(v2d, rect.xmax, rect.ymax-2, &rectf.xmax, &rectf.ymax);
 210
 211         /* filter data */
 212         filter= (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_LIST_VISIBLE | ANIMFILTER_LIST_CHANNELS | ANIMFILTER_NODUPLIS);
 213         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
 214
 215         /* get beztriple editing/validation funcs  */
 216         select_cb= ANIM_editkeyframes_select(selectmode);
 217
 218         if (ELEM(mode, ACTKEYS_BORDERSEL_FRAMERANGE, ACTKEYS_BORDERSEL_ALLKEYS))
 219                 ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAMERANGE);
 220         else
 221                 ok_cb= NULL;
 222
 223         /* init editing data */
 224         memset(&ked, 0, sizeof(KeyframeEditData));
 225
 226         /* loop over data, doing border select */
 227         for (ale= anim_data.first; ale; ale= ale->next) {
 228                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
 229
 230                 /* get new vertical minimum extent of channel */
 231                 ymin= ymax - ACHANNEL_STEP;
 232
 233                 /* set horizontal range (if applicable) */
 234                 if (ELEM(mode, ACTKEYS_BORDERSEL_FRAMERANGE, ACTKEYS_BORDERSEL_ALLKEYS)) {
 235                         /* if channel is mapped in NLA, apply correction */
 236                         if (adt) {
 237                                 ked.f1= BKE_nla_tweakedit_remap(adt, rectf.xmin, NLATIME_CONVERT_UNMAP);
 238                                 ked.f2= BKE_nla_tweakedit_remap(adt, rectf.xmax, NLATIME_CONVERT_UNMAP);
 239                         }
 240                         else {
 241                                 ked.f1= rectf.xmin;
 242                                 ked.f2= rectf.xmax;
 243                         }
 244                 }
 245
 246                 /* perform vertical suitability check (if applicable) */
 247                 if ( (mode == ACTKEYS_BORDERSEL_FRAMERANGE) ||
 248                         !((ymax < rectf.ymin) || (ymin > rectf.ymax)) )
 249                 {
 250                         /* loop over data selecting */
 251                         if (ale->type == ANIMTYPE_GPLAYER)
 252                                 borderselect_gplayer_frames(ale->data, rectf.xmin, rectf.xmax, selectmode);
 253                         else
 254                                 ANIM_animchannel_keyframes_loop(&ked, ac->ads, ale, ok_cb, select_cb, NULL);
 255                 }
 256
 257                 /* set minimum extent to be the maximum of the next channel */
 258                 ymax=ymin;
 259         }
 260
 261         /* cleanup */
 262         BLI_freelistN(&anim_data);
 263 }
 264
 265 /* ------------------- */
 266
 267 static int actkeys_borderselect_exec(bContext *C, wmOperator *op)
 268 {
 269         bAnimContext ac;
 270         rcti rect;
 271         short mode=0, selectmode=0;
 272         int gesture_mode;
 273
 274         /* get editor data */
 275         if (ANIM_animdata_get_context(C, &ac) == 0)
 276                 return OPERATOR_CANCELLED;
 277
 278         /* get settings from operator */
 279         rect.xmin= RNA_int_get(op->ptr, "xmin");
 280         rect.ymin= RNA_int_get(op->ptr, "ymin");
 281         rect.xmax= RNA_int_get(op->ptr, "xmax");
 282         rect.ymax= RNA_int_get(op->ptr, "ymax");
 283
 284         gesture_mode= RNA_int_get(op->ptr, "gesture_mode");
 285         if (gesture_mode == GESTURE_MODAL_SELECT)
 286                 selectmode = SELECT_ADD;
 287         else
 288                 selectmode = SELECT_SUBTRACT;
 289
 290         /* selection 'mode' depends on whether borderselect region only matters on one axis */
 291         if (RNA_boolean_get(op->ptr, "axis_range")) {
 292                 /* mode depends on which axis of the range is larger to determine which axis to use
 293                  *      - checking this in region-space is fine, as it's fundamentally still going to be a different rect size
 294                  *      - the frame-range select option is favoured over the channel one (x over y), as frame-range one is often
 295                  *        used for tweaking timing when "blocking", while channels is not that useful...
 296                  */
 297                 if ((rect.xmax - rect.xmin) >= (rect.ymax - rect.ymin))
 298                         mode= ACTKEYS_BORDERSEL_FRAMERANGE;
 299                 else
 300                         mode= ACTKEYS_BORDERSEL_CHANNELS;
 301         }
 302         else
 303                 mode= ACTKEYS_BORDERSEL_ALLKEYS;
 304
 305         /* apply borderselect action */
 306         borderselect_action(&ac, rect, mode, selectmode);
 307
 308         /* set notifier that keyframe selection have changed */
 309         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_SELECTED, NULL);
 310
 311         return OPERATOR_FINISHED;
 312 }
 313
 314 void ACTION_OT_select_border(wmOperatorType *ot)
 315 {
 316         /* identifiers */
 317         ot->name= "Border Select";
 318         ot->idname= "ACTION_OT_select_border";
 319         ot->description= "Select all keyframes within the specified region";
 320
 321         /* api callbacks */
 322         ot->invoke= WM_border_select_invoke;
 323         ot->exec= actkeys_borderselect_exec;
 324         ot->modal= WM_border_select_modal;
 325         ot->cancel= WM_border_select_cancel;
 326
 327         ot->poll= ED_operator_action_active;
 328
 329         /* flags */
 330         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 331
 332         /* rna */
 333         WM_operator_properties_gesture_border(ot, FALSE);
 334
 335         ot->prop= RNA_def_boolean(ot->srna, "axis_range", 0, "Axis Range", "");
 336 }
 337
 338 /* ******************** Column Select Operator **************************** */
 339 /* This operator works in one of four ways:
 340  *      - 1) select all keyframes in the same frame as a selected one  (KKEY)
 341  *      - 2) select all keyframes in the same frame as the current frame marker (CTRL-KKEY)
 342  *      - 3) select all keyframes in the same frame as a selected markers (SHIFT-KKEY)
 343  *      - 4) select all keyframes that occur between selected markers (ALT-KKEY)
 344  */
 345
 346 /* defines for column-select mode */
 347 static EnumPropertyItem prop_column_select_types[] = {
 348         {ACTKEYS_COLUMNSEL_KEYS, "KEYS", 0, "On Selected Keyframes", ""},
 349         {ACTKEYS_COLUMNSEL_CFRA, "CFRA", 0, "On Current Frame", ""},
 350         {ACTKEYS_COLUMNSEL_MARKERS_COLUMN, "MARKERS_COLUMN", 0, "On Selected Markers", ""},
 351         {ACTKEYS_COLUMNSEL_MARKERS_BETWEEN, "MARKERS_BETWEEN", 0, "Between Min/Max Selected Markers", ""},
 352         {0, NULL, 0, NULL, NULL}
 353 };
 354
 355 /* ------------------- */
 356
 357 /* Selects all visible keyframes between the specified markers */
 358 static void markers_selectkeys_between (bAnimContext *ac)
 359 {
 360         ListBase anim_data = {NULL, NULL};
 361         bAnimListElem *ale;
 362         int filter;
 363
 364         KeyframeEditFunc ok_cb, select_cb;
 365         KeyframeEditData ked= {{NULL}};
 366         float min, max;
 367
 368         /* get extreme markers */
 369         ED_markers_get_minmax(ac->markers, 1, &min, &max);
 370         min -= 0.5f;
 371         max += 0.5f;
 372
 373         /* get editing funcs + data */
 374         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAMERANGE);
 375         select_cb= ANIM_editkeyframes_select(SELECT_ADD);
 376
 377         ked.f1= min;
 378         ked.f2= max;
 379
 380         /* filter data */
 381         filter= (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_LIST_VISIBLE /*| ANIMFILTER_CURVESONLY */ | ANIMFILTER_NODUPLIS);
 382         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
 383
 384         /* select keys in-between */
 385         for (ale= anim_data.first; ale; ale= ale->next) {
 386                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
 387
 388                 if (adt) {
 389                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 0, 1);
 390                         ANIM_fcurve_keyframes_loop(&ked, ale->key_data, ok_cb, select_cb, NULL);
 391                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 1, 1);
 392                 }
 393                 else if (ale->type == ANIMTYPE_GPLAYER) {
 394                         borderselect_gplayer_frames(ale->data, min, max, SELECT_ADD);
 395                 }
 396                 else {
 397                         ANIM_fcurve_keyframes_loop(&ked, ale->key_data, ok_cb, select_cb, NULL);
 398                 }
 399         }
 400
 401         /* Cleanup */
 402         BLI_freelistN(&anim_data);
 403 }
 404
 405
 406 /* Selects all visible keyframes in the same frames as the specified elements */
 407 static void columnselect_action_keys (bAnimContext *ac, short mode)
 408 {
 409         ListBase anim_data= {NULL, NULL};
 410         bAnimListElem *ale;
 411         int filter;
 412
 413         Scene *scene= ac->scene;
 414         CfraElem *ce;
 415         KeyframeEditFunc select_cb, ok_cb;
 416         KeyframeEditData ked= {{NULL}};
 417
 418         /* initialise keyframe editing data */
 419
 420         /* build list of columns */
 421         switch (mode) {
 422                 case ACTKEYS_COLUMNSEL_KEYS: /* list of selected keys */
 423                         if (ac->datatype == ANIMCONT_GPENCIL) {
 424                                 filter= (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_LIST_VISIBLE);
 425                                 ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
 426
 427                                 for (ale= anim_data.first; ale; ale= ale->next)
 428                                         gplayer_make_cfra_list(ale->data, &ked.list, 1);
 429                         }
 430                         else {
 431                                 filter= (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_LIST_VISIBLE /*| ANIMFILTER_CURVESONLY*/);
 432                                 ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
 433
 434                                 for (ale= anim_data.first; ale; ale= ale->next)
 435                                         ANIM_fcurve_keyframes_loop(&ked, ale->key_data, NULL, bezt_to_cfraelem, NULL);
 436                         }
 437                         BLI_freelistN(&anim_data);
 438                         break;
 439
 440                 case ACTKEYS_COLUMNSEL_CFRA: /* current frame */
 441                         /* make a single CfraElem for storing this */
 442                         ce= MEM_callocN(sizeof(CfraElem), "cfraElem");
 443                         BLI_addtail(&ked.list, ce);
 444
 445                         ce->cfra= (float)CFRA;
 446                         break;
 447
 448                 case ACTKEYS_COLUMNSEL_MARKERS_COLUMN: /* list of selected markers */
 449                         ED_markers_make_cfra_list(ac->markers, &ked.list, SELECT);
 450                         break;
 451
 452                 default: /* invalid option */
 453                         return;
 454         }
 455
 456         /* set up BezTriple edit callbacks */
 457         select_cb= ANIM_editkeyframes_select(SELECT_ADD);
 458         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAME);
 459
 460         /* loop through all of the keys and select additional keyframes
 461          * based on the keys found to be selected above
 462          */
 463         if (ac->datatype == ANIMCONT_GPENCIL)
 464                 filter= (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_LIST_VISIBLE);
 465         else
 466                 filter= (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_LIST_VISIBLE /*| ANIMFILTER_CURVESONLY*/);
 467         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
 468
 469         for (ale= anim_data.first; ale; ale= ale->next) {
 470                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
 471
 472                 /* loop over cfraelems (stored in the KeyframeEditData->list)
 473                  *      - we need to do this here, as we can apply fewer NLA-mapping conversions
 474                  */
 475                 for (ce= ked.list.first; ce; ce= ce->next) {
 476                         /* set frame for validation callback to refer to */
 477                         if (adt)
 478                                 ked.f1= BKE_nla_tweakedit_remap(adt, ce->cfra, NLATIME_CONVERT_UNMAP);
 479                         else
 480                                 ked.f1= ce->cfra;
 481
 482                         /* select elements with frame number matching cfraelem */
 483                         if (ale->type == ANIMTYPE_GPLAYER)
 484                                 select_gpencil_frame(ale->data, ce->cfra, SELECT_ADD);
 485                         else
 486                                 ANIM_fcurve_keyframes_loop(&ked, ale->key_data, ok_cb, select_cb, NULL);
 487                 }
 488         }
 489
 490         /* free elements */
 491         BLI_freelistN(&ked.list);
 492         BLI_freelistN(&anim_data);
 493 }
 494
 495 /* ------------------- */
 496
 497 static int actkeys_columnselect_exec(bContext *C, wmOperator *op)
 498 {
 499         bAnimContext ac;
 500         short mode;
 501
 502         /* get editor data */
 503         if (ANIM_animdata_get_context(C, &ac) == 0)
 504                 return OPERATOR_CANCELLED;
 505
 506         /* action to take depends on the mode */
 507         mode= RNA_enum_get(op->ptr, "mode");
 508
 509         if (mode == ACTKEYS_COLUMNSEL_MARKERS_BETWEEN)
 510                 markers_selectkeys_between(&ac);
 511         else
 512                 columnselect_action_keys(&ac, mode);
 513
 514         /* set notifier that keyframe selection have changed */
 515         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_SELECTED, NULL);
 516
 517         return OPERATOR_FINISHED;
 518 }
 519
 520 void ACTION_OT_select_column (wmOperatorType *ot)
 521 {
 522         /* identifiers */
 523         ot->name= "Select All";
 524         ot->idname= "ACTION_OT_select_column";
 525         ot->description= "Select all keyframes on the specified frame(s)";
 526
 527         /* api callbacks */
 528         ot->exec= actkeys_columnselect_exec;
 529         ot->poll= ED_operator_action_active;
 530
 531         /* flags */
 532         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 533
 534         /* props */
 535         ot->prop= RNA_def_enum(ot->srna, "mode", prop_column_select_types, 0, "Mode", "");
 536 }
 537
 538 /* ******************** Select Linked Operator *********************** */
 539
 540 static int actkeys_select_linked_exec (bContext *C, wmOperator *UNUSED(op))
 541 {
 542         bAnimContext ac;
 543
 544         ListBase anim_data= {NULL, NULL};
 545         bAnimListElem *ale;
 546         int filter;
 547
 548         KeyframeEditFunc ok_cb = ANIM_editkeyframes_ok(BEZT_OK_SELECTED);
 549         KeyframeEditFunc sel_cb = ANIM_editkeyframes_select(SELECT_ADD);
 550
 551         /* get editor data */
 552         if (ANIM_animdata_get_context(C, &ac) == 0)
 553                 return OPERATOR_CANCELLED;
 554
 555         /* loop through all of the keys and select additional keyframes based on these */
 556         filter= (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_LIST_VISIBLE /*| ANIMFILTER_CURVESONLY*/ | ANIMFILTER_NODUPLIS);
 557         ANIM_animdata_filter(&ac, &anim_data, filter, ac.data, ac.datatype);
 558
 559         for (ale= anim_data.first; ale; ale= ale->next) {
 560                 FCurve *fcu= (FCurve *)ale->key_data;
 561
 562                 /* check if anything selected? */
 563                 if (ANIM_fcurve_keyframes_loop(NULL, fcu, NULL, ok_cb, NULL)) {
 564                         /* select every keyframe in this curve then */
 565                         ANIM_fcurve_keyframes_loop(NULL, fcu, NULL, sel_cb, NULL);
 566                 }
 567         }
 568
 569         /* Cleanup */
 570         BLI_freelistN(&anim_data);
 571
 572         /* set notifier that keyframe selection has changed */
 573         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_SELECTED, NULL);
 574
 575         return OPERATOR_FINISHED;
 576 }
 577
 578 void ACTION_OT_select_linked (wmOperatorType *ot)
 579 {
 580         /* identifiers */
 581         ot->name = "Select Linked";
 582         ot->idname= "ACTION_OT_select_linked";
 583         ot->description = "Select keyframes occurring the same F-Curves as selected ones";
 584
 585         /* api callbacks */
 586         ot->exec= actkeys_select_linked_exec;
 587         ot->poll= ED_operator_action_active;
 588
 589         /* flags */
 590         ot->flag= OPTYPE_REGISTER/*|OPTYPE_UNDO*/;
 591 }
 592
 593 /* ******************** Select More/Less Operators *********************** */
 594
 595 /* Common code to perform selection */
 596 static void select_moreless_action_keys (bAnimContext *ac, short mode)
 597 {
 598         ListBase anim_data= {NULL, NULL};
 599         bAnimListElem *ale;
 600         int filter;
 601
 602         KeyframeEditData ked= {{NULL}};
 603         KeyframeEditFunc build_cb;
 604
 605
 606         /* init selmap building data */
 607         build_cb= ANIM_editkeyframes_buildselmap(mode);
 608
 609         /* loop through all of the keys and select additional keyframes based on these */
 610         filter= (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_LIST_VISIBLE /*| ANIMFILTER_CURVESONLY*/ | ANIMFILTER_NODUPLIS);
 611         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
 612
 613         for (ale= anim_data.first; ale; ale= ale->next) {
 614                 FCurve *fcu= (FCurve *)ale->key_data;
 615
 616                 /* only continue if F-Curve has keyframes */
 617                 if (fcu->bezt == NULL)
 618                         continue;
 619
 620                 /* build up map of whether F-Curve's keyframes should be selected or not */
 621                 ked.data= MEM_callocN(fcu->totvert, "selmap actEdit more");
 622                 ANIM_fcurve_keyframes_loop(&ked, fcu, NULL, build_cb, NULL);
 623
 624                 /* based on this map, adjust the selection status of the keyframes */
 625                 ANIM_fcurve_keyframes_loop(&ked, fcu, NULL, bezt_selmap_flush, NULL);
 626
 627                 /* free the selmap used here */
 628                 MEM_freeN(ked.data);
 629                 ked.data= NULL;
 630         }
 631
 632         /* Cleanup */
 633         BLI_freelistN(&anim_data);
 634 }
 635
 636 /* ----------------- */
 637
 638 static int actkeys_select_more_exec (bContext *C, wmOperator *UNUSED(op))
 639 {
 640         bAnimContext ac;
 641
 642         /* get editor data */
 643         if (ANIM_animdata_get_context(C, &ac) == 0)
 644                 return OPERATOR_CANCELLED;
 645
 646         /* perform select changes */
 647         select_moreless_action_keys(&ac, SELMAP_MORE);
 648
 649         /* set notifier that keyframe selection has changed */
 650         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_SELECTED, NULL);
 651
 652         return OPERATOR_FINISHED;
 653 }
 654
 655 void ACTION_OT_select_more (wmOperatorType *ot)
 656 {
 657         /* identifiers */
 658         ot->name = "Select More";
 659         ot->idname= "ACTION_OT_select_more";
 660         ot->description = "Select keyframes beside already selected ones";
 661
 662         /* api callbacks */
 663         ot->exec= actkeys_select_more_exec;
 664         ot->poll= ED_operator_action_active;
 665
 666         /* flags */
 667         ot->flag= OPTYPE_REGISTER/*|OPTYPE_UNDO*/;
 668 }
 669
 670 /* ----------------- */
 671
 672 static int actkeys_select_less_exec (bContext *C, wmOperator *UNUSED(op))
 673 {
 674         bAnimContext ac;
 675
 676         /* get editor data */
 677         if (ANIM_animdata_get_context(C, &ac) == 0)
 678                 return OPERATOR_CANCELLED;
 679
 680         /* perform select changes */
 681         select_moreless_action_keys(&ac, SELMAP_LESS);
 682
 683         /* set notifier that keyframe selection has changed */
 684         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|NA_SELECTED, NULL);
 685
 686         return OPERATOR_FINISHED;
 687 }
 688
 689 void ACTION_OT_select_less (wmOperatorType *ot)
 690 {
 691         /* identifiers */
 692         ot->name = "Select Less";
 693         ot->idname= "ACTION_OT_select_less";
 694         ot->description = "Deselect keyframes on ends of selection islands";
 695
 696         /* api callbacks */
 697         ot->exec= actkeys_select_less_exec;
 698         ot->poll= ED_operator_action_active;
 699
 700         /* flags */
 701         ot->flag= OPTYPE_REGISTER/*|OPTYPE_UNDO*/;
 702 }
 703
 704 /* ******************** Select Left/Right Operator ************************* */
 705 /* Select keyframes left/right of the current frame indicator */
 706
 707 /* defines for left-right select tool */
 708 static EnumPropertyItem prop_actkeys_leftright_select_types[] = {
 709         {ACTKEYS_LRSEL_TEST, "CHECK", 0, "Check if Select Left or Right", ""},
 710         {ACTKEYS_LRSEL_LEFT, "LEFT", 0, "Before current frame", ""},
 711         {ACTKEYS_LRSEL_RIGHT, "RIGHT", 0, "After current frame", ""},
 712         {0, NULL, 0, NULL, NULL}
 713 };
 714
 715 /* --------------------------------- */
 716
 717 static void actkeys_select_leftright (bAnimContext *ac, short leftright, short select_mode)
 718 {
 719         ListBase anim_data = {NULL, NULL};
 720         bAnimListElem *ale;
 721         int filter;
 722
 723         KeyframeEditFunc ok_cb, select_cb;
 724         KeyframeEditData ked= {{NULL}};
 725         Scene *scene= ac->scene;
 726
 727         /* if select mode is replace, deselect all keyframes (and channels) first */
 728         if (select_mode==SELECT_REPLACE) {
 729                 select_mode= SELECT_ADD;
 730
 731                 /* deselect all other channels and keyframes */
 732                 ANIM_deselect_anim_channels(ac, ac->data, ac->datatype, 0, ACHANNEL_SETFLAG_CLEAR);
 733                 deselect_action_keys(ac, 0, SELECT_SUBTRACT);
 734         }
 735
 736         /* set callbacks and editing data */
 737         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAMERANGE);
 738         select_cb= ANIM_editkeyframes_select(select_mode);
 739
 740         if (leftright == ACTKEYS_LRSEL_LEFT) {
 741                 ked.f1 = MINAFRAMEF;
 742                 ked.f2 = (float)(CFRA + 0.1f);
 743         }
 744         else {
 745                 ked.f1 = (float)(CFRA - 0.1f);
 746                 ked.f2 = MAXFRAMEF;
 747         }
 748
 749         /* filter data */
 750         if (ac->datatype == ANIMCONT_GPENCIL)
 751                 filter= (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_LIST_VISIBLE | ANIMFILTER_NODUPLIS);
 752         else
 753                 filter= (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_LIST_VISIBLE /*| ANIMFILTER_CURVESONLY*/ | ANIMFILTER_NODUPLIS);
 754         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
 755
 756         /* select keys */
 757         for (ale= anim_data.first; ale; ale= ale->next) {
 758                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
 759
 760                 if (adt) {
 761                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 0, 1);
 762                         ANIM_fcurve_keyframes_loop(&ked, ale->key_data, ok_cb, select_cb, NULL);
 763                         ANIM_nla_mapping_apply_fcurve(adt, ale->key_data, 1, 1);
 764                 }
 765                 else if (ale->type == ANIMTYPE_GPLAYER)
 766                         borderselect_gplayer_frames(ale->data, ked.f1, ked.f2, select_mode);
 767                 else
 768                         ANIM_fcurve_keyframes_loop(&ked, ale->key_data, ok_cb, select_cb, NULL);
 769         }
 770
 771         /* Sync marker support */
 772         if (select_mode==SELECT_ADD) {
 773                 SpaceAction *saction= (SpaceAction *)ac->sl;
 774
 775                 if ((saction) && (saction->flag & SACTION_MARKERS_MOVE)) {
 776                         ListBase *markers = ED_animcontext_get_markers(ac);
 777                         TimeMarker *marker;
 778
 779                         for (marker= markers->first; marker; marker= marker->next) {
 780                                 if(     ((leftright == ACTKEYS_LRSEL_LEFT) && (marker->frame < CFRA)) ||
 781                                         ((leftright == ACTKEYS_LRSEL_RIGHT) && (marker->frame >= CFRA)) )
 782                                 {
 783                                         marker->flag |= SELECT;
 784                                 }
 785                                 else {
 786                                         marker->flag &= ~SELECT;
 787                                 }
 788                         }
 789                 }
 790         }
 791
 792         /* Cleanup */
 793         BLI_freelistN(&anim_data);
 794 }
 795
 796 /* ----------------- */
 797
 798 static int actkeys_select_leftright_exec (bContext *C, wmOperator *op)
 799 {
 800         bAnimContext ac;
 801         short leftright = RNA_enum_get(op->ptr, "mode");
 802         short selectmode;
 803
 804         /* get editor data */
 805         if (ANIM_animdata_get_context(C, &ac) == 0)
 806                 return OPERATOR_CANCELLED;
 807
 808         /* select mode is either replace (deselect all, then add) or add/extend */
 809         if (RNA_boolean_get(op->ptr, "extend"))
 810                 selectmode= SELECT_INVERT;
 811         else
 812                 selectmode= SELECT_REPLACE;
 813
 814         /* if "test" mode is set, we don't have any info to set this with */
 815         if (leftright == ACTKEYS_LRSEL_TEST)
 816                 return OPERATOR_CANCELLED;
 817
 818         /* do the selecting now */
 819         actkeys_select_leftright(&ac, leftright, selectmode);
 820
 821         /* set notifier that keyframe selection (and channels too) have changed */
 822         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|ND_ANIMCHAN|NA_SELECTED, NULL);
 823
 824         return OPERATOR_FINISHED;
 825 }
 826
 827 static int actkeys_select_leftright_invoke (bContext *C, wmOperator *op, wmEvent *event)
 828 {
 829         bAnimContext ac;
 830         short leftright = RNA_enum_get(op->ptr, "mode");
 831
 832         /* get editor data */
 833         if (ANIM_animdata_get_context(C, &ac) == 0)
 834                 return OPERATOR_CANCELLED;
 835
 836         /* handle mode-based testing */
 837         if (leftright == ACTKEYS_LRSEL_TEST) {
 838                 Scene *scene= ac.scene;
 839                 ARegion *ar= ac.ar;
 840                 View2D *v2d= &ar->v2d;
 841                 float x;
 842
 843                 /* determine which side of the current frame mouse is on */
 844                 UI_view2d_region_to_view(v2d, event->mval[0], event->mval[1], &x, NULL);
 845                 if (x < CFRA)
 846                         RNA_enum_set(op->ptr, "mode", ACTKEYS_LRSEL_LEFT);
 847                 else
 848                         RNA_enum_set(op->ptr, "mode", ACTKEYS_LRSEL_RIGHT);
 849         }
 850
 851         /* perform selection */
 852         return actkeys_select_leftright_exec(C, op);
 853 }
 854
 855 void ACTION_OT_select_leftright (wmOperatorType *ot)
 856 {
 857         /* identifiers */
 858         ot->name= "Select Left/Right";
 859         ot->idname= "ACTION_OT_select_leftright";
 860         ot->description= "Select keyframes to the left or the right of the current frame";
 861
 862         /* api callbacks  */
 863         ot->invoke= actkeys_select_leftright_invoke;
 864         ot->exec= actkeys_select_leftright_exec;
 865         ot->poll= ED_operator_action_active;
 866
 867         /* flags */
 868         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 869
 870         /* id-props */
 871         ot->prop= RNA_def_enum(ot->srna, "mode", prop_actkeys_leftright_select_types, ACTKEYS_LRSEL_TEST, "Mode", "");
 872         RNA_def_boolean(ot->srna, "extend", 0, "Extend Select", "");
 873 }
 874
 875 /* ******************** Mouse-Click Select Operator *********************** */
 876 /* This operator works in one of three ways:
 877  *      - 1) keyframe under mouse - no special modifiers
 878  *      - 2) all keyframes on the same side of current frame indicator as mouse - ALT modifier
 879  *      - 3) column select all keyframes in frame under mouse - CTRL modifier
 880  *
 881  * In addition to these basic options, the SHIFT modifier can be used to toggle the
 882  * selection mode between replacing the selection (without) and inverting the selection (with).
 883  */
 884
 885 /* sensitivity factor for frame-selections */
 886 #define FRAME_CLICK_THRESH              0.1f
 887
 888 /* ------------------- */
 889
 890 /* option 1) select keyframe directly under mouse */
 891 static void actkeys_mselect_single (bAnimContext *ac, bAnimListElem *ale, short select_mode, float selx)
 892 {
 893         KeyframeEditData ked= {{NULL}};
 894         KeyframeEditFunc select_cb, ok_cb;
 895
 896         /* get functions for selecting keyframes */
 897         select_cb= ANIM_editkeyframes_select(select_mode);
 898         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAME);
 899         ked.f1= selx;
 900
 901         /* select the nominated keyframe on the given frame */
 902         if (ale->type == ANIMTYPE_GPLAYER)
 903                 select_gpencil_frame(ale->data, selx, select_mode);
 904         else
 905                 ANIM_animchannel_keyframes_loop(&ked, ac->ads, ale, ok_cb, select_cb, NULL);
 906 }
 907
 908 /* Option 2) Selects all the keyframes on either side of the current frame (depends on which side the mouse is on) */
 909 /* (see actkeys_select_leftright) */
 910
 911 /* Option 3) Selects all visible keyframes in the same frame as the mouse click */
 912 static void actkeys_mselect_column(bAnimContext *ac, short select_mode, float selx)
 913 {
 914         ListBase anim_data= {NULL, NULL};
 915         bAnimListElem *ale;
 916         int filter;
 917
 918         KeyframeEditFunc select_cb, ok_cb;
 919         KeyframeEditData ked= {{NULL}};
 920
 921         /* initialise keyframe editing data */
 922
 923         /* set up BezTriple edit callbacks */
 924         select_cb= ANIM_editkeyframes_select(select_mode);
 925         ok_cb= ANIM_editkeyframes_ok(BEZT_OK_FRAME);
 926
 927         /* loop through all of the keys and select additional keyframes
 928          * based on the keys found to be selected above
 929          */
 930         if (ac->datatype == ANIMCONT_GPENCIL)
 931                 filter= (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_LIST_VISIBLE /*| ANIMFILTER_CURVESONLY */ | ANIMFILTER_NODUPLIS);
 932         else
 933                 filter= (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_LIST_VISIBLE | ANIMFILTER_NODUPLIS);
 934         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
 935
 936         for (ale= anim_data.first; ale; ale= ale->next) {
 937                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
 938
 939                 /* set frame for validation callback to refer to */
 940                 if (adt)
 941                         ked.f1= BKE_nla_tweakedit_remap(adt, selx, NLATIME_CONVERT_UNMAP);
 942                 else
 943                         ked.f1= selx;
 944
 945                 /* select elements with frame number matching cfra */
 946                 if (ale->type == ANIMTYPE_GPLAYER)
 947                         select_gpencil_frame(ale->key_data, selx, select_mode);
 948                 else
 949                         ANIM_fcurve_keyframes_loop(&ked, ale->key_data, ok_cb, select_cb, NULL);
 950         }
 951
 952         /* free elements */
 953         BLI_freelistN(&ked.list);
 954         BLI_freelistN(&anim_data);
 955 }
 956
 957 /* ------------------- */
 958
 959 static void mouse_action_keys (bAnimContext *ac, const int mval[2], short select_mode, short column)
 960 {
 961         ListBase anim_data = {NULL, NULL};
 962         DLRBT_Tree anim_keys;
 963         bAnimListElem *ale;
 964         int filter;
 965
 966         View2D *v2d= &ac->ar->v2d;
 967         bDopeSheet *ads = NULL;
 968         int channel_index;
 969         short found = 0;
 970         float selx = 0.0f;
 971         float x, y;
 972         rctf rectf;
 973
 974         /* get dopesheet info */
 975         if (ac->datatype == ANIMCONT_DOPESHEET)
 976                 ads= ac->data;
 977
 978         /* use View2D to determine the index of the channel (i.e a row in the list) where keyframe was */
 979         UI_view2d_region_to_view(v2d, mval[0], mval[1], &x, &y);
 980         UI_view2d_listview_view_to_cell(v2d, 0, ACHANNEL_STEP, 0, (float)ACHANNEL_HEIGHT_HALF, x, y, NULL, &channel_index);
 981
 982         /* x-range to check is +/- 7 (in screen/region-space) on either side of mouse click (size of keyframe icon) */
 983         UI_view2d_region_to_view(v2d, mval[0]-7, mval[1], &rectf.xmin, &rectf.ymin);
 984         UI_view2d_region_to_view(v2d, mval[0]+7, mval[1], &rectf.xmax, &rectf.ymax);
 985
 986         /* filter data */
 987         filter= (ANIMFILTER_DATA_VISIBLE | ANIMFILTER_LIST_VISIBLE | ANIMFILTER_LIST_CHANNELS);
 988         ANIM_animdata_filter(ac, &anim_data, filter, ac->data, ac->datatype);
 989
 990         /* try to get channel */
 991         ale= BLI_findlink(&anim_data, channel_index);
 992         if (ale == NULL) {
 993                 /* channel not found */
 994                 printf("Error: animation channel (index = %d) not found in mouse_action_keys() \n", channel_index);
 995                 BLI_freelistN(&anim_data);
 996                 return;
 997         }
 998         else {
 999                 /* found match - must return here... */
1000                 AnimData *adt= ANIM_nla_mapping_get(ac, ale);
1001                 ActKeyColumn *ak, *akn=NULL;
1002
1003                 /* make list of keyframes */
1004                 BLI_dlrbTree_init(&anim_keys);
1005
1006                 if (ale->key_data) {
1007                         switch (ale->datatype) {
1008                                 case ALE_SCE:
1009                                 {
1010                                         Scene *scene= (Scene *)ale->key_data;
1011                                         scene_to_keylist(ads, scene, &anim_keys, NULL);
1012                                 }
1013                                         break;
1014                                 case ALE_OB:
1015                                 {
1016                                         Object *ob= (Object *)ale->key_data;
1017                                         ob_to_keylist(ads, ob, &anim_keys, NULL);
1018                                 }
1019                                         break;
1020                                 case ALE_ACT:
1021                                 {
1022                                         bAction *act= (bAction *)ale->key_data;
1023                                         action_to_keylist(adt, act, &anim_keys, NULL);
1024                                 }
1025                                         break;
1026                                 case ALE_FCURVE:
1027                                 {
1028                                         FCurve *fcu= (FCurve *)ale->key_data;
1029                                         fcurve_to_keylist(adt, fcu, &anim_keys, NULL);
1030                                 }
1031                                         break;
1032                         }
1033                 }
1034                 else if (ale->type == ANIMTYPE_SUMMARY) {
1035                         /* dopesheet summary covers everything */
1036                         summary_to_keylist(ac, &anim_keys, NULL);
1037                 }
1038                 else if (ale->type == ANIMTYPE_GROUP) {
1039                         // TODO: why don't we just give groups key_data too?
1040                         bActionGroup *agrp= (bActionGroup *)ale->data;
1041                         agroup_to_keylist(adt, agrp, &anim_keys, NULL);
1042                 }
1043                 else if (ale->type == ANIMTYPE_GPLAYER) {
1044                         // TODO: why don't we just give gplayers key_data too?
1045                         bGPDlayer *gpl = (bGPDlayer *)ale->data;
1046                         gpl_to_keylist(ads, gpl, &anim_keys);
1047                 }
1048
1049                 /* start from keyframe at root of BST, traversing until we find one within the range that was clicked on */
1050                 for (ak= anim_keys.root; ak; ak= akn) {
1051                         if (IN_RANGE(ak->cfra, rectf.xmin, rectf.xmax)) {
1052                                 /* set the frame to use, and apply inverse-correction for NLA-mapping
1053                                  * so that the frame will get selected by the selection functions without
1054                                  * requiring to map each frame once again...
1055                                  */
1056                                 selx= BKE_nla_tweakedit_remap(adt, ak->cfra, NLATIME_CONVERT_UNMAP);
1057                                 found= 1;
1058                                 break;
1059                         }
1060                         else if (ak->cfra < rectf.xmin)
1061                                 akn= ak->right;
1062                         else
1063                                 akn= ak->left;
1064                 }
1065
1066                 /* remove active channel from list of channels for separate treatment (since it's needed later on) */
1067                 BLI_remlink(&anim_data, ale);
1068
1069                 /* cleanup temporary lists */
1070                 BLI_dlrbTree_free(&anim_keys);
1071
1072                 /* free list of channels, since it's not used anymore */
1073                 BLI_freelistN(&anim_data);
1074         }
1075
1076         /* for replacing selection, firstly need to clear existing selection */
1077         if (select_mode == SELECT_REPLACE) {
1078                 /* reset selection mode for next steps */
1079                 select_mode = SELECT_ADD;
1080
1081                 /* deselect all keyframes */
1082                 deselect_action_keys(ac, 0, SELECT_SUBTRACT);
1083
1084                 /* highlight channel clicked on */
1085                 if (ELEM(ac->datatype, ANIMCONT_ACTION, ANIMCONT_DOPESHEET)) {
1086                         /* deselect all other channels first */
1087                         ANIM_deselect_anim_channels(ac, ac->data, ac->datatype, 0, ACHANNEL_SETFLAG_CLEAR);
1088
1089                         /* Highlight Action-Group or F-Curve? */
1090                         if (ale && ale->data) {
1091                                 if (ale->type == ANIMTYPE_GROUP) {
1092                                         bActionGroup *agrp= ale->data;
1093
1094                                         agrp->flag |= AGRP_SELECTED;
1095                                         ANIM_set_active_channel(ac, ac->data, ac->datatype, filter, agrp, ANIMTYPE_GROUP);
1096                                 }
1097                                 else if (ale->type == ANIMTYPE_FCURVE) {
1098                                         FCurve *fcu= ale->data;
1099
1100                                         fcu->flag |= FCURVE_SELECTED;
1101                                         ANIM_set_active_channel(ac, ac->data, ac->datatype, filter, fcu, ANIMTYPE_FCURVE);
1102                                 }
1103                         }
1104                 }
1105                 else if (ac->datatype == ANIMCONT_GPENCIL) {
1106                         /* deselect all other channels first */
1107                         ANIM_deselect_anim_channels(ac, ac->data, ac->datatype, 0, ACHANNEL_SETFLAG_CLEAR);
1108
1109                         /* Highlight GPencil Layer */
1110                         if ((ale && ale->data) && (ale->type == ANIMTYPE_GPLAYER)) {
1111                                 bGPDlayer *gpl = ale->data;
1112
1113                                 gpl->flag |= GP_LAYER_SELECT;
1114                                 //gpencil_layer_setactive(gpd, gpl);
1115                         }
1116                 }
1117         }
1118
1119         /* only select keyframes if we clicked on a valid channel and hit something */
1120         if (ale) {
1121                 if (found) {
1122                         /* apply selection to keyframes */
1123                         if (column) {
1124                                 /* select all keyframes in the same frame as the one we hit on the active channel */
1125                                 actkeys_mselect_column(ac, select_mode, selx);
1126                         }
1127                         else {
1128                                 /* select the nominated keyframe on the given frame */
1129                                 actkeys_mselect_single(ac, ale, select_mode, selx);
1130                         }
1131                 }
1132
1133                 /* free this channel */
1134                 MEM_freeN(ale);
1135         }
1136 }
1137
1138 /* handle clicking */
1139 static int actkeys_clickselect_invoke(bContext *C, wmOperator *op, wmEvent *event)
1140 {
1141         bAnimContext ac;
1142         ARegion *ar;
1143         short selectmode, column;
1144
1145         /* get editor data */
1146         if (ANIM_animdata_get_context(C, &ac) == 0)
1147                 return OPERATOR_CANCELLED;
1148
1149         /* get useful pointers from animation context data */
1150         ar= ac.ar;
1151
1152         /* select mode is either replace (deselect all, then add) or add/extend */
1153         if (RNA_boolean_get(op->ptr, "extend"))
1154                 selectmode= SELECT_INVERT;
1155         else
1156                 selectmode= SELECT_REPLACE;
1157
1158         /* column selection */
1159         column= RNA_boolean_get(op->ptr, "column");
1160
1161         /* select keyframe(s) based upon mouse position*/
1162         mouse_action_keys(&ac, event->mval, selectmode, column);
1163
1164         /* set notifier that keyframe selection (and channels too) have changed */
1165         WM_event_add_notifier(C, NC_ANIMATION|ND_KEYFRAME|ND_ANIMCHAN|NA_SELECTED, NULL);
1166
1167         /* for tweak grab to work */
1168         return OPERATOR_FINISHED|OPERATOR_PASS_THROUGH;
1169 }
1170
1171 void ACTION_OT_clickselect (wmOperatorType *ot)
1172 {
1173         /* identifiers */
1174         ot->name= "Mouse Select Keys";
1175         ot->idname= "ACTION_OT_clickselect";
1176         ot->description= "Select keyframes by clicking on them";
1177
1178         /* api callbacks - absolutely no exec() this yet... */
1179         ot->invoke= actkeys_clickselect_invoke;
1180         ot->poll= ED_operator_action_active;
1181
1182         /* flags */
1183         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
1184
1185         /* id-props */
1186         RNA_def_boolean(ot->srna, "extend", 0, "Extend Select", ""); // SHIFTKEY
1187         RNA_def_boolean(ot->srna, "column", 0, "Column Select", ""); // ALTKEY
1188 }
1189
1190 /* ************************************************************************** */