U
    ù‘«jœz  ã                   @   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rôd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r¾i t|< ||  ¡ D ]¸}|| | tk�rT|t| k�rtd||t| | || | f ƒ || | t| |< td|||| | f ƒ t || | ¡ qÊtd|||| | || | tt ¡ ƒf ƒ qÊqªt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-16f7874t/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�rÐd	| d kröz|| d d	  }W n2 tk
rò   td
| d d	| d d	 f ƒ Y nX nÂd| 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
| ƒ�sÐt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
|¡s°t d|¡rÌt| ƒrÈt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|ƒ�r„t |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ø�qJqød\|d< }|D ].}|d }|dk�r†d|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|�rèt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 … } q²qŒd}t|ƒ}	t|ƒrÜt| |ƒ\}}
d|
 }t	|ƒ�r4t|ƒ�rd| |t
|	 t|	 |f }nd| |t
|	 |f }d|t
|	 f }�nNt|ƒ�r„t|ƒ�r^d| |t|ƒ|f }nd| |t|ƒ|f }d|t|ƒf }nþt|ƒ�r|d }tt|ƒƒ}d| ||t|	 d |¡|f }| |k�ræd| |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|ƒrêi }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|ƒrèd|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|rŠd |¡|d< nd|d< t|ƒr¤d	|d
< n&|d tkrÂt|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�ržt|  |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|ƒ�rÔt|ƒ|d< t|ƒ�rüt|t| |ƒƒ}t |d ¡}|d tk�rt|d  |d< t|ƒ�r¸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"}	g }
t(d0t)|	ƒd1ƒD ]&}|	| |ƒ�r”|
 |	|d  ¡ �q”t|ƒ�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�rˆd:| t+|d  f |d;< t|ƒ�r¢d<| | f |d;< t|ƒ�r¸d=|  |d;< |d t+k�ròd>| 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r�d	|d
< nd
|d
< i ad| k�r:| d  	¡ D ]„}|t
jkr²t
j| D ]j}|d }d| d | k�r(| d | d  	¡ D ]*}| d | d | |d krü|} �q(qü|d t|< qÊq²n0d| k�rj| d �rjtd|d t| d ƒf ƒ t| tƒ�pxd|d< t| ƒ�rÔd| 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| ƒ�rŒd ||f |d< t| ƒ�rÔt| d | ƒ|d!< |d! d"k�rÔt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| ƒp€d|d	< t| ƒp�d|d
< | d r¸t| d d ƒp°d|d< nd|d< t| ƒpÊd|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rÎd|d< n|d |d< |d dk�rt| ƒrúd|d< nd|d< |d tk�r$dt|d   |d< t| ƒ�r:t| ƒ|d< t	| ƒ�rÊd| k�rX| d }n| d }t
| d | ƒ�r˜| d | d |d< dg| d | d< ||d< t|| ƒ\|d< |d < t| ƒ�rêd!|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
| ƒ�r˜t	| ƒ�r˜d| 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|ƒr�t|ƒ|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