File: //lib/python3.9/site-packages/elftools/construct/lib/__pycache__/bitstream.cpython-39.pyc
a
�#�_� � @ s4 d dl mZmZ G dd� de�ZG dd� de�ZdS )� )�
encode_bin�
decode_binc @ s> e Zd Zg d�Zdd� Zdd� Zdd� Zdd d
�Zd d � Zd
S )�BitStreamReader)� substream�buffer�
total_sizec C s || _ d| _d| _d S )N� � )r r r ��selfr � r �D/usr/lib/python3.9/site-packages/elftools/construct/lib/bitstream.py�__init__ s zBitStreamReader.__init__c C s | j d dkrtd| j ��d S )N� r z/total size of read data must be a multiple of 8)r �
ValueError�r r r r
�close s �zBitStreamReader.closec C s
| j �� S �N)r �tellr r r r
r s zBitStreamReader.tellr c C s d| _ d| _| j�||� d S �Nr r )r r r �seek�r �pos�whencer r r
r s zBitStreamReader.seekc C s� |dk rt d��t| j�}|dkr(d}n|||krP| jd |� }| j|d � | _nT| j}||8 }|d }|d@ rv|d7 }t| j�|��}||d |� 7 }||d � | _| jt|�7 _|S )Nr zcount cannot be negativer r � r )r �lenr r r �readr )r �count�l�data�bytes�bufr r r
r s$
zBitStreamReader.readN)r ) �__name__�
__module__� __qualname__� __slots__r r r r r r r r r
r s
r c @ sF e Zd Zg d�Zdd� Zdd� Zdd� Zdd � Zdd d �Zd
d� Z dS )�BitStreamWriter�r r r c C s || _ g | _d| _d S )Nr r' r
r r r
r 3 s zBitStreamWriter.__init__c C s | � � d S r )�flushr r r r
r 8 s zBitStreamWriter.closec C s, t d�| j��}| j�|� g | _d| _d S r )r �joinr r �writer )r r r r r
r( ; s zBitStreamWriter.flushc C s | j �� | jd S )Nr )r r r r r r r
r A s zBitStreamWriter.tellr c C s | � � | j�||� d S r )r( r r r r r r
r D s zBitStreamWriter.seekc C s6 |sd S t |�tur&tdt |�f ��| j�|� d S )Nz data must be a string, not %r)�type�str� TypeErrorr �append)r r r r r
r* H s
zBitStreamWriter.writeN)r )
r" r# r$ r% r r r( r r r* r r r r
r& / s
r& N)Zbinaryr r �objectr r& r r r r
�<module> s ,