Math Node:
[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_modulo(float a, float b)
34 {
35         float result;
36
37         if (b == 0.0)
38                 result = 0.0;
39         else
40                 result = fmod(a, b);
41         
42         return result;
43 }
44
45 float safe_log(float a, float b)
46 {
47         if (a < 0.0 || b < 0.0)
48                 return 0.0;
49         
50         return log(a) / log(b);
51 }
52
53 shader node_math(
54         string type = "Add",
55         int Clamp = 0,
56         float Value1 = 0.0,
57         float Value2 = 0.0,
58         output float Value = 0.0)
59 {
60         /* OSL asin, acos, pow check for values that could give rise to nan */
61
62         if (type == "Add")
63                 Value = Value1 + Value2;
64         else if (type == "Subtract")
65                 Value = Value1 - Value2;
66         else if (type == "Multiply")
67                 Value = Value1 * Value2;
68         else if (type == "Divide")
69                 Value = safe_divide(Value1, Value2);
70         else if (type == "Sine")
71                 Value = sin(Value1);
72         else if (type == "Cosine")
73                 Value = cos(Value1);
74         else if (type == "Tangent")
75                 Value = tan(Value1);
76         else if (type == "Arcsine")
77                 Value = asin(Value1);
78         else if (type == "Arccosine")
79                 Value = acos(Value1);
80         else if (type == "Arctangent")
81                 Value = atan(Value1);
82         else if (type == "Power")
83                 Value = pow(Value1, Value2);
84         else if (type == "Logarithm")
85                 Value = safe_log(Value1, Value2);
86         else if (type == "Minimum")
87                 Value = min(Value1, Value2);
88         else if (type == "Maximum")
89                 Value = max(Value1, Value2);
90         else if (type == "Round")
91                 Value = floor(Value1 + 0.5);
92         else if (type == "Less Than")
93                 Value = Value1 < Value2;
94         else if (type == "Greater Than")
95                 Value = Value1 > Value2;
96         else if (type == "Modulo")
97                 Value = safe_modulo(Value1, Value2);
98
99         if (Clamp)
100                 Value = clamp(Value1, 0.0, 1.0);
101 }
102