215 lines
		
	
	
		
			6.2 KiB
		
	
	
	
		
			Plaintext
		
	
	
	
	
	
		
		
			
		
	
	
			215 lines
		
	
	
		
			6.2 KiB
		
	
	
	
		
			Plaintext
		
	
	
	
	
	
|   | //  (C) Copyright John Maddock 2005. | ||
|  | //  Use, modification and distribution are subject to the | ||
|  | //  Boost Software License, Version 1.0. (See accompanying file | ||
|  | //  LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt) | ||
|  | 
 | ||
|  | #ifndef BOOST_MATH_COMPLEX_ATANH_INCLUDED | ||
|  | #define BOOST_MATH_COMPLEX_ATANH_INCLUDED | ||
|  | 
 | ||
|  | #ifndef BOOST_MATH_COMPLEX_DETAILS_INCLUDED | ||
|  | #  include <boost/math/complex/details.hpp> | ||
|  | #endif | ||
|  | #ifndef BOOST_MATH_LOG1P_INCLUDED | ||
|  | #  include <boost/math/special_functions/log1p.hpp> | ||
|  | #endif | ||
|  | #include <boost/assert.hpp> | ||
|  | 
 | ||
|  | #ifdef BOOST_NO_STDC_NAMESPACE | ||
|  | namespace std{ using ::sqrt; using ::fabs; using ::acos; using ::asin; using ::atan; using ::atan2; } | ||
|  | #endif | ||
|  | 
 | ||
