a
    p=icz                     @   sf  d Z ddlmZ ej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T g dZd	Zg Zi ai Zd
d
d
ddddddddddddddZdddddddddddddddddZerdddddddddddddd ddddd!Zd"d#d"d$d$d%d&d'd(d)d*d+d,d-d-d.d/Zer6d"d#d0d%d1d%d2d3d(d4d*d+d5d6d,d-d7d8d!Zd"d#d%d%d2d(d*d+d9d9d9d:d;Zd
d
d<d=d=d>dd?d@ddAdBdCdDdEdF	dGdGdHdIdIdJdGdGdHdIdId>ddKd@ddAdLdMdHidMd<idMdKidMdidN	aetZg ZdOdP ZdQdQdRdSdSdTdUdVdSdWdXdYdXdZdZd[d\d]Zd^d_ Zd`da ZdvdbdcZddde Zdfdg Zdhdi Z djdk Z!dldm Z"dndo Z#dwdpdqZ$drds Z%dtdu Z&dS )xa)  

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 License.

NO WARRANTY IS EXPRESSED OR IMPLIED.  USE AT YOUR OWN RISK.
$Date: 2005/05/06 10:57:33 $
Pearu Peterson

   )__version__    N)markoutercomma)cb_rules)*)getctypegetstrlength
getarrdimsgetpydocsigngetarrdocsigngetinitsign2maproutsign2mapmodsign2mapcb_sign2mapcb_routsign2mapcommon_sign2mapTfloatintlongcomplexstring)doubler   long_doublecharsigned_charunsigned_charshortunsigned_shortr   r   	long_longunsignedcomplex_floatcomplex_doublecomplex_long_doubler   Z
NPY_DOUBLEZ	NPY_FLOATZ
NPY_STRINGZ	NPY_UBYTEZNPY_BYTEZ	NPY_SHORTZ
NPY_USHORTZNPY_INTZNPY_UINTZNPY_LONGZ
NPY_CFLOATZNPY_CDOUBLE)r   r   r   r   r   r   r   r   r   r    r   r   r!   r"   r#   r   ZNPY_LONGDOUBLEZ	NPY_ULONGZNPY_LONGLONGZNPY_ULONGLONG)r   r   r   r   r   r   r   r   r   r    r   unsigned_longr   unsigned_long_longr!   r"   r#   r   df1bswiulLFDc)r   r   r   r   r   r   r   r   r   r    r   r   r!   r"   r#   r   gBhHIqQGSNy)r   r   r   r   r   r   r   r   r!   r"   r#   r   r   r   ) 48Z1216r   r   r   r   r   r    r%   )	r>   r(   2r?   r@   -1z-2z-4z-8r!   r"   r#   )r>   r@   rA   Z24Z32r   )r>   r(   rB   r?   r@   r>   )	realintegerr   ZcomplexkindZlogicalzdouble complexzdouble precisionbyte	characterc                 C   s  t ta| d u r&d} tj| s&d S z|td|  t	| d$}t
|  i i }W d    n1 sl0    Y  | D ]6\}}| D ]}|| || < q|| || < q~| D ]}|tvri t|< ||  D ]}|| | tv rh|t| v r&td||t| | || | f  || | t| |< td|||| | f  t|| |  qtd|||| | || | tt f  qqtd W n2 ty } ztd|  W Y d }~n
d }~0 0 d S )	Nz.f2py_f2cmapzReading f2cmap from {!r} ...
rz3	Warning: redefinition of {'%s':{'%s':'%s'->'%s'}}
z	Mapping "%s(kind=%s)" to "%s"
z5	Ignoring map {'%s':{'%s':'%s'}}: '%s' must be in %s
z1Successfully applied user defined f2cmap changes
z;Failed to apply user defined f2cmap changes: %s. Skipping.
)copydeepcopyf2cmap_default
f2cmap_allospathisfileoutmessformatopenevalreadloweritemskeysc2py_mapf2cmap_mappedappenderrmesslist	Exception)Zf2cmap_filer'   r&   kd1k1msg rb   e/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/numpy/f2py/capi_maps.pyload_f2cmap_file   sF    
2"rd   z%gz%Lgz%dz%hhuz%hdz%huz%uz%ldz%luz(%g,%g)z	(%Lg,%Lg)%s)r   r   r   r   r   r   r   r   r   r    r   r$   r   r!   r"   r#   r   c                 C   s  d}t | rRd| v r| d }n| d }|| d v rBt| d | S td|  nxt| r^|S d| v r| d  tv r| d  }t| }|d }d| v rd	| d v rz|| d d	  }W n0 ty   td
| d d	| d d	 f  Y n0 nd| d v r|d tv rt|d  }z|| d d  }W n ty   |tv rRt| }z|t| d d  }W n@ ty   td|| d d ||| d d t	 f  Y n0 Y n0 nt
| std|   |S )z
    Determines C type
    voidresultnamevars,getctype: function %s has no return value?!
typespecr>   Zkindselectorr   z$getctype: "%s %s %s" not supported.
kindz}getctype: "%s(kind=%s)" is mapped to C "%s" (to override define dict(%s = dict(%s="<C typespec>")) in %s/.f2py_f2cmap file).
z2getctype: No C-type found in "%s", assuming void.
)
isfunctionr   r[   ZissubroutinerU   rL   KeyErrorstrrM   getcwd
isexternal)varctypeark   Zf2cmaprb   rb   rc   r      sR    



r   c                 C   s   t | rJd| v r| d }n| d }|| d v r>t| d | S td|  t| sbtdt|   d}d| v r| d }d|v r|d }nd	|v r|d	 }td
|std|rt| rtdt|   d}|S )Nrg   rh   ri   z0getstrlength: function %s has no return value?!
z;getstrlength: expected a signature of a string but got: %s
r(   Zcharselectorr   lenz\(\s*(\*|:)\s*\)z(\*|:)zMgetstrlength:intent(hide): expected a string with defined length but got: %s
rC   )isstringfunctionr   r[   isstringreprrematchisintent_hide)rr   rt   ru   rb   rb   rc   r      s2    


r   c           	      C   s  i }t |r8t|s8t||d< |d |d< d|d< nJt|r\d|d< d|d< d|d< n&t|rt|d }d||d< ztt|d |d< W n ty   Y n0 d	||d< tt	||d< tt	|d
g dd
 |d< t
t	|D ]|}g }|| tv r|| g}n,tD ]&}td| || r|| q|D ](}t|t| krHd||<  qqHqd\|d< }|D ].}|d }|dvrd|d ||f |d< q|d r|d d d
 |d< d\|d< }|d D ]|}|d }|dvrd|d ||f |d< nLt|rBtd|  d|d |df |d< n|rtdt| t|f  q|d r|d d d
 |d< |S )Ndimssizer(   rank0r>   	dimensionr   ,r   z	rank*[-1]z.*?\b%s\b.*)r>   r   Zsetdims)r   :z(*)z(:)z%s#varname#_Dims[%d]=%s,Z	cbsetdimsz?getarrdims:warning: assumed shape array, using 0 instead of %r
r   z]getarrdims: If in call-back function: array argument %s must have bounded dimensions: got %s
)rw   isarrayr   isscalarrI   joinrx   rS   r]   ru   rangedepargsry   rz   rZ   indexisintent_inrP   r[   )	rt   rr   verboseretdimr,   vvar&   rb   rb   rc   r	   <  sx    









r	   c                 C   s  t |rPd|v r|d }n|d }||d v r@t||d | S td|  dS | |  }}d}t|rld}nt|rxd}| }t|r|d	 D ]$}|d d
 dkr|d
d  } qqd}t|}	t|rt| |\}}
d|
 }t	|r4t|rd| |t
|	 t|	 |f }nd| |t
|	 |f }d|t
|	 f }nNt|rt|r^d| |t||f }nd| |t||f }d|t|f }nt|r|d }tt|}d| ||t|	 d||f }| |krd| |t|	 d|f }nd||t|	 d|| f }n~t|rvd}| tv rdt|  tv rddtt|   v rdtt|   d }|| ks`d| }nd}d| |f }|}ntd|   ||fS )Nrg   rh   ri   rj   r>   r>   r>   inputz	in/outputintent   out=z, optional\n    Default: %sz%s : %s rank-0 array(%s,'%s')%sz%s : %s %s%sz%s : %sz*%s : %s rank-0 array(string(len=%s),'c')%sz%s : %s string(len=%s)%sz%s : string(len=%s)r   z.%s : %s rank-%s array('%s') with bounds (%s)%sr   )%s : rank-%s array('%s') with bounds (%s)z8%s : rank-%s array('%s') with bounds (%s) and %s storageargnamez => %sz%s : call-back function%sz7getpydocsign: Could not resolve docsignature for "%s".
)rm   r
   r[   r   isintent_inoutisintent_outr   hasinitvaluer   r   rX   c2pycode_maprw   r   r   rx   ru   r   rq   lcb_maplcb2_map)rt   rr   afsigZsigoutoptout_ar^   initrs   showinitr   r~   Zuarb   rb   rc   r
   y  s    









