o
    1&i>                     @   sn  d Z ddlZddlZddlZddlZddlZddlmZ ddl	m
Z
 ddlmZmZmZ ddlmZ ddlmZ ddlmZ eed	ZejZejZeed
ZejZejZdZdZ dZ!dZ"dd Z#dd Z$dd Z%dd Z&dd Z'dd Z(dd Z)dd Z*dd  Z+d!d" Z,d#d$ Z-d%d& Z.dd(d)Z/e.ej0d* e.ej1d+Z2e.ej3d,Z4e.ej5d-Z6e.ej7d.Z8e.ej9d/Z:e/ej;d0d1Z<e/ej=d2d3Z>e/ej?d4d5Z@e/ejAd6d7ZBe/ejCd8d9ZDe/ejEd:d;ZFe/ejGd<d=ZHe/ejId>d?ZJe/ejKd@dAZLe/ejMdBdCZNe/ejOdDdEZPe/ejQdFdGZRe/ejSdHdIZTe/ejUdJdKZVe/ejWdLdMZXe/ejYdNdOdPZZe/ej[dQdRdPZ\e/ej]dSdTZ^e/ej_dUdVZ`e/ejadWdXdPZbe/ejcdYdZZdd[d\ Zed]d^ Zfd_d` Zgdadb Zhdcdd Zidedf Zjdgdh Zkdidj Zldkdl Zmdmdn Zndodp Zodqdr Zpdsdt Zqdudv Zrdwdx Zsdydz Zte-ejuet d{d| Zve-ejwev d}d~ Zxdd Zyeeydd Zzdd Z{dS )zA
Provide math calls that uses intrinsics or libc math functions.
    N)Constant)impl_ret_untracked)typesconfigcgutils)overload)	signature)trailing_zerosfloat32float64il        l    l            c                 C   s   |  d||S )z<
    Return a condition testing whether *val* is a NaN.
    Zuno)Zfcmp_unorderedbuilderval r   AC:\wamp64\www\opt\env\Lib\site-packages\numba/np/math/mathimpl.pyis_nan'   s   r   c                 C   sH   t |jtd}t |jtd}| d||}| d||}| ||S )zB
    Return a condition testing whether *val* is an infinite.
    z+infz-infz==)r   typefloatfcmp_orderedor_)r   r   Zpos_infZneg_infZisposinfZisneginfr   r   r   is_inf-   s
   r   c                 C   s   |  ||}| d||S )z?
    Return a condition testing whether *val* is a finite.
    ord)fsubr   )r   r   Zval_minus_valr   r   r   	is_finite7   s   r   c                 C   (   |j tj ks
J | |tjdS )z1
    Bitcast a double into a 64-bit integer.
    @   )r   llvmliteir
DoubleTypebitcastIntTyper   r   r   r   f64_as_int64?      r!   c                 C   (   |j tjdksJ | |tj S )z1
    Bitcast a 64-bit integer into a double.
    r   )r   r   r   r    r   r   r   r   r   r   int64_as_f64F      r$   c                 C   r   )z0
    Bitcast a float into a 32-bit integer.
        )r   r   r   	FloatTyper   r    r   r   r   r   f32_as_int32M   r"   r(   c                 C   r#   )z0
    Bitcast a 32-bit integer into a float.
    r&   )r   r   r   r    r   r'   r   r   r   r   int32_as_f32T   r%   r)   c                 C   s   |  t|jd|S )zB
    Negate real number *val*, with proper handling of zeros.
    g       )r   r   r   r   r   r   r   negate_real[   s   r*   c                 C   s(   | j }||dd |D }| ||S )z9
    Call a LLVM intrinsic floating-point operation.
    c                 S   s   g | ]}|j qS r   )r   ).0ar   r   r   
<listcomp>g   s    z%call_fp_intrinsic.<locals>.<listcomp>)moduleZdeclare_intrinsiccall)r   nameargsmodZintrr   r   r   call_fp_intrinsicb   s   r3   c                    s    fdd}|S )z
    Return an implementation factory to convert the single integral input
    argument to a float64, then defer to the *wrapped_impl*.
    c           	         sT   |\}|j d }| |||tj}ttjtj} | |||f}| ||tj|jS )Nr   )r1   castr   r   r   return_type)	contextr   sigr1   r   
