o
    ÃÚ·i	  ã                   @   s    d dl mZ G dd„ deƒZdS )é    )ÚTokenc                   @   s¨   e Zd ZdZdefdd„Zdd„ Zedd„ ƒZed	d
„ ƒZ	dd„ Z
dd„ Zdefdd„Zdefdd„Zdd„ Zdefdd„Zdefdd„Zdedefdd„Zd d!„ Zd"S )#ÚInputStream)ÚnameÚstrdataÚ_indexÚdataÚ_sizer   c                 C   s   d| _ || _|  ¡  d S )Nz<empty>)r   r   Ú_loadString)Úselfr   © r   úH/home/ubuntu/vllm_env/lib/python3.10/site-packages/antlr4/InputStream.pyÚ__init__   s   zInputStream.__init__c                 C   s(   d| _ dd„ | jD ƒ| _t| jƒ| _d S )Nr   c                 S   s   g | ]}t |ƒ‘qS r   )Úord)Ú.0Úcr   r   r   Ú
<listcomp>   s    z+InputStream._loadString.<locals>.<listcomp>)r   r   r   Úlenr   ©r
   r   r   r   r	      s   zInputStream._loadStringc                 C   ó   | j S ©N©r   r   r   r   r   Úindex   ó   zInputStream.indexc                 C   r   r   )r   r   r   r   r   Úsize   r   zInputStream.sizec                 C   s
   d| _ d S )Nr   r   r   r   r   r   Úreset'   ó   
zInputStream.resetc                 C   s:   | j | jkr|  d¡tjksJ ‚tdƒ‚|  j d7  _ d S )Né   zcannot consume EOF)r   r   ÚLAr   ÚEOFÚ	Exceptionr   r   r   r   Úconsume*   s   zInputStream.consumeÚoffsetc                 C   sL   |dkrdS |dk r|d7 }| j | d }|dk s|| jkr!tjS | j| S )Nr   r   )r   r   r   r   r   )r
   r!   Úposr   r   r   r   0   s   
zInputStream.LAc                 C   s
   |   |¡S r   )r   )r
   r!   r   r   r   ÚLT:   r   zInputStream.LTc                 C   s   dS )Néÿÿÿÿr   r   r   r   r   Úmark>   ó   zInputStream.markÚmarkerc                 C   s   d S r   r   )r
   r'   r   r   r   ÚreleaseA   r&   zInputStream.releaser   c                 C   s&   || j kr
|| _ d S t|| jƒ| _ d S r   )r   Úminr   )r
   r   r   r   r   ÚseekG   s   
zInputStream.seekÚstartÚstopc                 C   s4   || j kr
| j d }|| j krdS | j||d … S )Nr   Ú )r   r   )r
   r+   r,   r   r   r   ÚgetTextN   s
   


zInputStream.getTextc                 C   r   r   )r   r   r   r   r   Ú__str__V   s   zInputStream.__str__N)Ú__name__Ú
__module__Ú__qualname__Ú	__slots__Ústrr   r	   Úpropertyr   r   r   r    Úintr   r#   r%   r(   r*   r.   r/   r   r   r   r   r      s"    


r   N)Úantlr4.Tokenr   Úobjectr   r   r   r   r   Ú<module>   s   
