a
    bc                     @   s  d dl mZ ddlmZmZmZ d dlZzPd dlZee	e
ejdd d dk Zee	e
ejdd d d	k ZW n ey   dZd
Zd
ZY n0 erdZndZdd Zejdd Zd?ddZejdd Zdd Zdd Zejdd Zdd Zdd Zd d! Zd"d# Zd$d% Zedu r.d&d' Znd(d' Zd)d* Z d+d, Z!ejd@d-d.Z"ejd/d0 Z#d1d2 Z$ejd3d4 Z%d5d6 Z&ejd7d8 Z'd9d: Z(d;d< Z)d=d> Z*dS )A    )absolute_import   )CTypeCTypedefTypeCStructOrUnionTypeN.   )r   	      )r   r	      T__builtin__builtinsc                 C   s(   | d u rdS t | dd }|o&|ddS )NF
directivesZ
np_pythran)getattrget)envr    r   g/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/Cython/Compiler/Pythran.pyhas_np_pythran   s    r   c                 C   s   t | trt| jS | jS N)
isinstancer   is_pythran_supported_typeZtypedef_base_type
is_numerictype_r   r   r   is_pythran_supported_dtype    s    

r   ndarrayc                 C   s   | j r| j| j }}t|tr&|j}n2t|tr:| }nt|trL|j	}nt
d| trjd|||f S d||dd| f S | jr| jS | jr|  S t
d| t| f d S )Nzunsupported type %s!zpythonic::types::%s<%s,%d>z3pythonic::types::%s<%s,pythonic::types::pshape<%s>>,)longz unsupported pythran type %s (%s))Z	is_bufferndimdtyper   r   cnamer   sign_and_namer   Ztypedef_cname
ValueErrorpythran_is_pre_0_9joinis_pythran_exprpythran_typer   type)ZTyptyper   r    ctyper   r   r   r'   '   s"    



