o "`|-@sjddlmZmZeddZeddZeddZedd Zed;d d Zed dZeddZ edddZ ed?ddZed?ddZed?dd Zed!d"Zed?d#d$Zed%d&Zed'd(Zed)d*Zed+d,Zed-d.Zed/d0Zed1d2Zed3d4Zed5d6Zed7d8Zed9d:ZdS)@)defun defun_wrappedcCsL||d}d||j||dd|}||s$|||j}|S)Nrrrsquare_exp_argsqrtpiZhyp1f1_re_imjctxzz2vr?/usr/lib/python3/dist-packages/mpmath/functions/expintegrals.py _erf_complexs " rcCs|||dkr&||}|j|dd}||||j|dd|}nd||}||s]szerfinv..) r Z bad_domaininfninfabslnr signr r-Zfindroot)rr4Zxreaurr3rerfinvLs    (r=cCs>||}|||d d|d||d|jS)Nr)r rr r )rr4musigmarrrnpdf_s 4rAcCs@||||d}|dkr|| dSd||dS)Nrr>r)r r*r()rr4r?r@r;rrrncdfdsrBc Cs||krd}nv|s+|dkr|dkr|||}ne||||d||d||}nR||\}}|dkrW||j krL|j } |jd9_|| 7}n |dkrW|j|8_||||d||d|} ||||d||d|} | | |}|r||||}|S)Nr>rr)ZbetaZhyp2f1Z nint_distancer-Zeps) rr;bZx1Zx2 regularizedrmdhs1s2rrrbetaincls(&     rKNcCst|}||}|dur|j}d}n ||}||jk}|dur'|j}d}n ||}||jk}|sS|sS|rN||dkrA|jS||dkrK|jS|jS||S||krZ|jS||||krm||||| S|rz|rz| |||| S|r| |||S|r| |||SdS)NFr>) boolr zeror6rr/nangammagammainc_gamma3 _upper_gamma _lower_gamma)rrr;rDrEZlower_modifiedZupper_modifiedrrrrPs>      rPcsP|r t|jS|g|jddfdd}||gS)NTrcs2|gd|dggdgd|gf}|fS)Nrrr)rT1GrDrZnegbrrrHs,z_lower_gamma..h)isnpintr%r6r hypercomb)rrrDrErHrrVrrSs  rScs<|rDz5|r4|rt|jWSj}zjd7_||W|_WS|_w|WStyCYnw|dkrRdkrR|dS|dkrddks^dkrd|dSjdd |g|zfd d }j ||gdd WSj yfd d } ||gYSw)Nr1rrr>ryy?Trcs4|d}gd|ggd| ggdfgSNrrT)rrrWr;rZnegarrrHs,z_upper_gamma..h) force_seriescsRgd|dg|gggdf} |gd|dggdgd|gf}||fS)Nrr>rrT)rrUT2r\rrrHs.) isintrXr%rMr-Z_gamma_upper_intrOr"rrY NoConvergence)rrr;rEZorigrHrr\rrRs6      rRc Cs||}|r |r |jSzd|jd7_|j|||d}|j|||d}||}||t||||dkrD|W|jd8_S|sg|j|d||d}|j|d||d}||} |W|jd8_SW|jd8_t|jd8_w)N)rEir>)rXrMr-rPmagmaxr") rrr;rDrEZpolerUr^RrrrrQs, $  rQcCs||r||rz|||WStyYnw||s%||r)||S||jkr2d|S|dkrK||dkrDt||jS|j|dS|dkrW| | |S|dkri| | |d|dS||d| d||S)Nrr>rr) r_r!Z _expint_intr"Zisnanr6rr%r/rrP)rnrrrrexpints&  rfcCsT|r|dkr |jS|||||jS|s|S|dkr"|jS|||S)Nrr)rMeir9Zln2r7)rroffsetrrrlisricC*z||WSty||YSwr)Z_eir" _ei_genericr'rrrrg   rgcCs||jkr|S||jkr|jS||dkrZz9|j|}|||jddgg||jdd|}||}|dkrA||j |j 7}|dkrM||j |j 8}|WS|j yYYnw|| dddd||j }||r|d||||j|7}|S||t|7}|S)NrT)Zmaxtermsr]r>r?)r6r7rMrbr/rZhyperr-rr rr`Zhyp2f2Zeulerlogr8)rrr[rZimrrrrk&s6     "rkcCs,z||WSty|d|YSwrZ)Z_e1r"rfr'rrre1Bs   rocCrjr)Z_cir" _ci_genericr'rrrciIrlrqcCs||r||jkr |jS||jkr|jdS|j|j|dd}|j|dd}d||||}| |}| |}|dkrW|dkrL||jd7}|dkrW||jd8}|dkrq|dkrf||jd7}|dkrq||jd8}| |r|dkr| |}|S)N?Trrmr>?) isinfr6rMr7r r.rrrgr rr!)rrjznjzrzrealzimagrrrrpPs"    rpcCrjr)Z_sir" _si_genericr'rrrsidrlrzcCs||r||jkrd|jS||jkrd|jS||dkrc|j|j|dd}|j|dd}d||||}| |}|dkrL|d|j8}|dkrW|d|j7}| |ra| |}|S|| dd d d ||S) NrmrTryr>rrgп) rtr6r r7rbr.rrrgr r!hyp1f2)rrrurvrrwrrrryks    rycCs|j|dd}d||||}||}||}|dkr*||jd7}|S|dkr7||jd8}|S|dkrB||jd7}|S)NTrrmr>rsrr)rrgr rr )rrnzrrwrxrrrchis  r~cCs||dkr7|j|dd}d||||}||}|dkr*|d|j8}|dkr5|d|j7}|S||dddd ||S) NrTrrmr>rsrrg?)rbrrgrr r|)rrr}rrxrrrshis rc Cs^||jkr |dS||jkr|dS|j|dd|ddd|jd |d d S) Nrmr{r)rr+r)r+rr+)r6mpfr7r r|r'rrrfresnelss    6rc CsP||jkr |dS||jkr|dS||ddd|jd |ddS) Nrmr{)rr+r)r+rr+r)r6rr7r|r r'rrrfresnelcs    (r)rF)r>r)r>rF)r>NF)F)Z functionsrrrrr(r*r r0r=rArBrKrPrSrRrQrfrirgrkrorqrprzryr~rrrrrrrsn           (  &