a
    cDaU                     @   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 g dZ	ej
ZejjZejZi Zg ZdZdd	 Zd
d Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zd d! Zd"d# Z d$d% Z!d&d' Z"d(d) Z#d*d+ Z$d,d- Z%d.d/ Z&d0d1 Z'd2d3 Z(d4d5 Z)d6d7 Z*d8d9 Z+d:d; Z,d<d= Z-d>d? Z.d@dA Z/dBdC Z0dDdE Z1dFdG Z2dHdI Z3dJdK Z4dLdM Z5dNdO Z6dPdQ Z7dRdS Z8dTdU Z9dVdW Z:dXdY Z;dZd[ Z<d\d] Z=d^d_ Z>d`da Z?dbdc Z@ddde ZAdfdg ZBdhdi ZCdjdk ZDdldm ZEdndo ZFdpdq ZGdrds ZHdtdu ZIdvdw ZJdxdy ZKdzd{ ZLd|d} ZMd~d ZNdd ZOdd ZPdd ZQdd ZRdd ZSdd 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 dd dZhdd Zidd Zjdd Zkdd ZlddĄ ZmdddƄZnddȄ Zoi fddʄZpdd̄ Zqdd΄ ZrddЄ Zsdd҄ ZtddԄ Zuddք Zvdd؄ Zwddڄ Zxdd܄ Zyddd߄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)R
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_   c/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/numpy/f2py/auxfuncs.pyrS   >   s    rS   c                 C   s   dt v S )NZcapi)debugoptionsvarr_   r_   r`   r   C   s    r   c                 C   s   d| v 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    rC   c                 C   s   t | od| v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    rD   c                 C   s   t | o| d d dkS )N	dimensionz(*))rD   rb   r_   r_   r`   r   X   s    r   c                 C   s   d| v ot |  S Nrk   rf   rb   r_   r_   r`   r   ^   s    r   c                 C   s   t | pt| pt|  S rh   )r   rC   r#   rb   r_   r_   r`   rA   b   s    rA   c                 C   s   t | o| ddv S Nrd   )complexzdouble complexrA   rZ   rb   r_   r_   r`   r   f   s    r   c                 C   s   t | o| ddkS )Nrd   logicalrp   rb   r_   r_   r`   r4   k   s    r4   c                 C   s   t | o| ddkS )Nrd   integerrp   rb   r_   r_   r`   r'   o   s    r'   c                 C   s   t | o| ddkS )Nrd   realrp   rb   r_   r_   r`   isreals   s    rt   c                 C   sL   z| d d W S  t yF   z| d d W  Y S  t y@   Y n0 Y n0 d S )NZkindselector*kindKeyErrorrb   r_   r_   r`   get_kindw   s    ry   c                 C   s*   t | sdS | ddvrdS t| dkS )Nr   rd   rr   rq   8rA   rZ   ry   rb   r_   r_   r`   r9      s
    r9   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
    rJ   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
    rN   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
    rI   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
    rL   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
    r!   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
    r7   c                 C   s   t | sdS t| dkS )Nr   Z32)r   ry   rb   r_   r_   r`   r6      s    r6   c                 C   s   t | o| ddv S rn   )r   rZ   rb   r_   r_   r`   r      s    r   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    
