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
BufferWrapperZwin32c @ 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 Zbuf 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
r6 c C s t | | S )N)r detach)r Zdupfdr r r r4 _ s r4 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> lenrH maxr= r PAGESIZEr infor rD appendr? popr@ rA ) 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) rA KeyError_absorbr@ r? rO rJ Zinsortr> ) r rT rR rU rV Z
prev_block_Z
next_blockrQ r r r _free s(
z
Heap._freec C s^ |\}}}| j ||f= | j||f= || }| j| }|| |sV| j|= | j| ||fS )N)r@ rA r? remover> )r rT rR rU rV rQ rS r r r rY s
z Heap._absorbc C sF x@y| j }W n tk
r& P Y nX | j| | | qW d S )N)rE rP
IndexErrorrC r\ r[ )r rT 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: r3 formatr< acquirerE rO r^ rC r\ r[ release)r rT 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 )r3 r_ r1 maxsize
OverflowErrorr
r r: r r< r^ rH rL
_alignmentrW r[ rC add)r r rR rU rV Znew_stoprT r r r malloc s
z Heap.mallocN)r r! r" re r rM r staticmethodrH rW r[ rY r^ rb rg r r r r r8 h s
r8 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) r3 r_ r1 rc rd r _heaprg r r r) rb )r r rT r r r r s
zBufferWrapper.__init__c C s& | j \\}}}}t|j||| S )N)r
memoryviewr )r rR rU rV r r r r create_memoryview s z BufferWrapper.create_memoryviewN)r r! r" r8 rj r rl r r r r r s )rJ r r
r1 r# r; contextr r r __all__r2 r objectr r6 r4 registerr8 r r r r r
s$