a
    d                     @   s`   d dl T d dlmZ d dlZejd dkr8d dlmZ nd dlmZ dd ZG d	d
 d
e	Z
dS )    )*)StringION      )TextIOc                  C   s$  t  } | d | d | d | d | d | d | d | d | d	 | d
 | d | d | d | d | d | d | d | d | d | d | d | d | d | d | d | d | d | d | d | d | d | d  | d! | d" | d# | d$ | d% | d& | d' | d( | d) | d* | d+ | d, | d- | d. | d/ | d0 | d1 | d2 | d3 | d4 | d5 | d6 | d7 | d8 | d9 | d: | d; | d< | d= | d> | d? | d@ | dA | dB | dC | dD | dE | dF | dG | dH | dI | dJ | dK |  W  d    S 1 s0    Y  d S )LNu   悋Ꜫ脳맭䅼㯧瞆奤u   ·						z				
	
			z				z/
0z7
;
zC
FH
z				S
		U
		z	X				\
		^
			_	b
	z


f

j
mzt
z|
u   
u   

u   

u    
u   ®
u   ¯³
´z
 u   	
Î"%.z6
8>Ku   aegu   ­u    ²"##$$z%&&''(/
)/z*/+/,/-/.(z .).*.+.,.-/0z0.01127z3747
57z62636465z7	8:9;	:9:;z;<<==>G?Dz@AACB@CFDBz!DEEHFDG?GHHIzIJJKL LMMNzNOVPRQSRQz RSSUTPUXVTzVWWbXVY[Z\[zZ[\\^]Y^__]z_``baOa]bzcfdfecedfzgkhjihjmkizkllsmkntopzpqqrrtsnsozt}uvv|wxxyzyzz|{u{w|u   }{}~~}u   u   u   	u   u   
u   u   
u   u   
u   u   u   u   	u     u    ¡¡¢¢u   £®¤®¥®u   ¦®§®¨u   ®©®ª®u   «®¬®
­£u   ­¤­¥­u   ¦­§­¨u   ­©­ª­«u   ­¬®¯¯­u   ¯°°±³u   	²±³´´u   ²´µµ!.u   06:DGRV[_aeks{}u   ­¯´)r   writegetvalue)buf r
   u/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/omegaconf/grammar/gen/OmegaConfGrammarParser.pyserializedATN   s    











































































r   c                       s  e Zd ZdZe e Zdd eej	D Z
e Zg dZg dZdZdZdZd	Zd
ZdZdZdZdZdZdZdZdZdZdZdZg dZe j!Z!dZ"dZ#d	Z$d
Z%dZ&dZ'dZ(dZ)dZ*dZ+dZ,dZ-dZ.dZ/dZ0dZ1dZ2dZ3dZ4dZ5dZ6dZ7dZ8dZ9d Z:d!Z;d"Z<d#Z=e>j?fe@eAd$ fd%d&ZBG d'd( d(eCZDd)d* ZEG d+d, d,eCZFd-d. ZGG d/d0 d0eCZHd1d2 ZIG d3d4 d4eCZJd5d6 ZKG d7d8 d8eCZLd9d: ZMG d;d< d<eCZNd=d> ZOG d?d@ d@eCZPdAdB ZQG dCdD dDeCZRdEdF ZSG dGdH dHeCZTdIdJ ZUG dKdL dLeCZVdMdN ZWG dOdP dPeCZXdQdR ZYG dSdT dTeCZZdUdV Z[G dWdX dXeCZ\dYdZ Z]G d[d\ d\eCZ^d]d^ Z_G d_d` d`eCZ`dadb ZaG dcdd ddeCZbdedf Zc  ZdS )gOmegaConfGrammarParserzOmegaConfGrammarParser.g4c                 C   s   g | ]\}}t ||qS r
   )ZDFA).0iZdsr
   r
   r   
<listcomp>b       z!OmegaConfGrammarParser.<listcomp>)	<INVALID>r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   z'.'r   r   r   r   z'['z']')r   ANY_STR	ESC_INTERTOP_ESC
INTER_OPEN
BRACE_OPENBRACE_CLOSEQUOTE_OPEN_SINGLEQUOTE_OPEN_DOUBLECOMMABRACKET_OPENBRACKET_CLOSECOLONFLOATINTBOOLNULLUNQUOTED_CHARIDESCWSINTER_CLOSEDOT	INTER_KEYMATCHING_QUOTE_CLOSE
QUOTED_ESCDOLLARINTER_BRACKET_OPENINTER_BRACKET_CLOSEr   r            r            	   
                  )configValuesingleElementtextelementlistContainerdictContainerdictKeyValuePairsequenceinterpolationinterpolationNodeinterpolationResolver	configKeyresolverNamequotedValue	primitivedictKey                                       )inputoutputc                    s8   t  || | d t| | j| j| j| _d | _d S )Nz4.9.3)	super__init__ZcheckVersionZParserATNSimulatoratndecisionsToDFAsharedContextCache_interpZ_predicates)selfrY   rZ   	__class__r
   r   r\      s    
zOmegaConfGrammarParser.__init__c                       sl   e Zd ZdZdeed fddZdd Zd	d
 Zdd Z	e
dddZe
dddZedddZ  ZS )z)OmegaConfGrammarParser.ConfigValueContextparserNparentinvokingStatec                    s   t  || || _d S Nr[   r\   rd   ra   rd   rg   rh   rb   r
   r   r\      s    z2OmegaConfGrammarParser.ConfigValueContext.__init__c                 C   s   |  tjdS Nr   getTypedRuleContextr   TextContextra   r
   r
   r   r>      s    z.OmegaConfGrammarParser.ConfigValueContext.textc                 C   s   |  tjdS rl   getTokenr   EOFrp   r
   r
   r   rs      s    z-OmegaConfGrammarParser.ConfigValueContext.EOFc                 C   s   t jS ri   )r   RULE_configValuerp   r
   r
   r   getRuleIndex   s    z6OmegaConfGrammarParser.ConfigValueContext.getRuleIndexlistenerc                 C   s   t |dr||  d S )NenterConfigValue)hasattrrx   ra   rw   r
   r
   r   	enterRule   s    
z3OmegaConfGrammarParser.ConfigValueContext.enterRulec                 C   s   t |dr||  d S )NexitConfigValue)ry   r|   rz   r
   r
   r   exitRule   s    
