replace macros with bli math functions for nodes
[blender.git] / source / blender / nodes / texture / nodes / node_texture_proc.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): Robin Allen
24  *
25  * ***** END GPL LICENSE BLOCK *****
26  */
27
28 /** \file blender/nodes/texture/nodes/node_texture_proc.c
29  *  \ingroup texnodes
30  */
31
32
33 #include "node_texture_util.h"
34 #include "NOD_texture.h"
35
36 #include "RE_shader_ext.h"
37
38 /* 
39         In this file: wrappers to use procedural textures as nodes
40 */
41
42
43 static bNodeSocketTemplate outputs_both[]= {
44         { SOCK_RGBA, 0, "Color",  1.0f, 0.0f, 0.0f, 1.0f },
45         { SOCK_VECTOR, 0, "Normal", 0.0f, 0.0f, 1.0f, 0.0f, 0.0f, 1.0f, PROP_DIRECTION },
46         { -1, 0, "" }
47 };
48 static bNodeSocketTemplate outputs_color_only[]= {
49         { SOCK_RGBA, 0, "Color" },
50         { -1, 0, "" }
51 };
52
53 /* Inputs common to all, #defined because nodes will need their own inputs too */
54 #define I 2 /* count */
55 #define COMMON_INPUTS \
56         { SOCK_RGBA, 1, "Color 1", 0.0f, 0.0f, 0.0f, 1.0f }, \
57         { SOCK_RGBA, 1, "Color 2", 1.0f, 1.0f, 1.0f, 1.0f }
58
59 /* Calls multitex and copies the result to the outputs. Called by xxx_exec, which handles inputs. */
60 static void do_proc(float *result, TexParams *p, float *col1, float *col2, char is_normal, Tex *tex, short thread)
61 {
62         TexResult texres;
63         int textype;
64         
65         if(is_normal) {
66                 texres.nor = result;
67         }
68         else
69                 texres.nor = NULL;
70         
71         textype = multitex_nodes(tex, p->co, p->dxt, p->dyt, p->osatex,
72                 &texres, thread, 0, p->shi, p->mtex);
73         
74         if(is_normal)
75                 return;
76         
77         if(textype & TEX_RGB) {
78                 copy_v4_v4(result, &texres.tr);
79         }
80         else {
81                 copy_v4_v4(result, col1);
82                 ramp_blend(MA_RAMP_BLEND, result, result+1, result+2, texres.tin, col2);
83         }
84 }
85
86 typedef void (*MapFn) (Tex *tex, bNodeStack **in, TexParams *p, short thread);
87
88 static void texfn(
89         float *result, 
90         TexParams *p,
91         bNode *node, 
92         bNodeStack **in,
93         char is_normal, 
94         MapFn map_inputs,
95         short thread)
96 {
97         Tex tex = *((Tex*)(node->storage));
98         float col1[4], col2[4];
99         tex_input_rgba(col1, in[0], p, thread);
100         tex_input_rgba(col2, in[1], p, thread);
101         
102         map_inputs(&tex, in, p, thread);
103         
104         do_proc(result, p, col1, col2, is_normal, &tex, thread);
105 }
106
107 static int count_outputs(bNode *node)
108 {
109         bNodeSocket *sock;
110         int num = 0;
111         for(sock= node->outputs.first; sock; sock= sock->next) {
112                 num++;
113         }
114         return num;
115 }
116
117 /* Boilerplate generators */
118
119 #define ProcNoInputs(name) \
120                 static void name##_map_inputs(Tex *UNUSED(tex), bNodeStack **UNUSED(in), TexParams *UNUSED(p), short UNUSED(thread)) \
121                 {}
122
123 #define ProcDef(name) \
124                 static void name##_colorfn(float *result, TexParams *p, bNode *node, bNodeStack **in, short thread)  \
125                 {                                                                                                    \
126                                 texfn(result, p, node, in, 0, &name##_map_inputs, thread);                               \
127                 }                                                                                                    \
128                 static void name##_normalfn(float *result, TexParams *p, bNode *node, bNodeStack **in, short thread) \
129                 {                                                                                                    \
130                                 texfn(result, p, node, in, 1, &name##_map_inputs, thread);                               \
131                 }                                                                                                    \
132                 static void name##_exec(void *data, bNode *node, bNodeStack **in, bNodeStack **out)                  \
133                 {                                                                                                    \
134                                 int outs = count_outputs(node);                                                              \
135                                 if(outs >= 1) tex_output(node, in, out[0], &name##_colorfn, data);                                 \
136                                 if(outs >= 2) tex_output(node, in, out[1], &name##_normalfn, data);                                \
137                 }
138
139
140 /* --- VORONOI -- */
141 static bNodeSocketTemplate voronoi_inputs[]= {
142         COMMON_INPUTS,
143         { SOCK_FLOAT, 1, "W1", 1.0f, 0.0f, 0.0f, 0.0f,   -2.0f, 2.0f, PROP_NONE },
144         { SOCK_FLOAT, 1, "W2", 0.0f, 0.0f, 0.0f, 0.0f,   -2.0f, 2.0f, PROP_NONE },
145         { SOCK_FLOAT, 1, "W3", 0.0f, 0.0f, 0.0f, 0.0f,   -2.0f, 2.0f, PROP_NONE },
146         { SOCK_FLOAT, 1, "W4", 0.0f, 0.0f, 0.0f, 0.0f,   -2.0f, 2.0f, PROP_NONE },
147         
148         { SOCK_FLOAT, 1, "iScale", 1.0f, 0.0f, 0.0f, 0.0f,    0.01f,  10.0f, PROP_UNSIGNED },
149         { SOCK_FLOAT, 1, "Size",   0.25f, 0.0f, 0.0f, 0.0f,   0.0001f, 4.0f, PROP_UNSIGNED },
150         
151         { -1, 0, "" }
152 };
153 static void voronoi_map_inputs(Tex *tex, bNodeStack **in, TexParams *p, short thread)
154 {
155         tex->vn_w1 = tex_input_value(in[I+0], p, thread);
156         tex->vn_w2 = tex_input_value(in[I+1], p, thread);
157         tex->vn_w3 = tex_input_value(in[I+2], p, thread);
158         tex->vn_w4 = tex_input_value(in[I+3], p, thread);
159         
160         tex->ns_outscale = tex_input_value(in[I+4], p, thread);
161         tex->noisesize   = tex_input_value(in[I+5], p, thread);
162 }
163 ProcDef(voronoi)
164
165 /* --- BLEND -- */
166 static bNodeSocketTemplate blend_inputs[]= {
167         COMMON_INPUTS,
168         { -1, 0, "" }
169 };
170 ProcNoInputs(blend)
171 ProcDef(blend)
172
173 /* -- MAGIC -- */
174 static bNodeSocketTemplate magic_inputs[]= {
175         COMMON_INPUTS,
176         { SOCK_FLOAT, 1, "Turbulence", 5.0f, 0.0f, 0.0f, 0.0f,   0.0f, 200.0f, PROP_UNSIGNED },
177         { -1, 0, "" }
178 };
179 static void magic_map_inputs(Tex *tex, bNodeStack **in, TexParams *p, short thread)
180 {
181         tex->turbul = tex_input_value(in[I+0], p, thread);
182 }
183 ProcDef(magic)
184
185 /* --- MARBLE --- */
186 static bNodeSocketTemplate marble_inputs[]= {
187         COMMON_INPUTS,
188         { SOCK_FLOAT, 1, "Size",       0.25f, 0.0f, 0.0f, 0.0f,   0.0001f, 2.0f, PROP_UNSIGNED },
189         { SOCK_FLOAT, 1, "Turbulence", 5.0f,  0.0f, 0.0f, 0.0f,   0.0f, 200.0f, PROP_UNSIGNED },
190         { -1, 0, "" }
191 };
192 static void marble_map_inputs(Tex *tex, bNodeStack **in, TexParams *p, short thread)
193 {
194         tex->noisesize = tex_input_value(in[I+0], p, thread);
195         tex->turbul    = tex_input_value(in[I+1], p, thread);
196 }
197 ProcDef(marble)
198
199 /* --- CLOUDS --- */
200 static bNodeSocketTemplate clouds_inputs[]= {
201         COMMON_INPUTS,
202         { SOCK_FLOAT, 1, "Size",       0.25f, 0.0f, 0.0f, 0.0f,   0.0001f, 2.0f, PROP_UNSIGNED },
203         { -1, 0, "" }
204 };
205 static void clouds_map_inputs(Tex *tex, bNodeStack **in, TexParams *p, short thread)
206 {
207         tex->noisesize = tex_input_value(in[I+0], p, thread);
208 }
209 ProcDef(clouds)
210
211 /* --- DISTORTED NOISE --- */
212 static bNodeSocketTemplate distnoise_inputs[]= {
213         COMMON_INPUTS,
214         { SOCK_FLOAT, 1, "Size",       0.25f, 0.0f, 0.0f, 0.0f,   0.0001f,  2.0f, PROP_UNSIGNED },
215         { SOCK_FLOAT, 1, "Distortion", 1.00f, 0.0f, 0.0f, 0.0f,   0.0000f, 10.0f, PROP_UNSIGNED },
216         { -1, 0, "" }
217 };
218 static void distnoise_map_inputs(Tex *tex, bNodeStack **in, TexParams *p, short thread)
219 {
220         tex->noisesize   = tex_input_value(in[I+0], p, thread);
221         tex->dist_amount = tex_input_value(in[I+1], p, thread);
222 }
223 ProcDef(distnoise)
224
225 /* --- WOOD --- */
226 static bNodeSocketTemplate wood_inputs[]= {
227         COMMON_INPUTS,
228         { SOCK_FLOAT, 1, "Size",       0.25f, 0.0f, 0.0f, 0.0f,   0.0001f, 2.0f, PROP_UNSIGNED },
229         { SOCK_FLOAT, 1, "Turbulence", 5.0f,  0.0f, 0.0f, 0.0f,   0.0f, 200.0f, PROP_UNSIGNED },
230         { -1, 0, "" }
231 };
232 static void wood_map_inputs(Tex *tex, bNodeStack **in, TexParams *p, short thread)
233 {
234         tex->noisesize = tex_input_value(in[I+0], p, thread);
235         tex->turbul    = tex_input_value(in[I+1], p, thread);
236 }
237 ProcDef(wood)
238
239 /* --- MUSGRAVE --- */
240 static bNodeSocketTemplate musgrave_inputs[]= {
241         COMMON_INPUTS,
242         { SOCK_FLOAT, 1, "H",          1.0f, 0.0f, 0.0f, 0.0f,   0.0001f, 2.0f, PROP_UNSIGNED },
243         { SOCK_FLOAT, 1, "Lacunarity", 2.0f, 0.0f, 0.0f, 0.0f,   0.0f,    6.0f, PROP_UNSIGNED },
244         { SOCK_FLOAT, 1, "Octaves",    2.0f, 0.0f, 0.0f, 0.0f,   0.0f,    8.0f, PROP_UNSIGNED },
245         
246         { SOCK_FLOAT, 1, "iScale",     1.0f,  0.0f, 0.0f, 0.0f,  0.0f,   10.0f, PROP_UNSIGNED },
247         { SOCK_FLOAT, 1, "Size",       0.25f, 0.0f, 0.0f, 0.0f,  0.0001f, 2.0f, PROP_UNSIGNED },
248         { -1, 0, "" }
249 };
250 static void musgrave_map_inputs(Tex *tex, bNodeStack **in, TexParams *p, short thread)
251 {
252         tex->mg_H          = tex_input_value(in[I+0], p, thread);
253         tex->mg_lacunarity = tex_input_value(in[I+1], p, thread);
254         tex->mg_octaves    = tex_input_value(in[I+2], p, thread);
255         tex->ns_outscale   = tex_input_value(in[I+3], p, thread);
256         tex->noisesize     = tex_input_value(in[I+4], p, thread);
257 }
258 ProcDef(musgrave)
259
260 /* --- NOISE --- */
261 static bNodeSocketTemplate noise_inputs[]= {
262         COMMON_INPUTS,
263         { -1, 0, "" }
264 };
265 ProcNoInputs(noise)
266 ProcDef(noise)
267
268 /* --- STUCCI --- */
269 static bNodeSocketTemplate stucci_inputs[]= {
270         COMMON_INPUTS,
271         { SOCK_FLOAT, 1, "Size",       0.25f, 0.0f, 0.0f, 0.0f,   0.0001f, 2.0f, PROP_UNSIGNED },
272         { SOCK_FLOAT, 1, "Turbulence", 5.0f,  0.0f, 0.0f, 0.0f,   0.0f, 200.0f, PROP_UNSIGNED },
273         { -1, 0, "" }
274 };
275 static void stucci_map_inputs(Tex *tex, bNodeStack **in, TexParams *p, short thread)
276 {
277         tex->noisesize = tex_input_value(in[I+0], p, thread);
278         tex->turbul    = tex_input_value(in[I+1], p, thread);
279 }
280 ProcDef(stucci)
281
282 /* --- */
283
284 static void init(bNodeTree *UNUSED(ntree), bNode* node, bNodeTemplate *UNUSED(ntemp))
285 {
286         Tex *tex = MEM_callocN(sizeof(Tex), "Tex");
287         node->storage= tex;
288         
289         default_tex(tex);
290         tex->type = node->type - TEX_NODE_PROC;
291         
292         if(tex->type == TEX_WOOD)
293                 tex->stype = TEX_BANDNOISE;
294         
295 }
296
297 /* Node type definitions */
298 #define TexDef(TEXTYPE, outputs, name, Name) \
299 void register_node_type_tex_proc_##name(ListBase *lb) \
300 { \
301         static bNodeType ntype; \
302         \
303         node_type_base(&ntype, TEX_NODE_PROC+TEXTYPE, Name, NODE_CLASS_TEXTURE, NODE_PREVIEW|NODE_OPTIONS); \
304         node_type_socket_templates(&ntype, name##_inputs, outputs); \
305         node_type_size(&ntype, 140, 80, 140); \
306         node_type_init(&ntype, init); \
307         node_type_storage(&ntype, "Tex", node_free_standard_storage, node_copy_standard_storage); \
308         node_type_exec(&ntype, name##_exec); \
309         \
310         nodeRegisterType(lb, &ntype); \
311 }
312         
313 #define C outputs_color_only
314 #define CV outputs_both
315         
316 TexDef(TEX_VORONOI,   CV, voronoi,   "Voronoi"  )
317 TexDef(TEX_BLEND,     C,  blend,     "Blend"    )
318 TexDef(TEX_MAGIC,     C,  magic,     "Magic"    )
319 TexDef(TEX_MARBLE,    CV, marble,    "Marble"   )
320 TexDef(TEX_CLOUDS,    CV, clouds,    "Clouds"   )
321 TexDef(TEX_WOOD,      CV, wood,      "Wood"     )
322 TexDef(TEX_MUSGRAVE,  CV, musgrave,  "Musgrave" )
323 TexDef(TEX_NOISE,     C,  noise,     "Noise"    )
324 TexDef(TEX_STUCCI,    CV, stucci,    "Stucci"   )
325 TexDef(TEX_DISTNOISE, CV, distnoise, "Distorted Noise" )