U
    ev                  
   @   s  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	 ddlm
Z
 ddlmZmZ ddlmZ dd	lmZ dd
lmZ ddlmZ dZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%d Z&e!e"e#e$e%e&d!Z'd"d#d$d%d&d'd!Z(d(Z)d)Z*d*Z+d+Z,d,Z-d-d. Z.dd1d2Z/dd3d4Z0d5d6 Z1d7d8 Z2d9d: Z3dd;d<Z4G d=d> d>ej5Z6d?d@ Z7dAdB Z8dCdD Z9dEdF Z:dGdH Z;ddIdJZ<G dKdL dLe=Z>G dMdN dNe>Z?G dOdP dPe>Z@dQdR ZAdSdT ZBdUdV ZCdWdX ZDdYdZ ZEd[d\ ZFd]d^d_d`dadbdcZGddde ZHdfdg ZIdhdi ZJdjdk ZKddmdnZLddodpZMdqdr ZNe,e
jOe-ejPjQdsdtZReMdueRg dvZSeMdweRZTeMdxeUeRe
jOdyeSeTgdvZVeMdzZWeLd{ejXgd|ZYeMd}eRZZeMd~eRZ[eMdeRg d|Z\eLdeRe] ej^ej_eVeZe[e\gdvZ`dZaeSjbce` e\jbce` dlS )    )absolute_import   )CompileErrorerror)	ExprNodes)IntNodeNameNodeAttributeNode)Options)UtilityCodeTempitaUtilityCode)CythonUtilityCode)Buffer)
PyrexTypes)
ModuleNodezStart must not be given.z3Axis specification only allowed in the 'step' slot.z.Step must be omitted, 1, or a valid specifier.z>Cannot specify an array that is both C and Fortran contiguous.zInvalid axis specification.z+Variable was not cimported from cython.viewz=no expressions allowed in axis spec, only names and literals.z<Invalid axis specification for a C/Fortran contiguous array.zdCannot check if memoryview %s is initialized without the GIL, consider using initializedcheck(False)ZPyBUF_FORMATz#(PyBUF_C_CONTIGUOUS | PyBUF_FORMAT)z#(PyBUF_F_CONTIGUOUS | PyBUF_FORMAT)z%(PyBUF_ANY_CONTIGUOUS | PyBUF_FORMAT)ZPyBUF_FULL_ROZPyBUF_RECORDS_ROZ__Pyx_MEMVIEW_DIRECTZ__Pyx_MEMVIEW_PTRZ__Pyx_MEMVIEW_FULLZ__Pyx_MEMVIEW_CONTIGZ__Pyx_MEMVIEW_STRIDEDZ__Pyx_MEMVIEW_FOLLOW)directptrfullcontigstridedfollowdpfcs_z{ 0, 0, { 0 }, { 0 }, { 0 } }
memoryviewZ__pyx_memoryview_typeZ__pyx_memoryview_objZ__Pyx_memviewslicec                 C   s    | d|   | d|   d S )Nz%s.data = NULL;z%s.memview = NULL;)putln)Zmv_cnamecode r    >/tmp/pip-unpacked-wheel-7k3cqui1/Cython/Compiler/MemoryView.pyput_init_entryC   s    r"   FTc           	   	   C   s|   |j jst| p| }|r*| }n(|jj|dd}|d||	|f  t
| ||||||d |sx|j| dS )zEWe can avoid decreffing the lhs if we know it is the first assignmentFZ
manage_ref%s = %s;)have_gilfirst_assignmentN)typeZis_memoryviewsliceAssertionErrorresult_in_tempZ	is_simpleresult	funcstateallocate_tempr   Z	result_asput_assign_to_memviewslicerelease_temp)		lhs_cnameZlhs_typeZlhs_posrhsr   r%   r&   Z
pretty_rhsZrhstmpr    r    r!   put_acquire_memoryviewsliceL   s    
 r1   c                 C   sR   | |kr| d d S |s*|j| ||d | s<|| | d| |f  d S )Nz&/* memoryview self assignment no-op */r%   r$   )r   Zput_xdecrefr)   Zmake_owned_memoryviewslice)r/   r0   Z	rhs_cnameZmemviewslicetyper   r%   r&   r    r    r!   r-   a   s    

r-   c                 C   sD   t | \}}|rtS |rtS t|  \}}d|ks8d|kr<tS tS d S )Nr   r   )is_cf_contigmemview_c_contiguousmemview_f_contiguouszipmemview_full_accessmemview_strided_accessspecsis_c_contigis_f_contigaccesspackingr    r    r!   get_buf_flagss   s    r?   c                 C   s$   dg| }| | j t| j|S )Nr   r   )extendaxesr   MemoryViewSliceTypedtype)ZmemoryviewtypenrB   r    r    r!   insert_newaxes   s    
rF   c                 C   s<   t | j|j }| j|jk r*t| ||fS | t||fS d S N)absndimrF   )srcdstrE   r    r    r!   broadcast_types   s    rL   c                 C   s   | j r| jjrdS | tjkr dS | jrR| jdkrR| jjD ]}t	|j
s8 dS q8dS | jp| jrv|dk rvt	| j|d p| jp| jp| jp| jot	| jS )z
    Return whether type dtype can be used as the base type of a
    memoryview slice.

    We support structs, numeric types and objects
    FstructT   r   )Z
is_complexZ	real_typeZis_intr   Zc_bint_typeZ	is_structkindscopeZvar_entriesvalid_memslice_dtyper'   Zis_errorZis_arrayZ	base_typeZ
is_numericis_pyobjectZis_fusedZ
is_typedefZtypedef_base_type)rD   imemberr    r    r!   rQ      s*    

rQ   c                   @   sJ   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d ZdddZ	dd Z
dS )MemoryViewSliceBufferEntryz
    May be used during code generation time to be queried for
    shape/strides/suboffsets attributes, or to perform indexing or slicing.
    c                 C   sD   || _ |j| _|j| _d| j | _| j jj}t|| _|   d S )Nz%s.data)	entryr'   cnamebuf_ptrrD   r   ZCPtrTypeZbuf_ptr_typeZinit_attributes)selfrV   rD   r    r    r!   __init__   s    
z#MemoryViewSliceBufferEntry.__init__c                 C   s
   |  dS )N%s.suboffsets[%d]Z_for_all_ndimrY   r    r    r!   get_buf_suboffsetvars   s    z0MemoryViewSliceBufferEntry.get_buf_suboffsetvarsc                 C   s
   |  dS )N%s.strides[%d]r\   r]   r    r    r!   get_buf_stridevars   s    z-MemoryViewSliceBufferEntry.get_buf_stridevarsc                 C   s
   |  dS )N%s.shape[%d]r\   r]   r    r    r!   get_buf_shapevars   s    z,MemoryViewSliceBufferEntry.get_buf_shapevarsc                    s&    fddt | jjD }| ||S )Nc                    s$   g | ]\}\}}| | ||fqS r    r    ).0dimr=   r>   index_cnamesr    r!   
<listcomp>   s   
zJMemoryViewSliceBufferEntry.generate_buffer_lookup_code.<locals>.<listcomp>)	enumerater'   rB   _generate_buffer_lookup_code)rY   r   rf   rB   r    re   r!   generate_buffer_lookup_code   s    

z6MemoryViewSliceBufferEntry.generate_buffer_lookup_codeTc                 C   s  | j }| jj }|D ]\}}}}	d| j|f }
d| j|f }d| j|f }t||	}|dkr||jt d||||f }nr|dkrd|||f }d||f }nN|d	krd
|||f }n6|dkrd|||f }n|dkst	|d|||f }d||f }q|rd||f S |S )zl
        Generate a single expression that indexes the memory view slice
        in each dimension.
        ra   r_   r[   )genericgeneric_contiguousz-__pyx_memviewslice_index_full(%s, %s, %s, %s)indirectz(%s + %s * %s)z(*((char **) %s) + %s)indirect_contiguousz(*((char **) %s + %s) + %s)r   
contiguousz((char *) (((%s *) %s) + %s))z( /* dim=%d */ %s )z((%s *) %s))
rX   r'   rD   empty_declaration_coderW   get_memoryview_flagZglobalstateZuse_utility_codememviewslice_index_helpersr(   )rY   r   rB   Zcast_resultZbufp	type_declrd   indexr=   r>   shapeZstrideZ	suboffsetflagr    r    r!   ri      s2    

z7MemoryViewSliceBufferEntry._generate_buffer_lookup_codec              	      s
  | j } dt    dt    j|||d tdd | jjD }	g  fdd}
d}d	}|D ]}|jrd
D ]\}} d||||f  qz|d7 }qj|d7 }| jj| \}}|jrZt	t }d
 D ]8}t||}|j  }|d| < |r| nd||< q|d s<|d s<|d s<||d< d}nd} |j|d< |d7 }nv| }|dk}|r|dk}|d	krt|jd  S t	t t|d t|d d}|d r̈ |j|d< d}tj|d|d\}} | qjr jd	  d S )!a  
        Slice a memoryviewslice.

        indices     - list of index nodes. If not a SliceNode, or NoneNode,
                      then it must be coercible to Py_ssize_t

        Simply call __pyx_memoryview_slice_memviewslice with the right
        arguments, unless the dimension is omitted or a bare ':', in which
        case we copy over the shape/strides/suboffsets attributes directly
        for that dimension.
        z%(dst)s.data = %(src)s.data;z"%(dst)s.memview = %(src)s.memview;r2   c                 s   s   | ]\}}|d kV  qdS )r   Nr    )rc   r=   r>   r    r    r!   	<genexpr>  s     zHMemoryViewSliceBufferEntry.generate_buffer_slice_code.<locals>.<genexpr>c                     s6   s. j jtjdd}  d|   |  d S )NFr#   z%s = -1;r   )r+   r,   r   Z
c_int_typer   append)Zsuboffset_dimr   Zsuboffset_dim_tempr    r!   get_suboffset_dim  s
    
zPMemoryViewSliceBufferEntry.generate_buffer_slice_code.<locals>.get_suboffset_dimr   ))ru   r   )stridesr   )
suboffsetsr{   z%s.%s[%d] = %d;r   zstart stop stepZhave_0Z
have_startZ	have_stopZ	have_stepr=   ZSimpleSliceZ
ToughSlice
error_gotor   r   z7All preceding dimensions must be indexed and not sliced
wraparoundboundscheck)r   r   Z
SliceIndexMemoryView_C.c)contextN)rW   r   localsZput_incref_memoryviewsliceallr'   rB   is_noneis_slicedictsplitgetattrr*   r   posr   intr   Zload_as_stringputr+   r.   )rY   r   indicesrK   Zdst_typer%   have_slicesZ
directivesrJ   Zall_dimensions_directrz   rd   Znew_ndimrt   attribvaluer=   r>   r   r   idxZhave_idxZ	util_namerm   rk   r   implr    ry   r!   generate_buffer_slice_code   sb    







z5MemoryViewSliceBufferEntry.generate_buffer_slice_codeN)T)__name__
__module____qualname____doc__rZ   r^   r`   rb   rj   ri   r   r    r    r    r!   rU      s   
.rU   c                 C   s   t | }t j| |||dS )N)startstopstep)r   ZNoneNodeZ	SliceNode)r   noner    r    r!   empty_sliceX  s
    
 r   c                 C   s   g }d}d}dd | D }t | t | }| D ]j}t|tjr~d}t|j}|r\|| q|| d }	||g|	  d}q.|p|jp|j	}|| q.t |t | }