z2OmegaConfGrammarParser.ConfigValueContext.exitRulevisitorc                 C   s"   t |dr|| S || S d S )NvisitConfigValue)ry   r   visitChildrenra   r   r
   r
   r   accept   s    

z0OmegaConfGrammarParser.ConfigValueContext.accept)Nre   )__name__
__module____qualname__	__slots__ParserRuleContextintr\   r>   rs   ru   ParseTreeListenerr{   r}   ParseTreeVisitorr   __classcell__r
   r
   rb   r   ConfigValueContext   s   r   c              
   C   s   t | | j| j}| |d| j zz0| |d d| _|   d| _| t j	 W nF t
y } z.||_| j| | | j| | W Y d }~n
d }~0 0 W |   n
|   0 |S )Nr   r       !   )r   r   _ctxstater{   rt   enterOuterAltr>   matchrs   RecognitionException	exception_errHandlerreportErrorrecoverr}   ra   localctxrer
   r
   r   r<      s    &z"OmegaConfGrammarParser.configValuec                       sl   e Zd ZdZdeed fddZdd Zd	d
 Zdd Z	e
dddZe
dddZedddZ  ZS )z+OmegaConfGrammarParser.SingleElementContextrd   Nre   rf   c                    s   t  || || _d S ri   rj   rk   rb   r
   r   r\      s    z4OmegaConfGrammarParser.SingleElementContext.__init__c                 C   s   |  tjdS rl   rn   r   ElementContextrp   r
   r
   r   r?      s    z3OmegaConfGrammarParser.SingleElementContext.elementc                 C   s   |  tjdS rl   rq   rp   r
   r
   r   rs      s    z/OmegaConfGrammarParser.SingleElementContext.EOFc                 C   s   t jS ri   )r   RULE_singleElementrp   r
   r
   r   ru      s    z8OmegaConfGrammarParser.SingleElementContext.getRuleIndexrv   c                 C   s   t |dr||  d S )NenterSingleElement)ry   r   rz   r
   r
   r   r{      s    
z5OmegaConfGrammarParser.SingleElementContext.enterRulec                 C   s   t |dr||  d S )NexitSingleElement)ry   r   rz   r
   r
   r   r}      s    
z4OmegaConfGrammarParser.SingleElementContext.exitRuler~   c                 C   s"   t |dr|| S || S d S )NvisitSingleElement)ry   r   r   r   r
   r
   r   r     s    

z2OmegaConfGrammarParser.SingleElementContext.accept)Nre   )r   r   r   r   r   r   r\   r?   rs   ru   r   r{   r}   r   r   r   r
   r
   rb   r   SingleElementContext   s   r   c              
   C   s   t | | j| j}| |d| j zz0| |d d| _|   d| _| t j	 W nF t
y } z.||_| j| | | j| | W Y d }~n
d }~0 0 W |   n
|   0 |S )Nr/   r   #   $   )r   r   r   r   r{   r   r   r?   r   rs   r   r   r   r   r   r}   r   r
   r
   r   r=   
  s    &z$OmegaConfGrammarParser.singleElementc                       s   e Zd ZdZdeed fddZdeddd	Zd edd
dZd!edddZ	d"edddZ
d#edddZd$edddZdd ZedddZedddZedddZ  ZS )%z"OmegaConfGrammarParser.TextContextrd   Nre   rf   c                    s   t  || || _d S ri   rj   rk   rb   r
   r   r\      s    z+OmegaConfGrammarParser.TextContext.__init__r   c                 C   s&   |d u r|  tjS | tj|S d S ri   getTypedRuleContextsr   InterpolationContextrn   ra   r   r
   r
   r   rD   $  s    z0OmegaConfGrammarParser.TextContext.interpolationc                 C   s&   |d u r|  tjS | tj|S d S ri   )	getTokensr   r   rr   r   r
   r
   r   r   +  s    z*OmegaConfGrammarParser.TextContext.ANY_STRc                 C   s&   |d u r|  tjS | tj|S d S ri   r   r   r%   rr   r   r
   r
   r   r%   1  s    z&OmegaConfGrammarParser.TextContext.ESCc                 C   s&   |d u r|  tjS | tj|S d S ri   )r   r   r   rr   r   r
   r
   r   r   7  s    z,OmegaConfGrammarParser.TextContext.ESC_INTERc                 C   s&   |d u r|  tjS | tj|S d S ri   )r   r   r   rr   r   r
   r
   r   r   =  s    z*OmegaConfGrammarParser.TextContext.TOP_ESCc                 C   s&   |d u r|  tjS | tj|S d S ri   )r   r   r+   rr   r   r
   r
   r   r+   C  s    z-OmegaConfGrammarParser.TextContext.QUOTED_ESCc                 C   s   t jS ri   )r   	RULE_textrp   r
   r
   r   ru   I  s    z/OmegaConfGrammarParser.TextContext.getRuleIndexrv   c                 C   s   t |dr||  d S )N	enterText)ry   r   rz   r
   r
   r   r{   L  s    
z,OmegaConfGrammarParser.TextContext.enterRulec                 C   s   t |dr||  d S )NexitText)ry   r   rz   r
   r
   r   r}   P  s    
z+OmegaConfGrammarParser.TextContext.exitRuler~   c                 C   s"   t |dr|| S || S d S )N	visitText)ry   r   r   r   r
   r
   r   r   T  s    

z)OmegaConfGrammarParser.TextContext.accept)Nre   )N)N)N)N)N)N)r   r   r   r   r   r   r\   rD   r   r%   r   r   r+   ru   r   r{   r}   r   r   r   r
   r
   rb   r   ro     s   ro   c              
   C   s   t | | j| j}| |d| j d| _zz| |d d| _| j	|  | j
d}d| _| j	|  | j
d}|t jfv rd| _|   n|t jfv rd| _| t j n|t jfv rd| _| t j nl|t jfv rd| _| t j nL|t jfv rd	| _| t j n*|t jfv r8d
| _| t j nt| d| _| j	|  | j
d}|d@ dkrd|> dt j> dt j> B dt j> B dt j> B dt j> B dt j> B @ dksZqqZW nH ty } z.||_| j| | | j| | W Y d }~n
d }~0 0 W |   n
|   0 |S )Nr1   r   r   ,   &   '   (   )   *   +   .   )r   ro   r   r   r{   r   _lar   r   sync_inputLAr   rD   r   r   r%   r   r   r+   NoViableAltExceptionr   r   r   r   r}   ra   r   r   tokenr   r
   r
   r   r>   ]  s^    V
