B
+�[�/ � @ sr d dl mZmZ ddlmZmZ ddlmZ ddlm Z G dd� de
�Z e� e � e� e � dd
d �Z
d d
� Zd S )� )�Sequence�Hashable� )�islice�chain)�Integral)�plistc s( e Zd ZdZdZd=� fdd� Zedd� �Zedd � �Ze d
d � �Z
d d
� Z dd� Z e Z
edd� �Zd>dd�Zd?dd�Ze dd� �Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Ze d)d*� �Zd+d,� Zd-d.� Z d/d0� Z d1d2� Z d3d4� Z d5d6� Z e Z!d7d8� Z"d9d:� Z#d;d<� Z$e%j&Z&� Z'S )@�PDequea�
Persistent double ended queue (deque). Allows quick appends and pops in both ends. Implemented
using two persistent lists.
A maximum length can be specified to create a bounded queue.
Fully supports the Sequence and Hashable protocols including indexing and slicing but
if you need fast random access go for the PVector instead.
Do not instantiate directly, instead use the factory functions :py:func:`dq` or :py:func:`pdeque` to
create an instance.
Some examples:
>>> x = pdeque([1, 2, 3])
>>> x.left
1
>>> x.right
3
>>> x[0] == x.left
True
>>> x[-1] == x.right
True
>>> x.pop()
pdeque([1, 2])
>>> x.pop() == x[:-1]
True
>>> x.popleft()
pdeque([2, 3])
>>> x.append(4)
pdeque([1, 2, 3, 4])
>>> x.appendleft(4)
pdeque([4, 1, 2, 3])
>>> y = pdeque([1, 2, 3], maxlen=3)
>>> y.append(4)
pdeque([2, 3, 4], maxlen=3)
>>> y.appendleft(4)
pdeque([4, 1, 2], maxlen=3)
)�
_left_list� _right_list�_length�_maxlen� __weakref__Nc sV t t| ��| �}||_||_||_|d k rLt|t�s>> pdeque([1, 2]).pop()
pdeque([1])
>>> pdeque([1, 2]).pop(2)
pdeque([])
>>> pdeque([1, 2]).pop(-1)
pdeque([2])
r )�popleftr �
_pop_listsr r
�maxr r
)r �count�new_right_list�
new_left_listr r r �popi s z
PDeque.popc C sF |dk r| � | �S t�| j| j|�\}}t||t| j| d�| j�S )z�
Return new deque with leftmost element removed. Otherwise functionally
equivalent to pop().
>>> pdeque([1, 2]).popleft()
pdeque([2])
r )r0 r r+ r
r r, r r
)r r- r/ r. r r r r* } s zPDeque.popleftc C s` | }|}xN|dkrV|s|rV|d8 }|j r0|j }q
|rD|�� }t� }q
|�� j }t� }q
W ||fS )Nr r )�restr$ r )r" r# r- �new_primary_list�new_secondary_listr r r r+ � s
zPDeque._pop_listsc C s | j o| j S )N)r
r )r r r r � _is_empty� s zPDeque._is_emptyc C s t |t�stS t| �t|�k S )N)r r �NotImplemented�tuple)r �otherr r r �__lt__� s
z
PDeque.__lt__c C s: t |t�stS t| �t|�kr6t| �t|�ks2t�dS dS )NTF)r r r5 r6 �len�AssertionError)r r7 r r r �__eq__� s
z
PDeque.__eq__c C s t t| ��S )N)�hashr6 )r r r r �__hash__� s zPDeque.__hash__c C s | j S )N)r )r r r r �__len__� s zPDeque.__len__c C s( | � | j| j|�\}}}t|||| j�S )z�
Return new deque with elem as the rightmost element.
>>> pdeque([1, 2]).append(3)
pdeque([1, 2, 3])
)�_appendr
r r r
)r �elemr/ r. �
new_lengthr r r �append� s z
PDeque.appendc C s( | � | j| j|�\}}}t|||| j�S )z�
Return new deque with elem as the leftmost element.
>>> pdeque([1, 2]).appendleft(3)
pdeque([3, 1, 2])
)r? r r
r r
)r r@ r. r/ rA r r r �
appendleft� s zPDeque.appendleftc C sd | j d k rN| j| j krN| j dkr*||dfS t�||d�\}}||�|�| jfS ||�|�| jd fS )Nr r )r
r r r+ �cons)r r" r# r@ r2 r3 r r r r? � s
zPDeque._appendc C s, d}x |D ]}| � |�} |d7 }q
W | |fS )Nr r )rD )Zthe_list�iterabler- r@ r r r � _extend_list� s
zPDeque._extend_listc C s` t �||�\}}|}| j| }| jd k rV|| jkrV|| j }t �|||�\}}||8 }|||fS )N)r rF r r
r+ ) r r" r# rE r2 � extend_countr3 Z current_lenZpop_lenr r r �_extend� s
zPDeque._extendc C s. | � | j| j|�\}}}t||| j| | j�S )z�
Return new deque with all elements of iterable appended to the right.
>>> pdeque([1, 2]).extend([3, 4])
pdeque([1, 2, 3, 4])
)rH r r
r r r
)r rE r. r/ rG r r r �extend� s z
PDeque.extendc C s. | � | j| j|�\}}}t||| j| | j�S )a
Return new deque with all elements of iterable appended to the left.
NB! The elements will be inserted in reverse order compared to the order in the iterable.
>>> pdeque([1, 2]).extendleft([3, 4])
pdeque([4, 3, 1, 2])
)rH r
r r r r
)r rE r/ r. rG r r r �
extendleft� s zPDeque.extendleftc C s | j �|�| j�|� S )z�
Return the number of elements equal to elem present in the queue
>>> pdeque([1, 2, 1]).count(1)
2
)r
r- r )r r@ r r r r- � s z PDeque.countc C s~ y t | j�|�| j| jd �S tk
rx y$t | j| j�� �|��� | jd �S tk
rr td�|���Y nX Y nX dS )z�
Return new deque with first element from left equal to elem removed. If no such element is found
a ValueError is raised.
>>> pdeque([2, 1, 2]).remove(2)
pdeque([1, 2])
r z{0} not found in PDequeN)r r
�remover r r r$ r' )r r@ r r r rK s z
PDeque.removec C s t | j| j| j�S )z�
Return reversed deque.
>>> pdeque([1, 2, 3]).reverse()
pdeque([3, 2, 1])
Also supports the standard python reverse function.
>>> reversed(pdeque([1, 2, 3]))
pdeque([3, 2, 1])
)r r r
r )r r r r r$ s zPDeque.reversec C s8 | � |�}|dkr&|�t| �� |��S |�t| | ��S )z�
Return deque with elements rotated steps steps.
>>> x = pdeque([1, 2, 3])
>>> x.rotate(1)
pdeque([3, 1, 2])
>>> x.rotate(-2)
pdeque([3, 1, 2])
r )r0 rJ r r$ rI )r ZstepsZ popped_dequer r r �rotate) s
z
PDeque.rotatec C s t t| �| jffS )N)�pdequer( r
)r r r r �
__reduce__9 s zPDeque.__reduce__c C s� t |t�rz|jd k r4|jdkr4tt| �| | jd�S | }|jd k rT|�|j| j �}|j d k rv|�
| j|j | j �}|S t |t �s�t dt
|�j ��|dkr�| �|�jS t| �| }|dk r�td�|t| ����| �|�jS )Nr )r z-'%s' object cannot be interpreted as an indexr z!pdeque index {0} out of range {1})r �slice�steprM r6 r
�startr* r �stopr0 r r �type�__name__r r9 r! r' )r �index�resultZshiftedr r r � __getitem__= s$
zPDeque.__getitem__)N)r )r )(rT �
__module__� __qualname__�__doc__� __slots__r �propertyr r � staticmethodr r% r) �__str__r r0 r* r+ r4 r8 r; r= r> rB rC r? rF rH rI rJ r- rK r$ � __reversed__rL rN rW r rU �
__classcell__r r )r r r sB (
r r Nc C sd t | �}|dk r || d� }t|�}t|d �}t|d|� �}t||d� dd�}t||||�S )a*
Return deque containing the elements of iterable. If maxlen is specified then
len(iterable) - maxlen elements are discarded from the left to if len(iterable) > maxlen.
>>> pdeque([1, 2, 3])
pdeque([1, 2, 3])
>>> pdeque([1, 2, 3, 4], maxlen=2)
pdeque([3, 4], maxlen=2)
N� T)r$ )r6 r9 �intr r )rE r �tr �pivotr r r r r rM ^ s
rM c G s t | �S )z[
Return deque containing all arguments.
>>> dq(1, 2, 3)
pdeque([1, 2, 3])
)rM )�elementsr r r �dqq s rf )r N)�_compatr r � itertoolsr r �numbersr Zpyrsistent._plistr �objectr �registerrM rf r r r r � s U