5a8bc5da324cbeffc3afafec8389c61606e48f3d
[blender.git] / source / blender / editors / space_node / drawnode.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) 2005 Blender Foundation.
19  * All rights reserved.
20  *
21  * The Original Code is: all of this file.
22  *
23  * Contributor(s): David Millan Escriva, Juho Vepsäläinen, Bob Holcomb, Thomas Dinges
24  *
25  * ***** END GPL LICENSE BLOCK *****
26  */
27
28 /** \file blender/editors/space_node/drawnode.c
29  *  \ingroup spnode
30  */
31
32
33 #include <math.h>
34 #include <stdio.h>
35 #include <string.h>
36
37 #include "BLI_blenlib.h"
38 #include "BLI_math.h"
39 #include "BLI_utildefines.h"
40
41 #include "DNA_node_types.h"
42 #include "DNA_material_types.h"
43 #include "DNA_object_types.h"
44 #include "DNA_scene_types.h"
45 #include "DNA_space_types.h"
46 #include "DNA_screen_types.h"
47
48 #include "BKE_context.h"
49 #include "BKE_curve.h"
50 #include "BKE_global.h"
51 #include "BKE_image.h"
52 #include "BKE_library.h"
53 #include "BKE_main.h"
54 #include "BKE_node.h"
55
56 #include "BLF_api.h"
57 #include "BLF_translation.h"
58
59 #include "NOD_composite.h"
60 #include "NOD_shader.h"
61
62 #include "BIF_gl.h"
63 #include "BIF_glutil.h"
64
65 #include "BLF_api.h"
66
67 #include "MEM_guardedalloc.h"
68
69
70 #include "RNA_access.h"
71
72 #include "ED_node.h"
73
74 #include "WM_api.h"
75 #include "WM_types.h"
76
77 #include "UI_interface.h"
78 #include "UI_resources.h"
79
80 #include "IMB_imbuf.h"
81 #include "IMB_imbuf_types.h"
82
83 #include "node_intern.h"
84
85 // XXX interface.h
86 extern void ui_dropshadow(rctf *rct, float radius, float aspect, float alpha, int select);
87
88 /* ****************** SOCKET BUTTON DRAW FUNCTIONS ***************** */
89
90 static void node_sync_cb(bContext *UNUSED(C), void *snode_v, void *node_v)
91 {
92         SpaceNode *snode = snode_v;
93         
94         if (snode->treetype == NTREE_SHADER) {
95                 nodeShaderSynchronizeID(node_v, 1);
96                 // allqueue(REDRAWBUTSSHADING, 0);
97         }
98 }
99
100 static void node_socket_button_label(const bContext *UNUSED(C), uiBlock *block,
101                                      bNodeTree *UNUSED(ntree), bNode *UNUSED(node), bNodeSocket *sock,
102                                      const char *UNUSED(name), int x, int y, int width)
103 {
104         uiDefBut(block, LABEL, 0, IFACE_(sock->name), x, y, width, NODE_DY, NULL, 0, 0, 0, 0, "");
105 }
106
107 static void node_socket_button_default(const bContext *C, uiBlock *block,
108                                        bNodeTree *ntree, bNode *node, bNodeSocket *sock,
109                                        const char *name, int x, int y, int width)
110 {
111         if (sock->link || (sock->flag & SOCK_HIDE_VALUE))
112                 node_socket_button_label(C, block, ntree, node, sock, name, x, y, width);
113         else {
114                 PointerRNA ptr;
115                 uiBut *bt;
116                 
117                 RNA_pointer_create(&ntree->id, &RNA_NodeSocket, sock, &ptr);
118                 
119                 bt = uiDefButR(block, NUM, B_NODE_EXEC, IFACE_(name),
120                                x, y + 1, width, NODE_DY - 2,
121                                &ptr, "default_value", 0, 0, 0, -1, -1, NULL);
122                 if (node)
123                         uiButSetFunc(bt, node_sync_cb, CTX_wm_space_node(C), node);
124         }
125 }
126
127 typedef struct SocketComponentMenuArgs {
128         PointerRNA ptr;
129         int x, y, width;
130         uiButHandleFunc cb;
131         void *arg1, *arg2;
132 } SocketComponentMenuArgs;
133 /* NOTE: this is a block-menu, needs 0 events, otherwise the menu closes */
134 static uiBlock *socket_component_menu(bContext *C, ARegion *ar, void *args_v)
135 {
136         SocketComponentMenuArgs *args = (SocketComponentMenuArgs *)args_v;
137         uiBlock *block;
138         uiLayout *layout;
139         
140         block = uiBeginBlock(C, ar, __func__, UI_EMBOSS);
141         uiBlockSetFlag(block, UI_BLOCK_KEEP_OPEN);
142         
143         layout = uiLayoutColumn(uiBlockLayout(block, UI_LAYOUT_VERTICAL, UI_LAYOUT_PANEL, args->x, args->y + 2, args->width, NODE_DY, UI_GetStyle()), 0);
144         
145         uiItemR(layout, &args->ptr, "default_value", UI_ITEM_R_EXPAND, "", ICON_NONE);
146         
147         return block;
148 }
149 static void node_socket_button_components(const bContext *C, uiBlock *block,
150                                           bNodeTree *ntree, bNode *node, bNodeSocket *sock,
151                                           const char *name, int x, int y, int width)
152 {
153         if (sock->link || (sock->flag & SOCK_HIDE_VALUE))
154                 node_socket_button_label(C, block, ntree, node, sock, name, x, y, width);
155         else {
156                 PointerRNA ptr;
157                 SocketComponentMenuArgs *args;
158                 
159                 RNA_pointer_create(&ntree->id, &RNA_NodeSocket, sock, &ptr);
160                 
161                 args = MEM_callocN(sizeof(SocketComponentMenuArgs), "SocketComponentMenuArgs");
162                 
163                 args->ptr = ptr;
164                 args->x = x;
165                 args->y = y;
166                 args->width = width;
167                 args->cb = node_sync_cb;
168                 args->arg1 = CTX_wm_space_node(C);
169                 args->arg2 = node;
170                 
171                 uiDefBlockButN(block, socket_component_menu, args, IFACE_(name), x, y + 1, width, NODE_DY - 2, "");
172         }
173 }
174
175 static void node_socket_button_color(const bContext *C, uiBlock *block,
176                                      bNodeTree *ntree, bNode *node, bNodeSocket *sock,
177                                      const char *name, int x, int y, int width)
178 {
179         if (sock->link || (sock->flag & SOCK_HIDE_VALUE))
180                 node_socket_button_label(C, block, ntree, node, sock, IFACE_(name), x, y, width);
181         else {
182                 PointerRNA ptr;
183                 uiBut *bt;
184                 int labelw = width - 40;
185                 RNA_pointer_create(&ntree->id, &RNA_NodeSocket, sock, &ptr);
186                 
187                 bt = uiDefButR(block, COL, B_NODE_EXEC, "",
188                                x, y + 2, (labelw > 0 ? 40 : width), NODE_DY - 2,
189                                &ptr, "default_value", 0, 0, 0, -1, -1, NULL);
190                 if (node)
191                         uiButSetFunc(bt, node_sync_cb, CTX_wm_space_node(C), node);
192                 
193                 if (name[0] != '\0' && labelw > 0)
194                         uiDefBut(block, LABEL, 0, IFACE_(name), x + 40, y + 2, labelw, NODE_DY - 2, NULL, 0, 0, 0, 0, "");
195         }
196 }
197
198 /* standard draw function, display the default input value */
199 static void node_draw_input_default(const bContext *C, uiBlock *block,
200                                     bNodeTree *ntree, bNode *node, bNodeSocket *sock,
201                                     const char *name, int x, int y, int width)
202 {
203         bNodeSocketType *stype = ntreeGetSocketType(sock->type);
204         if (stype->buttonfunc)
205                 stype->buttonfunc(C, block, ntree, node, sock, name, x, y, width);
206         else
207                 node_socket_button_label(C, block, ntree, node, sock, IFACE_(name), x, y, width);
208 }
209
210 static void node_draw_output_default(const bContext *C, uiBlock *block,
211                                      bNodeTree *UNUSED(ntree), bNode *node, bNodeSocket *sock,
212                                      const char *name, int UNUSED(x), int UNUSED(y), int UNUSED(width))
213 {
214         SpaceNode *snode = CTX_wm_space_node(C);
215         float slen;
216         int ofs = 0;
217         const char *ui_name = IFACE_(name);
218         UI_ThemeColor(TH_TEXT);
219         slen = snode->aspect * UI_GetStringWidth(ui_name);
220         while (slen > node->width) {
221                 ofs++;
222                 slen = snode->aspect * UI_GetStringWidth(ui_name + ofs);
223         }
224         uiDefBut(block, LABEL, 0, ui_name + ofs, (short)(sock->locx - 15.0f - slen), (short)(sock->locy - 9.0f),
225                  (short)(node->width - NODE_DY), NODE_DY,  NULL, 0, 0, 0, 0, "");
226 }
227
228 /* ****************** BASE DRAW FUNCTIONS FOR NEW OPERATOR NODES ***************** */
229
230 #if 0 /* UNUSED */
231 static void node_draw_socket_new(bNodeSocket *sock, float size)
232 {
233         float x = sock->locx, y = sock->locy;
234         
235         /* 16 values of sin function */
236         static float si[16] = {
237                 0.00000000f, 0.39435585f, 0.72479278f, 0.93775213f,
238                 0.99871650f, 0.89780453f, 0.65137248f, 0.29936312f,
239                 -0.10116832f, -0.48530196f, -0.79077573f, -0.96807711f,
240                 -0.98846832f, -0.84864425f, -0.57126821f, -0.20129852f
241         };
242         /* 16 values of cos function */
243         static float co[16] = {
244                 1.00000000f, 0.91895781f, 0.68896691f, 0.34730525f,
245                 -0.05064916f, -0.44039415f, -0.75875812f, -0.95413925f,
246                 -0.99486932f, -0.87434661f, -0.61210598f, -0.25065253f,
247                 0.15142777f, 0.52896401f, 0.82076344f, 0.97952994f,
248         };
249         int a;
250         
251         glColor3ub(180, 180, 180);
252         
253         glBegin(GL_POLYGON);
254         for (a = 0; a < 16; a++)
255                 glVertex2f(x + size * si[a], y + size * co[a]);
256         glEnd();
257         
258         glColor4ub(0, 0, 0, 150);
259         glEnable(GL_BLEND);
260         glEnable(GL_LINE_SMOOTH);
261         glBegin(GL_LINE_LOOP);
262         for (a = 0; a < 16; a++)
263                 glVertex2f(x + size * si[a], y + size * co[a]);
264         glEnd();
265         glDisable(GL_LINE_SMOOTH);
266         glDisable(GL_BLEND);
267 }
268 #endif
269
270 /* ****************** BUTTON CALLBACKS FOR ALL TREES ***************** */
271
272 static void node_buts_value(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
273 {
274         PointerRNA sockptr;
275         PropertyRNA *prop;
276         
277         /* first socket stores value */
278         prop = RNA_struct_find_property(ptr, "outputs");
279         RNA_property_collection_lookup_int(ptr, prop, 0, &sockptr);
280         
281         uiItemR(layout, &sockptr, "default_value", 0, "", ICON_NONE);
282 }
283
284 static void node_buts_rgb(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
285 {
286         uiLayout *col;
287         PointerRNA sockptr;
288         PropertyRNA *prop;
289         
290         /* first socket stores value */
291         prop = RNA_struct_find_property(ptr, "outputs");
292         RNA_property_collection_lookup_int(ptr, prop, 0, &sockptr);
293         
294         col = uiLayoutColumn(layout, 0);
295         uiTemplateColorWheel(col, &sockptr, "default_value", 1, 0, 0, 0);
296         uiItemR(col, &sockptr, "default_value", 0, "", ICON_NONE);
297 }
298
299 static void node_buts_mix_rgb(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
300 {       
301         uiLayout *row;
302
303         bNodeTree *ntree = (bNodeTree *)ptr->id.data;
304
305         row = uiLayoutRow(layout, 1);
306         uiItemR(row, ptr, "blend_type", 0, "", ICON_NONE);
307         if (ntree->type == NTREE_COMPOSIT)
308                 uiItemR(row, ptr, "use_alpha", 0, "", ICON_IMAGE_RGB_ALPHA);
309 }
310
311 static void node_buts_time(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
312 {
313         uiLayout *row;
314 #if 0
315         /* XXX no context access here .. */
316         bNode *node = ptr->data;
317         CurveMapping *cumap = node->storage;
318         
319         if (cumap) {
320                 cumap->flag |= CUMA_DRAW_CFRA;
321                 if (node->custom1 < node->custom2)
322                         cumap->sample[0] = (float)(CFRA - node->custom1) / (float)(node->custom2 - node->custom1);
323         }
324 #endif
325
326         uiTemplateCurveMapping(layout, ptr, "curve", 's', 0, 0);
327
328         row = uiLayoutRow(layout, 1);
329         uiItemR(row, ptr, "frame_start", 0, IFACE_("Sta"), ICON_NONE);
330         uiItemR(row, ptr, "frame_end", 0, IFACE_("End"), ICON_NONE);
331 }
332
333 static void node_buts_colorramp(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
334 {
335         uiTemplateColorRamp(layout, ptr, "color_ramp", 0);
336 }
337
338 static void node_buts_curvevec(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
339 {
340         uiTemplateCurveMapping(layout, ptr, "mapping", 'v', 0, 0);
341 }
342
343 static float *_sample_col = NULL;    // bad bad, 2.5 will do better?
344 #if 0
345 static void node_curvemap_sample(float *col)
346 {
347         _sample_col = col;
348 }
349 #endif
350
351 static void node_buts_curvecol(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
352 {
353         bNode *node = ptr->data;
354         CurveMapping *cumap = node->storage;
355
356         if (_sample_col) {
357                 cumap->flag |= CUMA_DRAW_SAMPLE;
358                 copy_v3_v3(cumap->sample, _sample_col);
359         }
360         else 
361                 cumap->flag &= ~CUMA_DRAW_SAMPLE;
362
363         uiTemplateCurveMapping(layout, ptr, "mapping", 'c', 0, 0);
364 }
365
366 static void node_normal_cb(bContext *C, void *ntree_v, void *node_v)
367 {
368         Main *bmain = CTX_data_main(C);
369
370         ED_node_generic_update(bmain, ntree_v, node_v);
371         WM_event_add_notifier(C, NC_NODE | NA_EDITED, ntree_v);
372 }
373
374 static void node_buts_normal(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
375 {
376         uiBlock *block = uiLayoutAbsoluteBlock(layout);
377         bNodeTree *ntree = ptr->id.data;
378         bNode *node = ptr->data;
379         rctf *butr = &node->butr;
380         bNodeSocket *sock = node->outputs.first;     /* first socket stores normal */
381         float *nor = ((bNodeSocketValueVector *)sock->default_value)->value;
382         uiBut *bt;
383         
384         bt = uiDefButF(block, BUT_NORMAL, B_NODE_EXEC, "",
385                        (short)butr->xmin, (short)butr->xmin,
386                        butr->xmax - butr->xmin, butr->xmax - butr->xmin,
387                        nor, 0.0f, 1.0f, 0, 0, "");
388         uiButSetFunc(bt, node_normal_cb, ntree, node);
389 }
390 #if 0 // not used in 2.5x yet
391 static void node_browse_tex_cb(bContext *C, void *ntree_v, void *node_v)
392 {
393         Main *bmain = CTX_data_main(C);
394         bNodeTree *ntree = ntree_v;
395         bNode *node = node_v;
396         Tex *tex;
397         
398         if (node->menunr < 1) return;
399         
400         if (node->id) {
401                 node->id->us--;
402                 node->id = NULL;
403         }
404         tex = BLI_findlink(&bmain->tex, node->menunr - 1);
405
406         node->id = &tex->id;
407         id_us_plus(node->id);
408         BLI_strncpy(node->name, node->id->name + 2, sizeof(node->name));
409         
410         nodeSetActive(ntree, node);
411         
412         if (ntree->type == NTREE_TEXTURE)
413                 ntreeTexCheckCyclics(ntree);
414         
415         // allqueue(REDRAWBUTSSHADING, 0);
416         // allqueue(REDRAWNODE, 0);
417         NodeTagChanged(ntree, node); 
418         
419         node->menunr = 0;
420 }
421 #endif
422
423 static void node_buts_texture(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
424 {
425         bNode *node = ptr->data;
426
427         short multi = (
428             node->id &&
429             ((Tex *)node->id)->use_nodes &&
430             (node->type != CMP_NODE_TEXTURE) &&
431             (node->type != TEX_NODE_TEXTURE)
432             );
433         
434         uiItemR(layout, ptr, "texture", 0, "", ICON_NONE);
435         
436         if (multi) {
437                 /* Number Drawing not optimal here, better have a list*/
438                 uiItemR(layout, ptr, "node_output", 0, "", ICON_NONE);
439         }
440 }
441
442 static void node_buts_math(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
443
444         uiItemR(layout, ptr, "operation", 0, "", ICON_NONE);
445 }
446
447 static int node_resize_area_default(bNode *node, int x, int y)
448 {
449         if (node->flag & NODE_HIDDEN) {
450                 rctf totr = node->totr;
451                 /* right part of node */
452                 totr.xmin = node->totr.xmax - 20.0f;
453                 if (BLI_in_rctf(&totr, x, y))
454                         return NODE_RESIZE_RIGHT;
455                 else
456                         return 0;
457         }
458         else {
459                 const float size = 10.0f;
460                 rctf totr = node->totr;
461                 int dir = 0;
462                 
463                 if (x >= totr.xmax - size && x < totr.xmax && y >= totr.ymin && y < totr.ymax)
464                         dir |= NODE_RESIZE_RIGHT;
465                 if (x >= totr.xmin && x < totr.xmin + size && y >= totr.ymin && y < totr.ymax)
466                         dir |= NODE_RESIZE_LEFT;
467                 return dir;
468         }
469 }
470
471 /* ****************** BUTTON CALLBACKS FOR COMMON NODES ***************** */
472
473 /* width of socket columns in group display */
474 #define NODE_GROUP_FRAME        120
475
476 /* based on settings in node, sets drawing rect info. each redraw! */
477 /* note: this assumes only 1 group at a time is drawn (linked data) */
478 /* in node->totr the entire boundbox for the group is stored */
479 static void node_update_group(const bContext *C, bNodeTree *ntree, bNode *gnode)
480 {
481         if (!(gnode->flag & NODE_GROUP_EDIT)) {
482                 node_update_default(C, ntree, gnode);
483         }
484         else {
485                 bNodeTree *ngroup = (bNodeTree *)gnode->id;
486                 bNode *node;
487                 bNodeSocket *sock, *gsock;
488                 float locx, locy;
489                 rctf *rect = &gnode->totr;
490                 float node_group_frame = U.dpi * NODE_GROUP_FRAME / 72;
491                 float group_header = 26 * U.dpi / 72;
492                 int counter;
493                 int dy;
494                 
495                 /* get "global" coords */
496                 nodeToView(gnode, 0.0f, 0.0f, &locx, &locy);
497                 
498                 /* center them, is a bit of abuse of locx and locy though */
499                 node_update_nodetree(C, ngroup, locx, locy);
500                 
501                 rect->xmin = rect->xmax = locx;
502                 rect->ymin = rect->ymax = locy;
503                 
504                 counter = 1;
505                 for (node = ngroup->nodes.first; node; node = node->next) {
506                         if (counter) {
507                                 *rect = node->totr;
508                                 counter = 0;
509                         }
510                         else
511                                 BLI_union_rctf(rect, &node->totr);
512                 }
513                 
514                 /* add some room for links to group sockets */
515                 rect->xmin -= 4 * NODE_DY;
516                 rect->xmax += 4 * NODE_DY;
517                 rect->ymin -= NODE_DY;
518                 rect->ymax += NODE_DY;
519                 
520                 /* input sockets */
521                 dy = 0.5f * (rect->ymin + rect->ymax) + NODE_DY * (BLI_countlist(&gnode->inputs) - 1);
522                 gsock = ngroup->inputs.first;
523                 sock = gnode->inputs.first;
524                 while (gsock || sock) {
525                         while (sock && !sock->groupsock) {
526                                 sock->locx = rect->xmin - node_group_frame;
527                                 sock->locy = dy;
528
529                                 /* prevent long socket lists from growing out of the group box */
530                                 if (dy - 3 * NODE_DYS < rect->ymin)
531                                         rect->ymin = dy - 3 * NODE_DYS;
532                                 if (dy + 3 * NODE_DYS > rect->ymax)
533                                         rect->ymax = dy + 3 * NODE_DYS;
534                                 dy -= 2 * NODE_DY;
535                                 
536                                 sock = sock->next;
537                         }
538                         while (gsock && (!sock || sock->groupsock != gsock)) {
539                                 gsock->locx = rect->xmin;
540                                 gsock->locy = dy;
541                                 
542                                 /* prevent long socket lists from growing out of the group box */
543                                 if (dy - 3 * NODE_DYS < rect->ymin)
544                                         rect->ymin = dy - 3 * NODE_DYS;
545                                 if (dy + 3 * NODE_DYS > rect->ymax)
546                                         rect->ymax = dy + 3 * NODE_DYS;
547                                 dy -= 2 * NODE_DY;
548                                 
549                                 gsock = gsock->next;
550                         }
551                         while (sock && gsock && sock->groupsock == gsock) {
552                                 gsock->locx = rect->xmin;
553                                 sock->locx = rect->xmin - node_group_frame;
554                                 sock->locy = gsock->locy = dy;
555                                 
556                                 /* prevent long socket lists from growing out of the group box */
557                                 if (dy - 3 * NODE_DYS < rect->ymin)
558                                         rect->ymin = dy - 3 * NODE_DYS;
559                                 if (dy + 3 * NODE_DYS > rect->ymax)
560                                         rect->ymax = dy + 3 * NODE_DYS;
561                                 dy -= 2 * NODE_DY;
562                                 
563                                 sock = sock->next;
564                                 gsock = gsock->next;
565                         }
566                 }
567                 
568                 /* output sockets */
569                 dy = 0.5f * (rect->ymin + rect->ymax) + NODE_DY * (BLI_countlist(&gnode->outputs) - 1);
570                 gsock = ngroup->outputs.first;
571                 sock = gnode->outputs.first;
572                 while (gsock || sock) {
573                         while (sock && !sock->groupsock) {
574                                 sock->locx = rect->xmax + node_group_frame;
575                                 sock->locy = dy - NODE_DYS;
576                                 
577                                 /* prevent long socket lists from growing out of the group box */
578                                 if (dy - 3 * NODE_DYS < rect->ymin)
579                                         rect->ymin = dy - 3 * NODE_DYS;
580                                 if (dy + 3 * NODE_DYS > rect->ymax)
581                                         rect->ymax = dy + 3 * NODE_DYS;
582                                 dy -= 2 * NODE_DY;
583                                 
584                                 sock = sock->next;
585                         }
586                         while (gsock && (!sock || sock->groupsock != gsock)) {
587                                 gsock->locx = rect->xmax;
588                                 gsock->locy = dy - NODE_DYS;
589                                 
590                                 /* prevent long socket lists from growing out of the group box */
591                                 if (dy - 3 * NODE_DYS < rect->ymin)
592                                         rect->ymin = dy - 3 * NODE_DYS;
593                                 if (dy + 3 * NODE_DYS > rect->ymax)
594                                         rect->ymax = dy + 3 * NODE_DYS;
595                                 dy -= 2 * NODE_DY;
596                                 
597                                 gsock = gsock->next;
598                         }
599                         while (sock && gsock && sock->groupsock == gsock) {
600                                 gsock->locx = rect->xmax;
601                                 sock->locx = rect->xmax + node_group_frame;
602                                 sock->locy = gsock->locy = dy - NODE_DYS;
603                                 
604                                 /* prevent long socket lists from growing out of the group box */
605                                 if (dy - 3 * NODE_DYS < rect->ymin)
606                                         rect->ymin = dy - 3 * NODE_DYS;
607                                 if (dy + 3 * NODE_DYS > rect->ymax)
608                                         rect->ymax = dy + 3 * NODE_DYS;
609                                 dy -= 2 * NODE_DY;
610                                 
611                                 sock = sock->next;
612                                 gsock = gsock->next;
613                         }
614                 }
615                 
616                 /* Set the block bounds to clip mouse events from underlying nodes.
617                  * Add margin for header and input/output columns.
618                  */
619                 uiExplicitBoundsBlock(gnode->block,
620                                       rect->xmin - node_group_frame,
621                                       rect->ymin,
622                                       rect->xmax + node_group_frame,
623                                       rect->ymax + group_header);
624         }
625 }
626
627 static void update_group_input_cb(bContext *UNUSED(C), void *UNUSED(snode_v), void *ngroup_v)
628 {
629         bNodeTree *ngroup = (bNodeTree *)ngroup_v;
630         
631         ngroup->update |= NTREE_UPDATE_GROUP_IN;
632         ntreeUpdateTree(ngroup);
633 }
634
635 static void update_group_output_cb(bContext *UNUSED(C), void *UNUSED(snode_v), void *ngroup_v)
636 {
637         bNodeTree *ngroup = (bNodeTree *)ngroup_v;
638         
639         ngroup->update |= NTREE_UPDATE_GROUP_OUT;
640         ntreeUpdateTree(ngroup);
641 }
642
643 static void draw_group_socket_name(SpaceNode *snode, bNode *gnode, bNodeSocket *sock, int in_out, float xoffset, float yoffset)
644 {
645         bNodeTree *ngroup = (bNodeTree *)gnode->id;
646         uiBut *bt;
647         const char *ui_name = IFACE_(sock->name);
648         
649         if (sock->flag & SOCK_DYNAMIC) {
650                 bt = uiDefBut(gnode->block, TEX, 0, "", 
651                               sock->locx + xoffset, sock->locy + 1 + yoffset, 72, NODE_DY,
652                               sock->name, 0, sizeof(sock->name), 0, 0, "");
653                 if (in_out == SOCK_IN)
654                         uiButSetFunc(bt, update_group_input_cb, snode, ngroup);
655                 else
656                         uiButSetFunc(bt, update_group_output_cb, snode, ngroup);
657         }
658         else {
659                 uiDefBut(gnode->block, LABEL, 0, ui_name,
660                          sock->locx + xoffset, sock->locy + 1 + yoffset, 72, NODE_DY,
661                          NULL, 0, sizeof(ui_name), 0, 0, "");
662         }
663 }
664
665 static void draw_group_socket(const bContext *C, SpaceNode *snode, bNodeTree *ntree, bNode *gnode, bNodeSocket *sock, bNodeSocket *gsock, int index, int in_out)
666 {
667         bNodeTree *ngroup = (bNodeTree *)gnode->id;
668         bNodeSocketType *stype = ntreeGetSocketType(gsock ? gsock->type : sock->type);
669         uiBut *bt;
670         float offset;
671         int draw_value;
672         float node_group_frame = U.dpi * NODE_GROUP_FRAME / 72;
673         float socket_size = NODE_SOCKSIZE * U.dpi / 72;
674         float arrowbutw = 0.8f * UI_UNIT_X;
675         /* layout stuff for buttons on group left frame */
676         float colw = 0.6f * node_group_frame;
677         float col1 = 6 - node_group_frame;
678         float col2 = col1 + colw + 6;
679         float col3 = -arrowbutw - 6;
680         /* layout stuff for buttons on group right frame */
681         float cor1 = 6;
682         float cor2 = cor1 + arrowbutw + 6;
683         float cor3 = cor2 + arrowbutw + 6;
684         
685         /* node and group socket circles */
686         if (sock)
687                 node_socket_circle_draw(ntree, sock, socket_size, sock->flag & SELECT);
688         if (gsock)
689                 node_socket_circle_draw(ngroup, gsock, socket_size, gsock->flag & SELECT);
690         
691         /* socket name */
692         offset = (in_out == SOCK_IN ? col1 : cor3);
693         if (!gsock)
694                 offset += (in_out == SOCK_IN ? node_group_frame : -node_group_frame);
695         
696         /* draw both name and value button if:
697          * 1) input: not internal
698          * 2) output: (node type uses const outputs) and (group output is unlinked)
699          */
700         draw_value = 0;
701         switch (in_out) {
702                 case SOCK_IN:
703                         draw_value = !(gsock && (gsock->flag & SOCK_INTERNAL));
704                         break;
705                 case SOCK_OUT:
706                         if (gnode->typeinfo->flag & NODE_CONST_OUTPUT)
707                                 draw_value = !(gsock && gsock->link);
708                         break;
709         }
710         if (draw_value) {
711                 /* both name and value buttons */
712                 if (gsock) {
713                         draw_group_socket_name(snode, gnode, gsock, in_out, offset, 0);
714                         if (stype->buttonfunc)
715                                 stype->buttonfunc(C, gnode->block, ngroup, NULL, gsock, "", gsock->locx + offset, gsock->locy - NODE_DY, colw);
716                 }
717                 else {
718                         draw_group_socket_name(snode, gnode, sock, in_out, offset, 0);
719                         if (stype->buttonfunc)
720                                 stype->buttonfunc(C, gnode->block, ngroup, NULL, sock, "", sock->locx + offset, sock->locy - NODE_DY, colw);
721                 }
722         }
723         else {
724                 /* only name, no value button */
725                 if (gsock)
726                         draw_group_socket_name(snode, gnode, gsock, in_out, offset, -NODE_DYS);
727                 else
728                         draw_group_socket_name(snode, gnode, sock, in_out, offset, -NODE_DYS);
729         }
730         
731         if (gsock && (gsock->flag & SOCK_DYNAMIC)) {
732                 /* up/down buttons */
733                 offset = (in_out == SOCK_IN ? col2 : cor2);
734                 uiBlockSetDirection(gnode->block, UI_TOP);
735                 uiBlockBeginAlign(gnode->block);
736                 bt = uiDefIconButO(gnode->block, BUT, "NODE_OT_group_socket_move_up", 0, ICON_TRIA_UP,
737                                    gsock->locx + offset, gsock->locy, arrowbutw, arrowbutw, "");
738                 if (!gsock->prev || !(gsock->prev->flag & SOCK_DYNAMIC))
739                         uiButSetFlag(bt, UI_BUT_DISABLED);
740                 RNA_int_set(uiButGetOperatorPtrRNA(bt), "index", index);
741                 RNA_enum_set(uiButGetOperatorPtrRNA(bt), "in_out", in_out);
742                 bt = uiDefIconButO(gnode->block, BUT, "NODE_OT_group_socket_move_down", 0, ICON_TRIA_DOWN,
743                                    gsock->locx + offset, gsock->locy - arrowbutw, arrowbutw, arrowbutw, "");
744                 if (!gsock->next || !(gsock->next->flag & SOCK_DYNAMIC))
745                         uiButSetFlag(bt, UI_BUT_DISABLED);
746                 RNA_int_set(uiButGetOperatorPtrRNA(bt), "index", index);
747                 RNA_enum_set(uiButGetOperatorPtrRNA(bt), "in_out", in_out);
748                 uiBlockEndAlign(gnode->block);
749                 uiBlockSetDirection(gnode->block, 0);
750                 
751                 /* remove button */
752                 offset = (in_out == SOCK_IN ? col3 : cor1);
753                 uiBlockSetEmboss(gnode->block, UI_EMBOSSN);
754                 bt = uiDefIconButO(gnode->block, BUT, "NODE_OT_group_socket_remove", 0, ICON_X,
755                                    gsock->locx + offset, gsock->locy - 0.5f * arrowbutw, arrowbutw, arrowbutw, "");
756                 RNA_int_set(uiButGetOperatorPtrRNA(bt), "index", index);
757                 RNA_enum_set(uiButGetOperatorPtrRNA(bt), "in_out", in_out);
758                 uiBlockSetEmboss(gnode->block, UI_EMBOSS);
759         }
760 }
761
762 /* groups are, on creation, centered around 0,0 */
763 static void node_draw_group(const bContext *C, ARegion *ar, SpaceNode *snode, bNodeTree *ntree, bNode *gnode)
764 {
765         if (!(gnode->flag & NODE_GROUP_EDIT)) {
766                 node_draw_default(C, ar, snode, ntree, gnode);
767         }
768         else {
769                 bNodeTree *ngroup = (bNodeTree *)gnode->id;
770                 bNodeSocket *sock, *gsock;
771                 uiLayout *layout;
772                 PointerRNA ptr;
773                 rctf rect = gnode->totr;
774                 float node_group_frame = U.dpi * NODE_GROUP_FRAME / 72;
775                 float group_header = 26 * U.dpi / 72;
776                 
777                 int index;
778                 
779                 /* backdrop header */
780                 glEnable(GL_BLEND);
781                 uiSetRoundBox(UI_CNR_TOP_LEFT | UI_CNR_TOP_RIGHT);
782                 UI_ThemeColorShadeAlpha(TH_NODE_GROUP, 0, -70);
783                 uiDrawBox(GL_POLYGON, rect.xmin - node_group_frame, rect.ymax, rect.xmax + node_group_frame, rect.ymax + group_header, BASIS_RAD);
784                 
785                 /* backdrop body */
786                 UI_ThemeColorShadeAlpha(TH_BACK, -8, -70);
787                 uiSetRoundBox(UI_CNR_NONE);
788                 uiDrawBox(GL_POLYGON, rect.xmin, rect.ymin, rect.xmax, rect.ymax, BASIS_RAD);
789         
790                 /* input column */
791                 UI_ThemeColorShadeAlpha(TH_BACK, 10, -50);
792                 uiSetRoundBox(UI_CNR_BOTTOM_LEFT);
793                 uiDrawBox(GL_POLYGON, rect.xmin - node_group_frame, rect.ymin, rect.xmin, rect.ymax, BASIS_RAD);
794         
795                 /* output column */
796                 UI_ThemeColorShadeAlpha(TH_BACK, 10, -50);
797                 uiSetRoundBox(UI_CNR_BOTTOM_RIGHT);
798                 uiDrawBox(GL_POLYGON, rect.xmax, rect.ymin, rect.xmax + node_group_frame, rect.ymax, BASIS_RAD);
799         
800                 /* input column separator */
801                 glColor4ub(200, 200, 200, 140);
802                 glBegin(GL_LINES);
803                 glVertex2f(rect.xmin, rect.ymin);
804                 glVertex2f(rect.xmin, rect.ymax);
805                 glEnd();
806         
807                 /* output column separator */
808                 glColor4ub(200, 200, 200, 140);
809                 glBegin(GL_LINES);
810                 glVertex2f(rect.xmax, rect.ymin);
811                 glVertex2f(rect.xmax, rect.ymax);
812                 glEnd();
813         
814                 /* group node outline */
815                 uiSetRoundBox(UI_CNR_ALL);
816                 glColor4ub(200, 200, 200, 140);
817                 glEnable(GL_LINE_SMOOTH);
818                 uiDrawBox(GL_LINE_LOOP, rect.xmin - node_group_frame, rect.ymin, rect.xmax + node_group_frame, rect.ymax + group_header, BASIS_RAD);
819                 glDisable(GL_LINE_SMOOTH);
820                 glDisable(GL_BLEND);
821                 
822                 /* backdrop title */
823                 UI_ThemeColor(TH_TEXT_HI);
824         
825                 layout = uiBlockLayout(gnode->block, UI_LAYOUT_VERTICAL, UI_LAYOUT_PANEL, (short)(rect.xmin + 15), (short)(rect.ymax + group_header),
826                                        MIN2((int)(rect.xmax - rect.xmin - 18.0f), node_group_frame + 20), group_header, UI_GetStyle());
827                 RNA_pointer_create(&ntree->id, &RNA_Node, gnode, &ptr);
828                 uiTemplateIDBrowse(layout, (bContext *)C, &ptr, "node_tree", NULL, NULL, NULL);
829                 uiBlockLayoutResolve(gnode->block, NULL, NULL);
830         
831                 /* draw the internal tree nodes and links */
832                 node_draw_nodetree(C, ar, snode, ngroup);
833         
834                 /* group sockets */
835                 gsock = ngroup->inputs.first;
836                 sock = gnode->inputs.first;
837                 index = 0;
838                 while (gsock || sock) {
839                         while (sock && !sock->groupsock) {
840                                 draw_group_socket(C, snode, ntree, gnode, sock, NULL, index, SOCK_IN);
841                                 sock = sock->next;
842                         }
843                         while (gsock && (!sock || sock->groupsock != gsock)) {
844                                 draw_group_socket(C, snode, ntree, gnode, NULL, gsock, index, SOCK_IN);
845                                 gsock = gsock->next;
846                                 ++index;
847                         }
848                         while (sock && gsock && sock->groupsock == gsock) {
849                                 draw_group_socket(C, snode, ntree, gnode, sock, gsock, index, SOCK_IN);
850                                 sock = sock->next;
851                                 gsock = gsock->next;
852                                 ++index;
853                         }
854                 }
855                 gsock = ngroup->outputs.first;
856                 sock = gnode->outputs.first;
857                 index = 0;
858                 while (gsock || sock) {
859                         while (sock && !sock->groupsock) {
860                                 draw_group_socket(C, snode, ntree, gnode, sock, NULL, index, SOCK_OUT);
861                                 sock = sock->next;
862                         }
863                         while (gsock && (!sock || sock->groupsock != gsock)) {
864                                 draw_group_socket(C, snode, ntree, gnode, NULL, gsock, index, SOCK_OUT);
865                                 gsock = gsock->next;
866                                 ++index;
867                         }
868                         while (sock && gsock && sock->groupsock == gsock) {
869                                 draw_group_socket(C, snode, ntree, gnode, sock, gsock, index, SOCK_OUT);
870                                 sock = sock->next;
871                                 gsock = gsock->next;
872                                 ++index;
873                         }
874                 }
875                 
876                 uiEndBlock(C, gnode->block);
877                 uiDrawBlock(C, gnode->block);
878                 gnode->block = NULL;
879         }
880 }
881
882 void node_uifunc_group(uiLayout *layout, bContext *C, PointerRNA *ptr)
883 {
884         uiTemplateIDBrowse(layout, C, ptr, "node_tree", NULL, NULL, NULL);
885 }
886
887 static void node_common_buts_whileloop(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
888 {
889         uiItemR(layout, ptr, "max_iterations", 0, NULL, 0);
890 }
891
892 /* XXX Does a bounding box update by iterating over all children.
893  * Not ideal to do this in every draw call, but doing as transform callback doesn't work,
894  * since the child node totr rects are not updated properly at that point.
895  */
896 static void node_update_frame(const bContext *UNUSED(C), bNodeTree *ntree, bNode *node)
897 {
898         const float margin = 30.0f;
899         NodeFrame *data = (NodeFrame *)node->storage;
900         int bbinit;
901         bNode *tnode;
902         rctf rect, noderect;
903         float xmax, ymax;
904         
905         /* init rect from current frame size */
906         nodeToView(node, node->offsetx, node->offsety, &rect.xmin, &rect.ymax);
907         nodeToView(node, node->offsetx + node->width, node->offsety - node->height, &rect.xmax, &rect.ymin);
908         
909         /* frame can be resized manually only if shrinking is disabled or no children are attached */
910         data->flag |= NODE_FRAME_RESIZEABLE;
911         /* for shrinking bbox, initialize the rect from first child node */
912         bbinit = (data->flag & NODE_FRAME_SHRINK);
913         /* fit bounding box to all children */
914         for (tnode = ntree->nodes.first; tnode; tnode = tnode->next) {
915                 if (tnode->parent != node)
916                         continue;
917                 
918                 /* add margin to node rect */
919                 noderect = tnode->totr;
920                 noderect.xmin -= margin;
921                 noderect.xmax += margin;
922                 noderect.ymin -= margin;
923                 noderect.ymax += margin;
924                 
925                 /* first child initializes frame */
926                 if (bbinit) {
927                         bbinit = 0;
928                         rect = noderect;
929                         data->flag &= ~NODE_FRAME_RESIZEABLE;
930                 }
931                 else
932                         BLI_union_rctf(&rect, &noderect);
933         }
934         
935         /* now adjust the frame size from view-space bounding box */
936         nodeFromView(node, rect.xmin, rect.ymax, &node->offsetx, &node->offsety);
937         nodeFromView(node, rect.xmax, rect.ymin, &xmax, &ymax);
938         node->width = xmax - node->offsetx;
939         node->height = -ymax + node->offsety;
940         
941         node->totr = rect;
942 }
943
944 static void node_draw_frame_label(bNode *node)
945 {
946         /* XXX font id is crap design */
947         const int fontid = blf_mono_font;
948         NodeFrame *data = (NodeFrame *)node->storage;
949         rctf *rct = &node->totr;
950         int color_id = node_get_colorid(node);
951         char label[128];
952         /* XXX a bit hacky, should use separate align values for x and y */
953         float width, ascender;
954         float x, y;
955         
956         BLI_strncpy(label, nodeLabel(node), sizeof(label));
957         BLF_size(fontid, data->label_size, U.dpi);
958         
959         /* title color */
960         UI_ThemeColorBlendShade(TH_TEXT, color_id, 0.8f, 10);
961
962         width = BLF_width(fontid, label);
963         ascender = BLF_ascender(fontid);
964         
965         x = 0.5f * (rct->xmin + rct->xmax) - 0.5f * width;
966         y = rct->ymax - NODE_DYS - ascender;
967         
968         BLF_position(fontid, x, y, 0);
969         BLF_draw(fontid, label, BLF_DRAW_STR_DUMMY_MAX);
970 }
971
972 static void node_draw_frame(const bContext *C, ARegion *ar, SpaceNode *snode, bNodeTree *UNUSED(ntree), bNode *node)
973 {
974         rctf *rct = &node->totr;
975         int color_id = node_get_colorid(node);
976         unsigned char color[4];
977         float alpha;
978         
979         UI_GetThemeColor4ubv(TH_NODE_FRAME, color);
980         alpha = (float)(color[3])/255.0f;
981         
982         /* skip if out of view */
983         if (node->totr.xmax < ar->v2d.cur.xmin || node->totr.xmin > ar->v2d.cur.xmax ||
984             node->totr.ymax < ar->v2d.cur.ymin || node->totr.ymin > ar->v2d.cur.ymax) {
985                 
986                 uiEndBlock(C, node->block);
987                 node->block = NULL;
988                 return;
989         }
990         
991         /* shadow */
992         node_draw_shadow(snode, node, BASIS_RAD, alpha);
993         
994         /* body */
995         if (node->flag & NODE_CUSTOM_COLOR)
996                 glColor4f(node->color[0], node->color[1], node->color[2], alpha);
997         else
998                 UI_ThemeColor4(TH_NODE_FRAME);
999         glEnable(GL_BLEND);
1000         uiSetRoundBox(UI_CNR_ALL);
1001         uiRoundBox(rct->xmin, rct->ymin, rct->xmax, rct->ymax, BASIS_RAD);
1002         glDisable(GL_BLEND);
1003
1004         /* outline active and selected emphasis */
1005         if (node->flag & (NODE_ACTIVE | SELECT) ) {
1006                 glEnable(GL_BLEND);
1007                 glEnable(GL_LINE_SMOOTH);
1008                 
1009                 if (node->flag & NODE_ACTIVE)
1010                         UI_ThemeColorShadeAlpha(TH_ACTIVE, 0, -40);
1011                 else
1012                         UI_ThemeColorShadeAlpha(TH_SELECT, 0, -40);
1013                 uiSetRoundBox(UI_CNR_ALL);
1014                 uiDrawBox(GL_LINE_LOOP, rct->xmin, rct->ymin, rct->xmax, rct->ymax, BASIS_RAD);
1015                 
1016                 glDisable(GL_LINE_SMOOTH);
1017                 glDisable(GL_BLEND);
1018         }
1019         
1020         /* label */
1021         node_draw_frame_label(node);
1022         
1023         UI_ThemeClearColor(color_id);
1024                 
1025         uiEndBlock(C, node->block);
1026         uiDrawBlock(C, node->block);
1027         node->block = NULL;
1028 }
1029
1030 static int node_resize_area_frame(bNode *node, int x, int y)
1031 {
1032         const float size = 10.0f;
1033         NodeFrame *data = (NodeFrame *)node->storage;
1034         rctf totr = node->totr;
1035         int dir = 0;
1036         
1037         /* shrinking frame size is determined by child nodes */
1038         if (!(data->flag & NODE_FRAME_RESIZEABLE))
1039                 return 0;
1040         
1041         if (x >= totr.xmax - size && x < totr.xmax && y >= totr.ymin && y < totr.ymax)
1042                 dir |= NODE_RESIZE_RIGHT;
1043         if (x >= totr.xmin && x < totr.xmin + size && y >= totr.ymin && y < totr.ymax)
1044                 dir |= NODE_RESIZE_LEFT;
1045         if (x >= totr.xmin && x < totr.xmax && y >= totr.ymax - size && y < totr.ymax)
1046                 dir |= NODE_RESIZE_TOP;
1047         if (x >= totr.xmin && x < totr.xmax && y >= totr.ymin && y < totr.ymin + size)
1048                 dir |= NODE_RESIZE_BOTTOM;
1049         
1050         return dir;
1051 }
1052
1053 static void node_buts_frame_details(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1054 {
1055         uiItemR(layout, ptr, "label_size", 0, IFACE_("Label Size"), ICON_NONE);
1056         uiItemR(layout, ptr, "shrink", 0, IFACE_("Shrink"), ICON_NONE);
1057 }
1058
1059
1060 #define NODE_REROUTE_SIZE       8.0f
1061
1062 static void node_update_reroute(const bContext *UNUSED(C), bNodeTree *UNUSED(ntree), bNode *node)
1063 {
1064         bNodeSocket *nsock;
1065         float locx, locy;
1066         float size = NODE_REROUTE_SIZE;
1067         
1068         /* get "global" coords */
1069         nodeToView(node, 0.0f, 0.0f, &locx, &locy);
1070         
1071         /* reroute node has exactly one input and one output, both in the same place */
1072         nsock= node->outputs.first;
1073         nsock->locx= locx;
1074         nsock->locy= locy;
1075
1076         nsock= node->inputs.first;
1077         nsock->locx= locx;
1078         nsock->locy= locy;
1079         
1080         node->width = size*2;
1081         node->totr.xmin= locx - size;
1082         node->totr.xmax= locx + size;
1083         node->totr.ymax= locy + size;
1084         node->totr.ymin= locy - size;
1085 }
1086
1087 static void node_draw_reroute(const bContext *C, ARegion *ar, SpaceNode *UNUSED(snode),  bNodeTree *ntree, bNode *node)
1088 {
1089         bNodeSocket *sock;
1090         #if 0   /* UNUSED */
1091         rctf *rct= &node->totr;
1092         float size = NODE_REROUTE_SIZE;
1093         #endif
1094         float socket_size= NODE_SOCKSIZE;
1095
1096         /* skip if out of view */
1097         if (node->totr.xmax < ar->v2d.cur.xmin || node->totr.xmin > ar->v2d.cur.xmax ||
1098                         node->totr.ymax < ar->v2d.cur.ymin || node->totr.ymin > ar->v2d.cur.ymax) {
1099
1100                 uiEndBlock(C, node->block);
1101                 node->block= NULL;
1102                 return;
1103         }
1104
1105         /* XXX only kept for debugging
1106          * selection state is indicated by socket outline below!
1107          */
1108         #if 0
1109         /* body */
1110         uiSetRoundBox(15);
1111         UI_ThemeColor4(TH_NODE);
1112         glEnable(GL_BLEND);
1113         uiRoundBox(rct->xmin, rct->ymin, rct->xmax, rct->ymax, size);
1114         glDisable(GL_BLEND);
1115
1116         /* outline active and selected emphasis */
1117         if( node->flag & (NODE_ACTIVE|SELECT) ) {
1118                 glEnable(GL_BLEND);
1119                 glEnable( GL_LINE_SMOOTH );
1120                         /* using different shades of TH_TEXT_HI for the empasis, like triangle */
1121                         if( node->flag & NODE_ACTIVE )
1122                                 UI_ThemeColorShadeAlpha(TH_TEXT_HI, 0, -40);
1123                         else
1124                                 UI_ThemeColorShadeAlpha(TH_TEXT_HI, -20, -120);
1125                         uiDrawBox(GL_LINE_LOOP, rct->xmin, rct->ymin, rct->xmax, rct->ymax, size);
1126
1127                 glDisable( GL_LINE_SMOOTH );
1128                 glDisable(GL_BLEND);
1129         }
1130         #endif
1131
1132         /* only draw input socket. as they all are placed on the same position.
1133          * highlight also if node itself is selected, since we don't display the node body separately!
1134          */
1135         for(sock= node->inputs.first; sock; sock= sock->next) {
1136                 node_socket_circle_draw(ntree, sock, socket_size, (sock->flag & SELECT) || (node->flag & SELECT));
1137         }
1138
1139         uiEndBlock(C, node->block);
1140         uiDrawBlock(C, node->block);
1141         node->block= NULL;
1142 }
1143
1144 /* Special tweak area for reroute node.
1145  * Since this node is quite small, we use a larger tweak area for grabbing than for selection.
1146  */
1147 static int node_tweak_area_reroute(bNode *node, int x, int y)
1148 {
1149         /* square of tweak radius */
1150         static const float tweak_radius_sq = 576;       /* 24*24 */
1151         
1152         bNodeSocket *sock = node->inputs.first;
1153         float dx = sock->locx - x;
1154         float dy = sock->locy - y;
1155         return (dx*dx + dy*dy <= tweak_radius_sq);
1156 }
1157
1158 static void node_common_set_butfunc(bNodeType *ntype)
1159 {
1160         switch (ntype->type) {
1161                 case NODE_GROUP:
1162                         ntype->uifunc = node_uifunc_group;
1163                         ntype->drawfunc = node_draw_group;
1164                         ntype->drawupdatefunc = node_update_group;
1165                         break;
1166                 case NODE_FORLOOP:
1167 //                      ntype->uifunc= node_common_buts_group;
1168                         ntype->drawfunc = node_draw_group;
1169                         ntype->drawupdatefunc = node_update_group;
1170                         break;
1171                 case NODE_WHILELOOP:
1172                         ntype->uifunc = node_common_buts_whileloop;
1173                         ntype->drawfunc = node_draw_group;
1174                         ntype->drawupdatefunc = node_update_group;
1175                         break;
1176                 case NODE_FRAME:
1177                         ntype->drawfunc = node_draw_frame;
1178                         ntype->drawupdatefunc = node_update_frame;
1179                         ntype->uifuncbut = node_buts_frame_details;
1180                         ntype->resize_area_func = node_resize_area_frame;
1181                         break;
1182                 case NODE_REROUTE:
1183                         ntype->drawfunc= node_draw_reroute;
1184                         ntype->drawupdatefunc= node_update_reroute;
1185                         ntype->tweak_area_func= node_tweak_area_reroute;
1186                         break;
1187         }
1188 }
1189
1190 /* ****************** BUTTON CALLBACKS FOR SHADER NODES ***************** */
1191 static void node_shader_buts_material(uiLayout *layout, bContext *C, PointerRNA *ptr)
1192 {
1193         bNode *node = ptr->data;
1194         uiLayout *col;
1195         
1196         uiTemplateID(layout, C, ptr, "material", "MATERIAL_OT_new", NULL, NULL);
1197         
1198         if (!node->id) return;
1199         
1200         col = uiLayoutColumn(layout, 0);
1201         uiItemR(col, ptr, "use_diffuse", 0, NULL, ICON_NONE);
1202         uiItemR(col, ptr, "use_specular", 0, NULL, ICON_NONE);
1203         uiItemR(col, ptr, "invert_normal", 0, NULL, ICON_NONE);
1204 }
1205
1206 static void node_shader_buts_mapping(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1207 {
1208         uiLayout *row;
1209         
1210         uiItemL(layout, IFACE_("Location:"), ICON_NONE);
1211         row = uiLayoutRow(layout, 1);
1212         uiItemR(row, ptr, "translation", 0, "", ICON_NONE);
1213         
1214         uiItemL(layout, IFACE_("Rotation:"), ICON_NONE);
1215         row = uiLayoutRow(layout, 1);
1216         uiItemR(row, ptr, "rotation", 0, "", ICON_NONE);
1217         
1218         uiItemL(layout, IFACE_("Scale:"), ICON_NONE);
1219         row = uiLayoutRow(layout, 1);
1220         uiItemR(row, ptr, "scale", 0, "", ICON_NONE);
1221         
1222         row = uiLayoutRow(layout, 1);
1223         uiItemR(row, ptr, "use_min", 0, "Min", ICON_NONE);
1224         uiItemR(row, ptr, "min", 0, "", ICON_NONE);
1225         
1226         row = uiLayoutRow(layout, 1);
1227         uiItemR(row, ptr, "use_max", 0, "Max", ICON_NONE);
1228         uiItemR(row, ptr, "max", 0, "", ICON_NONE);
1229 }
1230
1231 static void node_shader_buts_vect_math(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1232
1233         uiItemR(layout, ptr, "operation", 0, "", ICON_NONE);
1234 }
1235
1236 static void node_shader_buts_geometry(uiLayout *layout, bContext *C, PointerRNA *ptr)
1237 {
1238         PointerRNA obptr = CTX_data_pointer_get(C, "active_object");
1239         uiLayout *col;
1240
1241         col = uiLayoutColumn(layout, 0);
1242
1243         if (obptr.data && RNA_enum_get(&obptr, "type") == OB_MESH) {
1244                 PointerRNA dataptr = RNA_pointer_get(&obptr, "data");
1245
1246                 uiItemPointerR(col, ptr, "uv_layer", &dataptr, "uv_textures", "", ICON_NONE);
1247                 uiItemPointerR(col, ptr, "color_layer", &dataptr, "vertex_colors", "", ICON_NONE);
1248         }
1249         else {
1250                 uiItemR(col, ptr, "uv_layer", 0, IFACE_("UV"), ICON_NONE);
1251                 uiItemR(col, ptr, "color_layer", 0, IFACE_("VCol"), ICON_NONE);
1252         }
1253 }
1254
1255 static void node_shader_buts_attribute(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1256 {
1257         uiItemR(layout, ptr, "attribute_name", 0, IFACE_("Name"), ICON_NONE);
1258 }
1259
1260 static void node_shader_buts_tex_image(uiLayout *layout, bContext *C, PointerRNA *ptr)
1261 {
1262         uiTemplateID(layout, C, ptr, "image", NULL, "IMAGE_OT_open", NULL);
1263         uiItemR(layout, ptr, "color_space", 0, "", ICON_NONE);
1264 }
1265
1266
1267 static void node_shader_buts_tex_environment(uiLayout *layout, bContext *C, PointerRNA *ptr)
1268 {
1269         uiTemplateID(layout, C, ptr, "image", NULL, "IMAGE_OT_open", NULL);
1270         uiItemR(layout, ptr, "color_space", 0, "", ICON_NONE);
1271         uiItemR(layout, ptr, "projection", 0, "", ICON_NONE);
1272 }
1273
1274 static void node_shader_buts_tex_sky(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1275 {
1276         uiItemR(layout, ptr, "sun_direction", 0, "", ICON_NONE);
1277         uiItemR(layout, ptr, "turbidity", 0, NULL, ICON_NONE);
1278 }
1279
1280 static void node_shader_buts_tex_gradient(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1281 {
1282         uiItemR(layout, ptr, "gradient_type", 0, "", ICON_NONE);
1283 }
1284
1285 static void node_shader_buts_tex_magic(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1286 {
1287         uiItemR(layout, ptr, "turbulence_depth", 0, NULL, ICON_NONE);
1288 }
1289
1290 static void node_shader_buts_tex_wave(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1291 {
1292         uiItemR(layout, ptr, "wave_type", 0, "", ICON_NONE);
1293 }
1294
1295 static void node_shader_buts_tex_musgrave(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1296 {
1297         uiItemR(layout, ptr, "musgrave_type", 0, "", ICON_NONE);
1298 }
1299
1300 static void node_shader_buts_tex_voronoi(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1301 {
1302         uiItemR(layout, ptr, "coloring", 0, "", ICON_NONE);
1303 }
1304
1305 static void node_shader_buts_glossy(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1306 {
1307         uiItemR(layout, ptr, "distribution", 0, "", ICON_NONE);
1308 }
1309
1310 /* only once called */
1311 static void node_shader_set_butfunc(bNodeType *ntype)
1312 {
1313         switch (ntype->type) {
1314                 /* case NODE_GROUP:      note, typeinfo for group is generated... see "XXX ugly hack" */
1315
1316                 case SH_NODE_MATERIAL:
1317                 case SH_NODE_MATERIAL_EXT:
1318                         ntype->uifunc = node_shader_buts_material;
1319                         break;
1320                 case SH_NODE_TEXTURE:
1321                         ntype->uifunc = node_buts_texture;
1322                         break;
1323                 case SH_NODE_NORMAL:
1324                         ntype->uifunc = node_buts_normal;
1325                         break;
1326                 case SH_NODE_CURVE_VEC:
1327                         ntype->uifunc = node_buts_curvevec;
1328                         break;
1329                 case SH_NODE_CURVE_RGB:
1330                         ntype->uifunc = node_buts_curvecol;
1331                         break;
1332                 case SH_NODE_MAPPING:
1333                         ntype->uifunc = node_shader_buts_mapping;
1334                         break;
1335                 case SH_NODE_VALUE:
1336                         ntype->uifunc = node_buts_value;
1337                         break;
1338                 case SH_NODE_RGB:
1339                         ntype->uifunc = node_buts_rgb;
1340                         break;
1341                 case SH_NODE_MIX_RGB:
1342                         ntype->uifunc = node_buts_mix_rgb;
1343                         break;
1344                 case SH_NODE_VALTORGB:
1345                         ntype->uifunc = node_buts_colorramp;
1346                         break;
1347                 case SH_NODE_MATH: 
1348                         ntype->uifunc = node_buts_math;
1349                         break; 
1350                 case SH_NODE_VECT_MATH: 
1351                         ntype->uifunc = node_shader_buts_vect_math;
1352                         break; 
1353                 case SH_NODE_GEOMETRY:
1354                         ntype->uifunc = node_shader_buts_geometry;
1355                         break;
1356                 case SH_NODE_ATTRIBUTE:
1357                         ntype->uifunc = node_shader_buts_attribute;
1358                         break;
1359                 case SH_NODE_TEX_SKY:
1360                         ntype->uifunc = node_shader_buts_tex_sky;
1361                         break;
1362                 case SH_NODE_TEX_IMAGE:
1363                         ntype->uifunc = node_shader_buts_tex_image;
1364                         break;
1365                 case SH_NODE_TEX_ENVIRONMENT:
1366                         ntype->uifunc = node_shader_buts_tex_environment;
1367                         break;
1368                 case SH_NODE_TEX_GRADIENT:
1369                         ntype->uifunc = node_shader_buts_tex_gradient;
1370                         break;
1371                 case SH_NODE_TEX_MAGIC:
1372                         ntype->uifunc = node_shader_buts_tex_magic;
1373                         break;
1374                 case SH_NODE_TEX_WAVE:
1375                         ntype->uifunc = node_shader_buts_tex_wave;
1376                         break;
1377                 case SH_NODE_TEX_MUSGRAVE:
1378                         ntype->uifunc = node_shader_buts_tex_musgrave;
1379                         break;
1380                 case SH_NODE_TEX_VORONOI:
1381                         ntype->uifunc = node_shader_buts_tex_voronoi;
1382                         break;
1383                 case SH_NODE_BSDF_GLOSSY:
1384                 case SH_NODE_BSDF_GLASS:
1385                         ntype->uifunc = node_shader_buts_glossy;
1386                         break;
1387         }
1388 }
1389
1390 /* ****************** BUTTON CALLBACKS FOR COMPOSITE NODES ***************** */
1391
1392 static void node_composit_buts_image(uiLayout *layout, bContext *C, PointerRNA *ptr)
1393 {
1394         uiLayout *col;
1395         bNode *node = ptr->data;
1396         PointerRNA imaptr;
1397         PropertyRNA *prop;
1398         int source;
1399         
1400         uiTemplateID(layout, C, ptr, "image", NULL, "IMAGE_OT_open", NULL);
1401         
1402         if (!node->id) return;
1403         
1404         prop = RNA_struct_find_property(ptr, "image");
1405         if (!prop || RNA_property_type(prop) != PROP_POINTER) return;
1406         imaptr = RNA_property_pointer_get(ptr, prop);
1407         
1408         col = uiLayoutColumn(layout, 0);
1409         
1410         uiItemR(col, &imaptr, "source", 0, NULL, ICON_NONE);
1411         
1412         source = RNA_enum_get(&imaptr, "source");
1413
1414         if (source == IMA_SRC_SEQUENCE) {
1415                 /* don't use iuser->framenr directly because it may not be updated if auto-refresh is off */
1416                 Scene *scene = CTX_data_scene(C);
1417                 ImageUser *iuser = node->storage;
1418                 char numstr[32];
1419                 const int framenr = BKE_image_user_frame_get(iuser, CFRA, 0);
1420                 BLI_snprintf(numstr, sizeof(numstr), IFACE_("Frame: %d"), framenr);
1421                 uiItemL(layout, numstr, ICON_NONE);
1422         }
1423
1424         if (ELEM(source, IMA_SRC_SEQUENCE, IMA_SRC_MOVIE)) {
1425                 col = uiLayoutColumn(layout, 1);
1426                 uiItemR(col, ptr, "frame_duration", 0, NULL, ICON_NONE);
1427                 uiItemR(col, ptr, "frame_start", 0, NULL, ICON_NONE);
1428                 uiItemR(col, ptr, "frame_offset", 0, NULL, ICON_NONE);
1429                 uiItemR(col, ptr, "use_cyclic", 0, NULL, ICON_NONE);
1430                 uiItemR(col, ptr, "use_auto_refresh", UI_ITEM_R_ICON_ONLY, NULL, ICON_NONE);
1431         }
1432
1433         col = uiLayoutColumn(layout, 0);
1434
1435         if (RNA_enum_get(&imaptr, "type") == IMA_TYPE_MULTILAYER)
1436                 uiItemR(col, ptr, "layer", 0, NULL, ICON_NONE);
1437 }
1438
1439 static void node_composit_buts_renderlayers(uiLayout *layout, bContext *C, PointerRNA *ptr)
1440 {
1441         bNode *node = ptr->data;
1442         uiLayout *col, *row;
1443         PointerRNA op_ptr;
1444         PointerRNA scn_ptr;
1445         PropertyRNA *prop;
1446         const char *layer_name;
1447         char scene_name[MAX_ID_NAME - 2];
1448         wmOperatorType *ot = WM_operatortype_find("RENDER_OT_render", 1);
1449
1450         BLI_assert(ot != 0);
1451
1452         uiTemplateID(layout, C, ptr, "scene", NULL, NULL, NULL);
1453         
1454         if (!node->id) return;
1455
1456         col = uiLayoutColumn(layout, 0);
1457         row = uiLayoutRow(col, 0);
1458         uiItemR(row, ptr, "layer", 0, "", ICON_NONE);
1459         
1460         prop = RNA_struct_find_property(ptr, "layer");
1461         if (!(RNA_property_enum_identifier(C, ptr, prop, RNA_property_enum_get(ptr, prop), &layer_name)))
1462                 return;
1463         
1464         scn_ptr = RNA_pointer_get(ptr, "scene");
1465         RNA_string_get(&scn_ptr, "name", scene_name);
1466         
1467         WM_operator_properties_create_ptr(&op_ptr, ot);
1468         RNA_string_set(&op_ptr, "layer", layer_name);
1469         RNA_string_set(&op_ptr, "scene", scene_name);
1470         uiItemFullO_ptr(row, ot, "", ICON_RENDER_STILL, op_ptr.data, WM_OP_INVOKE_DEFAULT, 0);
1471
1472 }
1473
1474
1475 static void node_composit_buts_blur(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1476 {
1477         uiLayout *col, *row;
1478         
1479         col = uiLayoutColumn(layout, 0);
1480         
1481         uiItemR(col, ptr, "filter_type", 0, "", ICON_NONE);
1482         if (RNA_enum_get(ptr, "filter_type") != R_FILTER_FAST_GAUSS) {
1483                 uiItemR(col, ptr, "use_bokeh", 0, NULL, ICON_NONE);
1484                 uiItemR(col, ptr, "use_gamma_correction", 0, NULL, ICON_NONE);
1485         }
1486         
1487         uiItemR(col, ptr, "use_relative", 0, NULL, ICON_NONE);
1488         
1489         if (RNA_boolean_get(ptr, "use_relative")) {
1490                 uiItemL(col, IFACE_("Aspect Correction"), 0);
1491                 row = uiLayoutRow(layout, 1);
1492                 uiItemR(row, ptr, "aspect_correction", UI_ITEM_R_EXPAND, NULL, 0);
1493                 
1494                 col = uiLayoutColumn(layout, 1);
1495                 uiItemR(col, ptr, "factor_x", 0, IFACE_("X"), ICON_NONE);
1496                 uiItemR(col, ptr, "factor_y", 0, IFACE_("Y"), ICON_NONE);
1497         }
1498         else {
1499                 col = uiLayoutColumn(layout, 1);
1500                 uiItemR(col, ptr, "size_x", 0, IFACE_("X"), ICON_NONE);
1501                 uiItemR(col, ptr, "size_y", 0, IFACE_("Y"), ICON_NONE);
1502         }
1503 }
1504
1505 static void node_composit_buts_dblur(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1506 {
1507         uiLayout *col;
1508         
1509         uiItemR(layout, ptr, "iterations", 0, NULL, ICON_NONE);
1510         uiItemR(layout, ptr, "use_wrap", 0, NULL, ICON_NONE);
1511         
1512         col = uiLayoutColumn(layout, 1);
1513         uiItemL(col, IFACE_("Center:"), ICON_NONE);
1514         uiItemR(col, ptr, "center_x", 0, IFACE_("X"), ICON_NONE);
1515         uiItemR(col, ptr, "center_y", 0, IFACE_("Y"), ICON_NONE);
1516         
1517         uiItemS(layout);
1518         
1519         col = uiLayoutColumn(layout, 1);
1520         uiItemR(col, ptr, "distance", 0, NULL, ICON_NONE);
1521         uiItemR(col, ptr, "angle", 0, NULL, ICON_NONE);
1522         
1523         uiItemS(layout);
1524         
1525         uiItemR(layout, ptr, "spin", 0, NULL, ICON_NONE);
1526         uiItemR(layout, ptr, "zoom", 0, NULL, ICON_NONE);
1527 }
1528
1529 static void node_composit_buts_bilateralblur(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1530 {       
1531         uiLayout *col;
1532         
1533         col = uiLayoutColumn(layout, 1);
1534         uiItemR(col, ptr, "iterations", 0, NULL, ICON_NONE);
1535         uiItemR(col, ptr, "sigma_color", 0, NULL, ICON_NONE);
1536         uiItemR(col, ptr, "sigma_space", 0, NULL, ICON_NONE);
1537 }
1538
1539 static void node_composit_buts_defocus(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1540 {
1541         uiLayout *sub, *col;
1542         
1543         col = uiLayoutColumn(layout, 0);
1544         uiItemL(col, IFACE_("Bokeh Type:"), ICON_NONE);
1545         uiItemR(col, ptr, "bokeh", 0, "", ICON_NONE);
1546         uiItemR(col, ptr, "angle", 0, NULL, ICON_NONE);
1547
1548         uiItemR(layout, ptr, "use_gamma_correction", 0, NULL, ICON_NONE);
1549
1550         col = uiLayoutColumn(layout, 0);
1551         uiLayoutSetActive(col, RNA_boolean_get(ptr, "use_zbuffer") == TRUE);
1552         uiItemR(col, ptr, "f_stop", 0, NULL, ICON_NONE);
1553
1554         uiItemR(layout, ptr, "blur_max", 0, NULL, ICON_NONE);
1555         uiItemR(layout, ptr, "threshold", 0, NULL, ICON_NONE);
1556
1557         col = uiLayoutColumn(layout, 0);
1558         uiItemR(col, ptr, "use_preview", 0, NULL, ICON_NONE);
1559         sub = uiLayoutColumn(col, 0);
1560         uiLayoutSetActive(sub, RNA_boolean_get(ptr, "use_preview"));
1561         uiItemR(sub, ptr, "samples", 0, NULL, ICON_NONE);
1562         
1563         col = uiLayoutColumn(layout, 0);
1564         uiItemR(col, ptr, "use_zbuffer", 0, NULL, ICON_NONE);
1565         sub = uiLayoutColumn(col, 0);
1566         uiLayoutSetActive(sub, RNA_boolean_get(ptr, "use_zbuffer") == FALSE);
1567         uiItemR(sub, ptr, "z_scale", 0, NULL, ICON_NONE);
1568 }
1569
1570 /* qdn: glare node */
1571 static void node_composit_buts_glare(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1572 {       
1573         uiItemR(layout, ptr, "glare_type", 0, "", ICON_NONE);
1574         uiItemR(layout, ptr, "quality", 0, "", ICON_NONE);
1575
1576         if (RNA_enum_get(ptr, "glare_type") != 1) {
1577                 uiItemR(layout, ptr, "iterations", 0, NULL, ICON_NONE);
1578         
1579                 if (RNA_enum_get(ptr, "glare_type") != 0)
1580                         uiItemR(layout, ptr, "color_modulation", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
1581         }
1582         
1583         uiItemR(layout, ptr, "mix", 0, NULL, ICON_NONE);
1584         uiItemR(layout, ptr, "threshold", 0, NULL, ICON_NONE);
1585
1586         if (RNA_enum_get(ptr, "glare_type") == 2) {
1587                 uiItemR(layout, ptr, "streaks", 0, NULL, ICON_NONE);
1588                 uiItemR(layout, ptr, "angle_offset", 0, NULL, ICON_NONE);
1589         }
1590         if (RNA_enum_get(ptr, "glare_type") == 0 || RNA_enum_get(ptr, "glare_type") == 2) {
1591                 uiItemR(layout, ptr, "fade", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
1592                 
1593                 if (RNA_enum_get(ptr, "glare_type") == 0)
1594                         uiItemR(layout, ptr, "use_rotate_45", 0, NULL, ICON_NONE);
1595         }
1596         if (RNA_enum_get(ptr, "glare_type") == 1) {
1597                 uiItemR(layout, ptr, "size", 0, NULL, ICON_NONE);
1598         }
1599 }
1600
1601 static void node_composit_buts_tonemap(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1602 {       
1603         uiLayout *col;
1604
1605         col = uiLayoutColumn(layout, 0);
1606         uiItemR(col, ptr, "tonemap_type", 0, "", ICON_NONE);
1607         if (RNA_enum_get(ptr, "tonemap_type") == 0) {
1608                 uiItemR(col, ptr, "key", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
1609                 uiItemR(col, ptr, "offset", 0, NULL, ICON_NONE);
1610                 uiItemR(col, ptr, "gamma", 0, NULL, ICON_NONE);
1611         }
1612         else {
1613                 uiItemR(col, ptr, "intensity", 0, NULL, ICON_NONE);
1614                 uiItemR(col, ptr, "contrast", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
1615                 uiItemR(col, ptr, "adaptation", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
1616                 uiItemR(col, ptr, "correction", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
1617         }
1618 }
1619
1620 static void node_composit_buts_lensdist(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1621 {
1622         uiLayout *col;
1623
1624         col = uiLayoutColumn(layout, 0);
1625         uiItemR(col, ptr, "use_projector", 0, NULL, ICON_NONE);
1626
1627         col = uiLayoutColumn(col, 0);
1628         uiLayoutSetActive(col, RNA_boolean_get(ptr, "use_projector") == FALSE);
1629         uiItemR(col, ptr, "use_jitter", 0, NULL, ICON_NONE);
1630         uiItemR(col, ptr, "use_fit", 0, NULL, ICON_NONE);
1631 }
1632
1633 static void node_composit_buts_vecblur(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1634 {
1635         uiLayout *col;
1636         
1637         col = uiLayoutColumn(layout, 0);
1638         uiItemR(col, ptr, "samples", 0, NULL, ICON_NONE);
1639         uiItemR(col, ptr, "factor", 0, IFACE_("Blur"), ICON_NONE);
1640         
1641         col = uiLayoutColumn(layout, 1);
1642         uiItemL(col, IFACE_("Speed:"), ICON_NONE);
1643         uiItemR(col, ptr, "speed_min", 0, IFACE_("Min"), ICON_NONE);
1644         uiItemR(col, ptr, "speed_max", 0, IFACE_("Max"), ICON_NONE);
1645
1646         uiItemR(layout, ptr, "use_curved", 0, NULL, ICON_NONE);
1647 }
1648
1649 static void node_composit_buts_filter(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1650 {
1651         uiItemR(layout, ptr, "filter_type", 0, "", ICON_NONE);
1652 }
1653
1654 static void node_composit_buts_flip(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1655 {
1656         uiItemR(layout, ptr, "axis", 0, "", ICON_NONE);
1657 }
1658
1659 static void node_composit_buts_crop(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1660 {
1661         uiLayout *col;
1662
1663         uiItemR(layout, ptr, "use_crop_size", 0, NULL, ICON_NONE);
1664         uiItemR(layout, ptr, "relative", 0, NULL, ICON_NONE);
1665
1666         col = uiLayoutColumn(layout, 1);
1667         if (RNA_boolean_get(ptr, "relative")) {
1668                 uiItemR(col, ptr, "rel_min_x", 0, IFACE_("Left"), ICON_NONE);
1669                 uiItemR(col, ptr, "rel_max_x", 0, IFACE_("Right"), ICON_NONE);
1670                 uiItemR(col, ptr, "rel_min_y", 0, IFACE_("Up"), ICON_NONE);
1671                 uiItemR(col, ptr, "rel_max_y", 0, IFACE_("Down"), ICON_NONE);
1672         }
1673         else {
1674                 uiItemR(col, ptr, "min_x", 0, IFACE_("Left"), ICON_NONE);
1675                 uiItemR(col, ptr, "max_x", 0, IFACE_("Right"), ICON_NONE);
1676                 uiItemR(col, ptr, "min_y", 0, IFACE_("Up"), ICON_NONE);
1677                 uiItemR(col, ptr, "max_y", 0, IFACE_("Down"), ICON_NONE);
1678         }
1679 }
1680
1681 static void node_composit_buts_splitviewer(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1682 {
1683         uiLayout *row, *col;
1684         
1685         col = uiLayoutColumn(layout, 0);
1686         row = uiLayoutRow(col, 0);
1687         uiItemR(row, ptr, "axis", UI_ITEM_R_EXPAND, NULL, ICON_NONE);
1688         uiItemR(col, ptr, "factor", 0, NULL, ICON_NONE);
1689 }
1690
1691 static void node_composit_buts_double_edge_mask(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1692 {
1693         uiLayout *col;
1694
1695         col = uiLayoutColumn(layout, 0);
1696
1697         uiItemL(col, IFACE_("Inner Edge:"), ICON_NONE);
1698         uiItemR(col, ptr, "inner_mode", 0, "", ICON_NONE);
1699         uiItemL(col, IFACE_("Buffer Edge:"), ICON_NONE);
1700         uiItemR(col, ptr, "edge_mode", 0, "", ICON_NONE);
1701 }
1702
1703 static void node_composit_buts_map_value(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1704 {
1705         uiLayout *sub, *col;
1706         
1707         col = uiLayoutColumn(layout, 1);
1708         uiItemR(col, ptr, "offset", 0, NULL, ICON_NONE);
1709         uiItemR(col, ptr, "size", 0, NULL, ICON_NONE);
1710         
1711         col = uiLayoutColumn(layout, 1);
1712         uiItemR(col, ptr, "use_min", 0, NULL, ICON_NONE);
1713         sub = uiLayoutColumn(col, 0);
1714         uiLayoutSetActive(sub, RNA_boolean_get(ptr, "use_min"));
1715         uiItemR(sub, ptr, "min", 0, "", ICON_NONE);
1716         
1717         col = uiLayoutColumn(layout, 1);
1718         uiItemR(col, ptr, "use_max", 0, NULL, ICON_NONE);
1719         sub = uiLayoutColumn(col, 0);
1720         uiLayoutSetActive(sub, RNA_boolean_get(ptr, "use_max"));
1721         uiItemR(sub, ptr, "max", 0, "", ICON_NONE);
1722 }
1723
1724 static void node_composit_buts_alphaover(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1725 {       
1726         uiLayout *col;
1727         
1728         col = uiLayoutColumn(layout, 1);
1729         uiItemR(col, ptr, "use_premultiply", 0, NULL, ICON_NONE);
1730         uiItemR(col, ptr, "premul", 0, NULL, ICON_NONE);
1731 }
1732
1733 static void node_composit_buts_zcombine(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1734 {       
1735         uiLayout *col;
1736         
1737         col = uiLayoutColumn(layout, 1);
1738         uiItemR(col, ptr, "use_alpha", 0, NULL, ICON_NONE);
1739 }
1740
1741
1742 static void node_composit_buts_hue_sat(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1743 {
1744         uiLayout *col;
1745         
1746         col = uiLayoutColumn(layout, 0);
1747         uiItemR(col, ptr, "color_hue", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
1748         uiItemR(col, ptr, "color_saturation", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
1749         uiItemR(col, ptr, "color_value", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
1750 }
1751
1752 static void node_composit_buts_dilateerode(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1753 {
1754         uiItemR(layout, ptr, "type", 0, NULL, ICON_NONE);
1755         uiItemR(layout, ptr, "distance", 0, NULL, ICON_NONE);
1756         if (RNA_enum_get(ptr, "type") == CMP_NODE_DILATEERODE_DISTANCE_THRESH) {
1757                 uiItemR(layout, ptr, "edge", 0, NULL, ICON_NONE);
1758         }
1759 }
1760
1761 static void node_composit_buts_diff_matte(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1762 {
1763         uiLayout *col;
1764         
1765         col = uiLayoutColumn(layout, 1);
1766         uiItemR(col, ptr, "tolerance", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
1767         uiItemR(col, ptr, "falloff", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
1768 }
1769
1770 static void node_composit_buts_distance_matte(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1771 {
1772         uiLayout *col, *row;
1773         
1774         col = uiLayoutColumn(layout, 1);
1775    
1776         uiItemL(layout, IFACE_("Color Space:"), ICON_NONE);
1777         row = uiLayoutRow(layout, 0);
1778         uiItemR(row, ptr, "channel", UI_ITEM_R_EXPAND, NULL, ICON_NONE);
1779
1780         uiItemR(col, ptr, "tolerance", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
1781         uiItemR(col, ptr, "falloff", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
1782 }
1783
1784 static void node_composit_buts_color_spill(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1785 {
1786         uiLayout *row, *col;
1787         
1788         uiItemL(layout, IFACE_("Despill Channel:"), ICON_NONE);
1789         row = uiLayoutRow(layout, 0);
1790         uiItemR(row, ptr, "channel", UI_ITEM_R_EXPAND, NULL, ICON_NONE);
1791
1792         col = uiLayoutColumn(layout, 0);
1793         uiItemR(col, ptr, "limit_method", 0, NULL, ICON_NONE);
1794
1795         if (RNA_enum_get(ptr, "limit_method") == 0) {
1796                 uiItemL(col, IFACE_("Limiting Channel:"), ICON_NONE);
1797                 row = uiLayoutRow(col, 0);
1798                 uiItemR(row, ptr, "limit_channel", UI_ITEM_R_EXPAND, NULL, ICON_NONE);
1799         }
1800
1801         uiItemR(col, ptr, "ratio", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
1802         uiItemR(col, ptr, "use_unspill", 0, NULL, ICON_NONE);
1803         if (RNA_boolean_get(ptr, "use_unspill") == TRUE) {
1804                 uiItemR(col, ptr, "unspill_red", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
1805                 uiItemR(col, ptr, "unspill_green", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
1806                 uiItemR(col, ptr, "unspill_blue", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
1807         }
1808 }
1809
1810 static void node_composit_buts_chroma_matte(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1811 {
1812         uiLayout *col;
1813         
1814         col = uiLayoutColumn(layout, 0);
1815         uiItemR(col, ptr, "tolerance", 0, NULL, ICON_NONE);
1816         uiItemR(col, ptr, "threshold", 0, NULL, ICON_NONE);
1817         
1818         col = uiLayoutColumn(layout, 1);
1819         /*uiItemR(col, ptr, "lift", UI_ITEM_R_SLIDER, NULL, ICON_NONE);  Removed for now */
1820         uiItemR(col, ptr, "gain", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
1821         /*uiItemR(col, ptr, "shadow_adjust", UI_ITEM_R_SLIDER, NULL, ICON_NONE);  Removed for now*/
1822 }
1823
1824 static void node_composit_buts_color_matte(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1825 {
1826         uiLayout *col;
1827         
1828         col = uiLayoutColumn(layout, 1);
1829         uiItemR(col, ptr, "color_hue", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
1830         uiItemR(col, ptr, "color_saturation", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
1831         uiItemR(col, ptr, "color_value", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
1832 }
1833
1834 static void node_composit_buts_channel_matte(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1835 {       
1836         uiLayout *col, *row;
1837
1838         uiItemL(layout, IFACE_("Color Space:"), ICON_NONE);
1839         row = uiLayoutRow(layout, 0);
1840         uiItemR(row, ptr, "color_space", UI_ITEM_R_EXPAND, NULL, ICON_NONE);
1841
1842         col = uiLayoutColumn(layout, 0);
1843         uiItemL(col, IFACE_("Key Channel:"), ICON_NONE);
1844         row = uiLayoutRow(col, 0);
1845         uiItemR(row, ptr, "matte_channel", UI_ITEM_R_EXPAND, NULL, ICON_NONE);
1846
1847         col = uiLayoutColumn(layout, 0);
1848
1849         uiItemR(col, ptr, "limit_method", 0, NULL, ICON_NONE);
1850         if (RNA_enum_get(ptr, "limit_method") == 0) {
1851                 uiItemL(col, IFACE_("Limiting Channel:"), ICON_NONE);
1852                 row = uiLayoutRow(col, 0);
1853                 uiItemR(row, ptr, "limit_channel", UI_ITEM_R_EXPAND, NULL, ICON_NONE);
1854         }
1855
1856         uiItemR(col, ptr, "limit_max", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
1857         uiItemR(col, ptr, "limit_min", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
1858 }
1859
1860 static void node_composit_buts_luma_matte(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1861 {
1862         uiLayout *col;
1863         
1864         col = uiLayoutColumn(layout, 1);
1865         uiItemR(col, ptr, "limit_max", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
1866         uiItemR(col, ptr, "limit_min", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
1867 }
1868
1869 static void node_composit_buts_map_uv(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1870 {
1871         uiItemR(layout, ptr, "alpha", 0, NULL, ICON_NONE);
1872 }
1873
1874 static void node_composit_buts_id_mask(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1875 {
1876         uiItemR(layout, ptr, "index", 0, NULL, ICON_NONE);
1877         uiItemR(layout, ptr, "use_smooth_mask", 0, NULL, ICON_NONE);
1878 }
1879
1880 /* draw function for file output node sockets, displays only sub-path and format, no value button */
1881 static void node_draw_input_file_output(const bContext *C, uiBlock *block,
1882                                         bNodeTree *ntree, bNode *node, bNodeSocket *sock,
1883                                         const char *UNUSED(name), int x, int y, int width)
1884 {
1885         uiLayout *layout, *row;
1886         PointerRNA nodeptr, inputptr, imfptr;
1887         int imtype;
1888         int rx, ry;
1889         RNA_pointer_create(&ntree->id, &RNA_Node, node, &nodeptr);
1890         
1891         layout = uiBlockLayout(block, UI_LAYOUT_VERTICAL, UI_LAYOUT_PANEL, x, y + NODE_DY, width, 20, UI_GetStyle());
1892         row = uiLayoutRow(layout, 0);
1893         
1894         imfptr = RNA_pointer_get(&nodeptr, "format");
1895         imtype = RNA_enum_get(&imfptr, "file_format");
1896         if (imtype == R_IMF_IMTYPE_MULTILAYER) {
1897                 NodeImageMultiFileSocket *input = sock->storage;
1898                 RNA_pointer_create(&ntree->id, &RNA_NodeOutputFileSlotLayer, input, &inputptr);
1899                 
1900                 uiItemL(row, input->layer, 0);
1901         }
1902         else {
1903                 NodeImageMultiFileSocket *input = sock->storage;
1904                 PropertyRNA *imtype_prop;
1905                 const char *imtype_name;
1906                 RNA_pointer_create(&ntree->id, &RNA_NodeOutputFileSlotFile, input, &inputptr);
1907                 
1908                 uiItemL(row, input->path, 0);
1909                 
1910                 if (!RNA_boolean_get(&inputptr, "use_node_format"))
1911                         imfptr = RNA_pointer_get(&inputptr, "format");
1912                 
1913                 imtype_prop = RNA_struct_find_property(&imfptr, "file_format");
1914                 RNA_property_enum_name((bContext *)C, &imfptr, imtype_prop, RNA_property_enum_get(&imfptr, imtype_prop), &imtype_name);
1915                 uiBlockSetEmboss(block, UI_EMBOSSP);
1916                 uiItemL(row, imtype_name, 0);
1917                 uiBlockSetEmboss(block, UI_EMBOSSN);
1918         }
1919         
1920         uiBlockLayoutResolve(block, &rx, &ry);
1921 }
1922 static void node_composit_buts_file_output(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
1923 {
1924         PointerRNA imfptr = RNA_pointer_get(ptr, "format");
1925         int multilayer = (RNA_enum_get(&imfptr, "file_format") == R_IMF_IMTYPE_MULTILAYER);
1926         
1927         if (multilayer)
1928                 uiItemL(layout, IFACE_("Path:"), 0);
1929         else
1930                 uiItemL(layout, IFACE_("Base Path:"), 0);
1931         uiItemR(layout, ptr, "base_path", 0, "", ICON_NONE);
1932 }
1933 static void node_composit_buts_file_output_details(uiLayout *layout, bContext *C, PointerRNA *ptr)
1934 {
1935         PointerRNA imfptr = RNA_pointer_get(ptr, "format");
1936         PointerRNA active_input_ptr, op_ptr;
1937         uiLayout *row;
1938         int active_index;
1939         int multilayer = (RNA_enum_get(&imfptr, "file_format") == R_IMF_IMTYPE_MULTILAYER);
1940         
1941         node_composit_buts_file_output(layout, C, ptr);
1942         uiTemplateImageSettings(layout, &imfptr);
1943         
1944         uiItemS(layout);
1945         
1946         uiItemO(layout, IFACE_("Add Input"), ICON_ZOOMIN, "NODE_OT_output_file_add_socket");
1947         
1948         active_index = RNA_int_get(ptr, "active_input_index");
1949         /* using different collection properties if multilayer format is enabled */
1950         if (multilayer) {
1951                 uiTemplateList(layout, C, ptr, "layer_slots", ptr, "active_input_index", NULL, 0, 0, 0);
1952                 RNA_property_collection_lookup_int(ptr, RNA_struct_find_property(ptr, "layer_slots"), active_index, &active_input_ptr);
1953         }
1954         else {
1955                 uiTemplateList(layout, C, ptr, "file_slots", ptr, "active_input_index", NULL, 0, 0, 0);
1956                 RNA_property_collection_lookup_int(ptr, RNA_struct_find_property(ptr, "file_slots"), active_index, &active_input_ptr);
1957         }
1958         /* XXX collection lookup does not return the ID part of the pointer, setting this manually here */
1959         active_input_ptr.id.data = ptr->id.data;
1960         
1961         row = uiLayoutRow(layout, 1);
1962         op_ptr = uiItemFullO(row, "NODE_OT_output_file_move_active_socket", "", ICON_TRIA_UP, NULL, WM_OP_INVOKE_DEFAULT, UI_ITEM_O_RETURN_PROPS);
1963         RNA_enum_set(&op_ptr, "direction", 1);
1964         op_ptr = uiItemFullO(row, "NODE_OT_output_file_move_active_socket", "", ICON_TRIA_DOWN, NULL, WM_OP_INVOKE_DEFAULT, UI_ITEM_O_RETURN_PROPS);
1965         RNA_enum_set(&op_ptr, "direction", 2);
1966         
1967         if (active_input_ptr.data) {
1968                 if (multilayer) {
1969                         uiLayout *row, *col;
1970                         col = uiLayoutColumn(layout, 1);
1971                         
1972                         uiItemL(col, IFACE_("Layer:"), 0);
1973                         row = uiLayoutRow(col, 0);
1974                         uiItemR(row, &active_input_ptr, "name", 0, "", 0);
1975                         uiItemFullO(row, "NODE_OT_output_file_remove_active_socket", "", ICON_X, NULL, WM_OP_EXEC_DEFAULT, UI_ITEM_R_ICON_ONLY);
1976                 }
1977                 else {
1978                         uiLayout *row, *col;
1979                         col = uiLayoutColumn(layout, 1);
1980                         
1981                         uiItemL(col, IFACE_("File Path:"), 0);
1982                         row = uiLayoutRow(col, 0);
1983                         uiItemR(row, &active_input_ptr, "path", 0, "", 0);
1984                         uiItemFullO(row, "NODE_OT_output_file_remove_active_socket", "", ICON_X, NULL, WM_OP_EXEC_DEFAULT, UI_ITEM_R_ICON_ONLY);
1985                         
1986                         /* format details for individual files */
1987                         imfptr = RNA_pointer_get(&active_input_ptr, "format");
1988                         
1989                         col = uiLayoutColumn(layout, 1);
1990                         uiItemL(col, IFACE_("Format:"), 0);
1991                         uiItemR(col, &active_input_ptr, "use_node_format", 0, NULL, 0);
1992                         
1993                         col = uiLayoutColumn(layout, 0);
1994                         uiLayoutSetActive(col, RNA_boolean_get(&active_input_ptr, "use_node_format") == FALSE);
1995                         uiTemplateImageSettings(col, &imfptr);
1996                 }
1997         }
1998 }
1999
2000 static void node_composit_buts_scale(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
2001 {
2002         uiItemR(layout, ptr, "space", 0, "", ICON_NONE);
2003 }
2004
2005 static void node_composit_buts_rotate(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
2006 {
2007         uiItemR(layout, ptr, "filter_type", 0, "", ICON_NONE);
2008 }
2009
2010 static void node_composit_buts_invert(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
2011 {
2012         uiLayout *col;
2013         
2014         col = uiLayoutColumn(layout, 0);
2015         uiItemR(col, ptr, "invert_rgb", 0, NULL, ICON_NONE);
2016         uiItemR(col, ptr, "invert_alpha", 0, NULL, ICON_NONE);
2017 }
2018
2019 static void node_composit_buts_premulkey(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
2020 {
2021         uiItemR(layout, ptr, "mapping", 0, "", ICON_NONE);
2022 }
2023
2024 static void node_composit_buts_view_levels(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
2025 {
2026         uiItemR(layout, ptr, "channel", UI_ITEM_R_EXPAND, NULL, ICON_NONE);
2027 }
2028
2029 static void node_composit_buts_colorbalance(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
2030 {
2031         uiLayout *split, *col, *row;
2032         
2033         uiItemR(layout, ptr, "correction_method", 0, NULL, ICON_NONE);
2034         
2035         if (RNA_enum_get(ptr, "correction_method") == 0) {
2036         
2037                 split = uiLayoutSplit(layout, 0, 0);
2038                 col = uiLayoutColumn(split, 0);
2039                 uiTemplateColorWheel(col, ptr, "lift", 1, 1, 0, 1);
2040                 row = uiLayoutRow(col, 0);
2041                 uiItemR(row, ptr, "lift", 0, NULL, ICON_NONE);
2042                 
2043                 col = uiLayoutColumn(split, 0);
2044                 uiTemplateColorWheel(col, ptr, "gamma", 1, 1, 1, 1);
2045                 row = uiLayoutRow(col, 0);
2046                 uiItemR(row, ptr, "gamma", 0, NULL, ICON_NONE);
2047                 
2048                 col = uiLayoutColumn(split, 0);
2049                 uiTemplateColorWheel(col, ptr, "gain", 1, 1, 1, 1);
2050                 row = uiLayoutRow(col, 0);
2051                 uiItemR(row, ptr, "gain", 0, NULL, ICON_NONE);
2052
2053         }
2054         else {
2055                 
2056                 split = uiLayoutSplit(layout, 0, 0);
2057                 col = uiLayoutColumn(split, 0);
2058                 uiTemplateColorWheel(col, ptr, "offset", 1, 1, 0, 1);
2059                 row = uiLayoutRow(col, 0);
2060                 uiItemR(row, ptr, "offset", 0, NULL, ICON_NONE);
2061                 
2062                 col = uiLayoutColumn(split, 0);
2063                 uiTemplateColorWheel(col, ptr, "power", 1, 1, 0, 1);
2064                 row = uiLayoutRow(col, 0);
2065                 uiItemR(row, ptr, "power", 0, NULL, ICON_NONE);
2066                 
2067                 col = uiLayoutColumn(split, 0);
2068                 uiTemplateColorWheel(col, ptr, "slope", 1, 1, 0, 1);
2069                 row = uiLayoutRow(col, 0);
2070                 uiItemR(row, ptr, "slope", 0, NULL, ICON_NONE);
2071         }
2072
2073 }
2074 static void node_composit_buts_colorbalance_but(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
2075 {
2076         uiItemR(layout, ptr, "correction_method", 0, NULL, ICON_NONE);
2077
2078         if (RNA_enum_get(ptr, "correction_method") == 0) {
2079
2080                 uiTemplateColorWheel(layout, ptr, "lift", 1, 1, 0, 1);
2081                 uiItemR(layout, ptr, "lift", 0, NULL, ICON_NONE);
2082
2083                 uiTemplateColorWheel(layout, ptr, "gamma", 1, 1, 1, 1);
2084                 uiItemR(layout, ptr, "gamma", 0, NULL, ICON_NONE);
2085
2086                 uiTemplateColorWheel(layout, ptr, "gain", 1, 1, 1, 1);
2087                 uiItemR(layout, ptr, "gain", 0, NULL, ICON_NONE);
2088         }
2089         else {
2090                 uiTemplateColorWheel(layout, ptr, "offset", 1, 1, 0, 1);
2091                 uiItemR(layout, ptr, "offset", 0, NULL, ICON_NONE);
2092
2093                 uiTemplateColorWheel(layout, ptr, "power", 1, 1, 0, 1);
2094                 uiItemR(layout, ptr, "power", 0, NULL, ICON_NONE);
2095
2096                 uiTemplateColorWheel(layout, ptr, "slope", 1, 1, 0, 1);
2097                 uiItemR(layout, ptr, "slope", 0, NULL, ICON_NONE);
2098         }
2099 }
2100
2101
2102 static void node_composit_buts_huecorrect(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
2103 {
2104         uiTemplateCurveMapping(layout, ptr, "mapping", 'h', 0, 0);
2105 }
2106
2107 static void node_composit_buts_ycc(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
2108
2109         uiItemR(layout, ptr, "mode", 0, "", ICON_NONE);
2110 }
2111
2112 static void node_composit_buts_movieclip(uiLayout *layout, bContext *C, PointerRNA *ptr)
2113 {
2114         uiTemplateID(layout, C, ptr, "clip", NULL, "CLIP_OT_open", NULL);
2115 }
2116
2117 static void node_composit_buts_stabilize2d(uiLayout *layout, bContext *C, PointerRNA *ptr)
2118 {
2119         bNode *node = ptr->data;
2120
2121         uiTemplateID(layout, C, ptr, "clip", NULL, "CLIP_OT_open", NULL);
2122
2123         if (!node->id)
2124                 return;
2125
2126         uiItemR(layout, ptr, "filter_type", 0, "", 0);
2127 }
2128
2129 static void node_composit_buts_transform(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
2130 {
2131         uiItemR(layout, ptr, "filter_type", 0, "", 0);
2132 }
2133
2134 static void node_composit_buts_moviedistortion(uiLayout *layout, bContext *C, PointerRNA *ptr)
2135 {
2136         bNode *node = ptr->data;
2137
2138         uiTemplateID(layout, C, ptr, "clip", NULL, "CLIP_OT_open", NULL);
2139
2140         if (!node->id)
2141                 return;
2142
2143         uiItemR(layout, ptr, "distortion_type", 0, "", 0);
2144 }
2145
2146 static void node_composit_buts_colorcorrection(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
2147 {
2148         uiLayout *row;
2149         
2150         row = uiLayoutRow(layout, 0);
2151         uiItemR(row, ptr, "red", 0, NULL, ICON_NONE);
2152         uiItemR(row, ptr, "green", 0, NULL, ICON_NONE);
2153         uiItemR(row, ptr, "blue", 0, NULL, ICON_NONE);
2154
2155         row = uiLayoutRow(layout, 0);
2156         uiItemL(row, "", 0);
2157         uiItemL(row, IFACE_("Saturation"), 0);
2158         uiItemL(row, IFACE_("Contrast"), 0);
2159         uiItemL(row, IFACE_("Gamma"), 0);
2160         uiItemL(row, IFACE_("Gain"), 0);
2161         uiItemL(row, IFACE_("Lift"), 0);
2162
2163         row = uiLayoutRow(layout, 0);
2164         uiItemL(row, IFACE_("Master"), 0);
2165         uiItemR(row, ptr, "master_saturation", UI_ITEM_R_SLIDER, "", ICON_NONE);
2166         uiItemR(row, ptr, "master_contrast", UI_ITEM_R_SLIDER, "", ICON_NONE);
2167         uiItemR(row, ptr, "master_gamma", UI_ITEM_R_SLIDER, "", ICON_NONE);
2168         uiItemR(row, ptr, "master_gain", UI_ITEM_R_SLIDER, "", ICON_NONE);
2169         uiItemR(row, ptr, "master_lift", UI_ITEM_R_SLIDER, "", ICON_NONE);
2170
2171         row = uiLayoutRow(layout, 0);
2172         uiItemL(row, IFACE_("Highlights"), 0);
2173         uiItemR(row, ptr, "highlights_saturation", UI_ITEM_R_SLIDER, "", ICON_NONE);
2174         uiItemR(row, ptr, "highlights_contrast", UI_ITEM_R_SLIDER, "", ICON_NONE);
2175         uiItemR(row, ptr, "highlights_gamma", UI_ITEM_R_SLIDER, "", ICON_NONE);
2176         uiItemR(row, ptr, "highlights_gain", UI_ITEM_R_SLIDER, "", ICON_NONE);
2177         uiItemR(row, ptr, "highlights_lift", UI_ITEM_R_SLIDER, "", ICON_NONE);
2178
2179         row = uiLayoutRow(layout, 0);
2180         uiItemL(row, IFACE_("Midtones"), 0);
2181         uiItemR(row, ptr, "midtones_saturation", UI_ITEM_R_SLIDER, "", ICON_NONE);
2182         uiItemR(row, ptr, "midtones_contrast", UI_ITEM_R_SLIDER, "", ICON_NONE);
2183         uiItemR(row, ptr, "midtones_gamma", UI_ITEM_R_SLIDER, "", ICON_NONE);
2184         uiItemR(row, ptr, "midtones_gain", UI_ITEM_R_SLIDER, "", ICON_NONE);
2185         uiItemR(row, ptr, "midtones_lift", UI_ITEM_R_SLIDER, "", ICON_NONE);
2186
2187         row = uiLayoutRow(layout, 0);
2188         uiItemL(row, IFACE_("Shadows"), 0);
2189         uiItemR(row, ptr, "shadows_saturation", UI_ITEM_R_SLIDER, "", ICON_NONE);
2190         uiItemR(row, ptr, "shadows_contrast", UI_ITEM_R_SLIDER, "", ICON_NONE);
2191         uiItemR(row, ptr, "shadows_gamma", UI_ITEM_R_SLIDER, "", ICON_NONE);
2192         uiItemR(row, ptr, "shadows_gain", UI_ITEM_R_SLIDER, "", ICON_NONE);
2193         uiItemR(row, ptr, "shadows_lift", UI_ITEM_R_SLIDER, "", ICON_NONE);
2194
2195         row = uiLayoutRow(layout, 0);
2196         uiItemR(row, ptr, "midtones_start", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2197         uiItemR(row, ptr, "midtones_end", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2198 }
2199
2200 static void node_composit_buts_colorcorrection_but(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
2201 {
2202         uiLayout *row;
2203         
2204         row = uiLayoutRow(layout, 0);
2205         uiItemR(row, ptr, "red", 0, NULL, ICON_NONE);
2206         uiItemR(row, ptr, "green", 0, NULL, ICON_NONE);
2207         uiItemR(row, ptr, "blue", 0, NULL, ICON_NONE);
2208         row = layout;
2209         uiItemL(row, IFACE_("Saturation"), 0);
2210         uiItemR(row, ptr, "master_saturation", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2211         uiItemR(row, ptr, "highlights_saturation", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2212         uiItemR(row, ptr, "midtones_saturation", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2213         uiItemR(row, ptr, "shadows_saturation", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2214
2215         uiItemL(row, IFACE_("Contrast"), 0);
2216         uiItemR(row, ptr, "master_contrast", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2217         uiItemR(row, ptr, "highlights_contrast", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2218         uiItemR(row, ptr, "midtones_contrast", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2219         uiItemR(row, ptr, "shadows_contrast", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2220
2221         uiItemL(row, IFACE_("Gamma"), 0);
2222         uiItemR(row, ptr, "master_gamma", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2223         uiItemR(row, ptr, "highlights_gamma", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2224         uiItemR(row, ptr, "midtones_gamma", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2225         uiItemR(row, ptr, "shadows_gamma", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2226
2227         uiItemL(row, IFACE_("Gain"), 0);
2228         uiItemR(row, ptr, "master_gain", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2229         uiItemR(row, ptr, "highlights_gain", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2230         uiItemR(row, ptr, "midtones_gain", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2231         uiItemR(row, ptr, "shadows_gain", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2232         
2233         uiItemL(row, IFACE_("Lift"), 0);
2234         uiItemR(row, ptr, "master_lift", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2235         uiItemR(row, ptr, "highlights_lift", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2236         uiItemR(row, ptr, "midtones_lift", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2237         uiItemR(row, ptr, "shadows_lift", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2238
2239         row = uiLayoutRow(layout, 0);
2240         uiItemR(row, ptr, "midtones_start", 0, NULL, ICON_NONE);
2241         uiItemR(row, ptr, "midtones_end", 0, NULL, ICON_NONE);
2242 }
2243
2244 static void node_composit_buts_switch(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
2245 {
2246         uiItemR(layout, ptr, "check", 0, NULL, ICON_NONE);
2247 }
2248
2249 static void node_composit_buts_boxmask(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
2250 {
2251         uiLayout *row;
2252         
2253         row = uiLayoutRow(layout, 1);
2254         uiItemR(row, ptr, "x", 0, NULL, ICON_NONE);
2255         uiItemR(row, ptr, "y", 0, NULL, ICON_NONE);
2256         
2257         row = uiLayoutRow(layout, 1);
2258         uiItemR(row, ptr, "width", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2259         uiItemR(row, ptr, "height", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2260
2261         uiItemR(layout, ptr, "rotation", 0, NULL, ICON_NONE);
2262         uiItemR(layout, ptr, "mask_type", 0, NULL, ICON_NONE);
2263 }
2264
2265 static void node_composit_buts_bokehimage(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
2266 {
2267         uiItemR(layout, ptr, "flaps", 0, NULL, ICON_NONE);
2268         uiItemR(layout, ptr, "angle", 0, NULL, ICON_NONE);
2269         uiItemR(layout, ptr, "rounding", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2270         uiItemR(layout, ptr, "catadioptric", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2271         uiItemR(layout, ptr, "shift", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2272 }
2273
2274 void node_composit_backdrop_viewer(SpaceNode *snode, ImBuf *backdrop, bNode *node, int x, int y)
2275 {
2276 //      node_composit_backdrop_canvas(snode, backdrop, node, x, y);
2277         if (node->custom1 == 0) { /// @todo: why did we need this one?
2278                 const float backdropWidth = backdrop->x;
2279                 const float backdropHeight = backdrop->y;
2280                 const float cx  = x + snode->zoom * backdropWidth * node->custom3;
2281                 const float cy = y + snode->zoom * backdropHeight * node->custom4;
2282
2283                 glColor3f(1.0, 1.0, 1.0);
2284
2285                 glBegin(GL_LINES);
2286                 glVertex2f(cx - 25, cy - 25);
2287                 glVertex2f(cx + 25, cy + 25);
2288                 glVertex2f(cx + 25, cy - 25);
2289                 glVertex2f(cx - 25, cy + 25);
2290                 glEnd();
2291         }
2292 }
2293
2294 void node_composit_backdrop_boxmask(SpaceNode *snode, ImBuf *backdrop, bNode *node, int x, int y)
2295 {
2296         NodeBoxMask *boxmask = node->storage;
2297         const float backdropWidth = backdrop->x;
2298         const float backdropHeight = backdrop->y;
2299         const float aspect = backdropWidth / backdropHeight;
2300         const float rad = DEG2RADF(-boxmask->rotation);
2301         const float cosine = cosf(rad);
2302         const float sine = sinf(rad);
2303         const float halveBoxWidth = backdropWidth * (boxmask->width / 2.0f);
2304         const float halveBoxHeight = backdropHeight * (boxmask->height / 2.0f) * aspect;
2305
2306         float cx, cy, x1, x2, x3, x4;
2307         float y1, y2, y3, y4;
2308
2309
2310         /* keep this, saves us from a version patch */
2311         if (snode->zoom == 0.0f) snode->zoom = 1.0f;
2312
2313         glColor3f(1.0, 1.0, 1.0);
2314
2315         cx  = x + snode->zoom * backdropWidth * boxmask->x;
2316         cy = y + snode->zoom * backdropHeight * boxmask->y;
2317
2318         x1 = cx - (cosine * halveBoxWidth + sine * halveBoxHeight) * snode->zoom;
2319         x2 = cx - (cosine * -halveBoxWidth + sine * halveBoxHeight) * snode->zoom;
2320         x3 = cx - (cosine * -halveBoxWidth + sine * -halveBoxHeight) * snode->zoom;
2321         x4 = cx - (cosine * halveBoxWidth + sine * -halveBoxHeight) * snode->zoom;
2322         y1 = cy - (-sine * halveBoxWidth + cosine * halveBoxHeight) * snode->zoom;
2323         y2 = cy - (-sine * -halveBoxWidth + cosine * halveBoxHeight) * snode->zoom;
2324         y3 = cy - (-sine * -halveBoxWidth + cosine * -halveBoxHeight) * snode->zoom;
2325         y4 = cy - (-sine * halveBoxWidth + cosine * -halveBoxHeight) * snode->zoom;
2326
2327         glBegin(GL_LINE_LOOP);
2328         glVertex2f(x1, y1);
2329         glVertex2f(x2, y2);
2330         glVertex2f(x3, y3);
2331         glVertex2f(x4, y4);
2332         glEnd();
2333 }
2334
2335 void node_composit_backdrop_ellipsemask(SpaceNode *snode, ImBuf *backdrop, bNode *node, int x, int y)
2336 {
2337         NodeEllipseMask *ellipsemask = node->storage;
2338         const float backdropWidth = backdrop->x;
2339         const float backdropHeight = backdrop->y;
2340         const float aspect = backdropWidth / backdropHeight;
2341         const float rad = DEG2RADF(-ellipsemask->rotation);
2342         const float cosine = cosf(rad);
2343         const float sine = sinf(rad);
2344         const float halveBoxWidth = backdropWidth * (ellipsemask->width / 2.0f);
2345         const float halveBoxHeight = backdropHeight * (ellipsemask->height / 2.0f) * aspect;
2346
2347         float cx, cy, x1, x2, x3, x4;
2348         float y1, y2, y3, y4;
2349
2350
2351         /* keep this, saves us from a version patch */
2352         if (snode->zoom == 0.0f) snode->zoom = 1.0f;
2353
2354         glColor3f(1.0, 1.0, 1.0);
2355
2356         cx  = x + snode->zoom * backdropWidth * ellipsemask->x;
2357         cy = y + snode->zoom * backdropHeight * ellipsemask->y;
2358
2359         x1 = cx - (cosine * halveBoxWidth + sine * halveBoxHeight) * snode->zoom;
2360         x2 = cx - (cosine * -halveBoxWidth + sine * halveBoxHeight) * snode->zoom;
2361         x3 = cx - (cosine * -halveBoxWidth + sine * -halveBoxHeight) * snode->zoom;
2362         x4 = cx - (cosine * halveBoxWidth + sine * -halveBoxHeight) * snode->zoom;
2363         y1 = cy - (-sine * halveBoxWidth + cosine * halveBoxHeight) * snode->zoom;
2364         y2 = cy - (-sine * -halveBoxWidth + cosine * halveBoxHeight) * snode->zoom;
2365         y3 = cy - (-sine * -halveBoxWidth + cosine * -halveBoxHeight) * snode->zoom;
2366         y4 = cy - (-sine * halveBoxWidth + cosine * -halveBoxHeight) * snode->zoom;
2367
2368         glBegin(GL_LINE_LOOP);
2369
2370         glVertex2f(x1, y1);
2371         glVertex2f(x2, y2);
2372         glVertex2f(x3, y3);
2373         glVertex2f(x4, y4);
2374         glEnd();
2375 }
2376
2377 static void node_composit_buts_ellipsemask(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
2378 {
2379         uiLayout *row;
2380         row = uiLayoutRow(layout, 1);
2381         uiItemR(row, ptr, "x", 0, NULL, ICON_NONE);
2382         uiItemR(row, ptr, "y", 0, NULL, ICON_NONE);
2383         row = uiLayoutRow(layout, 1);
2384         uiItemR(row, ptr, "width", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2385         uiItemR(row, ptr, "height", UI_ITEM_R_SLIDER, NULL, ICON_NONE);
2386
2387         uiItemR(layout, ptr, "rotation", 0, NULL, ICON_NONE);
2388         uiItemR(layout, ptr, "mask_type", 0, NULL, ICON_NONE);
2389 }
2390
2391 static void node_composit_buts_viewer_but(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
2392 {
2393         uiLayout *col;
2394         
2395         uiItemR(layout, ptr, "tile_order", 0, NULL, ICON_NONE);
2396         if (RNA_enum_get(ptr, "tile_order") == 0) {
2397                 col = uiLayoutColumn(layout, 1);
2398                 uiItemR(col, ptr, "center_x", 0, NULL, ICON_NONE);
2399                 uiItemR(col, ptr, "center_y", 0, NULL, ICON_NONE);
2400         }
2401 }
2402
2403 static void node_composit_buts_mask(uiLayout *layout, bContext *C, PointerRNA *ptr)
2404 {
2405         uiTemplateID(layout, C, ptr, "mask", NULL, NULL, NULL);
2406 }
2407
2408 /* only once called */
2409 static void node_composit_set_butfunc(bNodeType *ntype)
2410 {
2411         switch (ntype->type) {
2412                 /* case NODE_GROUP:      note, typeinfo for group is generated... see "XXX ugly hack" */
2413
2414                 case CMP_NODE_IMAGE:
2415                         ntype->uifunc = node_composit_buts_image;
2416                         break;
2417                 case CMP_NODE_R_LAYERS:
2418                         ntype->uifunc = node_composit_buts_renderlayers;
2419                         break;
2420                 case CMP_NODE_NORMAL:
2421                         ntype->uifunc = node_buts_normal;
2422                         break;
2423                 case CMP_NODE_CURVE_VEC:
2424                         ntype->uifunc = node_buts_curvevec;
2425                         break;
2426                 case CMP_NODE_CURVE_RGB:
2427                         ntype->uifunc = node_buts_curvecol;
2428                         break;
2429                 case CMP_NODE_VALUE:
2430                         ntype->uifunc = node_buts_value;
2431                         break;
2432                 case CMP_NODE_RGB:
2433                         ntype->uifunc = node_buts_rgb;
2434                         break;
2435                 case CMP_NODE_FLIP:
2436                         ntype->uifunc = node_composit_buts_flip;
2437                         break;
2438                 case CMP_NODE_SPLITVIEWER:
2439                         ntype->uifunc = node_composit_buts_splitviewer;
2440                         break;
2441                 case CMP_NODE_MIX_RGB:
2442                         ntype->uifunc = node_buts_mix_rgb;
2443                         break;
2444                 case CMP_NODE_VALTORGB:
2445                         ntype->uifunc = node_buts_colorramp;
2446                         break;
2447                 case CMP_NODE_CROP:
2448                         ntype->uifunc = node_composit_buts_crop;
2449                         break;
2450                 case CMP_NODE_BLUR:
2451                         ntype->uifunc = node_composit_buts_blur;
2452                         break;
2453                 case CMP_NODE_DBLUR:
2454                         ntype->uifunc = node_composit_buts_dblur;
2455                         break;
2456                 case CMP_NODE_BILATERALBLUR:
2457                         ntype->uifunc = node_composit_buts_bilateralblur;
2458                         break;
2459                 case CMP_NODE_DEFOCUS:
2460                         ntype->uifunc = node_composit_buts_defocus;
2461                         break;
2462                 case CMP_NODE_GLARE:
2463                         ntype->uifunc = node_composit_buts_glare;
2464                         break;
2465                 case CMP_NODE_TONEMAP:
2466                         ntype->uifunc = node_composit_buts_tonemap;
2467                         break;
2468                 case CMP_NODE_LENSDIST:
2469                         ntype->uifunc = node_composit_buts_lensdist;
2470                         break;
2471                 case CMP_NODE_VECBLUR:
2472                         ntype->uifunc = node_composit_buts_vecblur;
2473                         break;
2474                 case CMP_NODE_FILTER:
2475                         ntype->uifunc = node_composit_buts_filter;
2476                         break;
2477                 case CMP_NODE_MAP_VALUE:
2478                         ntype->uifunc = node_composit_buts_map_value;
2479                         break;
2480                 case CMP_NODE_TIME:
2481                         ntype->uifunc = node_buts_time;
2482                         break;
2483                 case CMP_NODE_ALPHAOVER:
2484                         ntype->uifunc = node_composit_buts_alphaover;
2485                         break;
2486                 case CMP_NODE_HUE_SAT:
2487                         ntype->uifunc = node_composit_buts_hue_sat;
2488                         break;
2489                 case CMP_NODE_TEXTURE:
2490                         ntype->uifunc = node_buts_texture;
2491                         break;
2492                 case CMP_NODE_DILATEERODE:
2493                         ntype->uifunc = node_composit_buts_dilateerode;
2494                         break;
2495                 case CMP_NODE_OUTPUT_FILE:
2496                         ntype->uifunc = node_composit_buts_file_output;
2497                         ntype->uifuncbut = node_composit_buts_file_output_details;
2498                         ntype->drawinputfunc = node_draw_input_file_output;
2499                         break;
2500                 case CMP_NODE_DIFF_MATTE:
2501                         ntype->uifunc = node_composit_buts_diff_matte;
2502                         break;
2503                 case CMP_NODE_DIST_MATTE:
2504                         ntype->uifunc = node_composit_buts_distance_matte;
2505                         break;
2506                 case CMP_NODE_COLOR_SPILL:
2507                         ntype->uifunc = node_composit_buts_color_spill;
2508                         break;
2509                 case CMP_NODE_CHROMA_MATTE:
2510                         ntype->uifunc = node_composit_buts_chroma_matte;
2511                         break;
2512                 case CMP_NODE_COLOR_MATTE:
2513                         ntype->uifunc = node_composit_buts_color_matte;
2514                         break;
2515                 case CMP_NODE_SCALE:
2516                         ntype->uifunc = node_composit_buts_scale;
2517                         break;
2518                 case CMP_NODE_ROTATE:
2519                         ntype->uifunc = node_composit_buts_rotate;
2520                         break;
2521                 case CMP_NODE_CHANNEL_MATTE:
2522                         ntype->uifunc = node_composit_buts_channel_matte;
2523                         break;
2524                 case CMP_NODE_LUMA_MATTE:
2525                         ntype->uifunc = node_composit_buts_luma_matte;
2526                         break;
2527                 case CMP_NODE_MAP_UV:
2528                         ntype->uifunc = node_composit_buts_map_uv;
2529                         break;
2530                 case CMP_NODE_ID_MASK:
2531                         ntype->uifunc = node_composit_buts_id_mask;
2532                         break;
2533                 case CMP_NODE_DOUBLEEDGEMASK:
2534                         ntype->uifunc = node_composit_buts_double_edge_mask;
2535                         break;
2536                 case CMP_NODE_MATH:
2537                         ntype->uifunc = node_buts_math;
2538                         break;
2539                 case CMP_NODE_INVERT:
2540                         ntype->uifunc = node_composit_buts_invert;
2541                         break;
2542                 case CMP_NODE_PREMULKEY:
2543                         ntype->uifunc = node_composit_buts_premulkey;
2544                         break;
2545                 case CMP_NODE_VIEW_LEVELS:
2546                         ntype->uifunc = node_composit_buts_view_levels;
2547                         break;
2548                 case CMP_NODE_COLORBALANCE:
2549                         ntype->uifunc = node_composit_buts_colorbalance;
2550                         ntype->uifuncbut = node_composit_buts_colorbalance_but;
2551                         break;
2552                 case CMP_NODE_HUECORRECT:
2553                         ntype->uifunc = node_composit_buts_huecorrect;
2554                         break;
2555                 case CMP_NODE_ZCOMBINE:
2556                         ntype->uifunc = node_composit_buts_zcombine;
2557                         break;
2558                 case CMP_NODE_COMBYCCA:
2559                 case CMP_NODE_SEPYCCA:
2560                         ntype->uifunc = node_composit_buts_ycc;
2561                         break;
2562                 case CMP_NODE_MOVIECLIP:
2563                         ntype->uifunc = node_composit_buts_movieclip;
2564                         break;
2565                 case CMP_NODE_STABILIZE2D:
2566                         ntype->uifunc = node_composit_buts_stabilize2d;
2567                         break;
2568                 case CMP_NODE_TRANSFORM:
2569                         ntype->uifunc = node_composit_buts_transform;
2570                         break;
2571                 case CMP_NODE_MOVIEDISTORTION:
2572                         ntype->uifunc = node_composit_buts_moviedistortion;
2573                         break;
2574                 case CMP_NODE_COLORCORRECTION:
2575                         ntype->uifunc = node_composit_buts_colorcorrection;
2576                         ntype->uifuncbut = node_composit_buts_colorcorrection_but;
2577                         break;
2578                 case CMP_NODE_SWITCH:
2579                         ntype->uifunc = node_composit_buts_switch;
2580                         break;
2581                 case CMP_NODE_MASK_BOX:
2582                         ntype->uifunc = node_composit_buts_boxmask;
2583                         ntype->uibackdropfunc = node_composit_backdrop_boxmask;
2584                         break;
2585                 case CMP_NODE_MASK_ELLIPSE:
2586                         ntype->uifunc = node_composit_buts_ellipsemask;
2587                         ntype->uibackdropfunc = node_composit_backdrop_ellipsemask;
2588                         break;
2589                 case CMP_NODE_BOKEHIMAGE:
2590                         ntype->uifunc = node_composit_buts_bokehimage;
2591                         break;
2592                 case CMP_NODE_VIEWER:
2593                         ntype->uifunc = NULL;
2594                         ntype->uifuncbut = node_composit_buts_viewer_but;
2595                         ntype->uibackdropfunc = node_composit_backdrop_viewer;
2596                         break;
2597                 case CMP_NODE_MASK:
2598                         ntype->uifunc= node_composit_buts_mask;
2599                         break;
2600                 default:
2601                         ntype->uifunc = NULL;
2602         }
2603 }
2604
2605 /* ****************** BUTTON CALLBACKS FOR TEXTURE NODES ***************** */
2606
2607 static void node_texture_buts_bricks(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
2608 {
2609         uiLayout *col;
2610         
2611         col = uiLayoutColumn(layout, 1);
2612         uiItemR(col, ptr, "offset", 0, IFACE_("Offset"), ICON_NONE);
2613         uiItemR(col, ptr, "offset_frequency", 0, IFACE_("Frequency"), ICON_NONE);
2614         
2615         col = uiLayoutColumn(layout, 1);
2616         uiItemR(col, ptr, "squash", 0, IFACE_("Squash"), ICON_NONE);
2617         uiItemR(col, ptr, "squash_frequency", 0, IFACE_("Frequency"), ICON_NONE);
2618 }
2619
2620 static void node_texture_buts_proc(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
2621 {
2622         PointerRNA tex_ptr;
2623         bNode *node = ptr->data;
2624         ID *id = ptr->id.data;
2625         Tex *tex = (Tex *)node->storage;
2626         uiLayout *col, *row;
2627         
2628         RNA_pointer_create(id, &RNA_Texture, tex, &tex_ptr);
2629
2630         col = uiLayoutColumn(layout, 0);
2631
2632         switch (tex->type) {
2633                 case TEX_BLEND:
2634                         uiItemR(col, &tex_ptr, "progression", 0, "", ICON_NONE);
2635                         row = uiLayoutRow(col, 0);
2636                         uiItemR(row, &tex_ptr, "use_flip_axis", UI_ITEM_R_EXPAND, NULL, ICON_NONE);
2637                         break;
2638
2639                 case TEX_MARBLE:
2640                         row = uiLayoutRow(col, 0);
2641                         uiItemR(row, &tex_ptr, "marble_type", UI_ITEM_R_EXPAND, NULL, ICON_NONE);
2642                         row = uiLayoutRow(col, 0);
2643                         uiItemR(row, &tex_ptr, "noise_type", UI_ITEM_R_EXPAND, NULL, ICON_NONE);
2644                         row = uiLayoutRow(col, 0);
2645                         uiItemR(row, &tex_ptr, "noise_basis", 0, "", ICON_NONE);
2646                         row = uiLayoutRow(col, 0);
2647                         uiItemR(row, &tex_ptr, "noise_basis_2", UI_ITEM_R_EXPAND, NULL, ICON_NONE);
2648                         break;
2649
2650                 case TEX_MAGIC:
2651                         uiItemR(col, &tex_ptr, "noise_depth", 0, NULL, ICON_NONE);
2652                         break;
2653
2654                 case TEX_STUCCI:
2655                         row = uiLayoutRow(col, 0);
2656                         uiItemR(row, &tex_ptr, "stucci_type", UI_ITEM_R_EXPAND, NULL, ICON_NONE);
2657                         row = uiLayoutRow(col, 0);
2658                         uiItemR(row, &tex_ptr, "noise_type", UI_ITEM_R_EXPAND, NULL, ICON_NONE);
2659                         uiItemR(col, &tex_ptr, "noise_basis", 0, "", ICON_NONE);
2660                         break;
2661
2662                 case TEX_WOOD:
2663                         uiItemR(col, &tex_ptr, "noise_basis", 0, "", ICON_NONE);
2664                         uiItemR(col, &tex_ptr, "wood_type", 0, "", ICON_NONE);
2665                         row = uiLayoutRow(col, 0);
2666                         uiItemR(row, &tex_ptr, "noise_basis_2", UI_ITEM_R_EXPAND, NULL, ICON_NONE);
2667                         row = uiLayoutRow(col, 0);
2668                         uiLayoutSetActive(row, !(RNA_enum_get(&tex_ptr, "wood_type") == TEX_BAND || RNA_enum_get(&tex_ptr, "wood_type") == TEX_RING));
2669                         uiItemR(row, &tex_ptr, "noise_type", UI_ITEM_R_EXPAND, NULL, ICON_NONE);
2670                         break;
2671                         
2672                 case TEX_CLOUDS:
2673                         uiItemR(col, &tex_ptr, "noise_basis", 0, "", ICON_NONE);
2674                         row = uiLayoutRow(col, 0);
2675                         uiItemR(row, &tex_ptr, "cloud_type", UI_ITEM_R_EXPAND, NULL, ICON_NONE);
2676                         row = uiLayoutRow(col, 0);
2677                         uiItemR(row, &tex_ptr, "noise_type", UI_ITEM_R_EXPAND, NULL, ICON_NONE);
2678                         uiItemR(col, &tex_ptr, "noise_depth", UI_ITEM_R_EXPAND, IFACE_("Depth"), ICON_NONE);
2679                         break;
2680                         
2681                 case TEX_DISTNOISE:
2682                         uiItemR(col, &tex_ptr, "noise_basis", 0, "", ICON_NONE);
2683                         uiItemR(col, &tex_ptr, "noise_distortion", 0, "", ICON_NONE);
2684                         break;
2685
2686                 case TEX_MUSGRAVE:
2687                         uiItemR(col, &tex_ptr, "musgrave_type", 0, "", ICON_NONE);
2688                         uiItemR(col, &tex_ptr, "noise_basis", 0, "", ICON_NONE);
2689                         break;
2690                 case TEX_VORONOI:
2691                         uiItemR(col, &tex_ptr, "distance_metric", 0, "", ICON_NONE);
2692                         if (tex->vn_distm == TEX_MINKOVSKY) {
2693                                 uiItemR(col, &tex_ptr, "minkovsky_exponent", 0, NULL, ICON_NONE);
2694                         }
2695                         uiItemR(col, &tex_ptr, "color_mode", 0, "", ICON_NONE);
2696                         break;
2697         }
2698 }
2699
2700 static void node_texture_buts_image(uiLayout *layout, bContext *C, PointerRNA *ptr)
2701 {
2702         uiTemplateID(layout, C, ptr, "image", NULL, "IMAGE_OT_open", NULL);
2703 }
2704
2705 static void node_texture_buts_output(uiLayout *layout, bContext *UNUSED(C), PointerRNA *ptr)
2706 {
2707         uiItemR(layout, ptr, "filepath", 0, "", ICON_NONE);
2708 }
2709
2710 /* only once called */
2711 static void node_texture_set_butfunc(bNodeType *ntype)
2712 {
2713         if (ntype->type >= TEX_NODE_PROC && ntype->type < TEX_NODE_PROC_MAX) {
2714                 ntype->uifunc = node_texture_buts_proc;
2715         }
2716         else {
2717                 switch (ntype->type) {
2718
2719                         case TEX_NODE_MATH:
2720                                 ntype->uifunc = node_buts_math;
2721                                 break;
2722
2723                         case TEX_NODE_MIX_RGB:
2724                                 ntype->uifunc = node_buts_mix_rgb;
2725                                 break;
2726
2727                         case TEX_NODE_VALTORGB:
2728                                 ntype->uifunc = node_buts_colorramp;
2729                                 break;
2730
2731                         case TEX_NODE_CURVE_RGB:
2732                                 ntype->uifunc = node_buts_curvecol;
2733                                 break;
2734
2735                         case TEX_NODE_CURVE_TIME:
2736                                 ntype->uifunc = node_buts_time;
2737                                 break;
2738
2739                         case TEX_NODE_TEXTURE:
2740                                 ntype->uifunc = node_buts_texture;
2741                                 break;
2742
2743                         case TEX_NODE_BRICKS:
2744                                 ntype->uifunc = node_texture_buts_bricks;
2745                                 break;
2746
2747                         case TEX_NODE_IMAGE:
2748                                 ntype->uifunc = node_texture_buts_image;
2749                                 break;
2750
2751                         case TEX_NODE_OUTPUT:
2752                                 ntype->uifunc = node_texture_buts_output;
2753                                 break;
2754                 }
2755         }
2756 }
2757
2758 /* ******* init draw callbacks for all tree types, only called in usiblender.c, once ************* */
2759
2760 void ED_init_node_butfuncs(void)
2761 {
2762         bNodeTreeType *treetype;
2763         bNodeType *ntype;
2764         bNodeSocketType *stype;
2765         int i;
2766         
2767         /* node type ui functions */
2768         for (i = 0; i < NUM_NTREE_TYPES; ++i) {
2769                 treetype = ntreeGetType(i);
2770                 if (treetype) {
2771                         for (ntype = treetype->node_types.first; ntype; ntype = ntype->next) {
2772                                 /* default ui functions */
2773                                 ntype->drawfunc = node_draw_default;
2774                                 ntype->drawupdatefunc = node_update_default;
2775                                 ntype->select_area_func = node_select_area_default;
2776                                 ntype->tweak_area_func = node_tweak_area_default;
2777                                 ntype->uifunc = NULL;
2778                                 ntype->uifuncbut = NULL;
2779                                 ntype->drawinputfunc = node_draw_input_default;
2780                                 ntype->drawoutputfunc = node_draw_output_default;
2781                                 ntype->resize_area_func = node_resize_area_default;
2782                                 
2783                                 node_common_set_butfunc(ntype);
2784                                 
2785                                 switch (i) {
2786                                         case NTREE_COMPOSIT:
2787                                                 node_composit_set_butfunc(ntype);
2788                                                 break;
2789                                         case NTREE_SHADER:
2790                                                 node_shader_set_butfunc(ntype);
2791                                                 break;
2792                                         case NTREE_TEXTURE:
2793                                                 node_texture_set_butfunc(ntype);
2794                                                 break;
2795                                 }
2796                         }
2797                 }
2798         }
2799         
2800         /* socket type ui functions */
2801         for (i = 0; i < NUM_SOCKET_TYPES; ++i) {
2802                 stype = ntreeGetSocketType(i);
2803                 if (stype) {
2804                         switch (stype->type) {
2805                                 case SOCK_FLOAT:
2806                                 case SOCK_INT:
2807                                 case SOCK_BOOLEAN:
2808                                         stype->buttonfunc = node_socket_button_default;
2809                                         break;
2810                                 case SOCK_VECTOR:
2811                                         stype->buttonfunc = node_socket_button_components;
2812                                         break;
2813                                 case SOCK_RGBA:
2814                                         stype->buttonfunc = node_socket_button_color;
2815                                         break;
2816                                 case SOCK_SHADER:
2817                                         stype->buttonfunc = node_socket_button_label;
2818                                         break;
2819                                 default:
2820                                         stype->buttonfunc = NULL;
2821                         }
2822                 }
2823         }
2824 }
2825
2826 /* ************** Generic drawing ************** */
2827
2828 void draw_nodespace_back_pix(ARegion *ar, SpaceNode *snode, int color_manage)
2829 {
2830         
2831         if ((snode->flag & SNODE_BACKDRAW) && snode->treetype == NTREE_COMPOSIT) {
2832                 Image *ima = BKE_image_verify_viewer(IMA_TYPE_COMPOSITE, "Viewer Node");
2833                 void *lock;
2834                 ImBuf *ibuf = BKE_image_acquire_ibuf(ima, NULL, &lock);
2835                 if (ibuf) {
2836                         float x, y; 
2837                         
2838                         glMatrixMode(GL_PROJECTION);
2839                         glPushMatrix();
2840                         glMatrixMode(GL_MODELVIEW);
2841                         glPushMatrix();
2842
2843                         /* keep this, saves us from a version patch */
2844                         if (snode->zoom == 0.0f) snode->zoom = 1.0f;
2845                         
2846                         /* somehow the offset has to be calculated inverse */
2847                         
2848                         glaDefine2DArea(&ar->winrct);
2849                         /* ortho at pixel level curarea */
2850                         wmOrtho2(-0.375, ar->winx - 0.375, -0.375, ar->winy - 0.375);
2851                         
2852                         x = (ar->winx - snode->zoom * ibuf->x) / 2 + snode->xof;
2853                         y = (ar->winy - snode->zoom * ibuf->y) / 2 + snode->yof;
2854                         
2855                         if (!ibuf->rect) {
2856                                 if (color_manage)
2857                                         ibuf->profile = IB_PROFILE_LINEAR_RGB;
2858                                 else
2859                                         ibuf->profile = IB_PROFILE_NONE;
2860                                 IMB_rect_from_float(ibuf);
2861                         }
2862
2863                         if (ibuf->rect) {
2864                                 if (snode->flag & SNODE_SHOW_ALPHA) {
2865                                         glPixelZoom(snode->zoom, snode->zoom);
2866                                         /* swap bytes, so alpha is most significant one, then just draw it as luminance int */
2867                                         if (ENDIAN_ORDER == B_ENDIAN)
2868                                                 glPixelStorei(GL_UNPACK_SWAP_BYTES, 1);
2869                                         
2870                                         glaDrawPixelsSafe(x, y, ibuf->x, ibuf->y, ibuf->x, GL_LUMINANCE, GL_UNSIGNED_INT, ibuf->rect);
2871                                         
2872                                         glPixelStorei(GL_UNPACK_SWAP_BYTES, 0);
2873                                         glPixelZoom(1.0f, 1.0f);
2874                                 }
2875                                 else if (snode->flag & SNODE_USE_ALPHA) {
2876                                         glEnable(GL_BLEND);
2877                                         glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
2878                                         glPixelZoom(snode->zoom, snode->zoom);
2879                                         
2880                                         glaDrawPixelsSafe(x, y, ibuf->x, ibuf->y, ibuf->x, GL_RGBA, GL_UNSIGNED_BYTE, ibuf->rect);
2881                                         
2882                                         glPixelZoom(1.0f, 1.0f);
2883                                         glDisable(GL_BLEND);
2884                                 }
2885                                 else {
2886                                         glPixelZoom(snode->zoom, snode->zoom);
2887                                         
2888                                         glaDrawPixelsSafe(x, y, ibuf->x, ibuf->y, ibuf->x, GL_RGBA, GL_UNSIGNED_BYTE, ibuf->rect);
2889                                         
2890                                         glPixelZoom(1.0f, 1.0f);
2891                                 }
2892                         }
2893
2894                         /// @note draw selected info on backdrop
2895                         if (snode->edittree) {
2896                                 bNode *node = snode->edittree->nodes.first;
2897                                 while (node) {
2898                                         if (node->flag & NODE_SELECT) {
2899                                                 if (node->typeinfo->uibackdropfunc) {
2900                                                         node->typeinfo->uibackdropfunc(snode, ibuf, node, x, y);
2901                                                 }
2902                                         }
2903                                         node = node->next;
2904                                 }
2905                         }
2906                         
2907                         glMatrixMode(GL_PROJECTION);
2908                         glPopMatrix();
2909                         glMatrixMode(GL_MODELVIEW);
2910                         glPopMatrix();
2911                 }
2912
2913                 BKE_image_release_ibuf(ima, lock);
2914         }
2915 }
2916
2917 #if 0
2918 /* note: needs to be userpref or opengl profile option */
2919 static void draw_nodespace_back_tex(ScrArea *sa, SpaceNode *snode)
2920 {
2921
2922         draw_nodespace_grid(snode);
2923         
2924         if (snode->flag & SNODE_BACKDRAW) {
2925                 Image *ima = BKE_image_verify_viewer(IMA_TYPE_COMPOSITE, "Viewer Node");
2926                 ImBuf *ibuf = BKE_image_get_ibuf(ima, NULL);
2927                 if (ibuf) {
2928                         int x, y;
2929                         float zoom = 1.0;
2930
2931                         glMatrixMode(GL_PROJECTION);
2932                         glPushMatrix();
2933                         glMatrixMode(GL_MODELVIEW);
2934                         glPushMatrix();
2935                         
2936                         glaDefine2DArea(&sa->winrct);
2937
2938                         if (ibuf->x > sa->winx || ibuf->y > sa->winy) {
2939                                 float zoomx, zoomy;
2940                                 zoomx = (float)sa->winx / ibuf->x;
2941                                 zoomy = (float)sa->winy / ibuf->y;
2942                                 zoom = MIN2(zoomx, zoomy);
2943                         }
2944                         
2945                         x = (sa->winx - zoom * ibuf->x) / 2 + snode->xof;
2946                         y = (sa->winy - zoom * ibuf->y) / 2 + snode->yof;
2947
2948                         glPixelZoom(zoom, zoom);
2949
2950                         glColor4f(1.0, 1.0, 1.0, 1.0);
2951                         if (ibuf->rect)
2952                                 glaDrawPixelsTex(x, y, ibuf->x, ibuf->y, GL_UNSIGNED_BYTE, ibuf->rect);
2953                         else if (ibuf->channels == 4)
2954                                 glaDrawPixelsTex(x, y, ibuf->x, ibuf->y, GL_FLOAT, ibuf->rect_float);
2955
2956                         glPixelZoom(1.0, 1.0);
2957
2958                         glMatrixMode(GL_PROJECTION);
2959                         glPopMatrix();
2960                         glMatrixMode(GL_MODELVIEW);
2961                         glPopMatrix();
2962                 }
2963         }
2964 }
2965 #endif
2966
2967 /* if v2d not NULL, it clips and returns 0 if not visible */
2968 int node_link_bezier_points(View2D *v2d, SpaceNode *snode, bNodeLink *link, float coord_array[][2], int resol)
2969 {
2970         float dist, vec[4][2];
2971         float deltax, deltay;
2972         int toreroute, fromreroute;
2973         /* in v0 and v3 we put begin/end points */
2974         if (link->fromsock) {
2975                 vec[0][0] = link->fromsock->locx;
2976                 vec[0][1] = link->fromsock->locy;
2977                 fromreroute = (link->fromnode && link->fromnode->type == NODE_REROUTE);
2978         }
2979         else {
2980                 if (snode == NULL) return 0;
2981                 vec[0][0] = snode->mx;
2982                 vec[0][1] = snode->my;
2983                 fromreroute = 0;
2984         }
2985         if (link->tosock) {
2986                 vec[3][0] = link->tosock->locx;
2987                 vec[3][1] = link->tosock->locy;
2988                 toreroute = (link->tonode && link->tonode->type == NODE_REROUTE);
2989         }
2990         else {
2991                 if (snode == NULL) return 0;
2992                 vec[3][0] = snode->mx;
2993                 vec[3][1] = snode->my;
2994                 toreroute = 0;
2995         }
2996
2997         dist = UI_GetThemeValue(TH_NODE_CURVING) * 0.10f * ABS(vec[0][0] - vec[3][0]);
2998         deltax = vec[3][0] - vec[0][0];
2999         deltay = vec[3][1] - vec[0][1];
3000         /* check direction later, for top sockets */
3001         if (fromreroute) {
3002                 if (ABS(deltax)>ABS(deltay)) {
3003                         vec[1][1]= vec[0][1];
3004                         vec[1][0]= vec[0][0]+(deltax>0?dist:-dist);
3005                 } else {
3006                         vec[1][0]= vec[0][0];
3007                         vec[1][1]= vec[0][1]+(deltay>0?dist:-dist);
3008                 }
3009         } else {
3010                 vec[1][0] = vec[0][0] + dist;
3011                 vec[1][1] = vec[0][1];
3012         }
3013         if (toreroute) {
3014                 if (ABS(deltax)>ABS(deltay)) {
3015                         vec[2][1]= vec[3][1];
3016                         vec[2][0]= vec[3][0]+ (deltax>0?-dist:dist);
3017                 } else {
3018                         vec[2][0]= vec[3][0];
3019                         vec[2][1]= vec[3][1]+(deltay>0?-dist:dist);
3020                 }
3021
3022         } else {
3023                 vec[2][0] = vec[3][0] - dist;
3024                 vec[2][1] = vec[3][1];
3025         }
3026         if (v2d && MIN4(vec[0][0], vec[1][0], vec[2][0], vec[3][0]) > v2d->cur.xmax) ;  /* clipped */
3027         else if (v2d && MAX4(vec[0][0], vec[1][0], vec[2][0], vec[3][0]) < v2d->cur.xmin) ;  /* clipped */
3028         else {
3029                 
3030                 /* always do all three, to prevent data hanging around */
3031                 BKE_curve_forward_diff_bezier(vec[0][0], vec[1][0], vec[2][0], vec[3][0], coord_array[0], resol, sizeof(float) * 2);
3032                 BKE_curve_forward_diff_bezier(vec[0][1], vec[1][1], vec[2][1], vec[3][1], coord_array[0] + 1, resol, sizeof(float) * 2);
3033                 
3034                 return 1;
3035         }
3036         return 0;
3037 }
3038
3039 #define LINK_RESOL  24
3040 #define LINK_ARROW      12      /* position of arrow on the link, LINK_RESOL/2 */
3041 #define ARROW_SIZE 7
3042 void node_draw_link_bezier(View2D *v2d, SpaceNode *snode, bNodeLink *link, int th_col1, int do_shaded, int th_col2, int do_triple, int th_col3)
3043 {
3044         float coord_array[LINK_RESOL + 1][2];
3045         
3046         if (node_link_bezier_points(v2d, snode, link, coord_array, LINK_RESOL)) {
3047                 float dist, spline_step = 0.0f;
3048                 int i;
3049                 int drawarrow;
3050                 /* store current linewidth */
3051                 float linew;
3052                 float arrow[2], arrow1[2], arrow2[2];
3053                 glGetFloatv(GL_LINE_WIDTH, &linew);
3054                 
3055                 /* we can reuse the dist variable here to increment the GL curve eval amount*/
3056                 dist = 1.0f / (float)LINK_RESOL;
3057                 
3058                 glEnable(GL_LINE_SMOOTH);
3059                 
3060                 drawarrow = (link->tonode && (link->tonode->type == NODE_REROUTE)) && (link->fromnode && (link->fromnode->type == NODE_REROUTE));
3061                 if (drawarrow) {
3062                         // draw arrow in line segment LINK_ARROW
3063                         float dx, dy, len;
3064                         dx = coord_array[LINK_ARROW][0]-coord_array[LINK_ARROW-1][0];
3065                         dy = coord_array[LINK_ARROW][1]-coord_array[LINK_ARROW-1][1];
3066                         len = sqrtf(dx*dx+dy*dy);
3067                         dx = dx /len*ARROW_SIZE;
3068                         dy = dy /len*ARROW_SIZE;
3069                         arrow1[0] = coord_array[LINK_ARROW][0]-dx+dy;
3070                         arrow1[1] = coord_array[LINK_ARROW][1]-dy-dx;
3071                         arrow2[0] = coord_array[LINK_ARROW][0]-dx-dy;
3072                         arrow2[1] = coord_array[LINK_ARROW][1]-dy+dx;
3073                         arrow[0] = coord_array[LINK_ARROW][0];
3074                         arrow[1] = coord_array[LINK_ARROW][1];
3075                 }
3076                 if (do_triple) {
3077                         UI_ThemeColorShadeAlpha(th_col3, -80, -120);
3078                         glLineWidth(4.0f);
3079                         
3080                         glBegin(GL_LINE_STRIP);
3081                         for (i = 0; i <= LINK_RESOL; i++) {
3082                                 glVertex2fv(coord_array[i]);
3083                         }
3084                         glEnd();
3085                         if (drawarrow) {
3086                                 glBegin(GL_LINE_STRIP);
3087                                 glVertex2fv(arrow1);
3088                                 glVertex2fv(arrow);
3089                                 glVertex2fv(arrow);
3090                                 glVertex2fv(arrow2);
3091                                 glEnd();
3092              &nb