d27a41c59f5cfd93993d140ae9fc196a86a4019d
[blender-staging.git] / source / blender / src / header_node.c
1 /**
2  *
3  * $Id:
4  *
5  * ***** BEGIN GPL LICENSE BLOCK *****
6  *
7  * This program is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU General Public License
9  * as published by the Free Software Foundation; either version 2
10  * of the License, or (at your option) any later version. 
11  *
12  * This program is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with this program; if not, write to the Free Software Foundation,
19  * Inc., 59 Temple Place - Suite 330, Boston, MA        02111-1307, USA.
20  *
21  * The Original Code is Copyright (C) 2005 Blender Foundation.
22  * All rights reserved.
23  *
24  * The Original Code is: all of this file.
25  *
26  * Contributor(s): David Millan Escriva, Juho Vepsäläinen
27  *
28  * ***** END GPL LICENSE BLOCK *****
29  */
30
31 #include <stdlib.h>
32 #include <string.h>
33 #include <stdio.h>
34
35 #include "DNA_ID.h"
36 #include "DNA_material_types.h"
37 #include "DNA_node_types.h"
38 #include "DNA_object_types.h"
39 #include "DNA_screen_types.h"
40 #include "DNA_scene_types.h"
41 #include "DNA_space_types.h"
42 #include "DNA_view2d_types.h"
43 #include "DNA_userdef_types.h"
44
45 #include "BIF_gl.h"
46 #include "BIF_interface.h"
47 #include "BIF_previewrender.h"
48 #include "BIF_resources.h"
49 #include "BIF_screen.h"
50 #include "BIF_space.h"
51 #include "BIF_toolbox.h"
52 #include "BIF_butspace.h"
53
54 #include "BKE_global.h"
55 #include "BKE_main.h"
56 #include "BKE_material.h"
57 #include "BKE_node.h"
58
59 #include "BSE_drawipo.h"
60 #include "BSE_headerbuttons.h"
61 #include "BSE_node.h"
62
63 #include "blendef.h"
64 #include "butspace.h"
65 #include "mydevice.h"
66
67 void do_node_buttons(ScrArea *sa, unsigned short event)
68 {
69         SpaceNode *snode= sa->spacedata.first;
70         Material *ma;
71         
72         switch(event) {
73                 case B_NODE_USEMAT:
74                         ma= (Material *)snode->id;
75                         if(ma) {
76                                 if(ma->use_nodes && ma->nodetree==NULL) {
77                                         node_shader_default(ma);
78                                         snode_set_context(snode);
79                                 }
80                                 BIF_preview_changed(ID_MA);
81                                 allqueue(REDRAWNODE, 0);
82                                 allqueue(REDRAWBUTSSHADING, 0);
83                                 allqueue(REDRAWIPO, 0);
84                         }               
85                         break;
86                         
87                 case B_NODE_USESCENE:
88                         if(G.scene->use_nodes) {
89                                 if(G.scene->nodetree==NULL)
90                                         node_composit_default(G.scene);
91                                 addqueue(curarea->win, UI_BUT_EVENT, B_NODE_TREE_EXEC);
92                         }
93                         snode_set_context(snode);
94                         allqueue(REDRAWNODE, 0);
95                         break;
96         }
97 }
98
99 static void do_node_viewmenu(void *arg, int event)
100 {
101         SpaceNode *snode= curarea->spacedata.first; 
102         
103         switch(event) {
104                 case 1: /* Zoom in */
105                         snode_zoom_in(curarea);
106                         break;
107                 case 2: /* View all */
108                         snode_zoom_out(curarea);
109                         break;
110                 case 3: /* View all */
111                         snode_home(curarea, snode);
112                         break;
113                 case 4: /* Grease Pencil */
114                         add_blockhandler(curarea, NODES_HANDLER_GREASEPENCIL, UI_PNL_UNSTOW);
115                         break;
116         }
117         allqueue(REDRAWNODE, 0);
118 }
119
120 static uiBlock *node_viewmenu(void *arg_unused)
121 {
122         SpaceNode *snode= curarea->spacedata.first; 
123         uiBlock *block;
124         short yco= 0, menuwidth=120;
125         
126         block= uiNewBlock(&curarea->uiblocks, "node_viewmenu", 
127                                           UI_EMBOSSP, UI_HELV, curarea->headwin);
128         uiBlockSetButmFunc(block, do_node_viewmenu, NULL);
129
130         if (snode->nodetree) {
131                 uiDefIconTextBut(block, BUTM, 1, ICON_MENU_PANEL, "Grease Pencil...", 0, yco-=20, menuwidth, 19, NULL, 0.0, 0.0, 1, 4, "");
132                 
133                 uiDefBut(block, SEPR, 0, "",        0, yco-=6, menuwidth, 6, NULL, 0.0, 0.0, 0, 0, "");
134         }
135         
136         uiDefIconTextBut(block, BUTM, 1, ICON_BLANK1, "Zoom In|NumPad +", 0, yco-=20, menuwidth, 19, NULL, 0.0, 0.0, 1, 1, "");
137         uiDefIconTextBut(block, BUTM, 1, ICON_BLANK1, "Zoom Out|NumPad -", 0, yco-=20, menuwidth, 19, NULL, 0.0, 0.0, 1, 2, "");
138         
139         uiDefBut(block, SEPR, 0, "",        0, yco-=6, menuwidth, 6, NULL, 0.0, 0.0, 0, 0, "");
140         
141         uiDefIconTextBut(block, BUTM, 1, ICON_BLANK1, "View All|Home", 0, yco-=20, menuwidth, 19, NULL, 0.0, 0.0, 1, 3, "");
142         
143         if (!curarea->full) 
144                 uiDefIconTextBut(block, BUTM, B_FULL, ICON_BLANK1, "Maximize Window|Ctrl UpArrow", 0, yco-=20, menuwidth, 19, NULL, 0.0, 0.0, 0, 4, "");
145         else 
146                 uiDefIconTextBut(block, BUTM, B_FULL, ICON_BLANK1, "Tile Window|Ctrl DownArrow", 0, yco-=20, menuwidth, 19, NULL, 0.0, 0.0, 0, 4, "");
147         
148         if(curarea->headertype==HEADERTOP) {
149                 uiBlockSetDirection(block, UI_DOWN);
150         }
151         else {
152                 uiBlockSetDirection(block, UI_TOP);
153                 uiBlockFlipOrder(block);
154         }
155         
156         uiTextBoundsBlock(block, 50);
157         
158         return block;
159 }
160
161 static void do_node_selectmenu(void *arg, int event)
162 {
163         SpaceNode *snode= curarea->spacedata.first; 
164         
165         /* functions in editnode.c assume there's a tree */
166         if(snode->nodetree==NULL)
167                 return;
168         
169         switch(event) {
170                 case 1: /* border select */
171                         node_border_select(snode);
172                         break;
173                 case 2: /* select/deselect all */
174                         node_deselectall(snode, 1);
175                         break;
176                 case 3: /* select linked in */
177                         node_select_linked(snode, 0);
178                         break;
179                 case 4: /* select linked out */
180                         node_select_linked(snode, 1);
181                         break;
182         }
183         allqueue(REDRAWNODE, 0);
184 }
185
186 static uiBlock *node_selectmenu(void *arg_unused)
187 {
188         uiBlock *block;
189         short yco= 0, menuwidth=120;
190         
191         block= uiNewBlock(&curarea->uiblocks, "node_selectmenu", 
192                                           UI_EMBOSSP, UI_HELV, curarea->headwin);
193         uiBlockSetButmFunc(block, do_node_selectmenu, NULL);
194         
195         uiDefIconTextBut(block, BUTM, 1, ICON_BLANK1, "Border Select|B", 0, yco-=20, menuwidth, 19, NULL, 0.0, 0.0, 1, 1, "");
196         
197         uiDefBut(block, SEPR, 0, "",        0, yco-=6, menuwidth, 6, NULL, 0.0, 0.0, 0, 0, "");
198         
199         uiDefIconTextBut(block, BUTM, 1, ICON_BLANK1, "Select/Deselect All|A", 0, yco-=20, menuwidth, 19, NULL, 0.0, 0.0, 1, 2, "");
200         uiDefIconTextBut(block, BUTM, 1, ICON_BLANK1, "Select Linked From|L", 0, yco-=20, menuwidth, 19, NULL, 0.0, 0.0, 1, 3, "");
201         uiDefIconTextBut(block, BUTM, 1, ICON_BLANK1, "Select Linked To|Shift L", 0, yco-=20, menuwidth, 19, NULL, 0.0, 0.0, 1, 4, "");
202         
203         if(curarea->headertype==HEADERTOP) {
204                 uiBlockSetDirection(block, UI_DOWN);
205         }
206         else {
207                 uiBlockSetDirection(block, UI_TOP);
208                 uiBlockFlipOrder(block);
209         }
210         
211         uiTextBoundsBlock(block, 50);
212         
213         return block;
214 }
215
216 void do_node_addmenu(void *arg, int event)
217 {
218         SpaceNode *snode= curarea->spacedata.first;
219         bNode *node;
220         float locx, locy;
221         short mval[2];
222         
223         /* store selection in temp test flag */
224         for(node= snode->edittree->nodes.first; node; node= node->next) {
225                 if(node->flag & NODE_SELECT) node->flag |= NODE_TEST;
226                 else node->flag &= ~NODE_TEST;
227         }
228         
229         toolbox_mousepos(mval, 0 ); /* get initial mouse position */
230         areamouseco_to_ipoco(G.v2d, mval, &locx, &locy);
231         node= node_add_node(snode, event, locx, locy);
232         
233         /* uses test flag */
234         snode_autoconnect(snode, node, NODE_TEST);
235                 
236         addqueue(curarea->win, UI_BUT_EVENT, B_NODE_TREE_EXEC);
237         
238         BIF_undo_push("Add Node");
239         
240 }
241
242 static void node_make_addmenu(SpaceNode *snode, int nodeclass, uiBlock *block)
243 {
244         bNodeTree *ntree;
245         int tot= 0, a;
246         short yco= 0, menuwidth=120;
247         
248         ntree = snode->nodetree;
249         if(ntree) {
250                 /* mostly taken from toolbox.c, node_add_sublevel() */
251                 if(ntree) {
252                         if(nodeclass==NODE_CLASS_GROUP) {
253                                 bNodeTree *ngroup= G.main->nodetree.first;
254                                 for(; ngroup; ngroup= ngroup->id.next)
255                                         if(ngroup->type==ntree->type)
256                                                 tot++;
257                         }
258                         else {
259                                 bNodeType *type = ntree->alltypes.first;
260                                 while(type) {
261                                         if(type->nclass == nodeclass)
262                                                 tot++;
263                                         type= type->next;
264                                 }
265                         }
266                 }       
267                 
268                 if(tot==0) {
269                         uiDefBut(block, SEPR, 0, "",        0, yco-=6, menuwidth, 6, NULL, 0.0, 0.0, 0, 0, "");
270                         return;
271                 }
272                 
273                 if(nodeclass==NODE_CLASS_GROUP) {
274                         bNodeTree *ngroup= G.main->nodetree.first;
275                         for(tot=0, a=0; ngroup; ngroup= ngroup->id.next, tot++) {
276                                 if(ngroup->type==ntree->type) {
277                                         
278                                         uiDefIconTextBut(block, BUTM, 1, ICON_BLANK1, (ngroup->id.name+2), 0, 
279                                                 yco-=20, menuwidth, 19, NULL, 0.0, 0.0, 1, NODE_GROUP_MENU+tot, "");
280                                         a++;
281                                 }
282                         }
283                 }
284                 else {
285                         bNodeType *type;
286                         int script=0;
287                         for(a=0, type= ntree->alltypes.first; type; type=type->next) {
288                                 if( type->nclass == nodeclass ) {
289                                         if(type->type == NODE_DYNAMIC) {
290                                                 uiDefIconTextBut(block, BUTM, 1, ICON_BLANK1, type->name, 0, 
291                                                         yco-=20, menuwidth, 19, NULL, 0.0, 0.0, 1, NODE_DYNAMIC_MENU+script, "");
292                                                 script++;
293                                         } else {
294                                         uiDefIconTextBut(block, BUTM, 1, ICON_BLANK1, type->name, 0, 
295                                                 yco-=20, menuwidth, 19, NULL, 0.0, 0.0, 1, type->type, "");
296                                         }
297                                         a++;
298                                 }
299                         }
300                 }
301         } else {
302                 uiDefBut(block, SEPR, 0, "",        0, yco-=6, menuwidth, 6, NULL, 0.0, 0.0, 0, 0, "");
303                 return;
304         }
305 }
306
307 static uiBlock *node_add_inputmenu(void *arg_unused)
308 {
309         SpaceNode *snode= curarea->spacedata.first;
310         uiBlock *block;
311
312         block= uiNewBlock(&curarea->uiblocks, "node_add_inputmenu", UI_EMBOSSP, UI_HELV, G.curscreen->mainwin);
313         uiBlockSetButmFunc(block, do_node_addmenu, NULL);
314         
315         node_make_addmenu(snode, NODE_CLASS_INPUT, block);
316         
317         uiBlockSetDirection(block, UI_RIGHT);
318         uiTextBoundsBlock(block, 60);
319         
320         return block;
321 }
322 static uiBlock *node_add_outputmenu(void *arg_unused)
323 {
324         SpaceNode *snode= curarea->spacedata.first;
325         uiBlock *block;
326         
327         block= uiNewBlock(&curarea->uiblocks, "node_add_outputmenu", UI_EMBOSSP, UI_HELV, G.curscreen->mainwin);
328         uiBlockSetButmFunc(block, do_node_addmenu, NULL);
329         
330         node_make_addmenu(snode, NODE_CLASS_OUTPUT, block);
331         
332         uiBlockSetDirection(block, UI_RIGHT);
333         uiTextBoundsBlock(block, 60);
334         
335         return block;
336 }
337 static uiBlock *node_add_colormenu(void *arg_unused)
338 {
339         SpaceNode *snode= curarea->spacedata.first;
340         uiBlock *block;
341         
342         block= uiNewBlock(&curarea->uiblocks, "node_add_colormenu", UI_EMBOSSP, UI_HELV, G.curscreen->mainwin);
343         uiBlockSetButmFunc(block, do_node_addmenu, NULL);
344         
345         node_make_addmenu(snode, NODE_CLASS_OP_COLOR, block);
346         
347         uiBlockSetDirection(block, UI_RIGHT);
348         uiTextBoundsBlock(block, 60);
349         
350         return block;
351 }
352 static uiBlock *node_add_vectormenu(void *arg_unused)
353 {
354         SpaceNode *snode= curarea->spacedata.first;
355         uiBlock *block;
356         
357         block= uiNewBlock(&curarea->uiblocks, "node_add_vectormenu", UI_EMBOSSP, UI_HELV, G.curscreen->mainwin);
358         uiBlockSetButmFunc(block, do_node_addmenu, NULL);
359         
360         node_make_addmenu(snode, NODE_CLASS_OP_VECTOR, block);
361         
362         uiBlockSetDirection(block, UI_RIGHT);
363         uiTextBoundsBlock(block, 60);
364         
365         return block;
366 }
367 static uiBlock *node_add_filtermenu(void *arg_unused)
368 {
369         SpaceNode *snode= curarea->spacedata.first;
370         uiBlock *block;
371         
372         block= uiNewBlock(&curarea->uiblocks, "node_add_filtermenu", UI_EMBOSSP, UI_HELV, G.curscreen->mainwin);
373         uiBlockSetButmFunc(block, do_node_addmenu, NULL);
374         
375         node_make_addmenu(snode, NODE_CLASS_OP_FILTER, block);
376         
377         uiBlockSetDirection(block, UI_RIGHT);
378         uiTextBoundsBlock(block, 60);
379         
380         return block;
381 }
382 static uiBlock *node_add_convertermenu(void *arg_unused)
383 {
384         SpaceNode *snode= curarea->spacedata.first;
385         uiBlock *block;
386         
387         block= uiNewBlock(&curarea->uiblocks, "node_add_convertermenu", UI_EMBOSSP, UI_HELV, G.curscreen->mainwin);
388         uiBlockSetButmFunc(block, do_node_addmenu, NULL);
389         
390         node_make_addmenu(snode, NODE_CLASS_CONVERTOR, block);
391         
392         uiBlockSetDirection(block, UI_RIGHT);
393         uiTextBoundsBlock(block, 60);
394         
395         return block;
396 }
397 static uiBlock *node_add_mattemenu(void *arg_unused)
398 {
399         SpaceNode *snode= curarea->spacedata.first;
400         uiBlock *block;
401         
402         block= uiNewBlock(&curarea->uiblocks, "node_add_mattemenu", UI_EMBOSSP, UI_HELV, G.curscreen->mainwin);
403         uiBlockSetButmFunc(block, do_node_addmenu, NULL);
404         
405         node_make_addmenu(snode, NODE_CLASS_MATTE, block);
406         
407         uiBlockSetDirection(block, UI_RIGHT);
408         uiTextBoundsBlock(block, 60);
409         
410         return block;
411 }
412 static uiBlock *node_add_distortmenu(void *arg_unused)
413 {
414         SpaceNode *snode= curarea->spacedata.first;
415         uiBlock *block;
416         
417         block= uiNewBlock(&curarea->uiblocks, "node_add_distortmenu", UI_EMBOSSP, UI_HELV, G.curscreen->mainwin);
418         uiBlockSetButmFunc(block, do_node_addmenu, NULL);
419         
420         node_make_addmenu(snode, NODE_CLASS_DISTORT, block);
421         
422         uiBlockSetDirection(block, UI_RIGHT);
423         uiTextBoundsBlock(block, 60);
424         
425         return block;
426 }
427 static uiBlock *node_add_groupmenu(void *arg_unused)
428 {
429         SpaceNode *snode= curarea->spacedata.first;
430         uiBlock *block;
431         
432         block= uiNewBlock(&curarea->uiblocks, "node_add_groupmenu", UI_EMBOSSP, UI_HELV, G.curscreen->mainwin);
433         uiBlockSetButmFunc(block, do_node_addmenu, NULL);
434         
435         node_make_addmenu(snode, NODE_CLASS_GROUP, block);
436         
437         uiBlockSetDirection(block, UI_RIGHT);
438         uiTextBoundsBlock(block, 60);
439         
440         return block;
441 }
442
443 static uiBlock *node_add_dynamicmenu(void *arg_unused)
444 {
445         SpaceNode *snode= curarea->spacedata.first;
446         uiBlock *block;
447         
448         block= uiNewBlock(&curarea->uiblocks, "node_add_dynamicmenu", UI_EMBOSSP, UI_HELV, G.curscreen->mainwin);
449         uiBlockSetButmFunc(block, do_node_addmenu, NULL);
450         
451         node_make_addmenu(snode, NODE_CLASS_OP_DYNAMIC, block);
452         
453         uiBlockSetDirection(block, UI_RIGHT);
454         uiTextBoundsBlock(block, 60);
455         
456         return block;
457 }
458
459 static uiBlock *node_addmenu(void *arg_unused)
460 {
461         SpaceNode *snode= curarea->spacedata.first;
462         uiBlock *block;
463         short yco= 0, menuwidth=120;
464         
465         block= uiNewBlock(&curarea->uiblocks, "node_addmenu", 
466                                           UI_EMBOSSP, UI_HELV, curarea->headwin);
467         uiBlockSetButmFunc(block, do_node_addmenu, NULL);
468         
469         if(snode->treetype==NTREE_SHADER) {
470                 uiDefIconTextBlockBut(block, node_add_inputmenu, NULL, ICON_RIGHTARROW_THIN, "Input", 0, yco-=20, 120, 19, "");
471                 uiDefIconTextBlockBut(block, node_add_outputmenu, NULL, ICON_RIGHTARROW_THIN, "Output", 0, yco-=20, 120, 19, "");
472                 uiDefIconTextBlockBut(block, node_add_colormenu, NULL, ICON_RIGHTARROW_THIN, "Color", 0, yco-=20, 120, 19, "");
473                 uiDefIconTextBlockBut(block, node_add_vectormenu, NULL, ICON_RIGHTARROW_THIN, "Vector", 0, yco-=20, 120, 19, "");
474                 uiDefIconTextBlockBut(block, node_add_convertermenu, NULL, ICON_RIGHTARROW_THIN, "Convertor", 0, yco-=20, 120, 19, "");
475                 uiDefIconTextBlockBut(block, node_add_groupmenu, NULL, ICON_RIGHTARROW_THIN, "Group", 0, yco-=20, 120, 19, "");
476                 uiDefIconTextBlockBut(block, node_add_dynamicmenu, NULL, ICON_RIGHTARROW_THIN, "Dynamic", 0, yco-=20, 120, 19, "");
477         }
478         else if(snode->treetype==NTREE_COMPOSIT) {
479                 uiDefIconTextBlockBut(block, node_add_inputmenu, NULL, ICON_RIGHTARROW_THIN, "Input", 0, yco-=20, 120, 19, "");
480                 uiDefIconTextBlockBut(block, node_add_outputmenu, NULL, ICON_RIGHTARROW_THIN, "Output", 0, yco-=20, 120, 19, "");
481                 uiDefIconTextBlockBut(block, node_add_colormenu, NULL, ICON_RIGHTARROW_THIN, "Color", 0, yco-=20, 120, 19, "");
482                 uiDefIconTextBlockBut(block, node_add_vectormenu, NULL, ICON_RIGHTARROW_THIN, "Vector", 0, yco-=20, 120, 19, "");
483                 uiDefIconTextBlockBut(block, node_add_filtermenu, NULL, ICON_RIGHTARROW_THIN, "Filter", 0, yco-=20, 120, 19, "");
484                 uiDefIconTextBlockBut(block, node_add_convertermenu, NULL, ICON_RIGHTARROW_THIN, "Convertor", 0, yco-=20, 120, 19, "");
485                 uiDefIconTextBlockBut(block, node_add_mattemenu, NULL, ICON_RIGHTARROW_THIN, "Matte", 0, yco-=20, 120, 19, "");
486                 uiDefIconTextBlockBut(block, node_add_distortmenu, NULL, ICON_RIGHTARROW_THIN, "Distort", 0, yco-=20, 120, 19, "");
487                 uiDefIconTextBlockBut(block, node_add_groupmenu, NULL, ICON_RIGHTARROW_THIN, "Group", 0, yco-=20, 120, 19, "");
488
489         } else
490                 uiDefBut(block, SEPR, 0, "",        0, yco-=6, menuwidth, 6, NULL, 0.0, 0.0, 0, 0, ""); 
491         
492         if(curarea->headertype==HEADERTOP) {
493                 uiBlockSetDirection(block, UI_DOWN);
494         }
495         else {
496                 uiBlockSetDirection(block, UI_TOP);
497                 uiBlockFlipOrder(block);
498         }
499         
500         uiTextBoundsBlock(block, 50);
501         
502         return block;
503 }
504
505 static void do_node_nodemenu(void *arg, int event)
506 {
507         SpaceNode *snode= curarea->spacedata.first; 
508         int fromlib=0;
509         
510         /* functions in editnode.c assume there's a tree */
511         if(snode->nodetree==NULL)
512                 return;
513         fromlib= (snode->id && snode->id->lib);
514         
515         switch(event) {
516                 case 1: /* grab/move */
517                         node_transform_ext(0,0);
518                         break;
519                 case 2: /* duplicate */
520                         if(fromlib) fromlib= -1;
521                         else node_adduplicate(snode);
522                         break;
523                 case 3: /* delete */
524                         if(fromlib) fromlib= -1;
525                         else node_delete(snode);
526                         break;
527                 case 4: /* make group */
528                         node_make_group(snode);
529                         break;
530                 case 5: /* ungroup */
531                         node_ungroup(snode);
532                         break;
533                 case 6: /* edit group */
534                         if(fromlib) fromlib= -1;
535                         else snode_make_group_editable(snode, NULL);
536                         break;
537                 case 7: /* hide/unhide */
538                         node_hide(snode);
539                         break;
540                 case 8: /* read saved render layers */
541                         node_read_renderlayers(snode);
542                         break;
543                 case 9: /* show cyclic */
544                         ntreeSolveOrder(snode->edittree);
545                         break;
546                 case 10: /* execute */
547                         addqueue(curarea->win, UI_BUT_EVENT, B_NODE_TREE_EXEC);
548                         break;
549                 case 11: /* make link */
550                         node_make_link(snode);
551                         break;
552                 case 12: /* rename */
553                         node_rename(snode);
554                         break;
555                 case 13: /* read saved full sample layers */
556                         node_read_fullsamplelayers(snode);
557                         break;
558                 case 14: /* connect viewer */
559                         node_active_link_viewer(snode);
560                         break;
561                         
562         }
563         
564         if(fromlib==-1) error_libdata();
565         allqueue(REDRAWNODE, 0);
566 }
567
568 static uiBlock *node_nodemenu(void *arg_unused)
569 {
570         SpaceNode *snode= curarea->spacedata.first; 
571         uiBlock *block;
572         short yco= 0, menuwidth=120;
573         
574         block= uiNewBlock(&curarea->uiblocks, "node_nodemenu", 
575                                           UI_EMBOSSP, UI_HELV, curarea->headwin);
576         uiBlockSetButmFunc(block, do_node_nodemenu, NULL);
577         
578         uiDefIconTextBut(block, BUTM, 1, ICON_BLANK1, "Grab/Move|G", 0, yco-=20, menuwidth, 19, NULL, 0.0, 0.0, 1, 1, "");
579         
580         uiDefBut(block, SEPR, 0, "",        0, yco-=6, menuwidth, 6, NULL, 0.0, 0.0, 0, 0, "");
581         
582         uiDefIconTextBut(block, BUTM, 1, ICON_BLANK1, "Duplicate|Shift D", 0, yco-=20, menuwidth, 19, NULL, 0.0, 0.0, 1, 2, "");
583         uiDefIconTextBut(block, BUTM, 1, ICON_BLANK1, "Delete|X", 0, yco-=20, menuwidth, 19, NULL, 0.0, 0.0, 1, 3, "");
584         
585         uiDefBut(block, SEPR, 0, "",        0, yco-=6, menuwidth, 6, NULL, 0.0, 0.0, 0, 0, "");
586
587         uiDefIconTextBut(block, BUTM, 1, ICON_BLANK1, "Make Link|F", 0, yco-=20, menuwidth, 19, NULL, 0.0, 0.0, 1, 11, "");
588
589         uiDefBut(block, SEPR, 0, "",        0, yco-=6, menuwidth, 6, NULL, 0.0, 0.0, 0, 0, "");
590         
591         uiDefIconTextBut(block, BUTM, 1, ICON_BLANK1, "Make Group|Ctrl G", 0, yco-=20, menuwidth, 19, NULL, 0.0, 0.0, 1, 4, "");
592         uiDefIconTextBut(block, BUTM, 1, ICON_BLANK1, "Ungroup|Alt G", 0, yco-=20, menuwidth, 19, NULL, 0.0, 0.0, 1, 5, "");
593         uiDefIconTextBut(block, BUTM, 1, ICON_BLANK1, "Edit Group|Tab", 0, yco-=20, menuwidth, 19, NULL, 0.0, 0.0, 1, 6, "");
594         
595         uiDefBut(block, SEPR, 0, "",        0, yco-=6, menuwidth, 6, NULL, 0.0, 0.0, 0, 0, "");
596         
597         uiDefIconTextBut(block, BUTM, 1, ICON_BLANK1, "Hide/Unhide|H", 0, yco-=20, menuwidth, 19, NULL, 0.0, 0.0, 1, 7, "");
598         uiDefIconTextBut(block, BUTM, 1, ICON_BLANK1, "Rename|Ctrl R", 0, yco-=20, menuwidth, 19, NULL, 0.0, 0.0, 1, 12, "");
599         
600         uiDefBut(block, SEPR, 0, "",        0, yco-=6, menuwidth, 6, NULL, 0.0, 0.0, 0, 0, "");
601         
602         if(snode->treetype==NTREE_COMPOSIT) {
603                 uiDefIconTextBut(block, BUTM, 1, ICON_BLANK1, "Execute Composite|E", 0, yco-=20, menuwidth, 19, NULL, 0.0, 0.0, 1, 10, "");
604                 uiDefIconTextBut(block, BUTM, 1, ICON_BLANK1, "Read Saved Render Results|R", 0, yco-=20, menuwidth, 19, NULL, 0.0, 0.0, 1, 8, "");
605                 uiDefIconTextBut(block, BUTM, 1, ICON_BLANK1, "Read Saved Full Sample Results|Shift R", 0, yco-=20, menuwidth, 19, NULL, 0.0, 0.0, 1, 13, "");
606                 
607                 uiDefBut(block, SEPR, 0, "",        0, yco-=6, menuwidth, 6, NULL, 0.0, 0.0, 0, 0, "");
608                 
609                 uiDefIconTextBut(block, BUTM, 1, ICON_BLANK1, "Connect Node to Viewer|Ctrl RMB", 0, yco-=20, menuwidth, 19, NULL, 0.0, 0.0, 1, 14, "");
610                 
611                 uiDefBut(block, SEPR, 0, "",        0, yco-=6, menuwidth, 6, NULL, 0.0, 0.0, 0, 0, "");
612         }
613         uiDefIconTextBut(block, BUTM, 1, ICON_BLANK1, "Show Cyclic Dependencies|C", 0, yco-=20, menuwidth, 19, NULL, 0.0, 0.0, 1, 9, "");
614         
615         if(curarea->headertype==HEADERTOP) {
616                 uiBlockSetDirection(block, UI_DOWN);
617         }
618         else {
619                 uiBlockSetDirection(block, UI_TOP);
620                 uiBlockFlipOrder(block);
621         }
622         
623         uiTextBoundsBlock(block, 50);
624         
625         return block;
626 }
627
628 void node_buttons(ScrArea *sa)
629 {
630         SpaceNode *snode= sa->spacedata.first;
631         uiBlock *block;
632         short xco, xmax;
633         char name[256];
634         
635         sprintf(name, "header %d", sa->headwin);
636         block= uiNewBlock(&sa->uiblocks, name, UI_EMBOSS, UI_HELV, sa->headwin);
637
638         if(area_is_active_area(sa)) uiBlockSetCol(block, TH_HEADER);
639         else uiBlockSetCol(block, TH_HEADERDESEL);
640         
641         sa->butspacetype= SPACE_NODE;
642
643         xco = 8;
644         
645         uiDefIconTextButC(block, ICONTEXTROW,B_NEWSPACE, ICON_VIEW3D, 
646                                           windowtype_pup(), xco, 0, XIC+10, YIC, 
647                                           &(sa->butspacetype), 1.0, SPACEICONMAX, 0, 0, 
648                                           "Displays Current Window Type");
649
650         xco += XIC + 14;
651
652         uiBlockSetEmboss(block, UI_EMBOSSN);
653         if (sa->flag & HEADER_NO_PULLDOWN) {
654                 uiDefIconButBitS(block, TOG, HEADER_NO_PULLDOWN, B_FLIPINFOMENU, 
655                                           ICON_DISCLOSURE_TRI_RIGHT, xco,2,XIC,YIC-2,
656                                           &(sa->flag), 0, 0, 0, 0,  "Show pulldown menus");
657         }
658         else {
659                 uiDefIconButBitS(block, TOG, HEADER_NO_PULLDOWN, B_FLIPINFOMENU, 
660                                           ICON_DISCLOSURE_TRI_DOWN,  xco,2,XIC,YIC-2,
661                                           &(sa->flag), 0, 0, 0, 0,  "Hide pulldown menus");
662         }
663         xco+=XIC;
664
665         if((sa->flag & HEADER_NO_PULLDOWN)==0) {
666                 /* pull down menus */
667                 uiBlockSetEmboss(block, UI_EMBOSSP);
668         
669                 xmax= GetButStringLength("View");
670                 uiDefPulldownBut(block, node_viewmenu, NULL, 
671                                           "View", xco, -2, xmax-3, 24, "");
672                 xco+= xmax;
673                 
674                 xmax= GetButStringLength("Select");
675                 uiDefPulldownBut(block, node_selectmenu, NULL, 
676                                                  "Select", xco, -2, xmax-3, 24, "");
677                 xco+= xmax;
678                 
679                 xmax= GetButStringLength("Add");
680                 uiDefPulldownBut(block, node_addmenu, NULL, 
681                                                  "Add", xco, -2, xmax-3, 24, "");
682                 xco+= xmax;
683                 
684                 xmax= GetButStringLength("Node");
685                 uiDefPulldownBut(block, node_nodemenu, NULL, 
686                                                  "Node", xco, -2, xmax-3, 24, "");
687                 xco+= xmax;
688         }
689         
690         uiBlockSetEmboss(block, UI_EMBOSS);
691         
692         /* main type choosing */
693         uiBlockBeginAlign(block);
694         uiDefIconButI(block, ROW, B_REDR, ICON_MATERIAL_DEHLT, xco,2,XIC,YIC-2,
695                                   &(snode->treetype), 2, 0, 0, 0, "Material Nodes");
696         xco+= XIC;
697         uiDefIconButI(block, ROW, B_REDR, ICON_IMAGE_DEHLT, xco,2,XIC,YIC-2,
698                                   &(snode->treetype), 2, 1, 0, 0, "Composite Nodes");
699         xco+= 2*XIC;
700         uiBlockEndAlign(block);
701         
702         /* find and set the context */
703         snode_set_context(snode);
704         
705         if(snode->treetype==NTREE_SHADER) {
706                 if(snode->from) {
707                                                                                 /* 0, NULL -> pin */
708                         xco= std_libbuttons(block, xco, 0, 0, NULL, B_MATBROWSE, ID_MA, 1, snode->id, snode->from, &(snode->menunr), 
709                                            B_MATALONE, B_MATLOCAL, B_MATDELETE, B_AUTOMATNAME, B_KEEPDATA);
710                         
711                         if(snode->id) {
712                                 Material *ma= (Material *)snode->id;
713                                 uiDefButC(block, TOG, B_NODE_USEMAT, "Use Nodes", xco+5,0,70,19, &ma->use_nodes, 0.0f, 0.0f, 0, 0, "");
714                                 xco+=80;
715                         }
716                 }
717         }
718         else if(snode->treetype==NTREE_COMPOSIT) {
719                 uiDefButS(block, TOG, B_NODE_USESCENE, "Use Nodes", xco+5,0,70,19, &G.scene->use_nodes, 0.0f, 0.0f, 0, 0, "Indicate this Scene will use Nodes and execute them while editing");
720                 xco+= 80;
721                 uiDefButBitI(block, TOG, R_COMP_FREE, B_NOP, "Free Unused", xco+5,0,80,19, &G.scene->r.scemode, 0.0f, 0.0f, 0, 0, "Free Nodes that are not used while composite");
722                 xco+= 80;
723                 uiDefButBitS(block, TOG, SNODE_BACKDRAW, REDRAWNODE, "Backdrop", xco+5,0,80,19, &snode->flag, 0.0f, 0.0f, 0, 0, "Use active Viewer Node output as backdrop");
724                 xco+= 80;
725         }
726         
727         /* always as last  */
728         sa->headbutlen= xco+2*XIC;
729
730         uiDrawBlock(block);
731 }