summaryrefslogtreecommitdiffstats
path: root/src/armadillo/include/armadillo_bits/arma_cmath.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armadillo/include/armadillo_bits/arma_cmath.hpp')
-rw-r--r--src/armadillo/include/armadillo_bits/arma_cmath.hpp378
1 files changed, 378 insertions, 0 deletions
diff --git a/src/armadillo/include/armadillo_bits/arma_cmath.hpp b/src/armadillo/include/armadillo_bits/arma_cmath.hpp
new file mode 100644
index 0000000..22df4bf
--- /dev/null
+++ b/src/armadillo/include/armadillo_bits/arma_cmath.hpp
@@ -0,0 +1,378 @@
+// SPDX-License-Identifier: Apache-2.0
+//
+// Copyright 2008-2016 Conrad Sanderson (http://conradsanderson.id.au)
+// Copyright 2008-2016 National ICT Australia (NICTA)
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+// ------------------------------------------------------------------------
+
+
+
+//! \addtogroup arma_cmath
+//! @{
+
+
+
+//
+// wrappers for isfinite
+
+
+template<typename eT>
+inline
+bool
+arma_isfinite(eT)
+ {
+ return true;
+ }
+
+
+
+template<>
+inline
+bool
+arma_isfinite(float x)
+ {
+ return std::isfinite(x);
+ }
+
+
+
+template<>
+inline
+bool
+arma_isfinite(double x)
+ {
+ return std::isfinite(x);
+ }
+
+
+
+template<typename T>
+inline
+bool
+arma_isfinite(const std::complex<T>& x)
+ {
+ return ( arma_isfinite(x.real()) && arma_isfinite(x.imag()) );
+ }
+
+
+
+//
+// wrappers for isinf
+
+
+template<typename eT>
+inline
+bool
+arma_isinf(eT)
+ {
+ return false;
+ }
+
+
+
+template<>
+inline
+bool
+arma_isinf(float x)
+ {
+ return std::isinf(x);
+ }
+
+
+
+template<>
+inline
+bool
+arma_isinf(double x)
+ {
+ return std::isinf(x);
+ }
+
+
+
+template<typename T>
+inline
+bool
+arma_isinf(const std::complex<T>& x)
+ {
+ return ( arma_isinf(x.real()) || arma_isinf(x.imag()) );
+ }
+
+
+
+//
+// wrappers for isnan
+
+
+template<typename eT>
+inline
+bool
+arma_isnan(eT val)
+ {
+ arma_ignore(val);
+
+ return false;
+ }
+
+
+
+template<>
+inline
+bool
+arma_isnan(float x)
+ {
+ return std::isnan(x);
+ }
+
+
+
+template<>
+inline
+bool
+arma_isnan(double x)
+ {
+ return std::isnan(x);
+ }
+
+
+
+template<typename T>
+inline
+bool
+arma_isnan(const std::complex<T>& x)
+ {
+ return ( arma_isnan(x.real()) || arma_isnan(x.imag()) );
+ }
+
+
+
+//
+// implementation of arma_sign()
+
+
+template<typename eT>
+constexpr
+typename arma_unsigned_integral_only<eT>::result
+arma_sign(const eT x)
+ {
+ return (x > eT(0)) ? eT(+1) : eT(0);
+ }
+
+
+
+template<typename eT>
+constexpr
+typename arma_signed_integral_only<eT>::result
+arma_sign(const eT x)
+ {
+ return (x > eT(0)) ? eT(+1) : ( (x < eT(0)) ? eT(-1) : eT(0) );
+ }
+
+
+
+template<typename eT>
+constexpr
+typename arma_real_only<eT>::result
+arma_sign(const eT x)
+ {
+ return (x > eT(0)) ? eT(+1) : ( (x < eT(0)) ? eT(-1) : ((x == eT(0)) ? eT(0) : x) );
+ }
+
+
+
+template<typename eT>
+inline
+typename arma_cx_only<eT>::result
+arma_sign(const eT& x)
+ {
+ typedef typename eT::value_type T;
+
+ const T abs_x = std::abs(x);
+
+ return (abs_x != T(0)) ? (x / abs_x) : x;
+ }
+
+
+
+//
+// wrappers for hypot(x, y) = sqrt(x^2 + y^2)
+
+
+template<typename eT>
+inline
+eT
+arma_hypot(const eT x, const eT y)
+ {
+ arma_ignore(x);
+ arma_ignore(y);
+
+ arma_stop_runtime_error("arma_hypot(): not implemented for integer or complex element types");
+
+ return eT(0);
+ }
+
+
+
+template<>
+inline
+float
+arma_hypot(const float x, const float y)
+ {
+ return std::hypot(x, y);
+ }
+
+
+
+template<>
+inline
+double
+arma_hypot(const double x, const double y)
+ {
+ return std::hypot(x, y);
+ }
+
+
+
+//
+// implementation of arma_sinc()
+
+
+template<typename eT>
+inline
+eT
+arma_sinc_generic(const eT x)
+ {
+ typedef typename get_pod_type<eT>::result T;
+
+ const eT tmp = Datum<T>::pi * x;
+
+ return (tmp == eT(0)) ? eT(1) : eT( std::sin(tmp) / tmp );
+ }
+
+
+
+template<typename eT>
+inline
+eT
+arma_sinc(const eT x)
+ {
+ return eT( arma_sinc_generic( double(x) ) );
+ }
+
+
+
+template<>
+inline
+float
+arma_sinc(const float x)
+ {
+ return arma_sinc_generic(x);
+ }
+
+
+
+template<>
+inline
+double
+arma_sinc(const double x)
+ {
+ return arma_sinc_generic(x);
+ }
+
+
+
+template<typename T>
+inline
+std::complex<T>
+arma_sinc(const std::complex<T>& x)
+ {
+ return arma_sinc_generic(x);
+ }
+
+
+
+//
+// wrappers for arg()
+
+
+template<typename eT>
+struct arma_arg
+ {
+ static
+ inline
+ eT
+ eval(const eT x)
+ {
+ return eT( std::arg(x) );
+ }
+ };
+
+
+
+template<>
+struct arma_arg<float>
+ {
+ static
+ inline
+ float
+ eval(const float x)
+ {
+ return std::arg(x);
+ }
+ };
+
+
+
+template<>
+struct arma_arg<double>
+ {
+ static
+ inline
+ double
+ eval(const double x)
+ {
+ return std::arg(x);
+ }
+ };
+
+
+
+template<>
+struct arma_arg< std::complex<float> >
+ {
+ static
+ inline
+ float
+ eval(const std::complex<float>& x)
+ {
+ return std::arg(x);
+ }
+ };
+
+
+
+template<>
+struct arma_arg< std::complex<double> >
+ {
+ static
+ inline
+ double
+ eval(const std::complex<double>& x)
+ {
+ return std::arg(x);
+ }
+ };
+
+
+
+//! @}