o
    Jez                     @   sP  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mZ ddlT g d	Zg Zi ai Zi 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Zi 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/Zi d
d0dd1dd2dd3dd4dd3dd5dd6dd7dd8dd9d'd:dd;d*d<dd=dd>dd?d@dAd/Zd0d1d3d3d5d7d9d:dBdBdBdCdAdDZddd
dddEddddddddd*dF	ddddddGddddddEddddddHdIdidId
idIdidJaeteaetZg ZdKdL Zi d
dMddMddNddOddOddPddQddRddOddSddTd'dUddTddVddVddWddXddYiZdZd[ Zd\d] Zd^d_ Zdvd`daZ dbdc 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)iso_c_binding_map)*)getctypegetstrlength
getarrdimsgetpydocsigngetarrdocsigngetinitsign2maproutsign2mapmodsign2mapcb_sign2mapcb_routsign2mapcommon_sign2mapdoublefloatZlong_doublecharintsigned_charZunsigned_charshortZunsigned_shortlong	long_longZunsignedcomplex_floatcomplexcomplex_doublecomplex_long_doublestring	characterbytesZ
NPY_DOUBLEZ	NPY_FLOATZNPY_LONGDOUBLEZNPY_BYTEZ	NPY_UBYTEZ	NPY_SHORTZ
NPY_USHORTZNPY_INTZNPY_UINTZNPY_LONGZunsigned_longZ	NPY_ULONGZNPY_LONGLONGZunsigned_long_longZNPY_ULONGLONGZ
NPY_CFLOATZNPY_CDOUBLEZ
NPY_STRING)r    r!   dfgbBhHiIlLqQFDGScNy)r   r   r   r   r   r   r   r   r   r   r   r    r!   ) 48Z1216)	r7   12r8   r9   -1z-2z-4z-8)r7   r9   r:   Z24Z32)r7   r;   r<   r8   r9   r7   )realintegerr   ZcomplexkindZlogicalzdouble complexzdouble precisionbytec                 C   s  t ta| d u rd} tj| sd S ztd|  t	| }t
|  i i }W d    n1 s4w   Y  | D ]\}}| D ]
}|| || < qE|| || < q=| D ]k}|tvrgi t|< ||  D ]Z}|| | tv r|t| v rtd||t| | || | f  || | t| |< td|||| | f  t|| |  qmtd|||| | || | tt f  qmq]td W d S  ty } ztd|  W Y d }~d S d }~ww )Nz.f2py_f2cmapzReading f2cmap from {!r} ...
z3	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#   kZd1Zk1msg rX   b/home/seandotau/au.memorygallery.club/gallery/lib/python3.10/site-packages/numpy/f2py/capi_maps.pyload_f2cmap_file   sN   

"
rZ   z%gz%Lgz%dz%hhuz%hdz%huz%uz%ldz%luz(%g,%g)z	(%Lg,%Lg)z\"%s\"z'%c'c                 C   s  d}t | r)d| v r| d }n| d }|| d v r!t| d | S td|  |S t| r/|S t| r5dS t| r;dS d| v r| d  tv r| d  }t| }|d	 }d
| v rd| d
 v rz|| d
 d  }W |S  ty   td| d d| d
 d f  Y |S w d| d
 v r|d tv rt|d  }z|| d
 d  }W |S  ty   |tv rt| }z|t	| d
 d  }W Y |S  ty   td|| d
 d ||| d
 d t
 f  Y nw Y |S w |S t| std|   |S )z
    Determines C type
    voidresultnamevars,getctype: function %s has no return value?!
