U
    ,‰dd½  ã                   @   sô	  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mZmZmZmZ d dlmZmZmZmZmZmZ d dlmZmZ d dlmZ dd	„ Zd
d„ Zdd„ Zdd„ Z dd„ Z!dd„ Z"dd„ Z#e
e$ej%ej%ƒdd„ ƒZ&e
ej'ej%ej%ƒe
ej(ej%ej%ƒdd„ ƒƒZ)e
ej*ej%ej%ƒe
ej+ej%ej%ƒdd„ ƒƒZ,e
ej-ej%ej%ƒe
ej.ej%ej%ƒdd„ ƒƒZ/dd„ Z0d d!„ Z1e
ej2ej%ej3ƒe
ej4ej%ej3ƒe
ej2ej5ej3ƒe
ej4ej5ej3ƒd"d#„ ƒƒƒƒZ6d$d%„ Z7d&d'„ Z8d(d)„ Z9d*d+„ Z:d,d-„ Z;d.d/„ Z<d0d1„ Z=d2d3„ Z>d4d5„ Z?d6d7„ Z@d8d9„ ZAd:d;„ ZBd<d=„ ZCd>d?„ ZDd@dA„ ZEdBdC„ ZFdDdE„ ZGdFdG„ ZHdHdI„ ZIdJdK„ ZJdLdM„ ZKdNdO„ ZLdPdQ„ ZMe
ejNejOejOƒe?ƒ e
ejPejOejOƒe@ƒ e
ejQejOejOƒe;ƒ e
ejRejOejOƒe<ƒ e
ejSejOejOƒe=ƒ e
ejTejOejOƒe>ƒ e
ejUejOƒeLƒ e
ejVejOƒeMƒ dRdS„ ZWdTdU„ ZXeWƒ  eXƒ  dVdW„ ZYdXdY„ ZZdZd[„ Z[d\d]„ Z\d^d_„ Z]d`da„ Z^e
e$ej5ej5ƒdÌdbdc„ƒZ_dÍddde„Z`dÎdfdg„Zadhdi„ Zbdjdk„ Zcdldm„ Zddndo„ Zedpdq„ Zfdrds„ Zgdtdu„ Zhdvdw„ Zidxdy„ Zjdzd{„ Zkd|d}„ Zlej5Zme
ejnememƒeYƒ e
ejoememƒeYƒ e
ejpememƒeZƒ e
ejqememƒeZƒ e
ejrememƒe[ƒ e
ejsememƒe[ƒ e
ej'ememƒeaƒ e
ej(ememƒeaƒ e
ej*ememƒe\ƒ e
ej+ememƒe\ƒ e
ej-ememƒe`ƒ e
ej.ememƒe`ƒ e
ej2ememƒebƒ e
ej4ememƒebƒ e
e2ememƒebƒ e
ejNememƒegƒ e
ejPememƒehƒ e
ejQememƒecƒ e
ejRememƒedƒ e
ejSememƒeeƒ e
ejTememƒefƒ e
etemƒeiƒ e
ejUemƒejƒ e
ejVemƒekƒ [meejud~ƒdd€„ ƒZveejudƒd‚dƒ„ ƒZwe
d„ejuƒd…d†„ ƒZxd‡dˆ„ Zyd‰dŠ„ Zzd‹dŒ„ Z{ej5ej%fD ]0Z|ee|d~ƒeyƒ ee|dƒezƒ e
d„e|ƒe{ƒ qe
ej2ejuejuƒe
ej4ejuejuƒe
e2ejuejuƒddŽ„ ƒƒƒZ}dd„ Z~d‘d’„ Zd“d”„ Z€ed•ƒZ‚d–d—„ Zƒd˜d™„ Z„dšd›„ Z…dœd„ Z†dždŸ„ Z‡d d¡„ ZˆejuZme
ejnememƒe~ƒ e
ejoememƒe~ƒ e
ejpememƒeƒ e
ejqememƒeƒ e
ejrememƒe€ƒ e
ejsememƒe€ƒ e
ej*ememƒeƒƒ e
ej+ememƒeƒƒ e
ejUemƒe„ƒ e
ejVemƒe…ƒ e
ejNememƒe†ƒ e
ejPememƒe‡ƒ e
etemƒeˆƒ [me
d¢ej‰ƒe
d¢ejŠƒd£d¤„ ƒƒZ‹d¥d¦„ ZŒe
eej‰ƒd§d¨„ ƒZŽe
eej%ƒd©dª„ ƒZe
eej5ƒd«d¬„ ƒZe
eejuƒd­d®„ ƒZ‘ej%ej5ejufD ]Zme
ej’emƒeŒƒ q<e
ej’ejOƒeŒƒ eej3ej%ƒeej3ej5ƒeej3ejuƒd¯d°„ ƒƒƒZ“eej%ej%ƒd±d²„ ƒZ”eej%ej•ƒd³d´„ ƒZ–eej5ej5ƒdµd¶„ ƒZ—eej%ej5ƒd·d¸„ ƒZ˜eej5ej%ƒd¹dº„ ƒZ™eej5ejuƒeej%ejuƒd»d¼„ ƒƒZšeejuejuƒd½d¾„ ƒZ›eejœej‰ƒd¿dÀ„ ƒZeej‰ejŠƒdÁdÂ„ ƒZžeej3ej‰ƒeejŸej‰ƒdÃdÄ„ ƒƒZ eejuƒdÅdÆ„ ƒZ¡eej%ƒeej5ƒeej‰ƒdÇdÈ„ ƒƒƒZ¢dÉdÊ„ Z£eej5dËƒe£ƒ eej%dËƒe£ƒ dS )Ïé    N)Úir)ÚConstant)Úlower_builtinÚlower_getattrÚlower_getattr_genericÚ
lower_castÚlower_constantÚimpl_ret_borrowedÚimpl_ret_untracked)ÚtypingÚtypesÚutilsÚerrorsÚcgutilsÚoptional)Ú	intrinsicÚoverload_method©Úviewerc                 C   s   | j rdgS g S dS )z;
    Return the modifier flags for integer arithmetic.
    ZnswN)Úsigned)Zrettype© r   ú9/tmp/pip-unpacked-wheel-eu7e0c37/numba/cpython/numbers.pyÚ_int_arith_flags   s    r   c                 C   s\   |\}}|j \}}|  ||||j¡}|  ||||j¡}	|j||	t|jƒd}
