U
    ;vfX                  X   @   s  d Z ddlZddlZddlZddlmZ ddlmZ ddlmZ ddd	d
dddddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^gXZ	ej
ZejjZejZi Zg ZdZd_dY Zd`d Zdadb Zdcdd Zded  Zdfd Zdgd ZdhdJ ZdidH ZdjdI Zdkd Zdld ZdmdF Zdnd! Z dod9 Z!dpd, Z"dqdr Z#dsdt Z$dud* Z%dvd> Z&dwdP Z'dxdT Z(dydO Z)dzdR Z*d{d% Z+d|d< Z,d}d; Z-d~d" Z.dd+ Z/ddQ Z0ddU Z1dd Z2ddS Z3dd Z4dd Z5dd Z6ddG Z7dd Z8dd Z9ddA Z:dd@ Z;dd( Z<dd) Z=ddL Z>ddM Z?dd^ Z@dd ZAdd ZBddE ZCdd: ZDdd? ZEdd= ZFdd# ZGdd$ ZHddK ZIdd ZJddN ZKdd ZLddB ZMdd' ZNddD ZOdd3 ZPdd4 ZQdd7 ZRdd2 ZSdd6 ZTdd. ZUdd ZVdd0 ZWdd8 ZXdd/ ZYdd5 ZZdd- Z[dd Z\dd Z]dd Z^ePdeQdeRdeSdeVdeUdeMdeZde\de]de^diZ_ddC Z`dd ZaddɄ Zbdd Zcdd Zddd Zedd΄ ZfddЄ Zgdd Zhdd ZiddԄ Zjddք ZkG dd؄ delZmG dd] d]ZnddV ZoddX ZpddW Zqdd& Zrdd ZsdddZtdd Zui fddZvdd Zwdd Zxdd Zydd Zzdd Z{dd Z|dd Z}dd Z~dd\ ZdddZZdd	 Zi fddZdS )aW  

Auxiliary functions for f2py2e.

Copyright 1999,2000 Pearu Peterson all rights reserved,
Pearu Peterson <pearu@ioc.ee>
Permission to use, modify, and distribute this software is given under the
terms of the NumPy (BSD style) LICENSE.


NO WARRANTY IS EXPRESSED OR IMPLIED.  USE AT YOUR OWN RISK.
$Date: 2005/07/24 19:01:55 $
Pearu Peterson

    N)reduce   )__version__)cfuncs
