
    Ng                     b    d dl mZ d dlmZ d dlmZ d dlmZ d dlm	Z	m
Z
  G d de          ZdS )	    )IntervalSet)RuleContext)Token)ATNType)ATNStateDecisionStatec                       e Zd ZdZdZdedefdZdede	fdZ
defd	Zddede	fdZdefdZdefdZdefdZdefdZdede	fdZd
S )ATN)
grammarTypemaxTokenTypestatesdecisionToStateruleToStartStateruleToStopStatemodeNameToStartStateruleToTokenTypelexerActionsmodeToStartStater   r   r   c                     || _         || _        g | _        g | _        g | _        d | _        t                      | _        d | _        d | _	        g | _
        d S N)r   r   r   r   r   r   dictr   r   r   r   )selfr   r   s      J/var/www/html/ai-engine/env/lib/python3.11/site-packages/antlr4/atn/ATN.py__init__zATN.__init__   sa    &(  " "#$(FF!  $ ! "    sctxc                 R    ddl m}  ||           }|                    ||          S )Nr   )LL1Analyzer)r   )antlr4.LL1Analyzerr   LOOK)r   r   r   r   anals        r   nextTokensInContextzATN.nextTokensInContext6   s9    222222{4  yyy$$$r   c                 z    |j         |j         S |                     |d           |_         d|j         _        |j         S )NT)nextTokenWithinRuler#   readonlyr   r   s     r   nextTokensNoContextzATN.nextTokensNoContext>   s@     ,(( $ 8 8D A A)-&$$r   Nc                 d    |d k    r|                      |          S |                     ||          S r   )r(   r#   )r   r   r   s      r   
nextTokenszATN.nextTokensE   s5    99++A...++As333r   statec                 ~    | | |_         t          | j                  |_        | j                            |           d S r   )atnlenr   stateNumberappendr   r+   s     r   addStatezATN.addStateK   s=    EI #DK 0 0E5!!!!!r   c                 $    d | j         |j        <   d S r   )r   r/   r1   s     r   removeStatezATN.removeStateQ   s    )-E%&&&r   c                 |    | j                             |           t          | j                   dz
  |_        |j        S )N   )r   r0   r.   decisionr'   s     r   defineDecisionStatezATN.defineDecisionStateT   s8    ##A&&&-..q0
zr   r7   c                 P    t          | j                  dk    rd S | j        |         S )Nr   )r.   r   )r   r7   s     r   getDecisionStatezATN.getDecisionStateY   s+    t#$$a''4'11r   r/   c                    |dk     s|t          | j                  k    rt          d          | j        |         }|                     |          }t          j        |vr|S t                      }|                    |           |                    t          j                   |d k    r|j	        dk    rt          j        |v r| j        |j	                 }|j
        d         }|                     |j                  }|                    |           |                    t          j                   |j        }|d k    r|j	        dk    rt          j        |v t          j        |v r|                    t          j                   |S )Nr   zInvalid state number.)r.   r   	Exceptionr*   r   EPSILONr   addSet	removeOneinvokingStatetransitionsfollowState	parentCtxaddOneEOF)r   r/   r   r   	followingexpectedr@   rts           r   getExpectedTokenszATN.getExpectedTokensq   s[   ??kS-=-===3444K$OOA&&	=	))==	"""5=)))d{{s0A55%-9:T:T K(9:M*1-B77IOOI&&&u}----C d{{s0A55%-9:T:T =I%%OOEI&&&r   r   )__name__
__module____qualname__	__slots__INVALID_ALT_NUMBERr   intr   r   r   r#   r(   r*   r2   r4   r   r8   r:   rI    r   r   r
   r
      s7       I #7 ## # # # #<%H %+ % % % %%H % % % %4 48 4 4 4 4 4"X " " " ". . . . .M    
2 2 2 2 20C [      r   r
   N)antlr4.IntervalSetr   antlr4.RuleContextr   antlr4.Tokenr   antlr4.atn.ATNTyper   antlr4.atn.ATNStater   r   objectr
   rP   r   r   <module>rW      s   
 + * * * * * * * * * * *       & & & & & & 7 7 7 7 7 7 7 7v v v v v& v v v v vr   