Merge branch 'master' into blender2.8
[blender.git] / intern / cycles / kernel / shaders / node_environment_texture.osl
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 #include "stdosl.h"
18 #include "node_color.h"
19
20 vector environment_texture_direction_to_equirectangular(vector dir)
21 {
22         float u = -atan2(dir[1], dir[0]) / (M_2PI) + 0.5;
23         float v = atan2(dir[2], hypot(dir[0], dir[1])) / M_PI + 0.5;
24
25         return vector(u, v, 0.0);
26 }
27
28 vector environment_texture_direction_to_mirrorball(vector idir)
29 {
30         vector dir = idir;
31         dir[1] -= 1.0;
32
33         float div = 2.0 * sqrt(max(-0.5 * dir[1], 0.0));
34         if (div > 0.0)
35                 dir /= div;
36
37         float u = 0.5 * (dir[0] + 1.0);
38         float v = 0.5 * (dir[2] + 1.0);
39
40         return vector(u, v, 0.0);
41 }
42
43 shader node_environment_texture(
44         int use_mapping = 0,
45         matrix mapping = matrix(0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0),
46         vector Vector = P,
47         string filename = "",
48         string projection = "equirectangular",
49         string interpolation = "linear",
50         string color_space = "sRGB",
51         int is_float = 1,
52         int use_alpha = 1,
53         output color Color = 0.0,
54         output float Alpha = 1.0)
55 {
56         vector p = Vector;
57
58         if (use_mapping)
59                 p = transform(mapping, p);
60         
61         p = normalize(p);
62
63         if (projection == "equirectangular")
64                 p = environment_texture_direction_to_equirectangular(p);
65         else
66                 p = environment_texture_direction_to_mirrorball(p);
67
68         /* todo: use environment for better texture filtering of equirectangular */
69         Color = (color)texture(filename, p[0], 1.0 - p[1], "wrap", "periodic", "interp", interpolation, "alpha", Alpha);
70
71         if (use_alpha) {
72                 Color = color_unpremultiply(Color, Alpha);
73
74                 if (!is_float)
75                         Color = min(Color, 1.0);
76         }
77
78         if (color_space == "sRGB")
79                 Color = color_srgb_to_scene_linear(Color);
80 }
81