o
    id<                     @   s  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
mZ ddlmZ ddlmZmZmZmZ ddlmZ ddlmZ dd	lmZ e
d
Zej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  Z0d!d" Z1d#d$ Z2d%d& Z3d'd( Z4dd*d+Z5e4ej6d, e4ej7d-Z8e4ej9d.Z:e4ej;d/Z<e4ej=d0Z>e4ej?d1Z@e5ejAd2d3ZBe5ejCd4d5ZDe5ejEd6d7ZFe5ejGd8d9ZHe5ejId:d;ZJe5ejKd<d=ZLe5ejMd>d?ZNe5ejOd@dAZPe5ejQdBdCZRe5ejSdDdEZTe5ejUdFdGZVe5ejWdHdIZXe5ejYdJdKZZe5ej[dLdMZ\e5ej]dNdOZ^e5ej_dPdQdRZ`e5ejadSdTdRZbe5ejcdUdVZde5ejedWdXZfe5ejgdYdZdRZhe5ejid[d\Zjeejkejld]d^ Zmeejkejnd_d` Zoeejpejldadb Zqeejpejndcdd Zreejsejldedf Zteejsejndgdh Zueejvejlejldidj Zweejxejldkdl Zyeejzejlej{dmdn Z|eej}ej~ej~dodp Zeej}ejejdqdr Zeej}ejlejldsdt Zeejej~ej~dudv Zeejejejdwdx Zeejejlejldydz Zeejejld{d| Ze3eje eejejld}d~ Ze3eje eejejlejleejejlejndd Zdd Zeedd Zdd Zeejejnejne dS )zA
Provide math calls that uses intrinsics or libc math functions.
    N)Constant)Registryimpl_ret_untracked)typeof)typesutilsconfigcgutils)overload)	signature)trailing_zerosZmathimpl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   e/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/numba/cpython/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_finite8   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_f64G      r'   c                 C   r   )z0
    Bitcast a float into a 32-bit integer.
        )r   r   r    	FloatTyper"   r#   r   r   r   r   f32_as_int32N   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_f32U   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>h   s    z%call_fp_intrinsic.<locals>.<listcomp>)moduleZdeclare_intrinsiccall)r   nameargsmodZintrr   r   r   call_fp_intrinsicc   s   r6   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   )r4   castr   r   r   return_type)	contextr   sigr4   r   
input_typeZfpvalZ	inner_sigreswrapped_implr   r   implementerq   s   
z2_unary_int_input_wrapper_impl.<locals>.implementerr   )r>   r?   r   r=   r   _unary_int_input_wrapper_impll   s   r@   c                 C   s   t |}t| tj| d S N)r@   lowerr   Integer)fn
float_implimplr   r   r   unary_math_int_impl{   s   rG   c                    s&   t | tj fdd}t| | |S )zO
    Implement the math function *fn* using the LLVM intrinsic *intrcode*.
    c                    s   t | |}t| ||j|S rA   )r6   r   r8   r9   r   r:   r4   r<   intrcoder   r   rE      s   z#unary_math_intr.<locals>.float_impl)rB   r   FloatrG   )rD   rJ   rE   r   rI   r   unary_math_intr   s   

rL   Fc                    s:   |rt jnd} fdd}t| t j| 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   r3   )r1   r4   get_value_typer   r   r   r   r    FunctionTyper	   insert_pure_functionr2   r7   r8   r   )r9   r   r:   r4   r   r5   r;   lty	func_namefntyrD   r<   	f32extern	f64externr   r   rE      s   

z%unary_math_extern.<locals>.float_impl)r   int64rB   rK   rG   )rD   rU   rV   Zint_restypeZ	f_restyperE   r   rT   r   unary_math_extern   s
   
