File manager

�
e f�@s#
dd
lZdd
l
Z
dd
lZdd
lZdd
lZdd
lZddlmZ
ddlmZ
ddlm	Z	
dg
Z
ejdkr�dd
lZGdd	�d	e
�Zn>Gd
d	�d	e
�Zdd�Zd
d�Zejee�
Gdd�de
�ZGdd�de
�Zd
S)�N�
)
�context)
�	reduction)
�util�
BufferWrapper�win32c@s@eZ
dZej�Zd
d�Zdd�Zdd�ZdS)�ArenacCs�|
|_xwt
d
�
D]]}dtj�t|j�
f}tjd|
d|�
}tj�dkriPn|j	�
qWt
d�
�
||_||_|j|jf|_
dS)N�dz	pym-%d-%sr�tagnamer
zCannot find name for new mmap���)�size�range�os�getpid�next�_rand�mmap�_winapiZGetLastError�close�FileExistsError�name�buffer�_state)�selfr�
irZbuf�r�9/opt/alt/python34/lib64/python3.4/multiprocessing/heap.py�__init__#s
	





	
	
zArena.__init__c

Cstj
|�

|jS)
N)rZassert_spawningr)
rrrr�__getstate__2s


zArena.__getstate__cCs>|
\|_|_
|_tjd|jd|j
�
|_dS)Nrr
r)rrrrr)r�staterrr�__setstate__6s

!
zArena.__setstate__N)	�__name__�
__module__�__qualname__�tempfileZ_RandomNameSequencerrrr rrrrrsrc@seZ
dZddd�
ZdS)rrc
Cs�|
|_||_
|d	kr�tjddtj�dtj��\|_
}tj|�

tj	|tj
|j
f
�
t|j
ddd�
�}|jd|
�

WdQXnt
j
|j
|j�|_dS)
Nr�prefixzpym-%d-�dir�wb�closefdFs
r)r�fdr$ZmkstemprrrZget_temp_dir�unlink�Finalizer�open�writerr)rrr)r�
frrrr?s
	
	

	
(





zArena.__init__Nr)r!r"r#rrrrrr=sc

Cs=|jdkrt
d�
�
nt|jtj|j�
ffS)NrzDArena is unpicklable because forking was enabled when it was createdr)r)�
ValueError�
rebuild_arenarrZDupFd)
�
arrr�reduce_arenaKs

r2cCst||
j
��S)
N)r�detach)rZdupfdrrrr0Qs
r0c@s�eZ
dZd
Zejdd�
Zedd��
Zdd�Z	dd	�Z
d
d�Zdd
�Zdd�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)rrrrrr^s


	
	
	
	
	

	z
Heap.__init__cCs|
d
}|||@S)Nrr)�
nZ	alignment�maskrrr�_roundupks

z
Heap._roundupc	Cs	
tj
|j|
�}|t|j�
kr�|jt|j|
�tj�}|jd
9_t	j
d|�
t|�
}|jj
|�

|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=)	rrr�length�arena�seq�block�start�stoprrr�_mallocqs"

!














zHeap._mallocc	Cs$
|
\}}}y|j||f}Wnt
k
r:


YnX|j|�
\}}y|j||f}Wnt
k
r{


YnX|j|�
\}}|||f}
||}y|j|j|
�

Wn5t
k
r�


|
g
|j|<tj|j|�
Yn
X|
|j||f<|
|j||f<dS)
N)	r=�KeyError�_absorbr<r;rKrFZinsortr:)	rrPrNrQrRZ
prev_block�
_Z
next_blockrMrrr�_free�s(















z
Heap._freecCs�|
\}}}|j||f=|j
||f=||}|j|}|j|
�

|sv|j|=|jj|�

n||fS)
N)r<r=r;�remover:)rrPrNrQrRrMrOrrrrU�s










zHeap._absorbc
CsQxJy|jj
�}
Wntk
r+


PYn
X|jj|
�

|j|
�

qWdS)
N)rArL�
IndexErrorr?rXrW)rrPrrr�_free_pending_blocks�s






zHeap._free_pending_blocksc
Cse|jj
d
�
s%|jj|
�

n<z+|j�
|jj|
�

|j|
�

Wd|jj�
XdS)NF)	r8�acquirerArKrZr?rXrW�release)rrPrrr�free�s



z	Heap.freec
Cs�tj
�|jkr"|j�
n|jj�
|j�
z�|jt|
d
�|j	�}
|j
|
�
\}}}||
}||kr�|j|||f�

n|||f}|jj
|�

|SWd|jj�
XdS)Nr)rrr6rr8r[rZrDrH�
_alignmentrSrWr?�addr\)rrrNrQrRZnew_stoprPrrr�malloc�s















zHeap.mallocN)r!r"r#r^rrIr�staticmethodrDrSrWrUrZr]r`rrrrr4Zs

r4c@s1eZ
dZe�Zd
d�Zdd�ZdS)rcCsDtj
j|
�
}||
f|_tj|tj
jd
|f
�

dS)N�args)r�_heapr`rrr+r])rrrPrrrr�s

zBufferWrapper.__init__c
Cs3|j\\}
}}}t
|
j�
|||�S)
N)r�
memoryviewr)rrNrQrRrrrr�create_memoryview�s

zBufferWrapper.create_memoryviewN)r!r"r#r4rcrrerrrrr�s	)rFrr�sysr$r7�rrr�__all__�platformr�objectrr2r0�registerr4rrrrr�<module>
s$






	