��Yf� �@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�||_xttd�D]Z}dtj�t|j�f}tjd|d|�}tj�dkrfP|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/python35/lib64/python3.5/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__6szArena.__setstate__N) �__name__� __module__�__qualname__�tempfileZ_RandomNameSequencerrrr rrrrrsrc@seZdZddd�ZdS)rrcCs||_||_|d kr�tjddtj�dtj��\|_}tj|�tj |tj |jf�t|jddd��c}d}||kr�d |}x%t||�D]}|j |�q�W~|j d ||�WdQRXtj|j|j�|_dS)Nr�prefixzpym-%d-�dir�wb�closefdFisri)r�fdr$ZmkstemprrrZget_temp_dir�unlink�Finalizer�openr �writerr)rrr)r�fZbsZzeros�_rrrrAs" ( zArena.__init__Nr)r!r"r#rrrrrr?scCs:|jdkrtd��t|jtj|j�ffS)NrzDArena is unpicklable because forking was enabled when it was createdr)r)� ValueError� rebuild_arenarrZDupFd)�arrr�reduce_arenaTsr3cCst||j��S)N)r�detach)rZdupfdrrrr1Zsr1c@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)rrrrrrgs z Heap.__init__cCs|d}|||@S)Nrr)�nZ alignment�maskrrr�_roundupts z Heap._roundupc Cstj|j|�}|t|j�kr�|jt|j|�tj�}|jd9_t j d|�t|�}|jj |�|d|fS|j|}|j|}|j�}|s�|j|=|j|=|\}}}|j||f=|j||f=|S)N�z"allocating a new mmap of length %dr)�bisectZbisect_leftr;�lenrE�maxr:r�PAGESIZEr�inforrA�appendr<�popr=r>) rrr�length�arena�seq�block�start�stoprrr�_malloczs"! 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|�KeyError�_absorbr=r<rLrGZinsortr;) rrQrOrRrSZ prev_blockr/Z next_blockrNrrr�_free�s( z Heap._freecCs}|\}}}|j||f=|j||f=||}|j|}|j|�|ss|j|=|jj|�||fS)N)r=r>r<�remover;)rrQrOrRrSrNrPrrrrV�s zHeap._absorbcCsQxJy|jj�}Wntk r+PYnX|jj|�|j|�qWdS)N)rBrM� IndexErrorr@rXrW)rrQrrr�_free_pending_blocks�s zHeap._free_pending_blocksc Cse|jjd�s%|jj|�n<z+|j�|jj|�|j|�Wd|jj�XdS)NF) r9�acquirerBrLrZr@rXrW�release)rrQrrr�free�s z Heap.freecCs�tj�|jkr|j�|j��|j�|jt|d�|j�}|j |�\}}}||}||kr�|j |||f�|||f}|jj|�|SWdQRXdS)Nr) rrr7rr9rZrErI� _alignmentrTrWr@�add)rrrOrRrSZnew_stoprQrrr�malloc�s zHeap.mallocN)r!r"r#r^rrJr�staticmethodrErTrWrVrZr]r`rrrrr5cs r5c@s1eZdZe�Zdd�Zdd�ZdS)rcCsDtjj|�}||f|_tj|tjjd|f�dS)N�args)r�_heapr`rrr+r])rrrQrrrr�szBufferWrapper.__init__cCs3|j\\}}}}t|j�|||�S)N)r� memoryviewr)rrOrRrSrrrr�create_memoryview�szBufferWrapper.create_memoryviewN)r!r"r#r5rcrrerrrrr�s )rGrr�sysr$r8�rrr�__all__�platformr�objectrr3r1�registerr5rrrrr� s$ �