|
|k rd}||
 }	|t| d jg|	  |||fS )NFc                 S   s   g | ]}|j r|qS r    )r   )rc   Znewaxisr    r    r!   rg   c  s      zunellipsify.<locals>.<listcomp>Tr   r{   )
len
isinstancer   ZEllipsisNoder   r   rx   rA   r   r   )r   rI   r*   Zseen_ellipsisr   ZnewaxesZ	n_indicesrt   Z
full_sliceZnslicesZresult_lengthr    r    r!   unellipsify^  s,    
r   c                 C   s   | dkr|dkrdS | dkr(|dkr(dS | dkr<|dkr<dS | dkrP|dkrPdS | d	krd|dkrdd
S | |fdks|t | |fdS d S )Nr   )r   r   rk   r   rl   r   rm   rn   r   r   r   r   ro   r(   )r=   r>   r    r    r!   rq   ~  s    rq   c                 C   s   | dkst d| |f S )NCFz!__pyx_memviewslice_is_contig_%s%dr   )contig_typerI   r    r    r!   get_is_contig_func_name  s    r   c                 C   s.   | dkst tt|| d}td|tgd}|S )Nr   )rI   r   ZMemviewSliceCheckContigrequires)r(   r   r   load_memview_c_utilityis_contig_utility)r   rI   r   Zutilityr    r    r!   get_is_contig_utility  s    r   c                 C   s0   | j s| jr|st| |||S t| |||S d S rG   )r;   r<   ContigSliceIterStridedSliceIter)
slice_typeslice_resultrI   r   Zforce_stridedr    r    r!   
slice_iter  s    r   c                   @   s   e Zd Zdd ZdS )	SliceIterc                 C   s   || _ || _|| _|| _d S rG   )r   r   r   rI   )rY   r   r   rI   r   r    r    r!   rZ     s    zSliceIter.__init__N)r   r   r   rZ   r    r    r    r!   r     s   r   c                   @   s   e Zd Zdd Zdd ZdS )r   c                    st    j }|   jj }d fddt jD }|d|  |d |d|| j	f  |d dS )	Nz * c                 3   s   | ]}d  j |f V  qdS )ra   N)r   )rc   rS   r]   r    r!   rw     s   z.ContigSliceIter.start_loops.<locals>.<genexpr>z"Py_ssize_t __pyx_temp_extent = %s;zPy_ssize_t __pyx_temp_idx;z(%s *__pyx_temp_pointer = (%s *) %s.data;zPfor (__pyx_temp_idx = 0; __pyx_temp_idx < __pyx_temp_extent; __pyx_temp_idx++) {Z__pyx_temp_pointer)
r   begin_blockr   rD   rp   joinrangerI   r   r   )rY   r   rs   Z
total_sizer    r]   r!   start_loops  s    
  