r'   c                 C   s   d|  S )Nz(typename std::remove_reference<%s>::typer   )tyr   r   r   type_remove_ref?   s    r,   c                 C   s6   | dkrdt |t |f S dt || t |f S d S )Nz**zSdecltype(pythonic::numpy::functor::power{}(std::declval<%s>(), std::declval<%s>()))z2decltype(std::declval<%s>() %s std::declval<%s>())r'   )opZtAZtBr   r   r   pythran_binop_typeD   s    r/   c                 C   s   d| t |f S )Nzdecltype(%sstd::declval<%s>())r-   )r.   r   r   r   r   pythran_unaryop_typeM   s    r0   c                    s0   d  fdd|D }t|dkr(dnd| S )Nr   c                    s   g | ]} |qS r   r   ).0idx
index_coder   r   
<listcomp>T       z!_index_access.<locals>.<listcomp>r   z[%s]z(%s))r%   len)r4   indicesZindexingr   r3   r   _index_accessR   s    r9   c                 C   sj   | \}}|j r8dt|jj  }dtddg| f S |jrJd|  S |jrZd|j	 S t
d| d S )Nr   z"pythonic::%s::functor::slice{}(%s)r   0std::declval<%s>()zunsupported indexing type %s!)is_sliceintstepis_nonepythran_builtinsr%   is_intr"   r&   r'   r#   )Zindex_with_typer2   Z
index_typenr   r   r   _index_type_codeX   s    
rC   c                 C   s   | j rP| j| j| jf}| jjr0d}|d d }nd}d|ddd |D f S | jjr`t| S | jj	rp| 
 S td| j d S )	NZcontiguous_slicer   slicezpythonic::types::%s(%s)r   c                 s   s   | ]}|  V  qd S r   )pythran_result)r1   vr   r   r   	<genexpr>o   r6   z_index_code.<locals>.<genexpr>zunsupported indexing type %s)r<   startstopr>   r?   r%   r(   rA   
to_pythranr&   rE   r#   )r2   valuesfuncr   r   r   _index_codef   s    rM   c                 C   s   t dt| tt|f S )Nzdecltype(std::declval<%s>()%s))r,   r'   r9   rC   )r   r8   r   r   r   pythran_indexing_typew   s    rN   c                 C   s
   t t| S r   )r9   rM   )r8   r   r   r   pythran_indexing_code~   s    rO   c                 C   s   | j s
g S t| j| jg S r   )Zis_numpy_attributenp_func_to_listobj	attributerL   r   r   r   rP      s    rP   c                 C   s   dS )NFr   )namer   r   r   pythran_is_numpy_func_supported   s    rU   c                 C   s<   t jjd }t| }|D ]}||d }|d u r dS qdS )NZnumpyFT)pythranZtablesZMODULESrP   r   )rL   ZCurFZFLFr   r   r   rU      s    c                 C   s0   t | } d| d d dg }d|| d f S )Nz::Zfunctorzpythonic::numpy::%s::%srP   r%   )rL   Z
submodulesr   r   r   pythran_functor   s    rZ   c                 C   s$   d dd |D }dt| |f S )Nr   c                 s   s   | ]}d t |j V  qdS )r;   N)r'   r(   )r1   ar   r   r   rG      r6   z$pythran_func_type.<locals>.<genexpr>zdecltype(%s{}(%s)))r%   rZ   )rL   argsr   r   r   pythran_func_type   s    r]   c                 C   sj   | j }|jr||  S t|g dr0|  S | jr>dt S |d u rNt|}| j jsZJ d|| 	 f S )N)r&   r   is_float
is_complexzpythonic::%s::Nonezfrom_python<%s>(%s))
r(   rA   Z	cast_coderesultis_typer?   r@   r'   Zis_pyobjectZ	py_result)r.   r)   Zop_typer   r   r   rJ      s    rJ   c                 C   s    |D ]}t | |dr dS qdS )NFT)r   )r   typesattrr   r   r   ra      s    ra   c                 C   s   | j pt| jS r   )r?   r   r(   )noder   r   r   !is_pythran_supported_node_or_none   s    re   c                 C   s   d}t | |pt| S )N)r&   rA   r   r^   r?   r_   ra   r&   r   Zpythran_supportedr   r   r   r      s    r   c                 C   s   d}t | |pt| S )N)r&   rA   r   r^   r_   rf   rg   r   r   r   #is_pythran_supported_operation_type   s    rh   c                 C   s   | j S r   )r&   r   r   r   r   r&      s    r&   c                 C   s"   | j o t| jo | jdv o | j S )N)cZstrided)Zis_numpy_bufferr   r    modecastr   r   r   r   is_pythran_buffer   s
    rl   c                 C   s   t | } dd|  S )Nzpythonic/numpy/%s.hpp/rY   rS   r   r   r   pythran_get_func_include_file   s    rn   c                 C   s   |  d |  d |  d |  d |  d |  dt  |  d dD ] }|  d	|  |  d
|  qNdD ]}|  d|  qtd S )Nzpythonic/core.hppzpythonic/python/core.hppzpythonic/types/bool.hppzpythonic/types/ndarray.hppzpythonic/numpy/power.hppzpythonic/%s/slice.hppz<new>)          @   zpythonic/types/uint%d.hppzpythonic/types/int%d.hpp)
floatZfloat32Zfloat64setrD   tupler=   complexZ	complex64Z
complex128zpythonic/types/%s.hpp)Zadd_include_filer@   )r   itr   r   r   include_pythran_generic   s    





ry   )r   )N)+
__future__r   Z
PyrexTypesr   r   r   ZcythonrV   ru   mapr=   __version__splitr$   Zpythran_is_pre_0_9_6ImportErrorr@   r   Zccallr   r'   Zcfuncr,   r/   r0   r9   rC   rM   rN   rO   rP   rU   rZ   r]   rJ   ra   re   r   rh   r&   rl   rn   ry   r   r   r   r   <module>   s\   "&



	


	


