a
    åzeô5  ã                   @   sd   d dl mZmZmZ G dd„ dejƒZG dd„ deƒZG dd„ deƒZG dd	„ d	eƒZ	ddd„Z
dS )é   )ÚgrammarÚtokenÚtokenizec                   @   s   e Zd ZdS )ÚPgenGrammarN)Ú__name__Ú
__module__Ú__qualname__© r	   r	   ú(/usr/lib/python3.9/lib2to3/pgen2/pgen.pyr      s   r   c                   @   s    e Zd Zd&dd„Zdd„ Zdd„ Zdd	„ Zd
d„ Zdd„ Zdd„ Z	dd„ Z
dd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zd'd d!„Zd"d#„ Zd$d%„ ZdS )(ÚParserGeneratorNc                 C   sl   d }|d u rt |ƒ}|j}|| _|| _t |j¡| _|  ¡  |  	¡ \| _
| _|d urZ|ƒ  i | _|  ¡  d S ©N)ÚopenÚcloseÚfilenameÚstreamr   Úgenerate_tokensÚreadlineÚ	generatorÚgettokenÚparseÚdfasÚstartsymbolÚfirstÚaddfirstsets)Úselfr   r   Zclose_streamr	   r	   r
   Ú__init__   s    zParserGenerator.__init__c              	   C   s  t ƒ }t| j ¡ ƒ}| ¡  | | j¡ | d| j¡ |D ]&}dt|j	ƒ }||j	|< ||j