*

r
   c                 C   s   t |}t|r*t|s*d| t|f }nVt|rJd| t| t| f }n6t|r|d }tt|}d| |t| d	|f }|S )Nz%%s : rank-0 array(string(len=%s),'c')z%s : rank-0 array(%s,'%s')r   r   r   )
r   rw   r   r   r   rX   r   rx   ru   r   )rt   rr   rs   r   r   r~   rb   rb   rc   r     s&    
r   c                 C   sL  t |rd\}}nd\}}t|rD|d }|}t|s@t|ri }zb|d }d|v rzt|dd d\|d< |d	< n*t|i i }t|jt|j	 |d< |d	< W n" t
y   td
|| f Y n0 t|rd|d |d	 f }n\t |rD|s d\}}|d dkr&d|dd dd }|d dkrDd|dd  }||fS )N)z""z''r   =r   r   r   @,@init.rinit.izMgetinit: expected complex number `(r,i)' but got `%s' as initial value of %r.z (capi_c.r=%s,capi_c.i=%s,capi_c)r   'z"%s""z\"z'%s')rw   r   	iscomplexiscomplexarrayr   splitrS   ro   rD   imagr]   
ValueErrorr   replace)rt   rr   r   r   r   r   rb   rb   rc   r     sD    


"


r   c              "   C   s<  | }t |r:|d D ]$}|dd dkr|dd } q:q| |t|d}g }t D ]\}}||rV|d|  qV|rd||d< nd|d< t|rd	|d
< n&|d tv rt|d  |d
< nd|d
< t| |\|d< |d< t	|r$t
|r$t|s$t|d dd d\|d< |d< t|r| |d< | tv rt|  |d< tt|   d |d< tt|   d |d< tt|   d |d< tt|   d |d< n | |d< td| tt f  t|rt||d< t|rt|t| |}t|d }|d tv rt|d  |d< t|rtd t d!td"td#td$td%t
d&tt t!t
d'td(td)t"d*t#d+t$d,tt%t!t$d-td.t&d.t'd/g"}	g }
t(d0t)|	d1D ]&}|	| |r|
|	|d   qt|r|
d2| |d f  t|rd3t*d4d5 |d |}|
d6|  t|r6d7| |d d3|
f |d8< n"d9|d | |d d3|
f |d8< t |r|d t+v rd:| t+|d  f |d;< t|rd<| | f |d;< t|rd=|  |d;< |d t+v rd>| t+|d  f |d?< d@t+|d   |dA< t|rdB| | f |d?< t,| |\|dC< |dD< t-|r8|dE |dE< |S )Fz
    varname,ctype,atype
    init,init.r,init.i,pytype
    vardebuginfo,vardebugshowvalue,varshowvalue
    varrformat

    intent
    r   Nr   r   )varnameZ