r!   r    typespecr7   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   rS   ZissubroutineZischaracter_or_characterarrayZisstring_or_stringarrayrM   rD   KeyErrorstrrE   getcwd
isexternal)varctypear`   Zf2cmaprX   rX   rY   r      sp   
#
r   c                 C   s   t dd| } | S )a^  Rewrite Fortran expression as f2py supported C expression.

    Due to the lack of a proper expression parser in f2py, this
    function uses a heuristic approach that assumes that Fortran
    arithmetic expressions are valid C arithmetic expressions when
    mapping Fortran function calls to the corresponding C function/CPP
    macros calls.

    z\blen\bZ	f2py_slen)resub)exprrX   rX   rY   f2cexpr  s   rm   c                 C   s   t | r%d| v r| d }n| d }|| d v rt| d | S td|  t| s1tdt|   d}d| v rN| d }d|v rD|d }n
d	|v rNt|d	 }td
|sZtd|rht| rftdt|   d}|S )Nr\   r]   r^   z0getstrlength: function %s has no return value?!
z;getstrlength: expected a signature of a string but got: %s
r;   charselectorr   lenz\(\s*(\*|:)\s*\)z(\*|:)zMgetstrlength:intent(hide): expected a string with defined length but got: %s
r=   )	isstringfunctionr	   rS   isstringreprrm   rj   matchisintent_hide)rg   ri   ro   rX   rX   rY   r	     s2   


r	   c           	      C   sn  i }t |rt|st||d< d|d< d|d< |S t|r,d|d< d|d< d|d< |S t|r5t|d }d||d< ztt|d |d< W n	 tyT   Y nw d	||d< tt	||d< tt	|d
g dd
 |d< t
t	|D ]9}g }|| tv r|| g}ntD ]}td| || r|| q|D ]}t|t| krd||<  nqqyd\|d< }|D ]}|d }|dvrd|d ||f |d< q|d r|d d d
 |d< d\|d< }|d D ]<}|d }|dvrd|d ||f |d< qt|rtd|  d|d |df |d< q|r%tdt| t|f  q|d r5|d d d
 |d< |S )Nsize0rankr7   dimsr;   	dimensionr   ,r   z	rank*[-1]z.*?\b%s\b.*)r7   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
)rq   isarrayr	   isscalarrA   joinrr   rK   rU   ro   rangedepargsrj   rs   rR   indexisintent_inrH   rS   )	ri   rg   verboseretdimr*   vvar#   rX   rX   rY   r
   ,  s   51







r
   c                 C   s  t |r(d|v r|d }n|d }||d v r t||d | S td|  dS | | }}d}t|r6d}nt|r<d}| }t|rY|d	 D ]}|d d
 dkrX|d
d  } nqFd}t|}	t|rnt| |\}}
d|
 }t	|rt|rd| |t
|	 t|	 |f }n
d| |t
|	 |f }d|t
|	 f }||fS t|rt|rd| |t||f }n
d| |t||f }d|t|f }||fS t|r|d }tt|}d| ||t|	 d||f }| |krd| |t|	 d|f }||fS d||t|	 d|| f }||fS t|rDd}| tv r8t|  tv r8dtt|   v r8tt|   d }|| ks6d| }nd}d| |f }|}||fS td|   ||fS )Nr\   r]   r^   r_   r7   r7   r7   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)ry   z.%s : %s rank-%s array('%s') with bounds (%s)%srz   )%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".
)rb   r   rS   r   isintent_inoutisintent_outr   hasinitvaluer   r~   rP   c2pycode_maprq   r	   r}   rr   ro   r   rf   lcb_maplcb2_map)ri   rg   afsigZsigoutoptout_arV   initrh   showinitr   rw   ZuarX   rX   rY   r   i  s   

#


*

r   c                 C   s   t |}t|rt|sd| t|f }|S t|r'd| t| t| f }|S t|rB|d }tt|}d| |t| d	|f }|S )Nz%%s : rank-0 array(string(len=%s),'c')z%s : rank-0 array(%s,'%s')ry   r   rz   )
r   rq   r}   r	   r~   rP   r   rr   ro   r   )ri   rg   rh   r   r   rw   rX   rX   rY   r     s*   
	r   c                 C   sD  t |r	d\}}nd\}}t|r|d }|}t|st|rti }z1|d }d|v r<t|dd d\|d< |d	< nt|i i }t|jt|j	|d< |d	< W n t
ya   td
|| f w t|rpd|d |d	 f }||fS t |r|s~d\}}|d dkrd|dd dd }|d dkrd|dd  }||fS )N)z""z''r   =rz   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')rq   r   	iscomplexiscomplexarrayr   splitrK   rd   r>   imagrU   
ValueErrorr}   replace)ri   rg   r   r   r   r   rX   rX   rY   r     sJ   


	r   c                 C   sD   t | st| rt| }| d d|}|S t| st| r dS dS )Nrn   Zf2py_lenr;   )rq   isstringarrayr	   getZischaracterZischaracterarray)rg   elsizerX   rX   rY   
get_elsize  s   r   c                 C   s~  | }t |r|d D ]}|dd dkr|dd } nq
| |t|d}g }t D ]\}}||r:|d|  q+|rEd||d< nd|d< t|rRd	|d
< n|d tv rat|d  |d
< nd|d
< t| |\|d< |d< t	|rt
|rt|st|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||d < t|rg t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(d0}	g }
t)d1t*|	d2D ]}|	| |r}|
|	|d   qkt|r|
d3| |d f  t|rd4t+d5d6 |d |}|
d7|  t|rd8| |d d4|
f |d9< nd:|d | |d d4|
f |d9< t!|r|d t,v rd;| t,|d  f |d<< t|rd=| | f |d<< t|rd>|  |d<< |d t,v rd?| t,|d  f |d@< dAt,|d   |dB< t|r'dC| | f |d@< t-| |\|dD< |dE< t.|r=|dF |dF< |S )Gz
    varname,ctype,atype
    init,init.r,init.i,pytype
    vardebuginfo,vardebugshowvalue,varshowvalue
    varrformat

    intent
    r   Nr   r   )varnameZ
outvarnamerh   zF2PY_%s|ZF2PY_INTENT_INr5   Z
varrformatrh   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.
lengthry   atyper   r   outputZinoutputrequiredoptionalZhiddenzcomplex scalarZscalarr    arrayzcomplex arrayzstring arrayzcomplex functionfunctioncallbackZ	auxiliaryr      zslen(%s)=%srz   c                 S   s   d| |f S )Nz%s|%srX   )xr6   rX   rX   rY   <lambda>D  s    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varshowvalue%sshowvalueformatz#name#:slen(%s)=%%d %s=\"%%s\"	pydocsignpydocsignoutnote)/r   r   Zisintent_dictrN   rR   r   r}   c2buildvalue_mapr   r   r   r   r   rf   r   r   rS   rT   rO   rq   r	   
dictappendr
   rA   
c2capi_mapr   	debugcapir   r   Z
isrequired
isoptionalrt   Zl_andr~   Zl_notr   r   iscomplexfunctionrb   isintent_callbackZisintent_auxr   ro   mapcformat_mapr   hasnote)ri   rg   r   rV   r   Zintent_flagsr$   sr   ilZrlr*   ZddimrX   rX   rY   r     s*  	

		












r   c           	      C   s  | d }t | }||dd| | t|td| || t| p%dt| p*dt| p/dd}d|v r;d|d< nd|d< d|v rHd	|d
< nd
|d
< i ad| v r| d 	 D ]@}|t
jv rt
j| D ]3}|d }d| d | v r| d | d 	 D ]}| d | d | |d kr|} nq||d t|< qdqXnd| v r| d rtd|d t| d f  t| tpd|d< t| rbd| 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 r1d|t|d  f |d< t| r>d ||f |d< t| rbt| d | |d!< |d! d"krbtd#t| d   d$|d!< t| rr| d |d< dg| d< |S )%zV
    name,NAME,begintitle,endtitle
    rname,ctype,rformat
    routdebugshowvalue
    r]   _\_	end of %sr7   )r]   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callprotoargumentr\   rnamer   r   r^   rh   r   Z
resultnoteSee elsewhere.Zrformatr   z.routsign2map: no c2buildvalue key for type %s
r   Zroutdebugshowvaluer   Zrlengthr=   zzroutsign2map: expected explicit specification of the length of the string returned by the fortran function %s; taking 10.
10)Zgetfortrannamer   rM   uppergentitleZgetcallstatementgetusercodegetusercode1r   rO   r   Zcb_maprS   rr   Zgetcallprotoargumentrb   r   r   Zhasresultnoter   r   r   rp   r	   r   )	routr]   fnamer   uunlnrV   ri   rX   rX   rY   r   `  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| r;| d |d< t| p@d|d	< t| pHd|d
< | d r\t| d d pXd|d< nd|d< t| ped|d< d| v rr| d |d< d| v r|| d |d< |S )z
    modulename
    r]   r   r   )Zf90modulenameZF90MODULENAMEZtexf90modulename)
