utf8 OSX - disabling utf8 at KeyUp, otherwise TextObject doesn't work.
[blender.git] / 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, 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 /** \file blender/editors/space_node/node_header.c
30  *  \ingroup spnode
31  */
32
33
34 #include <string.h>
35 #include <stdio.h>
36
37 #include "DNA_space_types.h"
38 #include "DNA_node_types.h"
39 #include "DNA_scene_types.h"
40 #include "DNA_screen_types.h"
41
42 #include "MEM_guardedalloc.h"
43
44 #include "BLI_blenlib.h"
45 #include "BLI_utildefines.h"
46
47 #include "BLF_translation.h"
48
49 #include "BKE_context.h"
50 #include "BKE_global.h"
51 #include "BKE_screen.h"
52 #include "BKE_node.h"
53 #include "BKE_main.h"
54
55 #include "RNA_access.h"
56
57 #include "WM_api.h"
58 #include "WM_types.h"
59
60 #include "UI_interface.h"
61 #include "UI_interface_icons.h"
62 #include "UI_resources.h"
63 #include "UI_view2d.h"
64
65 #include "node_intern.h"
66
67 /* ************************ add menu *********************** */
68
69 static void do_node_add(bContext *C, bNodeTemplate *ntemp)
70 {
71         Main *bmain= CTX_data_main(C);
72         Scene *scene= CTX_data_scene(C);
73         SpaceNode *snode= CTX_wm_space_node(C);
74         ScrArea *sa= CTX_wm_area(C);
75         ARegion *ar;
76         bNode *node;
77         
78         /* get location to add node at mouse */
79         for(ar=sa->regionbase.first; ar; ar=ar->next) {
80                 if(ar->regiontype == RGN_TYPE_WINDOW) {
81                         wmWindow *win= CTX_wm_window(C);
82                         int x= win->eventstate->x - ar->winrct.xmin;
83                         int y= win->eventstate->y - ar->winrct.ymin;
84                         
85                         if(y < 60) y+= 60;
86                         UI_view2d_region_to_view(&ar->v2d, x, y, &snode->mx, &snode->my);
87                 }
88         }
89         
90         /* store selection in temp test flag */
91         for(node= snode->edittree->nodes.first; node; node= node->next) {
92                 if(node->flag & NODE_SELECT) node->flag |= NODE_TEST;
93                 else node->flag &= ~NODE_TEST;
94         }
95         
96         /* node= */ node_add_node(snode, bmain, scene, ntemp, snode->mx, snode->my);
97         
98         /* select previous selection before autoconnect */
99         for(node= snode->edittree->nodes.first; node; node= node->next) {
100                 if(node->flag & NODE_TEST) node->flag |= NODE_SELECT;
101         }
102         
103         /* deselect after autoconnection */
104         for(node= snode->edittree->nodes.first; node; node= node->next) {
105                 if(node->flag & NODE_TEST) node->flag &= ~NODE_SELECT;
106         }
107                 
108         snode_notify(C, snode);
109         snode_dag_update(C, snode);
110 }
111
112 static void do_node_add_static(bContext *C, void *UNUSED(arg), int event)
113 {
114         bNodeTemplate ntemp;
115         ntemp.type = event;
116         do_node_add(C, &ntemp);
117 }
118
119 static void do_node_add_group(bContext *C, void *UNUSED(arg), int event)
120 {
121         SpaceNode *snode= CTX_wm_space_node(C);
122         bNodeTemplate ntemp;
123         
124         if (event>=0) {
125                 ntemp.ngroup= BLI_findlink(&G.main->nodetree, event);
126                 ntemp.type = ntemp.ngroup->nodetype;
127         }
128         else {
129                 ntemp.type = -event;
130                 switch (ntemp.type) {
131                 case NODE_GROUP:
132                         ntemp.ngroup = ntreeAddTree("Group", snode->treetype, ntemp.type);
133                         break;
134                 case NODE_FORLOOP:
135                         ntemp.ngroup = ntreeAddTree("For Loop", snode->treetype, ntemp.type);
136                         break;
137                 case NODE_WHILELOOP:
138                         ntemp.ngroup = ntreeAddTree("While Loop", snode->treetype, ntemp.type);
139                         break;
140                 default:
141                         ntemp.ngroup = NULL;
142                 }
143         }
144         if (!ntemp.ngroup)
145                 return;
146         
147         do_node_add(C, &ntemp);
148 }
149
150 #if 0 /* disabled */
151 static void do_node_add_dynamic(bContext *C, void *UNUSED(arg), int event)
152 {
153         bNodeTemplate ntemp;
154         ntemp.type = NODE_DYNAMIC;
155         do_node_add(C, &ntemp);
156 }
157 #endif
158
159 static int node_tree_has_type(int treetype, int nodetype)
160 {
161         bNodeTreeType *ttype= ntreeGetType(treetype);
162         bNodeType *ntype;
163         for (ntype=ttype->node_types.first; ntype; ntype=ntype->next) {
164                 if (ntype->type==nodetype)
165                         return 1;
166         }
167         return 0;
168 }
169
170 static void node_add_menu(bContext *C, uiLayout *layout, void *arg_nodeclass)
171 {
172         Main *bmain= CTX_data_main(C);
173         SpaceNode *snode= CTX_wm_space_node(C);
174         bNodeTree *ntree;
175         int nodeclass= GET_INT_FROM_POINTER(arg_nodeclass);
176         int event;
177         
178         ntree = snode->nodetree;
179         
180         if(!ntree) {
181                 uiItemS(layout);
182                 return;
183         }
184         
185         if (nodeclass==NODE_CLASS_GROUP) {
186                 bNodeTree *ngroup;
187                 
188                 uiLayoutSetFunc(layout, do_node_add_group, NULL);
189                 
190                 /* XXX hack: negative numbers used for empty group types */
191                 if (node_tree_has_type(ntree->type, NODE_GROUP))
192                         uiItemV(layout, "New Group", 0, -NODE_GROUP);
193                 if (node_tree_has_type(ntree->type, NODE_FORLOOP))
194                         uiItemV(layout, "New For Loop", 0, -NODE_FORLOOP);
195                 if (node_tree_has_type(ntree->type, NODE_WHILELOOP))
196                         uiItemV(layout, "New While Loop", 0, -NODE_WHILELOOP);
197                 uiItemS(layout);
198                 
199                 for(ngroup=bmain->nodetree.first, event=0; ngroup; ngroup= ngroup->id.next, ++event) {
200                         /* only use group trees */
201                         if (ngroup->type==ntree->type && ELEM3(ngroup->nodetype, NODE_GROUP, NODE_FORLOOP, NODE_WHILELOOP)) {
202                                 uiItemV(layout, ngroup->id.name+2, 0, event);
203                         }
204                 }
205         }
206         else if (nodeclass==NODE_DYNAMIC) {
207                 /* disabled */
208         }
209         else {
210                 bNodeType *ntype;
211                 
212                 uiLayoutSetFunc(layout, do_node_add_static, NULL);
213                 
214                 for (ntype=ntreeGetType(ntree->type)->node_types.first; ntype; ntype=ntype->next) {
215                         if(ntype->nclass==nodeclass && ntype->name)
216                                 uiItemV(layout, ntype->name, 0, ntype->type);
217                 }
218         }
219 }
220
221 static void node_menu_add(const bContext *C, Menu *menu)
222 {
223         SpaceNode *snode= CTX_wm_space_node(C);
224         uiLayout *layout= menu->layout;
225
226         if(!snode->nodetree)
227                 uiLayoutSetActive(layout, 0);
228
229         if(snode->treetype==NTREE_SHADER) {
230                 uiItemMenuF(layout, IFACE_("Input"), 0, node_add_menu, SET_INT_IN_POINTER(NODE_CLASS_INPUT));
231                 uiItemMenuF(layout, IFACE_("Output"), 0, node_add_menu, SET_INT_IN_POINTER(NODE_CLASS_OUTPUT));
232                 uiItemMenuF(layout, IFACE_("Color"), 0, node_add_menu, SET_INT_IN_POINTER(NODE_CLASS_OP_COLOR));
233                 uiItemMenuF(layout, IFACE_("Vector"), 0, node_add_menu, SET_INT_IN_POINTER(NODE_CLASS_OP_VECTOR));
234                 uiItemMenuF(layout, IFACE_("Convertor"), 0, node_add_menu, SET_INT_IN_POINTER(NODE_CLASS_CONVERTOR));
235                 uiItemMenuF(layout, IFACE_("Group"), 0, node_add_menu, SET_INT_IN_POINTER(NODE_CLASS_GROUP));
236                 uiItemMenuF(layout, IFACE_("Dynamic"), 0, node_add_menu, SET_INT_IN_POINTER(NODE_CLASS_OP_DYNAMIC));
237                 uiItemMenuF(layout, IFACE_("Layout"), 0, node_add_menu, SET_INT_IN_POINTER(NODE_CLASS_LAYOUT));
238         }
239         else if(snode->treetype==NTREE_COMPOSIT) {
240                 uiItemMenuF(layout, IFACE_("Input"), 0, node_add_menu, SET_INT_IN_POINTER(NODE_CLASS_INPUT));
241                 uiItemMenuF(layout, IFACE_("Output"), 0, node_add_menu, SET_INT_IN_POINTER(NODE_CLASS_OUTPUT));
242                 uiItemMenuF(layout, IFACE_("Color"), 0, node_add_menu, SET_INT_IN_POINTER(NODE_CLASS_OP_COLOR));
243                 uiItemMenuF(layout, IFACE_("Vector"), 0, node_add_menu, SET_INT_IN_POINTER(NODE_CLASS_OP_VECTOR));
244                 uiItemMenuF(layout, IFACE_("Filter"), 0, node_add_menu, SET_INT_IN_POINTER(NODE_CLASS_OP_FILTER));
245                 uiItemMenuF(layout, IFACE_("Convertor"), 0, node_add_menu, SET_INT_IN_POINTER(NODE_CLASS_CONVERTOR));
246                 uiItemMenuF(layout, IFACE_("Matte"), 0, node_add_menu, SET_INT_IN_POINTER(NODE_CLASS_MATTE));
247                 uiItemMenuF(layout, IFACE_("Distort"), 0, node_add_menu, SET_INT_IN_POINTER(NODE_CLASS_DISTORT));
248                 uiItemMenuF(layout, IFACE_("Group"), 0, node_add_menu, SET_INT_IN_POINTER(NODE_CLASS_GROUP));
249                 uiItemMenuF(layout, IFACE_("Layout"), 0, node_add_menu, SET_INT_IN_POINTER(NODE_CLASS_LAYOUT));
250         }
251         else if(snode->treetype==NTREE_TEXTURE) {
252                 uiItemMenuF(layout, IFACE_("Input"), 0, node_add_menu, SET_INT_IN_POINTER(NODE_CLASS_INPUT));
253                 uiItemMenuF(layout, IFACE_("Output"), 0, node_add_menu, SET_INT_IN_POINTER(NODE_CLASS_OUTPUT));
254                 uiItemMenuF(layout, IFACE_("Color"), 0, node_add_menu, SET_INT_IN_POINTER(NODE_CLASS_OP_COLOR));
255                 uiItemMenuF(layout, IFACE_("Patterns"), 0, node_add_menu, SET_INT_IN_POINTER(NODE_CLASS_PATTERN));
256                 uiItemMenuF(layout, IFACE_("Textures"), 0, node_add_menu, SET_INT_IN_POINTER(NODE_CLASS_TEXTURE));
257                 uiItemMenuF(layout, IFACE_("Convertor"), 0, node_add_menu, SET_INT_IN_POINTER(NODE_CLASS_CONVERTOR));
258                 uiItemMenuF(layout, IFACE_("Distort"), 0, node_add_menu, SET_INT_IN_POINTER(NODE_CLASS_DISTORT));
259                 uiItemMenuF(layout, IFACE_("Group"), 0, node_add_menu, SET_INT_IN_POINTER(NODE_CLASS_GROUP));
260                 uiItemMenuF(layout, IFACE_("Layout"), 0, node_add_menu, SET_INT_IN_POINTER(NODE_CLASS_LAYOUT));
261         }
262 }
263
264 void node_menus_register(void)
265 {
266         MenuType *mt;
267
268         mt= MEM_callocN(sizeof(MenuType), "spacetype node menu add");
269         strcpy(mt->idname, "NODE_MT_add");
270         strcpy(mt->label, "Add");
271         mt->draw= node_menu_add;
272         WM_menutype_add(mt);
273 }
274