Code cleanup:
[blender.git] / source / blender / editors / space_node / node_buttons.c
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version. 
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * The Original Code is Copyright (C) 2009 Blender Foundation.
19  * All rights reserved.
20  *
21  * 
22  * Contributor(s): Blender Foundation
23  *
24  * ***** END GPL LICENSE BLOCK *****
25  */
26
27 /** \file blender/editors/space_node/node_buttons.c
28  *  \ingroup spnode
29  */
30
31
32 #include <string.h>
33 #include <stdio.h>
34 #include <math.h>
35 #include <float.h>
36
37 #include "MEM_guardedalloc.h"
38
39 #include "DNA_node_types.h"
40 #include "DNA_scene_types.h"
41
42 #include "BLI_math.h"
43 #include "BLI_blenlib.h"
44 #include "BLI_rand.h"
45 #include "BLI_utildefines.h"
46
47 #include "BKE_context.h"
48 #include "BKE_global.h"
49 #include "BKE_node.h"
50 #include "BKE_screen.h"
51
52 #include "WM_api.h"
53 #include "WM_types.h"
54
55 #include "RNA_access.h"
56
57 #include "ED_gpencil.h"
58 #include "ED_screen.h"
59
60 #include "UI_interface.h"
61 #include "UI_resources.h"
62
63 #include "node_intern.h"        // own include
64
65
66 /* ******************* node space & buttons ************** */
67
68 /* poll callback for active node */
69 static int active_node_poll(const bContext *C, PanelType *UNUSED(pt))
70 {
71         SpaceNode *snode= CTX_wm_space_node(C);
72         
73         // TODO: include check for whether there is an active node...
74         return (snode && snode->nodetree);
75 }
76
77 /* active node */
78 static void active_node_panel(const bContext *C, Panel *pa)
79 {
80         SpaceNode *snode= CTX_wm_space_node(C);
81         bNodeTree *ntree= (snode) ? snode->edittree : NULL;
82         bNode *node = (ntree) ? nodeGetActive(ntree) : NULL; // xxx... for editing group nodes
83         uiLayout *layout= pa->layout;
84         PointerRNA ptr;
85         
86         /* verify pointers, and create RNA pointer for the node */
87         if (ELEM(NULL, ntree, node))
88                 return;
89         //if (node->id) /* for group nodes */
90         //      RNA_pointer_create(node->id, &RNA_Node, node, &ptr);
91         //else
92                 RNA_pointer_create(&ntree->id, &RNA_Node, node, &ptr); 
93         
94         /* draw this node's name, etc. */
95         uiItemR(layout, &ptr, "label", 0, NULL, ICON_NODE);
96         uiItemS(layout);
97         uiItemR(layout, &ptr, "name", 0, NULL, ICON_NODE);
98         uiItemS(layout);
99         
100         uiItemO(layout, NULL, 0, "NODE_OT_hide_socket_toggle");
101         uiItemS(layout);
102
103         /* draw this node's settings */
104         if (node->typeinfo && node->typeinfo->uifuncbut)
105                 node->typeinfo->uifuncbut(layout, (bContext *)C, &ptr);
106         else if (node->typeinfo && node->typeinfo->uifunc)
107                 node->typeinfo->uifunc(layout, (bContext *)C, &ptr);
108 }
109
110 static int node_sockets_poll(const bContext *C, PanelType *UNUSED(pt))
111 {
112         SpaceNode *snode= CTX_wm_space_node(C);
113         
114         return (snode && snode->nodetree && G.rt == 777);
115 }
116
117 static void node_sockets_panel(const bContext *C, Panel *pa)
118 {
119         SpaceNode *snode= CTX_wm_space_node(C);
120         bNodeTree *ntree= (snode) ? snode->edittree : NULL;
121         bNode *node = (ntree) ? nodeGetActive(ntree) : NULL;
122         bNodeSocket *sock;
123         uiLayout *layout= pa->layout, *split;
124         char name[UI_MAX_NAME_STR];
125         
126         if (ELEM(NULL, ntree, node))
127                 return;
128         
129         for (sock=node->inputs.first; sock; sock=sock->next) {
130                 BLI_snprintf(name, sizeof(name), "%s:", sock->name);
131
132                 split = uiLayoutSplit(layout, 0.35f, 0);
133                 uiItemL(split, name, ICON_NONE);
134                 uiTemplateNodeLink(split, ntree, node, sock);
135         }
136 }
137
138 /* ******************* node buttons registration ************** */
139
140 void node_buttons_register(ARegionType *art)
141 {
142         PanelType *pt;
143         
144         pt= MEM_callocN(sizeof(PanelType), "spacetype node panel active node");
145         strcpy(pt->idname, "NODE_PT_item");
146         strcpy(pt->label, "Active Node");
147         pt->draw= active_node_panel;
148         pt->poll= active_node_poll;
149         BLI_addtail(&art->paneltypes, pt);
150
151         pt= MEM_callocN(sizeof(PanelType), "spacetype node panel node sockets");
152         strcpy(pt->idname, "NODE_PT_sockets");
153         strcpy(pt->label, "Sockets");
154         pt->draw= node_sockets_panel;
155         pt->poll= node_sockets_poll;
156         pt->flag |= PNL_DEFAULT_CLOSED;
157         BLI_addtail(&art->paneltypes, pt);
158         
159         pt= MEM_callocN(sizeof(PanelType), "spacetype node panel gpencil");
160         strcpy(pt->idname, "NODE_PT_gpencil");
161         strcpy(pt->label, "Grease Pencil");
162         pt->draw= gpencil_panel_standard;
163         pt->poll= active_node_poll;
164         BLI_addtail(&art->paneltypes, pt);
165 }
166
167 static int node_properties(bContext *C, wmOperator *UNUSED(op))
168 {
169         ScrArea *sa= CTX_wm_area(C);
170         ARegion *ar= node_has_buttons_region(sa);
171         
172         if (ar)
173                 ED_region_toggle_hidden(C, ar);
174
175         return OPERATOR_FINISHED;
176 }
177
178 /* non-standard poll operator which doesn't care if there are any nodes */
179 static int node_properties_poll(bContext *C)
180 {
181         ScrArea *sa= CTX_wm_area(C);
182         return (sa && (sa->spacetype == SPACE_NODE));
183 }
184
185 void NODE_OT_properties(wmOperatorType *ot)
186 {
187         ot->name = "Properties";
188         ot->description = "Toggles the properties panel display";
189         ot->idname = "NODE_OT_properties";
190         
191         ot->exec = node_properties;
192         ot->poll = node_properties_poll;
193         
194         /* flags */
195         ot->flag = 0;
196 }