doxygen: blender/editors tagged.
[blender.git] / source / blender / editors / space_node / node_select.c
index 6746c21ebcfb467e1ce44c445c16e3ad7e3c7570..4aa56ef370d4fbcd3b44e4f6fa2dc4523bdd3170 100644 (file)
@@ -1,5 +1,5 @@
-/**
- * $Id:
+/*
+ * $Id$
  *
  * ***** BEGIN GPL LICENSE BLOCK *****
  *
  *
  * ***** BEGIN GPL LICENSE BLOCK *****
  *
@@ -15,7 +15,7 @@
  *
  * You should have received a copy of the GNU General Public License
  * along with this program; if not, write to the Free Software Foundation,
  *
  * 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
+ * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  *
  * The Original Code is Copyright (C) 2008 Blender Foundation.
  * All rights reserved.
  *
  * The Original Code is Copyright (C) 2008 Blender Foundation.
  * All rights reserved.
  * ***** END GPL LICENSE BLOCK *****
  */
 
  * ***** END GPL LICENSE BLOCK *****
  */
 
+/** \file blender/editors/space_node/node_select.c
+ *  \ingroup spnode
+ */
+
+
 #include <stdio.h>
 
 #include "DNA_node_types.h"
 #include <stdio.h>
 
 #include "DNA_node_types.h"
-#include "DNA_material_types.h"
-#include "DNA_texture_types.h"
 #include "DNA_scene_types.h"
 #include "DNA_scene_types.h"
-#include "DNA_screen_types.h"
-#include "DNA_space_types.h"
 
 #include "BKE_context.h"
 
 #include "BKE_context.h"
-#include "BKE_node.h"
-#include "BKE_global.h"
 
 #include "BLI_rect.h"
 
 #include "BLI_rect.h"
+#include "BLI_utildefines.h"
 
 
-#include "ED_space_api.h"
 #include "ED_screen.h"
 #include "ED_types.h"
 
 #include "ED_screen.h"
 #include "ED_types.h"
 
 #include "UI_view2d.h"
  
 #include "node_intern.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 void node_mouse_select(SpaceNode *snode, ARegion *ar, short *mval, short extend)
+static bNode *node_mouse_select(SpaceNode *snode, ARegion *ar, short *mval, short extend)
 {
        bNode *node;
        float mx, my;
        
 {
        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);
        
        mx= (float)mval[0];
        my= (float)mval[1];
        
        UI_view2d_region_to_view(&ar->v2d, mval[0], mval[1], &mx, &my);
        
-       for(next_node(snode->edittree); (node=next_node(NULL));) {
-               
-               /* first check for the headers or scaling widget */
-               /* XXX if(node->flag & NODE_HIDDEN) {
-                       if(do_header_hidden_node(snode, node, mx, my))
-                               return 1;
-               }
-               else {
-                       if(do_header_node(snode, node, mx, my))
-                               return 1;
-               }*/
-               
-               /* node body */
-               if(BLI_in_rctf(&node->totr, mx, my))
-                       break;
-       }
-       if(node) {
-               if((extend & KM_SHIFT)==0)
-                       node_deselectall(snode, 0);
-               
-               if(extend & KM_SHIFT) {
-                       if(node->flag & SELECT)
-                               node->flag &= ~SELECT;
-                       else
-                               node->flag |= SELECT;
+       /* 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
                }
                else
-                       node->flag |= SELECT;
-               
+                       node->flag ^= SELECT;
+                       
                node_set_active(snode, node);
                node_set_active(snode, node);
-               
-               /* viewer linking */
-               if(extend & KM_CTRL)
-                       ;//     node_link_viewer(snode, node);
-               
-               //std_rmouse_transform(node_transform_ext);     /* does undo push for select */
        }
        }
+
+       return node;
 }
 
 static int node_select_exec(bContext *C, wmOperator *op)
 {
 }
 
 static int node_select_exec(bContext *C, wmOperator *op)
 {
-       // XXX wmWindow *window=  CTX_wm_window(C);
-       SpaceNode *snode= (SpaceNode*)CTX_wm_space_data(C);
+       SpaceNode *snode= CTX_wm_space_node(C);
        ARegion *ar= CTX_wm_region(C);
        ARegion *ar= CTX_wm_region(C);
-       int select_type;
        short mval[2];
        short extend;
        short mval[2];
        short extend;
-
-       select_type = RNA_enum_get(op->ptr, "select_type");
+       bNode *node= NULL;
        
        
-       switch (select_type) {
-               case NODE_SELECT_MOUSE:
-                       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");
-                       node_mouse_select(snode, ar, mval, extend);
-                       break;
-       }
-
-       /* need refresh/a notifier vs compo notifier */
-       // XXX WM_event_add_notifier(C, NC_SCENE|ND_NODES, NULL); /* Do we need to pass the scene? */
-       ED_region_tag_redraw(ar);
+       /* 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;
 }
 
        
        /* allow tweak event to work too */
        return OPERATOR_FINISHED|OPERATOR_PASS_THROUGH;
 }
 