&zOmegaConfGrammarParser.textc                       s|   e Zd ZdZdeed fddZdd Zd	d
 Zdd Z	dd Z
dd ZedddZedddZedddZ  ZS )z%OmegaConfGrammarParser.ElementContextrd   Nre   rf   c                    s   t  || || _d S ri   rj   rk   rb   r
   r   r\     s    z.OmegaConfGrammarParser.ElementContext.__init__c                 C   s   |  tjdS rl   )rn   r   PrimitiveContextrp   r
   r
   r   rJ     s    z/OmegaConfGrammarParser.ElementContext.primitivec                 C   s   |  tjdS rl   )rn   r   QuotedValueContextrp   r
   r
   r   rI     s    z1OmegaConfGrammarParser.ElementContext.quotedValuec                 C   s   |  tjdS rl   )rn   r   ListContainerContextrp   r
   r
   r   r@     s    z3OmegaConfGrammarParser.ElementContext.listContainerc                 C   s   |  tjdS rl   )rn   r   DictContainerContextrp   r
   r
   r   rA     s    z3OmegaConfGrammarParser.ElementContext.dictContainerc                 C   s   t jS ri   )r   RULE_elementrp   r
   r
   r   ru     s    z2OmegaConfGrammarParser.ElementContext.getRuleIndexrv   c                 C   s   t |dr||  d S )NenterElement)ry   r   rz   r
   r
   r   r{     s    
z/OmegaConfGrammarParser.ElementContext.enterRulec                 C   s   t |dr||  d S )NexitElement)ry   r   rz   r
   r
   r   r}     s    
z.OmegaConfGrammarParser.ElementContext.exitRuler~   c                 C   s"   t |dr|| S || S d S )NvisitElement)ry   r   r   r   r
   r
   r   r     s    

z,OmegaConfGrammarParser.ElementContext.accept)Nre   )r   r   r   r   r   r   r\   rJ   rI   r@   rA   ru   r   r{   r}   r   r   r   r
   r
   rb   r   r     s   r   c                 C   s~  t | | j| j}| |d| j zJzd| _| j|  | j	d}|t j
t jt jt jt jt jt jt jt jt jf
v r| |d d| _|   n|t jt jfv r| |d d| _|   nZ|t jfv r| |d d| _|   n2|t jfv r| |d	 d
| _|   nt| W nH tyb } z.||_| j | | | j!| | W Y d }~n
d }~0 0 W | "  n
| "  0 |S )Nr2   4   r   0   r/   1   r0   2   r1   3   )#r   r   r   r   r{   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r   rJ   r   r   rI   r   r@   r   rA   r   r   r   r   r   r}   ra   r   r   r   r
   r
   r   r?     sB    0&zOmegaConfGrammarParser.elementc                       st   e Zd ZdZdeed fddZdd Zd	d
 Zdd Z	dd Z
edddZedddZedddZ  ZS )z+OmegaConfGrammarParser.ListContainerContextrd   Nre   rf   c                    s   t  || || _d S ri   rj   rk   rb   r
   r   r\     s    z4OmegaConfGrammarParser.ListContainerContext.__init__c                 C   s   |  tjdS rl   )rr   r   r   rp   r
   r
   r   r     s    z8OmegaConfGrammarParser.ListContainerContext.BRACKET_OPENc                 C   s   |  tjdS rl   )rr   r   r   rp   r
   r
   r   r     s    z9OmegaConfGrammarParser.ListContainerContext.BRACKET_CLOSEc                 C   s   |  tjdS rl   rn   r   SequenceContextrp   r
   r
   r   rC     s    z4OmegaConfGrammarParser.ListContainerContext.sequencec                 C   s   t jS ri   )r   RULE_listContainerrp   r
   r
   r   ru     s    z8OmegaConfGrammarParser.ListContainerContext.getRuleIndexrv   c                 C   s   t |dr||  d S )NenterListContainer)ry   r   rz   r
   r
   r   r{     s    
z5OmegaConfGrammarParser.ListContainerContext.enterRulec                 C   s   t |dr||  d S )NexitListContainer)ry   r   rz   r
   r
   r   r}      s    
z4OmegaConfGrammarParser.ListContainerContext.exitRuler~   c                 C   s"   t |dr|| S || S d S )NvisitListContainer)ry   r   r   r   r
   r
   r   r     s    

z2OmegaConfGrammarParser.ListContainerContext.accept)Nre   )r   r   r   r   r   r   r\   r   r   rC   ru   r   r{   r}   r   r   r   r
   r
   rb   r   r     s   r   c              
   C   s  t | | j| j}| |d| j d| _zjz| |d d| _| t j	 d| _| j
|  | jd}|d@ dkr,d|> dt j> dt j> B dt j> B dt j> B dt j> B dt j	> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B @ dkr,d| _|   d| _| t j W nH ty } z.||_| j
 | | | j
!| | W Y d }~n
d }~0 0 W | "  n
| "  0 |S )	Nr4   r   r   6   8   r   7   :   )#r   r   r   r   r{   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   rC   r   r   r   r   r   r}   ra   r   r   r   r
   r
   r   r@     s*    &z$OmegaConfGrammarParser.listContainerc                       s   e Zd ZdZdeed fddZdd Zd	d
 ZdedddZ	dedddZ
dd ZedddZedddZedddZ  ZS )z+OmegaConfGrammarParser.DictContainerContextrd   Nre   rf   c                    s   t  || || _d S ri   rj   rk   rb   r
   r   r\   ,  s    z4OmegaConfGrammarParser.DictContainerContext.__init__c                 C   s   |  tjdS rl   )rr   r   r   rp   r
   r
   r   r   0  s    z6OmegaConfGrammarParser.DictContainerContext.BRACE_OPENc                 C   s   |  tjdS rl   rr   r   r   rp   r
   r
   r   r   3  s    z7OmegaConfGrammarParser.DictContainerContext.BRACE_CLOSEr   c                 C   s&   |d u r|  tjS | tj|S d S ri   )r   r   DictKeyValuePairContextrn   r   r
   r
   r   rB   6  s    z<OmegaConfGrammarParser.DictContainerContext.dictKeyValuePairc                 C   s&   |d u r|  tjS | tj|S d S ri   r   r   r   rr   r   r
   r
   r   r   =  s    z1OmegaConfGrammarParser.DictContainerContext.COMMAc                 C   s   t jS ri   )r   RULE_dictContainerrp   r
   r
   r   ru   C  s    z8OmegaConfGrammarParser.DictContainerContext.getRuleIndexrv   c                 C   s   t |dr||  d S )NenterDictContainer)ry   r   rz   r
   r
   r   r{   F  s    
z5OmegaConfGrammarParser.DictContainerContext.enterRulec                 C   s   t |dr||  d S )NexitDictContainer)ry   r   rz   r
   r
   r   r}   J  s    