zContigSliceIter.start_loopsc                 C   s&   | j d | j d | j   d S )Nz__pyx_temp_pointer += 1;})r   r   	end_blockr]   r    r    r!   	end_loops  s    zContigSliceIter.end_loopsNr   r   r   r   r   r    r    r    r!   r     s   r   c                   @   s   e Zd Zdd Zdd ZdS )r   c                 C   s   | j }|  t| jD ]H}|| j|f}|d|  |d|  |d|  |d|  q|d| j  t| jD ]8}|dkr|d||d f  |d	||||f  q|d
| jd  S )Nz/Py_ssize_t __pyx_temp_extent_%d = %s.shape[%d];z1Py_ssize_t __pyx_temp_stride_%d = %s.strides[%d];zchar *__pyx_temp_pointer_%d;zPy_ssize_t __pyx_temp_idx_%d;z__pyx_temp_pointer_0 = %s.data;r   z.__pyx_temp_pointer_%d = __pyx_temp_pointer_%d;r   z\for (__pyx_temp_idx_%d = 0; __pyx_temp_idx_%d < __pyx_temp_extent_%d; __pyx_temp_idx_%d++) {z__pyx_temp_pointer_%d)r   r   r   rI   r   r   )rY   r   rS   tr    r    r!   r     s     
zStridedSliceIter.start_loopsc                 C   sF   | j }t| jd ddD ] }|d||f  |d q|  d S )Nr   r{   z.__pyx_temp_pointer_%d += __pyx_temp_stride_%d;r   )r   r   rI   r   r   )rY   r   rS   r    r    r!   r     s
    zStridedSliceIter.end_loopsNr   r    r    r    r!   r     s   r   c                 C   s    | j rd}nd}d|  |f S )Nr   r   z!__pyx_memoryview_copy_slice_%s_%s)r;   Zspecialization_suffix)ZmemviewZc_or_fr    r    r!   copy_c_or_fortran_cname  s     r   c                 C   s   |j |j kr0|j jr"|j j|j ks0t| d d S t|jt|jkrRt| d d S |jsl|jslt| d d S |jD ] \}}|dkrrt| d  d S qr|jrd}t}n|jst	d}t
}tdtt||j  ||jt|t|j jd	tgd
S )Nzdtypes must be the same!z!number of dimensions must be samez%to_memview must be c or f contiguous.r   z2cannot handle 'full' or 'ptr' access at this time.r   fortranZCopyContentsUtility)modeZ
dtype_declcontig_flagrI   Z
func_cnameZdtype_is_objectr   r   )rD   Zis_cv_qualifiedZcv_base_typer   r   rB   r;   r<   r4   r(   r5   r   r   r   rp   rI   r   r   rR   copy_contents_new_utility)r   Zfrom_memviewZ
to_memviewr=   r>   r   r   r    r    r!   get_copy_new_utility  sH    





