node_select.c

Go to the documentation of this file.

/*
 * $Id: node_select.c 37246 2011-06-06 11:04:54Z nazgul $
 *
 * ***** BEGIN GPL LICENSE BLOCK *****
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version. 
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 *
 * The Original Code is Copyright (C) 2008 Blender Foundation.
 * All rights reserved.
 *
 * 
 * Contributor(s): Blender Foundation, Nathan Letwory
 *
 * ***** END GPL LICENSE BLOCK *****
 */

#include <stdio.h>

#include "DNA_node_types.h"
#include "DNA_scene_types.h"

#include "BKE_context.h"

#include "BLI_rect.h"
#include "BLI_utildefines.h"

#include "ED_screen.h"
#include "ED_types.h"

#include "RNA_access.h"
#include "RNA_define.h"

#include "WM_api.h"
#include "WM_types.h"

#include "UI_view2d.h"
 
#include "node_intern.h"

/* ****** helpers ****** */

static bNode *node_under_mouse(bNodeTree *ntree, int mx, int my)
{
        bNode *node;
        
        for(next_node(ntree); (node=next_node(NULL));) {
                /* node body (header and scale are in other operators) */
                if (BLI_in_rctf(&node->totr, mx, my))
                        return node;
        }
        return NULL;
}

/* ****** Click Select ****** */
 
static bNode *node_mouse_select(SpaceNode *snode, ARegion *ar, const int mval[2], short extend)
{
        bNode *node;
        float mx, my;
        
        /* get mouse coordinates in view2d space */
        mx= (float)mval[0];
        my= (float)mval[1];
        
        UI_view2d_region_to_view(&ar->v2d, mval[0], mval[1], &mx, &my);
        
        /* find the closest visible node */
        node = node_under_mouse(snode->edittree, mx, my);
        
        if (node) {
                if (extend == 0) {
                        node_deselectall(snode);
                        node->flag |= SELECT;
                }
                else
                        node->flag ^= SELECT;
                        
                node_set_active(snode, node);
        }

        return node;
}

static int node_select_exec(bContext *C, wmOperator *op)
{
        SpaceNode *snode= CTX_wm_space_node(C);
        ARegion *ar= CTX_wm_region(C);
        int mval[2];
        short extend;
        bNode *node= NULL;
        
        /* get settings from RNA properties for operator */
        mval[0] = RNA_int_get(op->ptr, "mouse_x");
        mval[1] = RNA_int_get(op->ptr, "mouse_y");
        
        extend = RNA_boolean_get(op->ptr, "extend");
        
        /* perform the select */
        node= node_mouse_select(snode, ar, mval, extend);
        
        /* send notifiers */
        WM_event_add_notifier(C, NC_NODE|NA_SELECTED, NULL);
        
        /* allow tweak event to work too */
        return OPERATOR_FINISHED|OPERATOR_PASS_THROUGH;
}

static int node_select_invoke(bContext *C, wmOperator *op, wmEvent *event)
{
        RNA_int_set(op->ptr, "mouse_x", event->mval[0]);
        RNA_int_set(op->ptr, "mouse_y", event->mval[1]);

        return node_select_exec(C,op);
}


void NODE_OT_select(wmOperatorType *ot)
{
        /* identifiers */
        ot->name= "Select";
        ot->idname= "NODE_OT_select";
        ot->description= "Select node under cursor";
        
        /* api callbacks */
        ot->invoke= node_select_invoke;
        ot->poll= ED_operator_node_active;
        
        /* flags */
        ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
        
        /* properties */
        RNA_def_int(ot->srna, "mouse_x", 0, INT_MIN, INT_MAX, "Mouse X", "", INT_MIN, INT_MAX);
        RNA_def_int(ot->srna, "mouse_y", 0, INT_MIN, INT_MAX, "Mouse Y", "", INT_MIN, INT_MAX);
        RNA_def_boolean(ot->srna, "extend", 0, "Extend", "");
}

/* ****** Border Select ****** */

