Fix passing the same argument twice to BLI_strncpy
[blender.git] / source / blender / editors / space_node / node_templates.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  * Contributor(s): Blender Foundation 2009.
19  *
20  * ***** END GPL LICENSE BLOCK *****
21  */
22
23 /** \file blender/editors/space_node/node_templates.c
24  *  \ingroup edinterface
25  */
26
27 #include <stdlib.h>
28 #include <string.h>
29
30 #include "MEM_guardedalloc.h"
31
32 #include "DNA_node_types.h"
33 #include "DNA_screen_types.h"
34
35 #include "BLI_array.h"
36 #include "BLI_listbase.h"
37 #include "BLI_string.h"
38
39 #include "BLT_translation.h"
40
41 #include "BKE_context.h"
42 #include "BKE_global.h"
43 #include "BKE_library.h"
44 #include "BKE_main.h"
45 #include "BKE_scene.h"
46
47 #include "RNA_access.h"
48
49 #include "NOD_socket.h"
50
51 #include "UI_interface.h"
52 #include "../interface/interface_intern.h"  /* XXX bad level */
53
54 #include "ED_node.h"  /* own include */
55
56 #include "ED_util.h"
57
58
59 /************************* Node Socket Manipulation **************************/
60
61 /* describes an instance of a node type and a specific socket to link */
62 typedef struct NodeLinkItem {
63         int socket_index;                       /* index for linking */
64         int socket_type;                        /* socket type for compatibility check */
65         const char *socket_name;        /* ui label of the socket */
66         const char *node_name;          /* ui label of the node */
67         
68         /* extra settings */
69         bNodeTree *ngroup;              /* group node tree */
70 } NodeLinkItem;
71
72 /* Compare an existing node to a link item to see if it can be reused.
73  * item must be for the same node type!
74  * XXX should become a node type callback
75  */
76 static bool node_link_item_compare(bNode *node, NodeLinkItem *item)
77 {
78         if (node->type == NODE_GROUP) {
79                 return (node->id == (ID *)item->ngroup);
80         }
81         else
82                 return true;
83 }
84
85 static void node_link_item_apply(bNode *node, NodeLinkItem *item)
86 {
87         if (node->type == NODE_GROUP) {
88                 node->id = (ID *)item->ngroup;
89                 ntreeUpdateTree(G.main, item->ngroup);
90         }
91         else {
92                 /* nothing to do for now */
93         }
94         
95         if (node->id)
96                 id_us_plus(node->id);
97 }
98
99 static void node_tag_recursive(bNode *node)
100 {
101         bNodeSocket *input;
102
103         if (!node || (node->flag & NODE_TEST))
104                 return;  /* in case of cycles */
105
106         node->flag |= NODE_TEST;
107
108         for (input = node->inputs.first; input; input = input->next)
109                 if (input->link)
110                         node_tag_recursive(input->link->fromnode);
111 }
112
113 static void node_clear_recursive(bNode *node)
114 {
115         bNodeSocket *input;
116
117         if (!node || !(node->flag & NODE_TEST))
118                 return;  /* in case of cycles */
119
120         node->flag &= ~NODE_TEST;
121
122         for (input = node->inputs.first; input; input = input->next)
123                 if (input->link)
124                         node_clear_recursive(input->link->fromnode);
125 }
126
127 static void node_remove_linked(bNodeTree *ntree, bNode *rem_node)
128 {
129         bNode *node, *next;
130         bNodeSocket *sock;
131
132         if (!rem_node)
133                 return;
134
135         /* tag linked nodes to be removed */
136         for (node = ntree->nodes.first; node; node = node->next)
137                 node->flag &= ~NODE_TEST;
138
139         node_tag_recursive(rem_node);
140
141         /* clear tags on nodes that are still used by other nodes */
142         for (node = ntree->nodes.first; node; node = node->next)
143                 if (!(node->flag & NODE_TEST))
144                         for (sock = node->inputs.first; sock; sock = sock->next)
145                                 if (sock->link && sock->link->fromnode != rem_node)
146                                         node_clear_recursive(sock->link->fromnode);
147
148         /* remove nodes */
149         for (node = ntree->nodes.first; node; node = next) {
150                 next = node->next;
151
152                 if (node->flag & NODE_TEST) {
153                         if (node->id)
154                                 id_us_min(node->id);
155                         nodeFreeNode(ntree, node);
156                 }
157         }
158 }
159
160 /* disconnect socket from the node it is connected to */
161 static void node_socket_disconnect(Main *bmain, bNodeTree *ntree, bNode *node_to, bNodeSocket *sock_to)
162 {
163         if (!sock_to->link)
164                 return;
165
166         nodeRemLink(ntree, sock_to->link);
167         sock_to->flag |= SOCK_COLLAPSED;
168
169         nodeUpdate(ntree, node_to);
170         ntreeUpdateTree(bmain, ntree);
171
172         ED_node_tag_update_nodetree(bmain, ntree, node_to);
173 }
174
175 /* remove all nodes connected to this socket, if they aren't connected to other nodes */
176 static void node_socket_remove(Main *bmain, bNodeTree *ntree, bNode *node_to, bNodeSocket *sock_to)
177 {
178         if (!sock_to->link)
179                 return;
180
181         node_remove_linked(ntree, sock_to->link->fromnode);
182         sock_to->flag |= SOCK_COLLAPSED;
183
184         nodeUpdate(ntree, node_to);
185         ntreeUpdateTree(bmain, ntree);
186
187         ED_node_tag_update_nodetree(bmain, ntree, node_to);
188 }
189
190 /* add new node connected to this socket, or replace an existing one */
191 static void node_socket_add_replace(const bContext *C, bNodeTree *ntree, bNode *node_to, bNodeSocket *sock_to,
192                                     int type, NodeLinkItem *item)
193 {
194         bNode *node_from;
195         bNodeSocket *sock_from_tmp;
196         bNode *node_prev = NULL;
197
198         /* unlink existing node */
199         if (sock_to->link) {
200                 node_prev = sock_to->link->fromnode;
201                 nodeRemLink(ntree, sock_to->link);
202         }
203
204         /* find existing node that we can use */
205         for (node_from = ntree->nodes.first; node_from; node_from = node_from->next)
206                 if (node_from->type == type)
207                         break;
208
209         if (node_from)
210                 if (node_from->inputs.first || node_from->typeinfo->draw_buttons || node_from->typeinfo->draw_buttons_ex)
211                         node_from = NULL;
212
213         if (node_prev && node_prev->type == type && node_link_item_compare(node_prev, item)) {
214                 /* keep the previous node if it's the same type */
215                 node_from = node_prev;
216         }
217         else if (!node_from) {
218                 node_from = nodeAddStaticNode(C, ntree, type);
219                 if (node_prev != NULL) {
220                         /* If we're replacing existing node, use it's location. */
221                         node_from->locx = node_prev->locx;
222                         node_from->locy = node_prev->locy;
223                         node_from->offsetx = node_prev->offsetx;
224                         node_from->offsety = node_prev->offsety;
225                 }
226                 else {
227                         /* Avoid exact intersection of nodes.
228                          * TODO(sergey): Still not ideal, but better than nothing.
229                          */
230                         int index = BLI_findindex(&node_to->inputs, sock_to);
231                         BLI_assert(index != -1);
232                         node_from->locx = node_to->locx - (node_from->typeinfo->width + 50);
233                         node_from->locy = node_to->locy - (node_from->typeinfo->height * index);
234                 }
235                 
236                 node_link_item_apply(node_from, item);
237         }
238
239         nodeSetActive(ntree, node_from);
240
241         /* add link */
242         sock_from_tmp = BLI_findlink(&node_from->outputs, item->socket_index);
243         nodeAddLink(ntree, node_from, sock_from_tmp, node_to, sock_to);
244         sock_to->flag &= ~SOCK_COLLAPSED;
245
246         /* copy input sockets from previous node */
247         if (node_prev && node_from != node_prev) {
248                 bNodeSocket *sock_prev, *sock_from;
249
250                 for (sock_prev = node_prev->inputs.first; sock_prev; sock_prev = sock_prev->next) {
251                         for (sock_from = node_from->inputs.first; sock_from; sock_from = sock_from->next) {
252                                 if (nodeCountSocketLinks(ntree, sock_from) >= sock_from->limit)
253                                         continue;
254
255                                 if (STREQ(sock_prev->name, sock_from->name) && sock_prev->type == sock_from->type) {
256                                         bNodeLink *link = sock_prev->link;
257
258                                         if (link && link->fromnode) {
259                                                 nodeAddLink(ntree, link->fromnode, link->fromsock, node_from, sock_from);
260                                                 nodeRemLink(ntree, link);
261                                         }
262
263                                         node_socket_copy_default_value(sock_from, sock_prev);
264                                 }
265                         }
266                 }
267
268                 /* also preserve mapping for texture nodes */
269                 if (node_from->typeinfo->nclass == NODE_CLASS_TEXTURE &&
270                     node_prev->typeinfo->nclass == NODE_CLASS_TEXTURE)
271                 {
272                         memcpy(node_from->storage, node_prev->storage, sizeof(NodeTexBase));
273                 }
274
275                 /* remove node */
276                 node_remove_linked(ntree, node_prev);
277         }
278
279         nodeUpdate(ntree, node_from);
280         nodeUpdate(ntree, node_to);
281         ntreeUpdateTree(CTX_data_main(C), ntree);
282
283         ED_node_tag_update_nodetree(CTX_data_main(C), ntree, node_to);
284 }
285
286 /****************************** Node Link Menu *******************************/
287
288 // #define UI_NODE_LINK_ADD        0
289 #define UI_NODE_LINK_DISCONNECT -1
290 #define UI_NODE_LINK_REMOVE     -2
291
292 typedef struct NodeLinkArg {
293         Main *bmain;
294         Scene *scene;
295         bNodeTree *ntree;
296         bNode *node;
297         bNodeSocket *sock;
298
299         bNodeType *node_type;
300         NodeLinkItem item;
301
302         uiLayout *layout;
303 } NodeLinkArg;
304
305 static void ui_node_link_items(NodeLinkArg *arg, int in_out, NodeLinkItem **r_items, int *r_totitems)
306 {
307         /* XXX this should become a callback for node types! */
308         NodeLinkItem *items = NULL;
309         int totitems = 0;
310         
311         if (arg->node_type->type == NODE_GROUP) {
312                 bNodeTree *ngroup;
313                 int i;
314                 
315                 for (ngroup = arg->bmain->nodetree.first; ngroup; ngroup = ngroup->id.next) {
316                         ListBase *lb = ((in_out == SOCK_IN) ? &ngroup->inputs : &ngroup->outputs);
317                         totitems += BLI_listbase_count(lb);
318                 }
319                 
320                 if (totitems > 0) {
321                         items = MEM_callocN(sizeof(NodeLinkItem) * totitems, "ui node link items");
322                         
323                         i = 0;
324                         for (ngroup = arg->bmain->nodetree.first; ngroup; ngroup = ngroup->id.next) {
325                                 ListBase *lb = (in_out == SOCK_IN ? &ngroup->inputs : &ngroup->outputs);
326                                 bNodeSocket *stemp;
327                                 int index;
328                                 for (stemp = lb->first, index = 0; stemp; stemp = stemp->next, ++index, ++i) {
329                                         NodeLinkItem *item = &items[i];
330                                         
331                                         item->socket_index = index;
332                                         /* note: int stemp->type is not fully reliable, not used for node group
333                                          * interface sockets. use the typeinfo->type instead.
334                                          */
335                                         item->socket_type = stemp->typeinfo->type;
336                                         item->socket_name = stemp->name;
337                                         item->node_name = ngroup->id.name + 2;
338                                         item->ngroup = ngroup;
339                                 }
340                         }
341                 }
342         }
343         else {
344                 bNodeSocketTemplate *socket_templates = (in_out == SOCK_IN ? arg->node_type->inputs : arg->node_type->outputs);
345                 bNodeSocketTemplate *stemp;
346                 int i;
347                 
348                 for (stemp = socket_templates; stemp && stemp->type != -1; ++stemp)
349                         ++totitems;
350                 
351                 if (totitems > 0) {
352                         items = MEM_callocN(sizeof(NodeLinkItem) * totitems, "ui node link items");
353                         
354                         i = 0;
355                         for (stemp = socket_templates; stemp && stemp->type != -1; ++stemp, ++i) {
356                                 NodeLinkItem *item = &items[i];
357                                 
358                                 item->socket_index = i;
359                                 item->socket_type = stemp->type;
360                                 item->socket_name = stemp->name;
361                                 item->node_name = arg->node_type->ui_name;
362                         }
363                 }
364         }
365         
366         *r_items = items;
367         *r_totitems = totitems;
368 }
369
370 static void ui_node_link(bContext *C, void *arg_p, void *event_p)
371 {
372         NodeLinkArg *arg = (NodeLinkArg *)arg_p;
373         Main *bmain = arg->bmain;
374         bNode *node_to = arg->node;
375         bNodeSocket *sock_to = arg->sock;
376         bNodeTree *ntree = arg->ntree;
377         int event = GET_INT_FROM_POINTER(event_p);
378
379         if (event == UI_NODE_LINK_DISCONNECT)
380                 node_socket_disconnect(bmain, ntree, node_to, sock_to);
381         else if (event == UI_NODE_LINK_REMOVE)
382                 node_socket_remove(bmain, ntree, node_to, sock_to);
383         else
384                 node_socket_add_replace(C, ntree, node_to, sock_to, arg->node_type->type, &arg->item);
385
386         ED_undo_push(C, "Node input modify");
387 }
388
389 static void ui_node_sock_name(bNodeSocket *sock, char name[UI_MAX_NAME_STR])
390 {
391         if (sock->link && sock->link->fromnode) {
392                 bNode *node = sock->link->fromnode;
393                 char node_name[UI_MAX_NAME_STR];
394
395                 if (node->type == NODE_GROUP) {
396                         if (node->id)
397                                 BLI_strncpy(node_name, node->id->name + 2, UI_MAX_NAME_STR);
398                         else
399                                 BLI_strncpy(node_name, N_(node->typeinfo->ui_name), UI_MAX_NAME_STR);
400                 }
401                 else
402                         BLI_strncpy(node_name, node->typeinfo->ui_name, UI_MAX_NAME_STR);
403
404                 if (BLI_listbase_is_empty(&node->inputs) &&
405                     node->outputs.first != node->outputs.last)
406                 {
407                         BLI_snprintf(name, UI_MAX_NAME_STR, "%s | %s", IFACE_(node_name), IFACE_(sock->link->fromsock->name));
408                 }
409                 else {
410                         BLI_strncpy(name, IFACE_(node_name), UI_MAX_NAME_STR);
411                 }
412         }
413         else if (sock->type == SOCK_SHADER)
414                 BLI_strncpy(name, IFACE_("None"), UI_MAX_NAME_STR);
415         else
416                 BLI_strncpy(name, IFACE_("Default"), UI_MAX_NAME_STR);
417 }
418
419 static int ui_compatible_sockets(int typeA, int typeB)
420 {
421         return (typeA == typeB);
422 }
423
424 static int ui_node_item_name_compare(const void *a, const void *b)
425 {
426         const bNodeType *type_a = *(const bNodeType **)a;
427         const bNodeType *type_b = *(const bNodeType **)b;
428         return BLI_natstrcmp(type_a->ui_name, type_b->ui_name);
429 }
430
431 static bool ui_node_item_special_poll(const bNodeTree *UNUSED(ntree),
432                                       const bNodeType *ntype)
433 {
434         if (STREQ(ntype->idname, "ShaderNodeUVAlongStroke")) {
435                 /* TODO(sergey): Currently we don't have Freestyle nodes edited from
436                  * the buttons context, so can ignore it's nodes completely.
437                  *
438                  * However, we might want to do some extra checks here later.
439                  */
440                 return false;
441         }
442         return true;
443 }
444
445 static void ui_node_menu_column(NodeLinkArg *arg, int nclass, const char *cname)
446 {
447         bNodeTree *ntree = arg->ntree;
448         bNodeSocket *sock = arg->sock;
449         uiLayout *layout = arg->layout;
450         uiLayout *column = NULL;
451         uiBlock *block = uiLayoutGetBlock(layout);
452         uiBut *but;
453         NodeLinkArg *argN;
454         int first = 1;
455         int compatibility = 0;
456         
457         if (ntree->type == NTREE_SHADER) {
458                 if (BKE_scene_use_new_shading_nodes(arg->scene))
459                         compatibility = NODE_NEW_SHADING;
460                 else
461                         compatibility = NODE_OLD_SHADING;
462         }
463
464         /* generate array of node types sorted by UI name */
465         bNodeType **sorted_ntypes = NULL;
466         BLI_array_declare(sorted_ntypes);
467
468         NODE_TYPES_BEGIN(ntype) {
469                 if (compatibility && !(ntype->compatibility & compatibility)) {
470                         continue;
471                 }
472
473                 if (ntype->nclass != nclass) {
474                         continue;
475                 }
476
477                 if (!ui_node_item_special_poll(ntree, ntype)) {
478                         continue;
479                 }
480
481                 BLI_array_append(sorted_ntypes, ntype);
482         }
483         NODE_TYPES_END
484
485         qsort(sorted_ntypes, BLI_array_count(sorted_ntypes), sizeof(bNodeType *), ui_node_item_name_compare);
486
487         /* generate UI */
488         for (int j = 0; j < BLI_array_count(sorted_ntypes); j++) {
489                 bNodeType *ntype = sorted_ntypes[j];
490                 NodeLinkItem *items;
491                 int totitems;
492                 char name[UI_MAX_NAME_STR];
493                 const char *cur_node_name = NULL;
494                 int i, num = 0;
495                 int icon = ICON_NONE;
496                 
497                 arg->node_type = ntype;
498                 
499                 ui_node_link_items(arg, SOCK_OUT, &items, &totitems);
500                 
501                 for (i = 0; i < totitems; ++i)
502                         if (ui_compatible_sockets(items[i].socket_type, sock->type))
503                                 num++;
504                 
505                 for (i = 0; i < totitems; ++i) {
506                         if (!ui_compatible_sockets(items[i].socket_type, sock->type))
507                                 continue;
508                         
509                         if (first) {
510                                 column = uiLayoutColumn(layout, 0);
511                                 UI_block_layout_set_current(block, column);
512                                 
513                                 uiItemL(column, IFACE_(cname), ICON_NODE);
514                                 but = block->buttons.last;
515                                 
516                                 first = 0;
517                         }
518                         
519                         if (num > 1) {
520                                 if (!cur_node_name || !STREQ(cur_node_name, items[i].node_name)) {
521                                         cur_node_name = items[i].node_name;
522                                         /* XXX Do not use uiItemL here, it would add an empty icon as we are in a menu! */
523                                         uiDefBut(block, UI_BTYPE_LABEL, 0, IFACE_(cur_node_name), 0, 0, UI_UNIT_X * 4, UI_UNIT_Y,
524                                                  NULL, 0.0, 0.0, 0.0, 0.0, "");
525                                 }
526
527                                 BLI_snprintf(name, UI_MAX_NAME_STR, "%s", IFACE_(items[i].socket_name));
528                                 icon = ICON_BLANK1;
529                         }
530                         else {
531                                 BLI_strncpy(name, IFACE_(items[i].node_name), UI_MAX_NAME_STR);
532                                 icon = ICON_NONE;
533                         }
534                         
535                         but = uiDefIconTextBut(block, UI_BTYPE_BUT, 0, icon, name, 0, 0, UI_UNIT_X * 4, UI_UNIT_Y,
536                                                NULL, 0.0, 0.0, 0.0, 0.0, TIP_("Add node to input"));
537                         
538                         argN = MEM_dupallocN(arg);
539                         argN->item = items[i];
540                         UI_but_funcN_set(but, ui_node_link, argN, NULL);
541                 }
542                 
543                 if (items)
544                         MEM_freeN(items);
545         }
546
547         BLI_array_free(sorted_ntypes);
548 }
549
550 static void node_menu_column_foreach_cb(void *calldata, int nclass, const char *name)
551 {
552         NodeLinkArg *arg = (NodeLinkArg *)calldata;
553
554         if (!ELEM(nclass, NODE_CLASS_GROUP, NODE_CLASS_LAYOUT))
555                 ui_node_menu_column(arg, nclass, name);
556 }
557
558 static void ui_template_node_link_menu(bContext *C, uiLayout *layout, void *but_p)
559 {
560         Main *bmain = CTX_data_main(C);
561         Scene *scene = CTX_data_scene(C);
562         uiBlock *block = uiLayoutGetBlock(layout);
563         uiBut *but = (uiBut *)but_p;
564         uiLayout *split, *column;
565         NodeLinkArg *arg = (NodeLinkArg *)but->func_argN;
566         bNodeSocket *sock = arg->sock;
567         bNodeTreeType *ntreetype = arg->ntree->typeinfo;
568
569         UI_block_flag_enable(block, UI_BLOCK_NO_FLIP | UI_BLOCK_IS_FLIP);
570         UI_block_layout_set_current(block, layout);
571         split = uiLayoutSplit(layout, 0.0f, false);
572
573         arg->bmain = bmain;
574         arg->scene = scene;
575         arg->layout = split;
576
577         if (ntreetype && ntreetype->foreach_nodeclass)
578                 ntreetype->foreach_nodeclass(scene, arg, node_menu_column_foreach_cb);
579
580         column = uiLayoutColumn(split, false);
581         UI_block_layout_set_current(block, column);
582
583         if (sock->link) {
584                 uiItemL(column, IFACE_("Link"), ICON_NONE);
585                 but = block->buttons.last;
586                 but->drawflag = UI_BUT_TEXT_LEFT;
587
588                 but = uiDefBut(block, UI_BTYPE_BUT, 0, IFACE_("Remove"), 0, 0, UI_UNIT_X * 4, UI_UNIT_Y,
589                                NULL, 0.0, 0.0, 0.0, 0.0, TIP_("Remove nodes connected to the input"));
590                 UI_but_funcN_set(but, ui_node_link, MEM_dupallocN(arg), SET_INT_IN_POINTER(UI_NODE_LINK_REMOVE));
591
592                 but = uiDefBut(block, UI_BTYPE_BUT, 0, IFACE_("Disconnect"), 0, 0, UI_UNIT_X * 4, UI_UNIT_Y,
593                                NULL, 0.0, 0.0, 0.0, 0.0, TIP_("Disconnect nodes connected to the input"));
594                 UI_but_funcN_set(but, ui_node_link, MEM_dupallocN(arg), SET_INT_IN_POINTER(UI_NODE_LINK_DISCONNECT));
595         }
596
597         ui_node_menu_column(arg, NODE_CLASS_GROUP, N_("Group"));
598 }
599
600 void uiTemplateNodeLink(uiLayout *layout, bNodeTree *ntree, bNode *node, bNodeSocket *sock)
601 {
602         uiBlock *block = uiLayoutGetBlock(layout);
603         NodeLinkArg *arg;
604         uiBut *but;
605
606         arg = MEM_callocN(sizeof(NodeLinkArg), "NodeLinkArg");
607         arg->ntree = ntree;
608         arg->node = node;
609         arg->sock = sock;
610
611         UI_block_layout_set_current(block, layout);
612
613         if (sock->link || sock->type == SOCK_SHADER || (sock->flag & SOCK_HIDE_VALUE)) {
614                 char name[UI_MAX_NAME_STR];
615                 ui_node_sock_name(sock, name);
616                 but = uiDefMenuBut(block, ui_template_node_link_menu, NULL, name, 0, 0, UI_UNIT_X * 4, UI_UNIT_Y, "");
617         }
618         else
619                 but = uiDefIconMenuBut(block, ui_template_node_link_menu, NULL, ICON_NONE, 0, 0, UI_UNIT_X, UI_UNIT_Y, "");
620
621         UI_but_type_set_menu_from_pulldown(but);
622
623         but->flag |= UI_BUT_NODE_LINK;
624         but->poin = (char *)but;
625         but->func_argN = arg;
626
627         if (sock->link && sock->link->fromnode)
628                 if (sock->link->fromnode->flag & NODE_ACTIVE_TEXTURE)
629                         but->flag |= UI_BUT_NODE_ACTIVE;
630 }
631
632 /**************************** Node Tree Layout *******************************/
633
634 static void ui_node_draw_input(uiLayout *layout, bContext *C,
635                                bNodeTree *ntree, bNode *node, bNodeSocket *input, int depth);
636
637 static void ui_node_draw_node(uiLayout *layout, bContext *C, bNodeTree *ntree, bNode *node, int depth)
638 {
639         bNodeSocket *input;
640         uiLayout *col, *split;
641         PointerRNA nodeptr;
642
643         RNA_pointer_create(&ntree->id, &RNA_Node, node, &nodeptr);
644
645         if (node->typeinfo->draw_buttons) {
646                 if (node->type != NODE_GROUP) {
647                         split = uiLayoutSplit(layout, 0.35f, false);
648                         col = uiLayoutColumn(split, false);
649                         col = uiLayoutColumn(split, false);
650
651                         node->typeinfo->draw_buttons(col, C, &nodeptr);
652                 }
653         }
654
655         for (input = node->inputs.first; input; input = input->next)
656                 ui_node_draw_input(layout, C, ntree, node, input, depth + 1);
657 }
658
659 static void ui_node_draw_input(uiLayout *layout, bContext *C, bNodeTree *ntree, bNode *node, bNodeSocket *input, int depth)
660 {
661         PointerRNA inputptr, nodeptr;
662         uiBlock *block = uiLayoutGetBlock(layout);
663         uiBut *bt;
664         uiLayout *split, *row, *col;
665         bNode *lnode;
666         char label[UI_MAX_NAME_STR];
667         int i, indent = (depth > 1) ? 2 * (depth - 1) : 0;
668         int dependency_loop;
669
670         if (input->flag & SOCK_UNAVAIL)
671                 return;
672
673         /* to avoid eternal loops on cyclic dependencies */
674         node->flag |= NODE_TEST;
675         lnode = (input->link) ? input->link->fromnode : NULL;
676
677         dependency_loop = (lnode && (lnode->flag & NODE_TEST));
678         if (dependency_loop)
679                 lnode = NULL;
680
681         /* socket RNA pointer */
682         RNA_pointer_create(&ntree->id, &RNA_NodeSocket, input, &inputptr);
683         RNA_pointer_create(&ntree->id, &RNA_Node, node, &nodeptr);
684
685         /* indented label */
686         for (i = 0; i < indent; i++) {
687                 label[i] = ' ';
688         }
689         label[indent] = '\0';
690         BLI_snprintf(label + indent, UI_MAX_NAME_STR - indent, "%s:", IFACE_(input->name));
691
692         /* split in label and value */
693         split = uiLayoutSplit(layout, 0.35f, false);
694
695         row = uiLayoutRow(split, true);
696
697         if (depth > 0) {
698                 UI_block_emboss_set(block, UI_EMBOSS_NONE);
699
700                 if (lnode && (lnode->inputs.first || (lnode->typeinfo->draw_buttons && lnode->type != NODE_GROUP))) {
701                         int icon = (input->flag & SOCK_COLLAPSED) ? ICON_DISCLOSURE_TRI_RIGHT : ICON_DISCLOSURE_TRI_DOWN;
702                         uiItemR(row, &inputptr, "show_expanded", UI_ITEM_R_ICON_ONLY, "", icon);
703                 }
704                 else
705                         uiItemL(row, "", ICON_BLANK1);
706
707                 bt = block->buttons.last;
708                 bt->rect.xmax = UI_UNIT_X / 2;
709
710                 UI_block_emboss_set(block, UI_EMBOSS);
711         }
712
713         uiItemL(row, label, ICON_NONE);
714         bt = block->buttons.last;
715         bt->drawflag = UI_BUT_TEXT_LEFT;
716
717         if (dependency_loop) {
718                 row = uiLayoutRow(split, false);
719                 uiItemL(row, IFACE_("Dependency Loop"), ICON_ERROR);
720         }
721         else if (lnode) {
722                 /* input linked to a node */
723                 uiTemplateNodeLink(split, ntree, node, input);
724
725                 if (depth == 0 || !(input->flag & SOCK_COLLAPSED)) {
726                         if (depth == 0)
727                                 uiItemS(layout);
728
729                         ui_node_draw_node(layout, C, ntree, lnode, depth);
730                 }
731         }
732         else {
733                 /* input not linked, show value */
734                 if (!(input->flag & SOCK_HIDE_VALUE)) {
735                         switch (input->type) {
736                                 case SOCK_FLOAT:
737                                 case SOCK_INT:
738                                 case SOCK_BOOLEAN:
739                                 case SOCK_RGBA:
740                                 case SOCK_STRING:
741                                         row = uiLayoutRow(split, true);
742                                         uiItemR(row, &inputptr, "default_value", 0, "", ICON_NONE);
743                                         break;
744                                 case SOCK_VECTOR:
745                                         row = uiLayoutRow(split, false);
746                                         col = uiLayoutColumn(row, false);
747                                         uiItemR(col, &inputptr, "default_value", 0, "", ICON_NONE);
748                                         break;
749                                         
750                                 default:
751                                         row = uiLayoutRow(split, false);
752                                         break;
753                         }
754                 }
755                 else
756                         row = uiLayoutRow(split, false);
757
758                 uiTemplateNodeLink(row, ntree, node, input);
759         }
760
761         /* clear */
762         node->flag &= ~NODE_TEST;
763 }
764
765 void uiTemplateNodeView(uiLayout *layout, bContext *C, bNodeTree *ntree, bNode *node, bNodeSocket *input)
766 {
767         bNode *tnode;
768
769         if (!ntree)
770                 return;
771
772         /* clear for cycle check */
773         for (tnode = ntree->nodes.first; tnode; tnode = tnode->next)
774                 tnode->flag &= ~NODE_TEST;
775
776         if (input)
777                 ui_node_draw_input(layout, C, ntree, node, input, 0);
778         else
779                 ui_node_draw_node(layout, C, ntree, node, 0);
780 }
781