B <ZH*@s`dZddlZddlZddlZdZdZdZdZdZdZ d Z d Z d Z d Z d ZdZddZddZddZyedddZddZWn$eefk rejZejZYnXeejddZeeejZeeejZeejejZeejejZeejejZeejejZeejejZeejejZeej ej Z eej!ej!Z!eej"ej"Z"eej#ej#Z#eej$ddZ$eej%ddZ%eddddZ&ed dd!dZ'd"d#Z(d$d%Z)d&d'Z*d(d)Z+d*d+Z,ee*e,Z-ee)e+Z.d,d-Z/d.d/Z0d0Z1e1 Z2e1e1Z3d1Z4e1d2d2d3d4d5d6d7d8d9d:d;dd?d@dAdBdCdDdEfZ5e6e5dFZ7dGZ8dHZ9dIdJZ:dKdLZ;ee:e;ZdQdRZ?dSdTZ@dUdVdWdXdYdZdUd[d\d]g ZAd^d_ZBd`daZCeeBeCZDdbdcdddedfdgdhdidjg dddkZEd2dldmdndodpdqdrdsdtg dddkZFdudvZGdwdxZHdydzZId{d|ZJd}d~ZKddZLddddddddddddddddddddddddddddddddddddddddddg*ZMdZNdddZOdddZPdddZQddZRdddd dddddddddddddddddddddddgZSdddddddddg dddkZTd2ddddddddg dddkZUdddddddddddg ZVdddddddddddg ZWddZXdS)z This module complements the math and cmath builtin modules by providing fast machine precision versions of some additional functions (gamma, ...) and wrapping math/cmath functions so that they can be called with either real or complex arguments. Ng-DT! @giW @g;f?gw@gw?g9B.?gUk@gox?g]O?g m.f)__name__)r r r r )r r r _mathfun_reals rcsfdd}j|_|S)Nc sFt|tkr|Sy t|Sttfk r@t|SXdS)N)rrrrr)rr)r r r r r +s   z_mathfun..f)r)r r r r )r r r _mathfun*srcsfdd}j|_|S)Nc s>ydd|DSttfk r8dd|DSXdS)Ncss|]}t|VqdS)N)r).0rr r r 8sz(_mathfun_n..f..css|]}t|VqdS)N)r)rrr r r r:s)rr)argsr)r r r r r 6sz_mathfun_n..f)r)r r r r )r r r _mathfun_n5srgcCs|dkrtdt|S)Ngzmath domain error)rmathlog)rr r r math_logBsrcCs|dkrtdt|S)Ngzmath domain error)rrsqrt)rr r r math_sqrtFsrcCs t||S)N)r)ryr r r NsrcCstt|jt|jS)N)rrfloorrealimag)zr r r r`scCstt|jt|jS)N)rrceilrr)rr r r rbscCst|t|fS)N)rcossin)rr r r rescCst|t|fS)N)cmathr!r")rr r r rfscCs|dS)NgUUUUUU?r )rr r r rhscCs|dS)NgUUUUUU?r )rr r r rhsc Cs:d|}y t||Sttfk r4t||SXdS)Ng?)rrrr)rnrr r r nthrootjs  r&cCs|dkrt|  St|d\}}|t9}|d;}|dkrDt|S|dkrVt|S|dkrjt| S|dkr~t| SdS)Nrg?) _sinpi_realdivmodpirr"r!)rr$r%r r r r+qs    r+cCs||dkr| }t|d\}}|t9}|d;}|dkr>t|S|dkrRt| S|dkrft| S|dkrxt|SdS)Nrg?r'r(r)r*)r,r-rr!r")rr$r%r r r _cospi_real|s   r.cCs|jdkrt|  St|jd\}}tt||j}|d;}|dkrPt|S|dkrbt|S|dkrvt| S|dkrt| SdS)Nrg?r'r(r)r*) r_sinpi_complexr,r-rrr#r"r!)rr$r%r r r r/s     r/cCs|jdkr| }t|jd\}}tt||j}|d;}|dkrJt|S|dkr^t| S|dkrrt| S|dkrt|SdS)Nrg?r'r(r)r*)rr,r-rrr#r!r")rr$r%r r r _cospi_complexs    r0cCsPyt|t|Stk rJt|jdkr2dSt|jdkrDdSYnXdS)N y?y)sinpicospi OverflowErrorrr)rr r r tanpisr5cCsPyt|t|Stk rJt|jdkr2dSt|jdkrDdSYnXdS)Nr1yy?)r3r2r4rr)rr r r cotpisr6ggg?P5>cCst|}||kr0|dkr td|tkr0t|S|dkrPtt|td|S|d8}td}x*tdt dD]}|t|||7}qpW|t d}d||dt | |SdS)Nrzgamma function poleg?r(g?r)g' @) intZeroDivisionError_max_exact_gamma _exact_gammar-r+ _gamma_real _lanczos_prange _lanczos_grexp)r_intxr%itr r r r<s r<cCs|jstt|jS|jdkr6tt|td|S|d8}td}x*tdt dD]}|t|||7}qVW|t d}d||dt | |SdS)Ng?r(g?rr)g' @) rrr<rr-r/_gamma_complexr=r>r?r#r@)rr%rBrCr r r rDs  rDcCs*y dt|Stk r$|dSXdS)Ng?g)gammar9)rr r r rgammas rFcCs t|dS)Ng?)rE)rr r r factorialsrGcCs(t|tkrtd|St|j|jS)Ng)rrrZatan2rr)rr r r args  rHc Cs\t|ttfkr>y t|}Wn ttfk r<t|}YnXy|j}|j}Wntk rj|}d}YnX|dkrJt|dkrt t |}|dkr| }|Sd|}y|j}|j}Wntk r|}d}YnXt |}|dkrd}n|dkrd}nd}t dt|dt|tt t||t|dt ||S|dks^|dkrf|dSd} x(t|d kr| t |8} |d7}qlWd |dt ||} d|} | | } | d | 7} | | 9} | d | 7} | | 9} | d | 7} | | 9} | d| 7} | | 9} | d| 7} | | 9} | d| 7} | | 9} | d| 7} | | 9} | d| 7} | | S)Ngg?rr(y?g?g@ gdg?gUUUUUU?gllfgJ?g88Cg#+K?g}<ٰj_gAAz?gSˆB)rrrrrrrAttributeErrorabsrrE conjugaterr-logpir2loggamma) rZxrealZximagvrreZimZrefloorZimsignpsr%Zr2r r r rOsp        J         rOgUUUUUU?ggAAp?gqg|?gYYg^^^^^^ܿg柛n@gLQt:cCst|}||kr |dkr td|dkr>d|}tt|}nd}x |dkrb|d|8}|d7}qDW|d}|}x*tD]"}|||8}|dkrP||9}qvW|t|d|S) Nrzpolygamma poleg?g?gg$@g#B ;)r8r9r-r6 _psi_coeffr)rrArSx2rCcr r r _digamma_real;s&      rXcCs|jstt|jS|jdkr4d|}tt|}nd}x$t|dkr\|d|8}|d7}q:W|d}|}x.tD]&}|||8}t|dkrP||9}qpW|t |d|S)Ng?g?gg$@rTg#B ;) rrrXrr-r6rLrUr#r)rrSrVrCrWr r r _digamma_complexQs"       rYgS?g}^.@gCGb@g߸G?g1R5?gߵy?gLy~F?goO$y?gŕX3=?rIga o @ghr@g-@gF@gv#o?g,?gE㦰u?g󟿑?gan$hI?cCs.|d}x |ddD]}|||}qW|S)Nrr(r )ZcoeffsrrRrWr r r _polyvalsrZcCs~||}|}}d}x`t|dkrt|||9}||||d8}|d7}|||9}||||d7}|d7}qWd|S)Nr(gFFgrL)rrVrPr%rCrSr$r r r _erfc_asymps  r`cCsPt|}||kr|S|dkr(t|  S|dkrH|dkr ?gKX?)g^> ǿgKX?)g\S;S?gQSβ?)g\S;SgQSβ?)g|?gM _6?)g|ꢿgM _6?gD@Fc Csd|}d}}d}x.|||9}||7}t|dkr8P|d7}qW|t||}|rt|tkrr|j}|j}n|}d}|dkr|dkr|td7}n8t|tkr|jdkr|td7}|jdkr|td8}|S)Ng?r(gؗҜ            rpc Csxt|}t|ttfkrNyt|}t}Wn$ttfk rLt|}t}YnX|tkrh|jsht|jd}t| dd S)NgT)rc)rrrrrrrrp)rrkr r r e1Ls  rqgggSbQ?gH*2Q?g|ʎA?gE G?g92"?gcj?g9?g[:?gH1?g,[ ?g?g+9@?g ?gL?g9?gNe?ga;?g?g֕?g1@?g ?gZ>?g?g?gA gpgp6g|Upg%gE3+Ӿgf{g'( gѲG]g/" =g{%gh{srg9D"glXxkվg%ygծ*geuy-g]u:=gh#%.Jg`5'>>g^ҹgU3 gPF; ?g2Mg=Ƨ_ΚcgviT{?g\?gH$?ggI ;g.k^>gPNg/wogS' ?g7z2Mg)cg?g|ɐ?g?c Cst|ttfsry t|}WnVttfk rpy t|}|jsLtt|jSWnttfk rfYnXt YnX|dkrtd|dkrdd| d| St|}||kr|dkrt |S|dsd S|d krd|t |dt d |t d|td|S|dkrH|dkr6tt||dStt||dStt|tt|}dd| d| d | |S) z. Riemann zeta function, real argument r(z zeta(1) poleg?g@g@rr)gg?g@) isinstancerr8rrrrzetarNotImplementedError _zeta_intr-r+r<rZ_zeta_0_zeta_1_zeta_P_zeta_Q)rSr$rr r r rts:   8  rt)F)F)F)Y__doc__operatorrr#r-eZsqrt2Zsqrt5ZphiZln2Zln10rfZcatalanZkhinchinZaperyrNrrrrrrrrrpowr@r!r"ZtanZacosZasinZatanZcoshZsinhZtanhrr Zcos_sinZcbrtr&r+r.r/r0r3r2r5r6rhZNINFZNANZEPSr;lenr:r?r=r<rDrErFrGrHrOrUrXrYZdigammar\r]rZr[r^r`rarbrjrirergrprqrvryrzrxrwrtr r r r sx                    8    %