r   c                    s  |   jj}|  |j t fdddD }t fdddD }d\}}d\}}|d }	}
g }t|D ]\}}|jjst	|jj
t|jjst	|jj
t|jjr|||f qlt|jtr|j| d	krt	|jj
t||	d
f qlt|jttfr>t| |j}|jtkr.|t|j  nt	|jj
tqlt	|jj
tqld}d}t|D ]@\}\}}|d
kr^|rt	|jj
t|}|df||< d}q^|r|t|d	 krd}n0d}|r||d	  d dkrt	|| j
d|r.d}tt|D ](\}\}}|dkrt|| d	 }q|d	 }t|| }t||| D ]Z\}\}}|d	 | }|dkrt	|| j
d|dkrt	|| j
d||
f||< qR|r|d \}}|df|d< tdd |D ||| |S )z
    get_axes_specs(env, axes) -> list of (access, packing) specs for each axis.
    access is one of 'full', 'ptr' or 'direct'
    packing is one of 'contig', 'strided' or 'follow'
    c                    s   g | ]}  |qS r    lookuprc   name	viewscoper    r!   rg     s   z"get_axes_specs.<locals>.<listcomp>)r   r   r   c                    s   g | ]}  |qS r    r   r   r   r    r!   rg      s   r   r   r   )FFr@   r   r   Zcfcontigr   Fr   T)r   r   z>Fortran contiguous specifier must follow an indirect dimensionr{   r   r   r   z>Indirect dimension may not follow Fortran contiguous dimensionzDimension may not be contiguousc                 S   s   g | ]}|j jqS r    )r   r   rc   axisr    r    r!   rg   }  s     )global_scoper   cython_scopeload_cythonscoper   tuplerh   r   r   r   r   	START_ERRr   STOP_ERRr   rx   r   r   Zcompile_time_valueSTEP_ERRr   r	   _get_resolved_specr   view_constant_to_access_packingINVALID_ERRBOTH_CF_ERRr   reversedvalidate_axes_specs)envrB   Zcythonscopeaccess_specspacking_specsr<   r;   Zdefault_accessZdefault_packingZ	cf_accessZ
