???<!-- GIF89;a -->
123123123123
.....................................................................................................................................???<!-- GIF89;a -->
123123123123
.....................................................................................................................................3

  \                  @   s   d dl Z d dlZd dlZd dlZd dlZd dlZddlmZmZ ddl	m
Z
 dgZejdkrvd dlZG dd deZn,G d	d deZd
d Zdd Zejee G dd deZG dd deZdS )    N   )	reductionassert_spawning)utilBufferWrapperZwin32c               @   s,   e Zd Zej Zdd Zdd Zdd ZdS )Arenac             C   sz   || _ xTtdD ]@}dtj t| jf }tjd||d}tj dkrHP |j	  qW t
d|| _|| _| j | jf| _d S )Nd   z	pym-%d-%sr   )tagnamer   zCannot find name for new mmap)sizerangeosgetpidnext_randmmap_winapiZGetLastErrorcloseFileExistsErrornamebuffer_state)selfr   ir   Zbuf r   ,/usr/lib64/python3.6/multiprocessing/heap.py__init__"   s    zArena.__init__c             C   s   t |  | jS )N)r   r   )r   r   r   r   __getstate__1   s    zArena.__getstate__c             C   s,   | \| _ | _| _tjd| j | jd| _d S )Nr   )r	   r
   )r   r   r   r   r   )r   stater   r   r   __setstate__5   s    zArena.__setstate__N)	__name__
__module____qualname__tempfileZ_RandomNameSequencer   r   r   r   r   r   r   r   r      s   r   c               @   s   e Zd ZdddZdS )r   r   c             C   s   || _ || _|d	krtjdtj  tj d\| _}tj| tj	| tj
| jf t| jddd^}d
}||krd| }xt|| D ]}|j| qW ~|jd||   |j |kstW d Q R X tj| j| j | _d S )Nr   zpym-%d-)prefixdirwbF)closefdi       r
   i   )r   fdr#   Zmkstempr   r   r   Zget_temp_dirunlinkFinalizer   openr   writetellAssertionErrorr   r   )r   r   r)   r   fZbsZzeros_r   r   r   r   @   s"    
zArena.__init__Nr
   )r
   )r    r!   r"   r   r   r   r   r   r   >   s   c             C   s(   | j dkrtdt| jtj| j ffS )Nr   zDArena is unpicklable because forking was enabled when it was createdr
   )r)   
ValueErrorrebuild_arenar   r   ZDupFd)ar   r   r   reduce_arenaS   s    
r5   c             C   s   t | |j S )N)r   detach)r   Zdupfdr   r   r   r3   Y   s    r3   c               @   sZ   e Zd ZdZejfddZedd Zdd Z	dd	 Z
d
d Zdd Zdd Zdd ZdS )Heap   c             C   sJ   t j | _tj | _|| _g | _i | _i | _	i | _
t | _g | _g | _d S )N)r   r   _lastpid	threadingZLock_lock_size_lengths_len_to_seq_start_to_block_stop_to_blockset_allocated_blocks_arenas_pending_free_blocks)r   r   r   r   r   r   f   s    

zHeap.__init__c             C   s   |d }| | | @ S )Nr   r   )nZ	alignmentmaskr   r   r   _roundups   s    zHeap._roundupc       	      C   s   t j| j|}|t| jkrj| jt| j|tj}|  jd9  _t	j
d| t|}| jj| |d|fS | j| }| j| }|j }|s| j|= | j|= |\}}}| j||f= | j||f= |S )N   z"allocating a new mmap of length %dr   )bisectZbisect_leftr=   lenrG   maxr<   r   PAGESIZEr   infor   rC   appendr>   popr?   r@   )	r   r   r   lengtharenaseqblockstartstopr   r   r   _mallocy   s"    



zHeap._mallocc       	      C   s   |\}}}y| j ||f }W n tk
r0   Y nX | j|\}}y| j||f }W n tk
rf   Y nX | j|\}}|||f}|| }y| j| j| W n. tk
r   |g| j|< tj| j| Y nX || j||f< || j ||f< d S )N)	r@   KeyError_absorbr?   r>   rN   rI   Zinsortr=   )	r   rS   rQ   rT   rU   Z
prev_blockr1   Z
next_blockrP   r   r   r   _free   s(    

z
Heap._freec             C   s^   |\}}}| j ||f= | j||f= || }| j| }|j| |sV| j|= | jj| ||fS )N)r?   r@   r>   remover=   )r   rS   rQ   rT   rU   rP   rR   r   r   r   rX      s    


zHeap._absorbc             C   sF   x@y| j j }W n tk
r&   P Y nX | jj| | j| qW d S )N)rD   rO   
IndexErrorrB   rZ   rY   )r   rS   r   r   r   _free_pending_blocks   s    zHeap._free_pending_blocksc          
   C   s`   t j | jkst| jjds,| jj| n0z"| j  | j	j
| | j| W d | jj  X d S )NF)r   r   r9   r/   r;   acquirerD   rN   r\   rB   rZ   rY   release)r   rS   r   r   r   free   s    
z	Heap.freec             C   s   d|  kot jk n  sttj | jkr4| j  | jh | j  | j	t
|d| j}| j|\}}}|| }||k r| j|||f |||f}| jj| |S Q R X d S )Nr   r   )sysmaxsizer/   r   r   r9   r   r;   r\   rG   rK   
_alignmentrV   rY   rB   add)r   r   rQ   rT   rU   Znew_stoprS   r   r   r   malloc   s    
zHeap.mallocN)r    r!   r"   rb   r   rL   r   staticmethodrG   rV   rY   rX   r\   r_   rd   r   r   r   r   r7   b   s   
r7   c               @   s"   e Zd Ze Zdd Zdd ZdS )r   c             C   sN   d|  kot jk n  sttjj|}||f| _tj| tjj	|fd d S )Nr   )args)
r`   ra   r/   r   _heaprd   r   r   r+   r_   )r   r   rS   r   r   r   r      s    
zBufferWrapper.__init__c             C   s&   | j \\}}}}t|j|||  S )N)r   
memoryviewr   )r   rQ   rT   rU   r   r   r   r   create_memoryview   s    zBufferWrapper.create_memoryviewN)r    r!   r"   r7   rg   r   ri   r   r   r   r   r      s   )rI   r   r   r`   r#   r:   contextr   r    r   __all__platformr   objectr   r5   r3   registerr7   r   r   r   r   r   <module>
   s$   
  