Edit File: _internal.cpython-37.pyc
B ��FdS � @ s( d Z ddlmZmZmZ ddlZddlZddlmZ ddl m Z m Z m Z y ddl Z W n ek rl dZ Y nX ddlmZ ejdkr�d Znd Zd d � Zd d� Zdd� Zdd� Ze�d�Ze�d�Ze�d�ZdeiZdd� Z G dd� de �Z dd � Z de _ G d d � d e �Z!G d d � d e �Z"d!d"� Z#d#d$� Z$d%d&� Z%d'd(� Z&d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@�Z'dA�(e'�)� �Z*d)d*d+d,dBdCdDdEdDdEdFdGdHd6d7d9d:d<d=d>d?dI�Z+dA�(e+�)� �Z,dJdK� Z-dLdM� Z.dNdO� Z/dPdQ� Z0dRdS� Z1dTdU� Z2dVdW� Z3G dXdY� dYe4�Z5G dZd[� d[e6e7�Z8d\d]� Z9d^d_� Z:dS )`z_ A place for code to be called from core C-code. Some things are more easily handled Python. � )�division�absolute_import�print_functionN)� basestring� )�dtype�array�ndarray)�object_�little� <� >c C s g }t | �� �}x�|D ]�}| | }t|�}t|t�r<|dkrDtd��|dkrZ|d |krZqt|d �}|dk rvtd��t|d |d�}|dkr�|d } nd } |�|||| f� qW |j dd � d � d d � |D �} d d � |D �} dd � |D �} dd � |D �} | | | | fS )N)� � zentry not a 2- or 3- tupler r r zinvalid offset.)�alignc S s | d S )Nr � )�xr r �G/opt/alt/python37/lib64/python3.7/site-packages/numpy/core/_internal.py�<lambda>- � z!_makenames_list.<locals>.<lambda>)�keyc S s g | ] }|d �qS )r r )�.0r r r r � <listcomp>. s z#_makenames_list.<locals>.<listcomp>c S s g | ] }|d �qS )r r )r r r r r r / s c S s g | ] }|d �qS )r r )r r r r r r 0 s c S s g | ] }|d �qS )r r )r r r r r r 1 s ) �list�keys�len� isinstance�tuple� ValueError�intr �append�sort)�adictr Z allfields�fnames�fname�obj�n�num�format�title�names�formats�offsets�titlesr r r �_makenames_list s. r. c C s� y | d }W n t k r$ d }Y nX |d krBt| |�\}}}}ndg }g }g }xV|D ]N}| | }|�|d � |�|d � t|�dkr�|�|d � qT|�d � qTW t||||d�|�S )N���r r r )r* r+ r, r- )�KeyErrorr. r r r )r" r r* r+ r, r- �name�resr r r � _usefields8 s( r3 c sj | j � � d kr`| j}|d krL| jd kr,| jS | j�� }|rD| j|fS | jS nt|d �|d fS | j}� fdd�|D �}g }d}x�|D ]�}|d |kr�|d | }|�dd| f� ||7 }t|�dkr�|d |d f} n|d } |d j�r| t|d jd �|d jd f} n| t|d �f} ||d j 7 }|�| � q�W | j |k�rf| j | }|�dd| f� |S ) Nr r c s g | ]}� | |f �qS r r )r r )�fieldsr r r i s z _array_descr.<locals>.<listcomp>� z|V%dr r ) r4 �subdtype�metadata�str�copy� _array_descrr* r r �itemsize) Z descriptorr6 �newr* Zordered_fields�result�offset�fieldr' r1 �tupr )r4 r r: X sB r: c C s t �| ||�S )N)r �__new__)�subtype�shaper r r r � _reconstruct� s rD sx (?P<order1>[<>|=]?)(?P<repeats> *[(]?[ ,0-9L]*[)]? *)(?P<order2>[<>|=]?)(?P<dtype>[A-Za-z0-9.?]*(?:\[[a-zA-Z0-9,.]+\])?)s \s*,\s*s \s+$� =c C sh d}g }�xX|t | �k �rbtj| |d�}y|�� \}}}}W n0 ttfk rl tdt |�d | f ��Y nX |�� }|t | �k r�tj| |d�r�t | �}n2t j| |d�}|s�tdt |�d | f ��|�� }|dkr�|}nD|dkr�|}n6t � ||�}t � ||�}||k�rtd||f ��|}|ddt gk�r2d}|| }|dk�rJ|} n |t |�f} |�| � q W |S ) Nr )�posz*format number %d of "%s" is not recognizedr r z/inconsistent byte-order specification %s and %s� |rE )r � format_re�match�groups� TypeError�AttributeErrorr �end�space_re�sep_re� _convorder�get�_nbo�evalr ) �astrZ startindexr= �moZorder1ZrepeatsZorder2r �order�newitemr r r � _commastring� sL rX c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � dummy_ctypec C s || _ d S )N)�_cls)�self�clsr r r �__init__� s zdummy_ctype.__init__c C s | S )Nr )r[ �otherr r r �__mul__� s zdummy_ctype.__mul__c G s | � |�S )N)rZ )r[ r^ r r r �__call__� s zdummy_ctype.__call__c C s | j |j kS )N)rZ )r[ r^ r r r �__eq__� s zdummy_ctype.__eq__N)�__name__� __module__� __qualname__r] r_ r` ra r r r r rY � s rY c C sx t j} | d k r| S td kr.dd l}t|j�} n@td�j}|dkrHtj} n&|dkrXtj } n|dkrhtj } ntj } | t _| S )Nr �p�i�l�q) �_getintp_ctype�cache�ctypes�numpyrY Zintpr �char�c_int�c_long� c_longlong)�val�nprm r r r ri � s ri c @ s e Zd Zdd� Zdd� ZdS )�_missing_ctypesc C s |S )Nr )r[ r' r% r r r �cast� s z_missing_ctypes.castc C s |S )Nr )r[ r' r r r �c_void_p� s z_missing_ctypes.c_void_pN)rb rc rd rt ru r r r r rs � s rs c @ s� e Zd Zddd�Zdd� Zdd� Zdd � Zd d � Zd d � Zdd� Z dd� Z e eddd�Z e eddd�Z e e ddd�Ze e ddd�ZdS )�_ctypesNc C s> t r t | _nt� | _|| _|| _| jjdkr4d| _nd| _d S )Nr TF)rk rv rs �_arr�_data�ndim�_zerod)r[ r �ptrr r r r] � s z_ctypes.__init__c C s | j �| j|�S )N)rv rt rx )r[ r% r r r �data_as� s z_ctypes.data_asc C s | j r d S || jj | jj� S )N)rz rw ry rC )r[ r% r r r �shape_as s z_ctypes.shape_asc C s | j r d S || jj | jj� S )N)rz rw ry �strides)r[ r% r r r � strides_as s z_ctypes.strides_asc C s | j S )N)rx )r[ r r r �get_data s z_ctypes.get_datac C s | � t� �S )N)r} ri )r[ r r r � get_shape s z_ctypes.get_shapec C s | � t� �S )N)r ri )r[ r r r � get_strides s z_ctypes.get_stridesc C s | j �| j�S )N)rv ru rx )r[ r r r �get_as_parameter s z_ctypes.get_as_parameterz c-types data)�docz c-types shapezc-types stridesz_as parameter_)N)rb rc rd r] r| r} r r� r� r� r� �property�datarC r~ Z_as_parameter_r r r r rv � s rv c C s� | j }t|�}t|t�r |g}t|ttf�r|x>|D ]6}y|�|� W q2 tk rf td|f ��Y q2X q2W tt|�| �S td|f ��d S )Nzunknown field name: %szunsupported order value: %s)r* r r r8 r �remover )ZdatatyperV ZoldnamesZ nameslistr1 r r r � _newnames s r� c s0 | j � � j� fdd�� jD �d�}t| |dd�S )a Return copy of structured array with padding between fields removed. Parameters ---------- ary : ndarray Structured array from which to remove padding bytes Returns ------- ary_copy : ndarray Copy of ary with padding bytes removed c s g | ]}� j | d �qS )r )r4 )r r1 )�dtr r r < s z _copy_fields.<locals>.<listcomp>)r* r+ T)r r9 )r r* r )ZaryZ copy_dtyper )r� r � _copy_fields- s r� c C sj |j s | j rf|dkr || kr dS | jr^x6| jD ],}| j| d |kr.| j| d |kr.dS q.W td��dS )a Checks safety of getfield for object arrays. As in _view_is_safe, we need to check that memory containing objects is not reinterpreted as a non-object datatype and vice versa. Parameters ---------- oldtype : data-type Data type of the original ndarray. newtype : data-type Data type of the field being accessed by ndarray.getfield offset : int Offset of the field being accessed by ndarray.getfield Raises ------ TypeError If the field access is invalid r Nr z'Cannot get/set field of an object array)� hasobjectr* r4 rK )�oldtype�newtyper> r1 r r r �_getfield_is_safe? s r� c C s$ | |kr dS |j s| j r td��dS )az Checks safety of a view involving object arrays, for example when doing:: np.zeros(10, dtype=oldtype).view(newtype) Parameters ---------- oldtype : data-type Data type of original ndarray newtype : data-type Data type of the view Raises ------ TypeError If the new type is incompatible with the old type. Nz)Cannot change data-type for object array.)r� rK )r� r� r r r � _view_is_safe_ s r� �?ZS1�b�B�h�Hrf �Irg �Lrh �Q�e�f�d�g�F�D�G�S�U�O�V)r� �cr� r� r� r� rf r� rg r� rh r� r� r� r� r� �Zf�ZdZZg�s�wr� r r5 �i2�u2Zi4Zu4Zi8�u8�f2)r� r� r� r� r� r� rf r� rg r� rh r� r� r� r� r� r� r� r� r� r c C s, G dd� dt �}|| �}t|dd�\}}|S )Nc @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zed d � �Zd d � Z e Z d S )z#_dtype_from_pep3118.<locals>.Streamc S s || _ d| _d S )N�@)r� � byteorder)r[ r� r r r r] � s z,_dtype_from_pep3118.<locals>.Stream.__init__c S s" | j d |� }| j |d � | _ |S )N)r� )r[ r& r2 r r r �advance� s z+_dtype_from_pep3118.<locals>.Stream.advancec S s, | j d t|�� |kr(| �t|�� dS dS )NTF)r� r r� )r[ r� r r r �consume� s z+_dtype_from_pep3118.<locals>.Stream.consumec S sl t |�r@d}x(|t| j�k r4|| j| �s4|d }qW | �|�S | j�|�}| �|�}| �t|�� |S d S )Nr r )�callabler r� r� �index)r[ r� rf r2 r r r � consume_until� s z1_dtype_from_pep3118.<locals>.Stream.consume_untilc S s | j d S )Nr )r� )r[ r r r �next� s z(_dtype_from_pep3118.<locals>.Stream.nextc S s t | j�S )N)�boolr� )r[ r r r �__bool__� s z,_dtype_from_pep3118.<locals>.Stream.__bool__N) rb rc rd r] r� r� r� r� r� r� � __nonzero__r r r r �Stream� s r� F)� is_subdtype)�object�__dtype_from_pep3118)�specr� �streamr r r r r �_dtype_from_pep3118� s $r� c C s: t g g g dd�}d}d}d}�x�| �r�d }| �d�r6P d }| �d�rb| �d�}ttt|�d���}| jd kr�| �d�}|d kr�d }|| _ | j d kr�t } t } nt } t } | �d d� �} | r�t| �} nd} d}| �d�r�t| dd�\}} n�| j| k�rf| jdk�r| �d�}n | �d�}|dk}| | }|dk�r<|d| 7 }d} ddd ��| j | j �}t|| �}|j} ntd| j ��d}| j dk�r�| | }|j | }||7 }|dk�r�| dk�s�|d k �r�t|�dk�r�t||�}n||7 }t| |�}| dk�rt|| ff�}|d k �rt||f�}| �d��r2| �d�}nd }|�rF|d k�s�|d k �rj||d k�rjtd | ��|d �|� |d �|� |d �|� ||j7 }||7 }||d < q W | j dk�r�|d | | 7 < |d d gk�r"|d d dk�r"|d |d d jk�r"|�s"|d d }nt|� t|�}||fS ) Nr )r* r+ r, r; r F�}�(�)�,)r� �=�<�>�^�!r� r� )r� r� c S s | � � S )N)�isdigit)r� r r r r r z&__dtype_from_pep3118.<locals>.<lambda>zT{T)r� �Zr r ZUSVz%dr� z'Unknown PEP 3118 data type specifier %rr� �:r* z+Duplicate field name '%s' in PEP3118 formatr+ r, r; )�dictr� r� r �mapr �splitr� r� r� �_pep3118_native_map�_pep3118_native_typechars�_pep3118_standard_map�_pep3118_standard_typecharsr� rQ r � alignmentr r� r; �_prod�_add_trailing_padding�_lcm� RuntimeErrorr � _fix_names)r� r� � field_specr> Zcommon_alignmentZ is_padding�valuerC r� Ztype_mapZtype_map_charsZ itemsize_strr; r ZtypecharZ dtypecharZnumpy_byteorderZ extra_offsetZ start_paddingZ intra_paddingr1 �retr r r r� � s� " r� c C sZ | d }xLt |�D ]@\}}|dk r$qd}x d�|�}||kr>P |d }q*W |||< qW dS )z< Replace names which are None with the next unused f%d name r* Nr zf{}r )� enumerater( )r� r* rf r1 �jr r r r� j s r� c sv | j dkr$tdg| gdg| jd�}n:| j � | j}t|� fdd�|D �� fdd�|D �| jd�}|d |7 < t|�S ) zBInject the specified number of padding bytes at the end of a dtypeNZf0r )r* r+ r, r; c s g | ]}� | d �qS )r r )r r1 )r4 r r r � s z)_add_trailing_padding.<locals>.<listcomp>c s g | ]}� | d �qS )r r )r r1 )r4 r r r � s r; )r4 r� r; r* r )r� �paddingr� r* r )r4 r r� y s r� c C s d}x| D ] }||9 }q W |S )Nr r )�are r r r r r� � s r� c C s x|r|| | } }qW | S )z0Calculate the greatest common divisor of a and br )r� r� r r r �_gcd� s r� c C s | t | |� | S )N)r� )r� r� r r r r� � s r� c @ s e Zd ZdS )� TooHardErrorN)rb rc rd r r r r r� � s r� c s" e Zd ZdZd� fdd� Z� ZS )� AxisErrorz Axis supplied was invalid. Nc sJ |d kr|d kr|}n d� ||�}|d k r6d� ||�}tt| ��|� d S )Nz2axis {} is out of bounds for array of dimension {}z{}: {})r( �superr� r] )r[ Zaxisry Z msg_prefix�msg)� __class__r r r] � s zAxisError.__init__)NN)rb rc rd �__doc__r] � __classcell__r r )r� r r� � s r� c O sZ d� dd� |D �dd� |�� D � �}||�dd� }d� dd� |D ��}d �||||�S ) z= Format the error message for when __array_ufunc__ gives up. z, c S s g | ]}d � |��qS )z{!r})r( )r �argr r r r � s z0array_ufunc_errmsg_formatter.<locals>.<listcomp>c S s g | ]\}}d � ||��qS )z{}={!r})r( )r �k�vr r r r � s �outr c s s | ]}t t|�j�V qd S )N)�repr�typerb )r r� r r r � <genexpr>� s z/array_ufunc_errmsg_formatter.<locals>.<genexpr>zToperand type(s) all returned NotImplemented from __array_ufunc__({!r}, {!r}, {}): {})�join�itemsrQ r( )�dummy�ufunc�method�inputs�kwargsZ args_string�argsZ types_stringr r r � array_ufunc_errmsg_formatter� s r� c C s� | j dkrd}nd�dd� t| j �D ��}| jdkr:d}n>| jdkrJd}n.d jd�d d� t| j�D ��td | j �d �}d }| jdkr�d| }dj| j|||d�S )zz Builds a signature string which resembles PEP 457 This is used to construct the first line of the docstring r r z, c s s | ]}d � |d �V qdS )zx{}r N)r( )r rf r r r r� � s z1_ufunc_doc_signature_formatter.<locals>.<genexpr>r z , /, out=()z , /, out=Nonez%[, {positional}], / [, out={default}]c s s | ]}d � |d �V qdS )zout{}r N)r( )r rf r r r r� � s )N)� positional�defaultzM, casting='same_kind', order='K', dtype=None, subok=True[, signature, extobj]Nz , where=Truez&{name}({in_args}{out_args}, *{kwargs}))r1 �in_args�out_argsr� )�ninr� �range�noutr( r� � signaturerb )r� r� r� r� r r r � _ufunc_doc_signature_formatter� s& r� );r� � __future__r r r �re�sys� numpy.compatr � multiarrayr r r rk � ImportErrorZ numerictypesr r� rR r. r3 r: rD �compilerH rO rN rP rX r� rY ri rj rs rv r� r� r� r� r� r� r r� r� r� r� r� r� r� r� r� r� r� r� r � IndexErrorr� r� r� r r r r �<module> s� 0 / / !+