input_typeZfpvalZ	inner_sigreswrapped_implr   r   implementerp   s   
z2_unary_int_input_wrapper_impl.<locals>.implementerr   )r;   r<   r   r:   r   _unary_int_input_wrapper_implk   s   r=   c                 C   s   t |}d S N)r=   )fn
float_implimplr   r   r   unary_math_int_implz   s   rB   c                    s    fdd}t | | |S )zO
    Implement the math function *fn* using the LLVM intrinsic *intrcode*.
    c                    s   t | |}t| ||j|S r>   )r3   r   r5   r6   r   r7   r1   r9   intrcoder   r   r@      s   z#unary_math_intr.<locals>.float_impl)rB   )r?   rE   r@   r   rD   r   unary_math_intr~   s   
rF   Fc                    s@   t jr|rtjnd}n|rtjnd} fdd}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 }|jd }| |}tj tji| }tj||g}	t	j
|j |	|d}
||
|f}| ||||j}t| ||j|S )z9
        Implement *fn* for a types.Float input.
        r   r0   )r.   r1   get_value_typer   r
   r   r   r   FunctionTyper   insert_pure_functionr/   r4   r5   r   )r6   r   r7   r1   r   r2   r8   lty	func_namefntyr?   r9   	f32extern	f64externr   r   r@      s   

z%unary_math_extern.<locals>.float_impl)r   ZUSE_LEGACY_TYPE_SYSTEMr   Zint64Znp_int64rB   )r?   rO   rP   Zint_restypeZ	f_restyper@   r   rN   r   unary_math_extern   s   
rQ   z	llvm.fabszllvm.expzllvm.logz
llvm.log10zllvm.sinzllvm.cosZlog1pflog1pZexpm1fexpm1ZerfferfZerfcferfcZtanftanZasinfasinZacosfacosZatanfatanZasinhfasinhZacoshfacoshZatanhfatanhZsinhfsinhZcoshfcoshZtanhftanhZlog2flog2ZceilfceilTZfloorffloorZnumba_gammafZnumba_gammaZsqrtfsqrtZtruncftruncZlgammaflgammac                 C       |\}t ||}t| ||j|S r>   )r   r   r5   r6   r   r7   r1   r   r9   r   r   r   isnan_float_impl      
rh   c                 C      t j}t| ||j|S r>   r   Z	false_bitr   r5   rC   r   r   r   isnan_int_impl      rl   c                 C   rf   r>   )r   r   r5   rg   r   r   r   isinf_float_impl   ri   rn   c                 C   rj   r>   rk   rC   r   r   r   isinf_int_impl   rm   ro   c                 C   rf   r>   )r   r   r5   rg   r   r   r   isfinite_float_impl   ri   rp   c                 C   rj   r>   )r   Ztrue_bitr   r5   rC   r   r   r   isfinite_int_impl   rm   rq   c                 C   sN   |d j }|j}t|tj|||fd|j }|||}t	| ||j
|S )Nr   zllvm.copysign.%s)r   r.   r   get_or_insert_functionr   r   rI   Zintrinsic_namer/   r   r5   )r6   r   r7   r1   rK   r2   r?   r9   r   r   r   copysign_float_impl   s   
rs   c                 C   s   |\}|  |jd }|  |jd }tj||dd}tj||tj|f}dddt	| }	t
|j||	}
||
||f}t||||f}t| ||j|S )Nr      exprG   Znumba_frexpfZnumba_frexpr   double)get_data_typer1   r5   r   Zalloca_oncer   r   rI   ZPointerTypestrrr   r.   r/   Zmake_anonymous_structloadr   )r6   r   r7   r1   r   flttyinttyZexpptrrM   fnamer?   r9   r   r   r   
frexp_impl  s   r~   c                 C   sp   |\}}t | j|j\}}tj|||f}dddt| }	tj|j	||	d}
|
|
||f}t| ||j|S )NZnumba_ldexpfZnumba_ldexprv   rG   )maprx   r1   r   r   rI   ry   r   rJ   r.   r/   r   r5   )r6   r   r7   r1   r   ru   r{   r|   rM   r}   r?   r9   r   r   r   
ldexp_impl  s   r   c                 C   P   |\}}| |tj }| |tj }ttjtjtj}t| ||||fS r>   )sitofpr   r   r   r   r   r   atan2_float_implr6   r   r7   r1   yxfsigr   r   r   atan2_s64_impl(  
   r   c                 C   r   r>   )Zuitofpr   r   r   r   r   r   r   r   r   r   r   atan2_u64_impl0  r   r   c                 C   s~   t |dksJ |j}|jd }| |}tjdtjdi| }tj	|||f}t
j|j||d}	||	|}
t| ||j|
S )N   r   Zatan2fatan2rG   )lenr.   r1   rH   r   r
   r   r   r   rI   r   rJ   r/   r   r5   )r6   r   r7   r1   r2   tyrK   rL   rM   r?   r9   r   r   r   r   8  s   

