Aluode/PerceptionLabPortable
0
1o
2 �Yi�F � @ s� d Z ddlZddlZddlmZmZ ddlmZmZ ddl m3Z4 ddl m Z m
Z
ddlmZ ed�ZejZd d5� Zd d � Zd
d� Zeejej�dd� �Zeejej�dd� �Zeejej�dd� �Zeej �dd� �Z dd� Z e d�Z e d�Z!eej"ej�e dd � ��Z#eej$ej�e d d � ��Z%eej$ejej�d d � �Z&eej'�d!d"� �Z(eej)�d#d$� �Z*eej+�d%d&� �Z,eej-ej�d'd(� �Z.eej/ej�d)d*� �Z0eej1�d+d,� �Z2eej3ej�d-d.� �Z4eej5�d/d0� �Z6eej7ej�d1d2� �Z8eej9�d3d4� �Z:eej;ej�d5d6� �Z<eej=�d7d8� �Z>eej?ej�d9d:� �Z@eejAej�d;d<� �ZBeejCej�d=d>� �ZDeejEej�d?d@� �ZFdS )Az'6Implement the cmath module functions.7� N)�Registry�impl_ret_untracked)�types�cgutils)� signature)�builtins�mathimpl)�overloadZ cmathimplc C s | � d|j|j�S )NZuno)Zfcmp_unordered�real�imag��builder�z� r ��E:\DocsHouse\542 percep lab latest\PerceptionLab\PerceptionLab_Portable\python_embed\Lib\site-packages\numba/cpython/cmathimpl.py�is_nan s r c C � | � t�| |j�t�| |j��S �N)�or_r �is_infr8 r r r r r r � �r c C r r )�and_r � is_finiter9 r r r r r r r r c C �8 |j \}|\}| j|||d�}t||�}t| ||j|�S �N��value)�args� make_complexr r � return_type��contextr
�sigr �typr r �resr r r �isnan_float_impl �10 11r% c C r r )r r r r r r r r r �isinf_float_impl' r& r' c C r r )r r r r r r r r r �isfinite_float_impl0 r& r( c C s& t dd� | |fD ��rdd� }|S d S )Nc S s g | ]}t |tj��qS r )�12isinstancer �Float)�.0r# r r r �13<listcomp>; s z#impl_cmath_rect.<locals>.<listcomp>c S s� t �|�s| s t| �S t �| �rt| |�S t �|�}t �|�}|dkr-t �| �r-|| }n|| 9 }|dkr?t �| �r?|| }n|| 9 }t||�S )N� )�math�isfinite�abs�isinf�complex�cos�sin)�r�phir14 r r r r �impl< s 1516171819202122z impl_cmath_rect.<locals>.impl)�all)r5 r6 r7 r r r �impl_cmath_rect9 s �r9 c s � fdd�}|S )Nc
s� |j \}|\}| j|||d�}|j}|j}t�||�} t�||�}23t|jg|jfd t j24fd �R � } | � |� | ||| |25f�} t | ||| �S )Nr � )
r r r26 r r r r r �underlying_floatr �boolean�compile_internalr )
r! r
r" r r# r r �x�y� x_is_finite� y_is_finiteZ inner_sigr$ ��27inner_funcr r �wrapperU s �2829�z(intrinsic_complex_unary.<locals>.wrapperr )rC rD r rB r �intrinsic_complex_unaryT s rE �nan�infc C s� |r!|r t �|�}t �|�}t �| �}t|| || �S ttt�S t �| �r2|r-t| | �S t| |�S | dkr\|rWt �|�}t �|�}|dkrJ|| 9 }|dkrR|| 9 }t||�S t| t�S |rvt �| �}t �|�}t �|�}t|| || �S d}t||�S )zcmath.exp(x + y j)r- r )r. r3 r4 �expr2 �NAN�isnan) r> r? r@ rA �c�sr5 r30 r r r r �exp_implj s8 313233343536373839404142434445rM c C s( t �t �| |��}t �|| �}t||�S )zcmath.log(x + y j))r. �log�hypot�atan2r2 )r> r? r@ rA �a�br r r �log_impl� s 46rS c C s. |\}}dd� }| � ||||�}t| |||�S )zcmath.log(z, base)c S s t �| �t �|� S r )�cmathrN )r �baser r r �log_base� s z log_base_impl.<locals>.log_base�r= r )r! r
r" r r rU rV r$ r r r �
log_base_impl� s rX c s$ t | tj�sd S d� � fdd�}|S )NgU���k@c s t �| �} t| j� | j� �S )zcmath.log10(z))rT rN r2 r47 r �r �ZLN_10r r �48log10_impl� s 49z$impl_cmath_log10.<locals>.log10_impl�r) r �Complex)r r[ r rZ r �impl_cmath_log10� s50 r^ c C s t | tj�sdS dd� }|S )zcmath.phase(x + y j)Nc S s t �| j| j�S r )r. rP r r51 )r> r r r r7 � s zphase_impl.<locals>.implr\ �r> r7 r r r �52phase_impl� s r` c C � t | tj�sd S dd� }|S )Nc S s& | j | j}}t�||�t�||�fS r )r53 r r. rO rP )r> r5 �ir r r r7 � s zpolar_impl.<locals>.implr\ r_ r r r �54polar_impl� s rc c 55 s` d}d| }|j d j}|jdkrtjntj}|| � � fdd�}| �||||�} t| ||| �S )Ng�;f���?� �?r �@ c sR | j }| j}|dkr|dkrtt|�|�S t�|�r!tt|�|�S t�|�r+t||�S t�|�rL|dk rAtt|| �t�||��S t|t�|| |��S t|�� ksXt|�� krc|d9 }|d9 }d}nd}|dkrt�|t� ||� d �}|}|d| }n t�| t� ||� d �}t|�d| }t�||�}|r�t|d |�S t||�S )z
cmath.sqrt(z)r- � �?TFr � �?r: )56r57 r r2 r0 r. r1 rJ �copysign�sqrtrO )r rQ rR �scale�tr58 r ��THRESr r � sqrt_impl� s6 59606162 63z sqrt_impl.<locals>.sqrt_impl)r r; Zbitwidthr ZDBL_MAX�FLT_MAXr= r )64r! r
r" r ZSQRT2ZONE_PLUS_SQRT2Z theargflt�MAXrn r$ r rl r rn � s *rn c C �& dd� }| � ||||�}t| |||�S )Nc S s t �t| j | j��S )z cmath.cos(z) = cmath.cosh(z j))rT �coshr2 r r65 rY r r r �cos_impl s zcos_impl.<locals>.cos_implrW )r! r
r" r rs r$ r r r rs s rs c C ra )Nc S s� | j }| j}t�|�r@t�|�rt|�}|}n |dkr"t|�}|}nt�|t�|��}t�|t�|��}|dk r;| }t ||�S t t�|�t�66|� t�|�t� |� �S )z
cmath.cosh(z)r- ) r67 r r. r1 rJ r0 rh r3 r4 r2 rr �sinh�r r> r? r68 r r r r � cosh_impl s" 697071�z"impl_cmath_cosh.<locals>.cosh_implr\ )r rv r r r �impl_cmath_cosh � rw c C rq )Nc S �& t �t| j | j��}t|j|j �S )z#cmath.sin(z) = -j * cmath.sinh(z j))rT rt r2 r r72 �r r5 r r r �sin_impl7 � zsin_impl.<locals>.sin_implrW )r! r
r" r r{ r$ r r r r{ 5 � r{ c C ra )Nc S s� | j }| j}t�|�r6t�|�r|}|}n t�|�}t�|�}|dkr'||9 }|dkr1|t|�9 }t||�S tt�|�t� |� t�|�t�73|� �S )z
cmath.sinh(z)r- ) r74 r r. r1 rJ r3 r4 r0 r2 rt rr ru r r r � sinh_implD s 75767778 79�z"impl_cmath_sinh.<locals>.sinh_implr\ )r r~ r r r �impl_cmath_sinh? s r c C rq )Nc S ry )z#cmath.tan(z) = -j * cmath.tanh(z j))rT �tanhr2 r r80 rz r r r �tan_impl\ r| ztan_impl.<locals>.tan_implrW )r! r
r" r r� r$ r r r r� Z r} r� c C ra )Nc 81 S s� | j }| j}t�|�r)t�d|�}t�|�rd}n t�dt�d| ��}t||�S t�|�}t�|�}dt� |� }|| }d|| } t|d|| | || | | �S )z
cmath.tanh(z)rd r- � @)82r83 r r. r1 rh r4 r2 r� �tanrr )84r r> r? r85 r Ztx�ty�cxZtxty�denomr r r � tanh_implj s" 86 87888990 �z"impl_cmath_tanh.<locals>.tanh_implr\ )r r� r r r �impl_cmath_tanhe rx r� c �@ t �d�� tjd �� �fdd�}| �||||�}t| |||�S )N� c s� t | j��kst | j��kr4t�t | j�| j�}t�t�t�| jd | jd ��� | j �}t||�S t �91td| j | j ��}t �92td| j | j��}dt�|j|j� }t� |j|j |j|j �}t||�S )z
cmath.acos(z)rg rd r� ) r0 r93 r r. rP rh rN rO r2 rT ri �asinh�r r94 r �s1�s2��LN_4rm r r � acos_impl� s �95 96z acos_impl.<locals>.acos_impl�r. rN r ro r= r )r! r
r" r r� r$ r r� r r� � s97 9899r� c s6 t | tj�sd S t�d�� tjd �� �fdd�}|S )Nr� c s� t | j��kst | j��kr,t�t�| jd | jd ��� }t�| j| j�}t||�S t� t| jd | j��}t� t| jd | j��}t�100|j|j |j|j �}dt�|j|j� }t||�S )zcmath.acosh(z)rg rd r� ) r0 r101 r r. rN rO rP r2 rT ri r� r� r� r r �102acosh_impl� s "103 104z$impl_cmath_acosh.<locals>.acosh_impl)r) r r] r. rN r ro )r r� r r� r �impl_cmath_acosh� s 105106r� c r� )Nr� c s� t | j��kst | j��kr3t�t�t�| jd | jd ��� | j�}t�| jt | j��}t||�S t �107td| j | j ��}t �108td| j | j��}t� |j|j |j|j �}t�| j|j|j |j|j �}t||�S )zcmath.asinh(z)rg rd ) r0 r109 r r. rh rN rO rP r2 rT ri r� r� r� r r �110asinh_impl� s �111 "112z asinh_impl.<locals>.asinh_implr� )r! r
r" r r� r$ r r� r r� � s113 114115r� c C rq )Nc S ry )z%cmath.asin(z) = -j * cmath.asinh(z j))rT r� r2 r r116 rz r r r � asin_impl� r| z asin_impl.<locals>.asin_implrW )r! r
r" r r� r$ r r r r� � r} r� c C rq )Nc S sL t �t| j | j��}t�| j�r t�| j�r t|j|j�S t|j|j �S )z%cmath.atan(z) = -j * cmath.atanh(z j))rT �atanhr2 r r117 r. r1 rJ rz r r r � atan_impl� s z atan_impl.<locals>.atan_implrW )r! r
r" r r� r$ r r r r� � s r� c s^ t �d�}t �tjd ��t �tj��t jd � � ��fdd�}| �||||�}t| |||�S )Nr� r: c s� | j dk r d}| } nd}t| j�}t�| j �s!| j �ks!|�krWt�| j�r/t�d| j �}n t�| j �r8d}nt�| j d | jd �}| j d | | }t�� | j � }n`| j dkr�|�k r�|dkrjt}| j}nMt� t�118|�t�119t�|d�� � }t�t� d| �d | j�}n,|| }d | j }t� d| j || | �d120 }t� d | j |d | j | � d }t�| j�r�t
}|r�t| | �S t||�S ) zcmath.atanh(z)r- TFrg g @rd r� r: � rf g �)r121 r0 r r. rJ r1 rh rO �INFrN ri rP �log1prI r2 )r �negateZayr122 �hr ZsqayZzr1�ZPI_12Z THRES_LARGEZ THRES_SMALLr r �123atanh_impl� sD 124125 � 126 �� 127z atanh_impl.<locals>.atanh_impl) r. rN ri r ro ZFLT_MIN�pir= r )r! r
r" r r� r� r$ r r� r r� � s 128 129,r� )G�__doc__rT r. Znumba.core.imputilsr r Z130numba.corer r Znumba.core.typingr Z
numba.cpythonr r Znumba.core.extendingr �registry�lowerr r r rJ r] r% r1 r' r/ r( �rectr9 rE �floatrI r� rH rM rN rS rX �log10r^ Zphaser` Zpolarrc ri rn r3 rs rr rw r4 r{ rt r r� r� r� r� �acosr� �acoshr� r� r� �asinr� �atanr� r� r� r r r r �<module> s� 131 132 133134 ( 135 136137 138 139 140 <141142 143 144 145146 147 148149 150 151 152