t| ||j|
ƒS ©N)Úflags)ÚargsÚcastÚreturn_typeÚaddr   r
   ©ÚcontextÚbuilderÚsigr   ÚvaÚvbÚtaÚtbÚaÚbÚresr   r   r   Úint_add_impl#   s    
r*   c                 C   s\   |\}}|j \}}|  ||||j¡}|  ||||j¡}	|j||	t|jƒd}
t| ||j|
ƒS r   )r   r   r   Úsubr   r
   r   r   r   r   Úint_sub_impl,   s    
r,   c                 C   s\   |\}}|j \}}|  ||||j¡}|  ||||j¡}	|j||	t|jƒd}
t| ||j|
ƒS r   )r   r   r   Úmulr   r
   r   r   r   r   Úint_mul_impl5   s    
r.   c                 C   s`  |j |j kst‚|  d¡}|  d¡}t ||¡}t ||¡}| | d||  |j¡¡| d||  d¡¡¡}	|j| |	¡ddÈ | 	||¡}
| 
||¡}| d| ||¡|¡}| d||¡}| ||¡}| |¡j\}}| | |
|¡ | ||¡ W 5 Q R X |. | | |
|¡|¡ | | ||¡|¡ W 5 Q R X W 5 Q R X W 5 Q R X | |¡| |¡fS )	a@  
    Reference Objects/intobject.c
    xdivy = x / y;
    xmody = (long)(x - (unsigned long)xdivy * y);
    /* If the signs of x and y differ, and the remainder is non-0,
     * C89 doesn't define whether xdivy is now the floor or the
     * ceiling of the infinitely precise quotient.  We want the floor,
     * and we have it iff the remainder's sign matches y's.
     */
    if (xmody && ((y ^ xmody) < 0) /* i.e. and signs differ */) {
        xmody += y;
        --xdivy;
        assert(xmody && ((y ^ xmody) >= 0));
    }
    *p_xdivy = xdivy;
    *p_xmody = xmody;
    r   é   ú==éÿÿÿÿT©Zlikelyú<ú!=)ÚtypeÚAssertionErrorr   Zalloca_once_valueÚand_Úicmp_signedÚminvalÚif_thenÚnot_ZsdivZsremÚxorÚif_elseÚstorer+   r   Úload)r    r!   ÚtyÚxÚyÚZEROÚONEZresdivZresmodZis_overflowZxdivyZxmodyZy_xor_xmody_ltzZxmody_istrueZcondZif_different_signsZif_same_signsr   r   r   Úint_divmod_signed>   s.    

þ2rE   c                 C   s2   |j rt| ||||ƒS | ||¡| ||¡fS dS )zD
    Integer divmod(x, y).  The caller must ensure that y != 0.
    N)r   rE   ZudivZurem)r    r!   r@   rA   rB   r   r   r   Ú
int_divmodw   s    rF   c              
   C   s  |\}}|j \}}|j}	t|	tjƒr*|	j}	|  ||||	¡}
|  ||||	¡}tj||
j	dd}tj||
j	dd}|j
t ||¡dd~\}}|. | j ||f¡s¸| ||¡ | ||¡ W 5 Q R X |2 t| ||	|
|ƒ\}}| ||¡ | ||¡ W 5 Q R X W 5 Q R X ||fS )NÚquot©ÚnameÚremFr2   )r   r   Ú
isinstancer   ZUniTupleÚdtyper   r   Úalloca_oncer5   r=   Úis_scalar_zeroÚerror_modelÚfp_zero_divisionr>   rF   )r    r!   r"   r   Zzerodiv_messager#   r$   r%   r&   r@   r'   r(   rG   rJ   Úif_zeroÚif_non_zeroÚqÚrr   r   r   Ú_int_divmod_impl   s.    
 ÿ rU   c                 C   s0   t | |||dƒ\}}t || |¡| |¡f¡S )Nzinteger divmod by zero)rU   r   Ú
pack_arrayr?   ©r    r!   r"   r   rG   rJ   r   r   r   Úint_divmod_implŸ   s    
ÿÿrX   c                 C   s   t | |||dƒ\}}| |¡S )Nzinteger division by zero©rU   r?   rW   r   r   r   Úint_floordiv_impl¨   s    
ÿrZ   c              	   C   sx   |\}}|j \}}|  ||||j¡}|  ||||j¡}	t ||	¡ | j |d¡ W 5 Q R X | ||	¡}
t| ||j|
ƒS )N©zdivision by zero)	r   r   r   r   rQ   rO   rP   Úfdivr
   r   r   r   r   Úint_truediv_impl°   s    
r]   c                 C   s   t | |||dƒ\}}| |¡S )Nzinteger modulo by zerorY   rW   r   r   r   Úint_rem_impl½   s    
ÿr^   c                 C   s*   t |tjƒr"| jjs"d|jd > S dS d S )Nr1   r/   F)rK   r   ÚIntegerrO   Zraise_on_fp_zero_divisionÚbitwidth)r    r   r   r   r   Ú_get_power_zerodiv_returnÅ   s
    ÿra   c                    sR   t |jd tjƒ‰ |j‰t| ˆƒ‰‡ ‡‡fdd„}|  ||||¡}t| ||j|ƒS )z@
    a ^ b, where a is an integer or real, and b an integer
    r   c                    sº   ˆdƒ}ˆ| ƒ} |dk r`d}| }|dk r.t ‚ˆ rh| dkrJˆrBˆS tdƒ‚| dkrh| dkrhdS nd}|}|dkr€t | t|ƒ¡S |dkrª|d@ r˜|| 9 }|dL }| | 9 } q€|r¶d| S |S )	Nr/   r   Tú&0 cannot be raised to a negative powerr1   Fé   ç      ð?)ÚOverflowErrorÚZeroDivisionErrorÚmathÚpowÚfloat)r'   r(   rT   ÚinvertÚexp©Ú
