123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507 |
- /*
- * Copyright (c) 2020, Andreas Kling <kling@serenityos.org>
- * Copyright (c) 2020, Linus Groh <linusg@serenityos.org>
- *
- * SPDX-License-Identifier: BSD-2-Clause
- */
- #include <AK/Function.h>
- #include <AK/Random.h>
- #include <LibJS/Runtime/GlobalObject.h>
- #include <LibJS/Runtime/MathObject.h>
- #include <math.h>
- namespace JS {
- MathObject::MathObject(GlobalObject& global_object)
- : Object(*global_object.object_prototype())
- {
- }
- void MathObject::initialize(GlobalObject& global_object)
- {
- auto& vm = this->vm();
- Object::initialize(global_object);
- u8 attr = Attribute::Writable | Attribute::Configurable;
- define_native_function(vm.names.abs, abs, 1, attr);
- define_native_function(vm.names.random, random, 0, attr);
- define_native_function(vm.names.sqrt, sqrt, 1, attr);
- define_native_function(vm.names.floor, floor, 1, attr);
- define_native_function(vm.names.ceil, ceil, 1, attr);
- define_native_function(vm.names.round, round, 1, attr);
- define_native_function(vm.names.max, max, 2, attr);
- define_native_function(vm.names.min, min, 2, attr);
- define_native_function(vm.names.trunc, trunc, 1, attr);
- define_native_function(vm.names.sin, sin, 1, attr);
- define_native_function(vm.names.cos, cos, 1, attr);
- define_native_function(vm.names.tan, tan, 1, attr);
- define_native_function(vm.names.pow, pow, 2, attr);
- define_native_function(vm.names.exp, exp, 1, attr);
- define_native_function(vm.names.expm1, expm1, 1, attr);
- define_native_function(vm.names.sign, sign, 1, attr);
- define_native_function(vm.names.clz32, clz32, 1, attr);
- define_native_function(vm.names.acos, acos, 1, attr);
- define_native_function(vm.names.acosh, acosh, 1, attr);
- define_native_function(vm.names.asin, asin, 1, attr);
- define_native_function(vm.names.asinh, asinh, 1, attr);
- define_native_function(vm.names.atan, atan, 1, attr);
- define_native_function(vm.names.atanh, atanh, 1, attr);
- define_native_function(vm.names.log1p, log1p, 1, attr);
- define_native_function(vm.names.cbrt, cbrt, 1, attr);
- define_native_function(vm.names.atan2, atan2, 2, attr);
- define_native_function(vm.names.fround, fround, 1, attr);
- define_native_function(vm.names.hypot, hypot, 2, attr);
- define_native_function(vm.names.imul, imul, 2, attr);
- define_native_function(vm.names.log, log, 1, attr);
- define_native_function(vm.names.log2, log2, 1, attr);
- define_native_function(vm.names.log10, log10, 1, attr);
- define_native_function(vm.names.sinh, sinh, 1, attr);
- define_native_function(vm.names.cosh, cosh, 1, attr);
- define_native_function(vm.names.tanh, tanh, 1, attr);
- define_property(vm.names.E, Value(M_E), 0);
- define_property(vm.names.LN2, Value(M_LN2), 0);
- define_property(vm.names.LN10, Value(M_LN10), 0);
- define_property(vm.names.LOG2E, Value(::log2(M_E)), 0);
- define_property(vm.names.LOG10E, Value(::log10(M_E)), 0);
- define_property(vm.names.PI, Value(M_PI), 0);
- define_property(vm.names.SQRT1_2, Value(M_SQRT1_2), 0);
- define_property(vm.names.SQRT2, Value(M_SQRT2), 0);
- define_property(vm.well_known_symbol_to_string_tag(), js_string(vm.heap(), "Math"), Attribute::Configurable);
- }
- MathObject::~MathObject()
- {
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::abs)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- if (number.is_nan())
- return js_nan();
- return Value(number.as_double() >= 0 ? number.as_double() : -number.as_double());
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::random)
- {
- #ifdef __serenity__
- double r = (double)get_random<u32>() / (double)UINT32_MAX;
- #else
- double r = (double)rand() / (double)RAND_MAX;
- #endif
- return Value(r);
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::sqrt)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- if (number.is_nan())
- return js_nan();
- return Value(::sqrt(number.as_double()));
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::floor)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- if (number.is_nan())
- return js_nan();
- return Value(::floor(number.as_double()));
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::ceil)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- if (number.is_nan())
- return js_nan();
- auto number_double = number.as_double();
- if (number_double < 0 && number_double > -1)
- return Value(-0.f);
- return Value(::ceil(number.as_double()));
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::round)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- if (number.is_nan())
- return js_nan();
- double intpart = 0;
- double frac = modf(number.as_double(), &intpart);
- if (intpart >= 0) {
- if (frac >= 0.5)
- intpart += 1.0;
- } else {
- if (frac < -0.5)
- intpart -= 1.0;
- }
- return Value(intpart);
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::max)
- {
- if (!vm.argument_count())
- return js_negative_infinity();
- auto max = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- for (size_t i = 1; i < vm.argument_count(); ++i) {
- auto cur = vm.argument(i).to_number(global_object);
- if (vm.exception())
- return {};
- if ((max.is_negative_zero() && cur.is_positive_zero()) || cur.as_double() > max.as_double())
- max = cur;
- }
- return max;
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::min)
- {
- if (!vm.argument_count())
- return js_infinity();
- auto min = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- for (size_t i = 1; i < vm.argument_count(); ++i) {
- auto cur = vm.argument(i).to_number(global_object);
- if (vm.exception())
- return {};
- if ((min.is_positive_zero() && cur.is_negative_zero()) || cur.as_double() < min.as_double())
- min = cur;
- }
- return min;
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::trunc)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- if (number.is_nan())
- return js_nan();
- if (number.as_double() < 0)
- return MathObject::ceil(vm, global_object);
- return MathObject::floor(vm, global_object);
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::sin)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- if (number.is_nan())
- return js_nan();
- return Value(::sin(number.as_double()));
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::cos)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- if (number.is_nan())
- return js_nan();
- return Value(::cos(number.as_double()));
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::tan)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- if (number.is_nan())
- return js_nan();
- return Value(::tan(number.as_double()));
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::pow)
- {
- return JS::exp(global_object, vm.argument(0), vm.argument(1));
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::exp)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- if (number.is_nan())
- return js_nan();
- return Value(::exp(number.as_double()));
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::expm1)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- if (number.is_nan())
- return js_nan();
- return Value(::expm1(number.as_double()));
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::sign)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- if (number.is_positive_zero())
- return Value(0);
- if (number.is_negative_zero())
- return Value(-0.0);
- if (number.as_double() > 0)
- return Value(1);
- if (number.as_double() < 0)
- return Value(-1);
- return js_nan();
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::clz32)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- if (!number.is_finite_number() || (unsigned)number.as_double() == 0)
- return Value(32);
- return Value(__builtin_clz((unsigned)number.as_double()));
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::acos)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- if (number.is_nan() || number.as_double() > 1 || number.as_double() < -1)
- return js_nan();
- if (number.as_double() == 1)
- return Value(0);
- return Value(::acos(number.as_double()));
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::acosh)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- if (number.as_double() < 1)
- return js_nan();
- return Value(::acosh(number.as_double()));
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::asin)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- if (number.is_nan() || number.is_positive_zero() || number.is_negative_zero())
- return number;
- return Value(::asin(number.as_double()));
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::asinh)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- return Value(::asinh(number.as_double()));
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::atan)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- if (number.is_nan() || number.is_positive_zero() || number.is_negative_zero())
- return number;
- if (number.is_positive_infinity())
- return Value(M_PI_2);
- if (number.is_negative_infinity())
- return Value(-M_PI_2);
- return Value(::atan(number.as_double()));
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::atanh)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- if (number.as_double() > 1 || number.as_double() < -1)
- return js_nan();
- return Value(::atanh(number.as_double()));
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::log1p)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- if (number.as_double() < -1)
- return js_nan();
- return Value(::log1p(number.as_double()));
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::cbrt)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- return Value(::cbrt(number.as_double()));
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::atan2)
- {
- auto y = vm.argument(0).to_number(global_object), x = vm.argument(1).to_number(global_object);
- auto pi_4 = M_PI_2 / 2;
- auto three_pi_4 = pi_4 + M_PI_2;
- if (vm.exception())
- return {};
- if (x.is_positive_zero()) {
- if (y.is_positive_zero() || y.is_negative_zero())
- return y;
- else
- return (y.as_double() > 0) ? Value(M_PI_2) : Value(-M_PI_2);
- }
- if (x.is_negative_zero()) {
- if (y.is_positive_zero())
- return Value(M_PI);
- else if (y.is_negative_zero())
- return Value(-M_PI);
- else
- return (y.as_double() > 0) ? Value(M_PI_2) : Value(-M_PI_2);
- }
- if (x.is_positive_infinity()) {
- if (y.is_infinity())
- return (y.is_positive_infinity()) ? Value(pi_4) : Value(-pi_4);
- else
- return (y.as_double() > 0) ? Value(+0.0) : Value(-0.0);
- }
- if (x.is_negative_infinity()) {
- if (y.is_infinity())
- return (y.is_positive_infinity()) ? Value(three_pi_4) : Value(-three_pi_4);
- else
- return (y.as_double() > 0) ? Value(M_PI) : Value(-M_PI);
- }
- if (y.is_infinity())
- return (y.is_positive_infinity()) ? Value(M_PI_2) : Value(-M_PI_2);
- if (y.is_positive_zero())
- return (x.as_double() > 0) ? Value(+0.0) : Value(M_PI);
- if (y.is_negative_zero())
- return (x.as_double() > 0) ? Value(-0.0) : Value(-M_PI);
- return Value(::atan2(y.as_double(), x.as_double()));
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::fround)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- if (number.is_nan())
- return js_nan();
- return Value((float)number.as_double());
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::hypot)
- {
- if (!vm.argument_count())
- return Value(0);
- auto hypot = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- hypot = Value(hypot.as_double() * hypot.as_double());
- for (size_t i = 1; i < vm.argument_count(); ++i) {
- auto cur = vm.argument(i).to_number(global_object);
- if (vm.exception())
- return {};
- hypot = Value(hypot.as_double() + cur.as_double() * cur.as_double());
- }
- return Value(::sqrt(hypot.as_double()));
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::imul)
- {
- auto a = vm.argument(0).to_u32(global_object);
- if (vm.exception())
- return {};
- auto b = vm.argument(1).to_u32(global_object);
- if (vm.exception())
- return {};
- return Value(static_cast<i32>(a * b));
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::log)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- if (number.as_double() < 0)
- return js_nan();
- return Value(::log(number.as_double()));
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::log2)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- if (number.as_double() < 0)
- return js_nan();
- return Value(::log2(number.as_double()));
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::log10)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- if (number.as_double() < 0)
- return js_nan();
- return Value(::log10(number.as_double()));
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::sinh)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- if (number.is_nan())
- return js_nan();
- return Value(::sinh(number.as_double()));
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::cosh)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- if (number.is_nan())
- return js_nan();
- return Value(::cosh(number.as_double()));
- }
- JS_DEFINE_NATIVE_FUNCTION(MathObject::tanh)
- {
- auto number = vm.argument(0).to_number(global_object);
- if (vm.exception())
- return {};
- if (number.is_nan())
- return js_nan();
- if (number.is_positive_infinity())
- return Value(1);
- if (number.is_negative_infinity())
- return Value(-1);
- return Value(::tanh(number.as_double()));
- }
- }
|