split BKE_utildefines.h, now it only has blender specific defines like GS() MAKE_ID...
[blender.git] / source / blender / editors / space_node / node_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  * All rights reserved.
22  *
23  * 
24  * Contributor(s): Blender Foundation, Nathan Letwory
25  *
26  * ***** END GPL LICENSE BLOCK *****
27  */
28
29 #include <stdio.h>
30
31 #include "DNA_node_types.h"
32 #include "DNA_scene_types.h"
33
34 #include "BKE_context.h"
35
36 #include "BLI_rect.h"
37 #include "BLI_utildefines.h"
38
39 #include "ED_screen.h"
40 #include "ED_types.h"
41
42 #include "RNA_access.h"
43 #include "RNA_define.h"
44
45 #include "WM_api.h"
46 #include "WM_types.h"
47
48 #include "UI_view2d.h"
49  
50 #include "node_intern.h"
51
52 /* ****** helpers ****** */
53
54 static bNode *node_under_mouse(bNodeTree *ntree, int mx, int my)
55 {
56         bNode *node;
57         
58         for(next_node(ntree); (node=next_node(NULL));) {
59                 /* node body (header and scale are in other operators) */
60                 if (BLI_in_rctf(&node->totr, mx, my))
61                         return node;
62         }
63         return NULL;
64 }
65
66 /* ****** Click Select ****** */
67  
68 static bNode *node_mouse_select(SpaceNode *snode, ARegion *ar, short *mval, short extend)
69 {
70         bNode *node;
71         float mx, my;
72         
73         /* get mouse coordinates in view2d space */
74         mx= (float)mval[0];
75         my= (float)mval[1];
76         
77         UI_view2d_region_to_view(&ar->v2d, mval[0], mval[1], &mx, &my);
78         
79         /* find the closest visible node */
80         node = node_under_mouse(snode->edittree, mx, my);
81         
82         if (node) {
83                 if (extend == 0) {
84                         node_deselectall(snode);
85                         node->flag |= SELECT;
86                 }
87                 else
88                         node->flag ^= SELECT;
89                         
90                 node_set_active(snode, node);
91         }
92
93         return node;
94 }
95
96 static int node_select_exec(bContext *C, wmOperator *op)
97 {
98         SpaceNode *snode= CTX_wm_space_node(C);
99         ARegion *ar= CTX_wm_region(C);
100         short mval[2];
101         short extend;
102         bNode *node= NULL;
103         
104         /* get settings from RNA properties for operator */
105         mval[0] = RNA_int_get(op->ptr, "mouse_x");
106         mval[1] = RNA_int_get(op->ptr, "mouse_y");
107         
108         extend = RNA_boolean_get(op->ptr, "extend");
109         
110         /* perform the select */
111         node= node_mouse_select(snode, ar, mval, extend);
112         
113         /* send notifiers */
114         WM_event_add_notifier(C, NC_NODE|NA_SELECTED, NULL);
115         
116         /* allow tweak event to work too */
117         return OPERATOR_FINISHED|OPERATOR_PASS_THROUGH;
118 }
119
120 static int node_select_invoke(bContext *C, wmOperator *op, wmEvent *event)
121 {
122         ARegion *ar= CTX_wm_region(C);
123         short mval[2];  
124         
125         mval[0]= event->x - ar->winrct.xmin;
126         mval[1]= event->y - ar->winrct.ymin;
127         
128         RNA_int_set(op->ptr, "mouse_x", mval[0]);
129         RNA_int_set(op->ptr, "mouse_y", mval[1]);
130
131         return node_select_exec(C,op);
132 }
133
134
135 void NODE_OT_select(wmOperatorType *ot)
136 {
137         /* identifiers */
138         ot->name= "Select";
139         ot->idname= "NODE_OT_select";
140         
141         /* api callbacks */
142         ot->invoke= node_select_invoke;
143         ot->poll= ED_operator_node_active;
144         
145         /* flags */
146         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
147         
148         /* properties */
149         RNA_def_int(ot->srna, "mouse_x", 0, INT_MIN, INT_MAX, "Mouse X", "", INT_MIN, INT_MAX);
150         RNA_def_int(ot->srna, "mouse_y", 0, INT_MIN, INT_MAX, "Mouse Y", "", INT_MIN, INT_MAX);
151         RNA_def_boolean(ot->srna, "extend", 0, "Extend", "");
152 }
153
154 /* ****** Border Select ****** */
155
156 static int node_borderselect_exec(bContext *C, wmOperator *op)
157 {
158         SpaceNode *snode= CTX_wm_space_node(C);
159         ARegion *ar= CTX_wm_region(C);
160         bNode *node;
161         rcti rect;
162         rctf rectf;
163         int gesture_mode= RNA_int_get(op->ptr, "gesture_mode");
164         
165         rect.xmin= RNA_int_get(op->ptr, "xmin");
166         rect.ymin= RNA_int_get(op->ptr, "ymin");
167         UI_view2d_region_to_view(&ar->v2d, rect.xmin, rect.ymin, &rectf.xmin, &rectf.ymin);
168         
169         rect.xmax= RNA_int_get(op->ptr, "xmax");
170         rect.ymax= RNA_int_get(op->ptr, "ymax");
171         UI_view2d_region_to_view(&ar->v2d, rect.xmax, rect.ymax, &rectf.xmax, &rectf.ymax);
172         
173         for(node= snode->edittree->nodes.first; node; node= node->next) {
174                 if(BLI_isect_rctf(&rectf, &node->totr, NULL)) {
175                         if(gesture_mode==GESTURE_MODAL_SELECT)
176                                 node->flag |= SELECT;
177                         else
178                                 node->flag &= ~SELECT;
179                 }
180         }
181         
182         WM_event_add_notifier(C, NC_NODE|NA_SELECTED, NULL);
183
184         return OPERATOR_FINISHED;
185 }
186
187 static int node_border_select_invoke(bContext *C, wmOperator *op, wmEvent *event)
188 {
189         int tweak = RNA_boolean_get(op->ptr, "tweak");
190         
191         if (tweak) {
192                 /* prevent initiating the border select if the mouse is over a node */
193                 /* this allows border select on empty space, but drag-translate on nodes */
194                 SpaceNode *snode= CTX_wm_space_node(C);
195                 ARegion *ar= CTX_wm_region(C);
196                 short mval[2];
197                 float mx, my;
198                 
199                 mval[0]= event->x - ar->winrct.xmin;
200                 mval[1]= event->y - ar->winrct.ymin;
201                 
202                 /* get mouse coordinates in view2d space */
203                 mx= (float)mval[0];
204                 my= (float)mval[1];
205                 
206                 UI_view2d_region_to_view(&ar->v2d, mval[0], mval[1], &mx, &my);
207                 
208                 if (node_under_mouse(snode->edittree, mx, my))
209                         return OPERATOR_CANCELLED|OPERATOR_PASS_THROUGH;
210         }
211         
212         return WM_border_select_invoke(C, op, event);
213 }
214
215 void NODE_OT_select_border(wmOperatorType *ot)
216 {
217         /* identifiers */
218         ot->name= "Border Select";
219         ot->idname= "NODE_OT_select_border";
220         ot->description= "Use box selection to select nodes";
221         
222         /* api callbacks */
223         ot->invoke= node_border_select_invoke;
224         ot->exec= node_borderselect_exec;
225         ot->modal= WM_border_select_modal;
226         
227         ot->poll= ED_operator_node_active;
228         
229         /* flags */
230         ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
231         
232         /* rna */
233         WM_operator_properties_gesture_border(ot, FALSE);
234         RNA_def_boolean(ot->srna, "tweak", 0, "Tweak", "Only activate when mouse is not over a node - useful for tweak gesture");
235 }
236
237 /* ****** Select/Deselect All ****** */
238
239 static int node_select_all_exec(bContext *C, wmOperator *UNUSED(op))
240 {
241         SpaceNode *snode = CTX_wm_space_node(C);
242         bNode *first = snode->edittree->nodes.first;
243         bNode *node;
244         int count= 0;
245
246         for(node=first; node; node=node->next)
247                 if(node->flag & NODE_SELECT)
248                         count++;
249
250         if(count) {
251                 for(node=first; node; node=node->next)
252                         node->flag &= ~NODE_SELECT;
253         }
254         else {
255                 for(node=first; node; node=node->next)
256                         node->flag |= NODE_SELECT;
257         }
258         
259         WM_event_add_notifier(C, NC_NODE|NA_SELECTED, NULL);
260         return OPERATOR_FINISHED;
261 }
262
263 void NODE_OT_select_all(wmOperatorType *ot)
264 {
265         /* identifiers */
266         ot->name = "Select or Deselect All";
267         ot->description = "(De)select all nodes";
268         ot->idname = "NODE_OT_select_all";
269         
270         /* api callbacks */
271         ot->exec = node_select_all_exec;
272         ot->poll = ED_operator_node_active;
273         
274         /* flags */
275         ot->flag = OPTYPE_REGISTER|OPTYPE_UNDO;
276 }
277
278 /* ****** Select Linked To ****** */
279
280 static int node_select_linked_to_exec(bContext *C, wmOperator *UNUSED(op))
281 {
282         SpaceNode *snode = CTX_wm_space_node(C);
283         bNodeLink *link;
284         bNode *node;
285         
286         for (node=snode->edittree->nodes.first; node; node=node->next)
287                 node->flag &= ~NODE_TEST;
288
289         for (link=snode->edittree->links.first; link; link=link->next) {
290                 if (link->fromnode->flag & NODE_SELECT)
291                         link->tonode->flag |= NODE_TEST;
292         }
293         
294         for (node=snode->edittree->nodes.first; node; node=node->next) {
295                 if (node->flag & NODE_TEST)
296                         node->flag |= NODE_SELECT;
297         }
298         
299         WM_event_add_notifier(C, NC_NODE|NA_SELECTED, NULL);
300         return OPERATOR_FINISHED;
301 }
302
303 void NODE_OT_select_linked_to(wmOperatorType *ot)
304 {
305         /* identifiers */
306         ot->name = "Select Linked To";
307         ot->description = "Select nodes linked to the selected ones";
308         ot->idname = "NODE_OT_select_linked_to";
309         
310         /* api callbacks */
311         ot->exec = node_select_linked_to_exec;
312         ot->poll = ED_operator_node_active;
313         
314         /* flags */
315         ot->flag = OPTYPE_REGISTER|OPTYPE_UNDO;
316 }
317
318 /* ****** Select Linked From ****** */
319
320 static int node_select_linked_from_exec(bContext *C, wmOperator *UNUSED(op))
321 {
322         SpaceNode *snode = CTX_wm_space_node(C);
323         bNodeLink *link;
324         bNode *node;
325         
326         for(node=snode->edittree->nodes.first; node; node=node->next)
327                 node->flag &= ~NODE_TEST;
328
329         for(link=snode->edittree->links.first; link; link=link->next) {
330                 if(link->tonode->flag & NODE_SELECT)
331                         link->fromnode->flag |= NODE_TEST;
332         }
333         
334         for(node=snode->edittree->nodes.first; node; node=node->next) {
335                 if(node->flag & NODE_TEST)
336                         node->flag |= NODE_SELECT;
337         }
338         
339         WM_event_add_notifier(C, NC_NODE|NA_SELECTED, NULL);
340         return OPERATOR_FINISHED;
341 }
342
343 void NODE_OT_select_linked_from(wmOperatorType *ot)
344 {
345         /* identifiers */
346         ot->name = "Select Linked From";
347         ot->description = "Select nodes linked from the selected ones";
348         ot->idname = "NODE_OT_select_linked_from";
349         
350         /* api callbacks */
351         ot->exec = node_select_linked_from_exec;
352         ot->poll = ED_operator_node_active;
353         
354         /* flags */
355         ot->flag = OPTYPE_REGISTER|OPTYPE_UNDO;
356 }
357
358 /* ****** Select Same Type ****** */
359
360 static int node_select_same_type_exec(bContext *C, wmOperator *UNUSED(op))
361 {
362         SpaceNode *snode = CTX_wm_space_node(C);
363
364         node_select_same_type(snode);
365         WM_event_add_notifier(C, NC_NODE|NA_SELECTED, NULL);
366         return OPERATOR_FINISHED;
367 }
368
369 void NODE_OT_select_same_type(wmOperatorType *ot)
370 {
371         /* identifiers */
372         ot->name = "Select Same Type";
373         ot->description = "Select all the same type";
374         ot->idname = "NODE_OT_select_same_type";
375         
376         /* api callbacks */
377         ot->exec = node_select_same_type_exec;
378         ot->poll = ED_operator_node_active;
379         
380         /* flags */
381         ot->flag = OPTYPE_REGISTER|OPTYPE_UNDO;
382 }
383
384 /* ****** Select The Next/Prev Node Of The Same Type ****** */
385
386 static int node_select_same_type_next_exec(bContext *C, wmOperator *UNUSED(op))
387 {
388         SpaceNode *snode = CTX_wm_space_node(C);
389
390         node_select_same_type_np(snode, 0);
391         WM_event_add_notifier(C, NC_NODE|NA_SELECTED, NULL);
392         return OPERATOR_FINISHED;
393 }
394
395 void NODE_OT_select_same_type_next(wmOperatorType *ot)
396 {
397         /* identifiers */
398         ot->name = "Select Same Type Next";
399         ot->description = "Select the next node of the same type.";
400         ot->idname = "NODE_OT_select_same_type_next";
401         
402         /* api callbacks */
403         ot->exec = node_select_same_type_next_exec;
404         ot->poll = ED_operator_node_active;
405         
406         /* flags */
407         ot->flag = OPTYPE_REGISTER|OPTYPE_UNDO;
408 }
409
410 static int node_select_same_type_prev_exec(bContext *C, wmOperator *UNUSED(op))
411 {
412         SpaceNode *snode = CTX_wm_space_node(C);
413
414         node_select_same_type_np(snode, 1);
415         WM_event_add_notifier(C, NC_NODE|NA_SELECTED, NULL);
416         return OPERATOR_FINISHED;
417 }
418
419 void NODE_OT_select_same_type_prev(wmOperatorType *ot)
420 {
421         /* identifiers */
422         ot->name = "Select Same Type Prev";
423         ot->description = "Select the prev node of the same type.";
424         ot->idname = "NODE_OT_select_same_type_prev";
425         
426         /* api callbacks */
427         ot->exec = node_select_same_type_prev_exec;
428         ot->poll = ED_operator_node_active;
429         
430         /* flags */
431         ot->flag = OPTYPE_REGISTER|OPTYPE_UNDO;
432 }