z4OmegaConfGrammarParser.DictContainerContext.exitRuler~   c                 C   s"   t |dr|| S || S d S )NvisitDictContainer)ry   r   r   r   r
   r
   r   r   N  s    

z2OmegaConfGrammarParser.DictContainerContext.accept)Nre   )N)N)r   r   r   r   r   r   r\   r   r   rB   r   ru   r   r{   r}   r   r   r   r
   r
   rb   r   r   )  s   r   c              
   C   s  t | | j| j}| |d| j d| _zz8| |d d| _| t j	 d| _| j
|  | jd}|d@ dkrRd|> dt j> dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B @ dkrRd| _|   d| _| j
|  | jd}|t jkrRd	| _| t j d
| _|   d| _| j
|  | jd}qd| _| t j W nH ty } z.||_| j
| | | j
| | W Y d }~n
d }~0 0 W |   n
|   0 |S )Nr6   r   r   <   E   r   =   B   >   ?   D   G   )r   r   r   r   r{   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   rB   r   r   r   r   r   r   r}   r   r
   r
   r   rA   W  s@    l&z$OmegaConfGrammarParser.dictContainerc                       st   e Zd ZdZdeed fddZdd Zd	d
 Zdd Z	dd Z
edddZedddZedddZ  ZS )z.OmegaConfGrammarParser.DictKeyValuePairContextrd   Nre   rf   c                    s   t  || || _d S ri   rj   rk   rb   r
   r   r\     s    z7OmegaConfGrammarParser.DictKeyValuePairContext.__init__c                 C   s   |  tjdS rl   )rn   r   DictKeyContextrp   r
   r
   r   rK     s    z6OmegaConfGrammarParser.DictKeyValuePairContext.dictKeyc                 C   s   |  tjdS rl   rr   r   r   rp   r
   r
   r   r     s    z4OmegaConfGrammarParser.DictKeyValuePairContext.COLONc                 C   s   |  tjdS rl   r   rp   r
   r
   r   r?     s    z6OmegaConfGrammarParser.DictKeyValuePairContext.elementc                 C   s   t jS ri   )r   RULE_dictKeyValuePairrp   r
   r
   r   ru     s    z;OmegaConfGrammarParser.DictKeyValuePairContext.getRuleIndexrv   c                 C   s   t |dr||  d S )NenterDictKeyValuePair)ry   r   rz   r
   r
   r   r{     s    
z8OmegaConfGrammarParser.DictKeyValuePairContext.enterRulec                 C   s   t |dr||  d S )NexitDictKeyValuePair)ry   r   rz   r
   r
   r   r}     s    
z7OmegaConfGrammarParser.DictKeyValuePairContext.exitRuler~   c                 C   s"   t |dr|| S || S d S )NvisitDictKeyValuePair)ry   r   r   r   r
   r
   r   r     s    

z5OmegaConfGrammarParser.DictKeyValuePairContext.accept)Nre   )r   r   r   r   r   r   r\   rK   r   r?   ru   r   r{   r}   r   r   r   r
   r
   rb   r   r     s   r   c              
   C   s   t | | j| j}| |d| j zz>| |d d| _|   d| _| t j	 d| _| 
  W nF ty } z.||_| j| | | j| | W Y d }~n
d }~0 0 W |   n
|   0 |S )Nr8   r   I   J   K   )r   r   r   r   r{   r   r   rK   r   r   r?   r   r   r   r   r   r}   r   r
   r
   r   rB     s     &z'OmegaConfGrammarParser.dictKeyValuePairc                       s|   e Zd ZdZdeed fddZdeddd	Zdedd
dZdd Z	e
dddZe
dddZedddZ  ZS )z&OmegaConfGrammarParser.SequenceContextrd   Nre   rf   c                    s   t  || || _d S ri   rj   rk   rb   r
   r   r\     s    z/OmegaConfGrammarParser.SequenceContext.__init__r   c                 C   s&   |d u r|  tjS | tj|S d S ri   )r   r   r   rn   r   r
   r
   r   r?     s    z.OmegaConfGrammarParser.SequenceContext.elementc                 C   s&   |d u r|  tjS | tj|S d S ri   r   r   r
   r
   r   r     s    z,OmegaConfGrammarParser.SequenceContext.COMMAc                 C   s   t jS ri   )r   RULE_sequencerp   r
   r
   r   ru     s    z3OmegaConfGrammarParser.SequenceContext.getRuleIndexrv   c                 C   s   t |dr||  d S )NenterSequence)ry   r   rz   r
   r
   r   r{     s    
z0OmegaConfGrammarParser.SequenceContext.enterRulec                 C   s   t |dr||  d S )NexitSequence)ry   r   rz   r
   r
   r   r}     s    
z/OmegaConfGrammarParser.SequenceContext.exitRuler~   c                 C   s"   t |dr|| S || S d S )NvisitSequence)ry   r   r   r   r
   r
   r   r     s    

z-OmegaConfGrammarParser.SequenceContext.accept)Nre   )N)N)r   r   r   r   r   r   r\   r?   r   ru   r   r{   r}   r   r   r   r
   r
   rb   r   r     s   r   c                 C   s  t | | j| j}| |d| j d| _zdzd| _| j|  | j	
d}|t jt jt jt jt jt jt jt jt jt jt jt jt jt jfv r| |d d| _|   d| _| j|  | j	
d}|t jkr8d| _| t j d| _| j|  | j	
d}|d	@ dkrd|> dt j> dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B @ dkrd
| _|   d| _| j|  | j	
d}qȐnZ|t jfv r0| |d d| _| j|  | j	
d}d| _| t j d| _| j|  | j	
d}|d	@ dkrd|> dt j> dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B @ dkrd| _|   d| _| j|  | j	
d}|t jksq8qnt| W nH ty } z.||_| j | | | j!| | W Y d }~n
d }~0 0 W | "  n
| "  0 |S )Nr:   r   _   r   M   T   N   P   r   O   V   r/   [   W   Y   X   ]   )#r   r   r   r   r{   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r   r?   r   r   r   r   r   r   r   r}   ra   r   r   r   r   r
   r
   r   rC     sj    B&zOmegaConfGrammarParser.sequencec                       sl   e Zd ZdZdeed fddZdd Zd	d
 Zdd Z	e
