o
    ék)hð  ã                   @   sÐ   d dl Z d dlZ	 e dej¡Ze dej¡Ze dej¡Zdd„ Ze d¡Ze d¡Z	e d	¡Z
d
d„ Zdd„ Ze d¡Zdd„ Zdd„ Ze d¡Zdd„ Zdd„ Ze dej¡Zdd„ Zdd„ ZedƒZdS )é    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  | |¡}|du r	 |S | ¡ }t | || ¡ ¡r:	 |  d||¡}|dkr,n|}| ||d … dkr9nq |d7 }t | | ¡ ¡}|rN| ¡ d pQt| ƒ }}| 	||f¡ q)	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   TNÚ
éÿÿÿÿé   z
     $é   )
Úroutine_start_reÚsearchÚstartÚfunction_start_reÚmatchÚendÚrfindÚroutine_end_reÚlenÚappend)ÚastrZspanlistZindÚmr   Úir   © r   ú3lib/python3.10/site-packages/numpy/f2py/_src_pyf.pyÚparse_structure7   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 ]}|d  ¡ pt|ƒ}|d  dd¡}t|ƒ}|||< q	|S )Nr   r   ú\,ú@comma@)Únamed_reÚfindallÚstripÚ
unique_keyÚreplaceÚconv)r   ZrepsÚnamesZrepÚnameÚreplÚthelistr   r   r   Úfind_repl_patternsV   s   

r"   c                 C   s"   t | ƒ}t td| ¡d } | |fS )NÚ r   )r"   ÚreÚsubnr   )r   r   r   r   r   Úfind_and_remove_repl_patterns`   s   r&   z\A\\(?P<index>\d+)\Zc                 C   s`   |   d¡}dd„ |D ƒ}tt|ƒƒD ]}t || ¡}|r*t| d¡ƒ}|| ||< qd |¡S )Nú,c                 S   s   g | ]}|  ¡ ‘qS r   )r   )Ú.0Úxr   r   r   Ú
<listcomp>h   s    zconv.<locals>.<listcomp>Úindex)ÚsplitÚranger   Úitem_rer
   ÚintÚgroupÚjoin)r   ÚbÚlr   r   Újr   r   r   r   f   s   
€
r   c                 C   s>   t |  ¡ ƒ}d}d}|sd| }||v r|d7 }nd}|r|S )z( Obtain a unique key given a dictionary.Fr   z__l%sT)ÚlistÚkeys)ZadictZallkeysZdoneÚnZnewkeyr   r   r   r   p   s   
ûr   z\A\s*(\w[\w\d]*)\s*\Zc              	      sh  |   dd¡} |   dd¡} t| ƒ‰t d| ¡} ‡fdd„}t || ¡} d }d }i ‰t | ¡D ]Z}|ˆvr‡ˆ || |d ¡¡}|d u rGtd| ƒ‚||vrT| 	d	¡sT|||< d
d„ | 
d¡D ƒ}t|ƒ}|d u ro|}|ˆ|< |}q-||krx|ˆ|< q-td |d ˆ| ¡||¡ƒ q-ˆsŒ| S ‡ ‡fdd„}	d}
t|ƒD ]‰ |
t |	| ¡d 7 }
q™|
  dd¡}
|
  dd¡}
|
S )Nz\>z@rightarrow@z\<z@leftarrow@z<\1>c                    sj   t |  d¡ dd¡ƒ}t |¡rd| S d }ˆ  ¡ D ]
}ˆ | |kr$|}q|d u r1tˆ ƒ}|ˆ |< d| S )Nr   r   r   z<%s>)r   r0   r   Útemplate_name_rer
   r6   r   )Úmobjr!   r   Úkey)Úlnamesr   r   Úlistrepl…   s   
€zexpand_sub.<locals>.listreplzNo replicates found for <%s>Ú_c                 S   s   g | ]}|  d d¡‘qS )r   r'   )r   )r(   r   r   r   r   r*   Ÿ   s    zexpand_sub.<locals>.<listcomp>r'   zHMismatch in number of replacements (base <{}={}>) for <{}={}>. Ignoring.c                    s$   |   d¡}ˆ |ˆ d |g ¡ˆ  S )Nr   )r0   Úget)r9   r   )ÚkÚrulesr   r   Únamerepl®   s   
zexpand_sub.<locals>.namereplr#   z

ú>ú<)r   r"   r   ÚsubÚlist_reÚtemplate_rer   r>   Ú
ValueErrorÚ
startswithr,   r   ÚprintÚformatr1   r-   )Zsubstrr   r<   ZnumsubsZ	base_ruleÚrr!   ZruleZnumrA   Únewstrr   )r?   r;   r@   r   Ú
expand_sub   sJ   
ÿ€rM   c           	      C   s   | }d}t |ƒ}d}i }| t¡ |D ]*}t|||d … ƒ\}}||7 }| |¡ |t||d |d … |ƒ7 }|d }q|||d … 7 }|S )Nr#   r   r   )r   ÚupdateÚ_special_namesr&   rM   )	ZallstrrL   ZwritestrÚstructZoldendr   rD   Z
cleanedstrZdefsr   r   r   Úprocess_strº   s   


rQ   z8(\n|\A)\s*include\s*['\"](?P<name>[\w\d./\\]+\.src)['\"]c                 C   s¬   t j | ¡}t| ƒB}g }|D ]4}t |¡}|r>| d¡}t j |¡s*t j ||¡}t j 	|¡r8| 
t|ƒ¡ q| |¡ q| |¡ qW d   ƒ |S 1 sOw   Y  |S )Nr   )ÚosÚpathÚdirnameÚopenÚinclude_src_rer
   r0   Úisabsr1   ÚisfileÚextendÚresolve_includesr   )ÚsourceÚdZfidÚlinesÚliner   Úfnr   r   r   rZ   Ï   s$   


õ
þòrZ   c                 C   s   t | ƒ}td |¡ƒS )Nr#   )rZ   rQ   r1   )r[   r]   r   r   r   Úprocess_fileá   s   r`   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>
)rR   r$   ÚcompileÚIr   r   r	   r   rF   r   rE   r"   r&   r.   r   r   r8   rM   rQ   rV   rZ   r`   rO   r   r   r   r   Ú<module>   s,    -






;