��]� �@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�ir�buf�r�)/opt/alt/python35/lib64/python3.5/heap.py�__init__#s zArena.__init__cCstj|�|jS)N)r�assert_spawningr)rrrr�__getstate__2s zArena.__getstate__cCs>|\|_|_|_tjd|jd|j�|_dS)Nrr r)rrrrr)r�staterrr�__setstate__6szArena.__setstate__N) �__name__� __module__�__qualname__�tempfile�_RandomNameSequencerrr r"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&�mkstemprrr�get_temp_dir�unlink�Finalizer�openr �writerr)rrr,r�f�bsZzeros�_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_arenarr�DupFd)�arrr�reduce_arenaTsr:cCst||j��S)N)r�detach)rZdupfdrrrr7Zsr7c@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� threading�Lock�_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_leftrC�lenrM�maxrBr�PAGESIZEr�inforrI�appendrD�poprErF) 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|rrArbrMrQ� _alignmentr\r_rH�add)rrrWrZr[Znew_stoprYrrr�malloc�s zHeap.mallocN)r#r$r%rfrrRr�staticmethodrMr\r_r^rbrerhrrrrr<cs r<c@s1eZdZe�Zdd�Zdd�ZdS)rcCsDtjj|�}||f|_tj|tjjd|f�dS)N�args)r�_heaprhrrr0re)rrrYrrrr�szBufferWrapper.__init__cCs3|j\\}}}}t|j�|||�S)N)r� memoryviewr)rrWrZr[rrrr�create_memoryview�szBufferWrapper.create_memoryviewN)r#r$r%r<rkrrmrrrrr�s )rOrr�sysr&r?�rrr�__all__�platformr�objectrr:r7�registerr<rrrrr� s$ �