Current File : //usr/local/lib/python3.8/multiprocessing/__pycache__/heap.cpython-38.pyc
U
qaj- @ s d dl Z d dlmZ d dlZd dlZd dlZd dlZd dlZddlm Z m
Z
ddl m Z dgZ
ejdkrd dlZG dd d eZn,G d
d d eZd d Zd
d Ze ee G dd deZG dd deZdS ) N) defaultdict ) reductionassert_spawning)util
BufferWrapperwin32c @ s0 e Zd ZdZe Zdd Zdd Zdd Z dS ) ArenazL
A shared memory area backed by anonymous memory (Windows).
c C sx || _ tdD ]B}dt t| jf }tjd||d}t dkrH qZ| qt
d|| _ || _ | j | j f| _
d S )Nd z pym-%d-%sZtagnamer z Cannot find name for new mmap)sizerangeosgetpidnext_randmmap_winapiZ GetLastErrorcloseFileExistsErrornamebuffer_state)selfr
ir buf r 0/usr/local/lib/python3.8/multiprocessing/heap.py__init__& s
Arena.__init__c C s t | | jS N)r r )r r r r __getstate__5 s zArena.__getstate__c C s, | \| _ | _| _tjd| j | jd| _d S )Nr r )r
r r r r )r stater r r __setstate__9 s zArena.__setstate__N)
__name__
__module__ __qualname____doc__tempfile_RandomNameSequencer r r" r$ r r r r r s
r c @ s8 e Zd ZdZejdkr dgZng Zd
ddZdd Zd S ) r zJ
A shared memory area backed by a temporary file (POSIX).
linuxz/dev/shmr c C sx || _ || _|dkrbtjdt | |d\| _}t| t | tj
| jf t | j| t | j| j | _
d S )Nr zpym-%d-)prefixdir)r
fdr) mkstempr r _choose_dirunlinkr Finalizer ftruncater r )r r
r. r r r r r M s
r c C s6 | j D ]&}t|}|j|j |kr| S qt S r! )_dir_candidatesr statvfsf_bavailf_frsizer get_temp_dir)r r
dstr r r r0 [ s
zArena._choose_dirN)r ) r% r& r' r( sysplatformr4 r r0 r r r r r C 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 DupFd)ar r r reduce_arenad s
rA c C s t | | S r! )r detach)r
Zdupfdr r r r> j s r> c @ sz e Zd ZdZdZdZejfddZe dd Z
dd Z d d
Z d d Z
d
d Zdd Zdd Zdd Zdd Zdd ZdS )Heap i @ c C sX t | _t | _|| _g | _i | _i | _ i | _
t t | _
g | _g | _d| _d| _d S Nr )r r _lastpid threadingLock_lock_size_lengths _len_to_seq_start_to_block_stop_to_blockr set_allocated_blocks_arenas_pending_free_blocks
_n_mallocs_n_frees)r r
r r r r { s
z
Heap.__init__c C s |d }| | | @ S )Nr r )nZ alignmentmaskr r r _roundup s z
Heap._roundupc C sZ | t| j|tj}| j| jk r0| jd9 _td| t|}| j
| |d|fS )N z"allocating a new mmap of length %dr ) rW maxrJ r PAGESIZE_DOUBLE_ARENA_SIZE_UNTILr infor rQ append)r r
lengtharenar r r
_new_arena s zHeap._new_arenac C s |j }|| jk rd S | j|}|r(t| j|df= | j||f= | j| | j | }||d|f |s~| j |= | j
| d S rE ) r
_DISCARD_FREE_SPACE_LARGER_THANrP popAssertionErrorrM rN rQ removerL rK )r r_ r^ blocksseqr r r _discard_arena s
zHeap._discard_arenac C s| t | j|}|t| jkr&| |S | j| }| j| }| }|sV| j|= | j|= |\}}}| j||f= | j||f= |S r! ) bisect bisect_leftrK lenr` rL rb rM rN ) r r
r r^ rf blockr_ startstopr r r _malloc s
z Heap._mallocc C s |\}}}z| j ||f }W n tk
r0 Y nX | |\}}z| j||f }W n tk
rf Y nX | |\}}|||f}|| }z| j| | W n. tk
r |g| j|< t| j| Y nX || j||f< || j ||f< d S r! ) rN KeyError_absorbrM rL r] rh insortrK ) r rk r_ rl rm Z
prev_block_Z
next_blockr^ r r r _add_free_block s(
zHeap._add_free_blockc C s^ |\}}}| j ||f= | j||f= || }| j| }|| |sV| j|= | j| ||fS r! )rM rN rL rd rK )r rk r_ rl rm r^ rf r r r rp s
z Heap._absorbc C s4 |\}}}| j | }|||f |s0| | d S r! )rP rd rg )r rk r_ rl rm re r r r _remove_allocated_block s
z Heap._remove_allocated_blockc C sB z| j }W n tk
r& Y q>Y nX | | | | q d S r! )rR rb
IndexErrorrs rt r rk r r r _free_pending_blocks s
zHeap._free_pending_blocksc C s~ t | jkr$tdt | j| jds>| j| n<z.| j
d7 _
| | | |
| W 5 | j X d S )Nz$My pid ({0:n}) is not last pid {1:n}Fr )r r rF r= formatrI acquirerR r] releaserT rw rs rt rv r r r free s
z Heap.freec
C s |dk rt d|tj|kr.td|t | jkrD| | j | j
d7 _
| | t
|d| j}| |\}}}|| }||k r| |||f | j| ||f |||fW 5 Q R S Q R X d S )Nr Size {0:n} out of rangeSize {0:n} too larger )r= rx r; maxsize
OverflowErrorr r rF r rI rS rw rW rY
_alignmentrn rs rP add)r r
r_ rl rm Z real_stopr r r malloc( s
z Heap.mallocN)r% r& r' r ra r[ r rZ r staticmethodrW r` rg rn rs rp rt rw r{ r r r r r rC s s
rC 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 r| r} )args) r= rx r; r~ r r _heapr r r r2 r{ )r r
rk r r r r F s
zBufferWrapper.__init__c C s& | j \\}}}}t|j||| S r! )r
memoryviewr )r r_ rl rm r
r r r create_memoryviewO s z BufferWrapper.create_memoryviewN)r% r& r' rC r r r r r r r r B s )rh collectionsr r r r; r) rG contextr r r __all__r< r objectr rA r> registerrC r r r r r <module>
s&
$! P
Mini Shell