|< q:|D ] }| j| }g }|D ]`}g }t|j ¡ ƒD ]$\}	}
| |  ||	¡| |
¡f¡ q’|jrÒ| d| |¡f¡ | |¡ q||j |¡ ||  ||¡f|j|j	| < qf|j	| j |_|S )Né    é   )r   Úlistr   ÚkeysÚsortÚremover   ÚinsertÚlenÚsymbol2numberZnumber2symbolÚsortedÚarcsÚitemsÚappendÚ
make_labelÚindexÚisfinalÚstatesÚ
make_firstÚstart)r   ÚcÚnamesÚnameÚiÚdfar,   Ústater&   ÚlabelÚnextr	   r	   r
   Úmake_grammar   s.    

zParserGenerator.make_grammarc                 C   s4   | j | }i }t|ƒD ]}|  ||¡}d||< q|S ©Nr   )r   r%   r)   )r   r/   r1   Zrawfirstr   r5   Úilabelr	   r	   r
   r-   4   s    

zParserGenerator.make_firstc                 C   s&  t |jƒ}|d  ¡ rš||jv rZ||jv r4|j| S |j |j| d f¡ ||j|< |S n>tt|d ƒ}||jv rz|j| S |j |d f¡ ||j|< |S nˆt	|ƒ}|d  ¡ rä||j
v rÂ|j
| S |j tj|f¡ ||j
|< |S n>tj| }||jv r|j| S |j |d f¡ ||j|< |S d S ©Nr   )r#   ÚlabelsÚisalphar$   Zsymbol2labelr(   Úgetattrr   ÚtokensÚevalÚkeywordsÚNAMEr   Zopmap)r   r/   r5   r9   ZitokenÚvaluer	   r	   r
   r)   =   s6    













zParserGenerator.make_labelc                 C   s8   t | j ¡ ƒ}| ¡  |D ]}|| jvr|  |¡ qd S r   )r   r   r   r    r   Ú	calcfirst)r   r0   r1   r	   r	   r
   r   k   s
    
zParserGenerator.addfirstsetsc              	   C   sþ   | j | }d | j|< |d }i }i }|j ¡ D ]x\}}|| j v r’|| jv rj| j| }|d u r~td| ƒ‚n|  |¡ | j| }| |¡ |||< q.d||< |di||< q.i }	| ¡ D ]:\}}
|
D ],}||	v rätd||||	| f ƒ‚||	|< qÀq´|| j|< d S )Nr   zrecursion for rule %rr   zArule %s is ambiguous; %s is in the first sets of %s as well as %s)r   r   r&   r'   Ú
ValueErrorrC   Úupdate)r   r1   r3   r4   ZtotalsetZoverlapcheckr5   r6   ÚfsetZinverseZitsfirstÚsymbolr	   r	   r
   rC   s   s4    








þzParserGenerator.calcfirstc           	      C   s    i }d }| j tjkr˜| j tjkr*|  ¡  q|  tj¡}|  tjd¡ |  ¡ \}}|  tj¡ |  	||¡}t
|ƒ}|  |¡ t
|ƒ}|||< |d u r|}q||fS )Nú:)Útyper   Ú	ENDMARKERÚNEWLINEr   ÚexpectrA   ÚOPÚ	parse_rhsÚmake_dfar#   Úsimplify_dfa)	r   r   r   r1   ÚaÚzr3   ZoldlenZnewlenr	   r	   r
   r   ‘   s"    

zParserGenerator.parsec              	      s¾   ‡ fdd„}‡ fdd„‰ t ||ƒ|ƒg}|D ]Œ}i }|jD ].}|jD ]"\}}	|d urDˆ |	| |i ¡ƒ qDq:t| ¡ ƒD ]@\}}
|D ]}|j|
kr‚ qªq‚t |
|ƒ}| |¡ | ||¡ qvq,|S )Nc                    s   i }ˆ | |ƒ |S r   r	   )r4   Úbase©Ú
addclosurer	   r
   Úclosure°   s    
z)ParserGenerator.make_dfa.<locals>.closurec                    s:   | |v rd S d|| < | j D ]\}}|d u rˆ ||ƒ qd S r8   ©r&   )r4   rS   r5   r6   rT   r	   r
   rU   ´   s    z,ParserGenerator.make_dfa.<locals>.addclosure)ÚDFAStateÚnfasetr&   Ú
setdefaultr%   r'   r(   Úaddarc)r   r.   ÚfinishrV   r,   r4   r&   Znfastater5   r6   rY   Ústr	   rT   r
   rO   ©   s"    



zParserGenerator.make_dfac           
      C   sš   t d|ƒ |g}t|ƒD ]|\}}t d|||u r2dp4dƒ |jD ]T\}}||v rZ| |¡}	nt|ƒ}	| |¡ |d u r‚t d|	 ƒ q>t d||	f ƒ q>qd S )NzDump of NFA forú  Stateú(final)Ú z	    -> %dú    %s -> %d)ÚprintÚ	enumerater&   r*   r#   r(   )
r   r1   r.   r\   Ztodor2   r4   r5   r6   Újr	   r	   r
   Údump_nfaÍ   s    

zParserGenerator.dump_nfac                 C   sd   t d|ƒ t|ƒD ]L\}}t d||jr*dp,dƒ t|j ¡ ƒD ]\}}t d|| |¡f ƒ q>qd S )NzDump of DFA forr^   r_   r`   ra   )rb   rc   r+   r%   r&   r'   r*   )r   r1   r3   r2   r4   r5   r6   r	   r	   r
   Údump_dfaÝ   s
    
zParserGenerator.dump_dfac                 C   sp   d}|rld}t |ƒD ]T\}}t|d t|ƒƒD ]8}|| }||kr.||= |D ]}| ||¡ qLd} qq.qqd S )NTFr   )rc   Úranger#   Ú
unifystate)r   r3   Zchangesr2   Zstate_ird   Zstate_jr4   r	   r	   r
   rP   ä   s    zParserGenerator.simplify_dfac                 C   s~   |   ¡ \}}| jdkr||fS tƒ }tƒ }| |¡ | |¡ | jdkrr|  ¡  |   ¡ \}}| |¡ | |¡ q>||fS d S )Nú|)Ú	parse_altrB   ÚNFAStater[   r   )r   rQ   rR   ZaaÚzzr	   r	   r
   rN   ù   s    