is_integerÚtpÚzerodiv_returnr   r   Ú	int_powerÖ   s0    
z!int_power_impl.<locals>.int_power)rK   r   r   r_   r   ra   Úcompile_internalr
   )r    r!   r"   r   rp   r)   r   rl   r   Úint_power_implÎ   s    
rr   c                    sü   |j d j}t|tjƒst‚t|ƒdkr,t‚|dk }t|ƒ}|j}t|tj	ƒ‰t
| |ƒ‰|  ˆ |d |j d |¡}|j}‡ ‡fdd„}	|dƒ}
|}|dkrÀ|d@ r¬|	|
|ƒ}
|dL }|	||ƒ}q’|røˆrÖ‡fdd„}ndd„ }|  ˆ |t ||¡|
f¡}
|
S )	zH
    a ^ b, where a is an integer or real, and b a constant integer
    r/   rc   r   c                    s    ˆrˆ   | |¡S ˆ  | |¡S d S ©N)r-   Úfmul)r'   r(   )r!   rm   r   r   r-     s    zstatic_power_impl.<locals>.mulc                    s4   | dkrˆ rˆ S t dƒ‚| dkr,| dkr,dS | S d S )Nr   rb   r/   r1   )rf   ©r'   )ro   r   r   Úinvert_impl$  s    z&static_power_impl.<locals>.invert_implc                 S   s   d|  S )Nrd   r   ru   r   r   r   rv   1  s    )r   ÚvaluerK   ÚnumbersÚIntegralÚNotImplementedErrorÚabsr   r   r_   ra   r   r5   rq   r   Ú	signature)r    r!   r"   r   rk   rj   rn   ÚvalÚltyr-   r)   r'   rv   r   )r!   rm   ro   r   Ústatic_power_implù   s:    


 ÿr   c                 C   s   |j d|žŽ }t| ||j|ƒS ©Nr3   )r3   ©r8   r
   r   ©r    r!   r"   r   r)   r   r   r   Úint_slt_impl:  s    rƒ   c                 C   s   |j d|žŽ }t| ||j|ƒS ©Nú<=)r…   r   r‚   r   r   r   Úint_sle_impl?  s    r†   c                 C   s   |j d|žŽ }t| ||j|ƒS ©Nú>)rˆ   r   r‚   r   r   r   Úint_sgt_implD  s    r‰   c                 C   s   |j d|žŽ }t| ||j|ƒS ©Nú>=)r‹   r   r‚   r   r   r   Úint_sge_implI  s    rŒ   c                 C   s   |j d|žŽ }t| ||j|ƒS r€   ©Úicmp_unsignedr
   r   r‚   r   r   r   Úint_ult_implN  s    r   c                 C   s   |j d|žŽ }t| ||j|ƒS r„   r   r‚   r   r   r   Úint_ule_implS  s    r   c                 C   s   |j d|žŽ }t| ||j|ƒS r‡   r   r‚   r   r   r   Úint_ugt_implX  s    r‘   c                 C   s   |j d|žŽ }t| ||j|ƒS rŠ   r   r‚   r   r   r   Úint_uge_impl]  s    r’   c                 C   s   |j d|žŽ }t| ||j|ƒS ©Nr0   )r0   r   r‚   r   r   r   Úint_eq_implb  s    r”   c                 C   s   |j d|žŽ }t| ||j|ƒS ©Nr4   )r4   r   r‚   r   r   r   Úint_ne_implg  s    r–   c           	      C   sH   |\}t |jd ƒ}| d||¡}| |¡}| |||¡}t| ||j|ƒS )Nr3   )r   r5   r8   ÚnegÚselectr
   r   )	r    r!   r"   r   rA   rC   ZltzZnegatedr)   r   r   r   Úint_abs_impll  s    
r™   c                 C   s   |\}t | ||j|ƒS rs   ©r
   r   )r    r!   r"   r   rA   r   r   r   Úuint_abs_implu  s    r›   c           	      C   sR   |j \}}|\}}|  ||||j¡}|  ||||j¡}| ||¡}t| ||j|ƒS rs   )r   r   r   Zshlr
   ©	r    r!   r"   r   ZvaltyZamttyr}   Úamtr)   r   r   r   Úint_shl_implz  s    
