31 lines
697 B
Factor
31 lines
697 B
Factor
! Copyright (C) 2004, 2005 Slava Pestov.
|
|
! See http://factor.sf.net/license.txt for BSD license.
|
|
IN: math
|
|
USING: math math-internals kernel ;
|
|
|
|
! Power-related functions:
|
|
! exp log sqrt pow
|
|
|
|
: exp >rect swap fexp swap polar> ;
|
|
: log >polar swap flog swap rect> ;
|
|
|
|
: sqrt ( z -- sqrt )
|
|
>polar dup pi = [
|
|
drop fsqrt 0 swap rect>
|
|
] [
|
|
swap fsqrt swap 2 / polar>
|
|
] ifte ;
|
|
|
|
: ^mag ( w abs arg -- magnitude )
|
|
>r >r >rect swap r> swap fpow r> rot * fexp / ;
|
|
|
|
: ^theta ( w abs arg -- theta )
|
|
>r >r >rect r> flog * swap r> * + ;
|
|
|
|
: ^ ( z w -- z^w )
|
|
over real? over integer? and [
|
|
fpow
|
|
] [
|
|
swap >polar 3dup ^theta >r ^mag r> polar>
|
|
] ifte ;
|