Cycles: change __device and similar qualifiers to ccl_device in kernel code.
[blender.git] / intern / cycles / kernel / svm / svm_mix.h
1 /*
2  * Copyright 2011-2013 Blender Foundation
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License
15  */
16
17 CCL_NAMESPACE_BEGIN
18
19 ccl_device float3 svm_mix_blend(float t, float3 col1, float3 col2)
20 {
21         return interp(col1, col2, t);
22 }
23
24 ccl_device float3 svm_mix_add(float t, float3 col1, float3 col2)
25 {
26         return interp(col1, col1 + col2, t);
27 }
28
29 ccl_device float3 svm_mix_mul(float t, float3 col1, float3 col2)
30 {
31         return interp(col1, col1 * col2, t);
32 }
33
34 ccl_device float3 svm_mix_screen(float t, float3 col1, float3 col2)
35 {
36         float tm = 1.0f - t;
37         float3 one = make_float3(1.0f, 1.0f, 1.0f);
38         float3 tm3 = make_float3(tm, tm, tm);
39
40         return one - (tm3 + t*(one - col2))*(one - col1);
41 }
42
43 ccl_device float3 svm_mix_overlay(float t, float3 col1, float3 col2)
44 {
45         float tm = 1.0f - t;
46
47         float3 outcol = col1;
48
49         if(outcol.x < 0.5f)
50                 outcol.x *= tm + 2.0f*t*col2.x;
51         else
52                 outcol.x = 1.0f - (tm + 2.0f*t*(1.0f - col2.x))*(1.0f - outcol.x);
53
54         if(outcol.y < 0.5f)
55                 outcol.y *= tm + 2.0f*t*col2.y;
56         else
57                 outcol.y = 1.0f - (tm + 2.0f*t*(1.0f - col2.y))*(1.0f - outcol.y);
58
59         if(outcol.z < 0.5f)
60                 outcol.z *= tm + 2.0f*t*col2.z;
61         else
62                 outcol.z = 1.0f - (tm + 2.0f*t*(1.0f - col2.z))*(1.0f - outcol.z);
63         
64         return outcol;
65 }
66
67 ccl_device float3 svm_mix_sub(float t, float3 col1, float3 col2)
68 {
69         return interp(col1, col1 - col2, t);
70 }
71
72 ccl_device float3 svm_mix_div(float t, float3 col1, float3 col2)
73 {
74         float tm = 1.0f - t;
75
76         float3 outcol = col1;
77
78         if(col2.x != 0.0f) outcol.x = tm*outcol.x + t*outcol.x/col2.x;
79         if(col2.y != 0.0f) outcol.y = tm*outcol.y + t*outcol.y/col2.y;
80         if(col2.z != 0.0f) outcol.z = tm*outcol.z + t*outcol.z/col2.z;
81
82         return outcol;
83 }
84
85 ccl_device float3 svm_mix_diff(float t, float3 col1, float3 col2)
86 {
87         return interp(col1, fabs(col1 - col2), t);
88 }
89
90 ccl_device float3 svm_mix_dark(float t, float3 col1, float3 col2)
91 {
92         return min(col1, col2*t);
93 }
94
95 ccl_device float3 svm_mix_light(float t, float3 col1, float3 col2)
96 {
97         return max(col1, col2*t);
98 }
99
100 ccl_device float3 svm_mix_dodge(float t, float3 col1, float3 col2)
101 {
102         float3 outcol = col1;
103
104         if(outcol.x != 0.0f) {
105                 float tmp = 1.0f - t*col2.x;
106                 if(tmp <= 0.0f)
107                         outcol.x = 1.0f;
108                 else if((tmp = outcol.x/tmp) > 1.0f)
109                         outcol.x = 1.0f;
110                 else
111                         outcol.x = tmp;
112         }
113         if(outcol.y != 0.0f) {
114                 float tmp = 1.0f - t*col2.y;
115                 if(tmp <= 0.0f)
116                         outcol.y = 1.0f;
117                 else if((tmp = outcol.y/tmp) > 1.0f)
118                         outcol.y = 1.0f;
119                 else
120                         outcol.y = tmp;
121         }
122         if(outcol.z != 0.0f) {
123                 float tmp = 1.0f - t*col2.z;
124                 if(tmp <= 0.0f)
125                         outcol.z = 1.0f;
126                 else if((tmp = outcol.z/tmp) > 1.0f)
127                         outcol.z = 1.0f;
128                 else
129                         outcol.z = tmp;
130         }
131
132         return outcol;
133 }
134
135 ccl_device float3 svm_mix_burn(float t, float3 col1, float3 col2)
136 {
137         float tmp, tm = 1.0f - t;
138
139         float3 outcol = col1;
140
141         tmp = tm + t*col2.x;
142         if(tmp <= 0.0f)
143                 outcol.x = 0.0f;
144         else if((tmp = (1.0f - (1.0f - outcol.x)/tmp)) < 0.0f)
145                 outcol.x = 0.0f;
146         else if(tmp > 1.0f)
147                 outcol.x = 1.0f;
148         else
149                 outcol.x = tmp;
150
151         tmp = tm + t*col2.y;
152         if(tmp <= 0.0f)
153                 outcol.y = 0.0f;
154         else if((tmp = (1.0f - (1.0f - outcol.y)/tmp)) < 0.0f)
155                 outcol.y = 0.0f;
156         else if(tmp > 1.0f)
157                 outcol.y = 1.0f;
158         else
159                 outcol.y = tmp;
160
161         tmp = tm + t*col2.z;
162         if(tmp <= 0.0f)
163                 outcol.z = 0.0f;
164         else if((tmp = (1.0f - (1.0f - outcol.z)/tmp)) < 0.0f)
165                 outcol.z = 0.0f;
166         else if(tmp > 1.0f)
167                 outcol.z = 1.0f;
168         else
169                 outcol.z = tmp;
170         
171         return outcol;
172 }
173
174 ccl_device float3 svm_mix_hue(float t, float3 col1, float3 col2)
175 {
176         float3 outcol = col1;
177
178         float3 hsv2 = rgb_to_hsv(col2);
179
180         if(hsv2.y != 0.0f) {
181                 float3 hsv = rgb_to_hsv(outcol);
182                 hsv.x = hsv2.x;
183                 float3 tmp = hsv_to_rgb(hsv); 
184
185                 outcol = interp(outcol, tmp, t);
186         }
187
188         return outcol;
189 }
190
191 ccl_device float3 svm_mix_sat(float t, float3 col1, float3 col2)
192 {
193         float tm = 1.0f - t;
194
195         float3 outcol = col1;
196
197         float3 hsv = rgb_to_hsv(outcol);
198
199         if(hsv.y != 0.0f) {
200                 float3 hsv2 = rgb_to_hsv(col2);
201
202                 hsv.y = tm*hsv.y + t*hsv2.y;
203                 outcol = hsv_to_rgb(hsv);
204         }
205
206         return outcol;
207 }
208
209 ccl_device float3 svm_mix_val(float t, float3 col1, float3 col2)
210 {
211         float tm = 1.0f - t;
212
213         float3 hsv = rgb_to_hsv(col1);
214         float3 hsv2 = rgb_to_hsv(col2);
215
216         hsv.z = tm*hsv.z + t*hsv2.z;
217
218         return hsv_to_rgb(hsv);
219 }
220
221 ccl_device float3 svm_mix_color(float t, float3 col1, float3 col2)
222 {
223         float3 outcol = col1;
224         float3 hsv2 = rgb_to_hsv(col2);
225
226         if(hsv2.y != 0.0f) {
227                 float3 hsv = rgb_to_hsv(outcol);
228                 hsv.x = hsv2.x;
229                 hsv.y = hsv2.y;
230                 float3 tmp = hsv_to_rgb(hsv); 
231
232                 outcol = interp(outcol, tmp, t);
233         }
234
235         return outcol;
236 }
237
238 ccl_device float3 svm_mix_soft(float t, float3 col1, float3 col2)
239 {
240         float tm = 1.0f - t;
241
242         float3 one = make_float3(1.0f, 1.0f, 1.0f);
243         float3 scr = one - (one - col2)*(one - col1);
244
245         return tm*col1 + t*((one - col1)*col2*col1 + col1*scr);
246 }
247
248 ccl_device float3 svm_mix_linear(float t, float3 col1, float3 col2)
249 {
250         float3 outcol = col1;
251
252         if(col2.x > 0.5f)
253                 outcol.x = col1.x + t*(2.0f*(col2.x - 0.5f));
254         else
255                 outcol.x = col1.x + t*(2.0f*(col2.x) - 1.0f);
256
257         if(col2.y > 0.5f)
258                 outcol.y = col1.y + t*(2.0f*(col2.y - 0.5f));
259         else
260                 outcol.y = col1.y + t*(2.0f*(col2.y) - 1.0f);
261
262         if(col2.z > 0.5f)
263                 outcol.z = col1.z + t*(2.0f*(col2.z - 0.5f));
264         else
265                 outcol.z = col1.z + t*(2.0f*(col2.z) - 1.0f);
266         
267         return outcol;
268 }
269
270 ccl_device float3 svm_mix_clamp(float3 col)
271 {
272         float3 outcol = col;
273
274         outcol.x = clamp(col.x, 0.0f, 1.0f);
275         outcol.y = clamp(col.y, 0.0f, 1.0f);
276         outcol.z = clamp(col.z, 0.0f, 1.0f);
277
278         return outcol;
279 }
280
281 ccl_device float3 svm_mix(NodeMix type, float fac, float3 c1, float3 c2)
282 {
283         float t = clamp(fac, 0.0f, 1.0f);
284
285         switch(type) {
286                 case NODE_MIX_BLEND: return svm_mix_blend(t, c1, c2);
287                 case NODE_MIX_ADD: return svm_mix_add(t, c1, c2);
288                 case NODE_MIX_MUL: return svm_mix_mul(t, c1, c2);
289                 case NODE_MIX_SCREEN: return svm_mix_screen(t, c1, c2);
290                 case NODE_MIX_OVERLAY: return svm_mix_overlay(t, c1, c2);
291                 case NODE_MIX_SUB: return svm_mix_sub(t, c1, c2);
292                 case NODE_MIX_DIV: return svm_mix_div(t, c1, c2);
293                 case NODE_MIX_DIFF: return svm_mix_diff(t, c1, c2);
294                 case NODE_MIX_DARK: return svm_mix_dark(t, c1, c2);
295                 case NODE_MIX_LIGHT: return svm_mix_light(t, c1, c2);
296                 case NODE_MIX_DODGE: return svm_mix_dodge(t, c1, c2);
297                 case NODE_MIX_BURN: return svm_mix_burn(t, c1, c2);
298                 case NODE_MIX_HUE: return svm_mix_hue(t, c1, c2);
299                 case NODE_MIX_SAT: return svm_mix_sat(t, c1, c2);
300                 case NODE_MIX_VAL: return svm_mix_val (t, c1, c2);
301                 case NODE_MIX_COLOR: return svm_mix_color(t, c1, c2);
302                 case NODE_MIX_SOFT: return svm_mix_soft(t, c1, c2);
303                 case NODE_MIX_LINEAR: return svm_mix_linear(t, c1, c2);
304                 case NODE_MIX_CLAMP: return svm_mix_clamp(c1);
305         }
306
307         return make_float3(0.0f, 0.0f, 0.0f);
308 }
309
310 /* Node */
311
312 ccl_device void svm_node_mix(KernelGlobals *kg, ShaderData *sd, float *stack, uint fac_offset, uint c1_offset, uint c2_offset, int *offset)
313 {
314         /* read extra data */
315         uint4 node1 = read_node(kg, offset);
316
317         float fac = stack_load_float(stack, fac_offset);
318         float3 c1 = stack_load_float3(stack, c1_offset);
319         float3 c2 = stack_load_float3(stack, c2_offset);
320         float3 result = svm_mix((NodeMix)node1.y, fac, c1, c2);
321
322         stack_store_float3(stack, node1.z, result);
323 }
324
325 CCL_NAMESPACE_END
326