rž   c           	      C   sh   |j \}}|\}}|  ||||j¡}|  ||||j¡}|jjrL| ||¡}n| ||¡}t| ||j|ƒS rs   )r   r   r   r   ZashrZlshrr
   rœ   r   r   r   Úint_shr_implƒ  s    
rŸ   c                 C   sR   |j \}}|\}}|  ||||j¡}|  ||||j¡}	| ||	¡}
t| ||j|
ƒS rs   )r   r   r   r7   r
   ©r    r!   r"   r   ÚatZbtÚavZbvZcavZcbcr)   r   r   r   Úint_and_impl  s    
r£   c                 C   sR   |j \}}|\}}|  ||||j¡}|  ||||j¡}	| ||	¡}
t| ||j|
ƒS rs   )r   r   r   Úor_r
   r    r   r   r   Úint_or_impl˜  s    
r¥   c                 C   sR   |j \}}|\}}|  ||||j¡}|  ||||j¡}	| ||	¡}
t| ||j|
ƒS rs   )r   r   r   r<   r
   r    r   r   r   Úint_xor_impl¡  s    
r¦   c                 C   s:   |j \}|\}| |¡}|  ||||j¡}t| ||j|ƒS rs   )r   r—   r   r   r
   ©r    r!   r"   r   Útypr}   r)   r   r   r   Úint_negate_implª  s
    
r©   c                 C   s0   |j \}|\}|  ||||j¡}t| ||j|ƒS rs   ©r   r   r   r
   r§   r   r   r   Úint_positive_impl³  s    r«   c                 C   sR   |j \}|\}| |t|jtd|jj dƒƒ¡}|  ||||j¡}t| ||j|ƒS )NÚ1é   )	r   r<   r   r5   ÚintÚwidthr   r   r
   r§   r   r   r   Úint_invert_implº  s
    "r°   c              	   C   s`  |\}t |jdƒ}t |jdƒ}t |jdƒ}| d||¡}| d||¡}	t ||j¡}
| d¡}| d¡}| d¡}| d	¡}| d
¡}| |||¡ | |¡ | 	||
¡ | 
|¡ W 5 Q R X | |¡ | |	||¡ W 5 Q R X | |¡ | 	||
¡ | 
|¡ W 5 Q R X | |¡ | 	||
¡ | 
|¡ W 5 Q R X | |¡ | |
¡}t| ||j|ƒS )z
    np.sign(int)
    r/   r1   r   r0   rˆ   z.zeroz.postestz.posz.negz.exit)r   r5   rŽ   r8   r   rM   Úappend_basic_blockZcbranchZ
goto_blockr>   ÚbranchZposition_at_endr?   r
   r   )r    r!   r"   r   rA   ÚPOSÚNEGrC   Zcmp_zeroZcmp_posÚpresultZbb_zeroZ
bb_postestZbb_posZbb_negZbb_exitr)   r   r   r   Úint_sign_implÃ  s6    






r¶   c                 C   s:   |j \}|\}|  ||||j¡}| |¡}t| ||j|ƒS rs   )r   r   r   r—   r
   r§   r   r   r   Úbool_negate_implí  s
    
r·   c                 C   s0   |j \}|\}|  ||||j¡}t| ||j|ƒS rs   rª   r§   r   r   r   Úbool_unary_positive_implõ  s    r¸   c                  C   sÊ  t j} ttj| | ƒtƒ ttj| | ƒtƒ ttj| | ƒtƒ ttj	| | ƒtƒ ttj
| | ƒtƒ ttj| | ƒtƒ ttj| | ƒtƒ ttj| | ƒtƒ ttj| | ƒtƒ ttj| | ƒtƒ ttj| | ƒtƒ ttj| | ƒtƒ ttj| ƒtƒ ttj| ƒtƒ ttj| | ƒtƒ ttj| | ƒtƒ tt| | ƒtƒ t jD ]–} ttj| | ƒt ƒ ttj!| | ƒt"ƒ ttj#| | ƒt$ƒ ttj%| | ƒt&ƒ ttjt j'| ƒtƒ ttjt j'| ƒtƒ ttt j'| ƒtƒ tt(| ƒt)ƒ q8ttjt j*t j*ƒt+ƒ ttj#t j*t j*ƒt+ƒ ttj!t j*t j*ƒt+ƒ ttj%t j*t j*ƒt+ƒ t j,D ]–} ttj| | ƒt+ƒ ttj!| | ƒt-ƒ ttj#| | ƒt.ƒ ttj%| | ƒt/ƒ ttjt j'| ƒtƒ ttjt j'| ƒtƒ ttt j'| ƒtƒ tt(| ƒt0ƒ q.d S rs   )1r   r_   r   Úoperatorr   r*   Úiaddr+   r,   Úisubr-   r.   ÚimulÚeqr”   Úner–   Úlshiftrž   ÚilshiftÚrshiftrŸ   Úirshiftr—   r©   Úposr«   rh   rr   ÚipowZunsigned_domainÚltr   Úler   Úgtr‘   Úger’   ÚFloatr{   r›   ÚIntegerLiteralrƒ   Zsigned_domainr†   r‰   rŒ   r™   ©r@   r   r   r   Ú_implement_integer_operators  sP    