-static int node_select_modal(bContext *C, wmOperator *op, wmEvent *event)
-{
-       /* execute the events */
-       switch (event->type) {
-               case MOUSEMOVE:
-                       printf("%d %d\n", event->x, event->y);
-                       break;
-               case SELECTMOUSE:
-                       //if (event->val==0) {
-                               /* calculate overall delta mouse-movement for redo */
-                               printf("done translating\n");
-                               //WM_cursor_restore(CTX_wm_window(C));
-                               
-                               return OPERATOR_FINISHED;
-                       //}
-                       break;
-       }
-
-       return OPERATOR_RUNNING_MODAL;
-}
-
 static int node_select_invoke(bContext *C, wmOperator *op, wmEvent *event)
 {
        ARegion *ar= CTX_wm_region(C);
 static int node_select_invoke(bContext *C, wmOperator *op, wmEvent *event)
 {
        ARegion *ar= CTX_wm_region(C);
@@ -167,56 +136,22 @@ static int node_select_invoke(bContext *C, wmOperator *op, wmEvent *event)
        return node_select_exec(C,op);
 }
 
        return node_select_exec(C,op);
 }
 
-static int node_extend_select_invoke(bContext *C, wmOperator *op, wmEvent *event)
-{
-       RNA_boolean_set(op->ptr, "extend", KM_SHIFT);
-
-       return node_select_invoke(C, op, event);
-}
-
-/* operators */
-
-static EnumPropertyItem prop_select_items[] = {
-       {NODE_SELECT_MOUSE, "NORMAL", 0, "Normal Select", "Select using the mouse"},
-       {0, NULL, 0, NULL, NULL}};
-
-void NODE_OT_select_extend(wmOperatorType *ot)
-{
-       // XXX - Todo - This should just be a toggle option for NODE_OT_select not its own op
-       /* identifiers */
-       ot->name= "Activate/Select (Shift)";
-       ot->idname= "NODE_OT_select_extend";
-       
-       /* api callbacks */
-       ot->invoke= node_extend_select_invoke;
-       ot->poll= ED_operator_node_active;
-       
-       /* flags */
-       ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
-       
-       RNA_def_enum(ot->srna, "select_type", prop_select_items, 0, "Select Type", "");
-       
-       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", "");
-}
 
 void NODE_OT_select(wmOperatorType *ot)
 {
        /* identifiers */
 
 void NODE_OT_select(wmOperatorType *ot)
 {
        /* identifiers */
-       ot->name= "Activate/Select";
+       ot->name= "Select";
        ot->idname= "NODE_OT_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;
        
        /* api callbacks */
        ot->invoke= node_select_invoke;
        ot->poll= ED_operator_node_active;
-       ot->modal= node_select_modal;
        
        /* flags */
        ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
        
        
        /* flags */
        ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
        
-       RNA_def_enum(ot->srna, "select_type", prop_select_items, 0, "Select Type", "");
-       
+       /* 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", "");
        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", "");
@@ -224,22 +159,14 @@ void NODE_OT_select(wmOperatorType *ot)
 
 /* ****** Border Select ****** */
 
 
 /* ****** Border Select ****** */
 
-static EnumPropertyItem prop_select_types[] = {
-       {NODE_EXCLUSIVE, "EXCLUSIVE", 0, "Exclusive", ""}, /* right mouse */
-       {NODE_EXTEND, "EXTEND", 0, "Extend", ""}, /* left mouse */
-       {0, NULL, 0, NULL, NULL}
-};
-
 static int node_borderselect_exec(bContext *C, wmOperator *op)
 {
 static int node_borderselect_exec(bContext *C, wmOperator *op)
 {
-       SpaceNode *snode= (SpaceNode*)CTX_wm_space_data(C);
+       SpaceNode *snode= CTX_wm_space_node(C);
        ARegion *ar= CTX_wm_region(C);
        bNode *node;
        rcti rect;
        rctf rectf;
        ARegion *ar= CTX_wm_region(C);
        bNode *node;
        rcti rect;
        rctf rectf;
-       short val;
-       
-       val= RNA_int_get(op->ptr, "event_type");
+       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");
        
        rect.xmin= RNA_int_get(op->ptr, "xmin");
        rect.ymin= RNA_int_get(op->ptr, "ymin");
@@ -249,31 +176,57 @@ static int node_borderselect_exec(bContext *C, wmOperator *op)
        rect.ymax= RNA_int_get(op->ptr, "ymax");
        UI_view2d_region_to_view(&ar->v2d, rect.xmax, rect.ymax, &rectf.xmax, &rectf.ymax);
        
        rect.ymax= RNA_int_get(op->ptr, "ymax");
        UI_view2d_region_to_view(&ar->v2d, rect.xmax, rect.ymax, &rectf.xmax, &rectf.ymax);
        
-       if (snode->edittree == NULL) // XXX should this be in poll()? - campbell
-               return OPERATOR_FINISHED;
-
        for(node= snode->edittree->nodes.first; node; node= node->next) {
                if(BLI_isect_rctf(&rectf, &node->totr, NULL)) {
        for(node= snode->edittree->nodes.first; node; node= node->next) {
                if(BLI_isect_rctf(&rectf, &node->totr, NULL)) {
-                       if(val==NODE_EXTEND)
+                       if(gesture_mode==GESTURE_MODAL_SELECT)
                                node->flag |= SELECT;
                        else
                                node->flag &= ~SELECT;
                }
        }
        
                                node->flag |= SELECT;
                        else
                                node->flag &= ~SELECT;
                }
        }
        
-       WM_event_add_notifier(C, NC_SCENE|ND_NODES, NULL); /* Do we need to pass the scene? */
+       WM_event_add_notifier(C, NC_NODE|NA_SELECTED, NULL);
 
        return OPERATOR_FINISHED;
 }
 
 
        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);
+               short mval[2];
+               float mx, my;
+               
+               mval[0]= event->x - ar->winrct.xmin;
+               mval[1]= event->y - ar->winrct.ymin;
+               
+               /* 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);
+               
+               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";
 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 */
        
        /* api callbacks */
-       ot->invoke= WM_border_select_invoke;
+       ot->invoke= node_border_select_invoke;
        ot->exec= node_borderselect_exec;
        ot->modal= WM_border_select_modal;
        
        ot->exec= node_borderselect_exec;
        ot->modal= WM_border_select_modal;
        
@@ -283,11 +236,203 @@ void NODE_OT_select_border(wmOperatorType *ot)
        ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
        
        /* rna */
        ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
        
        /* rna */
-       RNA_def_int(ot->srna, "event_type", 0, INT_MIN, INT_MAX, "Event Type", "", INT_MIN, INT_MAX);
-       RNA_def_int(ot->srna, "xmin", 0, INT_MIN, INT_MAX, "X Min", "", INT_MIN, INT_MAX);
-       RNA_def_int(ot->srna, "xmax", 0, INT_MIN, INT_MAX, "X Max", "", INT_MIN, INT_MAX);
-       RNA_def_int(ot->srna, "ymin", 0, INT_MIN, INT_MAX, "Y Min", "", INT_MIN, INT_MAX);
-       RNA_def_int(ot->srna, "ymax", 0, INT_MIN, INT_MAX, "Y Max", "", INT_MIN, INT_MAX);
+       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;
 
 
-       RNA_def_enum(ot->srna, "type", prop_select_types, 0, "Type", "");
+       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;
 }
 }