B
'c Aɷ @ 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
Zeee G dd deZG dd deZdS ) N ) reductionassert_spawning)util
BufferWrapperwin32c @ s, e Zd Ze Zdd Zdd Zdd ZdS )Arenac C sz || _ xTtdD ]@}dt t| jf }tjd||d}t dkrHP | qW t
d|| _ || _ | j | j f| _
d S )Nd z pym-%d-%s)tagnamer z Cannot find name for new mmap)sizerangeosgetpidnext_randmmap_winapiZ GetLastErrorcloseFileExistsErrornamebuffer_state)selfr ir buf r 9/opt/alt/python37/lib64/python3.7/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 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 @ s4 e Zd ZejdkrdgZng Zd ddZdd ZdS )
r Zlinuxz/dev/shmr
c C sx || _ || _|dkrbtjdt | |d\| _}t| t | tj
| jf t | j| t | j| j | _
d S )Nr
zpym-%d-)prefixdir)r fdr% Zmkstempr r _choose_dirunlinkr Finalizer ftruncater r )r r r( r r r r r D s
zArena.__init__c C s6 x,| j D ]"}t|}|j|j |kr|S qW t S )N)_dir_candidatesr statvfsf_bavailf_frsizer Z get_temp_dir)r r dstr r r r) P s
zArena._choose_dirN)r
)r" r# r$ sysplatformr- r r) r r r r r > s
c C s( | j dkrtdt| jt| j ffS )Nr
zDArena is unpicklable because forking was enabled when it was created)r(
ValueError
rebuild_arenar r ZDupFd)ar r r reduce_arenaY s
r8 c C s t | | S )N)r detach)r Zdupfdr r r r6 _ s r6 c @ sZ e Zd ZdZejfddZedd Zdd Z dd Z
d
d Z d d
Z dd Z
dd ZdS )Heap c C sJ t | _t | _|| _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 l s
z
Heap.__init__c C s |d }| | | @ S )Nr r )nZ alignmentmaskr r r _roundupy s z
Heap._roundupc C s t | j|}|t| jkrj| t| j|tj}| jd9 _t
d| t |}| j
| |d|fS | j| }| j| }| }|s| j|= | j|= |\}}}| j||f= | j||f= |S )N z"allocating a new mmap of length %dr )bisectZ bisect_leftr@ lenrJ maxr? r PAGESIZEr infor rF appendrA poprB rC ) r r r lengtharenaseqblockstartstopr r r _malloc s"
z Heap._mallocc C s |\}}}y| j ||f }W n tk
r0 Y nX | |\}}y| j||f }W n tk
rf Y nX | |\}}|||f}|| }y| j| | W n. tk
r |g| j|< t| j| Y nX || j||f< || j ||f< d S )N) rC KeyError_absorbrB rA rQ rL Zinsortr@ ) r rV rT rW rX Z
prev_block_Z
next_blockrS r r r _free s(
z
Heap._freec C s^ |\}}}| j ||f= | j||f= || }| j| }|| |sV| j|= | j| ||fS )N)rB rC rA remover@ )r rV rT rW rX rS rU r r r r[ s
z Heap._absorbc C sF x@y| j }W n tk
r& P Y nX | j| | | qW d S )N)rG rR
IndexErrorrE r^ r] )r rV r r r _free_pending_blocks s zHeap._free_pending_blocksc C sr t | jkr$tdt | j| jds>| j| n0z"| | j
| | | W d | j
X d S )Nz$My pid ({0:n}) is not last pid {1:n}F)r r r< r5 formatr> acquirerG rQ r` rE r^ r] release)r rV r r r free s
z Heap.freec C s |dk rt d|tj|kr.td|t | jkrD| | j h |
| t |d| j
}| |\}}}|| }||k r| |||f |||f}| j| |S Q R X d S )Nr zSize {0:n} out of rangezSize {0:n} too larger )r5 ra r3 maxsize
OverflowErrorr r r< r r> r` rJ rN
_alignmentrY r] rE add)r r rT rW rX Znew_stoprV r r r malloc s
z Heap.mallocN)r" r# r$ rg r rO r staticmethodrJ rY r] r[ r` rd ri r r r r r: h s
r: c @ s" e Zd Ze Zdd Zdd ZdS )r c C s^ |dk rt d|tj|kr.td|tj|}||f| _t j
| tjj |fd d S )Nr zSize {0:n} out of rangezSize {0:n} too large)args) r5 ra r3 re rf r _heapri r r r+ rd )r r rV r r r r s
zBufferWrapper.__init__c C s& | j \\}}}}t|j||| S )N)r
memoryviewr )r rT rW rX r r r r create_memoryview s z BufferWrapper.create_memoryviewN)r" r# r$ r: rl r rn r r r r r s )rL r r r3 r% r= contextr r r __all__r4 r objectr r8 r6 registerr: r r r r r
s$