rÌ   c                  C   s’   t jt jfD ]€} ttj| | ƒtƒ ttj| | ƒtƒ ttj| | ƒt	ƒ ttj
| | ƒt	ƒ ttj| | ƒtƒ ttj| | ƒtƒ ttj| ƒtƒ qd S rs   )r   ÚBooleanr_   r   r¹   r7   r£   Úiandr¤   r¥   Úiorr<   r¦   Úixorrj   r°   rË   r   r   r   Ú_implement_bitwise_operators6  s    rÑ   c                 C   s   |j |Ž }t| ||j|ƒS rs   )Úfaddr
   r   r‚   r   r   r   Úreal_add_implF  s    
rÓ   c                 C   s   |j |Ž }t| ||j|ƒS rs   )Úfsubr
   r   r‚   r   r   r   Úreal_sub_implK  s    
rÕ   c                 C   s   |j |Ž }t| ||j|ƒS rs   )rt   r
   r   r‚   r   r   r   Úreal_mul_implP  s    
rÖ   c              	   C   sD   t  ||d ¡ | j |d¡ W 5 Q R X |j|Ž }t| ||j|ƒS )Nr/   r[   )r   rQ   rO   rP   r\   r
   r   r‚   r   r   r   Úreal_div_implU  s    
r×   c                 C   sÎ   |j |j kst‚|j }|j}|  d|j g¡}t |||t |¡f¡}t |||¡}|j	r¢d|_
t | d¡¡}	|j\}
}}t| |	|
|ƒ\}}|	 ||¡ |	 |¡ t ||¡}| ||||f¡}|| |¡fS )Nz.numba.python.remZlinkonce_odrÚentry)r5   r6   ÚmoduleZmanglerr   ÚFunctionTypeZPointerTyper   Úget_or_insert_functionZis_declarationÚlinkageZ	IRBuilderr±   r   Úreal_divmod_func_bodyr>   ÚretrM   Úcallr?   )r    r!   rA   rB   ZfloattyrÙ   ÚfnameÚfntyÚfnZ	fnbuilderZfxZfyÚpmodÚdivÚmodZquotientr   r   r   Úreal_divmod\  s     
ræ   c                 C   sn  t  ||j¡}t  ||j¡}t  ||j¡}| ||¡}| | ||¡|¡}| ||¡ | ||¡ | d¡}	| d¡}
| d¡}| d||	¡}| d||	¡}| d||	¡}|j	|dd\}}|R | 
d||¡}| |¡. | | ||¡|¡ | | ||¡|¡ W 5 Q R X W 5 Q R X |  | ||
|	¡}| ||¡ W 5 Q R X W 5 Q R X ~~| |¡}| d||	¡}| |¡ tjtjdœ}|t|jƒ }|  tjt ||¡¡}|||gƒ}| ||¡}| ||¡}t|jd	ƒ}| d
||¡}| |||¡}| ||¡ W 5 Q R X t  ||¡> | ||¡}| ||¡ | | ||¡|¡}| ||¡ W 5 Q R X | |¡| |¡fS )Nç        g       €rd   r4   r3   Tr2   )ri   Údoubleg      à?rˆ   )r   rM   r5   Úfremr\   rÔ   r>   Úfcmp_unorderedÚfcmp_orderedr=   rŽ   r:   rÒ   r˜   r?   r   Zfloat32Zfloat64ÚstrÚget_functionrg   Úfloorr   r|   r   Zifnotrt   )r    r!   ZvxZwxrã   ZpdivZ	pfloordivrå   rä   rC   ZNZEROrD   Z
mod_istrueZwx_ltzZmod_ltzZif_nonzero_modZif_zero_modZwx_ltz_ne_mod_ltzZ
div_istrueZrealtypemapZrealtypeZfloorfnÚfloordivZfloordivdiffZfloordivincrZHALFÚpredr   r   r   rÝ   r  s\    .


( 
ÿ
ÿrÝ   c              
   C   sø   |\}}t j||jdd}t j||jdd}|jt  ||¡dd”\}	}
|	F | j |d|¡sŽ| ||¡}| ||¡}| 	||¡ | 	||¡ W 5 Q R X |
0 t
| |||ƒ\}}| 	||¡ | 	||¡ W 5 Q R X W 5 Q R X t  || |¡| |¡f¡S )NrG   rH   rJ   Fr2   ©zmodulo by zero)r   rM   r5   r=   rN   rO   rP   r\   ré   r>   ræ   rV   r?   )r    r!   r"   r   ÚlocrA   rB   rG   rJ   rQ   rR   rS   rT   r   r   r   Úreal_divmod_implÞ  s,      ÿ ÿró   c              
   C   s¸   |\}}t  ||j¡}|jt  ||¡ddp\}}	|. | j |d|¡s`| ||¡}
| |
|¡ W 5 Q R X |	$ t	| |||ƒ\}}
| |
|¡ W 5 Q R X W 5 Q R X t
| ||j| |¡ƒS )NFr2   rñ   )r   rM   r5   r=   rN   rO   rP   ré   r>   ræ   r
   r   r?   )r    r!   r"   r   rò   rA   rB   r)   rQ   rR   rJ   Ú_r   r   r   Úreal_mod_implø  s$      ÿ 
ÿrõ   c              
   C   s¸   |\}}t  ||j¡}|jt  ||¡ddp\}}	|. | j |d|¡s`| ||¡}
