3
\ @ s d dl Z d dlZd dlZd dlZd dlZd dlZddlmZmZ ddl m
Z
dgZ ej dkrvd dl
Z
G dd deZn,G d d deZd
d Zd d
Zejee G dd deZG dd deZdS ) N ) reductionassert_spawning)util
BufferWrapperZwin32c @ s, e Zd Zej Zdd Zdd Zdd ZdS )Arenac C sz || _ xTtdD ]@}dtj t| jf }tjd||d}tj dkrHP |j qW t
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/lib64/python3.6/multiprocessing/heap.py__init__" s zArena.__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 )r stater r r __setstate__5 s zArena.__setstate__N) __name__
__module__ __qualname__tempfileZ_RandomNameSequencer r r r r r r r r s r c @ s e Zd ZdddZdS )r r c C s || _ || _|d krtjdtj tj d\| _}tj| tj | tj
| jf t | jdddN}d
}||krd| }x t || D ]}|j
| qW ~|j
d|| W d Q R X tj| j| j | _d S ) Nr zpym-%d-)prefixdirwbF)closefdi r
i )r fdr# Zmkstempr
r r Z get_temp_dirunlinkFinalizer openr writer r )r r r) r fZbsZzeros_r r r r @ s"
zArena.__init__Nr
)r
)r r! r" r r r r r r > s c C s( | j dkrtdt| jtj| j ffS )Nr zDArena is unpicklable because forking was enabled when it was createdr
)r)
ValueError
rebuild_arenar r ZDupFd)ar r r reduce_arenaS s
r3 c C s t | |j S )N)r detach)r Zdupfdr r r r1 Y s r1 c @ sZ e Zd ZdZejfddZedd Zdd Z dd Z
d
d Z d d
Z dd Z
dd ZdS )Heap c C sJ t j | _tj | _|| _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 f s
z
Heap.__init__c C s |d }| | | @ S )Nr r )nZ alignmentmaskr r r _roundups s z
Heap._roundupc C s t j| j|}|t| jkrj| jt| j|tj}| jd9 _t j
d| t |}| j j
| |d|fS | j| }| j| }|j }|s| j|= | j|= |\}}}| 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 lengtharenaseqblockstartstopr r r _mallocy s"
z Heap._mallocc C s |\}}}y| j ||f }W n tk
r0 Y nX | j|\}}y| j||f }W n tk
rf Y nX | j|\}}|||f}|| }y| j| j| W n. tk
r |g| j|< tj| j| Y nX || j||f< || j ||f< d S )N) r> KeyError_absorbr= r< rL rG Zinsortr; ) r rQ rO rR rS Z
prev_blockr/ Z
next_blockrN r r r _free s(
z
Heap._freec C s^ |\}}}| j ||f= | j||f= || }| j| }|j| |sV| j|= | jj| ||fS )N)r= r> r< remover; )r rQ rO rR rS rN rP r r r rV s
z Heap._absorbc C sF x@y| j j }W n tk
r& P Y nX | jj| | j| qW d S )N)rB rM
IndexErrorr@ rX rW )r rQ r r r _free_pending_blocks s zHeap._free_pending_blocksc
C sN | j jds| jj| n0z"| j | jj| | j| W d | j j X d S )NF) r9 acquirerB rL rZ r@ rX rW release)r rQ r r r free s z Heap.freec C s t j | jkr| j | jh | j | jt|d| j}| j |\}}}|| }||k rj| j
|||f |||f}| j j | |S Q R X d S )Nr )
r
r r7 r r9 rZ rE rI
_alignmentrT rW r@ add)r r rO rR rS Znew_stoprQ r r r malloc s
z Heap.mallocN)r r! r" r^ r rJ r staticmethodrE rT rW rV rZ r] r` r r r r r5 b s
r5 c @ s" e Zd Ze Zdd Zdd ZdS )r c C s0 t jj|}||f| _tj| t jj|fd d S )N)args)r _heapr` r r r+ r] )r r rQ r r r r s
zBufferWrapper.__init__c C s& | 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 rc r re r r r r r s )rG r r
sysr# r8 contextr r r __all__platformr objectr r3 r1 registerr5 r r r r r
s$