dddZe
dddZedddZ  ZS )z+OmegaConfGrammarParser.InterpolationContextrd   Nre   rf   c                    s   t  || || _d S ri   rj   rk   rb   r
   r   r\   (  s    z4OmegaConfGrammarParser.InterpolationContext.__init__c                 C   s   |  tjdS rl   )rn   r   InterpolationNodeContextrp   r
   r
   r   rE   ,  s    z=OmegaConfGrammarParser.InterpolationContext.interpolationNodec                 C   s   |  tjdS rl   )rn   r   InterpolationResolverContextrp   r
   r
   r   rF   0  s    zAOmegaConfGrammarParser.InterpolationContext.interpolationResolverc                 C   s   t jS ri   )r   RULE_interpolationrp   r
   r
   r   ru   4  s    z8OmegaConfGrammarParser.InterpolationContext.getRuleIndexrv   c                 C   s   t |dr||  d S )NenterInterpolation)ry   r  rz   r
   r
   r   r{   7  s    
z5OmegaConfGrammarParser.InterpolationContext.enterRulec                 C   s   t |dr||  d S )NexitInterpolation)ry   r  rz   r
   r
   r   r}   ;  s    
z4OmegaConfGrammarParser.InterpolationContext.exitRuler~   c                 C   s"   t |dr|| S || S d S )NvisitInterpolation)ry   r  r   r   r
   r
   r   r   ?  s    

z2OmegaConfGrammarParser.InterpolationContext.accept)Nre   )r   r   r   r   r   r   r\   rE   rF   ru   r   r{   r}   r   r   r   r
   r
   rb   r   r   %  s   r   c              
   C   s   t | | j| j}| |d| j zzpd| _| j|  | j	| j
