Cycles / Wavelength to RGB node:
[blender.git] / intern / cycles / kernel / svm / svm_hsv.h
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 #ifndef __SVM_HSV_H__
20 #define __SVM_HSV_H__
21
22 CCL_NAMESPACE_BEGIN
23
24 __device float3 rgb_to_hsv(float3 rgb)
25 {
26         float cmax, cmin, h, s, v, cdelta;
27         float3 c;
28
29         cmax = fmaxf(rgb.x, fmaxf(rgb.y, rgb.z));
30         cmin = min(rgb.x, min(rgb.y, rgb.z));
31         cdelta = cmax - cmin;
32
33         v = cmax;
34
35         if(cmax != 0.0f) {
36                 s = cdelta/cmax;
37         }
38         else {
39                 s = 0.0f;
40                 h = 0.0f;
41         }
42
43         if(s == 0.0f) {
44                 h = 0.0f;
45         }
46         else {
47                 float3 cmax3 = make_float3(cmax, cmax, cmax);
48                 c = (cmax3 - rgb)/cdelta;
49
50                 if(rgb.x == cmax) h = c.z - c.y;
51                 else if(rgb.y == cmax) h = 2.0f + c.x -  c.z;
52                 else h = 4.0f + c.y - c.x;
53
54                 h /= 6.0f;
55
56                 if(h < 0.0f)
57                         h += 1.0f;
58         }
59
60         return make_float3(h, s, v);
61 }
62
63 __device float3 hsv_to_rgb(float3 hsv)
64 {
65         float i, f, p, q, t, h, s, v;
66         float3 rgb;
67
68         h = hsv.x;
69         s = hsv.y;
70         v = hsv.z;
71
72         if(s == 0.0f) {
73                 rgb = make_float3(v, v, v);
74         }
75         else {
76                 if(h == 1.0f)
77                         h = 0.0f;
78                 
79                 h *= 6.0f;
80                 i = floorf(h);
81                 f = h - i;
82                 rgb = make_float3(f, f, f);
83                 p = v*(1.0f-s);
84                 q = v*(1.0f-(s*f));
85                 t = v*(1.0f-(s*(1.0f-f)));
86                 
87                 if(i == 0.0f) rgb = make_float3(v, t, p);
88                 else if(i == 1.0f) rgb = make_float3(q, v, p);
89                 else if(i == 2.0f) rgb = make_float3(p, v, t);
90                 else if(i == 3.0f) rgb = make_float3(p, q, v);
91                 else if(i == 4.0f) rgb = make_float3(t, p, v);
92                 else rgb = make_float3(v, p, q);
93         }
94
95         return rgb;
96 }
97
98 __device void svm_node_hsv(KernelGlobals *kg, ShaderData *sd, float *stack, uint in_color_offset, uint fac_offset, uint out_color_offset, int *offset)
99 {
100         /* read extra data */
101         uint4 node1 = read_node(kg, offset);
102
103         float fac = stack_load_float(stack, fac_offset);
104         float3 in_color = stack_load_float3(stack, in_color_offset);
105         float3 color = in_color;
106
107         float hue = stack_load_float(stack, node1.y);
108         float sat = stack_load_float(stack, node1.z);
109         float val = stack_load_float(stack, node1.w);
110
111         color = rgb_to_hsv(color);
112
113         /* remember: fmod doesn't work for negative numbers here */
114         color.x += hue + 0.5f;
115         color.x = fmodf(color.x, 1.0f);
116         color.y *= sat;
117         color.z *= val;
118
119         color = hsv_to_rgb(color);
120
121         color.x = fac*color.x + (1.0f - fac)*in_color.x;
122         color.y = fac*color.y + (1.0f - fac)*in_color.y;
123         color.z = fac*color.z + (1.0f - fac)*in_color.z;
124
125         if (stack_valid(out_color_offset))
126                 stack_store_float3(stack, out_color_offset, color);
127 }
128
129 CCL_NAMESPACE_END
130
131 #endif /* __SVM_HSV_H__ */
132