modulenameZ
MODULENAMEZtexmodulenameZrestdocr   r7   r   r   bodyr   Zinterface_usercodeZpymethoddefZcoutputZf2py_wrapper_output)ismoduler   r   Z
getrestdocr   r   r   Zgetpymethoddef)mr   rX   rX   rY   r     s.   

r   c                 C   s   d| i}|d |d< t ||d< |d tv r$t|d  |d< t||d< |d tv r4dt|d   |d< t|r@t|t| |}t| |\|d< |d	< t|rZ|d
 |d
< dg|d
< |S )Nr   Z	varname_irh   r   r   r   r   r   r   r   r   )	r   r   r   r   r}   r   r
   r   r   )ri   rg   r   r   rX   rX   rY   r     s   
r   c                 C   s  d| d |f dd}t | r0d| d v rd}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rgd|d< n|d |d< |d dkrt| r|d|d< nd|d< |d tv rdt|d   |d< t| rt| |d< t	| rd| v r| 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 rd| v r| d# D ]}| d | }ttt|r|d$ }t|r|d$ }qt||d%< t||d&< t
| rCt	| rCd| v rC| d |d'< dg| d< |S )(z\
    name,begintitle,endtitle,argname
    ctype,rctype,maxnofargs,nofoptargs,returncptr
    zcb_%s_in_%sr]   r7   )r]   