static int node_borderselect_exec(bContext *C, wmOperator *op)
{
        SpaceNode *snode= CTX_wm_space_node(C);
        ARegion *ar= CTX_wm_region(C);
        bNode *node;
        rcti rect;
        rctf rectf;
        int gesture_mode= RNA_int_get(op->ptr, "gesture_mode");
        
        rect.xmin= RNA_int_get(op->ptr, "xmin");
        rect.ymin= RNA_int_get(op->ptr, "ymin");
        UI_view2d_region_to_view(&ar->v2d, rect.xmin, rect.ymin, &rectf.xmin, &rectf.ymin);
        
        rect.xmax= RNA_int_get(op->ptr, "xmax");
        rect.ymax= RNA_int_get(op->ptr, "ymax");
        UI_view2d_region_to_view(&ar->v2d, rect.xmax, rect.ymax, &rectf.xmax, &rectf.ymax);
        
        for(node= snode->edittree->nodes.first; node; node= node->next) {
                if(BLI_isect_rctf(&rectf, &node->totr, NULL)) {
                        if(gesture_mode==GESTURE_MODAL_SELECT)
                                node->flag |= SELECT;
                        else
                                node->flag &= ~SELECT;
                }
        }
        
        WM_event_add_notifier(C, NC_NODE|NA_SELECTED, NULL);

        return OPERATOR_FINISHED;
}

static int node_border_select_invoke(bContext *C, wmOperator *op, wmEvent *event)
{
        int tweak = RNA_boolean_get(op->ptr, "tweak");
        
        if (tweak) {
                /* prevent initiating the border select if the mouse is over a node */
                /* this allows border select on empty space, but drag-translate on nodes */
                SpaceNode *snode= CTX_wm_space_node(C);
                ARegion *ar= CTX_wm_region(C);
                float mx, my;

                UI_view2d_region_to_view(&ar->v2d, event->mval[0], event->mval[1], &mx, &my);
                
                if (node_under_mouse(snode->edittree, mx, my))
                        return OPERATOR_CANCELLED|OPERATOR_PASS_THROUGH;
        }
        
        return WM_border_select_invoke(C, op, event);
}

void NODE_OT_select_border(wmOperatorType *ot)
{
        /* identifiers */
        ot->name= "Border Select";
        ot->idname= "NODE_OT_select_border";
        ot->description= "Use box selection to select nodes";
        
        /* api callbacks */
        ot->invoke= node_border_select_invoke;
        ot->exec= node_borderselect_exec;
        ot->modal= WM_border_select_modal;
        ot->cancel= WM_border_select_cancel;
        
        ot->poll= ED_operator_node_active;
        
        /* flags */
        ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
        
        /* rna */
        WM_operator_properties_gesture_border(ot, FALSE);
        RNA_def_boolean(ot->srna, "tweak", 0, "Tweak", "Only activate when mouse is not over a node - useful for tweak gesture");
}

/* ****** Select/Deselect All ****** */

static int node_select_all_exec(bContext *C, wmOperator *UNUSED(op))
{
        SpaceNode *snode = CTX_wm_space_node(C);
        bNode *first = snode->edittree->nodes.first;
        bNode *node;
        int count= 0;

        for(node=first; node; node=node->next)
                if(node->flag & NODE_SELECT)
                        count++;

        if(count) {
                for(node=first; node; node=node->next)
                        node->flag &= ~NODE_SELECT;
        }
        else {
                for(node=first; node; node=node->next)
                        node->flag |= NODE_SELECT;
        }
        
        WM_event_add_notifier(C, NC_NODE|NA_SELECTED, NULL);
        return OPERATOR_FINISHED;
}

void NODE_OT_select_all(wmOperatorType *ot)
{
        /* identifiers */
        ot->name = "Select or Deselect All";
        ot->description = "(De)select all nodes";
        ot->idname = "NODE_OT_select_all";
        
        /* api callbacks */
        ot->exec = node_select_all_exec;
        ot->poll = ED_operator_node_active;
        
        /* flags */
        ot->flag = OPTYPE_REGISTER|OPTYPE_UNDO;
}

/* ****** Select Linked To ****** */

static int node_select_linked_to_exec(bContext *C, wmOperator *UNUSED(op))
{
        SpaceNode *snode = CTX_wm_space_node(C);
        bNodeLink *link;
        bNode *node;
        
        for (node=snode->edittree->nodes.first; node; node=node->next)
                node->flag &= ~NODE_TEST;

        for (link=snode->edittree->links.first; link; link=link->next) {
                if (link->fromnode && link->tonode && (link->fromnode->flag & NODE_SELECT))
                        link->tonode->flag |= NODE_TEST;
        }
        
        for (node=snode->edittree->nodes.first; node; node=node->next) {
                if (node->flag & NODE_TEST)
                        node->flag |= NODE_SELECT;
        }
        
        WM_event_add_notifier(C, NC_NODE|NA_SELECTED, NULL);
        return OPERATOR_FINISHED;
}