| |
|¡ W 5 Q R X |	$ t	| |||ƒ\}
}| |
|¡ W 5 Q R X W 5 Q R X t
| ||j| |¡ƒS )NFr2   r[   )r   rM   r5   r=   rN   rO   rP   r\   r>   ræ   r
   r   r?   )r    r!   r"   r   rò   rA   rB   r)   rQ   rR   rG   rô   r   r   r   Úreal_floordiv_impl  s$      ÿ 
ÿrö   c           
      C   s^   |\}}|j }| jr.|  tj|¡}|||ƒ}n | d|jg¡}	| |	||f¡}t| ||j	|ƒS )Nzllvm.pow)
rÙ   Zimplement_powi_as_math_callrí   rg   rh   Zdeclare_intrinsicr5   rß   r
   r   )
r    r!   r"   r   rA   rB   rÙ   Úimpr)   râ   r   r   r   Úreal_power_impl  s    rø   c                 C   s   |j d|žŽ }t| ||j|ƒS r€   ©rë   r
   r   r‚   r   r   r   Úreal_lt_impl*  s    rú   c                 C   s   |j d|žŽ }t| ||j|ƒS r„   rù   r‚   r   r   r   Úreal_le_impl/  s    rû   c                 C   s   |j d|žŽ }t| ||j|ƒS r‡   rù   r‚   r   r   r   Úreal_gt_impl4  s    rü   c                 C   s   |j d|žŽ }t| ||j|ƒS rŠ   rù   r‚   r   r   r   Úreal_ge_impl9  s    rý   c                 C   s   |j d|žŽ }t| ||j|ƒS r“   rù   r‚   r   r   r   Úreal_eq_impl>  s    rþ   c                 C   s   |j d|žŽ }t| ||j|ƒS r•   )rê   r
   r   r‚   r   r   r   Úreal_ne_implC  s    rÿ   c                 C   s,   |j \}t ||¡}|  tj|¡}|||ƒS rs   )r   r   r|   rí   rg   Úfabs)r    r!   r"   r   r@   Úimplr   r   r   Úreal_abs_implH  s    r  c                 C   s,   ddl m} | ||d ¡}t| ||j|ƒS ©Nr   ©Úmathimpl)Únumba.cpythonr  Únegate_realr
   r   )r    r!   r"   r   r  r)   r   r   r   Úreal_negate_implO  s    r  c                 C   s0   |j \}|\}|  ||||j¡}t| ||j|ƒS rs   rª   r§   r   r   r   Úreal_positive_implU  s    r	  c                 C   s  |\}t |jdƒ}t |jdƒ}t |jdƒ}t ||j¡}| d||¡}	| d||¡}
| |	¡ˆ\}}| | ||¡ W 5 Q R X |X | |
¡B\}}| | ||¡ W 5 Q R X | | ||¡ W 5 Q R X W 5 Q R X W 5 Q R X W 5 Q R X | |¡}t| ||j	|ƒS )z
    np.sign(float)
    r/   r1   r   rˆ   r3   )
r   r5   r   rM   rë   r=   r>   r?   r
   r   )r    r!   r"   r   rA   r³   r´   rC   rµ   Zis_posZis_negZgt_zeroZnot_gt_zeroZlt_zeroZnot_lt_zeror)   r   r   r   Úreal_sign_impl\  s$    4
r
  Úrealc                 C   s$   | j |||d}|j}t| |||ƒS ©N©rw   )Úmake_complexr  r
   ©r    r!   r¨   rw   Zcplxr)   r   r   r   Úcomplex_real_impl›  s    r  Úimagc                 C   s$   | j |||d}|j}t| |||ƒS r  )r  r  r
   r  r   r   r   Úcomplex_imag_impl¡  s    r  zcomplex.conjugatec                 C   sL   ddl m} |  ||jd |d ¡}| ||j¡|_| ¡ }t| ||j|ƒS r  )	r  r  r  r   r  r  Ú	_getvaluer
   r   )r    r!   r"   r   r  Úzr)   r   r   r   Úcomplex_conjugate_impl§  s
    r  c                 C   s   t | |||ƒS rs   )r
   )r    r!   r¨   rw   r   r   r   Úreal_real_impl¯  s    r  c                 C   s   t  |j¡}t| |||ƒS rs   )r   Zget_null_valuer5   r
   )r    r!   r¨   rw   r)   r   r   r   Úreal_imag_impl²  s    r  c                 C   s   t | ||j|d ƒS ©Nr   rš   ©r    r!   r"   r   r   r   r   Úreal_conjugate_impl¶  s    r  c              
   C   sx  |\}}|j d }|j}| j|||d}| j|||d}	|  ||¡}
|j}| ¡ }|	 ¡ }|
 ¡ }|  |d¡}|  |d¡}| d|	j|¡}| d|	j|¡}| 	||¡}| 
|¡¨\}}|8 t| ||||fƒ}| j|||d}|j|
_|j|
_W 5 Q R X |R tjdtjdi| }t t ¡ |jgd ¡}t |||¡}| ||||f¡ W 5 Q R X W 5 Q R X | |¡}t| ||j|ƒS )Nr   r  r­   r0   Znumba_cpowfZ
numba_cpowé   )r   Úunderlying_floatZmake_helperrÙ   Z_getpointerÚget_constantrë   r  r  r7   r=   Úcomplex_mul_implr   Z	complex64Z
complex128r   rÚ   ZVoidTyper5   r   rÛ   rß   r?   r
   r   )r    r!   r"   r   ÚcaÚcbr@   Úftyr'   r(   ÚcrÙ   ÚpaZpbZpcZTWOrC   Zb_real_is_twoZb_imag_is_zeroZb_is_twoZthenZ	otherwiser)   ZcresÚ	func_namerá   Zcpowr   r   r   Úcomplex_power_impl¿  sD    
  þý&
