...

Source file src/pkg/math/cmplx/pow.go

     1	// Copyright 2010 The Go Authors. All rights reserved.
     2	// Use of this source code is governed by a BSD-style
     3	// license that can be found in the LICENSE file.
     4	
     5	package cmplx
     6	
     7	import "math"
     8	
     9	// The original C code, the long comment, and the constants
    10	// below are from http://netlib.sandia.gov/cephes/c9x-complex/clog.c.
    11	// The go code is a simplified version of the original C.
    12	//
    13	// Cephes Math Library Release 2.8:  June, 2000
    14	// Copyright 1984, 1987, 1989, 1992, 2000 by Stephen L. Moshier
    15	//
    16	// The readme file at http://netlib.sandia.gov/cephes/ says:
    17	//    Some software in this archive may be from the book _Methods and
    18	// Programs for Mathematical Functions_ (Prentice-Hall or Simon & Schuster
    19	// International, 1989) or from the Cephes Mathematical Library, a
    20	// commercial product. In either event, it is copyrighted by the author.
    21	// What you see here may be used freely but it comes with no support or
    22	// guarantee.
    23	//
    24	//   The two known misprints in the book are repaired here in the
    25	// source listings for the gamma function and the incomplete beta
    26	// integral.
    27	//
    28	//   Stephen L. Moshier
    29	//   moshier@na-net.ornl.gov
    30	
    31	// Complex power function
    32	//
    33	// DESCRIPTION:
    34	//
    35	// Raises complex A to the complex Zth power.
    36	// Definition is per AMS55 # 4.2.8,
    37	// analytically equivalent to cpow(a,z) = cexp(z clog(a)).
    38	//
    39	// ACCURACY:
    40	//
    41	//                      Relative error:
    42	// arithmetic   domain     # trials      peak         rms
    43	//    IEEE      -10,+10     30000       9.4e-15     1.5e-15
    44	
    45	// Pow returns x**y, the base-x exponential of y.
    46	// For generalized compatibility with math.Pow:
    47	//	Pow(0, ±0) returns 1+0i
    48	//	Pow(0, c) for real(c)<0 returns Inf+0i if imag(c) is zero, otherwise Inf+Inf i.
    49	func Pow(x, y complex128) complex128 {
    50		if x == 0 { // Guaranteed also true for x == -0.
    51			if IsNaN(y) {
    52				return NaN()
    53			}
    54			r, i := real(y), imag(y)
    55			switch {
    56			case r == 0:
    57				return 1
    58			case r < 0:
    59				if i == 0 {
    60					return complex(math.Inf(1), 0)
    61				}
    62				return Inf()
    63			case r > 0:
    64				return 0
    65			}
    66			panic("not reached")
    67		}
    68		modulus := Abs(x)
    69		if modulus == 0 {
    70			return complex(0, 0)
    71		}
    72		r := math.Pow(modulus, real(y))
    73		arg := Phase(x)
    74		theta := real(y) * arg
    75		if imag(y) != 0 {
    76			r *= math.Exp(-imag(y) * arg)
    77			theta += imag(y) * math.Log(modulus)
    78		}
    79		s, c := math.Sincos(theta)
    80		return complex(r*c, r*s)
    81	}
    82	

View as plain text