o
    6az                     @   s  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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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Zeri 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iZi d
d-dd.dd-dd/dd/dd0dd1dd2dd3dd4dd5dd6dd7dd8dd8dd9Zer@i d
d-dd.dd:dd0dd;dd0dd<dd=dd3dd>dd5d(d6dd?d+d@dd7dd8ddAddBiZd-d.d0d0d<d3d5d6dCdCdCdDdEZerS	 ddd
dddFddddddddd+dG	ddddddHddddddFddddddIdJdidJd
idJdidJdidK	aetZdLdM Zi d
dNddNddOddPddPddQddRddSddPddTddUd(dVddUddWddWddXddYZdZd[ Zd\d] Zdrd^d_Zd`da Zdbdc Zddde Zdfdg Z dhdi Z!djdk Z"dsdldmZ#dndo Z$dpdq Z%dS )ta)  

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doublefloatlong_doublecharintsigned_charunsigned_charshortunsigned_shortlong	long_longunsignedcomplex_floatcomplexcomplex_doublecomplex_long_doublestring
NPY_DOUBLE	NPY_FLOAT
NPY_STRING	NPY_UBYTENPY_BYTE	NPY_SHORT
NPY_USHORTNPY_INTNPY_UINTNPY_LONG
NPY_CFLOATNPY_CDOUBLENPY_LONGDOUBLEunsigned_long	NPY_ULONGNPY_LONGLONGunsigned_long_longNPY_ULONGLONGdf1bswiulLFDcgBhHIqQGSNy)r   r   r   r   r   r   r   r   r   r!   r"   r#   ) 481216)	rN   r8   2rO   rP   -1z-2z-4z-8)rN   rP   rR   2432)rN   r8   rS   rO   rP   rN   )	realintegerr    complexkindlogicalzdouble complexzdouble precisionbyte	characterc                 C   s  t ta| d u rd} tj| sd S ztd|  t	| d}t
| i i }W d    n1 s3w   Y  t| D ]\}}t| D ]
}|| || < qH|| || < q>t| D ]d}|tvrli t|< t||  D ]Q}|| | tv r|t| v rtd||t| | || | f  || | t| |< td|||| | f  qttd|||| | || | tt f  qtqbtd W d S  ty } ztd|  W Y d }~d S d }~ww )	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listitemskeyslowerc2py_maperrmess	Exception)f2cmap_filer7   r6   kd1k1msg rv   6/usr/lib/python3/dist-packages/numpy/f2py/capi_maps.pyload_f2cmap_file   sL   
"
rx   z%gz%Lgz%dz%hhuz%hdz%huz%uz%ldz%luz(%g,%g)z	(%Lg,%Lg)%sc                 C   s  d}t | r)d| v r| d }n| d }|| d v r!t| d | S td|  |S t| r/|S d| v r| d  tv r| d  }t| }|d }d| v rd	| d v rxz|| d d	  }W |S  tyw   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?!
typespecrN   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   ro   issubroutinerm   ra   KeyErrorstrrb   getcwd
isexternal)varctypear   f2cmaprv   rv   rw   r      sl   
!

r   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 rL| d }d|v rD|d }nd	|v rL|d	 }td
|sXtd|rft| rdt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
r8   charselectorr   lenz\(\s*(\*|:)\s*\)z(\*|:)zMgetstrlength:intent(hide): expected a string with defined length but got: %s
rT   )isstringfunctionr   ro   isstringreprrematchisintent_hide)r   r   r   rv   rv   rw   r   "  s2   


r   c           	      C   st  i }t |rt|st||d< |d |d< d|d< |S t|r.d|d< d|d< d|d< |S t|r8t|d }d||d< ztt|d |d< W n	 tyV   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qq{d\|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< qt|rtd|  d|d |df |d< q|r(tdt| t|f  q|d r8|d d d
 |d< |S )Ndimssizer8   rank0rN   	dimensionr   ,r   z	rank*[-1]z.*?\b%s\b.*)rN   r   setdims)r   :z(*)z(:)z%s#varname#_Dims[%d]=%s,	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
)r   isarrayr   isscalarr^   joinr   rh   rp   r   rangedepargsr   r   appendindexisintent_inre   ro   )	r   r   verboseretdimr<   vvar6   rv   rv   rw   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~   rN   rN   rN   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%sz8getpydocsign: Could not resolve docsignature for "%s".\n)r   r
   ro   r   isintent_inoutisintent_outr   hasinitvaluer   r   rn   c2pycode_mapr   r   r   r   r   r   r   lcb_maplcb2_map)r   r   afsigsigoutoptout_arr   initr   showinitr   r   uarv   rv   rw   r
   {  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')r   r   r   )
r   r   r   r   r   rn   r   r   r   r   )r   r   r   r   r   r   rv   rv   rw   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   =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')r   r   	iscomplexiscomplexarrayr   splitrh   r   rW   imagrp   
ValueErrorr   replace)r   r   r   r   r   r   rv   rv   rw   r     sJ   


	r   c                 C   sr  | }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|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'd/}	g }
t(d0t)|	d1D ]}|	| |rw|
|	|d   qet|r|
d2| |d f  t|rd3t*d4d5 |d |}|
d6|  t|rd7| |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 rd>| t+|d  f |d?< d@t+|d   |dA< t|r!dB| | f |d?< t,| |\|dC< |dD< t-|r7|dE |dE< |S )Fz
    varname,ctype,atype
    init,init.r,init.i,pytype
    vardebuginfo,vardebugshowvalue,varshowvalue
    varrfromat
    intent
    r   Nr   r   )varname
outvarnamer   zF2PY_%s|F2PY_INTENT_INrL   
varrformatr   Or   r   r   r   r   r   r   	cbnamekeycbname
maxnofargs
nofoptargsdocstrcbdocstrlatexdocstrcblatexdocstrz5sign2map: Confused: external %s is not in lcb_map%s.
lengthr   atyper   outputinoutputrequiredoptionalhiddenzcomplex scalarscalarr#   arrayzcomplex arrayzstring arrayzcomplex functionfunctioncallback	auxiliaryr      zslen(%s)=%sr   c                 S   s   d| |f S )Nz%s|%srv   )xrM   rv   rv   rw   <lambda>F  s    zsign2map.<locals>.<lambda>zdims(%s)zdebug-capi:%s=>%s:%svardebuginfozdebug-capi:%s %s=%s:%sdebug-capi:%s=%svardebugshowvalue"debug-capi:slen(%s)=%%d %s=\"%%s\"zdebug-capi:%s=%%pz#name#:%s=%svarshowvaluery   showvalueformatz#name#:slen(%s)=%%d %s=\"%%s\"	pydocsignpydocsignoutnote).r   r   isintent_dictrk   r   r   r   c2buildvalue_mapr   r   r   r   r   r   r   r   ro   rj   rl   r   r   
dictappendr	   r^   
c2capi_map	debugcapir   r   
isrequired
isoptionalr   l_andr   l_notr   isstringarrayiscomplexfunctionr   isintent_callbackisintent_auxr   r   mapcformat_mapr
   hasnote)r   r   r   rr   r   intent_flagsr7   r:   r   ilrlr<   ddimrv   rv   rw   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 %srN   )r|   texname
name_lowerNAME
begintitleendtitlefortrannameFORTRANNAMEcallstatementusercode	usercode1	F_FUNC_USF_FUNCF_WRAPPEDFUNC_USF_WRAPPEDFUNCuser   r   r   	externalszMroutsign2map: Confused: function %s has externals %s but no "use" statement.
callprotoargumentr{   rnamer   r   r}   r   r   
resultnoteSee elsewhere.rformatr   z.routsign2map: no c2buildvalue key for type %s
r   routdebugshowvaluer   rlengthrT   zzroutsign2map: expected explicit specification of the length of the string returned by the fortran function %s; taking 10.
10)getfortrannamer   rm   uppergentitlegetcallstatementgetusercodegetusercode1r   rl   r   cb_mapro   r   getcallprotoargumentr   r
   r   hasresultnoter   r   r   r   r   r  )	routr|   fnamer   r=   unlnrr   r   rv   rv   rw   r   b  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  )f90modulenameF90MODULENAMEtexf90modulename)
modulename
MODULENAMEtexmodulenamerestdocr   rN   r  r  bodyr   interface_usercodepymethoddefcoutputf2py_wrapper_output)ismoduler"  r   
getrestdocr  r%  r&  getpymethoddef)mr   rv   rv   rw   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|r:t|t| |}t| |\|d< |d< t|rT|d	 |d	< d
g|d	< |S )Nr   	varname_ir   r   ry   r   r   r   r   r  )r   r   r   r   r   r	   r
   r  )r   r   r   r   rv   rv   rw   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|   rN   )r|   
returncptrr  r  r  z	%s(%s,%s)callbacknameexternstaticr   r  r  r  r   rz   rctyper#   z3
#ifdef F2PY_CB_RETURNCOMPLEX
return_value=
#endif
r?  zreturn_value=ry   r   	strlengthr{   r}   r   r  r  r   r   z8
#ifdef F2PY_CB_RETURNCOMPLEX
#ctype#
#else
void
#endif
r   argsr   r   r   routnote)r   rm   r"  r#  r   r   r   r   r   r   r  r
   l_orr   r   r   r   )r*  umr   r  r   nofargsr   r   rv   rv   rw   r     s~   








r   c                 C   s   | t |d}t|rd|d< |d tv rt|d  |d< |d tv r-dt|d   |d< t|r:t|t| |}nt|rHt||d< d|d	< t	| |\|d
< |d< t
|rb|d |d< dg|d< t| ||d< |S )N)r   r   r   r   r   ry   r   r   r8   r   r   r   r   r  	arrdocstr)r   r   r   r   r   r   r	   r   r   r
   r  r   )r   r   r   rv   rv   rw   r   3  s$   
r   )r   )N)&__doc__rN   r   versionf2py_versionr^   r   rb   crackfortranr   r   auxfuncs__all__using_newcorer   r   r   rn   r   r   r   ra   r_   r`   rx   r   r   r   r	   r
   r   r   r   r   r   r   r   r   rv   rv   rv   rw   <module>   s  	
	
	
	
	

+	
1
=J#hQ
R