72461cfb2a8b3bd1193d4dedb8a00635c209f18f
[blender.git] / source / blender / editors / space_node / node_draw.c
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version. 
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * The Original Code is Copyright (C) 2008 Blender Foundation.
19  * All rights reserved.
20  *
21  * The Original Code is: all of this file.
22  * Contributor(s): Nathan Letwory
23  *
24  * ***** END GPL LICENSE BLOCK *****
25  */
26
27 /** \file blender/editors/space_node/node_draw.c
28  *  \ingroup spnode
29  *  \brief higher level node drawing for the node editor.
30  */
31
32 #include "DNA_node_types.h"
33 #include "DNA_object_types.h"
34 #include "DNA_space_types.h"
35 #include "DNA_screen_types.h"
36
37 #include "BLI_math.h"
38 #include "BLI_blenlib.h"
39
40 #include "BLF_translation.h"
41
42 #include "BKE_context.h"
43 #include "BKE_depsgraph.h"
44 #include "BKE_main.h"
45 #include "BKE_node.h"
46
47 #include "BIF_gl.h"
48 #include "BIF_glutil.h"
49
50 #include "WM_api.h"
51 #include "WM_types.h"
52
53 #include "ED_node.h"
54 #include "ED_gpencil.h"
55 #include "ED_space_api.h"
56
57 #include "UI_resources.h"
58 #include "UI_view2d.h"
59
60 #include "RNA_access.h"
61
62 #include "node_intern.h"  /* own include */
63 #include "COM_compositor.h"
64
65 /* XXX interface.h */
66 extern void ui_dropshadow(const rctf *rct, float radius, float aspect, float alpha, int select);
67
68 /* XXX update functions for node editor are a mess, needs a clear concept */
69 void ED_node_tree_update(SpaceNode *snode, Scene *scene)
70 {
71         snode_set_context(snode, scene);
72         
73         if (snode->nodetree && snode->nodetree->id.us == 0)
74                 snode->nodetree->id.us = 1;
75 }
76
77 void ED_node_changed_update(ID *id, bNode *node)
78 {
79         bNodeTree *nodetree, *edittree;
80         int treetype;
81
82         node_tree_from_ID(id, &nodetree, &edittree, &treetype);
83
84         if (treetype == NTREE_SHADER) {
85                 DAG_id_tag_update(id, 0);
86
87                 if (GS(id->name) == ID_MA)
88                         WM_main_add_notifier(NC_MATERIAL | ND_SHADING_DRAW, id);
89                 else if (GS(id->name) == ID_LA)
90                         WM_main_add_notifier(NC_LAMP | ND_LIGHTING_DRAW, id);
91                 else if (GS(id->name) == ID_WO)
92                         WM_main_add_notifier(NC_WORLD | ND_WORLD_DRAW, id);
93         }
94         else if (treetype == NTREE_COMPOSIT) {
95                 if (node)
96                         nodeUpdate(edittree, node);
97                 /* don't use NodeTagIDChanged, it gives far too many recomposites for image, scene layers, ... */
98                         
99                 node = node_tree_get_editgroup(nodetree);
100                 if (node)
101                         nodeUpdateID(nodetree, node->id);
102
103                 WM_main_add_notifier(NC_SCENE | ND_NODES, id);
104         }
105         else if (treetype == NTREE_TEXTURE) {
106                 DAG_id_tag_update(id, 0);
107                 WM_main_add_notifier(NC_TEXTURE | ND_NODES, id);
108         }
109 }
110
111 static int has_nodetree(bNodeTree *ntree, bNodeTree *lookup)
112 {
113         bNode *node;
114         
115         if (ntree == lookup)
116                 return 1;
117         
118         for (node = ntree->nodes.first; node; node = node->next)
119                 if (node->type == NODE_GROUP && node->id)
120                         if (has_nodetree((bNodeTree *)node->id, lookup))
121                                 return 1;
122         
123         return 0;
124 }
125
126 typedef struct NodeUpdateCalldata {
127         bNodeTree *ntree;
128         bNode *node;
129 } NodeUpdateCalldata;
130 static void node_generic_update_cb(void *calldata, ID *owner_id, bNodeTree *ntree)
131 {
132         NodeUpdateCalldata *cd = (NodeUpdateCalldata *)calldata;
133         /* check if nodetree uses the group stored in calldata */
134         if (has_nodetree(ntree, cd->ntree))
135                 ED_node_changed_update(owner_id, cd->node);
136 }
137 void ED_node_generic_update(Main *bmain, bNodeTree *ntree, bNode *node)
138 {
139         bNodeTreeType *tti = ntreeGetType(ntree->type);
140         NodeUpdateCalldata cd;
141         cd.ntree = ntree;
142         cd.node = node;
143         /* look through all datablocks, to support groups */
144         tti->foreach_nodetree(bmain, &cd, node_generic_update_cb);
145         
146         if (ntree->type == NTREE_TEXTURE)
147                 ntreeTexCheckCyclics(ntree);
148 }
149
150 static int compare_nodes(bNode *a, bNode *b)
151 {
152         bNode *parent;
153         /* These tell if either the node or any of the parent nodes is selected.
154          * A selected parent means an unselected node is also in foreground!
155          */
156         int a_select = (a->flag & NODE_SELECT), b_select = (b->flag & NODE_SELECT);
157         int a_active = (a->flag & NODE_ACTIVE), b_active = (b->flag & NODE_ACTIVE);
158         
159         /* if one is an ancestor of the other */
160         /* XXX there might be a better sorting algorithm for stable topological sort, this is O(n^2) worst case */
161         for (parent = a->parent; parent; parent = parent->parent) {
162                 /* if b is an ancestor, it is always behind a */
163                 if (parent == b)
164                         return 1;
165                 /* any selected ancestor moves the node forward */
166                 if (parent->flag & NODE_ACTIVE)
167                         a_active = 1;
168                 if (parent->flag & NODE_SELECT)
169                         a_select = 1;
170         }
171         for (parent = b->parent; parent; parent = parent->parent) {
172                 /* if a is an ancestor, it is always behind b */
173                 if (parent == a)
174                         return 0;
175                 /* any selected ancestor moves the node forward */
176                 if (parent->flag & NODE_ACTIVE)
177                         b_active = 1;
178                 if (parent->flag & NODE_SELECT)
179                         b_select = 1;
180         }
181
182         /* if one of the nodes is in the background and the other not */
183         if ((a->flag & NODE_BACKGROUND) && !(b->flag & NODE_BACKGROUND))
184                 return 0;
185         else if (!(a->flag & NODE_BACKGROUND) && (b->flag & NODE_BACKGROUND))
186                 return 1;
187         
188         /* if one has a higher selection state (active > selected > nothing) */
189         if (!b_active && a_active)
190                 return 1;
191         else if (!b_select && (a_active || a_select))
192                 return 1;
193         
194         return 0;
195 }
196
197 /* Sorts nodes by selection: unselected nodes first, then selected,
198  * then the active node at the very end. Relative order is kept intact!
199  */
200 void ED_node_sort(bNodeTree *ntree)
201 {
202         /* merge sort is the algorithm of choice here */
203         bNode *first_a, *first_b, *node_a, *node_b, *tmp;
204         int totnodes = BLI_countlist(&ntree->nodes);
205         int k, a, b;
206         
207         k = 1;
208         while (k < totnodes) {
209                 first_a = first_b = ntree->nodes.first;
210                 
211                 do {
212                         /* setup first_b pointer */
213                         for (b = 0; b < k && first_b; ++b) {
214                                 first_b = first_b->next;
215                         }
216                         /* all batches merged? */
217                         if (first_b == NULL)
218                                 break;
219                         
220                         /* merge batches */
221                         node_a = first_a;
222                         node_b = first_b;
223                         a = b = 0;
224                         while (a < k && b < k && node_b) {
225                                 if (compare_nodes(node_a, node_b) == 0) {
226                                         node_a = node_a->next;
227                                         a++;
228                                 }
229                                 else {
230                                         tmp = node_b;
231                                         node_b = node_b->next;
232                                         b++;
233                                         BLI_remlink(&ntree->nodes, tmp);
234                                         BLI_insertlinkbefore(&ntree->nodes, node_a, tmp);
235                                 }
236                         }
237
238                         /* setup first pointers for next batch */
239                         first_b = node_b;
240                         for (; b < k; ++b) {
241                                 /* all nodes sorted? */
242                                 if (first_b == NULL)
243                                         break;
244                                 first_b = first_b->next;
245                         }
246                         first_a = first_b;
247                 } while (first_b);
248                 
249                 k = k << 1;
250         }
251 }
252
253
254 static void do_node_internal_buttons(bContext *C, void *node_v, int event)
255 {
256         if (event == B_NODE_EXEC) {
257                 SpaceNode *snode = CTX_wm_space_node(C);
258                 if (snode && snode->id)
259                         ED_node_changed_update(snode->id, node_v);
260         }
261 }
262
263 static void node_uiblocks_init(const bContext *C, bNodeTree *ntree)
264 {
265         bNode *node;
266         char uiblockstr[32];
267         
268         /* add node uiBlocks in drawing order - prevents events going to overlapping nodes */
269         
270         for (node = ntree->nodes.first; node; node = node->next) {
271                 /* ui block */
272                 BLI_snprintf(uiblockstr, sizeof(uiblockstr), "node buttons %p", (void *)node);
273                 node->block = uiBeginBlock(C, CTX_wm_region(C), uiblockstr, UI_EMBOSS);
274                 uiBlockSetHandleFunc(node->block, do_node_internal_buttons, node);
275
276                 /* this cancels events for background nodes */
277                 uiBlockSetFlag(node->block, UI_BLOCK_CLIP_EVENTS);
278         }
279 }
280
281 /* based on settings in node, sets drawing rect info. each redraw! */
282 static void node_update_basis(const bContext *C, bNodeTree *ntree, bNode *node)
283 {
284         uiLayout *layout;
285         PointerRNA ptr;
286         bNodeSocket *nsock;
287         float locx, locy;
288         float dy;
289         int buty;
290         
291         /* get "global" coords */
292         nodeToView(node, 0.0f, 0.0f, &locx, &locy);
293         dy = locy;
294         
295         /* header */
296         dy -= NODE_DY;
297         
298         /* little bit space in top */
299         if (node->outputs.first)
300                 dy -= NODE_DYS / 2;
301
302         /* output sockets */
303         for (nsock = node->outputs.first; nsock; nsock = nsock->next) {
304                 if (!nodeSocketIsHidden(nsock)) {
305                         nsock->locx = locx + node->width;
306                         nsock->locy = dy - NODE_DYS;
307                         dy -= NODE_DY;
308                 }
309         }
310
311         node->prvr.xmin = locx + NODE_DYS;
312         node->prvr.xmax = locx + node->width - NODE_DYS;
313
314         /* preview rect? */
315         if (node->flag & NODE_PREVIEW) {
316                 if (node->preview && node->preview->rect) {
317                         float aspect = 1.0f;
318                         
319                         if (node->preview && node->preview->xsize && node->preview->ysize) 
320                                 aspect = (float)node->preview->ysize / (float)node->preview->xsize;
321                         
322                         dy -= NODE_DYS / 2;
323                         node->prvr.ymax = dy;
324                         
325                         if (aspect <= 1.0f)
326                                 node->prvr.ymin = dy - aspect * (node->width - NODE_DY);
327                         else {
328                                 /* width correction of image */
329                                 float dx = (node->width - NODE_DYS) - (node->width - NODE_DYS) / aspect;
330                                 
331                                 node->prvr.ymin = dy - (node->width - NODE_DY);
332                                 
333                                 node->prvr.xmin += 0.5f * dx;
334                                 node->prvr.xmax -= 0.5f * dx;
335                         }
336
337                         dy = node->prvr.ymin - NODE_DYS / 2;
338
339                         /* make sure that maximums are bigger or equal to minimums */
340                         if (node->prvr.xmax < node->prvr.xmin) SWAP(float, node->prvr.xmax, node->prvr.xmin);
341                         if (node->prvr.ymax < node->prvr.ymin) SWAP(float, node->prvr.ymax, node->prvr.ymin);
342                 }
343                 else {
344                         float oldh = BLI_rctf_size_y(&node->prvr);
345                         if (oldh == 0.0f)
346                                 oldh = 0.6f * node->width - NODE_DY;
347                         dy -= NODE_DYS / 2;
348                         node->prvr.ymax = dy;
349                         node->prvr.ymin = dy - oldh;
350                         dy = node->prvr.ymin - NODE_DYS / 2;
351                 }
352         }
353
354         /* buttons rect? */
355         if ((node->flag & NODE_OPTIONS) && node->typeinfo->uifunc) {
356                 dy -= NODE_DYS / 2;
357
358                 /* set this for uifunc() that don't use layout engine yet */
359                 node->butr.xmin = 0;
360                 node->butr.xmax = node->width - 2 * NODE_DYS;
361                 node->butr.ymin = 0;
362                 node->butr.ymax = 0;
363                 
364                 RNA_pointer_create(&ntree->id, &RNA_Node, node, &ptr);
365                 
366                 layout = uiBlockLayout(node->block, UI_LAYOUT_VERTICAL, UI_LAYOUT_PANEL,
367                                        locx + NODE_DYS, dy, node->butr.xmax, NODE_DY, UI_GetStyle());
368                 uiLayoutSetContextPointer(layout, "node", &ptr);
369                 
370                 node->typeinfo->uifunc(layout, (bContext *)C, &ptr);
371                 
372                 uiBlockEndAlign(node->block);
373                 uiBlockLayoutResolve(node->block, NULL, &buty);
374                 
375                 dy = buty - NODE_DYS / 2;
376         }
377
378         /* input sockets */
379         for (nsock = node->inputs.first; nsock; nsock = nsock->next) {
380                 if (!nodeSocketIsHidden(nsock)) {
381                         nsock->locx = locx;
382                         nsock->locy = dy - NODE_DYS;
383                         dy -= NODE_DY;
384                 }
385         }
386         
387         /* little bit space in end */
388         if (node->inputs.first || (node->flag & (NODE_OPTIONS | NODE_PREVIEW)) == 0)
389                 dy -= NODE_DYS / 2;
390
391         node->totr.xmin = locx;
392         node->totr.xmax = locx + node->width;
393         node->totr.ymax = locy;
394         node->totr.ymin = min_ff(dy, locy - 2 * NODE_DY);
395         
396         /* Set the block bounds to clip mouse events from underlying nodes.
397          * Add a margin for sockets on each side.
398          */
399         uiExplicitBoundsBlock(node->block,
400                               node->totr.xmin - NODE_SOCKSIZE,
401                               node->totr.ymin,
402                               node->totr.xmax + NODE_SOCKSIZE,
403                               node->totr.ymax);
404 }
405
406 /* based on settings in node, sets drawing rect info. each redraw! */
407 static void node_update_hidden(bNode *node)
408 {
409         bNodeSocket *nsock;
410         float locx, locy;
411         float rad, drad, hiddenrad = HIDDEN_RAD;
412         int totin = 0, totout = 0, tot;
413         
414         /* get "global" coords */
415         nodeToView(node, 0.0f, 0.0f, &locx, &locy);
416
417         /* calculate minimal radius */
418         for (nsock = node->inputs.first; nsock; nsock = nsock->next)
419                 if (!nodeSocketIsHidden(nsock))
420                         totin++;
421         for (nsock = node->outputs.first; nsock; nsock = nsock->next)
422                 if (!nodeSocketIsHidden(nsock))
423                         totout++;
424         
425         tot = MAX2(totin, totout);
426         if (tot > 4) {
427                 hiddenrad += 5.0f * (float)(tot - 4);
428         }
429         
430         node->totr.xmin = locx;
431         node->totr.xmax = locx + 3 * hiddenrad + node->miniwidth;
432         node->totr.ymax = locy + (hiddenrad - 0.5f * NODE_DY);
433         node->totr.ymin = node->totr.ymax - 2 * hiddenrad;
434         
435         /* output sockets */
436         rad = drad = (float)M_PI / (1.0f + (float)totout);
437         
438         for (nsock = node->outputs.first; nsock; nsock = nsock->next) {
439                 if (!nodeSocketIsHidden(nsock)) {
440                         nsock->locx = node->totr.xmax - hiddenrad + (float)sin(rad) * hiddenrad;
441                         nsock->locy = node->totr.ymin + hiddenrad + (float)cos(rad) * hiddenrad;
442                         rad += drad;
443                 }
444         }
445         
446         /* input sockets */
447         rad = drad = -(float)M_PI / (1.0f + (float)totin);
448         
449         for (nsock = node->inputs.first; nsock; nsock = nsock->next) {
450                 if (!nodeSocketIsHidden(nsock)) {
451                         nsock->locx = node->totr.xmin + hiddenrad + (float)sin(rad) * hiddenrad;
452                         nsock->locy = node->totr.ymin + hiddenrad + (float)cos(rad) * hiddenrad;
453                         rad += drad;
454                 }
455         }
456
457         /* Set the block bounds to clip mouse events from underlying nodes.
458          * Add a margin for sockets on each side.
459          */
460         uiExplicitBoundsBlock(node->block,
461                               node->totr.xmin - NODE_SOCKSIZE,
462                               node->totr.ymin,
463                               node->totr.xmax + NODE_SOCKSIZE,
464                               node->totr.ymax);
465 }
466
467 void node_update_default(const bContext *C, bNodeTree *ntree, bNode *node)
468 {
469         if (node->flag & NODE_HIDDEN)
470                 node_update_hidden(node);
471         else
472                 node_update_basis(C, ntree, node);
473 }
474
475 int node_select_area_default(bNode *node, int x, int y)
476 {
477         return BLI_rctf_isect_pt(&node->totr, x, y);
478 }
479
480 int node_tweak_area_default(bNode *node, int x, int y)
481 {
482         return BLI_rctf_isect_pt(&node->totr, x, y);
483 }
484
485 int node_get_colorid(bNode *node)
486 {
487         if (node->typeinfo->nclass == NODE_CLASS_INPUT)
488                 return TH_NODE_IN_OUT;
489         if (node->typeinfo->nclass == NODE_CLASS_OUTPUT) {
490                 if (node->flag & NODE_DO_OUTPUT)
491                         return TH_NODE_IN_OUT;
492                 else
493                         return TH_NODE;
494         }
495         if (node->typeinfo->nclass == NODE_CLASS_CONVERTOR)
496                 return TH_NODE_CONVERTOR;
497         if (ELEM3(node->typeinfo->nclass, NODE_CLASS_OP_COLOR, NODE_CLASS_OP_VECTOR, NODE_CLASS_OP_FILTER))
498                 return TH_NODE_OPERATOR;
499         if (node->typeinfo->nclass == NODE_CLASS_GROUP)
500                 return TH_NODE_GROUP;
501         return TH_NODE;
502 }
503
504 /* note: in cmp_util.c is similar code, for node_compo_pass_on()
505  *       the same goes for shader and texture nodes. */
506 /* note: in node_edit.c is similar code, for untangle node */
507 static void node_draw_mute_line(View2D *v2d, SpaceNode *snode, bNode *node)
508 {
509         bNodeLink *link;
510
511         glEnable(GL_BLEND);
512         glEnable(GL_LINE_SMOOTH);
513
514         for (link = node->internal_links.first; link; link = link->next)
515                 node_draw_link_bezier(v2d, snode, link, TH_REDALERT, 0, TH_WIRE, 0, TH_WIRE);
516
517         glDisable(GL_BLEND);
518         glDisable(GL_LINE_SMOOTH);
519 }
520
521 /* this might have some more generic use */
522 static void node_circle_draw(float x, float y, float size, char *col, int highlight)
523 {
524         /* 16 values of sin function */
525         static float si[16] = {
526                 0.00000000f, 0.39435585f, 0.72479278f, 0.93775213f,
527                 0.99871650f, 0.89780453f, 0.65137248f, 0.29936312f,
528                 -0.10116832f, -0.48530196f, -0.79077573f, -0.96807711f,
529                 -0.98846832f, -0.84864425f, -0.57126821f, -0.20129852f
530         };
531         /* 16 values of cos function */
532         static float co[16] = {
533                 1.00000000f, 0.91895781f, 0.68896691f, 0.34730525f,
534                 -0.05064916f, -0.44039415f, -0.75875812f, -0.95413925f,
535                 -0.99486932f, -0.87434661f, -0.61210598f, -0.25065253f,
536                 0.15142777f, 0.52896401f, 0.82076344f, 0.97952994f,
537         };
538         int a;
539         
540         glColor3ub(col[0], col[1], col[2]);
541         
542         glBegin(GL_POLYGON);
543         for (a = 0; a < 16; a++)
544                 glVertex2f(x + size * si[a], y + size * co[a]);
545         glEnd();
546         
547         if (highlight) {
548                 UI_ThemeColor(TH_TEXT_HI);
549                 glLineWidth(1.5f);
550         }
551         else {
552                 glColor4ub(0, 0, 0, 150);
553         }
554         glEnable(GL_BLEND);
555         glEnable(GL_LINE_SMOOTH);
556         glBegin(GL_LINE_LOOP);
557         for (a = 0; a < 16; a++)
558                 glVertex2f(x + size * si[a], y + size * co[a]);
559         glEnd();
560         glDisable(GL_LINE_SMOOTH);
561         glDisable(GL_BLEND);
562         glLineWidth(1.0f);
563 }
564
565 void node_socket_circle_draw(bNodeTree *UNUSED(ntree), bNodeSocket *sock, float size, int highlight)
566 {
567         bNodeSocketType *stype = ntreeGetSocketType(sock->type);
568         node_circle_draw(sock->locx, sock->locy, size, stype->ui_color, highlight);
569 }
570
571 /* **************  Socket callbacks *********** */
572
573 /* not a callback */
574 static void node_draw_preview(bNodePreview *preview, rctf *prv)
575 {
576         float xscale = BLI_rctf_size_x(prv) / ((float)preview->xsize);
577         float yscale = BLI_rctf_size_y(prv) / ((float)preview->ysize);
578         float tile   = BLI_rctf_size_x(prv) / 10.0f;
579         float x, y;
580         
581         /* draw checkerboard backdrop to show alpha */
582         glColor3ub(120, 120, 120);
583         glRectf(prv->xmin, prv->ymin, prv->xmax, prv->ymax);
584         glColor3ub(160, 160, 160);
585         
586         for (y = prv->ymin; y < prv->ymax; y += tile * 2) {
587                 for (x = prv->xmin; x < prv->xmax; x += tile * 2) {
588                         float tilex = tile, tiley = tile;
589
590                         if (x + tile > prv->xmax)
591                                 tilex = prv->xmax - x;
592                         if (y + tile > prv->ymax)
593                                 tiley = prv->ymax - y;
594
595                         glRectf(x, y, x + tilex, y + tiley);
596                 }
597         }
598         for (y = prv->ymin + tile; y < prv->ymax; y += tile * 2) {
599                 for (x = prv->xmin + tile; x < prv->xmax; x += tile * 2) {
600                         float tilex = tile, tiley = tile;
601
602                         if (x + tile > prv->xmax)
603                                 tilex = prv->xmax - x;
604                         if (y + tile > prv->ymax)
605                                 tiley = prv->ymax - y;
606
607                         glRectf(x, y, x + tilex, y + tiley);
608                 }
609         }
610         
611         glPixelZoom(xscale, yscale);
612
613         glEnable(GL_BLEND);
614         glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);  /* premul graphics */
615         
616         glColor4f(1.0, 1.0, 1.0, 1.0);
617         glaDrawPixelsTex(prv->xmin, prv->ymin, preview->xsize, preview->ysize, GL_UNSIGNED_BYTE, preview->rect);
618         
619         glDisable(GL_BLEND);
620         glPixelZoom(1.0f, 1.0f);
621
622         UI_ThemeColorShadeAlpha(TH_BACK, -15, +100);
623         fdrawbox(prv->xmin, prv->ymin, prv->xmax, prv->ymax);
624         
625 }
626
627 /* common handle function for operator buttons that need to select the node first */
628 static void node_toggle_button_cb(struct bContext *C, void *node_argv, void *op_argv)
629 {
630         bNode *node = (bNode *)node_argv;
631         const char *opname = (const char *)op_argv;
632         
633         /* select & activate only the button's node */
634         node_select_single(C, node);
635         
636         WM_operator_name_call(C, opname, WM_OP_INVOKE_DEFAULT, NULL);
637 }
638
639 void node_draw_shadow(SpaceNode *snode, bNode *node, float radius, float alpha)
640 {
641         rctf *rct = &node->totr;
642         
643         uiSetRoundBox(UI_CNR_ALL);
644         if (node->parent == NULL)
645                 ui_dropshadow(rct, radius, snode->aspect, alpha, node->flag & SELECT);
646         else {
647                 const float margin = 3.0f;
648                 
649                 glColor4f(0.0f, 0.0f, 0.0f, 0.33f);
650                 glEnable(GL_BLEND);
651                 uiRoundBox(rct->xmin - margin, rct->ymin - margin,
652                            rct->xmax + margin, rct->ymax + margin, radius + margin);
653                 glDisable(GL_BLEND);
654         }
655 }
656
657 static void node_draw_basis(const bContext *C, ARegion *ar, SpaceNode *snode, bNodeTree *ntree, bNode *node)
658 {
659         bNodeSocket *sock;
660         rctf *rct = &node->totr;
661         float iconofs;
662         /* float socket_size = NODE_SOCKSIZE*U.dpi/72; */ /* UNUSED */
663         float iconbutw = 0.8f * UI_UNIT_X;
664         int color_id = node_get_colorid(node);
665         char showname[128]; /* 128 used below */
666         View2D *v2d = &ar->v2d;
667         
668         /* hurmf... another candidate for callback, have to see how this works first */
669         if (node->id && node->block && snode->treetype == NTREE_SHADER)
670                 nodeShaderSynchronizeID(node, 0);
671         
672         /* skip if out of view */
673         if (BLI_rctf_isect(&node->totr, &ar->v2d.cur, NULL) == FALSE) {
674                 uiEndBlock(C, node->block);
675                 node->block = NULL;
676                 return;
677         }
678         
679         /* shadow */
680         node_draw_shadow(snode, node, BASIS_RAD, 1.0f);
681         
682         /* header */
683         if (color_id == TH_NODE)
684                 UI_ThemeColorShade(color_id, -20);
685         else
686                 UI_ThemeColor(color_id);
687         
688         if (node->flag & NODE_MUTED)
689                 UI_ThemeColorBlend(color_id, TH_REDALERT, 0.5f);
690
691 #ifdef WITH_COMPOSITOR
692         if (ntree->type == NTREE_COMPOSIT && (snode->flag & SNODE_SHOW_HIGHLIGHT)) {
693                 if (COM_isHighlightedbNode(node)) {
694                         UI_ThemeColorBlend(color_id, TH_ACTIVE, 0.5f);
695                 }
696         }
697 #endif
698
699         uiSetRoundBox(UI_CNR_TOP_LEFT | UI_CNR_TOP_RIGHT);
700         uiRoundBox(rct->xmin, rct->ymax - NODE_DY, rct->xmax, rct->ymax, BASIS_RAD);
701         
702         /* show/hide icons */
703         iconofs = rct->xmax - 7.0f;
704         
705         /* preview */
706         if (node->typeinfo->flag & NODE_PREVIEW) {
707                 uiBut *but;
708                 iconofs -= iconbutw;
709                 uiBlockSetEmboss(node->block, UI_EMBOSSN);
710                 but = uiDefIconBut(node->block, TOGBUT, B_REDR, ICON_MATERIAL,
711                                    iconofs, rct->ymax - NODE_DY, iconbutw, UI_UNIT_Y, NULL, 0, 0, 0, 0, "");
712                 uiButSetFunc(but, node_toggle_button_cb, node, (void *)"NODE_OT_preview_toggle");
713                 /* XXX this does not work when node is activated and the operator called right afterwards,
714                  * since active ID is not updated yet (needs to process the notifier).
715                  * This can only work as visual indicator!
716                  */
717 //              if (!(node->flag & (NODE_ACTIVE_ID|NODE_DO_OUTPUT)))
718 //                      uiButSetFlag(but, UI_BUT_DISABLED);
719                 uiBlockSetEmboss(node->block, UI_EMBOSS);
720         }
721         /* group edit */
722         if (node->type == NODE_GROUP) {
723                 uiBut *but;
724                 iconofs -= iconbutw;
725                 uiBlockSetEmboss(node->block, UI_EMBOSSN);
726                 but = uiDefIconBut(node->block, TOGBUT, B_REDR, ICON_NODETREE,
727                                    iconofs, rct->ymax - NODE_DY, iconbutw, UI_UNIT_Y, NULL, 0, 0, 0, 0, "");
728                 uiButSetFunc(but, node_toggle_button_cb, node, (void *)"NODE_OT_group_edit");
729                 uiBlockSetEmboss(node->block, UI_EMBOSS);
730         }
731         
732         /* title */
733         if (node->flag & SELECT) 
734                 UI_ThemeColor(TH_SELECT);
735         else
736                 UI_ThemeColorBlendShade(TH_TEXT, color_id, 0.4f, 10);
737         
738         /* open/close entirely? */
739         {
740                 uiBut *but;
741                 int but_size = UI_UNIT_X * 0.6f;
742                 /* XXX button uses a custom triangle draw below, so make it invisible without icon */
743                 uiBlockSetEmboss(node->block, UI_EMBOSSN);
744                 but = uiDefBut(node->block, TOGBUT, B_REDR, "",
745                                rct->xmin + 10.0f - but_size / 2, rct->ymax - NODE_DY / 2.0f - but_size / 2,
746                                but_size, but_size, NULL, 0, 0, 0, 0, "");
747                 uiButSetFunc(but, node_toggle_button_cb, node, (void *)"NODE_OT_hide_toggle");
748                 uiBlockSetEmboss(node->block, UI_EMBOSS);
749                 
750                 /* custom draw function for this button */
751                 UI_DrawTriIcon(rct->xmin + 10.0f, rct->ymax - NODE_DY / 2.0f, 'v');
752         }
753         
754         /* this isn't doing anything for the label, so commenting out */
755 #if 0
756         if (node->flag & SELECT) 
757                 UI_ThemeColor(TH_TEXT_HI);
758         else
759                 UI_ThemeColor(TH_TEXT);
760 #endif
761         
762         BLI_strncpy(showname, nodeLabel(node), sizeof(showname));
763         
764         //if (node->flag & NODE_MUTED)
765         //      BLI_snprintf(showname, sizeof(showname), "[%s]", showname); /* XXX - don't print into self! */
766         
767         uiDefBut(node->block, LABEL, 0, showname,
768                  (int)(rct->xmin + (NODE_MARGIN_X / snode->aspect_sqrt)), (int)(rct->ymax - NODE_DY),
769                  (short)(iconofs - rct->xmin - 18.0f), (short)NODE_DY,
770                  NULL, 0, 0, 0, 0, "");
771
772         /* body */
773         if (node->flag & NODE_CUSTOM_COLOR)
774                 glColor3fv(node->color);
775         else
776                 UI_ThemeColor4(TH_NODE);
777         glEnable(GL_BLEND);
778         uiSetRoundBox(UI_CNR_BOTTOM_LEFT | UI_CNR_BOTTOM_RIGHT);
779         uiRoundBox(rct->xmin, rct->ymin, rct->xmax, rct->ymax - NODE_DY, BASIS_RAD);
780         glDisable(GL_BLEND);
781
782         /* outline active and selected emphasis */
783         if (node->flag & SELECT) {
784                 glEnable(GL_BLEND);
785                 glEnable(GL_LINE_SMOOTH);
786                 
787                 if (node->flag & NODE_ACTIVE)
788                         UI_ThemeColorShadeAlpha(TH_ACTIVE, 0, -40);
789                 else
790                         UI_ThemeColorShadeAlpha(TH_SELECT, 0, -40);
791                 uiSetRoundBox(UI_CNR_ALL);
792                 uiDrawBox(GL_LINE_LOOP, rct->xmin, rct->ymin, rct->xmax, rct->ymax, BASIS_RAD);
793                 
794                 glDisable(GL_LINE_SMOOTH);
795                 glDisable(GL_BLEND);
796         }
797         
798         /* disable lines */
799         if (node->flag & NODE_MUTED)
800                 node_draw_mute_line(v2d, snode, node);
801
802         
803         /* socket inputs, buttons */
804         for (sock = node->inputs.first; sock; sock = sock->next) {
805                 if (nodeSocketIsHidden(sock))
806                         continue;
807                 
808                 node_socket_circle_draw(ntree, sock, NODE_SOCKSIZE, sock->flag & SELECT);
809                 
810                 node->typeinfo->drawinputfunc(C, node->block, ntree, node, sock, IFACE_(sock->name),
811                                               sock->locx + (NODE_DYS / snode->aspect_sqrt), sock->locy - NODE_DYS,
812                                               node->width - NODE_DY);
813         }
814         
815         /* socket outputs */
816         for (sock = node->outputs.first; sock; sock = sock->next) {
817                 if (nodeSocketIsHidden(sock))
818                         continue;
819                 
820                 node_socket_circle_draw(ntree, sock, NODE_SOCKSIZE, sock->flag & SELECT);
821                 
822                 node->typeinfo->drawoutputfunc(C, node->block, ntree, node, sock, IFACE_(sock->name),
823                                                sock->locx - node->width + (NODE_DYS / snode->aspect_sqrt), sock->locy - NODE_DYS,
824                                                node->width - NODE_DY);
825         }
826         
827         /* preview */
828         if (node->flag & NODE_PREVIEW) {
829                 if (node->preview && node->preview->rect && !BLI_rctf_is_empty(&node->prvr))
830                         node_draw_preview(node->preview, &node->prvr);
831         }
832         
833         UI_ThemeClearColor(color_id);
834                 
835         uiEndBlock(C, node->block);
836         uiDrawBlock(C, node->block);
837         node->block = NULL;
838 }
839
840 static void node_draw_hidden(const bContext *C, ARegion *ar, SpaceNode *snode, bNodeTree *ntree, bNode *node)
841 {
842         bNodeSocket *sock;
843         rctf *rct = &node->totr;
844         float dx, centy = BLI_rctf_cent_y(rct);
845         float hiddenrad = BLI_rctf_size_y(rct) / 2.0f;
846         float socket_size = NODE_SOCKSIZE * UI_DPI_ICON_FAC;
847         int color_id = node_get_colorid(node);
848         char showname[128]; /* 128 is used below */
849         
850         /* shadow */
851         node_draw_shadow(snode, node, hiddenrad, 1.0f);
852
853         /* body */
854         UI_ThemeColor(color_id);
855         if (node->flag & NODE_MUTED)
856                 UI_ThemeColorBlend(color_id, TH_REDALERT, 0.5f);
857
858 #ifdef WITH_COMPOSITOR
859         if (ntree->type == NTREE_COMPOSIT && (snode->flag & SNODE_SHOW_HIGHLIGHT)) {
860                 if (COM_isHighlightedbNode(node)) {
861                         UI_ThemeColorBlend(color_id, TH_ACTIVE, 0.5f);
862                 }
863         }
864 #else
865         (void)ntree;
866 #endif
867         
868         uiRoundBox(rct->xmin, rct->ymin, rct->xmax, rct->ymax, hiddenrad);
869         
870         /* outline active and selected emphasis */
871         if (node->flag & SELECT) {
872                 glEnable(GL_BLEND);
873                 glEnable(GL_LINE_SMOOTH);
874                 
875                 if (node->flag & NODE_ACTIVE)
876                         UI_ThemeColorShadeAlpha(TH_ACTIVE, 0, -40);
877                 else
878                         UI_ThemeColorShadeAlpha(TH_SELECT, 0, -40);
879                 uiDrawBox(GL_LINE_LOOP, rct->xmin, rct->ymin, rct->xmax, rct->ymax, hiddenrad);
880                 
881                 glDisable(GL_LINE_SMOOTH);
882                 glDisable(GL_BLEND);
883         }
884         
885         /* title */
886         if (node->flag & SELECT) 
887                 UI_ThemeColor(TH_SELECT);
888         else
889                 UI_ThemeColorBlendShade(TH_TEXT, color_id, 0.4f, 10);
890         
891         /* open entirely icon */
892         {
893                 uiBut *but;
894                 int but_size = UI_UNIT_X * 0.6f;
895                 /* XXX button uses a custom triangle draw below, so make it invisible without icon */
896                 uiBlockSetEmboss(node->block, UI_EMBOSSN);
897                 but = uiDefBut(node->block, TOGBUT, B_REDR, "",
898                                rct->xmin + 10.0f - but_size / 2, centy - but_size / 2,
899                                but_size, but_size, NULL, 0, 0, 0, 0, "");
900                 uiButSetFunc(but, node_toggle_button_cb, node, (void *)"NODE_OT_hide_toggle");
901                 uiBlockSetEmboss(node->block, UI_EMBOSS);
902                 
903                 /* custom draw function for this button */
904                 UI_DrawTriIcon(rct->xmin + 10.0f, centy, 'h');
905         }
906         
907         /* disable lines */
908         if (node->flag & NODE_MUTED)
909                 node_draw_mute_line(&ar->v2d, snode, node);
910         
911         if (node->flag & SELECT) 
912                 UI_ThemeColor(TH_SELECT);
913         else
914                 UI_ThemeColor(TH_TEXT);
915         
916         if (node->miniwidth > 0.0f) {
917                 BLI_strncpy(showname, nodeLabel(node), sizeof(showname));
918                 
919                 //if (node->flag & NODE_MUTED)
920                 //      BLI_snprintf(showname, sizeof(showname), "[%s]", showname); /* XXX - don't print into self! */
921
922                 uiDefBut(node->block, LABEL, 0, showname,
923                          (int)(rct->xmin + (NODE_MARGIN_X / snode->aspect_sqrt)), (int)(centy - 10),
924                          (short)(BLI_rctf_size_x(rct) - 18.0f - 12.0f), (short)NODE_DY,
925                          NULL, 0, 0, 0, 0, "");
926         }
927
928         /* scale widget thing */
929         UI_ThemeColorShade(color_id, -10);
930         dx = 10.0f;
931         fdrawline(rct->xmax - dx, centy - 4.0f, rct->xmax - dx, centy + 4.0f);
932         fdrawline(rct->xmax - dx - 3.0f * snode->aspect, centy - 4.0f, rct->xmax - dx - 3.0f * snode->aspect, centy + 4.0f);
933         
934         UI_ThemeColorShade(color_id, +30);
935         dx -= snode->aspect;
936         fdrawline(rct->xmax - dx, centy - 4.0f, rct->xmax - dx, centy + 4.0f);
937         fdrawline(rct->xmax - dx - 3.0f * snode->aspect, centy - 4.0f, rct->xmax - dx - 3.0f * snode->aspect, centy + 4.0f);
938
939         /* sockets */
940         for (sock = node->inputs.first; sock; sock = sock->next) {
941                 if (!nodeSocketIsHidden(sock))
942                         node_socket_circle_draw(snode->nodetree, sock, socket_size, sock->flag & SELECT);
943         }
944         
945         for (sock = node->outputs.first; sock; sock = sock->next) {
946                 if (!nodeSocketIsHidden(sock))
947                         node_socket_circle_draw(snode->nodetree, sock, socket_size, sock->flag & SELECT);
948         }
949         
950         uiEndBlock(C, node->block);
951         uiDrawBlock(C, node->block);
952         node->block = NULL;
953 }
954
955 int node_get_resize_cursor(int directions)
956 {
957         if (directions == 0)
958                 return CURSOR_STD;
959         else if ((directions & ~(NODE_RESIZE_TOP | NODE_RESIZE_BOTTOM)) == 0)
960                 return CURSOR_Y_MOVE;
961         else if ((directions & ~(NODE_RESIZE_RIGHT | NODE_RESIZE_LEFT)) == 0)
962                 return CURSOR_X_MOVE;
963         else
964                 return CURSOR_EDIT;
965 }
966
967 void node_set_cursor(wmWindow *win, SpaceNode *snode)
968 {
969         bNodeTree *ntree = snode->edittree;
970         bNode *node;
971         bNodeSocket *sock;
972         int cursor = CURSOR_STD;
973         
974         if (ntree) {
975                 if (node_find_indicated_socket(snode, &node, &sock, SOCK_IN | SOCK_OUT)) {
976                         /* pass */
977                 }
978                 else {
979                         /* check nodes front to back */
980                         for (node = ntree->nodes.last; node; node = node->prev) {
981                                 if (BLI_rctf_isect_pt(&node->totr, snode->cursor[0], snode->cursor[1]))
982                                         break;  /* first hit on node stops */
983                         }
984                         if (node) {
985                                 int dir = node->typeinfo->resize_area_func(node, snode->cursor[0], snode->cursor[1]);
986                                 cursor = node_get_resize_cursor(dir);
987                         }
988                 }
989         }
990         
991         WM_cursor_set(win, cursor);
992 }
993
994 void node_draw_default(const bContext *C, ARegion *ar, SpaceNode *snode, bNodeTree *ntree, bNode *node)
995 {
996         if (node->flag & NODE_HIDDEN)
997                 node_draw_hidden(C, ar, snode, ntree, node);
998         else
999                 node_draw_basis(C, ar, snode, ntree, node);
1000 }
1001
1002 static void node_update(const bContext *C, bNodeTree *ntree, bNode *node)
1003 {
1004         if (node->typeinfo->drawupdatefunc)
1005                 node->typeinfo->drawupdatefunc(C, ntree, node);
1006 }
1007
1008 void node_update_nodetree(const bContext *C, bNodeTree *ntree, float offsetx, float offsety)
1009 {
1010         bNode *node;
1011         
1012         /* update nodes front to back, so children sizes get updated before parents */
1013         for (node = ntree->nodes.last; node; node = node->prev) {
1014                 /* XXX little hack */
1015                 node->locx += offsetx;
1016                 node->locy += offsety;
1017                 
1018                 node_update(C, ntree, node);
1019                 
1020                 node->locx -= offsetx;
1021                 node->locy -= offsety;
1022         }
1023 }
1024
1025 static void node_draw(const bContext *C, ARegion *ar, SpaceNode *snode, bNodeTree *ntree, bNode *node)
1026 {
1027         if (node->typeinfo->drawfunc)
1028                 node->typeinfo->drawfunc(C, ar, snode, ntree, node);
1029 }
1030
1031 #define USE_DRAW_TOT_UPDATE
1032
1033 void node_draw_nodetree(const bContext *C, ARegion *ar, SpaceNode *snode, bNodeTree *ntree)
1034 {
1035         bNode *node;
1036         bNodeLink *link;
1037         int a;
1038         
1039         if (ntree == NULL) return;      /* groups... */
1040
1041 #ifdef USE_DRAW_TOT_UPDATE
1042         if (ntree->nodes.first) {
1043                 BLI_rctf_init_minmax(&ar->v2d.tot);
1044         }
1045 #endif
1046
1047         /* draw background nodes, last nodes in front */
1048         for (a = 0, node = ntree->nodes.first; node; node = node->next, a++) {
1049
1050 #ifdef USE_DRAW_TOT_UPDATE
1051                 /* unrelated to background nodes, update the v2d->tot,
1052                  * can be anywhere before we draw the scroll bars */
1053                 BLI_rctf_union(&ar->v2d.tot, &node->totr);
1054 #endif
1055
1056                 if (!(node->flag & NODE_BACKGROUND))
1057                         continue;
1058                 node->nr = a;        /* index of node in list, used for exec event code */
1059                 node_draw(C, ar, snode, ntree, node);
1060         }
1061         
1062         /* node lines */
1063         glEnable(GL_BLEND);
1064         glEnable(GL_LINE_SMOOTH);
1065         for (link = ntree->links.first; link; link = link->next)
1066                 node_draw_link(&ar->v2d, snode, link);
1067         glDisable(GL_LINE_SMOOTH);
1068         glDisable(GL_BLEND);
1069         
1070         /* draw foreground nodes, last nodes in front */
1071         for (a = 0, node = ntree->nodes.first; node; node = node->next, a++) {
1072                 if (node->flag & NODE_BACKGROUND)
1073                         continue;
1074                 node->nr = a;        /* index of node in list, used for exec event code */
1075                 node_draw(C, ar, snode, ntree, node);
1076         }
1077 }
1078
1079 void drawnodespace(const bContext *C, ARegion *ar, View2D *v2d)
1080 {
1081         View2DScrollers *scrollers;
1082         SpaceNode *snode = CTX_wm_space_node(C);
1083         bNodeLinkDrag *nldrag;
1084         LinkData *linkdata;
1085         
1086         UI_ThemeClearColor(TH_BACK);
1087         glClear(GL_COLOR_BUFFER_BIT);
1088
1089         UI_view2d_view_ortho(v2d);
1090         
1091         //uiFreeBlocksWin(&sa->uiblocks, sa->win);
1092
1093         ED_region_draw_cb_draw(C, ar, REGION_DRAW_PRE_VIEW);
1094
1095         /* only set once */
1096         glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
1097         glEnable(GL_MAP1_VERTEX_3);
1098
1099         /* aspect+font, set each time */
1100         snode->aspect = BLI_rctf_size_x(&v2d->cur) / (float)ar->winx;
1101         snode->aspect_sqrt = sqrtf(snode->aspect);
1102         // XXX snode->curfont = uiSetCurFont_ext(snode->aspect);
1103
1104         /* grid */
1105         UI_view2d_multi_grid_draw(v2d, 25.0f, 5, 2);
1106
1107         /* backdrop */
1108         draw_nodespace_back_pix(C, ar, snode);
1109         
1110         /* nodes */
1111         snode_set_context(snode, CTX_data_scene(C));
1112         
1113         if (snode->nodetree) {
1114                 bNode *node;
1115                 /* void** highlights = 0; */ /* UNUSED */
1116                 
1117                 node_uiblocks_init(C, snode->nodetree);
1118                 
1119                 /* uiBlocks must be initialized in drawing order for correct event clipping.
1120                  * Node group internal blocks added after the main group block.
1121                  */
1122                 for (node = snode->nodetree->nodes.first; node; node = node->next) {
1123                         if (node->flag & NODE_GROUP_EDIT)
1124                                 node_uiblocks_init(C, (bNodeTree *)node->id);
1125                 }
1126                 
1127                 node_update_nodetree(C, snode->nodetree, 0.0f, 0.0f);
1128
1129 #ifdef WITH_COMPOSITOR
1130                 if (snode->nodetree->type == NTREE_COMPOSIT) {
1131                         COM_startReadHighlights();
1132                 }
1133 #endif
1134
1135                 node_draw_nodetree(C, ar, snode, snode->nodetree);
1136                 
1137                 #if 0
1138                 /* active group */
1139                 for (node = snode->nodetree->nodes.first; node; node = node->next) {
1140                         if (node->flag & NODE_GROUP_EDIT)
1141                                 node_draw_group(C, ar, snode, snode->nodetree, node);
1142                 }
1143                 #endif
1144         }
1145         
1146         /* temporary links */
1147         glEnable(GL_BLEND);
1148         glEnable(GL_LINE_SMOOTH);
1149         for (nldrag = snode->linkdrag.first; nldrag; nldrag = nldrag->next) {
1150                 for (linkdata = nldrag->links.first; linkdata; linkdata = linkdata->next) {
1151                         node_draw_link(&ar->v2d, snode, (bNodeLink *)linkdata->data);
1152                 }
1153         }
1154         glDisable(GL_LINE_SMOOTH);
1155         glDisable(GL_BLEND);
1156         
1157         ED_region_draw_cb_draw(C, ar, REGION_DRAW_POST_VIEW);
1158         
1159         /* draw grease-pencil ('canvas' strokes) */
1160         if (snode->nodetree)
1161                 draw_gpencil_view2d(C, 1);
1162         
1163         /* reset view matrix */
1164         UI_view2d_view_restore(C);
1165         
1166         /* draw grease-pencil (screen strokes, and also paintbuffer) */
1167         if (snode->nodetree)
1168                 draw_gpencil_view2d(C, 0);
1169         
1170         /* scrollers */
1171         scrollers = UI_view2d_scrollers_calc(C, v2d, 10, V2D_GRID_CLAMP, V2D_ARG_DUMMY, V2D_ARG_DUMMY);
1172         UI_view2d_scrollers_draw(C, v2d, scrollers);
1173         UI_view2d_scrollers_free(scrollers);
1174 }