returncptrr   r   r   z	%s(%s,%s)ZcallbacknameZexternZstaticr   r   r   r   rh   r[   Zrctyper    z3
#ifdef F2PY_CB_RETURNCOMPLEX
return_value=
#endif
r   zreturn_value=r   r   Z	strlengthr\   r^   r   r   r   r   r   z8
#ifdef F2PY_CB_RETURNCOMPLEX
#ctype#
#else
void
#endif
r   argsr   r   r   Zroutnote)r   rM   r   r   r   r   r   rp   r	   rb   r   r   Zl_orr   r   r   rr   )r   Zumr   r   ri   Znofargsr   rg   rX   rX   rY   r     s~   








r   c                 C   s   | t |d}t|rd|d< |d tv r#t|d  |d< t||d< |d tv r3dt|d   |d< t|r@t|t| |}nt|rNt	||d< d	|d
< t
| |\|d< |d< t|rh|d |d< dg|d< t| ||d< |S )N)r   rh   r   rh   r   r   r   r   ru   r;   rw   r   r   r   r   Z	arrdocstr)r   r   r   r   r   r}   r   r
   rq   r	   r   r   r   )ri   rg   r   rX   rX   rY   r   2  s&   
r   )r   )N)+__doc__r7   r   versionZf2py_versionrA   rj   rE   Zcrackfortranr   r   Z	_isocbindr   Zauxfuncs__all__r   r   r   rP   r   r   r   rD   Z
deep_mergerB   rC   rQ   rZ   r   r   rm   r	   r
   r   r   r   r   r   r   r   r   r   r   rX   rX   rX   rY   <module>   s  	
	
	


,	
3
=J#jQ
R