U
    dz                     @   sz  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 dd	d
dddddddddg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/d0d1d2d3d4d5d6d7d8d8d9d:ZerJd-d.d;d0d<d0d=d>d3d?d5d6d@dAd7d8dBdCd,Zd-d.d0d0d=d3d5d6dDdDdDdEdFZdddGdHdHdIddJdKddLdMdNdOdPdQ	dRdRdSdTdTdUdRdRdSdTdTdIddVdKddLdWdXdSidXdGidXdVidXdidY	aetZg ZdZd[ Zd\d\d]d^d^d_d`dad^dbdcdddcdededfdgdhZdid Zdjd	 Zdudkd
Zdld Zdmd Zdnd Z dod Z!dpd Z"dqd Z#dvdrdZ$dsd Z%dtd Z&dS )wa)  

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complexkindlogicalzdouble complexzdouble precisionbyte	characterc                 C   s  t ta| d kr&d} tj| s&d S zhtd|  t	| d}t
|  i i }W 5 Q R X | D ]6\}}| D ]}|| || < qz|| || < qj| D ]}|tkri t|< ||  D ]}|| | tkrT|t| krtd||t| | || | f  || | t| |< td|||| | f  t|| |  qtd|||| | || | tt f  qqtd W n0 tk
r } ztd|  W 5 d }~X Y nX 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d1Zk1msg rb   8/tmp/pip-unpacked-wheel-c1z6a0q8/numpy/f2py/capi_maps.pyload_f2cmap_file   sN    
  
 
 
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| kr| d }n| d }|| d krBt| d | S td|  n~t| r^|S d| kr| d  tkr| d  }t| }|d }d| krd	| d krz|| d d	  }W n2 tk
r   td
| d d	| d d	 f  Y nX nd| d kr|d tkrt|d  }z|| d d  }W n tk
r   |tkrVt| }z|t| d d  }W nB tk
r   td|| d d ||| d d t	 f  Y nX Y nX 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issubroutinerV   rM   KeyErrorstrrN   getcwd
isexternal)varctypeark   Zf2cmaprb   rb   rc   r      sV    


 
 
c                 C   s   t | rJd| kr| d }n| d }|| d kr>t| d | S td|  t| sbtdt|   d}d| kr| d }d|kr|d }nd	|k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    


c           	      C   s  i }t |r8t|s8t||d< |d |d< d|d< nLt|r\d|d< d|d< d|d< n(t|rt|d }d||d< ztt|d |d< W n tk
r   Y nX d	||d< tt	||d< tt	|d
g dd
 |d< t
t	|D ]|}g }|| tkr|| g}n,tD ]&}td| || r|| q|D ](}t|t| krJd||<  qqJqd\|d< }|D ].}|d }|dkrd|d ||f |d< q|d r|d d d
 |d< d\|d< }|d D ]|}|d }|dkrd|d ||f |d< nLt|rDtd|  d|d |df |d< n|rtdt| t|f  q|d r|d d d
 |d< |S )NZ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   isscalarrJ   joinrx   rT   r^   ru   rangedepargsry   rz   r[   indexisintent_inrQ   r\   )	rt   rr   verboseretdimr,   vvar&   rb   rb   rc   r	   <  s    


  

  
  
c                 C   s  t |rPd|kr|d }n|d }||d k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krdt|  tkrddtt|   k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   rY   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    



 

      
 

*

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   rY   r   rx   ru   r   )rt   rr   rs   r   r   r}   rb   rb   rc   r     s&    
c                 C   sN  t |rd\}}nd\}}t|rF|d }|}t|s@t|ri }zb|d }d|krzt|dd d\|d< |d	< n*t|i i }t|jt|j	 |d< |d	< W n$ t
k
r   td
|| f Y nX t|rd|d |d	 f }n\t |rF|sd\}}|d dkr(d|dd dd }|d dkrFd|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   splitrT   ro   rD   imagr^   
ValueErrorr   replace)rt   rr   r   r   r   r   rb   rb   rc   r     sF    


"

 
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k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k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k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+krd:| t+|d  f |d;< t|rd<| | f |d;< t|rd=|  |d;< |d t+k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_dictrW   r[   r   r   c2buildvalue_mapr   r   r   r   r   rq   r   r   r\   r]   rX   rw   r   
dictappendr	   rJ   
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   ZilZrlr,   Zddimrb   rb   rc   r     s    	


 



                   


     

 


 



c           	      C   s  | d }t | }||dd| | t|td| || t| pJdt| pTdt| p^dd}d|krvd|d< nd|d< d|krd	|d
< nd
|d
< i ad| kr:| d 	 D ]}|t
jkrt
j| D ]j}|d }d| d | kr(| d | d 	 D ]*}| d | d | |d kr|} q(q|d t|< qqn0d| krj| d rjtd|d t| d f  t| tpxd|d< t| rd| kr| d }n| d }||d< t|| \|d< |d< t| d | |d< t| r| d | d |d< dg| d | d< |d tkr&t|d  |d< nd|d< tdt|d   t| r|d tk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   rV   uppergentitleZgetcallstatementgetusercodegetusercode1r   rX   r   Zcb_mapr\   rx   Zgetcallprotoargumentrm   r
   r   Zhasresultnoter   r   r   rv   r   r   )	routrh   fnamer   r-   Zunlnr_   rt   rb   rb   rc   r   a  s    








 






 


 




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| kr| d |d< d| k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.    

c                 C   s   d| i}|d |d< t ||d< |d tkr<t|d  |d< |d tk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    
c                 C   s  d| d |f dd}t | r`d| d k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kr$dt|d   |d< t| r:t| |d< t	| rd| k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#| krLd| krL| d# D ]<}| d | }ttt|r|d$ }t|r|d$ }qt||d%< t||d&< t
| rt	| rd| k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callbackname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   rV   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    














c                 C   s   | t |d}t|rd|d< |d tkr:t|d  |d< |d tk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   )N)'__doc__r>   r   versionZf2py_versionrJ   ry   rN   Zcrackfortranr   r   Zauxfuncs__all__Zusing_newcorer   r   r   rY   r   r   r   rM   rK   rL   rZ   rd   r   r   r   r	   r
   r   r   r   r   r   r   r   r   rb   rb   rb   rc   <module>   s                 
,/
=J#iQ
R