a
    d:"                     @   sB  d dl mZ d dlmZ d dlmZmZ dZdZG dd de	Z
G dd de
ZG d	d
 d
e
ZG dd de
ZG dd de
ZG dd de
ZG dd deZG dd de
ZG dd de
ZG dd deZG dd de
ZG dd deZee
jee
jee
jee
jee
jee
jee
jee
jee
jee
ji
e
_ [[d dl!T dS )    )IntervalSet)Token)	PredicatePrecedencePredicateNc                   @   sT   e 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 ZedddZdS )
Transition)target	isEpsilonlabel                        	   
   )INVALIDEPSILONRANGERULE	PREDICATEATOMACTIONSETNOT_SETWILDCARD
PRECEDENCEr   c                 C   s&   |d u rt d|| _d| _d | _d S )Nztarget cannot be null.F)	Exceptionr   r   r	   selfr    r#   ^/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/antlr4/atn/Transition.py__init__;   s
    zTransition.__init__N)__name__
__module____qualname__	__slots__r   r   r   r   r   r   r   r   r   r   ZserializationNamesdictserializationTypesATNStater%   r#   r#   r#   r$   r      s   r   c                       sJ   e Zd ZdZeed fddZdd Zeeeddd	Zd
d Z	  Z
S )AtomTransition)label_serializationType)r   r	   c                    s(   t  | || _|  | _| j| _d S N)superr%   r.   	makeLabelr	   r   r/   )r"   r   r	   	__class__r#   r$   r%   I   s    
zAtomTransition.__init__c                 C   s   t  }|| j |S r0   )r   ZaddOner.   r"   sr#   r#   r$   r2   O   s    zAtomTransition.makeLabelsymbolminVocabSymbolmaxVocabSymbolc                 C   s
   | j |kS r0   )r.   r"   r8   r9   r:   r#   r#   r$   matchesT   s    zAtomTransition.matchesc                 C   s
   t | jS r0   )strr.   r"   r#   r#   r$   __str__W   s    zAtomTransition.__str__r&   r'   r(   r)   r,   intr%   r2   r<   r?   __classcell__r#   r#   r3   r$   r-   F   s
   r-   c                       s>   e Zd ZdZeeeed fddZeeedddZ  Z	S )RuleTransition)	ruleIndex
precedencefollowStater/   )	ruleStartrD   rE   rF   c                    s0   t  | || _|| _|| _| j| _d| _d S NT)r1   r%   rD   rE   rF   r   r/   r   )r"   rG   rD   rE   rF   r3   r#   r$   r%   ]   s    zRuleTransition.__init__r7   c                 C   s   dS NFr#   r;   r#   r#   r$   r<   e   s    zRuleTransition.matches)
r&   r'   r(   r)   RuleStartStaterA   r,   r%   r<   rB   r#   r#   r3   r$   rC   Z   s   rC   c                       s<   e Zd ZdZd
 fdd	ZeeedddZdd	 Z  ZS )EpsilonTransition)r/   outermostPrecedenceReturnc                    s(   t t| | | j| _d| _|| _d S rH   )r1   rK   r%   r   r/   r   rL   )r"   r   rL   r3   r#   r$   r%   l   s    zEpsilonTransition.__init__r7   c                 C   s   dS rI   r#   r;   r#   r#   r$   r<   r   s    zEpsilonTransition.matchesc                 C   s   dS )Nepsilonr#   r>   r#   r#   r$   r?   u   s    zEpsilonTransition.__str__)rM   )	r&   r'   r(   r)   r%   rA   r<   r?   rB   r#   r#   r3   r$   rK   i   s   rK   c                       sL   e Zd ZdZeeed fddZdd Zeeeddd	Zd
d Z	  Z
S )RangeTransition)r/   startstop)r   rP   rQ   c                    s.   t  | | j| _|| _|| _|  | _d S r0   )r1   r%   r   r/   rP   rQ   r2   r	   )r"   r   rP   rQ   r3   r#   r$   r%   {   s
    zRangeTransition.__init__c                 C   s"   t  }|t| j| jd  |S Nr
   )r   addRangerangerP   rQ   r5   r#   r#   r$   r2      s    zRangeTransition.makeLabelr7   c                 C   s   || j ko|| jkS r0   )rP   rQ   r;   r#   r#   r$   r<      s    zRangeTransition.matchesc                 C   s    dt | j d t | j d S )N'z'..')chrrP   rQ   r>   r#   r#   r$   r?      s    zRangeTransition.__str__r@   r#   r#   r3   r$   rO   x   s
   rO   c                       s"   e Zd Zed fddZ  ZS )AbstractPredicateTransitionr   c                    s   t  | d S r0   )r1   r%   r!   r3   r#   r$   r%      s    z$AbstractPredicateTransition.__init__)r&   r'   r(   r,   r%   rB   r#   r#   r3   r$   rW      s   rW   c                       sN   e Zd ZdZeeeed fddZeeedddZdd	 Z	d
