2ec1a5f0a32365bc952b57ba28326fc62f23f006
[blender.git] / intern / cycles / kernel / shaders / node_math.osl
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 #include "stdosl.h"
20
21 float safe_divide(float a, float b)
22 {
23         float result;
24
25         if (b == 0.0)
26                 result = 0.0;
27         else
28                 result = a / b;
29         
30         return result;
31 }
32
33 float safe_log(float a, float b)
34 {
35         if (a < 0.0 || b < 0.0)
36                 return 0.0;
37         
38         return log(a) / log(b);
39 }
40
41 shader node_math(
42         string type = "Add",
43         int Clamp = 0,
44         float Value1 = 0.0,
45         float Value2 = 0.0,
46         output float Value = 0.0)
47 {
48         /* OSL asin, acos, pow check for values that could give rise to nan */
49
50         if (type == "Add")
51                 Value = Value1 + Value2;
52         else if (type == "Subtract")
53                 Value = Value1 - Value2;
54         else if (type == "Multiply")
55                 Value = Value1 * Value2;
56         else if (type == "Divide")
57                 Value = safe_divide(Value1, Value2);
58         else if (type == "Sine")
59                 Value = sin(Value1);
60         else if (type == "Cosine")
61                 Value = cos(Value1);
62         else if (type == "Tangent")
63                 Value = tan(Value1);
64         else if (type == "Arcsine")
65                 Value = asin(Value1);
66         else if (type == "Arccosine")
67                 Value = acos(Value1);
68         else if (type == "Arctangent")
69                 Value = atan(Value1);
70         else if (type == "Power")
71                 Value = pow(Value1, Value2);
72         else if (type == "Logarithm")
73                 Value = safe_log(Value1, Value2);
74         else if (type == "Minimum")
75                 Value = min(Value1, Value2);
76         else if (type == "Maximum")
77                 Value = max(Value1, Value2);
78         else if (type == "Round")
79                 Value = floor(Value1 + 0.5);
80         else if (type == "Less Than")
81                 Value = Value1 < Value2;
82         else if (type == "Greater Than")
83                 Value = Value1 > Value2;
84
85         if (Clamp)
86                 Value = clamp(Value1, 0.0, 1.0);
87 }
88