U
    dU                  R   @   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@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXgRZ	ej
ZejjZejZi Zg ZdZdYdT ZdZd Zd[d\ Zd]dD Zd^d_ Zd`dE Zdad Zdbd ZdcdB Zddd Zded5 Zdfd( Zdgdh Zdidj Z dkd: Z!dldK Z"dmdO Z#dndJ Z$dodM Z%dpd" Z&dqd8 Z'drd7 Z(dsd Z)dtd' Z*dudL Z+dvdP Z,dwdx Z-dydN Z.dzd{ Z/d|d} Z0d~d Z1ddC Z2dd Z3dd Z4dd= Z5dd< Z6dd% Z7dd& Z8ddG Z9ddH Z:dd Z;dd Z<ddA Z=dd6 Z>dd; Z?dd9 Z@dd  ZAdd! ZBddF ZCdd ZDddI ZEdd ZFdd> ZGdd$ ZHdd@ ZIdd/ ZJdd0 ZKdd3 ZLdd. ZMdd2 ZNdd* ZOdd ZPdd, ZQdd4 ZRdd+ ZSdd1 ZTdd) ZUdd ZVdd ZWdd ZXeJdeKdeLdeMdePdeOdeGdeTdeVdeWdeXdiZYdd? ZZdd Z[dd Z\dd Z]dd Z^dd Z_dd Z`ddĄ Zadd Zbdd ZcddȄ Zdddʄ ZeG dd̄ defZgG ddX dXZhddQ ZiddS ZjddR Zkdd# Zldd ZmdddԄZndd Zoi fddZpdd Zqdd Zrdd Zsddۄ Ztdd Zudd Zvdd Zwdd ZxddW ZydddUZzdd	 Z{i fddZ|d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	iscomplexiscomplexarrayiscomplexfunctioniscomplexfunction_warnisdoubleisdummyroutine
isexternal
isfunctionisfunction_wrap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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_errorc                 C   s   t ddrtj|  d S )Nverboser   )optionsgetsysstdoutwrite)t r_   7/tmp/pip-unpacked-wheel-c1z6a0q8/numpy/f2py/auxfuncs.pyrS   >   s    c                 C   s   dt kS )NZcapi)debugoptionsvarr_   r_   r`   r   C   s    c                 C   s   d| ko| d dkot |  S )Ntypespec	characterr#   rb   r_   r_   r`   	_isstringG   s    rg   c                 C   s   t | ot|  S N)rg   r   rb   r_   r_   r`   rC   L   s    c                 C   s   t | od| kS )NZcharselectorrC   rb   r_   r_   r`   ischaracterP   s    rj   c                 C   s   t | ot| S rh   )r   rg   rb   r_   r_   r`   rD   T   s    c                 C   s   t | o| d d dkS )N	dimensionz(*))rD   rb   r_   r_   r`   r   X   s    c                 C   s   d| kot |  S Nrk   rf   rb   r_   r_   r`   r   ^   s    c                 C   s   t | pt| pt|  S rh   )r   rC   r#   rb   r_   r_   r`   rA   b   s    c                 C   s   t | o| ddkS Nrd   )complexzdouble complexrA   rZ   rb   r_   r_   r`   r   f   s    c                 C   s   t | o| ddkS )Nrd   logicalrp   rb   r_   r_   r`   r4   k   s    c                 C   s   t | o| ddkS )Nrd   integerrp   rb   r_   r_   r`   r'   o   s    c                 C   s   t | o| ddkS )Nrd   realrp   rb   r_   r_   r`   isreals   s    rt   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KeyErrorrb   r_   r_   r`   get_kindw   s    ry   c                 C   s*   t | sdS | ddkrdS t| dkS )Nr   rd   rr   rq   8rA   rZ   ry   rb   r_   r_   r`   r9      s
    c                 C   s*   t | sdS | ddkrdS t| dkS )Nr   rd   rr   -1r|   rb   r_   r_   r`   rJ      s
    c                 C   s*   t | sdS | ddkrdS t| dkS )Nr   rd   rr   -2r|   rb   r_   r_   r`   rN      s
    c                 C   s*   t | sdS | ddkrdS t| dkS )Nr   rd   rr   -4r|   rb   r_   r_   r`   rI      s
    c                 C   s*   t | sdS | ddkrdS t| dkS )Nr   rd   rr   -8r|   rb   r_   r_   r`   rL      s
    c                 C   s*   t | sdS | ddksdS t| dkS )Nr   rd   rs   r{   r|   rb   r_   r_   r`   r!      s
    c                 C   s*   t | sdS | ddksdS t| dkS )Nr   rd   rs   Z16r|   rb   r_   r_   r`   r7      s
    c                 C   s   t | sdS t| dkS )Nr   Z32)r   ry   rb   r_   r_   r`   r6      s    c                 C   s   t | o| ddkS rn   )r   rZ   rb   r_   r_   r`   r      s    c                 C   s"   t | o | ddko t| dkS )Nrd   rr   1r   rZ   ry   rb   r_   r_   r`   r&      s    