r&   c                 C   s"   t | o | ddv o t| dkS )Nrd   rz   r}   r   rb   r_   r_   r`   rK      s    
rK   c                 C   s"   t | o | ddv o t| dkS )Nrd   rz   r~   r   rb   r_   r_   r`   rO      s    
rO   c                 C   s"   t | o | ddv o t| dkS )Nrd   rz   r   r   rb   r_   r_   r`   isunsignedarray   s    
r   c                 C   s"   t | o | ddv o t| dkS )Nrd   rz   r   r   rb   r_   r_   r`   rM      s    
rM   c                 C   s"   t | o | ddv o t| dkS )Nrd   rz   r   r   rb   r_   r_   r`   issigned_chararray   s    
r   c                 C   s"   t | o | ddv o t| dkS )Nrd   rz   2r   rb   r_   r_   r`   issigned_shortarray   s    
r   c                 C   s"   t | o | ddv o t| dkS )Nrd   rz   4r   rb   r_   r_   r`   issigned_array   s    
r   c                 C   s"   t | o | ddv o t| dkS )Nrd   rz   r{   r   rb   r_   r_   r`   rB      s    
rB   c                 C   s   d| v od| d v S )NattrspecZallocatabler_   rb   r_   r_   r`   r      s    r   c                 C   s   d| vpt |  S rm   ri   rb   r_   r_   r`   	ismutable   s    r   c                 C   s   d| v S )N
modulenamer_   routr_   r_   r`   r<      s    r<   c                 C   s   d| v od| d kS )Nblockmoduler_   r   r_   r_   r`   r;      s    r;   c                 C   s   d| v od| d kS )Nr   functionr_   r   r_   r_   r`   r$     s    r$   c                 C   s"   t | rdS to t| o t|  S Nr   )r)   	wrapfuncsr$   r#   r   r_   r_   r`   r%     s    r%   c                 C   s   d| v od| d kS )Nr   Z
subroutiner_   r   r_   r_   r`   rF     s    rF   c                 C   s   t | rdS t| ot| S r   )r)   rF   hasassumedshaper   r_   r_   r`   rG     s    rG   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    r@   c                 C   sF   t | sdS d| v r| d }n| d }|| d v rBt| d | S dS Nr   resultnamer   )r$   r4   r   r   r_   r_   r`   r5   (  s    
r5   c                 C   sF   t | sdS d| v r| d }n| d }|| d v rBt| d | S dS r   )r$   r9   r   r_   r_   r`   r:   4  s    
r:   c                 C   sF   t | sdS d| v r| d }n| d }|| d v rBt| d | S dS r   )r$   r7   r   r_   r_   r`   r8   @  s    
r8   c                 C   sF   t | sdS d| v r| d }n| d }|| d v rBt| d | S dS r   )r$   r   r   r_   r_   r`   r   L  s    
r   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    r    c                 C   sF   t | sdS d| v r| d }n| d }|| d v rBt| d | S dS r   )r$   rC   r   r_   r_   r`   rE   f  s    
rE   c                 C   s   d| v o| d S )NZ	externalsr_   r   r_   r_   r`   r   r  s    r   c                 C   s   d| v od| d v S )Nf2pyenhancementsZ
threadsafer_   r   r_   r_   r`   rH   v  s    
rH   c                 C   s   d| v o| d S )Nr   r_   r   r_   r_   r`   hasvariables{  s    r   c                 C   s(   d| v o&d| d v o&d| d vo&t | S )Nr   optionalrequired)r1   rb   r_   r_   r`   r=     s
    
r=   c                 C   s   d| v od| d v S )Nr   Zexternalr_   rb   r_   r_   r`   r#     s    r#   c                 C   s   t |  ot| S rh   )r=   r1   rb   r_   r_   r`   r?     s    r?   c                 C   sp   d| vrdS d| d v rdS d| d v r,dS d| d v r<dS d| d v rLdS d| d v r\dS d	| d v rldS dS )
Nintentr   hider   inplaceinoutinoutoutinr_   rb   r_   r_   r`   r.     s    r.   c                 C   sD   d| v oBd| d v s d| d v oBd| d voBd| d voBd| d vS )Nr   r   r   r   r   r   r_   rb   r_   r_   r`   r/     s    