applyrules	debugcapi
dictappenderrmessgentitlegetargs2getcallprotoargumentgetcallstatementgetfortrannamegetpymethoddef
getrestdocgetusercodegetusercode1hasbodyhascallstatement	hascommonhasexternalshasinitvaluehasnotehasresultnoteisallocatableisarrayisarrayofstringsischaracterischaracterarrayischaracter_or_characterarray	iscomplexiscomplexarrayiscomplexfunctioniscomplexfunction_warnisdoubleisdummyroutine
isexternal
isfunctionisfunction_wrapisint1isint1array	isintegerisintent_aux
isintent_cisintent_callbackisintent_copyisintent_dictisintent_hideisintent_inisintent_inoutisintent_inplaceisintent_nothideisintent_outisintent_overwrite	islogicalislogicalfunctionislong_complexislong_doubleislong_doublefunctionislong_longislong_longfunctionismoduleismoduleroutine
isoptional	isprivate
isrequired	isroutineisscalarissigned_long_longarrayisstringisstringarrayisstring_or_stringarrayisstringfunctionissubroutineissubroutine_wrapisthreadsafe
isunsignedisunsigned_charisunsigned_chararrayisunsigned_long_longisunsigned_long_longarrayisunsigned_shortisunsigned_shortarrayl_andl_notl_oroutmessreplaceshow
stripcommathrow_errorisattr_valuec                 C   s   t ddrtj|  d S )Nverboser   )optionsgetsysstdoutwrite)t re   7/tmp/pip-unpacked-wheel-sdaoq5e2/numpy/f2py/auxfuncs.pyrX   A   s    c                 C   s   dt kS )NZcapi)debugoptionsvarre   re   rf   r   F   s    c                 C   s   d| ko| d dkot |  S Ntypespec	characterr&   rh   re   re   rf   _ischaracterJ   s    rn   c                 C   s   d| ko| d dkot |  S rj   rm   rh   re   re   rf   	_isstringO   s    ro   c                 C   s   t | od| kS NZcharselectorrn   rh   re   re   rf   r   T   s    c                 C   s   t | ot|  S Nr   r   rh   re   re   rf   r   X   s    c                 C   s   t | ot| S rr   rs   rh   re   re   rf   r   \   s    c                 C   s   t | od| kS rp   rq   rh   re   re   rf   rI   `   s    c                 C   s   t | ot|  S rr   rI   r   rh   re   re   rf   rG   d   s    c                 C   s   t | ot| S rr   rt   rh   re   re   rf   rH   h   s    c                 C   s   t | o| d d dkS )N	dimensionz(*))rH   rh   re   re   rf   r   l   s    c                 C   s   d| kot |  S Nru   rm   rh   re   re   rf   r   r   s    c                 C   s   t | pt| pt|  S rr   )r   rG   r&   rh   re   re   rf   rE   v   s    c                 C   s   t | o| ddkS Nrk   )complexzdouble complexrE   r`   rh   re   re   rf   r    z   s    c                 C   s   t | o| ddkS )Nrk   logicalrz   rh   re   re   rf   r8      s    c                 C   s   t | o| ddkS )Nrk   integerrz   rh   re   re   rf   r+      s    c                 C   s   t | o| ddkS )Nrk   realrz   rh   re   re   rf   isreal   s    r~   c                 C   sP   z| d d W S  t k
rJ   z| d d W  Y S  t k
rD   Y nX Y nX d S )NZkindselector*kindKeyErrorrh   re   re   rf   get_kind   s    r   c                 C   s$   |  ddko"t| dko"t|  S Nrk   r|   1)r`   r   r   rh   re   re   rf   r)      s
    
c                 C   s*   t | sdS | ddkrdS t| dkS )Nr   rk   r|   r{   8rE   r`   r   rh   re   re   rf   r=      s
    c                 C   s*   t | sdS | ddkrdS t| dkS )Nr   rk   r|   -1r   rh   re   re   rf   rO      s
    c                 C   s*   t | sdS | ddkrdS t| dkS )Nr   rk   r|   -2r   rh   re   re   rf   rS      s
    c                 C   s*   t | sdS | ddkrdS t| dkS )Nr   rk   r|   -4r   rh   re   re   rf   rN      s
    c                 C   s*   t | sdS | ddkrdS t| dkS )Nr   rk   r|   -8r   rh   re   re   rf   rQ      s
    c                 C   s*   t | sdS | ddksdS t| dkS )Nr   rk   r}   r   r   rh   re   re   rf   r$      s
    c                 C   s*   t | sdS | ddksdS t| dkS )Nr   rk   r}   Z16r   rh   re   re   rf   r;      s
    c                 C   s   t | sdS t| dkS )Nr   Z32)r    r   rh   re   re   rf   r:      s    c                 C   s   t | o| ddkS rx   )r   r`   rh   re   re   rf   r!      s    c                 C   s"   t | o | ddko t| dkS r   r   r`   r   rh   re   re   rf   r*      s    
c                 C   s"   t | o | ddko t| dkS )Nrk   r   r   r   rh   re   re   rf   rP      s    
c                 C   s"   t | o | ddko t| dkS )Nrk   r   r   r   rh   re   re   rf   rT      s    
c                 C   s"   t | o | ddko t| dkS )Nrk   r   r   r   rh   re   re   rf   isunsignedarray   s    
r   c                 C   s"   t | o | ddko t| dkS )Nrk   r   r   r   rh   re   re   rf   rR      s    
c                 C   s"   t | o | ddko t| dkS )Nrk   r   r   r   rh   re   re   rf   issigned_chararray   s    
r   c                 C   s"   t | o | ddko t| dkS )Nrk   r   2r   rh   re   re   rf   issigned_shortarray   s    
r   c                 C   s"   t | o | ddko t| dkS )Nrk   r   4r   rh   re   re   rf   issigned_array   s    
r   c                 C   s"   t | o | ddko t| dkS )Nrk   r   r   r   rh   re   re   rf   rF     s    
c                 C   s   d| kod| d kS )NattrspecZallocatablere   rh   re   re   rf   r   
  s    c                 C   s   d| kpt |  S rw   )rG   rh   re   re   rf   	ismutable  s    r   c                 C   s   d| kS )N
