1 /* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */
2
3 #include "base/dictionary.hpp"
4 #include "base/function.hpp"
5 #include "base/functionwrapper.hpp"
6 #include "base/scriptframe.hpp"
7 #include "base/initialize.hpp"
8 #include "base/namespace.hpp"
9 #include <boost/math/special_functions/round.hpp>
10 #include <cmath>
11
12 using namespace icinga;
13
MathAbs(double x)14 static double MathAbs(double x)
15 {
16 return std::fabs(x);
17 }
18
MathAcos(double x)19 static double MathAcos(double x)
20 {
21 return std::acos(x);
22 }
23
MathAsin(double x)24 static double MathAsin(double x)
25 {
26 return std::asin(x);
27 }
28
MathAtan(double x)29 static double MathAtan(double x)
30 {
31 return std::atan(x);
32 }
33
MathAtan2(double y,double x)34 static double MathAtan2(double y, double x)
35 {
36 return std::atan2(y, x);
37 }
38
MathCeil(double x)39 static double MathCeil(double x)
40 {
41 return std::ceil(x);
42 }
43
MathCos(double x)44 static double MathCos(double x)
45 {
46 return std::cos(x);
47 }
48
MathExp(double x)49 static double MathExp(double x)
50 {
51 return std::exp(x);
52 }
53
MathFloor(double x)54 static double MathFloor(double x)
55 {
56 return std::floor(x);
57 }
58
MathLog(double x)59 static double MathLog(double x)
60 {
61 return std::log(x);
62 }
63
MathMax(const std::vector<Value> & args)64 static Value MathMax(const std::vector<Value>& args)
65 {
66 bool first = true;
67 Value result = -INFINITY;
68
69 for (const Value& arg : args) {
70 if (first || arg > result) {
71 first = false;
72 result = arg;
73 }
74 }
75
76 return result;
77 }
78
MathMin(const std::vector<Value> & args)79 static Value MathMin(const std::vector<Value>& args)
80 {
81 bool first = true;
82 Value result = INFINITY;
83
84 for (const Value& arg : args) {
85 if (first || arg < result) {
86 first = false;
87 result = arg;
88 }
89 }
90
91 return result;
92 }
93
MathPow(double x,double y)94 static double MathPow(double x, double y)
95 {
96 return std::pow(x, y);
97 }
98
MathRandom()99 static double MathRandom()
100 {
101 return (double)std::rand() / RAND_MAX;
102 }
103
MathRound(double x)104 static double MathRound(double x)
105 {
106 return boost::math::round(x);
107 }
108
MathSin(double x)109 static double MathSin(double x)
110 {
111 return std::sin(x);
112 }
113
MathSqrt(double x)114 static double MathSqrt(double x)
115 {
116 return std::sqrt(x);
117 }
118
MathTan(double x)119 static double MathTan(double x)
120 {
121 return std::tan(x);
122 }
123
MathIsnan(double x)124 static bool MathIsnan(double x)
125 {
126 return boost::math::isnan(x);
127 }
128
MathIsinf(double x)129 static bool MathIsinf(double x)
130 {
131 return boost::math::isinf(x);
132 }
133
MathSign(double x)134 static double MathSign(double x)
135 {
136 if (x > 0)
137 return 1;
138 else if (x < 0)
139 return -1;
140 else
141 return 0;
142 }
143
__anonba9593180102() 144 INITIALIZE_ONCE([]() {
145 auto mathNSBehavior = new ConstNamespaceBehavior();
146 Namespace::Ptr mathNS = new Namespace(mathNSBehavior);
147
148 /* Constants */
149 mathNS->Set("E", 2.71828182845904523536);
150 mathNS->Set("LN2", 0.693147180559945309417);
151 mathNS->Set("LN10", 2.30258509299404568402);
152 mathNS->Set("LOG2E", 1.44269504088896340736);
153 mathNS->Set("LOG10E", 0.434294481903251827651);
154 mathNS->Set("PI", 3.14159265358979323846);
155 mathNS->Set("SQRT1_2", 0.707106781186547524401);
156 mathNS->Set("SQRT2", 1.41421356237309504880);
157
158 /* Methods */
159 mathNS->Set("abs", new Function("Math#abs", MathAbs, { "x" }, true));
160 mathNS->Set("acos", new Function("Math#acos", MathAcos, { "x" }, true));
161 mathNS->Set("asin", new Function("Math#asin", MathAsin, { "x" }, true));
162 mathNS->Set("atan", new Function("Math#atan", MathAtan, { "x" }, true));
163 mathNS->Set("atan2", new Function("Math#atan2", MathAtan2, { "x", "y" }, true));
164 mathNS->Set("ceil", new Function("Math#ceil", MathCeil, { "x" }, true));
165 mathNS->Set("cos", new Function("Math#cos", MathCos, { "x" }, true));
166 mathNS->Set("exp", new Function("Math#exp", MathExp, { "x" }, true));
167 mathNS->Set("floor", new Function("Math#floor", MathFloor, { "x" }, true));
168 mathNS->Set("log", new Function("Math#log", MathLog, { "x" }, true));
169 mathNS->Set("max", new Function("Math#max", MathMax, {}, true));
170 mathNS->Set("min", new Function("Math#min", MathMin, {}, true));
171 mathNS->Set("pow", new Function("Math#pow", MathPow, { "x", "y" }, true));
172 mathNS->Set("random", new Function("Math#random", MathRandom, {}, true));
173 mathNS->Set("round", new Function("Math#round", MathRound, { "x" }, true));
174 mathNS->Set("sin", new Function("Math#sin", MathSin, { "x" }, true));
175 mathNS->Set("sqrt", new Function("Math#sqrt", MathSqrt, { "x" }, true));
176 mathNS->Set("tan", new Function("Math#tan", MathTan, { "x" }, true));
177 mathNS->Set("isnan", new Function("Math#isnan", MathIsnan, { "x" }, true));
178 mathNS->Set("isinf", new Function("Math#isinf", MathIsinf, { "x" }, true));
179 mathNS->Set("sign", new Function("Math#sign", MathSign, { "x" }, true));
180
181 mathNSBehavior->Freeze();
182
183 Namespace::Ptr systemNS = ScriptGlobal::Get("System");
184 systemNS->SetAttribute("Math", new ConstEmbeddedNamespaceValue(mathNS));
185 });
186