o
    ڷi                     @   sH   d dl mZ d dlmZ d dlmZ G dd deZG dd deZdS )	    )StringIO)ATNConfigSet)SemanticContextc                   @   s*   e Zd ZdZdedefddZdd ZdS )	PredPredictionaltpredr   r   c                 C   s   || _ || _d S Nr   )selfr   r    r   I/home/ubuntu/vllm_env/lib/python3.10/site-packages/antlr4/dfa/DFAState.py__init__   s   
zPredPrediction.__init__c                 C   s    dt | j d t | j d S )N(z, ))strr   r   r
   r   r   r   __str__   s    zPredPrediction.__str__N)__name__
__module____qualname__	__slots__r   intr   r   r   r   r   r   r      s    r   c                   @   sJ   e Zd ZdZde fdedefddZdd Zd	d
 Zdd Z	dd Z
dS )DFAStatestateNumberconfigsedgesisAcceptState
predictionlexerActionExecutorrequiresFullContext
predicatesr   r   c                 C   s4   || _ || _d | _d| _d| _d | _d| _d | _d S )NFr   r   )r
   r   r   r   r   r   r   5   s   
zDFAState.__init__c                 C   s&   | j d urtdd | j D pd S d S )Nc                 s   s    | ]}|j V  qd S r	   )r   ).0cfgr   r   r   	<genexpr>Y   s    z%DFAState.getAltSet.<locals>.<genexpr>)r   setr   r   r   r   	getAltSetW   s   
zDFAState.getAltSetc                 C   s
   t | jS r	   )hashr   r   r   r   r   __hash__\   s   
zDFAState.__hash__c                 C   s&   | |u rdS t |tsdS | j|jkS )NTF)
isinstancer   r   )r
   otherr   r   r   __eq__j   s
   
zDFAState.__eq__c                 C   s   t  ?}|t| j |d |t| j | jr7|d | jd ur/|t| j n|t| j | W  d    S 1 sEw   Y  d S )N:z=>)	r   writer   r   r   r   r!   r   getvalue)r
   bufr   r   r   r   s   s   


$zDFAState.__str__N)r   r   r   r   r   r   r   r'   r)   r,   r   r   r   r   r   r   /   s    "	r   N)	ior   antlr4.atn.ATNConfigSetr   antlr4.atn.SemanticContextr   objectr   r   r   r   r   r   <module>   s
   "