Implement more utility functions

refs #8169

diff --git a/lib/base/CMakeLists.txt b/lib/base/CMakeLists.txt
index 263124a5e47083c8f9c8fdf34464b2259288fee4..0fe01867597256ec53825ca3ac1732e6ac795320 100644 (file)
--- a/lib/base/CMakeLists.txt
+++ b/lib/base/CMakeLists.txt
@@ -25,7 +25,7 @@ mkclass_target(sysloglogger.ti sysloglogger.thpp)
 set(base_SOURCES
   application.cpp application-version.cpp application.thpp array.cpp array-script.cpp boolean.cpp boolean-script.cpp console.cpp context.cpp
   convert.cpp debuginfo.cpp dictionary.cpp dictionary-script.cpp dynamicobject.cpp dynamicobject.thpp dynamictype.cpp
-  exception.cpp fifo.cpp filelogger.cpp filelogger.thpp initialize.cpp json.cpp logger.cpp logger.thpp
+  exception.cpp fifo.cpp filelogger.cpp filelogger.thpp initialize.cpp json.cpp logger.cpp logger.thpp math-script.cpp
   netstring.cpp networkstream.cpp number.cpp number-script.cpp object.cpp object-script.cpp primitivetype.cpp process.cpp
   ringbuffer.cpp scriptframe.cpp scriptfunction.cpp scriptfunction-script.cpp scriptfunctionwrapper.cpp scriptglobal.cpp
   scriptutils.cpp serializer.cpp socket.cpp stacktrace.cpp

diff --git a/lib/base/math-script.cpp b/lib/base/math-script.cpp
new file mode 100644 (file)
index 0000000..a4deffa
--- /dev/null
+++ b/lib/base/math-script.cpp
@@ -0,0 +1,179 @@
+/******************************************************************************
+ * Icinga 2                                                                   *
+ * Copyright (C) 2012-2014 Icinga Development Team (http://www.icinga.org)    *
+ *                                                                            *
+ * This program is free software; you can redistribute it and/or              *
+ * modify it under the terms of the GNU General Public License                *
+ * as published by the Free Software Foundation; either version 2             *
+ * of the License, or (at your option) any later version.                     *
+ *                                                                            *
+ * This program is distributed in the hope that it will be useful,            *
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of             *
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the              *
+ * GNU General Public License for more details.                               *
+ *                                                                            *
+ * You should have received a copy of the GNU General Public License          *
+ * along with this program; if not, write to the Free Software Foundation     *
+ * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA.             *
+ ******************************************************************************/
+
+#include "base/dictionary.hpp"
+#include "base/scriptfunction.hpp"
+#include "base/scriptfunctionwrapper.hpp"
+#include "base/scriptframe.hpp"
+#include "base/initialize.hpp"
+#include <boost/math/special_functions/round.hpp>
+#include <boost/foreach.hpp>
+#include <cmath>
+
+using namespace icinga;
+
+static double MathAbs(double x)
+{
+       return std::fabs(x);
+}
+
+static double MathAcos(double x)
+{
+       return std::acos(x);
+}
+
+static double MathAsin(double x)
+{
+       return std::asin(x);
+}
+
+static double MathAtan(double x)
+{
+       return std::atan(x);
+}
+
+static double MathAtan2(double x, double y)
+{
+       return std::atan2(x, y);
+}
+
+static double MathCeil(double x)
+{
+       return std::ceil(x);
+}
+
+static double MathCos(double x)
+{
+       return std::cos(x);
+}
+
+static double MathExp(double x)
+{
+       return std::exp(x);
+}
+
+static double MathFloor(double x)
+{
+       return std::floor(x);
+}
+
+static double MathLog(double x)
+{
+       return std::log(x);
+}
+
+static Value MathMax(const std::vector<Value>& args)
+{
+       bool first = true;
+       Value result = -INFINITY;
+
+       BOOST_FOREACH(const Value& arg, args) {
+               if (first || arg > result) {
+                       first = false;
+                       result = arg;
+               }
+       }
+
+       return result;
+}
+
+static Value MathMin(const std::vector<Value>& args)
+{
+       bool first = true;
+       Value result = INFINITY;
+
+       BOOST_FOREACH(const Value& arg, args) {
+               if (first || arg < result) {
+                       first = false;
+                       result = arg;
+               }
+       }
+
+       return result;
+}
+
+static double MathPow(double x, double y)
+{
+       return std::pow(x, y);
+}
+
+static double MathRandom(void)
+{
+       return (double)std::rand() / RAND_MAX;
+}
+
+static double MathRound(double x)
+{
+       return boost::math::round(x);
+}
+
+static double MathSin(double x)
+{
+       return std::sin(x);
+}
+
+static double MathSqrt(double x)
+{
+       return std::sqrt(x);
+}
+
+static double MathTan(double x)
+{
+       return std::tan(x);
+}
+
+static void InitializeMathObj(void)
+{
+       Dictionary::Ptr mathObj = new Dictionary();
+
+       /* Constants */
+       mathObj->Set("E", 2.71828182845904523536);
+       mathObj->Set("LN2", 0.693147180559945309417);
+       mathObj->Set("LN10", 2.30258509299404568402);
+       mathObj->Set("LOG2E", 1.44269504088896340736);
+       mathObj->Set("LOG10E", 0.434294481903251827651);
+       mathObj->Set("PI", 3.14159265358979323846);
+       mathObj->Set("SQRT1_2", 0.707106781186547524401);
+       mathObj->Set("SQRT2", 1.41421356237309504880);
+
+       /* Methods */
+       mathObj->Set("abs", new ScriptFunction(WrapScriptFunction(MathAbs)));
+       mathObj->Set("acos", new ScriptFunction(WrapScriptFunction(MathAcos)));
+       mathObj->Set("asin", new ScriptFunction(WrapScriptFunction(MathAsin)));
+       mathObj->Set("atan", new ScriptFunction(WrapScriptFunction(MathAtan)));
+       mathObj->Set("atan2", new ScriptFunction(WrapScriptFunction(MathAtan2)));
+       mathObj->Set("ceil", new ScriptFunction(WrapScriptFunction(MathCeil)));
+       mathObj->Set("cos", new ScriptFunction(WrapScriptFunction(MathCos)));
+       mathObj->Set("exp", new ScriptFunction(WrapScriptFunction(MathExp)));
+       mathObj->Set("floor", new ScriptFunction(WrapScriptFunction(MathFloor)));
+       mathObj->Set("log", new ScriptFunction(WrapScriptFunction(MathLog)));
+       mathObj->Set("max", new ScriptFunction(WrapScriptFunction(MathMax)));
+       mathObj->Set("min", new ScriptFunction(WrapScriptFunction(MathMin)));
+       mathObj->Set("pow", new ScriptFunction(WrapScriptFunction(MathPow)));
+       mathObj->Set("random", new ScriptFunction(WrapScriptFunction(MathRandom)));
+       mathObj->Set("round", new ScriptFunction(WrapScriptFunction(MathRound)));
+       mathObj->Set("sin", new ScriptFunction(WrapScriptFunction(MathSin)));
+       mathObj->Set("sqrt", new ScriptFunction(WrapScriptFunction(MathSqrt)));
+       mathObj->Set("tan", new ScriptFunction(WrapScriptFunction(MathTan)));
+
+       ScriptGlobal::Set("Math", mathObj);
+}
+
+INITIALIZE_ONCE(InitializeMathObj);
+