ó <Zc@s¶ddlmZmZed„ƒZed„ƒZed„ƒZed„ƒZed„ƒZed„ƒZed„ƒZ ed „ƒZ ed d „ƒZ edd „ƒZ d S(i(tdefunt defun_wrappedc0Csd}d}d}||jkrP|j|ƒsá|j|}|j|j|ƒ|ƒ}|||?}|} } |} x<t| ƒ|kr´| ||?} | | |?} | | 7} qyWd|d>| d>} |j| | ƒ} qù|j|}|j|j|ƒ|ƒ} |j|j|ƒ|ƒ} | | | | |?}| | |d?}|}}|}}d|>|}|}xŠ|d|d|kr|||||?|||||?}}|||||?|||||?}}||7}||7}qwW|d>}|d>}|j|| ƒ}|j|| ƒ}|j||ƒ} n© |j|ƒ r|j|ƒ r|j|}|j|j|ƒ|ƒ}|||?}|} } |j|j|ƒd|ƒ\}}|j||ƒ}}|j||ƒ}}|||||?}|||d?}|||||?|||||?}}|| ||?} xqt| ƒ|krÚ| ||?} | | |?} |||||?|||||?}}| | ||?7} qjW| d>} |j| | ƒ} | |j |dƒ9} | S|j|ƒs~|j|}|j|j|ƒ|ƒ} |j|j|ƒ|ƒ} | | | | |?}| | |d?}|}}|}}|j|j|ƒd|ƒ\}}|j||ƒ}}|j||ƒ}}|||||?}|||d?}|||||?|||||?}}||||?}|||?}xÇ|d|d|kr.|||||?|||||?}}|||||?|||||?}}|||||?|||||?}}||||?7}||||?7}qhW|d>}|d>}|j|| ƒ}|j|| ƒ}|j||ƒ} n{|j|ƒs |j|}|j|j|ƒ|ƒ}|||?}|} } |j}||_|j|ƒ\}}||_|j|j|ƒ|ƒ}}|j|j|ƒ|ƒ}} |j|j|ƒ|ƒ}!}"|j|j|ƒ|ƒ}#}$||| | |"|"|$|$|?}%|| |"|$|d?}&||"| |$|d?}'||$| |"|d?}(||%||&|!|'|#|(|?})||&||%|!|(|#|'|?}*|!|%|#|&||'||(|?}+|!|&|#|%||(||'|?},|)}|*}|+}!|,}#|| ||?}| | ||?}xt| ƒ|krÉ | ||?} | | |?} ||%||&|!|'|#|(|?})||&||%|!|(|#|'|?}*|!|%|#|&||'||(|?}+|!|&|#|%||(||'|?},|)}|*}|+}!|,}#|| ||?7}|| ||?7}qÄW|d>}|d>}|j|| ƒ}|j|| ƒ}|j||ƒ} nà|j|}|j|j|ƒ|ƒ} |j|j|ƒ|ƒ} | | | | |?}| | |d?}|}}|}}|j}||_|j|ƒ\}}||_|j|j|ƒ|ƒ}}|j|j|ƒ|ƒ}} |j|j|ƒ|ƒ}!}"|j|j|ƒ|ƒ}#}$||| | |"|"|$|$|?}%|| |"|$|d?}&||"| |$|d?}'||$| |"|d?}(||%||&|!|'|#|(|?})||&||%|!|(|#|'|?}*|!|%|#|&||'||(|?}+|!|&|#|%||(||'|?},|)}|*}|+}!|,}#d}-|}.| }/||||||?}| |||||?}d}-|||||?}.|||||?}/||||||?}| |||||?}d}-x|d|d|kr¬|||||?|||||?}}|||||?|||||?}}||%||&|!|'|#|(|?})||&||%|!|(|#|'|?}*|!|%|#|&||'||(|?}+|!|&|#|%||(||'|?},|)}|*}|+}!|,}#|||||?}.|||||?}/||||||?7}||||||?7}|-d7}-q W|d>}|d>}|j|| ƒ}|j|| ƒ}|j||ƒ} | |j |dƒ9} | S( Ni iiitpreciii( tzerot_imRtto_fixedt_retabstldexptmpctcos_sintnthroot(0tctxtztqtextra1textra2tMINtwptxtx2tatbtstxretximtx2retx2imtaretbretaimtbimtsretsimtc1ts1tcntsntc2ts2tprec0tcnretc1retcnimtc1imtsnrets1retsnimts1imtc2retc2imts2rets2imtt1tt2tt3tt4tnttermrettermim((s5lib/python2.7/site-packages/mpmath/functions/theta.pyt_jacobi_theta2s†           $--    $--       &&&&&&&&&        &&&&&&&&&  c/Csïd}d}d}|j|ƒ r<|j|ƒ r<|j|}|j|j|ƒ|ƒ}|||?} | } } |j|j|ƒd|ƒ\} } |j| |ƒ}} |j| |ƒ}} | | | | |?}| | |d?}|||||?|||||?}}|d@r>| | |d||?}n| | |d||?}d}xºt| ƒ|kr| | |?} | | |?} |||||?|||||?}}|d@rë|| |d|d||?7}n"|| |d|d||?7}|d7}qaW|d> }|j|| ƒ}nk |j|ƒsX|j|}|j|j|ƒ|ƒ}|j|j|ƒ|ƒ}|||||?}|||d?}|}}|}}|j|j|ƒd|ƒ\} } |j| |ƒ}} |j| |ƒ}} | | | | |?}| | |d?}|||||?|||||?}}|d@r¬| ||d||?}||d||?}n0| ||d||?}||d||?}d}x"|d|d|kr|||||?|||||?}}|||||?|||||?}}|||||?|||||?}}|d@rÅ||||||?7}||||||?7}n4||||||?7}||||||?7}|d7}qåW|d> }|d> }|j|| ƒ}|j|| ƒ}|j||ƒ}nO |j|ƒs§ |j|}|j|j|ƒ|ƒ}|||?} | } } |j}||_|j|ƒ\} } ||_|j|j| ƒ|ƒ}} |j|j| ƒ|ƒ}!}"|j|j| ƒ|ƒ}#}$|j|j| ƒ|ƒ}%}&| | |"|"|$|$|&|&|?}'| |"|$|&|d?}(| |$|"|&|d?})| |&|"|$|d?}*||'|!|(|#|)|%|*|?}+||(|!|'|#|*|%|)|?},|#|'|%|(||)|!|*|?}-|#|(|%|'||*|!|)|?}.|+}|,}!|-}#|.}%|d@r¸|$| |#d||?}|&| |%d||?}n4| | |d||?}|"| |!d||?}d}xat| ƒ|krU | | |?} | | |?} ||'|!|(|#|)|%|*|?}+||(|!|'|#|*|%|)|?},|#|'|%|(||)|!|*|?}-|#|(|%|'||*|!|)|?}.|+}|,}!|-}#|.}%|d@r || |#|||?7}|| |%|||?7}n4|| ||||?7}|| |!|||?7}|d7}qõW|d> }|d> }|j|| ƒ}|j|| ƒ}|j||ƒ}n|j|}|j|j|ƒ|ƒ}|j|j|ƒ|ƒ}|||||?}|||d?}|}}|}}|j}||_|j|ƒ\} } ||_|j|j| ƒ|ƒ}} |j|j| ƒ|ƒ}!}"|j|j| ƒ|ƒ}#}$|j|j| ƒ|ƒ}%}&| | |"|"|$|$|&|&|?}'| |"|$|&|d?}(| |$|"|&|d?})| |&|"|$|d?}*||'|!|(|#|)|%|*|?}+||(|!|'|#|*|%|)|?},|#|'|%|(||)|!|*|?}-|#|(|%|'||*|!|)|?}.|+}|,}!|-}#|.}%|d@rG |$||#||%d||?}|&||%||#d||?}nD| ||||!d||?}|"||!||d||?}d}xÅ|d|d|krX|||||?|||||?}}|||||?|||||?}}||'|!|(|#|)|%|*|?}+||(|!|'|#|*|%|)|?},|#|'|%|(||)|!|*|?}-|#|(|%|'||*|!|)|?}.|+}|,}!|-}#|.}%|d@r|||#||%|||?7}|||#||%|||?7}nD|||||!|||?7}|||||!|||?7}|d7}q” W|d> }|d> }|j|| ƒ}|j|| ƒ}|j||ƒ}||j|dƒ9}|d@r×d |d|Sd d|d|SdS( Nii iRiiiiiÿÿÿÿ( RRRRR RRR R (/R R RtndRRRRRRRRR"R#R$R%R&R'RR9RRRRRRRRR R!R(R)R*R+R,R-R.R/R0R1R2R3R4R5R6R7R8((s5lib/python2.7/site-packages/mpmath/functions/theta.pyt_djacobi_theta2Øsf   $- - %"    $- -        &&&&& &&&&         &&&&& "%""&&&& "%""   c'Csn d}d}d}||jkrM|j|ƒsÞ|j|}|j|j|ƒ|ƒ}|}|} } |||?} x<t| ƒ|kr´| | |?} | | |?} || 7}qyWd|>|d>}|j|| ƒ}|S|j|}|j|j|ƒ|ƒ} |j|j|ƒ|ƒ} | | | | |?}| | |d?}| }}}| }}}xŠ|d|d|krñ|||||?|||||?}}|||||?|||||?}}||7}||7}qhWd|>|d>}|d>}|j|| ƒ}|j|| ƒ}|j||ƒ}|Sn |j|ƒ rº|j|ƒ rºd}|j|}|j|j|ƒ|ƒ}|} } |||?} |j|j|ƒdd|ƒ\}}|j||ƒ}|j||ƒ}|}|}|| ||?7}xqt| ƒ|kr| | |?} | | |?} |||||?|||||?}}|| ||?7}q Wd|>|d>}|j|| ƒ}|S|j|ƒsÞ|j|}|j|j|ƒ|ƒ} |j|j|ƒ|ƒ} | | | | |?}| | |d?}| }}| }}|j|j|ƒdd|ƒ\}}|j||ƒ}|j||ƒ}|}|}|||?}|||?}xÇ|d|d|kr…|||||?|||||?}}|||||?|||||?}}|||||?|||||?}}||||?7}||||?7}q¿Wd|>|d>}|d>}|j|| ƒ}|j|| ƒ}|j||ƒ}|S|j|ƒsZ|j|}|j|j|ƒ|ƒ}|} } |||?} |j}||_|jd|ƒ\}}||_|j|j|ƒ|ƒ}}|j|j|ƒ|ƒ}}|j|j|ƒ|ƒ}} |j|j|ƒ|ƒ}!}"| ||?}| ||?}xt| ƒ|kr| | |?} | | |?} |||||| |!|"|?}#||||||"|!| |?}$|||!||| ||"|?}%|||!|||"|| |?}&|#}|$}|%}|&}!|| ||?7}|| ||?7}qüWd|>|d>}|d>}|j|| ƒ}|j|| ƒ}|j||ƒ}|S|j|}|j|j|ƒ|ƒ} |j|j|ƒ|ƒ} | | | | |?}| | |d?}| }}| }}|j}||_|jd|ƒ\}}||_|j|j|ƒ|ƒ}}|j|j|ƒ|ƒ}}|j|j|ƒ|ƒ}} |j|j|ƒ|ƒ}!}"|||||?}|||||?}xZ|d|d|kr |||||?|||||?}}|||||?|||||?}}|||||| |!|"|?}#||||||"|!| |?}$|||!||| ||"|?}%|||!|||"|| |?}&|#}|$}|%}|&}!||||||?7}||||||?7}q¸ Wd|>|d>}|d>}|j|| ƒ}|j|| ƒ}|j||ƒ}|SdS(Ni iiiiR( RRRRRRRR R ('R R RRRRRRRRRRRRRRR RRR!RRR"R#R$R%R(R)R*R+R,R-R.R/R0R5R6R7R8((s5lib/python2.7/site-packages/mpmath/functions/theta.pyt_jacobi_theta3¡s<        (-   (-      &&&&       &&&& c)Csy d}d}d}|j|ƒ rÙ|j|ƒ rÙd}|j|}|j|j|ƒ|ƒ} | } } | | |?} |j|j|ƒdd|ƒ\} }|j| |ƒ} |j||ƒ}| }|}|d@rï|| ||?7}n|| ||?7}d}xªt| ƒ|kr³| | |?} | | |?} || |||?|| |||?}}|d@rŒ|| ||||?7}n|| ||||?7}|d7}q W||d> }|j|| ƒ}nn |j|ƒsˆ|j|}|j|j|ƒ|ƒ}|j|j|ƒ|ƒ}|||||?}|||d?}|}}|}}|j|j|ƒdd|ƒ\} }|j| |ƒ} |j||ƒ}| }|}|d@rè|||?}|||?}n|||?}|||?}d}x"|d|d|kr.|||||?|||||?}}|||||?|||||?}}|| |||?|| |||?}}|d@rí||||||?7}||||||?7}n4||||||?7}||||||?7}|d7}q W||d> }||d> }|j|| ƒ}|j|| ƒ}|j||ƒ}n¿|j|ƒs|j|}|j|j|ƒ|ƒ} | } } | | |?} |j}||_|jd|ƒ\} }||_|j|j| ƒ|ƒ}}|j|j| ƒ|ƒ}} |j|j|ƒ|ƒ}!}"|j|j|ƒ|ƒ}#}$|d@r°| |!|?}| |#|?}n| ||?}| ||?}d}xat| ƒ|kr5| | |?} | | |?} |||| |!|"|#|$|?}%|| |||!|$|#|"|?}&|!||#| ||"||$|?}'|!| |#|||$||"|?}(|%}|&}|'}!|(}#|d@rô|| |!|||?7}|| |#|||?7}n4|| ||||?7}|| ||||?7}|d7}qÕW||d> }||d> }|j|| ƒ}|j|| ƒ}|j||ƒ}n¸|j|}|j|j|ƒ|ƒ}|j|j|ƒ|ƒ}|||||?}|||d?}|}}|}}|j}||_|jd|ƒ\} }||_|j|j| ƒ|ƒ}}|j|j| ƒ|ƒ}} |j|j|ƒ|ƒ}!}"|j|j|ƒ|ƒ}#}$|d@r÷||!||#|?}||!||#|?}n,|||||?}|||||?}d}xÅ|d|d|krð |||||?|||||?}}|||||?|||||?}}|||| |!|"|#|$|?}%|| |||!|$|#|"|?}&|!||#| ||"||$|?}'|!| |#|||$||"|?}(|%}|&}|'}!|(}#|d@rŸ |||!||#|||?7}|||!||#|||?7}nD||||||||?7}||||||||?7}|d7}q, W||d> }||d> }|j|| ƒ}|j|| ƒ}|j||ƒ}|d@ra d|d|Sdd|d|SdS( s2nd=1,2,3 order of the derivative with respect to zii iiRiiÿÿÿÿN(RRRRR RRR ()R R RR=RRRRRRRRRR"R#R$R%R9RRRRRRRRR R!R(R)R*R+R,R-R.R/R0R5R6R7R8((s5lib/python2.7/site-packages/mpmath/functions/theta.pyt_djacobi_theta3Is2   ( -    ( -       &&&&        &&&& "%"" c Csyt|j|ƒ|j|j|ƒƒddƒ}}|jd|ƒ}|jd|d|ƒ}}||||}||} | } |jt| ƒ} xN|d7}||}|||||} t| ƒ| kráPn| | 7} q¡W|}|jd|ƒ}|}xN|d8}||}|||||} t| ƒ| krQPn| | 7} qW| |j|dƒ} | S(su case ctx._im(z) != 0 theta(2, z, q) = q**1/4 * Sum(q**(n*n + n) * exp(j*(2*n + 1)*z), n=-inf, inf) max term for minimum (2*n+1)*log(q).real - 2* ctx._im(z) n0 = int(ctx._im(z)/log(q).real - 1/2) theta(2, z, q) = q**1/4 * Sum(q**(n*n + n) * exp(j*(2*n + 1)*z), n=n0, inf) + q**1/4 * Sum(q**(n*n + n) * exp(j*(2*n + 1)*z), n, n0-1, -inf) iiiþÿÿÿi(tintRRtlogtexpjtepsRR ( R R RR9tn0te2tete0RttermRteps1((s5lib/python2.7/site-packages/mpmath/functions/theta.pyt_jacobi_theta2aôs4 7     c CsJt|j|ƒ t|j|j|ƒƒƒƒ}}|jd|ƒ}|jd||ƒ}}||||}} |jt| ƒ} xJ|d7}||}||||} t| ƒ| krÌPn|| 7}qW|}|jd|ƒ}|}xJ|d8}||}||||} t| ƒ| kr8Pn|| 7}qüW|S(s¾ case ctx._im(z) != 0 theta3(z, q) = Sum(q**(n*n) * exp(j*2*n*z), n, -inf, inf) max term for n*abs(log(q).real) + ctx._im(z) ~= 0 n0 = int(- ctx._im(z)/abs(log(q).real)) iiiþÿÿÿ(RARRRRBRCRD( R R RR9RERFRGRHRRIRJ((s5lib/python2.7/site-packages/mpmath/functions/theta.pyt_jacobi_theta3as.6    c Cs®t|j|ƒ|j|j|ƒƒddƒ}}|jd|ƒ}|jd|d|ƒ}}||||} d|d|| |} | } |jt| ƒ} x^|d7}||}d|d||||||} t| ƒ| krPn| | 7} q±W|}|jd|ƒ}|}x^|d8}||}d|d||||||} t| ƒ| krPn| | 7} q1W|j|| |j|dƒS(sê case ctx._im(z) != 0 dtheta(2, z, q, nd) = j* q**1/4 * Sum(q**(n*n + n) * (2*n+1)*exp(j*(2*n + 1)*z), n=-inf, inf) max term for (2*n0+1)*log(q).real - 2* ctx._im(z) ~= 0 n0 = int(ctx._im(z)/log(q).real - 1/2) iiiþÿÿÿi( RARRRBRCRDRtjR ( R R RR=R9RERFRGRHRRIRRJ((s5lib/python2.7/site-packages/mpmath/functions/theta.pyt_djacobi_theta2a<s2 7  &  &cCsçt|j|ƒ t|j|j|ƒƒƒƒ}}|jd|ƒ}|jd||ƒ}}||||} ||| } } |dkrª|jt| ƒ} n|jt| ƒ} xy|d7}||}||||} ||| } |dkrt| ƒ} n t| ƒ} | | kr+Pn| | 7} qÀW|}|jd|ƒ}|}xy|d8}||}||||} ||| } |dkrªt| ƒ} n t| ƒ} | | krÆPn| | 7} q[Wd|j|| S(sµ case ctx._im(z) != 0 djtheta3(z, q, nd) = (2*j)**nd * Sum(q**(n*n) * n**nd * exp(j*2*n*z), n, -inf, inf) max term for minimum n*abs(log(q).real) + ctx._im(z) iiiiþÿÿÿ(RARRRRBRCRDRM(R R RR=R9RERFRGRHRRRIRJtaterm((s5lib/python2.7/site-packages/mpmath/functions/theta.pyt_djacobi_theta3a`sD6           ic CsÝ|r|j||||ƒS|j|ƒ}|j|ƒ}t|ƒ|jkretd|jƒ‚nd}|r¾|j|ƒ}|dks¤|dkr¾|dkr¾|dt|ƒ7}q¾nd}d}|j} zù|j|7_|dkrµ|j|ƒr•t|j|ƒƒ|t|j|j |ƒƒƒkrf|j |7_ |j ||j d|ƒ} q²|j d7_ |j ||j d|ƒ} qË|j ||j d|ƒ} n|dkrd|j|ƒrOt|j|ƒƒ|t|j|j |ƒƒƒkr+|j |7_ |j ||ƒ} qa|j d7_ |j ||ƒ} qË|j ||ƒ} ng|d kr|j|ƒrþt|j|ƒƒ|t|j|j |ƒƒƒkrÚ|j |7_ |j||ƒ} q|j d7_ |j||ƒ} qË|j||ƒ} n¸|d krÅ|j|ƒr¯t|j|ƒƒ|t|j|j |ƒƒƒkrŠ|j |7_ |j|| ƒ} qÂ|j d7_ |j|| ƒ} qË|j|| ƒ} nt‚Wd| |_X| S( Nsabs(q) > THETA_Q_LIM = %fi iiiûÿÿÿigà?i2ii(t_djthetatconvertRt THETA_Q_LIMt ValueErrortmagRRRRBtdpsR<tpiRKR?RL( R R9R Rt derivativetextratMtczRR(tres((s5lib/python2.7/site-packages/mpmath/functions/theta.pytjthetaŽsf $  7    7 7 7  c Cs|j|ƒ}|j|ƒ}t|ƒ}t|ƒ|jkrUtd|jƒ‚nd|j|d}|r½|j|ƒ}|dks£|dkr½|dkr½|dt|ƒ7}q½nd}d} |j} z|j|7_|dkr½|j|ƒršt|j|ƒƒ|t|j|j |ƒƒƒkrh|j | 7_ |j ||j d||ƒ} qº|j d7_ |j ||j d||ƒ} qî|j ||j d||ƒ} n1|dkru|j|ƒr]t|j|ƒƒ|t|j|j |ƒƒƒkr6|j | 7_ |j |||ƒ} qr|j d7_ |j |||ƒ} qî|j |||ƒ} ny|d kr-|j|ƒrt|j|ƒƒ|t|j|j |ƒƒƒkrî|j | 7_ |j|||ƒ} q*|j d7_ |j|||ƒ} qî|j|||ƒ} nÁ|d krè|j|ƒrÏt|j|ƒƒ|t|j|j |ƒƒƒkr§|j | 7_ |j|| |ƒ} qå|j d7_ |j|| |ƒ} qî|j|| |ƒ} nt‚Wd| |_X| S( Nsabs(q) > THETA_Q_LIM = %fi iiiûÿÿÿigà?i2ii(RRRARRSRTRRURRRBRVR>RWRNR@RP( R R9R RRXR=RYRZR[RR(R\((s5lib/python2.7/site-packages/mpmath/functions/theta.pyRQÛsd $  7### 7 7 7  N( t functionsRRR<R>R?R@RKRLRNRPR]RQ(((s5lib/python2.7/site-packages/mpmath/functions/theta.pytsÕɨ«( $.L