a | d>} || | } qt|j|}||||} ||||} | | | | |?}| | |d?}|}}|}}d|>|}|}|d|d|kr|||||?|||||?}}|||||?|||||?}}||7}||7}q|d>}|d>}||| }||| }|||} n||sB||sB|j|}||||}|||?}|} } |j|||d\}}|||}}|||}}|||||?}|||d?}|||||?|||||?}}|| ||?} t| |kr| ||?} | | |?} |||||?|||||?}}| | ||?7} q| d>} || | } | | |d9} | S||sF|j|}||||} ||||} | | | | |?}| | |d?}|}}|}}|j|||d\}}|||}}|||}}|||||?}|||d?}|||||?|||||?}}||||?}|||?}|d|d|kr |||||?|||||?}}|||||?|||||?}}|||||?|||||?}}||||?7}||||?7}qR|d>}|d>}||| }||| }|||} n.||sL|j|}||||}|||?}|} } |j}||_||\}}||_||||}}||||}} ||||}!}"||||}#}$||| | |"|"|$|$|?}%|| |"|$|d?}&||"| |$|d?}'||$| |"|d?}(||%||&|!|'|#|(|?})||&||%|!|(|#|'|?}*|!|%|#|&||'||(|?}+|!|&|#|%||(||'|?},|)}|*}|+}!|,}#|| ||?}| | ||?}t| |kr| ||?} | | |?} ||%||&|!|'|#|(|?})||&||%|!|(|#|'|?}*|!|%|#|&||'||(|?}+|!|&|#|%||(||'|?},|)}|*}|+}!|,}#|| ||?7}|| ||?7}q&|d>}|d>}||| }||| }|||} n(|j|}||||} ||||} | | | | |?}| | |d?}|}}|}}|j}||_||\}}||_||||}}||||}} ||||}!}"||||}#}$||| | |"|"|$|$|?}%|| |"|$|d?}&||"| |$|d?}'||$| |"|d?}(||%||&|!|'|#|(|?})||&||%|!|(|#|'|?}*|!|%|#|&||'||(|?}+|!|&|#|%||(||'|?},|)}|*}|+}!|,}#d}-|}.| }/||||||?}| |||||?}d}-|||||?}.|||||?}/||||||?}| |||||?}d}-|d|d|k r<|||||?|||||?}}|||||?|||||?}}||%||&|!|'|#|(|?})||&||%|!|(|#|'|?}*|!|%|#|&||'||(|?}+|!|&|#|%||(||'|?},|)}|*}|+}!|,}#|||||?}.|||||?}/||||||?7}||||||?7}|-d7}- q|d>}|d>}||| }||| }|||} | | |d9} | S) N rprec) zero_imrto_fixed_reabsldexpmpccos_sinnthroot)0ctxzqextra1extra2MINwpxx2absxreximx2rex2imarebreaimbimsresimc1s1cnsnc2s2prec0cnrec1recnimc1imsnres1resnims1imc2rec2ims2res2imt1t2t3t4nZtermreZtermimrCf/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/mpmath/functions/theta.py_jacobi_theta2s             *  *  * *   $$$$$  $$$$ $$$$$$$$$  rEc/Cs d}d}d}||s||s|j|}||||}|||?} | } } |j|||d\} } || |}} || |}} | | | | |?}| | |d?}|||||?|||||?}}|d@r| | |d||?}n| | |d||?}d}t| |kr| | |?} | | |?} |||||?|||||?}}|d@r|| |d|d||?7}n || |d|d||?7}|d7}q|d> }||| } n||s|j|}||||}||||}|||||?}|||d?}|}}|}}|j|||d\} } || |}} || |}} | | | | |?}| | |d?}|||||?|||||?}}|d@r| ||d||?}||d||?}n,| ||d||?}||d||?}d}|d|d|krD|||||?|||||?}}|||||?|||||?}}|||||?|||||?}}|d@r||||||?7}||||||?7}n0||||||?7}||||||?7}|d7}q8|d> }|d> }||| }||| }|||}n||s2|j|}||||}|||?} | } } |j}||_||\} } ||_||| |}} ||| |}!}"||| |}#}$||| |}%}&| | |"|"|$|$|&|&|?}'| |"|$|&|d?}(| |$|"|&|d?})| |&|"|$|d?}*||'|!|(|#|)|%|*|?}+||(|!|'|#|*|%|)|?},|#|'|%|(||)|!|*|?}-|#|(|%|'||*|!|)|?}.|+}|,}!|-}#|.}%|d@r|$| |#d||?}|&| |%d||?}n0| | |d||?}|"| |!d||?}d}t| |kr| | |?} | | |?} ||'|!|(|#|)|%|*|?}+||(|!|'|#|*|%|)|?},|#|'|%|(||)|!|*|?}-|#|(|%|'||*|!|)|?}.|+}|,}!|-}#|.}%|d@r|| |#|||?7}|| |%|||?7}n0|| ||||?7}|| |!|||?7}|d7}q|d> }|d> }||| }||| }|||}nT|j|}||||}||||}|||||?}|||d?}|}}|}}|j}||_||\} } ||_||| |}} ||| |}!}"||| |}#}$||| |}%}&| | |"|"|$|$|&|&|?}'| |"|$|&|d?}(| |$|"|&|d?})| |&|"|$|d?}*||'|!|(|#|)|%|*|?}+||(|!|'|#|*|%|)|?},|#|'|%|(||)|!|*|?}-|#|(|%|'||*|!|)|?}.|+}|,}!|-}#|.}%|d@ rd|$||#||%d||?}|&||%||#d||?}n@| ||||!d||?}|"||!||d||?}d}|d|d|k rJ|||||?|||||?}}|||||?|||||?}}||'|!|(|#|)|%|*|?}+||(|!|'|#|*|%|)|?},|#|'|%|(||)|!|*|?}-|#|(|%|'||*|!|)|?}.|+}|,}!|-}#|.}%|d@ r|||#||%|||?7}|||#||%|||?7}n@|||||!|||?7}|||||!|||?7}|d7} q|d> }|d> }||| }||| }|||}|||d9}|d@ rd |d|Sd d|d|SdS) Nrrrrrr r r ) r rrrrrrrr)/rrrndrrrrrrrrr+r,r-r.r/r0r rBr!r"r#r$r%r&r'r(r)r*r1r2r3r4r5r6r7r8r9r:r;r<r=r>r?r@rArCrCrD_djacobi_theta2sn  *  * "     * *       $$$$$   $$$$     $$$$$  "  $$$$  "       rHc'Cs< d}d}d}||jkr||s|j|}||||}|}|} } |||?} t| |kr| | |?} | | |?} || 7}qVd|>|d>}||| }|S|j|}||||} ||||} | | | | |?}| | |d?}| }}}| }}}|d|d|kr|||||?|||||?}}|||||?|||||?}}||7}||7}qd|>|d>}|d>}||| }||| }|||}|Snb||s||sd}|j|}||||}|} } |||?} |j||d|d\}}|||}|||}|}|}|| ||?7}t| |kr| | |?} | | |?} |||||?|||||?}}|| ||?7}qnd|>|d>}||| }|S||s|j|}||||} ||||} | | | | |?}| | |d?}| }}| }}|j||d|d\}}|||}|||}|}|}|||?}|||?}|d|d|krn|||||?|||||?}}|||||?|||||?}}|||||?|||||?}}||||?7}||||?7}qd|>|d>}|d>}||| }||| }|||}|S||s|j|}||||}|} } |||?} |j}||_|d|\}}||_||||}}||||}}||||}} ||||}!}"| ||?}| ||?}t| |krl| | |?} | | |?} |||||| |!|"|?}#||||||"|!| |?}$|||!||| ||"|?}%|||!|||"|| |?}&|#}|$}|%}|&}!|| ||?7}|| ||?7}qd|>|d>}|d>}||| }||| }|||}|S|j|}||||} ||||} | | | | |?}| | |d?}| }}| }}|j}||_|d|\}}||_||||}}||||}}||||}} ||||}!}"|||||?}|||||?}|d|d|kr|||||?|||||?}}|||||?|||||?}}|||||| |!|"|?}#||||||"|!| |?}$|||!||| ||"|?}%|||!|||"|| |?}&|#}|$}|%}|&}!||||||?7}||||||?7}qd|>|d>}|d>}||| }||| }|||}|SdS)Nrrrrr) r r rrrrrrr)'rrrrrrrrr rrrr!r"r#r$r)r%r&r*r'r(r+r,r-r.r1r2r3r4r5r6r7r8r9r>r?r@rArCrCrD_jacobi_theta3sH                   *      *        $$$$  $$$$ rJc)Cs d}d}d}||sx||sxd}|j|}||||} | } } | | |?} |j||d|d\} }|| |} |||}| }|}|d@r|| ||?7}n|| ||?7}d}t| |krX| | |?} | | |?} || |||?|| |||?}}|d@r6|| ||||?7}n|| ||||?7}|d7}q||d> }||| }n||s|j|}||||}||||}|||||?}|||d?}|}}|}}|j||d|d\} }|| |} |||}| }|}|d@rF|||?}|||?}n|||?}|||?}d}|d|d|krn|||||?|||||?}}|||||?|||||?}}|| |||?|| |||?}}|d@r2||||||?7}||||||?7}n0||||||?7}||||||?7}|d7}qb||d> }||d> }||| }||| }|||}n||s4|j|}||||} | } } | | |?} |j}||_|d|\} }||_||| |}}||| |}} ||||}!}"||||}#}$|d@r| |!|?}| |#|?}n| ||?}| ||?}d}t| |kr| | |?} | | |?} |||| |!|"|#|$|?}%|| |||!|$|#|"|?}&|!||#| ||"||$|?}'|!| |#|||$||"|?}(|%}|&}|'}!|(}#|d@r|| |!|||?7}|| |#|||?7}n0|| ||||?7}|| ||||?7}|d7}q||d> }||d> }||| }||| }|||}n$|j|}||||}||||}|||||?}|||d?}|}}|}}|j}||_|d|\} }||_||| |}}||| |}} ||||}!}"||||}#}$|d@rF||!||#|?}||!||#|?}n(|||||?}|||||?}d}|d|d|k r|||||?|||||?}}|||||?|||||?}}|||| |!|"|#|$|?}%|| |||!|$|#|"|?}&|!||#| ||"||$|?}'|!| |#|||$||"|?}(|%}|&}|'}!|(}#|d@r|||!||#|||?7}|||!||#|||?7}n@||||||||?7}||||||||?7}|d7}qr||d> }||d> }||| }||| }|||}|d@ rrd|d|Sdd|d|SdS) z2nd=1,2,3 order of the derivative with respect to zrrrrIrrrFN)r rrrrrrr))rrrrGrrrr rrrrrr+r,r-r.rBr!r"r#r$r%r&r'r(r)r*r1r2r3r4r5r6r7r8r9r>r?r@rArCrCrD_djacobi_theta3Is:      *          *           $$$$    $$$$  "     rKc Cst|||||d}}|d|}|d|d|}}||||}||} | } |jt| } |d7}||}|||||} t| | krq| | 7} qx|}|d|}|}|d8}||}|||||} t| | krq| | 7} q| ||d} | S)au 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) ?rrr )intr rlogexpjepsrr) rrrrBn0e2ee0rtermr eps1rCrCrD_jacobi_theta2as0 &    rXc Cst|| t|||}}|d|}|d||}}||||}} |jt| } |d7}||}||||} t| | krq|| 7}qn|}|d|}|}|d8}||}||||} t| | krq|| 7}q|S)z 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)) rrrM)rNr rrrOrPrQ) rrrrBrRrSrTrUr rVrWrCrCrD_jacobi_theta3as*(    rYc CsTt|||||d}}|d|}|d|d|}}||||} d|d|| |} | } |jt| } |d7}||}d|d||||||} t| | krq| | 7} q|}|d|}|}|d8}||}d|d||||||} t| | kr0q:| | 7} q|j|| ||dS)z 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) rLrrrMr ) rNr rrOrPrQrjr) rrrrGrBrRrSrTrUrrVr rWrCrCrD_djacobi_theta2a<s. &$  $ r[cCsvt|| t|||}}|d|}|d||}}||||} ||| } } |dkr|jt| } n|jt| } |d7}||}||||} ||| } |dkrt| } nt| } | | krq| | 7} q|}|d|}|}|d8}||}||||} ||| } |dkrBt| } nt| } | | krXqd| | 7} qd|j|| S)z 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) rrIrrM)rNr rrrOrPrQrZ)rrrrGrBrRrSrTrUrr rVrWZatermrCrCrD_djacobi_theta3a`s@(        r\rIc Cs|r|||||S||}||}t||jkrDtd|jd}|r~||}|dksn|dkr~|dkr~|dt|7}d}d}|j} zT|j|7_|dkrB||r(t|||t|| |kr|j |7_ | ||j d|} n$|j d7_ | ||j d|} n| ||j d|} n|dkr||rt|||t|| |kr|j |7_ | ||} n|j d7_ | ||} n | ||} n|d krL||r>t|||t|| |kr"|j |7_ |||} n|j d7_ |||} n |||} n|d kr||rt|||t|| |kr|j |7_ ||| } n|j d7_ ||| } n||| } ntW| |_n| |_0| S Nzabs(q) > THETA_Q_LIM = %frr rrrL2r r )_djthetaconvertr THETA_Q_LIM ValueErrormagrr rrOdpsrEpirXrJrY) rrBrr derivativeextraMczrr1resrCrCrDjthetasf     (  (  (  (rlc Cs ||}||}t|}t||jkr8td|jd|j|d}|r||}|dksp|dkr|dkr|dt|7}d}d} |j} zl|j|7_|dkrJ||r.t|||t|| |kr|j | 7_ | ||j d||} n&|j d7_ | ||j d||} n| ||j d||} n|dkr||rt|||t|| |kr|j | 7_ | |||} n|j d7_ | |||} n| |||} n|d kr`||rPt|||t|| |kr2|j | 7_ ||||} n|j d7_ ||||} n||||} n|d kr||rt|||t|| |kr|j | 7_ ||| |} n|j d7_ ||| |} n||| |} ntW| |_n| |_0| Sr])rarNrrbrcrrdr rrOrerHrfr[rKr\) rrBrrrgrGrhrirjrr1rkrCrCrDr`sd     (  (  (  (r`N)rI)r) Z functionsrrrErHrJrKrXrYr[r\rlr`rCrCrCrDs0 U I ( + '  # - L