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 | jddd^}d
}||krd| }x t || D ]}|j
| qW ~|j
d|| |j |kstW 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 writetellAssertionErrorr 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
r5 c C s t | |j S )N)r detach)r Zdupfdr r r r3 Y s r3 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= lenrG maxr< r PAGESIZEr infor rC 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> rN rI Zinsortr= ) 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|= | jj| ||fS )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 tk
r& P Y nX | jj| | j| qW d S )N)rD rO
IndexErrorrB rZ rY )r rS r r r _free_pending_blocks s zHeap._free_pending_blocksc
C s` t j | jkst| jjds,| jj| n0z"| j | j j
| | j | W d | jj 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| kot jk n s ttj | jkr4| j | jh | j | j t
|d| j }| j |\}}}|| }||k r| j
|||f |||f}| jj| |S Q R X d S )Nr r )sysmaxsizer/ r
r r9 r r; r\ rG rK
_alignmentrV rY rB add)r r rQ rT rU Znew_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 Zd Ze Zdd Zdd ZdS )r c C sN d| kot jk n s ttjj|}||f| _tj| tjj |fd d S )Nr )args)
r` ra r/ r _heaprd r r r+ r_ )r r rS r r r r s
zBufferWrapper.__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
s$