outvarnamers   zF2PY_%s|ZF2PY_INTENT_INr<   Z
varrformatrs   Or   r   r   r   r   r   r   Z	cbnamekeyZcbname
maxnofargs
nofoptargsZdocstrZcbdocstrZlatexdocstrZcblatexdocstrz5sign2map: Confused: external %s is not in lcb_map%s.
lengthr   atyper   outputZinoutputrequiredoptionalhiddenzcomplex scalarZscalarr   arrayzcomplex arrayzstring arrayzcomplex functionfunctioncallbackZ	auxiliaryr      zslen(%s)=%sr   c                 S   s   d| |f S )Nz%s|%srb   )xr=   rb   rb   rc   <lambda>E      zsign2map.<locals>.<lambda>zdims(%s)zdebug-capi:%s=>%s:%sZvardebuginfozdebug-capi:%s %s=%s:%sdebug-capi:%s=%sZvardebugshowvalue"debug-capi:slen(%s)=%%d %s=\"%%s\"zdebug-capi:%s=%%pz#name#:%s=%sZvarshowvaluere   showvalueformatz#name#:slen(%s)=%%d %s=\"%%s\"	pydocsignpydocsignoutnote).r   r   Zisintent_dictrV   rZ   r   r   c2buildvalue_mapr   r   r   r   r   rq   r   r   r[   r\   rW   rw   r   
dictappendr	   rI   
c2capi_map	debugcapir   r   Z
isrequired
isoptionalr{   Zl_andr   Zl_notr   isstringarrayiscomplexfunctionrm   isintent_callbackZisintent_auxr   ru   mapcformat_mapr
   hasnote)rt   rr   r   r^   r   Zintent_flagsr'   r*   r   ilZrlr,   Zddimrb   rb   rc   r     s    	
















