doxygen: prevent GPL license block from being parsed as doxygen comment.
[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 #include "../SHD_util.h"
31
32
33 /* **************** CURVE VEC  ******************** */
34 static bNodeSocketType sh_node_curve_vec_in[]= {
35         {       SOCK_VALUE, 0, "Fac",   1.0f, 0.0f, 1.0f, 1.0f, 0.0f, 1.0f},
36         {       SOCK_VECTOR, 1, "Vector",       0.0f, 0.0f, 0.0f, 1.0f, -1.0f, 1.0f},
37         {       -1, 0, ""       }
38 };
39
40 static bNodeSocketType sh_node_curve_vec_out[]= {
41         {       SOCK_VECTOR, 0, "Vector",       0.0f, 0.0f, 1.0f, 1.0f, -1.0f, 1.0f},
42         {       -1, 0, ""       }
43 };
44
45 static void node_shader_exec_curve_vec(void *UNUSED(data), bNode *node, bNodeStack **in, bNodeStack **out)
46 {
47         float vec[3];
48         
49         /* stack order input:  vec */
50         /* stack order output: vec */
51         nodestack_get_vec(vec, SOCK_VECTOR, in[1]);
52         curvemapping_evaluate3F(node->storage, out[0]->vec, vec);
53 }
54
55 static void node_shader_init_curve_vec(bNode* node)
56 {
57    node->storage= curvemapping_add(3, -1.0f, -1.0f, 1.0f, 1.0f);
58 }
59
60 static int gpu_shader_curve_vec(GPUMaterial *mat, bNode *node, GPUNodeStack *in, GPUNodeStack *out)
61 {
62         float *array;
63         int size;
64
65         curvemapping_table_RGBA(node->storage, &array, &size);
66         return GPU_stack_link(mat, "curves_vec", in, out, GPU_texture(size, array));
67 }
68
69 void register_node_type_sh_curve_vec(ListBase *lb)
70 {
71         static bNodeType ntype;
72
73         node_type_base(&ntype, SH_NODE_CURVE_VEC, "Vector Curves", NODE_CLASS_OP_VECTOR, NODE_OPTIONS,
74                 sh_node_curve_vec_in, sh_node_curve_vec_out);
75         node_type_size(&ntype, 200, 140, 320);
76         node_type_init(&ntype, node_shader_init_curve_vec);
77         node_type_storage(&ntype, "CurveMapping", node_free_curves, node_copy_curves);
78         node_type_exec(&ntype, node_shader_exec_curve_vec);
79         node_type_gpu(&ntype, gpu_shader_curve_vec);
80
81         nodeRegisterType(lb, &ntype);
82 }
83
84
85 /* **************** CURVE RGB  ******************** */
86 static bNodeSocketType sh_node_curve_rgb_in[]= {
87         {       SOCK_VALUE, 1, "Fac",   1.0f, 0.0f, 0.0f, 1.0f, -1.0f, 1.0f},
88         {       SOCK_RGBA, 1, "Color",  0.0f, 0.0f, 0.0f, 1.0f, -1.0f, 1.0f},
89         {       -1, 0, ""       }
90 };
91
92 static bNodeSocketType sh_node_curve_rgb_out[]= {
93         {       SOCK_RGBA, 0, "Color",  0.0f, 0.0f, 1.0f, 1.0f, -1.0f, 1.0f},
94         {       -1, 0, ""       }
95 };
96
97 static void node_shader_exec_curve_rgb(void *UNUSED(data), bNode *node, bNodeStack **in, bNodeStack **out)
98 {
99         float vec[3];
100         
101         /* stack order input:  vec */
102         /* stack order output: vec */
103         nodestack_get_vec(vec, SOCK_VECTOR, in[1]);
104         curvemapping_evaluateRGBF(node->storage, out[0]->vec, vec);
105         if(in[0]->vec[0] != 1.0f) {
106                 interp_v3_v3v3(out[0]->vec, vec, out[0]->vec, *in[0]->vec);
107         }
108 }
109
110 static void node_shader_init_curve_rgb(bNode *node)
111 {
112    node->storage= curvemapping_add(4, 0.0f, 0.0f, 1.0f, 1.0f);
113 }
114
115 static int gpu_shader_curve_rgb(GPUMaterial *mat, bNode *node, GPUNodeStack *in, GPUNodeStack *out)
116 {
117         float *array;
118         int size;
119         curvemapping_table_RGBA(node->storage, &array, &size);
120         return GPU_stack_link(mat, "curves_rgb", in, out, GPU_texture(size, array));
121 }
122
123 void register_node_type_sh_curve_rgb(ListBase *lb)
124 {
125         static bNodeType ntype;
126
127         node_type_base(&ntype, SH_NODE_CURVE_RGB, "RGB Curves", NODE_CLASS_OP_COLOR, NODE_OPTIONS,
128                 sh_node_curve_rgb_in, sh_node_curve_rgb_out);
129         node_type_size(&ntype, 200, 140, 320);
130         node_type_init(&ntype, node_shader_init_curve_rgb);
131         node_type_storage(&ntype, "CurveMapping", node_free_curves, node_copy_curves);
132         node_type_exec(&ntype, node_shader_exec_curve_rgb);
133         node_type_gpu(&ntype, gpu_shader_curve_rgb);
134
135         nodeRegisterType(lb, &ntype);
136 }
137