d Z
  ZS )PredicateTransition)r/   rD   	predIndexisCtxDependent)r   rD   rY   rZ   c                    s0   t  | | j| _|| _|| _|| _d| _d S rH   )r1   r%   r   r/   rD   rY   rZ   r   )r"   r   rD   rY   rZ   r3   r#   r$   r%      s    zPredicateTransition.__init__r7   c                 C   s   dS rI   r#   r;   r#   r#   r$   r<      s    zPredicateTransition.matchesc                 C   s   t | j| j| jS r0   )r   rD   rY   rZ   r>   r#   r#   r$   getPredicate   s    z PredicateTransition.getPredicatec                 C   s   dt | j d t | j S )NZpred_:)r=   rD   rY   r>   r#   r#   r$   r?      s    zPredicateTransition.__str__)r&   r'   r(   r)   r,   rA   boolr%   r<   r[   r?   rB   r#   r#   r3   r$   rX      s
   rX   c                       sH   e Zd ZdZdeeeed fddZeeeddd	Zd
d Z	  Z
S )ActionTransition)r/   rD   actionIndexrZ   rM   F)r   rD   r_   rZ   c                    s0   t  | | j| _|| _|| _|| _d| _d S rH   )r1   r%   r   r/   rD   r_   rZ   r   )r"   r   rD   r_   rZ   r3   r#   r$   r%      s    zActionTransition.__init__r7   c                 C   s   dS rI   r#   r;   r#   r#   r$   r<      s    zActionTransition.matchesc                 C   s   d| j  d | j S )NZaction_r\   )rD   r_   r>   r#   r#   r$   r?      s    zActionTransition.__str__)rM   F)r&   r'   r(   r)   r,   rA   r]   r%   r<   r?   rB   r#   r#   r3   r$   r^      s   r^   c                       sB   e Zd ZdZeed fddZeeedddZdd	 Z	  Z
S )
SetTransitionr/   r   setc                    sJ   t  | | j| _|d ur$|| _n"t | _| jttj	tj	d  d S rR   )
r1   r%   r   r/   r	   r   rS   rT   r   ZINVALID_TYPEr"   r   rb   r3   r#   r$   r%      s    zSetTransition.__init__r7   c                 C   s
   || j v S r0   )r	   r;   r#   r#   r$   r<      s    zSetTransition.matchesc                 C   s
   t | jS r0   )r=   r	   r>   r#   r#   r$   r?      s    zSetTransition.__str__)r&   r'   r(   r)   r,   r   r%   rA   r<   r?   rB   r#   r#   r3   r$   r`      s   	r`   c                       sF   e Zd Zeed fddZeeed fddZ fddZ  Z	S )	NotSetTransitionra   c                    s   t  || | j| _d S r0   )r1   r%   r   r/   rc   r3   r#   r$   r%      s    zNotSetTransition.__init__r7   c                    s*   ||ko(||ko(t t| | ||| S r0   )r1   typer<   r;   r3   r#   r$   r<      s
    zNotSetTransition.matchesc                    s   dt t| |   S )N~)r1   re   r?   r>   r3   r#   r$   r?      s    zNotSetTransition.__str__)
r&   r'   r(   r,   r   r%   rA   r<   r?   rB   r#   r#   r3   r$   rd      s   rd   c                       s@   e Zd ZdZed fddZeeedddZdd	 Z  Z	S )
WildcardTransitionr/   r   c                    s   t  | | j| _d S r0   )r1   r%   r   r/   r!   r3   r#   r$   r%      s    zWildcardTransition.__init__r7   c                 C   s   ||ko||kS r0   r#   r;   r#   r#   r$   r<      s    zWildcardTransition.matchesc                 C   s   dS )N.r#   r>   r#   r#   r$   r?      s    zWildcardTransition.__str__)
r&   r'   r(   r)   r,   r%   rA   r<   r?   rB   r#   r#   r3   r$   rg      s   rg   c                       sJ   e Zd ZdZeed fddZeeedddZdd	 Zd
d Z	  Z
S )PrecedencePredicateTransition)r/   rE   )r   rE   c                    s$   t  | | j| _|| _d| _d S rH   )r1   r%   r   r/   rE   r   )r"   r   rE   r3   r#   r$   r%      s    z&PrecedencePredicateTransition.__init__r7   c                 C   s   dS rI   r#   r;   r#   r#   r$   r<      s    z%PrecedencePredicateTransition.matchesc                 C   s
   t | jS r0   )r   rE   r>   r#   r#   r$   r[      s    z*PrecedencePredicateTransition.getPredicatec                 C   s
   | j d S )Nz >= _p)rE   r>   r#   r#   r$   r?      s    z%PrecedencePredicateTransition.__str__)r&   r'   r(   r)   r,   rA   r%   r<   r[   r?   rB   r#   r#   r3   r$   ri      s
   ri   )*)"Zantlr4.IntervalSetr   Zantlr4.Tokenr   Zantlr4.atn.SemanticContextr   r   r,   rJ   objectr   r-   rC   rK   rO   rW   rX   r^   r`   rd   rg   ri   r   r   r   r   r   r   r   r   r   r   r+   Zantlr4.atn.ATNStater#   r#   r#   r$   <module>   s<   *