void NODE_OT_select_linked_to(wmOperatorType *ot)
{
        /* identifiers */
        ot->name = "Select Linked To";
        ot->description = "Select nodes linked to the selected ones";
        ot->idname = "NODE_OT_select_linked_to";
        
        /* api callbacks */
        ot->exec = node_select_linked_to_exec;
        ot->poll = ED_operator_node_active;
        
        /* flags */
        ot->flag = OPTYPE_REGISTER|OPTYPE_UNDO;
}

/* ****** Select Linked From ****** */

static int node_select_linked_from_exec(bContext *C, wmOperator *UNUSED(op))
{
        SpaceNode *snode = CTX_wm_space_node(C);
        bNodeLink *link;
        bNode *node;
        
        for(node=snode->edittree->nodes.first; node; node=node->next)
                node->flag &= ~NODE_TEST;

        for(link=snode->edittree->links.first; link; link=link->next) {
                if(link->fromnode && link->tonode && (link->tonode->flag & NODE_SELECT))
                        link->fromnode->flag |= NODE_TEST;
        }
        
        for(node=snode->edittree->nodes.first; node; node=node->next) {
                if(node->flag & NODE_TEST)
                        node->flag |= NODE_SELECT;
        }
        
        WM_event_add_notifier(C, NC_NODE|NA_SELECTED, NULL);
        return OPERATOR_FINISHED;
}

void NODE_OT_select_linked_from(wmOperatorType *ot)
{
        /* identifiers */
        ot->name = "Select Linked From";
        ot->description = "Select nodes linked from the selected ones";
        ot->idname = "NODE_OT_select_linked_from";
        
        /* api callbacks */
        ot->exec = node_select_linked_from_exec;
        ot->poll = ED_operator_node_active;
        
        /* flags */
        ot->flag = OPTYPE_REGISTER|OPTYPE_UNDO;
}

/* ****** Select Same Type ****** */

static int node_select_same_type_exec(bContext *C, wmOperator *UNUSED(op))
{
        SpaceNode *snode = CTX_wm_space_node(C);

        node_select_same_type(snode);
        WM_event_add_notifier(C, NC_NODE|NA_SELECTED, NULL);
        return OPERATOR_FINISHED;
}

void NODE_OT_select_same_type(wmOperatorType *ot)
{
        /* identifiers */
        ot->name = "Select Same Type";
        ot->description = "Select all the same type";
        ot->idname = "NODE_OT_select_same_type";
        
        /* api callbacks */
        ot->exec = node_select_same_type_exec;
        ot->poll = ED_operator_node_active;
        
        /* flags */
        ot->flag = OPTYPE_REGISTER|OPTYPE_UNDO;
}

/* ****** Select The Next/Prev Node Of The Same Type ****** */

static int node_select_same_type_next_exec(bContext *C, wmOperator *UNUSED(op))
{
        SpaceNode *snode = CTX_wm_space_node(C);

        node_select_same_type_np(snode, 0);
        WM_event_add_notifier(C, NC_NODE|NA_SELECTED, NULL);
        return OPERATOR_FINISHED;
}

void NODE_OT_select_same_type_next(wmOperatorType *ot)
{
        /* identifiers */
        ot->name = "Select Same Type Next";
        ot->description = "Select the next node of the same type.";
        ot->idname = "NODE_OT_select_same_type_next";
        
        /* api callbacks */
        ot->exec = node_select_same_type_next_exec;
        ot->poll = ED_operator_node_active;
        
        /* flags */
        ot->flag = OPTYPE_REGISTER|OPTYPE_UNDO;
}

static int node_select_same_type_prev_exec(bContext *C, wmOperator *UNUSED(op))
{
        SpaceNode *snode = CTX_wm_space_node(C);

        node_select_same_type_np(snode, 1);
        WM_event_add_notifier(C, NC_NODE|NA_SELECTED, NULL);
        return OPERATOR_FINISHED;
}

void NODE_OT_select_same_type_prev(wmOperatorType *ot)
{
        /* identifiers */
        ot->name = "Select Same Type Prev";
        ot->description = "Select the prev node of the same type.";
        ot->idname = "NODE_OT_select_same_type_prev";
        
        /* api callbacks */
        ot->exec = node_select_same_type_prev_exec;
        ot->poll = ED_operator_node_active;
        
        /* flags */
        ot->flag = OPTYPE_REGISTER|OPTYPE_UNDO;
}