Fix T58117: Crash with keyframing
[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 dir)
29 {
30         dir[1] -= 1.0;
31
32         float div = 2.0 * sqrt(max(-0.5 * dir[1], 0.0));
33         if (div > 0.0)
34                 dir /= div;
35
36         float u = 0.5 * (dir[0] + 1.0);
37         float v = 0.5 * (dir[2] + 1.0);
38
39         return vector(u, v, 0.0);
40 }
41
42 shader node_environment_texture(
43         int use_mapping = 0,
44         matrix mapping = matrix(0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0),
45         vector Vector = P,
46         string filename = "",
47         string projection = "equirectangular",
48         string interpolation = "linear",
49         string color_space = "sRGB",
50         int is_float = 1,
51         int use_alpha = 1,
52         output color Color = 0.0,
53         output float Alpha = 1.0)
54 {
55         vector p = Vector;
56
57         if (use_mapping)
58                 p = transform(mapping, p);
59         
60         p = normalize(p);
61
62         if (projection == "equirectangular")
63                 p = environment_texture_direction_to_equirectangular(p);
64         else
65                 p = environment_texture_direction_to_mirrorball(p);
66
67         /* todo: use environment for better texture filtering of equirectangular */
68         Color = (color)texture(filename, p[0], 1.0 - p[1], "wrap", "periodic", "interp", interpolation, "alpha", Alpha);
69
70         if (use_alpha) {
71                 Color = color_unpremultiply(Color, Alpha);
72
73                 if (!is_float)
74                         Color = min(Color, 1.0);
75         }
76
77         if (color_space == "sRGB")
78                 Color = color_srgb_to_scene_linear(Color);
79 }
80