o Tdd<@sdZddlZddlZddlZddlZddlZddlmZddl m Z m Z ddl m Z ddlmZmZmZmZddlmZddlmZdd lmZe d ZejZeed ZejZej Z!eed Z"e"jZ#e"j Z$d Z%dZ&dZ'dZ(ddZ)ddZ*ddZ+ddZ,ddZ-ddZ.ddZ/dd Z0d!d"Z1d#d$Z2d%d&Z3d'd(Z4dd*d+Z5e4ej6d,e4ej7d-Z8e4ej9d.Z:e4ej;d/Ze4ej?d1Z@e5ejAd2d3ZBe5ejCd4d5ZDe5ejEd6d7ZFe5ejGd8d9ZHe5ejId:d;ZJe5ejKdd?ZNe5ejOd@dAZPe5ejQdBdCZRe5ejSdDdEZTe5ejUdFdGZVe5ejWdHdIZXe5ejYdJdKZZe5ej[dLdMZ\e5ej]dNdOZ^e5ej_dPdQdRZ`e5ejadSdTdRZbe5ejcdUdVZde5ejedWdXZfe5ejgdYdZdRZhe5ejid[d\Zjeejkejld]d^Zmeejkejnd_d`ZoeejpejldadbZqeejpejndcddZreejsejldedfZteejsejndgdhZueejvejlejldidjZweejxejldkdlZyeejzejlej{dmdnZ|eej}ej~ej~dodpZeej}ejejdqdrZeej}ejlejldsdtZeejej~ej~dudvZeejejejdwdxZeejejlejldydzZeejejld{d|Ze3ejeeejejld}d~Ze3ejeeejejlejleejejlejnddZddZeeddZddZeejejnejnedS)zA Provide math calls that uses intrinsics or libc math functions. N)Constant)Registryimpl_ret_untracked)typeof)typesutilsconfigcgutils)overload) signature)trailing_zerosmathimplfloat32float64illlcCs|d||S)z< Return a condition testing whether *val* is a NaN. uno)fcmp_unorderedbuildervalri/home/ncw/WWW/www-new/content/articles/pi-bbp/venv/lib/python3.10/site-packages/numba/cpython/mathimpl.pyis_nan(srcCsHt|jtd}t|jtd}|d||}|d||}|||S)zB Return a condition testing whether *val* is an infinite. z+infz-infz==)rtypefloat fcmp_orderedor_)rrpos_infneg_infisposinfisneginfrrris_inf.s  r cCs|||}|d||S)z? Return a condition testing whether *val* is a finite. ord)fsubr)rr val_minus_valrrr is_finite8s r$cC(|jtjks J||tjdS)z1 Bitcast a double into a 64-bit integer. @)rllvmliteir DoubleTypebitcastIntTyperrrr f64_as_int64@r,cC(|jtjdks J||tjS)z1 Bitcast a 64-bit integer into a double. r&)rr'r(r+r*r)rrrr int64_as_f64Gr/cCr%)z0 Bitcast a float into a 32-bit integer. )rr'r( FloatTyper*r+rrrr f32_as_int32Nr-r3cCr.)z0 Bitcast a 32-bit integer into a float. r1)rr'r(r+r*r2rrrr int32_as_f32Ur0r4cCs|t|jd|S)zB Negate real number *val*, with proper handling of zeros. g)r"rrrrrr negate_real\sr5cCs(|j}||dd|D}|||S)z9 Call a LLVM intrinsic floating-point operation. cSsg|]}|jqSr)r).0arrr hsz%call_fp_intrinsic..)moduledeclare_intrinsiccall)rnameargsmodintrrrrcall_fp_intrinsiccs r@csfdd}|S)z Return an implementation factory to convert the single integral input argument to a float64, then defer to the *wrapped_impl*. c sT|\}|jd}||||tj}ttjtj}||||f}|||tj|jS)Nr)r=castrrr return_type) contextrsigr=r input_typefpval inner_sigres wrapped_implrr implementerqs  z2_unary_int_input_wrapper_impl..implementerr)rJrKrrIr_unary_int_input_wrapper_implls rLcCst|}t|tj|dSN)rLlowerrInteger)fn float_implimplrrrunary_math_int_impl{srScs&t|tjfdd}t|||S)zO Implement the math function *fn* using the LLVM intrinsic *intrcode*. cst||}t|||j|SrM)r@rrBrCrrDr=rHintrcoderrrQs z#unary_math_intr..float_impl)rNrFloatrS)rPrVrQrrUrunary_math_intrs  rXFcs:|rtjnd}fdd}t|tj|t|||S)a! Register implementations of Python function *fn* using the external function named *f32extern* and *f64extern* (for float32 and float64 inputs, respectively). If *int_restype* is true, then the function's return value should be integral, otherwise floating-point. Nc s|\}|j}|jd}||}tjtji|}tj||g} t j |j| |d} | | |f} | || ||j } t|||j | S)z9 Implement *fn* for a types.Float input. rr<)r9r=get_value_typerrrr'r( FunctionTyper insert_pure_functionr;rArBr) rCrrDr=rr>rElty func_namefntyrPrH f32extern f64externrrrQs  z%unary_math_extern..float_impl)rint64rNrWrS)rPrarb int_restype f_restyperQrr`runary_math_externs  rfz llvm.fabszllvm.expzllvm.logz llvm.log10zllvm.sinzllvm.coslog1pflog1pexpm1fexpm1erfferferfcferfctanftanasinfasinacosfacosatanfatanasinhfasinhacoshfacoshatanhfatanhsinhfsinhcoshfcoshtanhftanhlog2flog2ceilfceilTfloorffloor numba_gammaf numba_gammasqrtfsqrttruncftrunclgammaflgammacC |\}t||}t|||j|SrM)rrrBrCrrDr=rrHrrrisnan_float_impl rcCtj}t|||j|SrMr false_bitrrBrTrrrisnan_int_implrcCrrM)r rrBrrrrisinf_float_implrrcCrrMrrTrrrisinf_int_implrrcCrrM)r$rrBrrrrisfinite_float_implrrcCrrM)r true_bitrrBrTrrrisfinite_int_implrrcCsN|dj}|j}t|tj|||fd|j}|||}t |||j |S)Nrzllvm.copysign.%s) rr9r get_or_insert_functionr'r(r[intrinsic_namer;rrB)rCrrDr=r]r>rPrHrrrcopysign_float_impls  rc Cs|\}||jd}||jd}tj||dd}tj||tj|f}dddt |} t |j || } | | ||f} t || ||f} t|||j| S)NrexprY numba_frexpf numba_frexprdouble) get_data_typer=rBr alloca_oncer'r(r[ PointerTypestrrr9r;make_anonymous_structloadr) rCrrDr=rflttyinttyexpptrr_fnamerPrHrrr frexp_implsrc Csp|\}}t|j|j\}}tj|||f}dddt|} tj|j || d} | | ||f} t |||j | S)N numba_ldexpf numba_ldexprrY) maprr=r'r(r[rr r\r9r;rrB) rCrrDr=rrrrr_rrPrHrrr ldexp_implsrcCP|\}}||tj}||tj}ttjtjtj}t|||||fSrM)sitofpr'r(r)r rratan2_float_implrCrrDr=yxfsigrrratan2_s64_impl% rcCrrM)uitofpr'r(r)r rrrrrrratan2_u64_impl-rrc Cs~t|dksJ|j}|jd}||}tjdtjdi|}tj |||f}t j |j||d} | | |} t |||j| S)Nratan2fatan2rY)lenr9r=rZrrrr'r(r[r r\r;rrB) rCrrDr=r>tyr]r^r_rPrHrrrr5s   rcC`|\}}||tj}||tj}ttjtjtj}t|||||f}t|||j |SrM rr'r(r)r rrhypot_float_implrrBrCrrDr=rrrrHrrrhypot_s64_implH rcCrrMrrrrrhypot_u64_implRrrc s|j\}}||kr|jksJJ|\}}tjtjdkr!dndtjtjdkr+dndi|}t||tjdkrNtj dkrN|t dfdd } nfd d } | || ||} t |||j| S) Nwin32_hypotfhypotf_hypothypotr1infcs"t|s t|r S||SrM)mathisinfrrr plat_hypotrr hypot_implms z$hypot_float_impl..hypot_implcs ||SrMrr)rrrrrs ) r=rBrrsysplatformrExternalFunctionr MACHINE_BITSrcompile_internalr) rCrrDr=xtyytyrrrrrHrrrr\s    rcCs6|\}||jtjd}|||}t|||j|SN get_constantrBrpifmulrrCrrDr=rcoefrHrrrradians_float_impl{ rcCs6|\}||jdtj}|||}t|||j|Srrrrrrdegrees_float_implrrcCs|tj|}|||SrM) get_functionoperatorpow)rCrrDr=rRrrrpow_impls rcCsdS)z8Convert integer to unsigned integer of equivalent width.NrTrrr _unsignedsrcs>|tjvr ddS|tjvrttd|jfddSdS)NcSs|SrMrrrrrsz _unsigned_impl..zuint{}cs|SrMrrnewTrrrs)runsigned_domain signed_domaingetattrformatbitwidthrrrr_unsigned_impls   rc CsX|j\}}||kr|jksJJ|\}}dd}|||||} t|||j| S)Nc Sst|}|dkr t|S|dkrt|St|}t|}t||}ttt||}ttt||}||krR||krB||}}||8}t|t|}||ks9t|||}|S)zO Stein's algorithm, heavily cribbed from Julia implementation. r)rabsr minrnp right_shift left_shift) r7brzazbkuvrrrrgcds   zgcd_impl..gcd)r=rBrr) rCrrDr=rrrrrrHrrrgcd_impls r )F)__doc__rrrnumpyr llvmlite.irr'rnumba.core.imputilsrrnumbar numba.corerrrr numba.core.extendingr numba.core.typingr numba.cpython.unsafe.numbersr registryrNfinfodtype _NP_FLT_FINFOmaxFLT_MAXtinyFLT_MIN _NP_DBL_FINFODBL_MAXDBL_MINFLOAT_ABS_MASKFLOAT_SIGN_MASKDOUBLE_ABS_MASKDOUBLE_SIGN_MASKrr r$r,r/r3r4r5r@rLrSrXrffabsrexp_implloglog_impllog10 log10_implsinsin_implcoscos_implrh log1p_implrj expm1_implrlerf_implrn erfc_implrptan_implrr asin_implrt acos_implrv atan_implrx asinh_implrz acosh_implr| atanh_implr~ sinh_implr cosh_implr tanh_implr log2_implr ceil_implr floor_implgamma gamma_implr sqrt_implr trunc_implr lgamma_implisnanrWrrOrrrrisfiniterrcopysignrfrexprldexpintcrrrcruint64rrrrrrradiansrdegreesrrrrrr rrrrrs        $