File: //usr/lib/python3.5/multiprocessing/__pycache__/heap.cpython-35.pyc
Fa @ 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
BufferWrapperZwin32c @ s@ e Z d Z e j Z d d Z d d Z d d Z d S)Arenac C s | | _ xt t d D]Z } d t j t | j f } t j d | d | } t j d k rf P| j q Wt
d | | _ | | _ | j | j f | _
d S)Nd z pym-%d-%sr tagnamer z Cannot find name for new mmap)sizerangeosgetpidnext_randmmap_winapiZ GetLastErrorcloseFileExistsErrornamebuffer_state)selfr ir Zbuf r */usr/lib/python3.5/multiprocessing/heap.py__init__# s zArena.__init__c C s t j | | j S)N)r Zassert_spawningr )r r r r __getstate__2 s
zArena.__getstate__c C s> | \ | _ | _ | _ t j d | j d | j | _ d S)Nr r r
)r r r r r )r stater r r __setstate__6 s zArena.__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 d t j d t j \ | _ } t j | t j | t j
| j f t | j d d d { } d } | | k r d | } x% t | | D] } | j
| q W~ | j
d | | | j | k st Wd QRXt j | j | j | _ d S) Nr prefixzpym-%d-dirwbclosefdFi s r
i )r fdr# Zmkstempr
r r Z get_temp_dirunlinkFinalizer openr writetellAssertionErrorr r )r r r( r fZbsZzeros_r r r r A s" (
zArena.__init__Nr
)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 ZDupFd)ar r r reduce_arenaT s r4 c C s t | | j S)N)r detach)r Zdupfdr r r r2 Z s r2 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 threadingZLock_lock_size_lengths _len_to_seq_start_to_block_stop_to_blockset_allocated_blocks_arenas_pending_free_blocks)r r r r r r g s z
Heap.__init__c C s | d } | | | @S)Nr r )nZ alignmentmaskr r r _roundupt 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 | =| \ } } } | j | | f =| j | | f =| S)N z"allocating a new mmap of length %dr )bisectZ bisect_leftr< lenrF maxr; r PAGESIZEr infor rB appendr= popr> r? ) r r r lengtharenaseqblockstartstopr r r _mallocz 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= rM rH Zinsortr< ) r rR rP rS rT Z
prev_blockr0 Z
next_blockrO r r r _free s(
z
Heap._freec C s} | \ } } } | j | | f =| j | | f =| | } | j | } | j | | ss | j | =| j j | | | f S)N)r> r? r= remover< )r rR rP rS rT rO rQ r r r rW s
z Heap._absorbc C sQ xJ y | j j } Wn t k
r+ PYn X| j j | | j | q Wd S)N)rC rN
IndexErrorrA rY rX )r rR r r r _free_pending_blocks s
zHeap._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 r8 r. r: acquirerC rM r[ rA rY rX release)r rR r r r free s
z Heap.freec C s d | k o t j k n s% t t j | j k rD | j | j | j | j t
| d | j } | j | \ } } } | | } | | k r | j
| | | f | | | f } | j j | | SWd QRXd S)Nr r )sysmaxsizer. r
r r8 r r: r[ rF rJ
_alignmentrU rX rA add)r r rP rS rT Znew_stoprR r r r malloc s %
z Heap.mallocN)r r! r" ra r rK r staticmethodrF rU rX rW r[ r^ rc r r r r r6 c s
r6 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 rR r r r r s %zBufferWrapper.__init__c C s3 | j \ \ } } } } t | j | | | S)N)r
memoryviewr )r rP rS rT r r r r create_memoryview s z BufferWrapper.create_memoryviewN)r r! r" r6 rf r rh r r r r r s )rH r r
r_ r# r9 r r r __all__platformr objectr r4 r2 registerr6 r r r r r <module>
s$