r   c                 C   `   |\}}| |tj }| |tj }ttjtjtj}t| ||||f}t| ||j	|S r>   
r   r   r   r   r   r   r   hypot_float_implr   r5   r6   r   r7   r1   r   r   r   r9   r   r   r   hypot_s64_implK     r   c                 C   r   r>   r   r   r   r   r   hypot_u64_implU  r   r   c                    s   |j \}}||  kr|jksJ  J |\}}tjtjdkr!dndtjtjdkr+dndi| }t||tjdkrNtj	dkrN|t
d  fdd	}	nfd
d	}	| ||	||}
t| ||j|
S )Nwin32Z_hypotfZhypotfZ_hypothypotr&   infc                    s"   t | s
t |r S | |S r>   )mathisinfr   r   r   
plat_hypotr   r   
hypot_implo  s   
z$hypot_float_impl.<locals>.hypot_implc                    s
    | |S r>   r   r   )r   r   r   r   t  s   
)r1   r5   r   r
   sysplatformr   ZExternalFunctionr   ZMACHINE_BITSr   compile_internalr   )r6   r   r7   r1   xtyytyr   r   r}   r   r9   r   r   r   r   _  s   
r   c                 C   s6   |\}|  |jtjd }|||}t| ||j|S N   Zget_constantr5   r   piZfmulr   r6   r   r7   r1   r   Zcoefr9   r   r   r   radians_float_impl~     r   c                 C   s6   |\}|  |jdtj }|||}t| ||j|S r   r   r   r   r   r   degrees_float_impl  r   r   c                 C   s   |  tj|}|||S r>   )Zget_functionoperatorpow)r6   r   r7   r1   rA   r   r   r   pow_impl  s   
r   c                 C   s   dS )z8Convert integer to unsigned integer of equivalent width.Nr   Tr   r   r   	_unsigned  s   r   c                    s>   | t jv r	dd S | t jv rtt d| j  fddS d S )Nc                 S   s   | S r>   r   r   r   r   r   <lambda>  s    z _unsigned_impl.<locals>.<lambda>zuint{}c                    s    | S r>   r   r   ZnewTr   r   r     s    )r   Zunsigned_domainZsigned_domaingetattrformatZbitwidthr   r   r   r   _unsigned_impl  s   

r   c           
      C   sX   |j \}}||  kr|jksJ  J |\}}dd }| ||||}	t| ||j|	S )Nc           	      S   s   t | }| dkrt|S |dkrt| S t| }t|}t||}ttt| |}ttt||}||krR||krB||}}||8 }t|t|}||ks9t|||}|S )zO
        Stein's algorithm, heavily cribbed from Julia implementation.
        r   )r   absr	   minr   npZright_shiftZ
left_shift)	r,   br   ZzaZzbkuvrr   r   r   gcd  s    

zgcd_impl.<locals>.gcd)r1   r5   r   r   )
r6   r   r7   r1   r   r   r   r   r   r9   r   r   r   gcd_impl  s   
r   )F)|__doc__r   r   r   numpyr   Zllvmlite.irr   r   Znumba.core.imputilsr   Z
numba.corer   r   r   Znumba.core.extendingr   Znumba.core.typingr   Znumba.cpython.unsafe.numbersr	   ZfinfoZdtypeZ_NP_FLT_FINFOmaxZFLT_MAXZtinyZFLT_MINZ_NP_DBL_FINFOZDBL_MAXZDBL_MINZFLOAT_ABS_MASKZFLOAT_SIGN_MASKZDOUBLE_ABS_MASKZDOUBLE_SIGN_MASKr   r   r   r!   r$   r(   r)   r*   r3   r=   rB   rF   rQ   fabsru   Zexp_impllogZlog_impllog10Z
log10_implsinZsin_implcosZcos_implrR   Z
log1p_implrS   Z
expm1_implrT   Zerf_implrU   Z	erfc_implrV   Ztan_implrW   Z	asin_implrX   Z	acos_implrY   Z	atan_implrZ   Z
asinh_implr[   Z
acosh_implr\   Z
atanh_implr]   Z	sinh_implr^   Z	cosh_implr_   Z	tanh_implr`   Z	log2_implra   Z	ceil_implrb   Z
floor_implgammaZ
gamma_implrc   Z	sqrt_implrd   Z
trunc_implre   Zlgamma_implrh   rl   rn   ro   rp   rq   rs   r~   r   r   r   r   r   r   r   r   radiansr   degreesr   r   r   r   r   r   r   r   <module>   s    
	
'