r%  c                 C   sŠ   |\}}|j d }| j|||d}| j|||d}|  ||¡}	|j}
|j}|j}|j}| |
|¡|	_| ||¡|	_|	 ¡ }t| ||j|ƒS ©Nr   r  )r   r  r  r  rÒ   r  r
   r   ©r    r!   r"   r   ÚcxÚcyr@   rA   rB   r  r'   r(   r"  Údr)   r   r   r   Úcomplex_add_implë  s    
r+  c                 C   sŠ   |\}}|j d }| j|||d}| j|||d}|  ||¡}	|j}
|j}|j}|j}| |
|¡|	_| ||¡|	_|	 ¡ }t| ||j|ƒS r&  )r   r  r  r  rÔ   r  r
   r   r'  r   r   r   Úcomplex_sub_implû  s    
r,  c                 C   sº   |\}}|j d }| j|||d}| j|||d}|  ||¡}	|j}
|j}|j}|j}| |
|¡}| ||¡}| |
|¡}| ||¡}| ||¡|	_| ||¡|	_|	 ¡ }t| ||j	|ƒS )z'
    (a+bi)(c+di)=(ac-bd)+i(ad+bc)
    r   r  )
r   r  r  r  rt   rÔ   rÒ   r  r
   r   )r    r!   r"   r   r(  r)  r@   rA   rB   r  r'   r(   r"  r*  ÚacZbdÚadZbcr)   r   r   r   r    s"    
r  Únanc                 C   s(   dd„ }|   ||||¡}t| ||j|ƒS )Nc                 S   sÌ   | j }| j}|j }|j}|s(|s(tdƒ‚t|ƒt|ƒkr||sFtttƒS || }|||  }t|||  | |||  | ƒS |sŠtttƒS || }|| | }t| j | | j | | j| | j  | ƒS d S )Nzcomplex division by zero)r  r  rf   r{   ÚcomplexÚNAN)r'   r(   ZarealZaimagZbrealZbimagZratioZdenomr   r   r   Úcomplex_div%  s.    
þ
þz%complex_div_impl.<locals>.complex_div©rq   r
   r   )r    r!   r"   r   r2  r)   r   r   r   Úcomplex_div_impl$  s    r4  c           	      C   sn   ddl m} |j\}|\}| j|||d}|  ||¡}| ||j¡|_| ||j¡|_| ¡ }t| ||j	|ƒS )Nr   r  r  )
r  r  r   r  r  r  r  r  r
   r   )	r    r!   r"   r   r  r¨   r}   Úcmplxr)   r   r   r   Úcomplex_negate_implD  s    r6  c                 C   s   |\}t | ||j|ƒS rs   rš   ©r    r!   r"   r   r}   r   r   r   Úcomplex_positive_implP  s    r8  c                 C   sr   |\}}|j d }| j|||d}| j|||d}| d|j|j¡}	| d|j|j¡}
| |	|
¡}t| ||j|ƒS )Nr   r  r0   )r   r  rë   r  r  r7   r
   r   )r    r!   r"   r   r(  r)  r¨   rA   rB   Zreals_are_eqZimags_are_eqr)   r   r   r   Úcomplex_eq_implU  s    
r9  c                 C   sr   |\}}|j d }| j|||d}| j|||d}| d|j|j¡}	| d|j|j¡}
| |	|
¡}t| ||j|ƒS )Nr   r  r4   )r   r  rê   r  r  r¤   r
   r   )r    r!   r"   r   r(  r)  r¨   rA   rB   Zreals_are_neZimags_are_ner)   r   r   r   Úcomplex_ne_impla  s    
r:  c                 C   s(   dd„ }|   ||||¡}t| ||j|ƒS )z)
    abs(z) := hypot(z.real, z.imag)
    c                 S   s   t  | j| j¡S rs   )rg   Úhypotr  r  )r  r   r   r   Úcomplex_absq  s    z%complex_abs_impl.<locals>.complex_absr3  )r    r!   r"   r   r<  r)   r   r   r   Úcomplex_abs_implm  s    r=  znumber.itemc                 C   s   |d S )z;
    The no-op .item() method on booleans and numbers.
    r   r   r  r   r   r   Únumber_item_implŽ  s    r>  c                 C   s:   |j \}|\}|  ||||j¡}| |¡}t| ||j|ƒS rs   )r   r   r   r;   r
   )r    r!   r"   r   r¨   r}   Zistruer)   r   r   r   Únumber_not_implš  s
    
r?  c                 C   s
   |\}|S rs   r   r7  r   r   r   Úbool_as_bool¡  s    r@  c                 C   s   |\}|  d|t|jdƒ¡S )Nr4   r   )rŽ   r   r5   r7  r   r   r   Úint_as_bool¦  s    rA  c                 C   s   |\}|  d|t|jdƒ¡S )Nr4   rç   )rê   r   r5   r7  r   r   r   Úfloat_as_bool«  s    rB  c                 C   s^   |j \}|\}|  |||¡}|j|j }}t|jdƒ}	| d||	¡}