zParserGenerator.parse_rhsc                 C   sL   |   ¡ \}}| jdv s(| jtjtjfv rD|   ¡ \}}| |¡ |}q||fS )N)ú(ú[)Ú
parse_itemrB   rI   r   rA   ÚSTRINGr[   )r   rQ   Úbr/   Údr	   r	   r
   rj   
  s    
ÿ
zParserGenerator.parse_altc                 C   sŽ   | j dkr>|  ¡  |  ¡ \}}|  tjd¡ | |¡ ||fS |  ¡ \}}| j }|dvr`||fS |  ¡  | |¡ |dkr‚||fS ||fS d S )Nrn   ú])ú+Ú*rt   )rB   r   rN   rL   r   rM   r[   Ú
parse_atom)r   rQ   rR   rB   r	   r	   r
   ro     s    


zParserGenerator.parse_itemc                 C   s†   | j dkr4|  ¡  |  ¡ \}}|  tjd¡ ||fS | jtjtjfv rpt	ƒ }t	ƒ }| 
|| j ¡ |  ¡  ||fS |  d| j| j ¡ d S )Nrm   ú)z+expected (...) or NAME or STRING, got %s/%s)rB   r   rN   rL   r   rM   rI   rA   rp   rk   r[   Úraise_error)r   rQ   rR   r	   r	   r
   rv   (  s    
ÿzParserGenerator.parse_atomc                 C   sD   | j |ks|d ur2| j|kr2|  d||| j | j¡ | j}|  ¡  |S )Nzexpected %s/%s, got %s/%s)rI   rB   rx   r   )r   rI   rB   r	   r	   r
   rL   9  s    ÿzParserGenerator.expectc                 C   sF   t | jƒ}|d tjtjfv r*t | jƒ}q
|\| _| _| _| _| _	d S r:   )
r6   r   r   ÚCOMMENTÚNLrI   rB   ZbeginÚendÚline)r   Útupr	   r	   r
   r   A  s    
zParserGenerator.gettokenc              
   G   s^   |r8z|| }W n&   d  |gttt|ƒƒ ¡}Y n0 t|| j| jd | jd | jfƒ‚d S )Nú r   r   )Újoinr   ÚmapÚstrÚSyntaxErrorr   r{   r|   )r   ÚmsgÚargsr	   r	   r
   rx   H  s     ÿzParserGenerator.raise_error)N)N)r   r   r   r   r7   r-   r)   r   rC   r   rO   re   rf   rP   rN   rj   ro   rv   rL   r   rx   r	   r	   r	   r
   r   
   s$   
	.$

r   c                   @   s   e Zd Zdd„ Zddd„ZdS )rk   c                 C   s
   g | _ d S r   rW   )r   r	   r	   r
   r   S  s    zNFAState.__init__Nc                 C   s   | j  ||f¡ d S r   )r&   r(   ©r   r6   r5   r	   r	   r
   r[   V  s    zNFAState.addarc)N)r   r   r   r   r[   r	   r	   r	   r
   rk   Q  s   rk   c                   @   s0   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	Zd	S )
rX   c                 C   s   || _ ||v | _i | _d S r   )rY   r+   r&   )r   rY   Úfinalr	   r	   r
   r   ]  s    
zDFAState.__init__c                 C   s   || j |< d S r   rW   r…   r	   r	   r
   r[   e  s    zDFAState.addarcc                 C   s*   | j  ¡ D ]\}}||u r
|| j |< q
d S r   )r&   r'   )r   ÚoldÚnewr5   r6   r	   r	   r
   rh   k  s    zDFAState.unifystatec                 C   sV   | j |j krdS t| jƒt|jƒkr(dS | j ¡ D ]\}}||j |¡ur2 dS q2dS )NFT)r+   r#   r&   r'   Úget)r   Úotherr5   r6   r	   r	   r
   Ú__eq__p  s    zDFAState.__eq__N)r   r   r   r   r[   rh   r‹   Ú__hash__r	   r	   r	   r
   rX   [  s
   rX   úGrammar.txtc                 C   s   t | ƒ}| ¡ S r   )r   r7   )r   Úpr	   r	   r
   Úgenerate_grammar€  s    r   N)r   )r`   r   r   r   ZGrammarr   Úobjectr   rk   rX   r   r	   r	   r	   r
   Ú<module>   s     I
%