3
���h � @ s� d Z ddgZddlZddlZddlZejdej�Zejdej�Zejdej�Z dd � Z
ejd
�Z ejd �Z ejd �Z
d
d� Zdd� Zejd�Zdd� Zdd� Zejd�Zdd� Zdd� Zejdej�Zdd � Zd d� Zed �Zd d � Zed!kr�e� dS )"a�
process_file(filename)
takes templated file .xxx.src and produces .xxx file where .xxx
is .pyf .f90 or .f using the following template rules:
'<..>' denotes a template.
All function and subroutine blocks in a source file with names that
contain '<..>' will be replicated according to the rules in '<..>'.
The number of comma-separated words in '<..>' will determine the number of
replicates.
'<..>' may have two different forms, named and short. For example,
named:
<p=d,s,z,c> where anywhere inside a block '<p>' will be replaced with
'd', 's', 'z', and 'c' for each replicate of the block.
<_c> is already defined: <_c=s,d,c,z>
<_t> is already defined: <_t=real,double precision,complex,double complex>
short:
<s,d,c,z>, a short form of the named, useful when no <p> appears inside
a block.
In general, '<..>' contains a comma separated list of arbitrary
expressions. If these expression must contain a comma|leftarrow|rightarrow,
then prepend the comma|leftarrow|rightarrow with a backslash.
If an expression matches '\<index>' then it will be replaced
by <index>-th expression.
Note that all '<..>' forms in a block must have the same number of
comma-separated entries.
Predefined named template rules:
<prefix=s,d,c,z>
<ftype=real,double precision,complex,double complex>
<ftypereal=real,double precision,\0,\1>
<ctype=float,double,complex_float,complex_double>
<ctypereal=float,double,\0,\1>
� process_str� process_file� Nz2(\n|\A)(( (\$|\*))|)\s*(subroutine|function)\bz+\n\s*end\s*(subroutine|function)\b.*(\n|\Z)z\n (\$|\*)\s*function\bc C s� g }d}x�t j| |�}|dkr P |j� }tj| ||j� �rrx6| jd||�}|dkrTP |}| ||d � dkr<P q<W |d7 }tj| |j� �}|r�|j� d p�t| � }}|j ||f� q
W |S )z� Return a list of tuples for each function or subroutine each
tuple is the start and end of a subroutine or function to be
expanded.
r N�
� � z
$���)
�routine_start_re�search�start�function_start_re�match�end�rfind�routine_end_re�len�append)�astrZspanlist�ind�mr
�ir
� r �>/tmp/pip-build-5_djhm0z/numpy/numpy/distutils/from_template.py�parse_structure: s( r z<\s*(\w[\w\d]*)\s*>z <\s*(\w[\w\d]*)\s*=\s*(.*?)\s*>z<\s*((.*?))\s*>c C sT t j| �}i }x@|D ]8}|d j� p*t|�}|d jdd�}t|�}|||< qW |S )Nr r z\,z@comma@)�named_re�findall�strip�
unique_key�replace�conv)r Zreps�names�rep�name�repl�thelistr r r �find_repl_patternsY s
r$ c C s" t | �}tjtd| �d } | |fS )N� r )r$ �re�subnr )r r r r r � find_and_remove_repl_patternsc s r( z\A\\(?P<index>\d+)\Zc C sd | j d�}dd� |D �}x@tt|��D ]0}tj|| �}|r&t|jd��}|| ||< q&W dj|�S )N�,c S s g | ] }|j � �qS r )r )�.0�xr r r �
<listcomp>k s zconv.<locals>.<listcomp>�index)�split�ranger �item_rer �int�group�join)r �b�lr r �jr r r r i s
r c C s@ t | j� �}d}d}x&|s:d| }||kr4|d7 }qd}qW |S )z( Obtain a unique key given a dictionary.Fr z__l%sT)�list�keys)�adictZallkeys�done�nZnewkeyr r r r s s
r z\A\s*(\w[\w\d]*)\s*\Zc sv | j dd�} | j dd�} t| ��tjd| �} �fdd�}tj|| �} d }d }i �x�tj| �D ]�}|�kr\�j||j|d ��}|d kr�td| ��||kr�|j d � r�|||< d
d � |j
d �D �}t |�}|d kr�|}|�|< |}q\||kr�|�|< q\t d
|d j
�| �||f � q\W ��s | S � �fdd�} d}
x&t|�D ]� |
tj| | �d 7 }
�q<W |
j dd�}
|
j dd�}
|
S )Nz\>z @rightarrow@z\<z @leftarrow@z<\1>c sn t | jd�jdd��}tj|�r(d| S d }x � j� D ]}� | |kr6|}q6W |d krft� �}|� |< d| S )Nr z\,z@comma@z<%s>)r r2 r �template_name_rer r8 r )�mobjr# r! �key)�lnamesr r �listrepl� s
z expand_sub.<locals>.listreplz No replicates found for <%s>�_c S s g | ]}|j d d��qS )z@comma@r) )r )r* r r r r r, � s z expand_sub.<locals>.<listcomp>r) zHMismatch in number of replacements (base <%s=%s>) for <%s=%s>. Ignoring.c s$ | j d�}�j|� d |g �� S )Nr )r2 �get)r= r! )�k�rulesr r �namerepl� s
z expand_sub.<locals>.namereplr% z
�>�<)r r$ r �sub�list_re� template_rer rB �
ValueError�
startswithr. r �printr3 r/ ) �substrr r@ ZnumsubsZ base_rule�rr# Zrule�numrE �newstrr )rC r? rD r �
expand_sub� sF
rR c C s� | }d}t |�}d}i }|jt� x\|D ]T}t|||d � �\}}||7 }|j|� |t||d |d � |�7 }|d }q(W |||d � 7 }|S )Nr% r r )r �update�_special_namesr( rR ) �allstrrQ �writestr�structZoldendr rH Z
cleanedstrZdefsr r r r � s
z9(\n|\A)\s*include\s*['\"](?P<name>[\w\d./\\]+[.]src)['\"]c C s� t jj| �}t| ���}g }xz|D ]r}tj|�}|r�|jd�}t jj|�sVt jj||�}t jj |�r|t
d|� |j t |�� q�|j
|� q |j
|� q W W d Q R X |S )Nr! zIncluding file)�os�path�dirname�open�include_src_rer r2 �isabsr3 �isfilerM �extend�resolve_includesr )�source�d�fid�lines�liner �fnr r r r` � s
r` c C s t | �}tdj|��S )Nr% )r` r r3 )ra rd r r r r � s z�
<_c=s,d,c,z>
<_t=real,double precision,complex,double complex>
<prefix=s,d,c,z>
<ftype=real,double precision,complex,double complex>
<ctype=float,double,complex_float,complex_double>
<ftypereal=real,double precision,\0,\1>
<ctypereal=float,double,\0,\1>
c C sv yt jd } W n tk
r. t j}t j}Y n*X t| d�}tjj| �\}}|}t|d�}|j � }t
|�}|j |� d S )Nr rO �w) �sys�argv�
IndexError�stdin�stdoutr[ rX rY �splitext�readr �write)�filerc �outfile�base�extZnewnamerU rV r r r �main� s
rt �__main__) �__doc__�__all__rX rh r& �compile�Ir r r r rJ r rI r$ r( r0 r r r< rR r r\ r` r rT rt �__name__r r r r �<module>/ s6
< |