o
    6a=                     @   sl   d Z ddlmZ ejZddlmZmZmZm	Z	m
Z
 ddlmZ ddlmZ ddlmZ dddZd	d
 ZdS )aM  

Build common block mechanism for f2py2e.

Copyright 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__)hasbody	hascommonhasnoteisintent_hideoutmess)	capi_maps)	func2subr)	rmbadnamec           
         s   g }t  r# d  D ]\}} fdd|D }||||f qnt r5 d D ]	}|t|d }q+|rTg }g }|D ]}	|	d |vrQ||	d  ||	 q=|S |S )Ncommonc                    s   i | ]	}| d  | qS )vars ).0vblockr   9/usr/lib/python3/dist-packages/numpy/f2py/common_rules.py
<dictcomp>   s    z$findcommonblocks.<locals>.<dictcomp>body    )r   itemsappendr   findcommonblocks)
r   topretkeyvaluevars_btretnamestr   r   r   r      s&   
r   c                 C   s  g g dgd}dg}|fdd}dg}|fdd}dg}|fdd	}dg}|fd
d}	t | D ]\}
}}|
 }g g }}|D ]}t|| rN|| q@|| q@|rftd|
d|d|f  ntd|
d|f  |d|
  |d |D ]
}|t|| q}|
dkr|dd|  n|d|
d|f  |dd|  |d |d|
  g }|D ]?}t	|| }tj
| }t||| }|d r|d|d   n|d |d  }|sd}|d||d ||f  q|d t|}ddd |D }|d|
|f  |d  |D ]}|d!|
|f  q|d" d#|v r0d$}nd%}|d&|||
 dd'gt| f  |d(|
  |d)|||
 |
f  |d* |d+|
|
f  |d,|
  |d- |
d#d.}|	d/|  |	d0 |D ]-}|	d1t|||   t|| r|| d2 }t|trd3|}|	d4|  q|	d5 |d6 d7|
dtd8d9 ||f  q/||d:< ||d;< |d< |d=< t|d6 d>krd|d6< ||d< fS )?Nz"COMMON blocks:\n")commonhooksinitcommonhooksdocs c                 S      d|d | f |d< d S )Nz%s
      %sr   r   linesr   r   r   fadd4      zbuildhooks.<locals>.faddc                 S   r&   Nz%s
%sr   r   r'   r   r   r   cadd8   r+   zbuildhooks.<locals>.caddc                 S   r&   r,   r   r'   r   r   r   iadd<   r+   zbuildhooks.<locals>.iaddc                 S   r&   r,   r   r'   r   r   r   dadd@   r+   zbuildhooks.<locals>.daddzF		Constructing COMMON block support for "%s"...
		  %s
		  Hidden: %s
,z7		Constructing COMMON block support for "%s"...
		  %s
z subroutine f2pyinit%s(setupfunc)zexternal setupfunc_BLNK_z	common %szcommon /%s/ %szcall setupfunc(%s)zend
z'static FortranDataDef f2py_%s_def[] = {dimsz(%s)z-1z	{"%s",%s,{{%s}},%s},rankz
	{NULL}
};c                 S   s   g | ]}d | qS )zchar *r   )r   r)   r   r   r   
<listcomp>j   s    zbuildhooks.<locals>.<listcomp>zstatic void f2py_setup_%s(%s) {z	int i_f2py=0;z!	f2py_%s_def[i_f2py++].data = %s;}_	F_FUNC_USF_FUNCz3extern void %s(f2pyinit%s,F2PYINIT%s)(void(*)(%s));zchar*z static void f2py_init_%s(void) {z*	%s(f2pyinit%s,F2PYINIT%s)(f2py_setup_%s);z}
z5	tmp = PyFortranObject_New(f2py_%s_def,f2py_init_%s);z&	F2PyDict_SetItemString(d, "%s", tmp);z	Py_DECREF(tmp);z\_z&\subsection{Common block \texttt{%s}}
z\begin{description}z\item[]{{}\verb@%s@{}}note
z--- %sz\end{description}r$   z"	/%s/ %s\n"c                 S   s   | | S )Nr   )r   dr   r   r   <lambda>   s    zbuildhooks.<locals>.<lambda>r"   r#   r   latexdocr   )r   lowerr   r   r   joinr	   var2fixfortranr   getctype
c2capi_map
getarrdimsstripr
   upperlenreplacegetarrdocsignr   
isinstancelistmap)mr   fwrapr*   chooksr-   ihooksr.   docr/   namevnamesr   
lower_namehnamesinamesnidimsctatdmdmsinames1inames1_tpsr8   tnamer9   r   r   r   
buildhooks/   s   







r_   N)r   )__doc__r%   r   versionf2py_versionauxfuncsr   r   r   r   r   r   r	   crackfortranr
   r   r_   r   r   r   r   <module>   s   
