clamp for Mix node
[blender.git] / intern / cycles / kernel / osl / nodes / node_mix.osl
1 /*
2  * Copyright 2011, Blender Foundation.
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
19 #include "stdosl.h"
20 #include "node_color.h"
21
22 color node_mix_blend(float t, color col1, color col2)
23 {
24         return mix(col1, col2, t);
25 }
26
27 color node_mix_add(float t, color col1, color col2)
28 {
29         return mix(col1, col1 + col2, t);
30 }
31
32 color node_mix_mul(float t, color col1, color col2)
33 {
34         return mix(col1, col1 * col2, t);
35 }
36
37 color node_mix_screen(float t, color col1, color col2)
38 {
39         float tm = 1.0 - t;
40
41         return color(1.0) - (color(tm) + t*(color(1.0) - col2))*(color(1.0) - col1);
42 }
43
44 color node_mix_overlay(float t, color col1, color col2)
45 {
46         float tm = 1.0 - t;
47
48         color outcol = col1;
49
50         if(outcol[0] < 0.5)
51                 outcol[0] *= tm + 2.0*t*col2[0];
52         else
53                 outcol[0] = 1.0 - (tm + 2.0*t*(1.0 - col2[0]))*(1.0 - outcol[0]);
54
55         if(outcol[1] < 0.5)
56                 outcol[1] *= tm + 2.0*t*col2[1];
57         else
58                 outcol[1] = 1.0 - (tm + 2.0*t*(1.0 - col2[1]))*(1.0 - outcol[1]);
59
60         if(outcol[2] < 0.5)
61                 outcol[2] *= tm + 2.0*t*col2[2];
62         else
63                 outcol[2] = 1.0 - (tm + 2.0*t*(1.0 - col2[2]))*(1.0 - outcol[2]);
64         
65         return outcol;
66 }
67
68 color node_mix_sub(float t, color col1, color col2)
69 {
70         return mix(col1, col1 - col2, t);
71 }
72
73 color node_mix_div(float t, color col1, color col2)
74 {
75         float tm = 1.0 - t;
76
77         color outcol = col1;
78
79         if(col2[0] != 0.0) outcol[0] = tm*outcol[0] + t*outcol[0]/col2[0];
80         if(col2[1] != 0.0) outcol[1] = tm*outcol[1] + t*outcol[1]/col2[1];
81         if(col2[2] != 0.0) outcol[2] = tm*outcol[2] + t*outcol[2]/col2[2];
82
83         return outcol;
84 }
85
86 color node_mix_diff(float t, color col1, color col2)
87 {
88         return mix(col1, abs(col1 - col2), t);
89 }
90
91 color node_mix_dark(float t, color col1, color col2)
92 {
93         return min(col1, col2*t);
94 }
95
96 color node_mix_light(float t, color col1, color col2)
97 {
98         return max(col1, col2*t);
99 }
100
101 color node_mix_dodge(float t, color col1, color col2)
102 {
103         color outcol = col1;
104
105         if(outcol[0] != 0.0) {
106                 float tmp = 1.0 - t*col2[0];
107                 if(tmp <= 0.0)
108                         outcol[0] = 1.0;
109                 else if((tmp = outcol[0]/tmp) > 1.0)
110                         outcol[0] = 1.0;
111                 else
112                         outcol[0] = tmp;
113         }
114         if(outcol[1] != 0.0) {
115                 float tmp = 1.0 - t*col2[1];
116                 if(tmp <= 0.0)
117                         outcol[1] = 1.0;
118                 else if((tmp = outcol[1]/tmp) > 1.0)
119                         outcol[1] = 1.0;
120                 else
121                         outcol[1] = tmp;
122         }
123         if(outcol[2] != 0.0) {
124                 float tmp = 1.0 - t*col2[2];
125                 if(tmp <= 0.0)
126                         outcol[2] = 1.0;
127                 else if((tmp = outcol[2]/tmp) > 1.0)
128                         outcol[2] = 1.0;
129                 else
130                         outcol[2] = tmp;
131         }
132
133         return outcol;
134 }
135
136 color node_mix_burn(float t, color col1, color col2)
137 {
138         float tmp, tm = 1.0 - t;
139
140         color outcol = col1;
141
142         tmp = tm + t*col2[0];
143         if(tmp <= 0.0)
144                 outcol[0] = 0.0;
145         else if((tmp = (1.0 - (1.0 - outcol[0])/tmp)) < 0.0)
146                 outcol[0] = 0.0;
147         else if(tmp > 1.0)
148                 outcol[0] = 1.0;
149         else
150                 outcol[0] = tmp;
151
152         tmp = tm + t*col2[1];
153         if(tmp <= 0.0)
154                 outcol[1] = 0.0;
155         else if((tmp = (1.0 - (1.0 - outcol[1])/tmp)) < 0.0)
156                 outcol[1] = 0.0;
157         else if(tmp > 1.0)
158                 outcol[1] = 1.0;
159         else
160                 outcol[1] = tmp;
161
162         tmp = tm + t*col2[2];
163         if(tmp <= 0.0)
164                 outcol[2] = 0.0;
165         else if((tmp = (1.0 - (1.0 - outcol[2])/tmp)) < 0.0)
166                 outcol[2] = 0.0;
167         else if(tmp > 1.0)
168                 outcol[2] = 1.0;
169         else
170                 outcol[2] = tmp;
171         
172         return outcol;
173 }
174
175 color node_mix_hue(float t, color col1, color col2)
176 {
177         color outcol = col1;
178         color hsv2 = rgb_to_hsv(col2);
179
180         if(hsv2[1] != 0.0) {
181                 color hsv = rgb_to_hsv(outcol);
182                 hsv[0] = hsv2[0];
183                 color tmp = hsv_to_rgb(hsv); 
184
185                 outcol = mix(outcol, tmp, t);
186         }
187
188         return outcol;
189 }
190
191 color node_mix_sat(float t, color col1, color col2)
192 {
193         float tm = 1.0 - t;
194
195         color outcol = col1;
196
197         color hsv = rgb_to_hsv(outcol);
198
199         if(hsv[1] != 0.0) {
200                 color hsv2 = rgb_to_hsv(col2);
201
202                 hsv[1] = tm*hsv[1] + t*hsv2[1];
203                 outcol = hsv_to_rgb(hsv);
204         }
205
206         return outcol;
207 }
208
209 color node_mix_val(float t, color col1, color col2)
210 {
211         float tm = 1.0 - t;
212
213         color hsv = rgb_to_hsv(col1);
214         color hsv2 = rgb_to_hsv(col2);
215
216         hsv[2] = tm*hsv[2] + t*hsv2[2];
217
218         return hsv_to_rgb(hsv);
219 }
220
221 color node_mix_color(float t, color col1, color col2)
222 {
223         color outcol = col1;
224         color hsv2 = rgb_to_hsv(col2);
225
226         if(hsv2[1] != 0.0) {
227                 color hsv = rgb_to_hsv(outcol);
228                 hsv[0] = hsv2[0];
229                 hsv[1] = hsv2[1];
230                 color tmp = hsv_to_rgb(hsv); 
231
232                 outcol = mix(outcol, tmp, t);
233         }
234
235         return outcol;
236 }
237
238 color node_mix_soft(float t, color col1, color col2)
239 {
240         float tm = 1.0 - t;
241
242         color one= color(1.0);
243         color scr= one - (one - col2)*(one - col1);
244
245         return tm*col1 + t*((one - col1)*col2*col1 + col1*scr);
246 }
247
248 color node_mix_linear(float t, color col1, color col2)
249 {
250         color outcol = col1;
251
252         if(col2[0] > 0.5)
253                 outcol[0]= col1[0] + t*(2.0*(col2[0] - 0.5));
254         else
255                 outcol[0]= col1[0] + t*(2.0*(col2[0]) - 1.0);
256
257         if(col2[1] > 0.5)
258                 outcol[1]= col1[1] + t*(2.0*(col2[1] - 0.5));
259         else
260                 outcol[1]= col1[1] + t*(2.0*(col2[1]) - 1.0);
261
262         if(col2[2] > 0.5)
263                 outcol[2]= col1[2] + t*(2.0*(col2[2] - 0.5));
264         else
265                 outcol[2]= col1[2] + t*(2.0*(col2[2]) - 1.0);
266         
267         return outcol;
268 }
269
270 color node_mix_clamp(color col)
271 {
272         color outcol = col;
273
274         outcol[0] = clamp(col[0], 0.0, 1.0);
275         outcol[1] = clamp(col[2], 0.0, 1.0);
276         outcol[2] = clamp(col[2], 0.0, 1.0);
277
278         return outcol;
279 }
280
281 shader node_mix(
282         string type = "Mix",
283         int Clamp = false,
284         float Fac = 0.5,
285         color Color1 = color(0.0, 0.0, 0.0),
286         color Color2 = color(0.0, 0.0, 0.0),
287         output color Color = color(0.0, 0.0, 0.0))
288 {
289         float t = clamp(Fac, 0.0, 1.0);
290
291         if(type == "Mix")
292                 Color = node_mix_blend(t, Color1, Color2);
293         if(type == "Add")
294                 Color = node_mix_add(t, Color1, Color2);
295         if(type == "Multiply")
296                 Color = node_mix_mul(t, Color1, Color2);
297         if(type == "Screen")
298                 Color = node_mix_screen(t, Color1, Color2);
299         if(type == "Overlay")
300                 Color = node_mix_overlay(t, Color1, Color2);
301         if(type == "Subtract")
302                 Color = node_mix_sub(t, Color1, Color2);
303         if(type == "Divide")
304                 Color = node_mix_div(t, Color1, Color2);
305         if(type == "Difference")
306                 Color = node_mix_diff(t, Color1, Color2);
307         if(type == "Darken")
308                 Color = node_mix_dark(t, Color1, Color2);
309         if(type == "Lighten")
310                 Color = node_mix_light(t, Color1, Color2);
311         if(type == "Dodge")
312                 Color = node_mix_dodge(t, Color1, Color2);
313         if(type == "Burn")
314                 Color = node_mix_burn(t, Color1, Color2);
315         if(type == "Hue")
316                 Color = node_mix_hue(t, Color1, Color2);
317         if(type == "Saturation")
318                 Color = node_mix_sat(t, Color1, Color2);
319         if(type == "Value")
320                 Color = node_mix_val (t, Color1, Color2);
321         if(type == "Color")
322                 Color = node_mix_color(t, Color1, Color2);
323         if(type == "Soft Light")
324                 Color = node_mix_soft(t, Color1, Color2);
325         if(type == "Linear Light")
326                 Color = node_mix_linear(t, Color1, Color2);
327
328         if(Clamp)
329                 Color = node_mix_clamp(Color);
330 }
331