Cycles bugfix: [32431] Cycles Math Node : Clamp does not work
[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         int Clamp = false,
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         if(type == "Subtract")
53                 Value = Value1 - Value2;
54         if(type == "Multiply")
55                 Value = Value1*Value2;
56         if(type == "Divide")
57                 Value = safe_divide(Value1, Value2);
58         if(type == "Sine")
59                 Value = sin(Value1);
60         if(type == "Cosine")
61                 Value = cos(Value1);
62         if(type == "Tangent")
63                 Value = tan(Value1);
64         if(type == "Arcsine")
65                 Value = asin(Value1);
66         if(type == "Arccosine")
67                 Value = acos(Value1);
68         if(type == "Arctangent")
69                 Value = atan(Value1);
70         if(type == "Power")
71                 Value = pow(Value1, Value2);
72         if(type == "Logarithm")
73                 Value = safe_log(Value1, Value2);
74         if(type == "Minimum")
75                 Value = min(Value1, Value2);
76         if(type == "Maximum")
77                 Value = max(Value1, Value2);
78         if(type == "Round")
79                 Value = floor(Value1 + 0.5);
80         if(type == "Less Than")
81                 Value = Value1 < Value2;
82         if(type == "Greater Than")
83                 Value = Value1 > Value2;
84
85         if(Clamp)
86                 Value = clamp(Value1, 0.0, 1.0);
87 }
88