add BLI_strcpy_rlen, replace strcat, which was used in misleading way.
[blender.git] / intern / cycles / util / util_color.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 __UTIL_COLOR_H__
20 #define __UTIL_COLOR_H__
21
22 #include "util_math.h"
23 #include "util_types.h"
24
25 CCL_NAMESPACE_BEGIN
26
27 __device float color_srgb_to_scene_linear(float c)
28 {
29         if(c < 0.04045f)
30                 return (c < 0.0f)? 0.0f: c * (1.0f/12.92f);
31         else
32                 return powf((c + 0.055f) * (1.0f / 1.055f), 2.4f);
33 }
34
35 __device float color_scene_linear_to_srgb(float c)
36 {
37         if(c < 0.0031308f)
38                 return (c < 0.0f)? 0.0f: c * 12.92f;
39         else
40                 return 1.055f * powf(c, 1.0f / 2.4f) - 0.055f;
41 }
42
43 __device float3 rgb_to_hsv(float3 rgb)
44 {
45         float cmax, cmin, h, s, v, cdelta;
46         float3 c;
47
48         cmax = fmaxf(rgb.x, fmaxf(rgb.y, rgb.z));
49         cmin = min(rgb.x, min(rgb.y, rgb.z));
50         cdelta = cmax - cmin;
51
52         v = cmax;
53
54         if(cmax != 0.0f) {
55                 s = cdelta/cmax;
56         }
57         else {
58                 s = 0.0f;
59                 h = 0.0f;
60         }
61
62         if(s == 0.0f) {
63                 h = 0.0f;
64         }
65         else {
66                 float3 cmax3 = make_float3(cmax, cmax, cmax);
67                 c = (cmax3 - rgb)/cdelta;
68
69                 if(rgb.x == cmax) h = c.z - c.y;
70                 else if(rgb.y == cmax) h = 2.0f + c.x -  c.z;
71                 else h = 4.0f + c.y - c.x;
72
73                 h /= 6.0f;
74
75                 if(h < 0.0f)
76                         h += 1.0f;
77         }
78
79         return make_float3(h, s, v);
80 }
81
82 __device float3 hsv_to_rgb(float3 hsv)
83 {
84         float i, f, p, q, t, h, s, v;
85         float3 rgb;
86
87         h = hsv.x;
88         s = hsv.y;
89         v = hsv.z;
90
91         if(s == 0.0f) {
92                 rgb = make_float3(v, v, v);
93         }
94         else {
95                 if(h == 1.0f)
96                         h = 0.0f;
97                 
98                 h *= 6.0f;
99                 i = floorf(h);
100                 f = h - i;
101                 rgb = make_float3(f, f, f);
102                 p = v*(1.0f-s);
103                 q = v*(1.0f-(s*f));
104                 t = v*(1.0f-(s*(1.0f-f)));
105                 
106                 if(i == 0.0f) rgb = make_float3(v, t, p);
107                 else if(i == 1.0f) rgb = make_float3(q, v, p);
108                 else if(i == 2.0f) rgb = make_float3(p, v, t);
109                 else if(i == 3.0f) rgb = make_float3(p, q, v);
110                 else if(i == 4.0f) rgb = make_float3(t, p, v);
111                 else rgb = make_float3(v, p, q);
112         }
113
114         return rgb;
115 }
116
117 __device float3 xyY_to_xyz(float x, float y, float Y)
118 {
119         float X, Z;
120
121         if(y != 0.0f) X = (x / y) * Y;
122         else X = 0.0f;
123
124         if(y != 0.0f && Y != 0.0f) Z = (1.0f - x - y) / y * Y;
125         else Z = 0.0f;
126
127         return make_float3(X, Y, Z);
128 }
129
130 __device float3 xyz_to_rgb(float x, float y, float z)
131 {
132         return make_float3(3.240479f * x + -1.537150f * y + -0.498535f * z,
133                                           -0.969256f * x +  1.875991f * y +  0.041556f * z,
134                                            0.055648f * x + -0.204043f * y +  1.057311f * z);
135 }
136
137 #ifndef __KERNEL_OPENCL__
138
139 __device float3 color_srgb_to_scene_linear(float3 c)
140 {
141         return make_float3(
142                 color_srgb_to_scene_linear(c.x),
143                 color_srgb_to_scene_linear(c.y),
144                 color_srgb_to_scene_linear(c.z));
145 }
146
147 __device float3 color_scene_linear_to_srgb(float3 c)
148 {
149         return make_float3(
150                 color_scene_linear_to_srgb(c.x),
151                 color_scene_linear_to_srgb(c.y),
152                 color_scene_linear_to_srgb(c.z));
153 }
154
155 #endif
156
157 __device float linear_rgb_to_gray(float3 c)
158 {
159         return c.x*0.2126f + c.y*0.7152f + c.z*0.0722f;
160 }
161
162 CCL_NAMESPACE_END
163
164 #endif /* __UTIL_COLOR_H__ */
165