be9bb71d5113ed9dd2c76a912dd751ee0c746968
[blender.git] / intern / cycles / kernel / osl / nodes / 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         float Value1 = 0.0,
44         float Value2 = 0.0,
45         output float Value = 0.0)
46 {
47         /* OSL asin, acos, pow check for values that could give rise to nan */
48
49         if(type == "Add")
50                 Value = Value1 + Value2;
51         if(type == "Subtract")
52                 Value = Value1 - Value2;
53         if(type == "Multiply")
54                 Value = Value1*Value2;
55         if(type == "Divide")
56                 Value = safe_divide(Value1, Value2);
57         if(type == "Sine")
58                 Value = sin(Value1);
59         if(type == "Cosine")
60                 Value = cos(Value1);
61         if(type == "Tangent")
62                 Value = tan(Value1);
63         if(type == "Arcsine")
64                 Value = asin(Value1);
65         if(type == "Arccosine")
66                 Value = acos(Value1);
67         if(type == "Arctangent")
68                 Value = atan(Value1);
69         if(type == "Power")
70                 Value = pow(Value1, Value2);
71         if(type == "Logarithm")
72                 Value = safe_log(Value1, Value2);
73         if(type == "Minimum")
74                 Value = min(Value1, Value2);
75         if(type == "Maximum")
76                 Value = max(Value1, Value2);
77         if(type == "Round")
78                 Value = floor(Value1 + 0.5);
79         if(type == "Less Than")
80                 Value = Value1 < Value2;
81         if(type == "Greater Than")
82                 Value = Value1 > Value2;
83 }
84