d| j}|dkrp| |d d| _|   n"|dkr| |d d| _|   W nF ty } z.||_| j| | | j| | W Y d }~n
d }~0 0 W |   n
|   0 |S )NrL   c   r7   r   a   r/   b   )r   r   r   r   r{   r  r   r   r`   ZadaptivePredictr   r   rE   rF   r   r   r   r   r}   )ra   r   Zla_r   r
   r
   r   rD   H  s,    &z$OmegaConfGrammarParser.interpolationc                       s   e Zd ZdZdeed fddZdd Zd	d
 ZdedddZ	d edddZ
d!edddZd"edddZdd ZedddZedddZedddZ  ZS )#z/OmegaConfGrammarParser.InterpolationNodeContextrd   Nre   rf   c                    s   t  || || _d S ri   rj   rk   rb   r
   r   r\   i  s    z8OmegaConfGrammarParser.InterpolationNodeContext.__init__c                 C   s   |  tjdS rl   rr   r   r   rp   r
   r
   r   r   m  s    z:OmegaConfGrammarParser.InterpolationNodeContext.INTER_OPENc                 C   s   |  tjdS rl   )rr   r   r'   rp   r
   r
   r   r'   p  s    z;OmegaConfGrammarParser.InterpolationNodeContext.INTER_CLOSEr   c                 C   s&   |d u r|  tjS | tj|S d S ri   )r   r   ConfigKeyContextrn   r   r
   r
   r   rG   s  s    z9OmegaConfGrammarParser.InterpolationNodeContext.configKeyc                 C   s&   |d u r|  tjS | tj|S d S ri   )r   r   r   rr   r   r
   r
   r   r   z  s    z<OmegaConfGrammarParser.InterpolationNodeContext.BRACKET_OPENc                 C   s&   |d u r|  tjS | tj|S d S ri   )r   r   r   rr   r   r
   r
   r   r     s    z=OmegaConfGrammarParser.InterpolationNodeContext.BRACKET_CLOSEc                 C   s&   |d u r|  tjS | tj|S d S ri   r   r   r(   rr   r   r
   r
   r   r(     s    z3OmegaConfGrammarParser.InterpolationNodeContext.DOTc                 C   s   t jS ri   )r   RULE_interpolationNoderp   r
   r
   r   ru     s    z<OmegaConfGrammarParser.InterpolationNodeContext.getRuleIndexrv   c                 C   s   t |dr||  d S )NenterInterpolationNode)ry   r  rz   r
   r
   r   r{     s    
z9OmegaConfGrammarParser.InterpolationNodeContext.enterRulec                 C   s   t |dr||  d S )NexitInterpolationNode)ry   r  rz   r
   r
   r   r}     s    
z8OmegaConfGrammarParser.InterpolationNodeContext.exitRuler~   c                 C   s"   t |dr|| S || S d S )NvisitInterpolationNode)ry   r  r   r   r
   r
   r   r     s    

z6OmegaConfGrammarParser.InterpolationNodeContext.accept)Nre   )N)N)N)N)r   r   r   r   r   r   r\   r   r'   rG   r   r   r(   ru   r   r{   r}   r   r   r   r
   r
   rb   r   r  f  s   r  c              
   C   s  t | | j| j}| |d| j d| _zbz
| |d d| _| t j	 d| _| j
|  | jd}|t jkrd| _| t j d| _| j
|  | jd}qld| _| j
|  | jd}|t j	t jt jfv rd	| _|   nJ|t jfv r,d
| _| t j d| _|   d| _| t j nt| d| _| j
|  | jd}|t jksj|t jkr$d| _| j
|  | jd}|t jfv rd| _| t j d| _|   nJ|t jfv rd| _| t j d| _|   d| _| t j nt| d| _| j
|  | jd}qRd| _| t j W nH ty } z.||_| j
| | | j
| | W Y d }~n
d }~0 0 W |   n
|   0 |S )NrN   r   r   e   i   f   k   q   l   m   n   o   {   y   s   t   u   v   w   }   ~   )r   r  r   r   r{   r  r   r   r   r   r   r   r   r   r(   r$   r)   rG   r   r   r   r'   r   r   r   r   r}   r   r
   r
   r   rE     s    
&z(OmegaConfGrammarParser.interpolationNodec                       s   e Zd ZdZdeed fddZdd Zd	d
 Zdd Z	dd Z
dd Zdd ZedddZedddZedddZ  ZS )z3OmegaConfGrammarParser.InterpolationResolverContextrd   Nre   rf   c                    s   t  || || _d S ri   rj   rk   rb   r
   r   r\     s    z<OmegaConfGrammarParser.InterpolationResolverContext.__init__c                 C   s   |  tjdS rl   r  rp   r
   r
   r   r     s    z>OmegaConfGrammarParser.InterpolationResolverContext.INTER_OPENc                 C   s   |  tjdS rl   )rn   r   ResolverNameContextrp   r
   r
   r   rH     s    z@OmegaConfGrammarParser.InterpolationResolverContext.resolverNamec                 C   s   |  tjdS rl   r   rp   r
   r
   r   r     s    z9OmegaConfGrammarParser.InterpolationResolverContext.COLONc                 C   s   |  tjdS rl   r   rp   r
   r
   r   r     s    z?OmegaConfGrammarParser.InterpolationResolverContext.BRACE_CLOSEc                 C   s   |  tjdS rl   r   rp   r
   r
   r   rC      s    z<OmegaConfGrammarParser.InterpolationResolverContext.sequencec                 C   s   t jS ri   )r   RULE_interpolationResolverrp   r
   r
   r   ru     s    z@OmegaConfGrammarParser.InterpolationResolverContext.getRuleIndexrv   c                 C   s   t |dr||  d S )NenterInterpolationResolver)ry   r&  rz   r
   r
   r   r{     s    
z=OmegaConfGrammarParser.InterpolationResolverContext.enterRulec                 C   s   t |dr||  d S )NexitInterpolationResolver)ry   r'  rz   r
   r
   r   r}     s    
z<OmegaConfGrammarParser.InterpolationResolverContext.exitRuler~   c                 C   s"   t |dr|| S || S d S )NvisitInterpolationResolver)ry   r(  r   r   r
   r
   r   r     s    

z:OmegaConfGrammarParser.InterpolationResolverContext.accept)Nre   )r   r   r   r   r   r   r\   r   rH   r   r   rC   ru   r   r{   r}   r   r   r   r
   r
   rb   r   r    s   r  c              
   C   s  t | | j| j}| |d| j d| _zz2| |d d| _| t j	 d| _| 
  d| _| t j d| _| j|  | jd}|d@ dkrLd|> dt j	> dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B @ dkrLd	| _|   d
| _| t j W nH ty } z.||_ | j!| | | j"| | W Y d }~n
d }~0 0 W | #  n
| #  0 |S )NrP   r   r               r         )$r   r  r   r   r{   r%  r   r   r   r   rH   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   rC   r   r   r   r   r   r}   r   r
   r
   r   rF     s2    &z,OmegaConfGrammarParser.interpolationResolverc                       st   e Zd ZdZdeed fddZdd Zd	d
 Zdd Z	dd Z
edddZedddZedddZ  ZS )z'OmegaConfGrammarParser.ConfigKeyContextrd   Nre   rf   c                    s   t  || || _d S ri   rj   rk   rb   r
   r   r\   ;  s    z0OmegaConfGrammarParser.ConfigKeyContext.__init__c                 C   s   |  tjdS rl   )rn   r   r   rp   r
   r
   r   rD   ?  s    z5OmegaConfGrammarParser.ConfigKeyContext.interpolationc                 C   s   |  tjdS rl   )rr   r   r$   rp   r
   r
   r   r$   C  s    z*OmegaConfGrammarParser.ConfigKeyContext.IDc                 C   s   |  tjdS rl   )rr   r   r)   rp   r
   r
   r   r)   F  s    z1OmegaConfGrammarParser.ConfigKeyContext.INTER_KEYc                 C   s   t jS ri   )r   RULE_configKeyrp   r
   r
   r   ru   I  s    z4OmegaConfGrammarParser.ConfigKeyContext.getRuleIndexrv   c                 C   s   t |dr||  d S )NenterConfigKey)ry   r0  rz   r
   r
   r   r{   L  s    
z1OmegaConfGrammarParser.ConfigKeyContext.enterRulec                 C   s   t |dr||  d S )NexitConfigKey)ry   r1  rz   r
   r
   r   r}   P  s    
z0OmegaConfGrammarParser.ConfigKeyContext.exitRuler~   c                 C   s"   t |dr|| S || S d S )NvisitConfigKey)ry   r2  r   r   r
   r
   r   r   T  s    

z.OmegaConfGrammarParser.ConfigKeyContext.accept)Nre   )r   r   r   r   r   r   r\   rD   r$   r)   ru   r   r{   r}   r   r   r   r
   r
   rb   r   r  8  s   r  c              
   C   s4  t | | j| j}| |d| j z zd| _| j|  | j	d}|t j
fv rn| |d d| _|   n`|t jfv r| |d d| _| t j n4|t jfv r| |d d| _| t j nt| W nH ty } z.||_| j| | | j| | W Y d }~n
d }~0 0 W |   n
|   0 |S )	NrR      r      r/      r0      )r   r  r   r   r{   r/  r   r   r   r   r   r   rD   r$   r   r)   r   r   r   r   r   r}   r   r
   r
   r   rG   ]  s8    &z OmegaConfGrammarParser.configKeyc                       s   e Zd ZdZdeed fddZdeddd	Zdedd
dZdedddZ	dd Z
edddZedddZedddZ  ZS )z*OmegaConfGrammarParser.ResolverNameContextrd   Nre   rf   c                    s   t  || || _d S ri   rj   rk   rb   r
   r   r\     s    z3OmegaConfGrammarParser.ResolverNameContext.__init__r   c                 C   s&   |d u r|  tjS | tj|S d S ri   r   r   r
   r
   r   rD     s    z8OmegaConfGrammarParser.ResolverNameContext.interpolationc                 C   s&   |d u r|  tjS | tj|S d S ri   r   r   r$   rr   r   r
   r
   r   r$     s    z-OmegaConfGrammarParser.ResolverNameContext.IDc                 C   s&   |d u r|  tjS | tj|S d S ri   r  r   r
   r
   r   r(     s    z.OmegaConfGrammarParser.ResolverNameContext.DOTc                 C   s   t jS ri   )r   RULE_resolverNamerp   r
   r
   r   ru     s    z7OmegaConfGrammarParser.ResolverNameContext.getRuleIndexrv   c                 C   s   t |dr||  d S )NenterResolverName)ry   r9  rz   r
   r
   r   r{     s    
z4OmegaConfGrammarParser.ResolverNameContext.enterRulec                 C   s   t |dr||  d S )NexitResolverName)ry   r:  rz   r
   r
   r   r}     s    
