B <Z|-@sjddlmZmZeddZeddZeddZedd Zed;d d Zed dZeddZ edddZ 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)N)rr)r)square_exp_argsqrtpihyp1f1_re_imj)ctxzz2vr]szerfinv..) r Z bad_domaininfninfabslnr Zsignrr%Zfindroot)rr+Zxreaur)rr+rerfinvLs     (r3cCs>||}|||d d|d||d|jS)Nr)rrrr )rr+musigmarrrnpdf_s r7cCsD||||d}|dkr.|| dSd||dSdS)Nrr4r)rr"r!)rr+r5r6r1rrrncdfdsr8c 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)Nr4rr)ZbetaZhyp2f1Z nint_distancer%Zeps) rr1bZx1Zx2 regularizedrmdhs1s2rrrbetaincls(&     rANcCs t|}||}|dkr&|j}d}n||}||jk}|dkrN|j}d}n||}||jk}|s|s|r||dkr|jS||dkr|jS|jS||S||kr|jS||||kr||||| S|r|r| |||| S|r| |||S|r| |||SdS)NFr4) boolrzeror-rr'nangammagammainc_gamma3 _upper_gamma _lower_gamma)rrr1r:r;Zlower_modifiedZupper_modifiedrrrrFs<      rFcsP|rt|jS|g|jddfdd}||gS)NT)rcs2|gd|dggdgd|gf}|fS)Nrr)r)rT1)Gr:rnegbrrr>s,z_lower_gamma..h)isnpintrr-r hypercomb)rrr:r;r>r)rKr:rrLrrIs   rIcs8|ryb|r^|r(t|jSj}z$jd7_||S|_Xn |SWntk rYnX|dkrdkr|dS|dkrˆdksdkr|dSjdd |g|y$fd d }j ||gdd Sj k r2fd d } ||gSXdS)Nr)rrr4ryy?T)rcs4|d}gd|ggd| ggdfgS)Nr)r)rr)rKr1rnegarrr>sz_upper_gamma..h) force_seriescsRgd|dg|gggdf} |gd|dggdgd|gf}||fS)Nrr4r)r)rrJT2)rKr1rrPrrr>s.) isintrMrrCr%Z_gamma_upper_intrErrrN NoConvergence)rrr1r;Zorigr>r)rKr1rrPrrHs2      rHc Cs||}|r|r|jSz|jd7_|j|||d}|j|||d}||}||t||||dkrx|S|s|j|d||d}|j|d||d}||}|SWd|jd8_XtdS)N)r;ir4)rMrCr%rFmagmaxr) rrr1r:r;ZpolerJrRRrrrrGs" $rGcCs||r6||r6y |||Stk r4YnX||sJ||rR||S||jkrdd|S|dkr||dkrt||jS|j|dS|dkr| | |S|dkr| | |d|dS||d| d||S)Nrr4rr) rSrZ _expint_intrZisnanr-rrr'rrF)rnrrrrexpints$  rZcCsT|r.|dkr|jS|||||jS|s6|S|dkrD|jS|||S)Nrr)rCeir0Zln2r.)rroffsetrrrlisr]cCs*y ||Stk r$||SXdS)N)Z_eir _ei_generic)rrrrrr[s r[cCs||jkr|S||jkr|jS||dkryp|j|}|||jddgg||jdd|}||}|dkr||j |j 7}|dkr||j |j 8}|S|j k rYnX|| dddd||j }||r|d||||j|7}n||t|7}|S)NrT)ZmaxtermsrQr4rg?)r-r.rCrVr'rZhyperr%r r r rTZhyp2f2Zeulerlogr/)rrrOrZimrrrr^&s*   (  $r^cCs,y ||Stk r&|d|SXdS)Nr)Z_e1rrZ)rrrrre1Bs r`cCs*y ||Stk r$||SXdS)N)Z_cir _ci_generic)rrrrrciIs rbcCs||r.||jkr|jS||jkr.|jdS|j|j|dd}|j|dd}d||||}| |}| |}|dkr|dkr||jd7}|dkr||jd8}|dkr|dkr||jd7}|dkr||jd8}| |r|dkr| |}|S)Ny?T)rg?r4y?) isinfr-rCr.r r&r rr[r r r)rrjznjzrzrealzimagrrrraPs.      racCs*y ||Stk r$||SXdS)N)Z_sir _si_generic)rrrrrsids ricCs||r2||jkrd|jS||jkr2d|jS||dkr|j|j|dd}|j|dd}d||||}| |}|dkr|d|j8}|dkr|d|j7}| |r| |}|S|| dd d d ||SdS) Ng?grT)ryr4)rr)rrgп) rcr-r r.rVr&r rr[r rhyp1f2)rrrdrerrfrrrrhks$      rhcCs|j|dd}d||||}||}||}|dkrR||jd7}n.|dkrj||jd8}n|dkr||jd7}|S)NT)rg?r4y?y?)rr[r r r )rrnzrrfrgrrrchis  rlcCs||dkrn|j|dd}d||||}||}|dkrT|d|j8}|dkrj|d|j7}|S||dddd ||SdS) NrT)rg?r4y?)rr)rrg?)rVrr[r r rj)rrrkrrgrrrshis rmc Cs^||jkr|dS||jkr(|dS|j|dd|ddd|jd |d d S) Ng?gr)rr#)rr)r#rr#)r-mpfr.r rj)rrrrrfresnelss     rrc CsP||jkr|dS||jkr(|dS||ddd|jd |ddS) Ng?g)rr#)rr)r#rr#rp)r-rqr.rjr )rrrrrfresnelcs     rt)rF)r4r)r4r)r4rF)r4NF)F)F)F)F)Z functionsrrrrr!r"rr(r3r7r8rArFrIrHrGrZr]r[r^r`rbrarirhrlrmrrrtrrrrsH       (  &