|  | namespace boost{ namespace math{ | ||
|  | 
 | ||
|  | template<class T>  | ||
|  | std::complex<T> atanh(const std::complex<T>& z) | ||
|  | { | ||
|  |    // | ||
|  |    // References: | ||
|  |    // | ||
|  |    // Eric W. Weisstein. "Inverse Hyperbolic Tangent."  | ||
|  |    // From MathWorld--A Wolfram Web Resource.  | ||
|  |    // http://mathworld.wolfram.com/InverseHyperbolicTangent.html | ||
|  |    // | ||
|  |    // Also: The Wolfram Functions Site, | ||
|  |    // http://functions.wolfram.com/ElementaryFunctions/ArcTanh/ | ||
|  |    // | ||
|  |    // Also "Abramowitz and Stegun. Handbook of Mathematical Functions." | ||
|  |    // at : http://jove.prohosting.com/~skripty/toc.htm | ||
|  |    // | ||
|  |    // See also: https://svn.boost.org/trac/boost/ticket/7291 | ||
|  |    // | ||
|  |     | ||
|  |    static const T pi = boost::math::constants::pi<T>(); | ||
|  |    static const T half_pi = pi / 2; | ||
|  |    static const T one = static_cast<T>(1.0L); | ||
|  |    static const T two = static_cast<T>(2.0L); | ||
|  |    static const T four = static_cast<T>(4.0L); | ||
|  |    static const T zero = static_cast<T>(0); | ||
|  |    static const T log_two = boost::math::constants::ln_two<T>(); | ||
|  | 
 | ||
|  | #ifdef BOOST_MSVC | ||
|  | #pragma warning(push) | ||
|  | #pragma warning(disable:4127) | ||
|  | #endif | ||
|  | 
 | ||
|  |    T x = std::fabs(z.real()); | ||
|  |    T y = std::fabs(z.imag()); | ||
|  | 
 | ||
|  |    T real, imag;  // our results | ||
|  | 
 | ||
|  |    T safe_upper = detail::safe_max(two); | ||
|  |    T safe_lower = detail::safe_min(static_cast<T>(2)); | ||
|  | 
 | ||
|  |    // | ||
|  |    // Begin by handling the special cases specified in C99: | ||
|  |    // | ||
|  |    if((boost::math::isnan)(x)) | ||
|  |    { | ||
|  |       if((boost::math::isnan)(y)) | ||
|  |          return std::complex<T>(x, x); | ||
|  |       else if((boost::math::isinf)(y)) | ||
|  |          return std::complex<T>(0, ((boost::math::signbit)(z.imag()) ? -half_pi : half_pi)); | ||
|  |       else | ||
|  |          return std::complex<T>(x, x); | ||
|  |    } | ||
|  |    else if((boost::math::isnan)(y)) | ||
|  |    { | ||
|  |       if(x == 0) | ||
|  |          return std::complex<T>(x, y); | ||
|  |       if((boost::math::isinf)(x)) | ||
|  |          return std::complex<T>(0, y); | ||
|  |       else | ||
|  |          return std::complex<T>(y, y); | ||
|  |    } | ||
|  |    else if((x > safe_lower) && (x < safe_upper) && (y > safe_lower) && (y < safe_upper)) | ||
|  |    { | ||
|  | 
 | ||
|  |       T yy = y*y; | ||
|  |       T mxm1 = one - x; | ||
|  |       /// | ||
|  |       // The real part is given by: | ||
|  |       //  | ||
|  |       // real(atanh(z)) == log1p(4*x / ((x-1)*(x-1) + y^2)) | ||
|  |       //  | ||
|  |       real = boost::math::log1p(four * x / (mxm1*mxm1 + yy)); | ||
|  |       real /= four; | ||
|  |       if((boost::math::signbit)(z.real())) | ||
|  |          real = (boost::math::changesign)(real); | ||
|  | 
 | ||
|  |       imag = std::atan2((y * two), (mxm1*(one+x) - yy)); | ||
|  |       imag /= two; | ||
|  |       if(z.imag() < 0) | ||
|  |          imag = (boost::math::changesign)(imag); | ||
|  |    } | ||
|  |    else | ||
|  |    { | ||
|  |       // | ||
|  |       // This section handles exception cases that would normally cause | ||
|  |       // underflow or overflow in the main formulas. | ||
|  |       // | ||
|  |       // Begin by working out the real part, we need to approximate | ||
|  |       //    real = boost::math::log1p(4x / ((x-1)^2 + y^2)) | ||
|  |       // without either overflow or underflow in the squared terms. | ||
|  |       // | ||
|  |       T mxm1 = one - x; | ||
|  |       if(x >= safe_upper) | ||
|  |       { | ||
|  |          // x-1 = x to machine precision: | ||
|  |          if((boost::math::isinf)(x) || (boost::math::isinf)(y)) | ||
|  |          { | ||
|  |             real = 0; | ||
|  |          } | ||
|  |          else if(y >= safe_upper) | ||
|  |          { | ||
|  |             // Big x and y: divide through by x*y: | ||
|  |             real = boost::math::log1p((four/y) / (x/y + y/x)); | ||
|  |          } | ||
|  |          else if(y > one) | ||
|  |          { | ||
|  |             // Big x: divide through by x: | ||
|  |             real = boost::math::log1p(four / (x + y*y/x)); | ||
|  |          } | ||
|  |          else | ||
|  |          { | ||
|  |             // Big x small y, as above but neglect y^2/x: | ||
|  |             real = boost::math::log1p(four/x); | ||
|  |          } | ||
|  |       } | ||
|  |       else if(y >= safe_upper) | ||
|  |       { | ||
|  |          if(x > one) | ||
|  |          { | ||
|  |             // Big y, medium x, divide through by y: | ||
|  |             real = boost::math::log1p((four*x/y) / (y + mxm1*mxm1/y)); | ||
|  |          } | ||
|  |          else | ||
|  |          { | ||
|  |             // Small or medium x, large y: | ||
|  |             real = four*x/y/y; | ||
|  |          } | ||
|  |       } | ||
|  |       else if (x != one) | ||
|  |       { | ||
|  |          // y is small, calculate divisor carefully: | ||
|  |          T div = mxm1*mxm1; | ||
|  |          if(y > safe_lower) | ||
|  |             div += y*y; | ||
|  |          real = boost::math::log1p(four*x/div); | ||
|  |       } | ||
|  |       else | ||
|  |          real = boost::math::changesign(two * (std::log(y) - log_two)); | ||
|  | 
 | ||
|  |       real /= four; | ||
|  |       if((boost::math::signbit)(z.real())) | ||
|  |          real = (boost::math::changesign)(real); | ||
|  | 
 | ||
|  |       // | ||
|  |       // Now handle imaginary part, this is much easier, | ||
|  |       // if x or y are large, then the formula: | ||
|  |       //    atan2(2y, (1-x)*(1+x) - y^2) | ||
|  |       // evaluates to +-(PI - theta) where theta is negligible compared to PI. | ||
|  |       // | ||
|  |       if((x >= safe_upper) || (y >= safe_upper)) | ||
|  |       { | ||
|  |          imag = pi; | ||
|  |       } | ||
|  |       else if(x <= safe_lower) | ||
|  |       { | ||
|  |          // | ||
|  |          // If both x and y are small then atan(2y), | ||
|  |          // otherwise just x^2 is negligible in the divisor: | ||
|  |          // | ||
|  |          if(y <= safe_lower) | ||
|  |             imag = std::atan2(two*y, one); | ||
|  |          else | ||
|  |          { | ||
|  |             if((y == zero) && (x == zero)) | ||
|  |                imag = 0; | ||
|  |             else | ||
|  |                imag = std::atan2(two*y, one - y*y); | ||
|  |          } | ||
|  |       } | ||
|  |       else | ||
|  |       { | ||
|  |          // | ||
|  |          // y^2 is negligible: | ||
|  |          // | ||
|  |          if((y == zero) && (x == one)) | ||
|  |             imag = 0; | ||
|  |          else | ||
|  |             imag = std::atan2(two*y, mxm1*(one+x)); | ||
|  |       } | ||
|  |       imag /= two; | ||
|  |       if((boost::math::signbit)(z.imag())) | ||
|  |          imag = (boost::math::changesign)(imag); | ||
|  |    } | ||
|  |    return std::complex<T>(real, imag); | ||
|  | #ifdef BOOST_MSVC | ||
|  | #pragma warning(pop) | ||
|  | #endif | ||
|  | } | ||
|  | 
 | ||
|  | } } // namespaces | ||
|  | 
 | ||
|  | #endif // BOOST_MATH_COMPLEX_ATANH_INCLUDED |