U
|
O ±dî ã @ sø d Z ddgZddlZddlZddlZe dej¡Ze dej¡Ze dej¡Z dd Z
|
e d
|
¡Z e d ¡Z e d ¡Z d d Zdd Ze d¡Zdd Zdd Ze d¡Zdd Zdd Ze dej¡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}t | |¡}|dkr q°| ¡ }t | || ¡ ¡rl| d||¡}|dkrPql|}| ||d
dkr8qlq8|d7 }t | | ¡ ¡}|r| ¡ d pt| }}| ||f¡ q|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 úTd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\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 sP t | ¡}i }|D ]8}|d ¡ p(t|}|d dd¡}t|}|||< q|S )Nr r ú\,ú@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 | }t td| ¡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 s` | d¡}dd |D }tt|D ]0}t || ¡}|r$t| d¡}|| ||< q$d |¡S )Nú,c S s g | ] }| ¡ 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 | ¡ }d}d}|s8d| }||kr2|d7 }qd}q|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 sl | dd¡} | dd¡} t| t d| ¡} fdd}t || ¡} d }d }i t | ¡D ]´}|krZ || |d ¡¡}|d krtd| ||kr¨| d ¡s¨|||< d
|
d |
|
d ¡D }t |}|d krÞ|}||< |}qZ||krð||< qZt d |d | ¡||f qZs| S fdd} d}
|
t|D ] |
|
t | | ¡d 7 }
|
q4|
|
dd¡}
|
|
|
dd¡}
|
|
|
S )Nz\>z @rightarrow@z\<z @leftarrow@z<\1>c sj t | d¡ dd¡}t |¡r(d| S d } ¡ D ]} | |kr4|}q4|d krbt }| |< d| S )Nr r r z<%s>)r r4 r Útemplate_name_rer r: r )Úmobjr% r# Úkey)Úlnamesr r Úlistrepl s
|
z expand_sub.<locals>.listreplz No replicates found for <%s>Ú_c S s g | ]}| d d¡qS )r 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$ | d¡} | d |g ¡ S )Nr )r4 Úget)r? r# )ÚkÚrulesr r Únamerepl² s
|
z expand_sub.<locals>.namereplr' z
|
|
ú>ú<)r r& r ÚsubÚlist_reÚ template_rer rD Ú
|
ValueErrorÚ
|
startswithr0 r Úprintr5 r1 ) Úsubstrr! rB ZnumsubsZ base_ruleÚrr% ZruleÚnumrG Únewstrr )rE rA rF r Ú
|
expand_sub sH
|
þ rT c C s | }d}t |}d}i }| t¡ |D ]T}t|||d
\}}||7 }| |¡ |t||d |d
|7 }|d }q&|||d
7 }|S )Nr' r r )r ÚupdateÚ_special_namesr* rT ) ÚallstrrS ÚwritestrÚstructZoldendr! rJ Z
|
cleanedstrZdefsr r r r ¾ s
|
|
|
z8(\n|\A)\s*include\s*['\"](?P<name>[\w\d./\\]+\.src)['\"]c C s t j | ¡}t| x}g }|D ]h}t |¡}|r|| d¡}t j |¡sTt j ||¡}t j |¡rp|
|
t |¡ q| |¡ q | |¡ q W 5 Q R X |S )Nr# ) ÚosÚpathÚdirnameÚopenÚinclude_src_rer r4 Úisabsr5 ÚisfileÚextendÚresolve_includesr )ÚsourceÚdÚfidÚlinesÚliner Úfnr r r rb Ó s
|
|
|
rb c C s t | }td |¡S )Nr' )rb r r5 )rc rf 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 zt jd } W n tk
|
r. t j}t j}Y n*X t| d}tj | ¡\}}|}t|d}| ¡ }t
|
|}| |¡ d S )Nr rQ Úw) ÚsysÚargvÚ
|
IndexErrorÚstdinÚstdoutr] rZ r[ ÚsplitextÚreadr Úwrite)Úfilere ÚoutfileÚbaseÚextZnewnamerW rX r r r Úmainó s
|
|
rv Ú__main__) Ú__doc__Ú__all__rZ rj r( ÚcompileÚIr r r r rL r rK r& r* r2 r r r> rT r r^ rb r rV rv Ú__name__r r r r Ú<module> s4 .
|
|
|
|
|
|
|
<
|
|