r   c           	      C   s  | d }t | }||dd| | t|td| || t| pJdt| pTdt| p^dd}d|v rvd|d< nd|d< d|v rd	|d
< nd
|d
< i ad| v r:| d 	 D ]}|t
jv rt
j| D ]j}|d }d| d | v r(| d | d 	 D ]*}| d | d | |d kr|} q(q|d t|< qqn0d| v rj| d rjtd|d t| d f  t| tpxd|d< t| rd| v r| d }n| d }||d< t|| \|d< |d< t| d | |d< t| r| d | d |d< dg| d | d< |d tv r&t|d  |d< nd|d< tdt|d   t| r|d tv rrd|t|d  f |d< t| rd ||f |d< t| rt| d | |d!< |d! d"krtd#t| d   d$|d!< t| r| d |d< dg| d< |S )%zV
    name,NAME,begintitle,endtitle
    rname,ctype,rformat
    routdebugshowvalue
    rh   _\_	end of %sr>   )rh   ZtexnameZ
name_lowerNAME
begintitleendtitleZfortrannameZFORTRANNAMEZcallstatementusercode	usercode1	F_FUNC_USF_FUNCZF_WRAPPEDFUNC_USZF_WRAPPEDFUNCZuser   r   r   Z	externalszMroutsign2map: Confused: function %s has externals %s but no "use" statement.
Zcallprotoargumentrg   rnamer   r   ri   rs   r   Z
resultnoteSee elsewhere.Zrformatr   z.routsign2map: no c2buildvalue key for type %s
r   Zroutdebugshowvaluer   ZrlengthrC   zzroutsign2map: expected explicit specification of the length of the string returned by the fortran function %s; taking 10.
10)Zgetfortrannamer   rU   uppergentitleZgetcallstatementgetusercodegetusercode1r   rW   r   Zcb_mapr[   rx   Zgetcallprotoargumentrm   r
   r   Zhasresultnoter   r   r   rv   r   r   )	routrh   fnamer   r-   unlnr^   rt   rb   rb   rc   r   a  s    





















