source/blender/editors/space_action/action_select.c

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