r/   c                 C   s   d|  dg v S )Nr   r   r   rb   r_   r_   r`   r2     s    r2   c                 C   s<   d| v o:d| d v p:d| d v o:d| d vo:t tt|  S )Nr   r   r   r   )rR   r/   r0   rb   r_   r_   r`   r-     s    r-   c                 C   s
   t |  S rh   )r-   rb   r_   r_   r`   r1     s    r1   c                 C   s   d|  dg v S )Ncr   r   rb   r_   r_   r`   r)     s    r)   c                 C   s   d|  dg v S )Ncacher   r   rb   r_   r_   r`   isintent_cache  s    r   c                 C   s   d|  dg v S )Ncopyr   r   rb   r_   r_   r`   r+     s    r+   c                 C   s   d|  dg v S )N	overwriter   r   rb   r_   r_   r`   r3     s    r3   c                 C   s   d|  dg v S )Ncallbackr   r   rb   r_   r_   r`   r*     s    r*   c                 C   s   d|  dg v S )Nr   r   r   rb   r_   r_   r`   r0     s    r0   c                 C   s   d|  dg v S )NZauxr   r   rb   r_   r_   r`   r(     s    r(   c                 C   s   d|  dg v S )NZaligned4r   r   rb   r_   r_   r`   isintent_aligned4  s    r   c                 C   s   d|  dg v S )NZaligned8r   r   rb   r_   r_   r`   isintent_aligned8  s    r   c                 C   s   d|  dg v S )NZ	aligned16r   r   rb   r_   r_   r`   isintent_aligned16  s    r   Z	INTENT_INZINTENT_INOUTZ
INTENT_OUTZINTENT_HIDEZINTENT_CACHEZINTENT_CZOPTIONALZINTENT_INPLACEZINTENT_ALIGNED4ZINTENT_ALIGNED8ZINTENT_ALIGNED16c                 C   s   d| v od| d v S )Nr   privater_   rb   r_   r_   r`   r>     s    r>   c                 C   s   d| v S )N=r_   rb   r_   r_   r`   r     s    r   c                 C   s   t | sdS | d d dv S )Nr   r   )"')r   rb   r_   r_   r`   hasinitvalueasstring  s    r   c                 C   s   d| v S )NZnoter_   rb   r_   r_   r`   r     s    r   c                 C   sF   t | sdS d| v r| d }n| d }|| d v rBt| d | S dS r   )r$   r   r   r_   r_   r`   r     s    
r   c                 C   s   d| v S )Ncommonr_   r   r_   r_   r`   r     s    r   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| v S )Nr   r_   r   r_   r_   r`   r     s    r   c                 C   s   t | d uS rh   )r   r   r_   r_   r`   r     s    r   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   rW   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
    
rP   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
    
rR   c                 C   s   t dS )Nzlambda v,f=f:not f(v))r   )r   r_   r_   r`   rQ   H  s    rQ   c                 C   s,   z| d d dkW S  t y&   Y dS 0 d S )Nr   fortranname r   rw   r   r_   r_   r`   r"   L  s    r"   c                 C   sT   z4| d d }|dkrt |s2td| d   t W n t yN   | d }Y n0 |S )Nr   r   r   z"Failed to use fortranname from %s
r   )rx   r	   )r   r   r_   r_   r`   r   S  s    r   c                 C   s  z| d  |}W n ty&   Y d S 0 |s0d S |dkrFt|trFd S t|trh|t|kr`d 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    r   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||v 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>    

r   c                 C   s
   t | dS )Nusercoder   r   r_   r_   r`   r     s    r   c                 C   s   t | dddS )Nr   r   )r   r   r   r_   r_   r`   r     s    r   c                 C   s
   t | dS )NZpymethoddefr   r   r_   r_   r`   r     s    r   c                 C   sp   g g  }}d| v rh| d }d| v r`| d D ]}||v r*| | q*|D ]}||v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v r|d D ]}| v rF|| qF D ]}||vrb|| qbn|d  } |fS )Nr   c                    s(   g | ] }t d  | r| v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    r   c                 C   s<   d| v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    r   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    r
   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    rV   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 v r|| d v 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     r   zreplace.<locals>.<listcomp>c                    s   g | ]}t | qS r_   r   r   )r   r   r_   r`   r     r   r   separatorsforz#%s#)r   r   r   rT   r   r   )r   r   r   r   sepr_   )r   r   r   r`   rT     s    

 rT   c                 C   s2  t |tr"|D ]}t| |} q| S | D ] }|d dkr>q*|| v r t | | trd| | g| |< t | | trt || tr| | ||  | |< n| | ||  ntt | | tr,t || tr,|dkr||  D ]$}|| | vr|| | | | |< qnt| | || | |< q*|| | |< q*| S )Nr   _r   )r   r   r   r   r   r   dict)rdarr   r   k1r_   r_   r`   r     s.    


r   c                 C   s  i }t | tr>| D ]&}t|||}t||}d|v r q:q|S d| v rV| d |sV|S d| v rtd| d i||}d|v rt|d  |  D ]}|dkr| | ||< qt | | trt| | |||< nt | | tr$g ||< | | D ].}t||i||}	||	v 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|v r|d }nd	}|| 	t|| qnR| | |
 }t |t
r(td|i|}d|v 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   )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   



$	