r   c                 C   s   t | r.| d | d  | d ddd}n$| d | d  | d ddd}t| p\g |d< t| rv| d |d< t| pd|d	< t| pd|d
< | d rt| d d pd|d< nd|d< t| pd|d< d| v r| d |d< d| v r| d |d< |S )z
    modulename
    rh   r   r   )Zf90modulenameZF90MODULENAMEZtexf90modulename)
modulenameZ
MODULENAMEZtexmodulenameZrestdocr   r>   r   r   bodyr   Zinterface_usercodeZpymethoddefZcoutputZf2py_wrapper_output)ismoduler   r   Z
getrestdocr   r   r   Zgetpymethoddef)mr   rb   rb   rc   r     s.    

r   c                 C   s   d| i}|d |d< t ||d< |d tv r<t|d  |d< |d tv r\dt|d   |d< t|rtt|t| |}t| |\|d< |d< t|r|d	 |d	< d
g|d	< |S )Nr   Z	varname_irs   r   re   r   r   r   r   r   )r   r   r   r   r   r	   r
   r   )rt   rr   r   r   rb   rb   rc   r     s    
r   c                 C   s  d| d |f dd}t | r`d| d v r0d}nd}d|| d  | d  f |d	< d
|d< n|d |d	< d|d< | d |d< t|d |d< td|d  |d< t| |d< d|d< |d dkrd|d< n|d |d< |d dkrt| rd|d< nd|d< |d tv r$dt|d   |d< t| r:t| |d< t	| rd| v rX| d }n| d }t
| d | r| d | d |d< dg| d | d< ||d< t|| \|d< |d < t| rd!|d< n t
| r| d |d< dg| d< d"}d"}d#| v rLd| v rL| d# D ]<}| d | }ttt|r|d$ }t|r|d$ }qt||d%< t||d&< t
| rt	| rd| v r| d |d'< dg| d< |S )(z\
    name,begintitle,endtitle,argname
    ctype,rctype,maxnofargs,nofoptargs,returncptr
    zcb_%s_in_%srh   r>   )rh   
returncptrr   r   r   z	%s(%s,%s)ZcallbacknameZexternZstaticr   r   r   r   rs   rf   Zrctyper   z3
#ifdef F2PY_CB_RETURNCOMPLEX
return_value=
#endif
r   zreturn_value=re   r   Z	strlengthrg   ri   r   r   r   r   r   z8
#ifdef F2PY_CB_RETURNCOMPLEX
#ctype#
#else
void
#endif
r   argsr   r   r   Zroutnote)r   rU   r   r   r   r   r   rv   r   rm   r   r
   Zl_orr   r   r   rx   )r   Zumr   r   rt   Znofargsr   rr   rb   rb   rc   r     sz    














r   c                 C   s   | t |d}t|rd|d< |d tv r:t|d  |d< |d tv rZdt|d   |d< t|rtt|t| |}nt|rt||d< d|d	< t	| |\|d
< |d< t
|r|d |d< dg|d< t| ||d< |S )N)r   rs   r   rs   r   re   r   r}   r(   r~   r   r   r   r   Z	arrdocstr)r   r   r   r   r   r   r	   rw   r   r
   r   r   )rt   rr   r   rb   rb   rc   r   2  s$    
r   )r   )N)'__doc__r>   r   versionZf2py_versionrI   ry   rM   Zcrackfortranr   r   Zauxfuncs__all__Zusing_newcorer   r   r   rX   r   r   r   rL   rJ   rK   rY   rd   r   r   r   r	   r
   r   r   r   r   r   r   r   r   rb   rb   rb   rc   <module>   s\  
,/
=J#iQ
R