fix [#33501] Grease pencil in OpenGL render
[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 #include "MEM_guardedalloc.h"
32
33 #include "DNA_node_types.h"
34
35 #include "BLI_math.h"
36 #include "BLI_blenlib.h"
37
38 #include "BLF_translation.h"
39
40 #include "BKE_context.h"
41 #include "BKE_global.h"
42 #include "BKE_node.h"
43 #include "BKE_screen.h"
44
45 #include "WM_api.h"
46 #include "WM_types.h"
47
48 #include "RNA_access.h"
49
50 #include "ED_gpencil.h"
51 #include "ED_screen.h"
52
53 #include "UI_resources.h"
54
55 #include "node_intern.h"  /* own include */
56
57
58 /* ******************* node space & buttons ************** */
59
60 /* poll for active nodetree */
61 static int active_nodetree_poll(const bContext *C, PanelType *UNUSED(pt))
62 {
63         SpaceNode *snode = CTX_wm_space_node(C);
64         
65         return (snode && snode->nodetree);
66 }
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         return (snode && snode->edittree && nodeGetActive(snode->edittree));
74 }
75
76 /* active node */
77 static void active_node_panel(const bContext *C, Panel *pa)
78 {
79         SpaceNode *snode = CTX_wm_space_node(C);
80         bNodeTree *ntree = (snode) ? snode->edittree : NULL;
81         bNode *node = (ntree) ? nodeGetActive(ntree) : NULL; // xxx... for editing group nodes
82         uiLayout *layout, *row, *col, *sub;
83         PointerRNA ptr, opptr;
84         
85         /* verify pointers, and create RNA pointer for the node */
86         if (ELEM(NULL, ntree, node))
87                 return;
88         //if (node->id) /* for group nodes */
89         //      RNA_pointer_create(node->id, &RNA_Node, node, &ptr);
90         //else
91         RNA_pointer_create(&ntree->id, &RNA_Node, node, &ptr);
92         
93         layout = uiLayoutColumn(pa->layout, FALSE);
94         uiLayoutSetContextPointer(layout, "node", &ptr);
95         
96         /* draw this node's name, etc. */
97         uiItemR(layout, &ptr, "label", 0, NULL, ICON_NODE);
98         uiItemS(layout);
99         uiItemR(layout, &ptr, "name", 0, NULL, ICON_NODE);
100         uiItemS(layout);
101         
102         uiItemO(layout, NULL, 0, "NODE_OT_hide_socket_toggle");
103         uiItemS(layout);
104         uiItemS(layout);
105
106         row = uiLayoutRow(layout, FALSE);
107         
108         col = uiLayoutColumn(row, TRUE);
109         uiItemM(col, (bContext *)C, "NODE_MT_node_color_presets", NULL, 0);
110         uiItemR(col, &ptr, "use_custom_color", UI_ITEM_R_ICON_ONLY, NULL, ICON_NONE);
111         sub = uiLayoutRow(col, FALSE);
112         if (!(node->flag & NODE_CUSTOM_COLOR))
113                 uiLayoutSetEnabled(sub, FALSE);
114         uiItemR(sub, &ptr, "color", 0, "", ICON_NONE);
115         
116         col = uiLayoutColumn(row, TRUE);
117         uiItemO(col, "", ICON_ZOOMIN, "node.node_color_preset_add");
118         opptr = uiItemFullO(col, "node.node_color_preset_add", "", ICON_ZOOMOUT, NULL, WM_OP_INVOKE_DEFAULT, UI_ITEM_O_RETURN_PROPS);
119         RNA_boolean_set(&opptr, "remove_active", 1);
120         uiItemM(col, (bContext *)C, "NODE_MT_node_color_specials", "", ICON_DOWNARROW_HLT);
121         
122         /* draw this node's settings */
123         if (node->typeinfo && node->typeinfo->uifuncbut) {
124                 uiItemS(layout);
125                 uiItemS(layout);
126                 node->typeinfo->uifuncbut(layout, (bContext *)C, &ptr);
127         }
128         else if (node->typeinfo && node->typeinfo->uifunc) {
129                 uiItemS(layout);
130                 uiItemS(layout);
131                 node->typeinfo->uifunc(layout, (bContext *)C, &ptr);
132         }
133 }
134
135 static int node_sockets_poll(const bContext *C, PanelType *UNUSED(pt))
136 {
137         SpaceNode *snode = CTX_wm_space_node(C);
138         
139         return (snode && snode->nodetree && G.debug_value == 777);
140 }
141
142 static void node_sockets_panel(const bContext *C, Panel *pa)
143 {
144         SpaceNode *snode = CTX_wm_space_node(C);
145         bNodeTree *ntree = (snode) ? snode->edittree : NULL;
146         bNode *node = (ntree) ? nodeGetActive(ntree) : NULL;
147         bNodeSocket *sock;
148         uiLayout *layout = pa->layout, *split;
149         char name[UI_MAX_NAME_STR];
150         
151         if (ELEM(NULL, ntree, node))
152                 return;
153         
154         for (sock = node->inputs.first; sock; sock = sock->next) {
155                 BLI_snprintf(name, sizeof(name), "%s:", sock->name);
156
157                 split = uiLayoutSplit(layout, 0.35f, FALSE);
158                 uiItemL(split, name, ICON_NONE);
159                 uiTemplateNodeLink(split, ntree, node, sock);
160         }
161 }
162
163 /* ******************* node buttons registration ************** */
164
165 void node_buttons_register(ARegionType *art)
166 {
167         PanelType *pt;
168         
169         pt = MEM_callocN(sizeof(PanelType), "spacetype node panel active node");
170         strcpy(pt->idname, "NODE_PT_item");
171         strcpy(pt->label, IFACE_("Active Node"));
172         pt->draw = active_node_panel;
173         pt->poll = active_node_poll;
174         BLI_addtail(&art->paneltypes, pt);
175
176         pt = MEM_callocN(sizeof(PanelType), "spacetype node panel node sockets");
177         strcpy(pt->idname, "NODE_PT_sockets");
178         strcpy(pt->label, "Sockets");
179         pt->draw = node_sockets_panel;
180         pt->poll = node_sockets_poll;
181         pt->flag |= PNL_DEFAULT_CLOSED;
182         BLI_addtail(&art->paneltypes, pt);
183         
184         pt = MEM_callocN(sizeof(PanelType), "spacetype node panel gpencil");
185         strcpy(pt->idname, "NODE_PT_gpencil");
186         strcpy(pt->label, "Grease Pencil");
187         pt->draw_header = gpencil_panel_standard_header;
188         pt->draw = gpencil_panel_standard;
189         pt->poll = active_nodetree_poll;
190         BLI_addtail(&art->paneltypes, pt);
191 }
192
193 static int node_properties(bContext *C, wmOperator *UNUSED(op))
194 {
195         ScrArea *sa = CTX_wm_area(C);
196         ARegion *ar = node_has_buttons_region(sa);
197         
198         if (ar)
199                 ED_region_toggle_hidden(C, ar);
200
201         return OPERATOR_FINISHED;
202 }
203
204 /* non-standard poll operator which doesn't care if there are any nodes */
205 static int node_properties_poll(bContext *C)
206 {
207         ScrArea *sa = CTX_wm_area(C);
208         return (sa && (sa->spacetype == SPACE_NODE));
209 }
210
211 void NODE_OT_properties(wmOperatorType *ot)
212 {
213         ot->name = "Properties";
214         ot->description = "Toggles the properties panel display";
215         ot->idname = "NODE_OT_properties";
216         
217         ot->exec = node_properties;
218         ot->poll = node_properties_poll;
219         
220         /* flags */
221         ot->flag = 0;
222 }