cf_packingZ
axes_specsr   r   rV   Z
contig_dimZ	is_contigr=   r>   r   r   ar   r    r   r!   get_axes_specs  s    



	



r   c                 C   s    t |tjkrt| d dS dS )NzGMore dimensions than the maximum number of buffer dimensions were used.FT)r   r
   buffer_max_dimsr   )r   rB   r    r    r!   validate_axes  s    
r   c                 C   s   d }}t | dkr$| dgkr$d}nb| d dkrPtdd | d d D rPd}n6t | dkr| d dkrtd	d | dd  D rd}||fS )
NFr   r   Tr{   c                 s   s   | ]}|d kV  qdS )r   r   Nr    r   r    r    r!   rw     s     zis_cf_contig.<locals>.<genexpr>r   c                 s   s   | ]}|d kV  qdS r   r    r   r    r    r!   rw     s     )r   r   )r:   r;   r<   r    r    r!   r3     s    
r3   c                 C   s<   t | \}}|rdS |rdS | D ]\}}|dkr  dS q dS )Nr   r   r   r   r   )r3   r9   r    r    r!   get_mode  s    r   )r   r   r@   )r   r   )r   r   r   )r   r   )rk   r   rm   rl   ro   rn   c                 C   sD  d}d}d } } }}	d}
t |D ]\}\}}|dkr$|}
q$t t| |D ]\}\}\}}||krl||ksvt|d|dkrd}n|d	kr|rt|d
|
d t|d f}||kr|dkr|
d t|d krd| }nd|d  }t|d| |dk}n0|dkr2|rt|d|s2|s2t|d|dkrLd}qLd S )Nr   )r   r   r   Fr{   r   zInvalid axes specification.r   Tr   z1Only one direct contiguous axis may be specified.r   zdimensions %d and %dzdimension %dr   z$Only %s may be contiguous and directr   zAA memoryview cannot have both follow and strided axis specifiers.z$Invalid use of the follow specifier.r   )rh   r6   r   r   )Z	positionsr:   r;   r<   r   r   Z
has_contigZ
has_followZhas_stridedZhas_generic_contigZlast_indirect_dimensionr   r=   r>   r   Zvalid_contig_dimsZdimsr    r    r!   r     s>    