c                 C   s"   t | o | ddko t| dkS )Nrd   rz   r}   r   rb   r_   r_   r`   rK      s    
c                 C   s"   t | o | ddko t| dkS )Nrd   rz   r~   r   rb   r_   r_   r`   rO      s    
c                 C   s"   t | o | ddko t| dkS )Nrd   rz   r   r   rb   r_   r_   r`   isunsignedarray   s    
r   c                 C   s"   t | o | ddko t| dkS )Nrd   rz   r   r   rb   r_   r_   r`   rM      s    
c                 C   s"   t | o | ddko t| dkS )Nrd   rz   r   r   rb   r_   r_   r`   issigned_chararray   s    
r   c                 C   s"   t | o | ddko t| dkS )Nrd   rz   2r   rb   r_   r_   r`   issigned_shortarray   s    
r   c                 C   s"   t | o | ddko t| dkS )Nrd   rz   4r   rb   r_   r_   r`   issigned_array   s    
r   c                 C   s"   t | o | ddko t| dkS )Nrd   rz   r{   r   rb   r_   r_   r`   rB      s    
c                 C   s   d| kod| d kS )NattrspecZallocatabler_   rb   r_   r_   r`   r      s    c                 C   s   d| kpt |  S rm   ri   rb   r_   r_   r`   	ismutable   s    r   c                 C   s   d| kS )N
modulenamer_   routr_   r_   r`   r<      s    c                 C   s   d| kod| d kS )Nblockmoduler_   r   r_   r_   r`   r;      s    c                 C   s   d| kod| d kS )Nr   functionr_   r   r_   r_   r`   r$     s    c                 C   s"   t | rdS to t| o t|  S Nr   )r)   	wrapfuncsr$   r#   r   r_   r_   r`   r%     s    c                 C   s   d| kod| d kS )Nr   Z
subroutiner_   r   r_   r_   r`   rF     s    c                 C   s   t | rdS t| ot| S r   )r)   rF   hasassumedshaper   r_   r_   r`   rG     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varsrk   :FrZ   )r   adr_   r_   r`   r     s    
r   c                 C   s   t | pt| S rh   )r<   r   r   r_   r_   r`   requiresf90wrapper   s    r   c                 C   s   t | pt| S rh   )r$   rF   r   r_   r_   r`   r@   $  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$   r4   r   r   r_   r_   r`   r5   (  s    
c                 C   sF   t | sdS d| kr| d }n| d }|| d krBt| d | S dS r   )r$   r9   r   r_   r_   r`   r:   4  s    
c                 C   sF   t | sdS d| kr| d }n| d }|| d krBt| d | S dS r   )r$   r7   r   r_   r_   r`   r8   @  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   r_   r_   r`   r   L  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   rS   r   r_   r_   r`   r    X  s    c                 C   sF   t | sdS d| kr| d }n| d }|| d krBt| d | S dS r   )r$   rC   r   r_   r_   r`   rE   f  s    
c                 C   s   d| ko| d S )NZ	externalsr_   r   r_   r_   r`   r   r  s    c                 C   s   d| kod| d kS )Nf2pyenhancementsZ
threadsafer_   r   r_   r_   r`   rH   v  s    
c                 C   s   d| ko| d S )Nr   r_   r   r_   r_   r`   hasvariables{  s    r   c                 C   s(   d| ko&d| d ko&d| d ko&t | S )Nr   optionalrequired)r1   rb   r_   r_   r`   r=     s
    
c                 C   s   d| kod| d kS )Nr   Zexternalr_   rb   r_   r_   r`   r#     s    c                 C   s   t |  ot| S rh   )r=   r1   rb   r_   r_   r`   r?     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outinr_   rb   r_   r_   r`   r.     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   r_   rb   r_   r_   r`   r/     s    



c                 C   s   d|  dg kS )Nr   r   r   rb   r_   r_   r`   r2     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   )rR   r/   r0   rb   r_   r_   r`   r-     s    c                 C   s
   t |  S rh   )r-   rb   r_   r_   r`   r1     s    c                 C   s   d|  dg kS )Ncr   r   rb   r_   r_   r`   r)     s    c                 C   s   d|  dg kS )Ncacher   r   rb   r_   r_   r`   isintent_cache  s    r   c                 C   s   d|  dg kS )Ncopyr   r   rb   r_   r_   r`   r+     s    c                 C   s   d|  dg kS )N	overwriter   r   rb   r_   r_   r`   r3     s    c                 C   s   d|  dg kS )Ncallbackr   r   rb   r_   r_   r`   r*     s    c                 C   s   d|  dg kS )Nr   r   r   rb   r_   r_   r`   r0     s    c                 C   s   d|  dg kS )NZauxr   r   rb   r_   r_   r`   r(     s    c                 C   s   d|  dg kS )NZaligned4r   r   rb   r_   r_   r`   isintent_aligned4  s    r   c                 C   s   d|  dg kS )NZaligned8r   r   rb   r_   r_   r`   isintent_aligned8  s    r   c                 C   s   d|  dg kS )NZ	aligned16r   r   rb   r_   r_   r`   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   privater_   rb   r_   r_   r`   r>     s    c                 C   s   d| kS )N=r_   rb   r_   r_   r`   r     s    c                 C   s   t | sdS | d d dkS )Nr   r   )"')r   rb   r_   r_   r`   hasinitvalueasstring  s    r   c                 C   s   d| kS )NZnoter_   rb   r_   r_   r`   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   r_   r_   r`   r     s    