modulenamere   routre   re   rf   r@     s    c                 C   s   d| kod| d kS )Nblockmodulere   r   re   re   rf   r?     s    c                 C   s   d| kod| d kS )Nr   functionre   r   re   re   rf   r'     s    c                 C   s"   t | rdS to t| o t|  S Nr   )r-   	wrapfuncsr'   r&   r   re   re   rf   r(     s    c                 C   s   d| kod| d kS )Nr   Z
subroutinere   r   re   re   rf   rK   $  s    c                 C   s   t | rdS t| ot| S r   )r-   rK   hasassumedshaper   re   re   rf   rL   (  s    c                 C   s   d|  dg kS )Nvaluer   r`   rh   re   re   rf   r]   -  s    c                 C   sV   |  drdS | d D ]:}| d  |i  dg D ]}|dkr2d| d<   dS q2qdS )Nr   Targsvarsru   :Fr   )r   adre   re   rf   r   1  s    
r   c                 C   s   t | pt| S rr   )r@   r   r   re   re   rf   requiresf90wrapper<  s    r   c                 C   s   t | pt| S rr   )r'   rK   r   re   re   rf   rD   @  s    c                 C   sF   t | sdS d| kr| d }n| d }|| d krBt| d | S dS Nr   resultnamer   )r'   r8   r   r   re   re   rf   r9   D  s    
c                 C   sF   t | sdS d| kr| d }n| d }|| d krBt| d | S dS r   )r'   r=   r   re   re   rf   r>   P  s    
c                 C   sF   t | sdS d| kr| d }n| d }|| d krBt| d | S dS r   )r'   r;   r   re   re   rf   r<   \  s    
c                 C   sF   t | sdS d| kr| d }n| d }|| d krBt| d | S dS r   )r'   r    r   re   re   rf   r"   h  s    
c                 C   s   t | rtd dS dS )Na      **************************************************************
        Warning: code with a function returning complex value
        may not work correctly with your Fortran compiler.
        When using GNU gcc/g77 compilers, codes should work
        correctly for callbacks with:
        f2py -c -DF2PY_CB_RETURNCOMPLEX
    **************************************************************
r   r   )r"   rX   r   re   re   rf   r#   t  s    c                 C   sF   t | sdS d| kr| d }n| d }|| d krBt| d | S dS r   )r'   rG   r   re   re   rf   rJ     s    
c                 C   s   d| ko| d S )NZ	externalsre   r   re   re   rf   r     s    c                 C   s   d| kod| d kS )Nf2pyenhancementsZ
threadsafere   r   re   re   rf   rM     s    
c                 C   s   d| ko| d S )Nr   re   r   re   re   rf   hasvariables  s    r   c                 C   s(   d| ko&d| d ko&d| d ko&t | S )Nr   optionalrequired)r5   rh   re   re   rf   rA     s
    
c                 C   s   d| kod| d kS )Nr   Zexternalre   rh   re   re   rf   r&     s    c                 C   s   t |  ot| S rr   )rA   r5   rh   re   re   rf   rC     s    c                 C   sp   d| krdS d| d krdS d| d kr,dS d| d kr<dS d| d krLdS d| d kr\dS d	| d krldS dS )
Nintentr   hider   inplaceinoutinoutoutinre   rh   re   re   rf   r2     s    c                 C   sD   d| koBd| d ks d| d koBd| d koBd| d koBd| d kS )Nr   r   r   r   r   r   re   rh   re   re   rf   r3     s    



c                 C   s   d|  dg kS )Nr   r   r   rh   re   re   rf   r6     s    c                 C   s<   d| ko:d| d kp:d| d ko:d| d ko:t tt|  S )Nr   r   r   r   )rW   r3   r4   rh   re   re   rf   r1     s    c                 C   s
   t |  S rr   )r1   rh   re   re   rf   r5     s    c                 C   s   d|  dg kS )Ncr   r   rh   re   re   rf   r-     s    c                 C   s   d|  dg kS )Ncacher   r   rh   re   re   rf   isintent_cache  s    r   c                 C   s   d|  dg kS )Ncopyr   r   rh   re   re   rf   r/     s    c                 C   s   d|  dg kS )N	overwriter   r   rh   re   re   rf   r7     s    c                 C   s   d|  dg kS )Ncallbackr   r   rh   re   re   rf   r.     s    c                 C   s   d|  dg kS )Nr   r   r   rh   re   re   rf   r4     s    c                 C   s   d|  dg kS )NZauxr   r   rh   re   re   rf   r,     s    c                 C   s   d|  dg kS )NZaligned4r   r   rh   re   re   rf   isintent_aligned4  s    r   c                 C   s   d|  dg kS )NZaligned8r   r   rh   re   re   rf   isintent_aligned8  s    r   c                 C   s   d|  dg kS )NZ	aligned16r   r   rh   re   re   rf   isintent_aligned16  s    r   Z	INTENT_INZINTENT_INOUTZ
INTENT_OUTZINTENT_HIDEZINTENT_CACHEZINTENT_COPTIONALZINTENT_INPLACEZINTENT_ALIGNED4ZINTENT_ALIGNED8ZINTENT_ALIGNED16c                 C   s   d| kod| d kS )Nr   privatere   rh   re   re   rf   rB     s    c                 C   s   d| kS )N=re   rh   re   re   rf   r     s    c                 C   s   t | sdS | d d dkS )Nr   r   )"')r   rh   re   re   rf   hasinitvalueasstring	  s    r   c                 C   s   d| kS )NZnotere   rh   re   re   rf   r     s    c                 C   sF   t | sdS d| kr| d }n| d }|| d krBt| d | S dS r   )r'   r   r   re   re   rf   r     s    
c                 C   s   d| kS )Ncommonre   r   re   re   rf   r     s    c                 C   s4   t | rdS t| r0| d D ]}t|r dS qdS )Nr   bodyr   )r   r   containscommon)r   bre   re   rf   r   #  s    r   c                 C   s8   t | rdS t| sdS | d D ]}t|r  dS q dS )Nr   r   r   )r?   r   containsmodule)r   r   re   re   rf   r   -  s    r   c                 C   s   d| kS )Nr   re   r   re   re   rf   r   8  s    c                 C   s   t | d k	S rr   )r   r   re   re   rf   r   <  s    c                 C   s   dS )Nr   re   rh   re   re   rf   istrue@  s    r   c                 C   s   dS r   re   rh   re   re   rf   isfalseD  s    r   c                   @   s   e Zd ZdS )	F2PYErrorN)__name__
__module____qualname__re   re   re   rf   r   H  s   r   c                   @   s   e Zd Zdd Zdd ZdS )r\   c                 C   s
   || _ d S rr   )mess)selfr   re   re   rf   __init__N  s    zthrow_error.__init__c                 C   s   d|| j f }t|d S )Nz

  var = %s
  Message: %s
)r   r   )r   ri   r   re   re   rf   __call__Q  s    zthrow_error.__call__N)r   r   r   r   r   re   re   re   rf   r\   L  s   c                  G   sN   dg  }}t t| D ] }d|||f }|d|  qtd|d|f S )Nlambda v%s,f%d=f[%d]f%d(v)%s:%sz and rangelenappendevaljoinfl1l2ire   re   rf   rU   V  s
    
c                  G   sN   dg  }}t t| D ] }d|||f }|d|  qtd|d|f S )Nr   r   r   r   z or r   r   re   re   rf   rW   ^  s
    
c                 C   s   t dS )Nzlambda v,f=f:not f(v))r   )r   re   re   rf   rV   f  s    c                 C   s.   z| d d dkW S  t k
r(   Y dS X d S )Nr   fortranname r   r   r   re   re   rf   r%   j  s    c                 C   sV   z4| d d }|dkrt |s2td| d   t W n t k
rP   | d }Y nX |S )Nr   r   r   z"Failed to use fortranname from %s
r   )r   r	   )r   r   re   re   rf   r   q  s    c                 C   s
  z| d  |}W n tk
r(   Y d S X |s2d S |dkrHt|trHd S t|trj|t|krbd S || }|d d dkr|rd| d t| d |dd   }n|dd  }|dd  dkr|r|d d d	 t| d
 }n|d d }ntd|t|f  |S )Nr   r      z'''z
	/* start z multiline (z) */
z
	/* end multiline (z)*/z-%s multiline block should end with `'''`: %s
)r`   r   
isinstancestrlistr   reprr	   )r   Z	blocknamecommentcounterrre   re   rf   getmultilineblock  s@    


r   c                 C   s
   t | dS )NZcallstatementr   r   re   re   rf   r     s    c           
      C   s  t | ddd}|r|S t| r*td d S ddlm} g g  }}tttt| r`|	ddg | d	 D ]}| d
 | }t
|rqh||kr|| d }nZ||}ttttt|rnt|rnt|s|d }t|st|st|r|d || qhd|| }	|	sd}	|	S )NZcallprotoargumentr   )r   z<warning: callstatement is defined without callprotoargument
r   )getctypezchar*Zsize_tr   r   Z_typedefr   ,void)r   r   rX   Z	capi_mapsr   rU   rJ   rV   r(   extendr.   r-   rW   rE   r    rG   r]   r   rH   r   r   )
r   Zcb_mapr   r   Z	arg_typesZ
arg_types2nri   ctypeZ
proto_argsre   re   rf   r     sH    

c                 C   s
   t | dS )Nusercoder   r   re   re   rf   r     s    c                 C   s   t | dddS )Nr   r   )r   r   r   re   re   rf   r     s    c                 C   s
   t | dS )NZpymethoddefr   r   re   re   rf   r     s    c                 C   sp   g g  }}d| krh| d }d| kr`| d D ]}||kr*| | q*|D ]}||krF| | qFn| d }||fS )Nr   sortvars)r   )r   sortargsr   r   re   re   rf   getargs  s    
r   c                    s   g  dg  }  fddd  D }|   dkr|d D ]}| krF|| qF D ]}||krb|| qbn|d  } |fS )Nr   c                    s(   g | ] }t d  | r| kr|qS )r   )r,   ).0r   r   r   re   rf   
<listcomp>  s     zgetargs2.<locals>.<listcomp>r   r   )r`   keysr   )r   r   Zauxvarsr   re   r   rf   r     s    c                 C   s<   d| krd S d }| d dkr,| d | d f}| d  |d S )NZf2pymultilinesr   zpython moduler   r   )r   kre   re   rf   r     s    c                 C   s*   dt |  d d }d|d | |d f S )NP         z/*%s %s %s*/r   )r   )r   lnre   re   rf   r
     s    c                 C   s$   t | trttfdd| g S | gS )Nc                 S   s   | || S rr   re   )xyr   re   re   rf   <lambda>      zflatlist.<locals>.<lambda>)r   r   r   flatlist)lstre   re   rf   r    s    
r  c                 C   s    | r| d dkr| d d S | S )Nrv   r   re   )sre   re   rf   r[   	  s    r   c              	      s   t  trfdd D S t tr< fddD S dt   D ]v}|dkrZqLd kr|| d kr| d | }n}t  | trd| |t | qLd|  | qLS )Nc                    s   g | ]}t | qS re   rY   r   _m)
defaultsepr   re   rf   r     s     zreplace.<locals>.<listcomp>c                    s   g | ]}t | qS re   r
  r  )r   r  re   rf   r     s     r  separatorsforz#%s#)r   r   r   rY   r   r  )r   r   r  r   sepre   )r   r  r   rf   rY     s    

 c                 C   s2  t |tr"|D ]}t| |} q| S | D ] }|d dkr>q*|| kr t | | trd| | g| |< t | | trt || tr| | ||  | |< n| | ||  ntt | | tr,t || tr,|dkr||  D ]$}|| | kr|| | | | |< qnt| | || | |< q*|| | |< q*| S )Nr   _r  )r   r   r   r   r   r   dict)rdarr   r   k1re   re   rf   r   "  s.    


c                 C   s  i }t | tr>| D ]&}t|||}t||}d|kr q:q|S d| krV| d |sV|S d| krtd| d i||}d|krt|d  |  D ]}|dkr| | ||< qt | | trt| | |||< nt | | tr$g ||< | | D ].}t||i||}	||	kr|| 	|	|  qn2|d dkr8qnt | | t
rBg ||< | |  D ]}
t |
tjr\|
|r\t | | |
 tr| | |
 D ]N}t |t
rtd|i||}d|kr|d }nd	}|| 	t|| qnR| | |
 }t |t
r(td|i|}d|kr$|d }nd	}|| 	t|| q\ntd
t| |   t || trt|| dkr|| d ||< || g kr||= q|S )N_break_checkZneedZneedsr  r   r  Z	supertextr   zapplyrules: ignoring rule %s.
r   )r   r   r   r   r   Zappend_needsr   r   rY   r   r  typesFunctionTyper	   r   r   )rulesr   ri   retr   rrresr   r   r  r  re   re   rf   r   ?  sl    





)r   r   )r   )__doc__pprintra   r  	functoolsr   r   r   r   __all__versionZf2py_versionstderrrc   r	   rZ   r_   rg   r   rX   r   rn   ro   r   r   r   rI   rG   rH   r   r   rE   r    r8   r+   r~   r   r)   r=   rO   rS   rN   rQ   r$   r;   r:   r!   r*   rP   rT   r   rR   r   r   r   rF   r   r   r@   r?   r'   r(   rK   rL   r]   r   r   rD   r9   r>   r<   r"   r#   rJ   r   rM   r   rA   r&   rC   r2   r3   r6   r1   r5   r-   r   r/   r7   r.   r4   r,   r   r   r   r0   rB   r   r   r   r   r   r   r   r   r   r   r   	Exceptionr   r\   rU   rW   rV   r%   r   r   r   r   r   r   r   r   r   r   r
   r  r[   rY   r   r   re   re   re   rf   <module>   s                                                                 
           


'	