rX   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 rA   )r   r   r8   r9   r   r:   r4   r   r<   r   r   r   isnan_float_impl      
ro   c                 C      t j}t| ||j|S rA   r	   Z	false_bitr   r8   rH   r   r   r   isnan_int_impl      rs   c                 C   rm   rA   )r   r   r8   rn   r   r   r   isinf_float_impl   rp   ru   c                 C   rq   rA   rr   rH   r   r   r   isinf_int_impl   rt   rv   c                 C   rm   rA   )r   r   r8   rn   r   r   r   isfinite_float_impl   rp   rw   c                 C   rq   rA   )r	   Ztrue_bitr   r8   rH   r   r   r   isfinite_int_impl   rt   rx   c                 C   sN   |d j }|j}t|tj|||fd|j }|||}t	| ||j
|S )Nr   zllvm.copysign.%s)r   r1   r	   get_or_insert_functionr   r    rO   Zintrinsic_namer2   r   r8   )r9   r   r:   r4   rQ   r5   rD   r<   r   r   r   copysign_float_impl   s   
rz   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      exprM   Znumba_frexpfZnumba_frexpr   double)get_data_typer4   r8   r	   Zalloca_oncer   r    rO   ZPointerTypestrry   r1   r2   Zmake_anonymous_structloadr   )r9   r   r:   r4   r   flttyinttyZexpptrrS   fnamerD   r<   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_ldexpr}   rM   )mapr   r4   r   r    rO   r   r	   rP   r1   r2   r   r8   )r9   r   r:   r4   r   r|   r   r   rS   r   rD   r<   r   r   r   
ldexp_impl  s   r   c                 C   P   |\}}| |tj }| |tj }ttjtjtj}t| ||||fS rA   )sitofpr   r    r!   r   r   r   atan2_float_implr9   r   r:   r4   yxfsigr   r   r   atan2_s64_impl%  
   r   c                 C   r   rA   )Zuitofpr   r    r!   r   r   r   r   r   r   r   r   atan2_u64_impl-  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atan2rM   )lenr1   r4   rN   r   r   r   r   r    rO   r	   rP   r2   r   r8   )r9   r   r:   r4   r5   tyrQ   rR   rS   rD   r<   r   r   r   r   5  s   

r   c                 C   `   |\}}| |tj }| |tj }ttjtjtj}t| ||||f}t| ||j	|S rA   
r   r   r    r!   r   r   r   hypot_float_implr   r8   r9   r   r:   r4   r   r   r   r<   r   r   r   hypot_s64_implH     r   c                 C   r   rA   r   r   r   r   r   hypot_u64_implR  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 rA   )mathisinfr   r   r   
plat_hypotr   r   
hypot_implm  s   
z$hypot_float_impl.<locals>.hypot_implc                    s
    | |S rA   r   r   )r   r   r   r   r  s   
)r4   r8   r   r   sysplatformr   ZExternalFunctionr   ZMACHINE_BITSr   compile_internalr   )r9   r   r:   r4   xtyytyr   r   r   r   r<   r   r   r   r   \  s   
r   c                 C   s6   |\}|  |jtjd }|||}t| ||j|S N   Zget_constantr8   r   piZfmulr   r9   r   r:   r4   r   Zcoefr<   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 rA   )Zget_functionoperatorpow)r9   r   r:   r4   rF   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 rA   r   r   r   r   r   <lambda>  s    z _unsigned_impl.<locals>.<lambda>zuint{}c                    s    | S rA   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)r4   r8   r   r   )
r9   r   r:   r4   r   r   r   r   r   r<   r   r   r   gcd_impl  s   
r   )F)__doc__r   r   r   numpyr   Zllvmlite.irr   r   Znumba.core.imputilsr   r   Znumbar   Z
numba.corer   r   r   r	   Znumba.core.extendingr
   Znumba.core.typingr   Znumba.cpython.unsafe.numbersr   registryrB   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-   r6   r@   rG   rL   rX   fabsr|   Zexp_impllogZlog_impllog10Z
log10_implsinZsin_implcosZcos_implrY   Z
log1p_implrZ   Z
expm1_implr[   Zerf_implr\   Z	erfc_implr]   Ztan_implr^   Z	asin_implr_   Z	acos_implr`   Z	atan_implra   Z
asinh_implrb   Z
acosh_implrc   Z
atanh_implrd   Z	sinh_implre   Z	cosh_implrf   Z	tanh_implrg   Z	log2_implrh   Z	ceil_implri   Z
floor_implgammaZ
gamma_implrj   Z	sqrt_implrk   Z
trunc_implrl   Zlgamma_implisnanrK   ro   rC   rs   r   ru   rv   isfiniterw   rx   copysignrz   frexpr   ldexpZintcr   r   rW   r   Zuint64r   r   r   r   r   r   radiansr   degreesr   r   r   r   r   r   r   r   r   r   r   <module>   s    
	
$












	
	



