� e f�@s#ddlZddlZddlZddlZddlZddlZddlmZddlmZddlm Z dgZ ejdkr�ddlZGdd �d e �Zn>Gd d �d e �Zdd�Zd d�Zejee�Gdd�de �ZGdd�de �ZdS)�N�)�context)� reduction)�util� BufferWrapper�win32c@s@eZdZej�Zdd�Zdd�Zdd�ZdS)�ArenacCs�||_xwtd�D]]}dtj�t|j�f}tjd|d|�}tj�dkriPn|j �qWt d��||_||_|j|jf|_ dS)N�dz pym-%d-%sr�tagnamerzCannot find name for new mmap��)�size�range�os�getpid�next�_rand�mmap�_winapiZGetLastError�close�FileExistsError�name�buffer�_state)�selfr�irZbuf�r�9/opt/alt/python34/lib64/python3.4/multiprocessing/heap.py�__init__#s zArena.__init__cCstj|�|jS)N)rZassert_spawningr)rrrr�__getstate__2s zArena.__getstate__cCs>|\|_|_|_tjd|jd|j�|_dS)Nrr r)rrrrr)r�staterrr�__setstate__6s!zArena.__setstate__N) �__name__� __module__�__qualname__�tempfileZ_RandomNameSequencerrrr rrrrrsrc@seZdZddd�ZdS)rrc Cs�||_||_|d kr�tjddtj�dtj��\|_}tj|�tj |tj |jf�t|jddd��}|jd|�WdQXnt j |j|j�|_dS) Nr�prefixzpym-%d-�dir�wb�closefdFsr)r�fdr$ZmkstemprrrZget_temp_dir�unlink�Finalizer�open�writerr)rrr)r�frrrr?s ( zArena.__init__Nr)r!r"r#rrrrrr=scCs=|jdkrtd��nt|jtj|j�ffS)NrzDArena is unpicklable because forking was enabled when it was createdr)r)� ValueError� rebuild_arenarrZDupFd)�arrr�reduce_arenaKsr2cCst||j��S)N)r�detach)rZdupfdrrrr0Qsr0c@s�eZdZdZejdd�Zedd��Zdd�Z dd �Z d d�Zdd �Zdd�Z dd�ZdS)�Heap�cCsmtj�|_tj�|_||_g|_i|_i|_ i|_ t�|_g|_ g|_dS)N)rr�_lastpid� threadingZLock�_lock�_size�_lengths�_len_to_seq�_start_to_block�_stop_to_block�set�_allocated_blocks�_arenas�_pending_free_blocks)rrrrrr^s z Heap.__init__cCs|d}|||@S)Nrr)�nZ alignment�maskrrr�_roundupks z Heap._roundupc Cs tj|j|�}|t|j�kr�|jt|j|�tj�}|jd9_t j d|�t|�}|jj |�|d|fS|j|}|j|}|j�}|s�|j|=|j|=n|\}}}|j||f=|j||f=|S)N�z"allocating a new mmap of length %dr)�bisectZbisect_leftr:�lenrD�maxr9r�PAGESIZEr�inforr@�appendr;�popr<r=) rrr�length�arena�seq�block�start�stoprrr�_mallocqs"! zHeap._mallocc Cs$|\}}}y|j||f}Wntk r:YnX|j|�\}}y|j||f}Wntk r{YnX|j|�\}}|||f}||}y|j|j|�Wn5tk r�|g|j| s$ �