c                 C   s   d| kS )Ncommonr_   r   r_   r_   r`   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   br_   r_   r`   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   r_   r_   r`   r     s    r   c                 C   s   d| kS )Nr   r_   r   r_   r_   r`   r     s    c                 C   s   t | d k	S rh   )r   r   r_   r_   r`   r     s    c                 C   s   dS )Nr   r_   rb   r_   r_   r`   istrue"  s    r   c                 C   s   dS r   r_   rb   r_   r_   r`   isfalse&  s    r   c                   @   s   e Zd ZdS )	F2PYErrorN)__name__
__module____qualname__r_   r_   r_   r`   r   *  s   r   c                   @   s   e Zd Zdd Zdd ZdS )rW   c                 C   s
   || _ d S rh   )mess)selfr   r_   r_   r`   __init__0  s    zthrow_error.__init__c                 C   s   d|| j f }t|d S )Nz

  var = %s
  Message: %s
)r   r   )r   rc   r   r_   r_   r`   __call__3  s    zthrow_error.__call__N)r   r   r   r   r   r_   r_   r_   r`   rW   .  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ll2ir_   r_   r`   rP   8  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   r_   r_   r`   rR   @  s
    
c                 C   s   t dS )Nzlambda v,f=f:not f(v))r   )r   r_   r_   r`   rQ   H  s    c                 C   s.   z| d d dkW S  t k
r(   Y dS X d S )Nr   fortranname r   rw   r   r_   r_   r`   r"   L  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   )rx   r	   )r   r   r_   r_   r`   r   S  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
)rZ   rx   
isinstancestrlistr   reprr	   )r   Z	blocknamecommentcounterrr_   r_   r`   getmultilineblocka  s@    


r   c                 C   s
   t | dS )NZcallstatementr   r   r_   r_   r`   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 }nJ||}ttttt|rnt|rn|d }t|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_typedefru   ,void)r   r   rS   Z	capi_mapsr   rP   rE   rQ   r%   extendr*   r)   rR   rA   r   rC   r   r   r   )
r   Zcb_mapr   r   Z	arg_typesZ
arg_types2nrc   ctypeZ
proto_argsr_   r_   r`   r     s>    

c                 C   s
   t | dS )Nusercoder   r   r_   r_   r`   r     s    c                 C   s   t | dddS )Nr   r   )r   r   r   r_   r_   r`   r     s    c                 C   s
   t | dS )NZpymethoddefr   r   r_   r_   r`   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   r_   r_   r`   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   r_   r`   
<listcomp>  s     zgetargs2.<locals>.<listcomp>r   r   )rZ   keysr   )r   r   Zauxvarsr   r_   r   r`   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   kr_   r_   r`   r     s    c                 C   s*   dt |  d d }d|d | |d f S )NP         z/*%s %s %s*/ru   )r   )r   r   r_   r_   r`   r
     s    c                 C   s$   t | trttfdd| g S | gS )Nc                 S   s   | || S rh   r_   )xyr   r_   r_   r`   <lambda>      zflatlist.<locals>.<lambda>)r   r   r   flatlist)r   r_   r_   r`   r     s    
r   c                 C   s    | r| d dkr| d d S | S )Nrl   r   r_   )sr_   r_   r`   rV     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 r_   rT   r   _m)
defaultsepr   r_   r`   r     s     zreplace.<locals>.<listcomp>c                    s   g | ]}t | qS r_   r   r   )r   r   r_   r`   r     s     r   separatorsforz#%s#)r   r   r   rT   r   r   )r   r   r   r   sepr_   )r   r   r   r`   rT     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   k1r_   r_   r`   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   rT   r   r  typesFunctionTyper	   r   r   )rulesr   rc   retr   rrresr   r   r  r  r_   r_   r`   r     sl    





)r   r   )r   )}__doc__pprintr[   r	  	functoolsr   r   r   r   __all__versionZf2py_versionstderrr]   r	   rU   rY   ra   r   rS   r   rg   rC   rj   rD   r   r   rA   r   r4   r'   rt   ry   r9   rJ   rN   rI   rL   r!   r7   r6   r   r&   rK   rO   r   rM   r   r   r   rB   r   r   r<   r;   r$   r%   rF   rG   r   r   r@   r5   r:   r8   r   r    rE   r   rH   r   r=   r#   r?   r.   r/   r2   r-   r1   r)   r   r+   r3   r*   r0   r(   r   r   r   r,   r>   r   r   r   r   r   r   r   r   r   r   r   	Exceptionr   rW   rP   rR   rQ   r"   r   r   r   r   r   r   r   r   r   r   r
   r   rV   rT   r   r   r_   r_   r_   r`   <module>   s                                                              
           


$	