z3OmegaConfGrammarParser.ResolverNameContext.exitRuler~   c                 C   s"   t |dr|| S || S d S )NvisitResolverName)ry   r;  r   r   r
   r
   r   r     s    

z1OmegaConfGrammarParser.ResolverNameContext.accept)Nre   )N)N)N)r   r   r   r   r   r   r\   rD   r$   r(   ru   r   r{   r}   r   r   r   r
   r
   rb   r   r$    s   r$  c              
   C   s  t | | j| j}| |d| j d| _zz4| |d d| _| j	|  | j
d}|t jfv rvd| _|   n(|t jfv rd| _| t j nt| d| _| j	|  | j
d}|t jkr`d| _| t j d	| _| j	|  | j
d}|t jfv rd
| _|   n*|t jfv r8d| _| t j nt| d| _| j	|  | j
d}qW nH ty } z.||_| j| | | j| | W Y d }~n
d }~0 0 W |   n
|   0 |S )NrT   r   r                              )r   r$  r   r   r{   r8  r   r   r   r   r   r   r   rD   r$   r   r   r(   r   r   r   r   r}   r  r
   r
   r   rH     sX    &z#OmegaConfGrammarParser.resolverNamec                       s|   e Zd ZdZdeed fddZdd Zd	d
 Zdd Z	dd Z
dd ZedddZedddZedddZ  ZS )z)OmegaConfGrammarParser.QuotedValueContextrd   Nre   rf   c                    s   t  || || _d S ri   rj   rk   rb   r
   r   r\     s    z2OmegaConfGrammarParser.QuotedValueContext.__init__c                 C   s   |  tjdS rl   )rr   r   r*   rp   r
   r
   r   r*     s    z>OmegaConfGrammarParser.QuotedValueContext.MATCHING_QUOTE_CLOSEc                 C   s   |  tjdS rl   )rr   r   r   rp   r
   r
   r   r     s    z;OmegaConfGrammarParser.QuotedValueContext.QUOTE_OPEN_SINGLEc                 C   s   |  tjdS rl   )rr   r   r   rp   r
   r
   r   r     s    z;OmegaConfGrammarParser.QuotedValueContext.QUOTE_OPEN_DOUBLEc                 C   s   |  tjdS rl   rm   rp   r
   r
   r   r>     s    z.OmegaConfGrammarParser.QuotedValueContext.textc                 C   s   t jS ri   )r   RULE_quotedValuerp   r
   r
   r   ru     s    z6OmegaConfGrammarParser.QuotedValueContext.getRuleIndexrv   c                 C   s   t |dr||  d S )NenterQuotedValue)ry   rF  rz   r
   r
   r   r{     s    
z3OmegaConfGrammarParser.QuotedValueContext.enterRulec                 C   s   t |dr||  d S )NexitQuotedValue)ry   rG  rz   r
   r
   r   r}     s    
z2OmegaConfGrammarParser.QuotedValueContext.exitRuler~   c                 C   s"   t |dr|| S || S d S )NvisitQuotedValue)ry   rH  r   r   r
   r
   r   r     s    

z0OmegaConfGrammarParser.QuotedValueContext.accept)Nre   )r   r   r   r   r   r   r\   r*   r   r   r>   ru   r   r{   r}   r   r   r   r
   r
   rb   r   r     s   r   c              
   C   s~  t | | j| j}| |d| j d| _zDz| |d d| _| j	d}|t j
ksn|t jksn| j|  n| j|  |   d| _| j|  | j	d}|d@ dkrd|> dt j> dt j> B dt j> B dt j> B dt j> B dt j> B @ dkrd| _|   d| _| t j W nH tyb } z.||_| j| | | j| | W Y d }~n
d }~0 0 W |   n
|   0 |S )	NrV   r   r         r         )r   r   r   r   r{   rE  r   r   r   r   r   r   r   recoverInlinereportMatchconsumer   r   r   r   r   r%   r+   r>   r   r*   r   r   r   r   r}   r   r
   r
   r   rI     s2    X&z"OmegaConfGrammarParser.quotedValuec                       s   e Zd ZdZd&eed fddZd'eddd	Zd(edd
dZd)edddZ	d*edddZ
d+edddZd,edddZd-edddZd.edddZd/edddZd0edddZdd Zeddd Zedd!d"Zed#d$d%Z  ZS )1z'OmegaConfGrammarParser.PrimitiveContextrd   Nre   rf   c                    s   t  || || _d S ri   rj   rk   rb   r
   r   r\   0  s    z0OmegaConfGrammarParser.PrimitiveContext.__init__r   c                 C   s&   |d u r|  tjS | tj|S d S ri   r7  r   r
   r
   r   r$   4  s    z*OmegaConfGrammarParser.PrimitiveContext.IDc                 C   s&   |d u r|  tjS | tj|S d S ri   r   r   r"   rr   r   r
   r
   r   r"   :  s    z,OmegaConfGrammarParser.PrimitiveContext.NULLc                 C   s&   |d u r|  tjS | tj|S d S ri   r   r   r    rr   r   r
   r
   r   r    @  s    z+OmegaConfGrammarParser.PrimitiveContext.INTc                 C   s&   |d u r|  tjS | tj|S d S ri   r   r   r   rr   r   r
   r
   r   r   F  s    z-OmegaConfGrammarParser.PrimitiveContext.FLOATc                 C   s&   |d u r|  tjS | tj|S d S ri   r   r   r!   rr   r   r
   r
   r   r!   L  s    z,OmegaConfGrammarParser.PrimitiveContext.BOOLc                 C   s&   |d u r|  tjS | tj|S d S ri   r   r   r#   rr   r   r
   r
   r   r#   R  s    z5OmegaConfGrammarParser.PrimitiveContext.UNQUOTED_CHARc                 C   s&   |d u r|  tjS | tj|S d S ri   )r   r   r   rr   r   r
   r
   r   r   X  s    z-OmegaConfGrammarParser.PrimitiveContext.COLONc                 C   s&   |d u r|  tjS | tj|S d S ri   r   r   r
   r
   r   r%   ^  s    z+OmegaConfGrammarParser.PrimitiveContext.ESCc                 C   s&   |d u r|  tjS | tj|S d S ri   r   r   r&   rr   r   r
   r
   r   r&   d  s    z*OmegaConfGrammarParser.PrimitiveContext.WSc                 C   s&   |d u r|  tjS | tj|S d S ri   r   r   r
   r
   r   rD   j  s    z5OmegaConfGrammarParser.PrimitiveContext.interpolationc                 C   s   t jS ri   )r   RULE_primitiverp   r
   r
   r   ru   q  s    z4OmegaConfGrammarParser.PrimitiveContext.getRuleIndexrv   c                 C   s   t |dr||  d S )NenterPrimitive)ry   rW  rz   r
   r
   r   r{   t  s    
z1OmegaConfGrammarParser.PrimitiveContext.enterRulec                 C   s   t |dr||  d S )NexitPrimitive)ry   rX  rz   r
   r
   r   r}   x  s    
z0OmegaConfGrammarParser.PrimitiveContext.exitRuler~   c                 C   s"   t |dr|| S || S d S )NvisitPrimitive)ry   rY  r   r   r
   r
   r   r   |  s    

z.OmegaConfGrammarParser.PrimitiveContext.accept)Nre   )N)N)N)N)N)N)N)N)N)N)r   r   r   r   r   r   r\   r$   r"   r    r   r!   r#   r   r%   r&   rD   ru   r   r{   r}   r   r   r   r
   r
   rb   r   r   -  s    r   c              
   C   s  t | | j| j}| |d| j d| _zzB| |d d| _| j	|  | j
d}d| _| j	|  | j
d}|t jfv rd| _| t j n2|t jfv rd| _| t j n|t jfv rd| _| t j n|t jfv rd| _| t j n|t jfv rd	| _| t j n|t jfv r@d
| _| t j n|t jfv rbd| _| t j nj|t jfv rd| _| t j nH|t jfv rd| _| t j n&|t jfv rd| _|   nt| d| _| j	|  | j
d}|d@ dkrnd|> dt j> dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B @ dksZqnqZW nH ty } z.||_| j| | | j| | W Y d }~n
d }~0 0 W |   n
|   0 |S )NrX   r   r                                       r   )r   r   r   r   r{   rV  r   r   r   r   r   r   r$   r   r"   r    r   r!   r#   r   r%   r&   r   rD   r   r   r   r   r   r}   r   r
   r
   r   rJ     s~    ~
&z OmegaConfGrammarParser.primitivec                       s   e Zd ZdZd"eed fddZd#eddd	Zd$edd
dZd%edddZ	d&edddZ
d'edddZd(edddZd)edddZd*edddZdd ZedddZedddZedd d!Z  ZS )+z%OmegaConfGrammarParser.DictKeyContextrd   Nre   rf   c                    s   t  || || _d S ri   rj   rk   rb   r
   r   r\     s    z.OmegaConfGrammarParser.DictKeyContext.__init__r   c                 C   s&   |d u r|  tjS | tj|S d S ri   r7  r   r
   r
   r   r$     s    z(OmegaConfGrammarParser.DictKeyContext.IDc                 C   s&   |d u r|  tjS | tj|S d S ri   rP  r   r
   r
   r   r"     s    z*OmegaConfGrammarParser.DictKeyContext.NULLc                 C   s&   |d u r|  tjS | tj|S d S ri   rQ  r   r
   r
   r   r      s    z)OmegaConfGrammarParser.DictKeyContext.INTc                 C   s&   |d u r|  tjS | tj|S d S ri   rR  r   r
   r
   r   r     s    z+OmegaConfGrammarParser.DictKeyContext.FLOATc                 C   s&   |d u r|  tjS | tj|S d S ri   rS  r   r
   r
   r   r!     s    z*OmegaConfGrammarParser.DictKeyContext.BOOLc                 C   s&   |d u r|  tjS | tj|S d S ri   rT  r   r
   r
   r   r#     s    z3OmegaConfGrammarParser.DictKeyContext.UNQUOTED_CHARc                 C   s&   |d u r|  tjS | tj|S d S ri   r   r   r
   r
   r   r%     s    z)OmegaConfGrammarParser.DictKeyContext.ESCc                 C   s&   |d u r|  tjS | tj|S d S ri   rU  r   r
   r
   r   r&     s    z(OmegaConfGrammarParser.DictKeyContext.WSc                 C   s   t jS ri   )r   RULE_dictKeyrp   r
   r
   r   ru     s    z2OmegaConfGrammarParser.DictKeyContext.getRuleIndexrv   c                 C   s   t |dr||  d S )NenterDictKey)ry   rg  rz   r
   r
   r   r{     s    
z/OmegaConfGrammarParser.DictKeyContext.enterRulec                 C   s   t |dr||  d S )NexitDictKey)ry   rh  rz   r
   r
   r   r}     s    
z.OmegaConfGrammarParser.DictKeyContext.exitRuler~   c                 C   s"   t |dr|| S || S d S )NvisitDictKey)ry   ri  r   r   r
   r
   r   r     s    

z,OmegaConfGrammarParser.DictKeyContext.accept)Nre   )N)N)N)N)N)N)N)N)r   r   r   r   r   r   r\   r$   r"   r    r   r!   r#   r%   r&   ru   r   r{   r}   r   r   r   r
   r
   rb   r   r     s   r   c              
   C   s  t | | j| j}| |d| j d| _zzX| |d d| _| j	|  | j
d}d| _| j
d}|d@ dkrd|> dt j> dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B @ dks| j|  n| j|  |   d| _| j	|  | j
d}|d@ dkrd|> dt j> dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B dt j> B @ dksZqqZW nH ty } z.||_| j| | | j| | W Y d }~n
d }~0 0 W |   n
|   0 |S )N   r   r         r      )r   r   r   r   r{   rf  r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   rM  rN  rO  r   r   r   r   r}   r   r
   r
   r   rK     s2    hj
&zOmegaConfGrammarParser.dictKey)er   r   r   ZgrammarFileNameZATNDeserializerZdeserializer   r]   	enumerateZdecisionToStater^   ZPredictionContextCacher_   ZliteralNamesZsymbolicNamesrt   r   r   r   r   r   r   r   r  r  r%  r/  r8  rE  rV  rf  Z	ruleNamesTokenrs   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   sysstdoutZTokenStreamr   r\   r   r   r<   r   r=   ro   r>   r   r?   r   r@   r   rA   r   rB   r   rC   r   rD   r  rE   r  rF   r  rG   r$  rH   r   rI   r   rJ   r   rK   r   r
   r
   rb   r   r   \   s   	""@8+(%.(&(C#:L, %#.6(!XHKr   )Zantlr4ior   rp  version_infotypingr   Z	typing.ior   Parserr   r
   r
   r
   r   <module>   s   P