| d||	¡}| |
|¡S )Nrç   r4   )r   r  r  r  r   r5   rê   r¤   )r    r!   r"   r   r¨   r}   r5  r  r  ZzeroZreal_istrueZimag_istruer   r   r   Úcomplex_as_bool°  s    rC  c                 C   s$   |   ||j|j¡}|  |||j|¡S rs   )Úget_constant_genericÚliteral_typeÚliteral_valuer   ©r    r!   ÚfromtyÚtotyr}   Zlitr   r   r   Úliteral_int_to_numberÈ  s    ýrJ  c                 C   s\   |j |j kr|S |j |j k r.| ||  |¡¡S |jrF| ||  |¡¡S | ||  |¡¡S d S rs   )r`   ÚtruncÚget_value_typer   ZsextÚzext©r    r!   rH  rI  r}   r   r   r   Úinteger_to_integerÔ  s    rO  c                 C   s   |  ||  |¡¡S rs   )ZinttoptrrL  rN  r   r   r   Úinteger_to_voidptrã  s    rP  c                 C   s2   |   |¡}|j|jk r"| ||¡S | ||¡S d S rs   )rL  r`   ZfpextZfptrunc©r    r!   rH  rI  r}   r~   r   r   r   Úfloat_to_floatç  s    
rR  c                 C   s,   |   |¡}|jr| ||¡S | ||¡S d S rs   )rL  r   ZsitofpZuitofprQ  r   r   r   Úinteger_to_floatï  s    
rS  c                 C   s,   |   |¡}|jr| ||¡S | ||¡S d S rs   )rL  r   ZfptosiZfptouirQ  r   r   r   Úfloat_to_integer÷  s    
rT  c                 C   s@   |   ||||j¡}|  |jd¡}|  ||¡}||_||_| ¡ S r  )r   r  r  r  r  r  r  )r    r!   rH  rI  r}   r  r  r5  r   r   r   Únon_complex_to_complexÿ  s    rU  c           	      C   sX   |j }|j }| j|||d}|  ||¡}|  ||j||¡|_|  ||j||¡|_| ¡ S r  )r  r  r   r  r  r  )	r    r!   rH  rI  r}   ZsrctyZdsttyÚsrcÚdstr   r   r   Úcomplex_to_complex
  s    rX  c                 C   s   |   |||¡S rs   )Úis_truerN  r   r   r   Úany_to_boolean  s    rZ  c                 C   s$   |  |t d¡¡}|  ||tj|¡S )Né    )rM  r   ZIntTyper   r   Zint32)r    r!   rH  rI  r}   Zasintr   r   r   Úboolean_to_any  s    r\  c                 C   s"   |   ||j|j¡}|  ||j|¡S rs   )rD  rE  rF  rY  rG  r   r   r   Úliteral_int_to_boolean  s    ýr]  c                 C   s4   |j }|  |||j¡}|  |||j¡}t ||f¡S rs   )r  rD  r  r  r   Zliteral_struct)r    r!   r@   Úpyvalr!  r  r  r   r   r   Úconstant_complex,  s    r_  c                 C   s&   t |tjƒrt|ƒ}|  |¡}||ƒS rs   )rK   ÚnpZbool_ÚboolrL  )r    r!   r@   r^  r~   r   r   r   Úconstant_integer3  s    
rb  c                 C   sH   t | tjtjfƒrDt |tjjƒrD| j|jjkr8t 	d¡‚dd„ }|S dS )z) Typing for the np scalar 'view' method. zOChanging the dtype of a 0d array is only supported if the itemsize is unchangedc                 S   s
   t | |ƒS rs   r   )ÚscalarÚviewtyr   r   r   r  M  s    zscalar_view.<locals>.implN)
rK   r   rÉ   r_   ZabstractZ	DTypeSpecr`   rL   r   ZTypingError)rc  rd  r  r   r   r   Úscalar_viewD  s    ÿÿre  Úview)N)N)N)¤rg   rx   Znumpyr`  r¹   Zllvmliter   Zllvmlite.irr   Znumba.core.imputilsr   r   r   r   r   r	   r
   Z
numba.corer   r   r   r   r   r   Znumba.core.extendingr   r   Znumba.cpython.unsafe.numbersr   r   r*   r,   r.   rE   rF   rU   Údivmodr_   rX   rï   Ú	ifloordivrZ   ÚtruedivÚitruedivr]   rå   Úimodr^   ra   rr   rh   rÊ   rÄ   rÉ   r   rƒ   r†   r‰   rŒ   r   r   r‘   r’   r”   r–   r™   r›   rž   rŸ   r£   r¥   r¦   r©   r«   r°   r¶   r·   r¸   r½   Úbooleanr¾   rÅ   rÆ   rÇ   rÈ   r—   rÃ   rÌ   rÑ   rÓ   rÕ   rÖ   r×   ræ   rÝ   ró   rõ   rö   rø   rú   rû   rü   rý   rþ   rÿ   r  r  r	  r
  r@   r   rº   r+   r»   r-   r¼   r{   ÚComplexr  r  r  r  r  r  Úclsr%  r+  r,  r  ri   r1  r4  r6  r8  r9  r:  r=  rÍ   ÚNumberr>  r?  ra  r@  rA  rB  rC  r;   rJ  rO  ZvoidptrrP  rR  rS  rT  rU  rX  ÚAnyrZ  r\  ZBooleanLiteralr]  r_  rb  re  r   r   r   r   Ú<module>   s   $ 			9

	+=							*0l







) 










		




	




