Cycles / Wavelength to RGB node:
[blender.git] / intern / cycles / kernel / shaders / node_color.h
1 /*
2  * This program is free software; you can redistribute it and/or
3  * modify it under the terms of the GNU General Public License
4  * as published by the Free Software Foundation; either version 2
5  * of the License, or (at your option) any later version.
6  *
7  * This program is distributed in the hope that it will be useful,
8  * but WITHOUT ANY WARRANTY; without even the implied warranty of
9  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10  * GNU General Public License for more details.
11  *
12  * You should have received a copy of the GNU General Public License
13  * along with this program; if not, write to the Free Software Foundation,
14  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
15  */
16
17 /* Color Management */
18
19 float color_srgb_to_scene_linear(float c)
20 {
21         if (c < 0.04045)
22                 return (c < 0.0) ? 0.0 : c * (1.0 / 12.92);
23         else
24                 return pow((c + 0.055) * (1.0 / 1.055), 2.4);
25 }
26
27 float color_scene_linear_to_srgb(float c)
28 {
29         if (c < 0.0031308)
30                 return (c < 0.0) ? 0.0 : c * 12.92;
31         else
32                 return 1.055 * pow(c, 1.0 / 2.4) - 0.055;
33 }
34
35 color color_srgb_to_scene_linear(color c)
36 {
37         return color(
38                 color_srgb_to_scene_linear(c[0]),
39                 color_srgb_to_scene_linear(c[1]),
40                 color_srgb_to_scene_linear(c[2]));
41 }
42
43 color color_scene_linear_to_srgb(color c)
44 {
45         return color(
46                 color_scene_linear_to_srgb(c[0]),
47                 color_scene_linear_to_srgb(c[1]),
48                 color_scene_linear_to_srgb(c[2]));
49 }
50
51 color color_unpremultiply(color c, float alpha)
52 {
53         if(alpha != 1.0 && alpha != 0.0)
54                 return c/alpha;
55
56         return c;
57 }
58
59 /* Color Operations */
60
61 color xyY_to_xyz(float x, float y, float Y)
62 {
63         float X, Z;
64
65         if (y != 0.0) X = (x / y) * Y;
66         else X = 0.0;
67
68         if (y != 0.0 && Y != 0.0) Z = ((1.0 - x - y) / y) * Y;
69         else Z = 0.0;
70
71         return color(X, Y, Z);
72 }
73
74 color xyz_to_rgb(float x, float y, float z)
75 {
76         return color( 3.240479 * x + -1.537150 * y + -0.498535 * z,
77                      -0.969256 * x +  1.875991 * y +  0.041556 * z,
78                       0.055648 * x + -0.204043 * y +  1.057311 * z);
79 }
80
81 color rgb_to_hsv(color rgb)
82 {
83         float cmax, cmin, h, s, v, cdelta;
84         color c;
85
86         cmax = max(rgb[0], max(rgb[1], rgb[2]));
87         cmin = min(rgb[0], min(rgb[1], rgb[2]));
88         cdelta = cmax - cmin;
89
90         v = cmax;
91
92         if (cmax != 0.0) {
93                 s = cdelta / cmax;
94         }
95         else {
96                 s = 0.0;
97                 h = 0.0;
98         }
99
100         if (s == 0.0) {
101                 h = 0.0;
102         }
103         else {
104                 c = (color(cmax, cmax, cmax) - rgb) / cdelta;
105
106                 if (rgb[0] == cmax) h = c[2] - c[1];
107                 else if (rgb[1] == cmax) h = 2.0 + c[0] -  c[2];
108                 else h = 4.0 + c[1] - c[0];
109
110                 h /= 6.0;
111
112                 if (h < 0.0)
113                         h += 1.0;
114         }
115
116         return color(h, s, v);
117 }
118
119 color hsv_to_rgb(color hsv)
120 {
121         float i, f, p, q, t, h, s, v;
122         color rgb;
123
124         h = hsv[0];
125         s = hsv[1];
126         v = hsv[2];
127
128         if (s == 0.0) {
129                 rgb = color(v, v, v);
130         }
131         else {
132                 if (h == 1.0)
133                         h = 0.0;
134                 
135                 h *= 6.0;
136                 i = floor(h);
137                 f = h - i;
138                 rgb = color(f, f, f);
139                 p = v * (1.0 - s);
140                 q = v * (1.0 - (s * f));
141                 t = v * (1.0 - (s * (1.0 - f)));
142
143                 if (i == 0.0) rgb = color(v, t, p);
144                 else if (i == 1.0) rgb = color(q, v, p);
145                 else if (i == 2.0) rgb = color(p, v, t);
146                 else if (i == 3.0) rgb = color(p, q, v);
147                 else if (i == 4.0) rgb = color(t, p, v);
148                 else rgb = color(v, p, q);
149         }
150
151         return rgb;
152 }
153