source/blender/editors/space_node/node_header.c

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