converted more mixed tab/space indentations to tabs. only whitespace changes.
[blender.git] / source / blender / nodes / intern / SHD_nodes / SHD_curves.c
1 /*
2  * $Id$
3  *
4  * ***** BEGIN GPL LICENSE BLOCK *****
5  *
6  * This program is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU General Public License
8  * as published by the Free Software Foundation; either version 2
9  * of the License, or (at your option) any later version. 
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software Foundation,
18  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
19  *
20  * The Original Code is Copyright (C) 2005 Blender Foundation.
21  * All rights reserved.
22  *
23  * The Original Code is: all of this file.
24  *
25  * Contributor(s): none yet.
26  *
27  * ***** END GPL LICENSE BLOCK *****
28  */
29
30 /** \file blender/nodes/intern/SHD_nodes/SHD_curves.c
31  *  \ingroup shdnodes
32  */
33
34
35 #include "../SHD_util.h"
36
37
38 /* **************** CURVE VEC  ******************** */
39 static bNodeSocketType sh_node_curve_vec_in[]= {
40         {       SOCK_VALUE, 0, "Fac",   1.0f, 0.0f, 1.0f, 1.0f, 0.0f, 1.0f},
41         {       SOCK_VECTOR, 1, "Vector",       0.0f, 0.0f, 0.0f, 1.0f, -1.0f, 1.0f},
42         {       -1, 0, ""       }
43 };
44
45 static bNodeSocketType sh_node_curve_vec_out[]= {
46         {       SOCK_VECTOR, 0, "Vector",       0.0f, 0.0f, 1.0f, 1.0f, -1.0f, 1.0f},
47         {       -1, 0, ""       }
48 };
49
50 static void node_shader_exec_curve_vec(void *UNUSED(data), bNode *node, bNodeStack **in, bNodeStack **out)
51 {
52         float vec[3];
53         
54         /* stack order input:  vec */
55         /* stack order output: vec */
56         nodestack_get_vec(vec, SOCK_VECTOR, in[1]);
57         curvemapping_evaluate3F(node->storage, out[0]->vec, vec);
58 }
59
60 static void node_shader_init_curve_vec(bNode* node)
61 {
62         node->storage= curvemapping_add(3, -1.0f, -1.0f, 1.0f, 1.0f);
63 }
64
65 static int gpu_shader_curve_vec(GPUMaterial *mat, bNode *node, GPUNodeStack *in, GPUNodeStack *out)
66 {
67         float *array;
68         int size;
69
70         curvemapping_table_RGBA(node->storage, &array, &size);
71         return GPU_stack_link(mat, "curves_vec", in, out, GPU_texture(size, array));
72 }
73
74 void register_node_type_sh_curve_vec(ListBase *lb)
75 {
76         static bNodeType ntype;
77
78         node_type_base(&ntype, SH_NODE_CURVE_VEC, "Vector Curves", NODE_CLASS_OP_VECTOR, NODE_OPTIONS,
79                 sh_node_curve_vec_in, sh_node_curve_vec_out);
80         node_type_size(&ntype, 200, 140, 320);
81         node_type_init(&ntype, node_shader_init_curve_vec);
82         node_type_storage(&ntype, "CurveMapping", node_free_curves, node_copy_curves);
83         node_type_exec(&ntype, node_shader_exec_curve_vec);
84         node_type_gpu(&ntype, gpu_shader_curve_vec);
85
86         nodeRegisterType(lb, &ntype);
87 }
88
89
90 /* **************** CURVE RGB  ******************** */
91 static bNodeSocketType sh_node_curve_rgb_in[]= {
92         {       SOCK_VALUE, 1, "Fac",   1.0f, 0.0f, 0.0f, 1.0f, -1.0f, 1.0f},
93         {       SOCK_RGBA, 1, "Color",  0.0f, 0.0f, 0.0f, 1.0f, -1.0f, 1.0f},
94         {       -1, 0, ""       }
95 };
96
97 static bNodeSocketType sh_node_curve_rgb_out[]= {
98         {       SOCK_RGBA, 0, "Color",  0.0f, 0.0f, 1.0f, 1.0f, -1.0f, 1.0f},
99         {       -1, 0, ""       }
100 };
101
102 static void node_shader_exec_curve_rgb(void *UNUSED(data), bNode *node, bNodeStack **in, bNodeStack **out)
103 {
104         float vec[3];
105         
106         /* stack order input:  vec */
107         /* stack order output: vec */
108         nodestack_get_vec(vec, SOCK_VECTOR, in[1]);
109         curvemapping_evaluateRGBF(node->storage, out[0]->vec, vec);
110         if(in[0]->vec[0] != 1.0f) {
111                 interp_v3_v3v3(out[0]->vec, vec, out[0]->vec, *in[0]->vec);
112         }
113 }
114
115 static void node_shader_init_curve_rgb(bNode *node)
116 {
117         node->storage= curvemapping_add(4, 0.0f, 0.0f, 1.0f, 1.0f);
118 }
119
120 static int gpu_shader_curve_rgb(GPUMaterial *mat, bNode *node, GPUNodeStack *in, GPUNodeStack *out)
121 {
122         float *array;
123         int size;
124         curvemapping_table_RGBA(node->storage, &array, &size);
125         return GPU_stack_link(mat, "curves_rgb", in, out, GPU_texture(size, array));
126 }
127
128 void register_node_type_sh_curve_rgb(ListBase *lb)
129 {
130         static bNodeType ntype;
131
132         node_type_base(&ntype, SH_NODE_CURVE_RGB, "RGB Curves", NODE_CLASS_OP_COLOR, NODE_OPTIONS,
133                 sh_node_curve_rgb_in, sh_node_curve_rgb_out);
134         node_type_size(&ntype, 200, 140, 320);
135         node_type_init(&ntype, node_shader_init_curve_rgb);
136         node_type_storage(&ntype, "CurveMapping", node_free_curves, node_copy_curves);
137         node_type_exec(&ntype, node_shader_exec_curve_rgb);
138         node_type_gpu(&ntype, gpu_shader_curve_rgb);
139
140         nodeRegisterType(lb, &ntype);
141 }
142