
][\                 @   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 d d l m Z d d l m	 Z	 d g Z
 e j d k r d  d l Z Gd d	   d	 e  Z n> Gd
 d	   d	 e  Z d d   Z d d   Z e j e e  Gd d   d e  Z Gd d   d e  Z d S)    N   )context)	reduction)utilBufferWrapperZwin32c               @   s@   e  Z d  Z e j   Z d d   Z d d   Z d d   Z d S)Arenac             C   s   | |  _  xw t d  D]] } d t j   t |  j  f } t j d | d | } t j   d k ri Pn  | j	   q Wt
 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_winapiGetLastErrorcloseFileExistsErrornamebuffer_state)selfr   ir   Zbuf r   */usr/lib/python3.4/multiprocessing/heap.py__init__#   s    			zArena.__init__c             C   s   t  j |   |  j S)N)r   Zassert_spawningr   )r   r   r   r   __getstate__2   s    zArena.__getstate__c             C   sY   | \ |  _  |  _ |  _ t j d |  j  d |  j |  _ t j   t j k sU t  d  S)Nr   r	   r
   )	r   r   r   r   r   r   r   ZERROR_ALREADY_EXISTSAssertionError)r   stater   r   r   __setstate__6   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 r t j d d t j   d t j    \ |  _ } t j |  t j	 |  t j
 |  j f  t |  j d d d  } | j d |  Wd  QXn  t j |  j |  j   |  _ d  S)
Nr   prefixzpym-%d-dirwbclosefdFs    r
   )r   fdr%   Zmkstempr   r   r   Zget_temp_dirunlinkFinalizer   openwriter   r   )r   r   r*   r   fr   r   r   r   ?   s    			(zArena.__init__Nr
   )r"   r#   r$   r   r   r   r   r   r   =   s   c             C   s=   |  j  d k r t d   n  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_arenaK   s    r3   c             C   s   t  |  | j    S)N)r   detach)r   Zdupfdr   r   r   r1   Q   s    r1   c               @   s   e  Z d  Z d Z e j 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   sm   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   ^   s    						zHeap.__init__c             C   s   | d } |  | | @S)Nr   r   )nZ	alignmentmaskr   r   r   _roundupk   s    
zHeap._roundupc       	      C   s	  t  j |  j |  } | t |  j  k r |  j t |  j |  t j  } |  j d 9_ t	 j
 d |  t |  } |  j j |  | d | f S|  j | } |  j | } | j   } | s |  j | =|  j | =n  | \ } } } |  j | | f =|  j | | f =| S)N   z"allocating a new mmap of length %dr   )bisectZbisect_leftr;   lenrE   maxr:   r   PAGESIZEr   infor   rA   appendr<   popr=   r>   )	r   r   r   lengtharenaseqblockstartstopr   r   r   _mallocq   s"    !zHeap._mallocc       	      C   s$  | \ } } } y |  j  | | f } Wn t k
 r: Yn X|  j |  \ } } y |  j | | f } Wn t k
 r{ Yn X|  j |  \ } } | | | f } | | } y |  j | j |  Wn5 t k
 r | g |  j | <t j |  j |  Yn X| |  j | | f <| |  j  | | f <d  S)N)	r>   KeyError_absorbr=   r<   rL   rG   Zinsortr;   )	r   rQ   rO   rR   rS   Z
prev_block_Z
next_blockrN   r   r   r   _free   s(    
z
Heap._freec             C   s   | \ } } } |  j  | | f =|  j | | f =| | } |  j | } | j |  | sv |  j | =|  j j |  n  | | f S)N)r=   r>   r<   remover;   )r   rQ   rO   rR   rS   rN   rP   r   r   r   rV      s    

zHeap._absorbc             C   sP   xI y |  j  j   } Wn t k
 r+ PYn X|  j j |  |  j |  q d  S)N)rB   rM   
IndexErrorr@   rY   rX   )r   rQ   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 |  n< z+ |  j   |  j	 j
 |  |  j |  Wd  |  j j   Xd  S)NF)r   r   r7   r   r9   acquirerB   rL   r[   r@   rY   rX   release)r   rQ   r   r   r   free   s    

z	Heap.freec          
   C   s   d | k o t  j k  n s% t  t j   |  j k rG |  j   n  |  j j   |  j	   z |  j
 t | d  |  j  } |  j |  \ } } } | | } | | k  r |  j | | | f  n  | | | f } |  j j |  | SWd  |  j j   Xd  S)Nr   r   )sysmaxsizer   r   r   r7   r   r9   r\   r[   rE   rI   
_alignmentrT   rX   r@   addr]   )r   r   rO   rR   rS   Znew_stoprQ   r   r   r   malloc   s    %

zHeap.mallocN)r"   r#   r$   ra   r   rJ   r   staticmethodrE   rT   rX   rV   r[   r^   rc   r   r   r   r   r5   Z   s   
r5   c               @   s1   e  Z d  Z e   Z d d   Z d d   Z d S)r   c             C   si   d | k o t  j k  n s% t  t j j |  } | | f |  _ t j |  t j j	 d | f d  S)Nr   args)
r_   r`   r   r   _heaprc   r   r   r,   r^   )r   r   rQ   r   r   r   r      s    %zBufferWrapper.__init__c             C   s3   |  j  \ \ } } } } t | j  | | |  S)N)r   
memoryviewr   )r   rO   rR   rS   r   r   r   r   create_memoryview   s    zBufferWrapper.create_memoryviewN)r"   r#   r$   r5   rf   r   rh   r   r   r   r   r      s   	)rG   r   r   r_   r%   r8    r   r   r   __all__platformr   objectr   r3   r1   registerr5   r   r   r   r   r   <module>
   s$   	