r   c                 C   s8   t |trt| |S t |tr(t| |S t|jtd S rG   )r   r   _resolve_NameNoder	   _resolve_AttributeNoder   r   r   )r   specr    r    r!   r     s
    



r   c                 C   sd   z|  |jj}W n  tk
r2   t|jtY nX |  jjj	}| |}|d kr`t|jt
|S rG   )r   r   AttributeErrorr   r   r   r   r   r   r   NOT_CIMPORTED_ERR)r   nodeZresolved_namer   rV   r    r    r!   r     s    
r   c                 C   s   g }t |tr$|d|j |j}qt |tr>|d|j nt|jt	|d d }|s^t
| }|D ].}||}|r~|jst|jd| |j}qf||d }|st|jd|d  |S )Nr   r{   zundeclared name not builtin: %szNo such attribute '%s')r   r	   insert	attributeobjr   r   r   r   EXPR_ERRr(   r   Z	as_module)r   r   pathmodnamesrP   modnamemodrV   r    r    r!   r     s,    



 r   Nc                 K   s   t j| dfd|i|S )NzMemoryView.pyxr   )r   loadZutil_code_namer   kwargsr    r    r!   load_memview_cy_utility  s
    r   c                 K   s4   |d krt j| df|S tj| dfd|i|S d S )Nr   r   )r   r   r   r   r    r    r!   r      s    r   c                 C   s&   |   jj}|  d|jd_d S )NTarray_cwrapper)r   r   r   r   r   r   used)r   r   r    r    r!   use_cython_array_utility_code'  s    r   rN   )Zmemview_struct_nameZmax_dimsZmemviewslice_nameZmemslice_initZTHREAD_LOCKS_PREALLOCATEDZMemviewSliceStructr   ZAtomicsZMemviewSliceInit)ZBUF_MAX_NDIMSZMemviewSliceIndexZBufferFormatFromTypeInfor   ZMemviewSliceIsContigZOverlappingSlicesZMemviewSliceCopyTemplatezView.MemoryView)arrayr   r   rk   r   rm   ro   rn   )FT)FF)r   )F)N)N)d
__future__r   ZErrorsr   r    r   r   r   r	   r
   ZCoder   r   r   r   r   r   r   r   r   r   r   r   r   ZCF_ERRZERR_UNINITIALIZEDZformat_flagr4   r5   Zmemview_any_contiguousr7   r8   ZMEMVIEW_DIRECTZMEMVIEW_PTRZMEMVIEW_FULLZMEMVIEW_CONTIGZMEMVIEW_STRIDEDZMEMVIEW_FOLLOWZ_spec_to_constZ_spec_to_abbrevZmemslice_entry_initZmemview_nameZmemview_typeptr_cnameZmemview_objstruct_cnameZmemviewslice_cnamer"   r1   r-   r?   rF   rL   rQ   ZBufferEntryrU   r   r   rq   r   r   r   objectr   r   r   r   r   r   r   r3   r   r   r   r   r   r   r   r   r   r   rC   default_valuer   Zmemviewslice_declare_codeZatomic_utilityr   Zmemviewslice_init_coderr   Z_typeinfo_to_format_codeZtypeinfo_to_format_coder   Zoverlapping_utilityr   ZGetAndReleaseBufferUtilityCodeZbuffer_struct_declare_codeZbuffer_formats_declare_codeZview_utility_codeZview_utility_allowlistr   rx   r    r    r    r!   <module>   s  
	
   
   

! ' 
!
&r		.	


 

