B
�/g^�" c @ s� d dl Z d dlZd dlZd dlZd dlZd dlZddlmZmZ ddl m
Z
dgZ ej dkrvd dl
Z
G dd� de�Zn,G d d� de�Zd
d � Zd d
� Ze�ee� G dd� de�ZG dd� de�ZdS )� N� )� reduction�assert_spawning)�util�
BufferWrapperZwin32c @ s, e Zd Ze�� Zdd� Zdd� Zdd� ZdS )�Arenac C sz || _ xTtd�D ]@}dt�� t| j�f }tjd||d�}t�� dkrHP |� � qW t
d��|| _ || _ | j | j f| _
d S )Nid z pym-%d-%s������Ztagnamer z Cannot find name for new mmap)�sizeZrange�os�getpidZnext�_rand�mmap�_winapiZ GetLastError�closeZFileExistsError�name�buffer�_state)�selfr �ir Zbuf� r �,/usr/lib64/python3.7/multiprocessing/heap.py�__init__" s �Arena.__init__c C s t | � | jS �N)r r )r r r r � __getstate__1 s zArena.__getstate__c C s, | \| _ | _| _tjd| j | jd�| _d S )Nr r )r r r r
r )r Zstater r r � __setstate__5 s zArena.__setstate__N) �__name__�
__module__� __qualname__�tempfileZ_RandomNameSequencer r r r r r r r r s r c @ s4 e Zd ZejdkrdgZng Zd dd�Zdd� ZdS )
r Zlinuxz/dev/shmr c C sx || _ || _|dkrbtjdt�� | �|�d�\| _}t�|� t� | tj
| jf� t� | j|� t � | j| j �| _
d S )Nr zpym-%d-)ZprefixZdir)r �fdr Zmkstempr
r � _choose_dirZunlinkr �Finalizer Z ftruncater
r )r r r r r r r r D s
r c C s6 x,| j D ]"}t�|�}|j|j |kr|S qW t�� S r )�_dir_candidatesr
ZstatvfsZf_bavailZf_frsizer Z get_temp_dir)r r ZdZstr r r r! P s
zArena._choose_dirN)r )r r r �sys�platformr# r r! r r r r r > s
c C s( | j dkrtd��t| jt�| j �ffS )Nr zDArena is unpicklable because forking was enabled when it was created)r �
ValueError�
rebuild_arenar r ZDupFd)Zar r r � reduce_arenaY s
r( c C s t | |�� �S r )r Zdetach)r Zdupfdr r r r' _ s r' c @ sZ e Zd ZdZejfdd�Zedd� �Zdd� Z dd � Z
d
d � Z d d
� Z dd� Z
dd� ZdS )�Heapi c C sJ t �� | _t�� | _|| _g | _i | _i | _ i | _
t � | _ g | _
g | _d S r )r
r �_lastpid� threadingZLock�_lock�_size�_lengths� _len_to_seq�_start_to_block�_stop_to_blockZset�_allocated_blocks�_arenas�_pending_free_blocks)r r r r r r l s
z
Heap.__init__c C s |d }| | | @ S )Nr r )ZnZ alignmentZmaskr r r �_roundupy s z
Heap._roundupc C s� t �| j|�}|t| j�krj| �t| j|�tj�}| jd9 _t �
d|� t |�}| j �
|� |d|fS | j| }| j| }|�� }|s�| j|= | j|= |\}}}| j||f= | j||f= |S )Ni z"allocating a new mmap of length %dr )�bisectZ bisect_leftr. Zlenr5 �maxr- r
�PAGESIZEr Zinfor r3 �appendr/ �popr0 r1 ) r r r �length�arena�seq�block�start�stopr r r �_malloc s"
z Heap._mallocc C s� |\}}}y| j ||f }W n tk
r0 Y nX | �|�\}}y| j||f }W n tk
rf Y nX | �|�\}}|||f}|| }y| j| �|� W n. tk
r� |g| j|< t�| j|� Y nX || j||f< || j ||f< d S r ) r1 ZKeyError�_absorbr0 r/ r9 r6 Zinsortr. ) r r> r< r? r@ Z
prev_blockZ_Z
next_blockr; r r r �_free� s(
z
Heap._freec C s^ |\}}}| j ||f= | j||f= || }| j| }|�|� |sV| j|= | j�|� ||fS r )r0 r1 r/ �remover. )r r> r< r? r@ r; r= r r r rB � s
z Heap._absorbc C sF x@y| j �� }W n tk
r& P Y nX | j�|� | �|� qW d S r )r4 r: Z
IndexErrorr2 rD rC �r r> r r r �_free_pending_blocks� s zHeap._free_pending_blocksc C sr t �� | jkr$td�t �� | j���| j�d�s>| j�|� n0z"| � � | j
� |� | � |� W d | j�
� X d S )Nz$My pid ({0:n}) is not last pid {1:n}F)r
r r* r& �formatr, Zacquirer4 r9 rF r2 rD rC ZreleaserE r r r �free� s
z Heap.freec C s� |dk rt d�|���tj|kr.td�|���t�� | jkrD| �� | j �h | �
� | � t |d�| j
�}| �|�\}}}|| }||k r�| �|||f� |||f}| j�|� |S Q R X d S )Nr �Size {0:n} out of range�Size {0:n} too larger )r& rG r$ �maxsize�
OverflowErrorr
r r* r r, rF r5 r7 �
_alignmentrA rC r2 Zadd)r r r< r? r@ Znew_stopr> r r r �malloc� s
z Heap.mallocN)r r r rM r
r8 r Z staticmethodr5 rA rC rB rF rH rN r r r r r) h s
r) c @ s" e Zd Ze� Zdd� Zdd� ZdS )r c C s^ |dk rt d�|���tj|kr.td�|���tj�|�}||f| _t j
| tjj |fd� d S )Nr rI rJ )Zargs) r& rG r$ rK rL r �_heaprN r r r" rH )r r r> r r r r s
zBufferWrapper.__init__c C s& | j \\}}}}t|j�||| � S r )r Z
memoryviewr )r r< r? r@ r r r r �create_memoryview s z BufferWrapper.create_memoryviewN)r r r r) rO r rP r r r r r � s )r6 r
r
r$ r r+ Zcontextr r Z r Z__all__r% r Zobjectr r( r' Zregisterr) r r r r r Z<module>
s$