a ddZ ed?ddZ ed@ddZedAddZedBdd Zed!d"ZedCd#d$Zed%d&Zed'd(Zed)d*Zed+d,Zed-d.Zed/d0Zed1d2Zed3d4Zed5d6Zed7d8Zed9d:ZdS)D)defun defun_wrappedcCsL||d}d||j||dd|}||sH|||j}|S)Nrrrsquare_exp_argsqrtpiZhyp1f1_re_imjctxzz2vrm/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/mpmath/functions/expintegrals.py _erf_complexs  " rcCs|||dkrL||}|j|dd}||||j|dd|}nd||}||sxd| ||j }|S)NrTexactrr) rer fnegexpr r Zhyperurr rr)rrrZnz2rrrr _erfc_complex s & rcCsx||}||r4z ||WSty2Yn0||rn|jsnzt|||jWStylYn0||SN) convert _is_real_typeZ_erfNotImplementedError_is_complex_typeimagtyperealrrrrrrerfs     r'cCsx||}||r4z ||WSty2Yn0||rn|jsnzt|||jWStylYn0||Sr) rr Z_erfcr!r"r#r$r%rr&rrrerfc's     r(FcCsb|jdd}|r6|j|||d}|j|j||d}n|j|||d}|dkr^|j||dd}|S)N)precrTr)r+fmulZfdivone)rrZmultZ reciprocalr+rrrrr 6sr cCsR|s|S||}d||j||dd|}||sN|||j}|S)Nrrrr rrrrerfiBs " r.cs}|ks"|dks"|dkr,dS|s8SdkrFjSdkrTjStdkrvddd}nHdjtdd}||d}j d 7_ fd d |S) Nrrz*erfinv(x) is defined only for -1 <= x <= 1g?goe1?rgϷ?r cs|Sr)r')trxrr]zerfinv..) r Z bad_domaininfninfabslnr signr r+Zfindroot)rr2Zxreaurr1rerfinvLs    (r<cCs>||}|||d d|d||d|jS)Nr)rrr r )rr2musigmarrrnpdf_s r@cCsD||||d}|dkr.|| dSd||dSdS)Nrr=r)r r(r')rr2r>r?r:rrrncdfdsrAc Cs||krd}n|sV|dkr0|dkr0|||}q||||d||d||}n||\}}|dkr||j kr|j } |jd9_|| 7}n|dkr|j|8_||||d||d|} ||||d||d|} | | |}|r||||}|S)Nr=rr)betaZhyp2f1Z nint_distancer+Zeps) rr:bx1Zx2 regularizedrmdhs1s2rrrbetaincls(&     rLNcCs t|}||}|dur&|j}d}n||}||jk}|durN|j}d}n||}||jk}|s|s|r||dkr|jS||dkr|jS|jS||S||kr|jS||||kr||||| S|r|r| |||| S|r| |||S|r| |||SdS)NFr=) boolrzeror5rr-nangammagammainc_gamma3 _upper_gamma _lower_gamma)rrr:rDrFZlower_modifiedZupper_modifiedrrrrQs<      rQcsP|rt|jS|g|jddfdd}||gS)NTrcs2|gd|dggdgd|gf}|fS)Nrrr)rT1GrDrZnegbrrrIs,z_lower_gamma..h)isnpintr$r5r hypercomb)rrrDrFrIrrWrrTs   rTcsH|rzp|rj|r*t|jWSj}z.jd7_||W|_WS|_0n|WSWntyYn0|dkrdkr|dS|dkrΈdksˆdkr|dSjdd |g|z&fd d }j ||gdd WSj yBfd d } ||gYS0dS)Nr/rrr=ryy?Trcs4|d}gd|ggd| ggdfgSNrrU)rrrXr:rZnegarrrIsz_upper_gamma..h) force_seriescsRgd|dg|gggdf} |gd|dggdgd|gf}||fS)Nrr=rrU)rrVT2r]rrrIs.) isintrYr$rNr+Z_gamma_upper_intrPr!rrZ NoConvergence)rrr:rForigrIrr]rrSs6       rSc Cs||}|r|r|jSz|jd7_|j|||d}|j|||d}||}||t||||dkr|W|jd8_S|s|j|d||d}|j|d||d}||}|W|jd8_SW|jd8_n|jd8_0tdS)N)rFir=)rYrNr+rQmagmaxr!) rrr:rDrFZpolerVr_RrrrrRs* $  rRcCs||r6||r6z|||WSty4Yn0||sJ||rR||S||jkrdd|S|dkr||dkrt||jS|j|dS|dkr| | |S|dkr| | |d|dS||d| d||S)Nrr=rr) r`r Z _expint_intr!isnanr5rr$r-rrQ)rnrrrrexpints$  ricCsT|r.|dkr|jS|||||jS|s6|S|dkrD|jS|||S)Nrr)rNeir8Zln2r6)rroffsetrrrlisrlcCs.z ||WSty(||YS0dSr)Z_eir! _ei_genericr&rrrrjs  rjcCs||jkr|S||jkr|jS||dkrzr|j|}|||jddgg||jdd|}||}|dkr||j |j 7}|dkr||j |j 8}|WS|j yYn0|| dddd||j }||r|d||||j|7}n||t|7}|S)NrT)Zmaxtermsr^r=r?)r5r6rNrdr-rZhyperr+rr rraZhyp2f2Zeulerlogr7)rrr\rZimrrrrm&s2     $rmcCs0z ||WSty*|d|YS0dSr[)Z_e1r!rir&rrre1Bs  rpcCs.z ||WSty(||YS0dSr)Z_cir! _ci_genericr&rrrciIs  rrcCs||r.||jkr|jS||jkr.|jdS|j|j|dd}|j|dd}d||||}| |}| |}|dkr|dkr||jd7}|dkr||jd8}|dkr|dkr||jd7}|dkr||jd8}| |r|dkr| |}|S)N?Trrnr=?) isinfr5rNr6r r,rrrjr rr )rrjznjzrzrealzimagrrrrqPs"    rqcCs.z ||WSty(||YS0dSr)Z_sir! _si_genericr&rrrsids  r{cCs||r2||jkrd|jS||jkr2d|jS||dkr|j|j|dd}|j|dd}d||||}| |}|dkr|d|j8}|dkr|d|j7}| |r| |}|S|| dd d d ||SdS) NrnrTryr=rrgп) rur5r r6rdr,rrrjr r hyp1f2)rrrvrwrrxrrrrzks     rzcCs|j|dd}d||||}||}||}|dkrR||jd7}n.|dkrj||jd8}n|dkr||jd7}|S)NTrrnr=rtrs)rrjr rr )rrnzrrxryrrrchis  rcCs||dkrn|j|dd}d||||}||}|dkrT|d|j8}|dkrj|d|j7}|S||dddd ||SdS) NrTrrnr=rtrrg?)rdrrjrr r})rrr~rryrrrshis rc Cs^||jkr|dS||jkr(|dS|j|dd|ddd|jd |d d S) Nrnr|r)rr)r)r)rr))r5mpfr6r r}r&rrrfresnelss     rc CsP||jkr|dS||jkr(|dS||ddd|jd |ddS) Nrnr|)rr)r)r)rr)r)r5rr6r}r r&rrrfresnelcs     r)rF)r=r)r=r)r=rF)r=NF)F)F)F)F)Z functionsrrrrr'r(r r.r<r@rArLrQrTrSrRrirlrjrmrprrrqr{rzrrrrrrrrsl           (  &