o
    wi                     @   s  d dl mZmZmZ ddlZz(d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yE   dZd	Zd	ZY nw erKd
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 rd%d& Znd'd& Zd(d) Zd*d+ Zejd?d,d-Z ejd.d/ Z!d0d1 Z"ejd2d3 Z#d4d5 Z$ejd6d7 Z%d8d9 Z&d:d; Z'd<d= Z(dS )@   )CTypeCTypedefTypeCStructOrUnionType    N.   )r   	      )r   r      T__builtin__builtinsc                 C   s(   | d u rdS t | dd }|o|ddS )NF
directives
np_pythran)getattrget)envr    r   T/home/ubuntu/sommelier/.venv/lib/python3.10/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_typetypedef_base_type
is_numerictype_r   r   r   is_pythran_supported_dtype   s   

r   ndarrayc                 C   s   | j rA| j| j}}t|tr|j}nt|tr| }nt|tr&|j	}nt
d| tr5d|||f S d||dd| f S | jrG| jS | jrN|  S t
d| t| f )Nzunsupported type %s!zpythonic::types::%s<%s,%d>z3pythonic::types::%s<%s,pythonic::types::pshape<%s>>,)longz unsupported pythran type %s (%s))	is_bufferndimdtyper   r   cnamer   sign_and_namer   typedef_cname
ValueErrorpythran_is_pre_0_9joinis_pythran_exprpythran_typer   type)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   r0   c                 C   s2   | dkrdt |t |f S dt || t |f S )Nz**zSdecltype(pythonic::numpy::functor::power{}(std::declval<%s>(), std::declval<%s>()))z2decltype(std::declval<%s>() %s std::declval<%s>())r*   )optAtBr   r   r   pythran_binop_type@   s   r5   c                 C   s   d| t |f S )Nzdecltype(%sstd::declval<%s>())r1   )r2   r   r   r   r   pythran_unaryop_typeI   s   r6   c                    s4   d  fdd|D }t|dkrd| S d| S )Nr   c                    s   g | ]} |qS r   r   ).0idx
index_coder   r   
<listcomp>P   s    z!_index_access.<locals>.<listcomp>r   z[%s]z(%s))r(   len)r:   indicesindexingr   r9   r   _index_accessN   s   r?   c                 C   sf   | \}}|j rdt|jj  }dtddg| f S |jr%d|  S |jr-d|j	 S t
d| )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&   )index_with_typer8   
index_typenr   r   r   _index_type_codeT   s   
rK   c                 C   s~   | j r(| j| j| jf}| jjrd}|d d }nd}d|ddd |D f S | jjr0t| S | jj	r8| 
 S td| j )	Ncontiguous_slicer   slicezpythonic::types::%s(%s)r   c                 s   s    | ]}|  V  qd S r   )pythran_result)r7   vr   r   r   	<genexpr>k   s    z_index_code.<locals>.<genexpr>zunsupported indexing type %s)rB   startstoprD   rE   r(   r+   rG   
to_pythranr)   rN   r&   )r8   valuesfuncr   r   r   _index_codeb   s   rV   c                 C   s   t dt| tt|f S )Nzdecltype(std::declval<%s>()%s))r0   r*   r?   rK   )r   r=   r   r   r   pythran_indexing_types   s   rW   c                 C   s
   t t| S r   )r?   rV   )r=   r   r   r   pythran_indexing_codez   s   
rX   c                 C   s   | j sg S t| j| jg S r   )is_numpy_attributenp_func_to_listobj	attributerU   r   r   r   rZ   }   s   rZ   c                 C   s   dS )NFr   )namer   r   r   pythran_is_numpy_func_supported   s   r_   c                 C   s<   t jjd }t| }|D ]}||d }|d u r dS qdS )NnumpyFT)pythrantablesMODULESrZ   r   )rU   CurFFLFr   r   r   r_      s   c                 C   s0   t | } d| d d dg }d|| d f S )Nz::functorzpythonic::numpy::%s::%srZ   r(   )rU   
submodulesr   r   r   pythran_functor   s   rk   c                 C   s$   d dd |D }dt| |f S )Nr   c                 s   s    | ]
}d t |j V  qdS )rA   N)r*   r+   )r7   ar   r   r   rP      s    z$pythran_func_type.<locals>.<genexpr>zdecltype(%s{}(%s)))r(   rk   )rU   argsr   r   r   pythran_func_type   s   rn   c                 C   s   | j }|jr||  S t|r(|  st| dd r(d|   d|   dS t|g dr3|  S | jr:dt	 S |d u rBt
|}| j jsHJ d||  f S )Nentryz	decltype(z){})r)   r   is_float
is_complexzpythonic::%s::Nonezfrom_python<%s>(%s))r+   rG   	cast_coderesultr)   result_in_tempr   is_typerE   rF   r*   is_pyobject	py_result)r2   r-   op_typer   r   r   rS      s   rS   c                 C   s    |D ]}t | |dr dS qdS )NFT)r   )r   typesattrr   r   r   rv      s
   rv   c                 C   s   | j pt| jS r   )rE   r   r+   )noder   r   r   !is_pythran_supported_node_or_none   s   r}   c                 C      d}t | |p
t| S )N)r)   rG   r   rq   rE   rr   rv   r)   r   pythran_supportedr   r   r   r      s   r   c                 C   r~   )N)r)   rG   r   rq   rr   r   r   r   r   r   #is_pythran_supported_operation_type   s   r   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)cstrided)is_numpy_bufferr   r"   modecastr   r   r   r   is_pythran_buffer   s
   r   c                 C   s   t | } dd|  S )Nzpythonic/numpy/%s.hpp/ri   r]   r   r   r   pythran_get_func_include_file   s   r   c                 C   s   |  d |  d |  d |  d |  d |  dt  |  d dD ]}|  d	|  |  d
|  q'dD ]	}|  d|  q:d 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)
floatfloat32float64setrM   tuplerC   complex	complex64
complex128zpythonic/types/%s.hpp)add_include_filerF   )r   itr   r   r   include_pythran_generic   s   





r   )r   r   ))
PyrexTypesr   r   r   cythonra   r   maprC   __version__splitr'   pythran_is_pre_0_9_6ImportErrorrF   r   ccallr   r*   cfuncr0   r5   r6   r?   rK   rV   rW   rX   rZ   r_   rk   rn   rS   rv   r}   r   r   r)   r   r   r   r   r   r   r   <module>   s^    "&


	

	


