/usr/lib/python3.10/multiprocessing/__pycache__
Edit: /usr/lib/python3.10/multiprocessing/__pycache__/heap.cpython-310.pyc (7668B)
o
9jj- @ s d dl Z d dlmZ d dlZd dlZd dlZd dlZd dlZddlm Z m
Z
ddl m Z dgZ
ejdkrAd dlZG dd d eZnG 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 ]!}dt t| jf }tjd||d}t dkr$ n | qt
d|| _ || _ | j | j f| _
d S )Nd z pym-%d-%stagnamer z Cannot find name for new mmap)sizerangeosgetpidnext_randmmap_winapi GetLastErrorcloseFileExistsErrornamebuffer_state)selfr ir buf r +/usr/lib/python3.10/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 ZdZejdkrdgZng Zd
ddZdd Zd S ) r zJ
A shared memory area backed by a temporary file (POSIX).
linuxz/dev/shmr c C sx || _ || _|dkr1tjdt | |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 r0 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 r2 [ s
zArena._choose_dirN)r ) r' r( r) r* sysplatformr6 r! r2 r r r r r C s
c C s( | j dkr tdt| jt| j ffS )Nr zDArena is unpicklable because forking was enabled when it was created)r0
ValueError
rebuild_arenar r DupFd)ar r r reduce_arenad s
rC c C s t | | S r# )r detach)r dupfdr 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 )n alignmentmaskr r r _roundup s z
Heap._roundupc C sZ | t| j|tj}| j| jk r| jd9 _td| t|}| j
| |d|fS )N z"allocating a new mmap of length %dr ) r[ maxrM r PAGESIZE_DOUBLE_ARENA_SIZE_UNTILr infor rT append)r r lengtharenar r r
_new_arena s
zHeap._new_arenac C s |j }|| jk r
d S | j|}|rJ | j|df= | j||f= | j| | j| }||d|f |sA| j|= | j | d S d S rH )
r _DISCARD_FREE_SPACE_LARGER_THANrS poprP rQ rT removerO rN )r rc rb blocksseqr r r _discard_arena s
zHeap._discard_arenac C s| t | j|}|t| jkr| |S | j| }| j| }| }|s+| j|= | j|= |\}}}| j||f= | j||f= |S r# ) bisect bisect_leftrN lenrd rO rf rP rQ ) r r r rb ri blockrc startstopr r r _malloc s
z Heap._mallocc C s |\}}}z | j ||f }W n ty Y nw | |\}}z | j||f }W n ty1 Y nw | |\}}|||f}|| }z
| j| | W n tyb |g| j|< t| j| Y nw || j||f< || j ||f< d S r# ) rQ KeyError_absorbrP rO ra rk insortrN ) r rn rc ro rp
prev_block_
next_blockrb r r r _add_free_block s.
zHeap._add_free_blockc C s^ |\}}}| j ||f= | j||f= || }| j| }|| |s+| j|= | j| ||fS r# )rP rQ rO rg rN )r rn rc ro rp rb ri r r r rs s
z Heap._absorbc C s8 |\}}}| j | }|||f |s| | d S d S r# )rS rg rj )r rn rc ro rp rh r r r _remove_allocated_block s
z Heap._remove_allocated_blockc C s< z| j }W n
ty Y d S w | | | | qr# )rU rf
IndexErrorrx ry r rn r r r _free_pending_blocks s
zHeap._free_pending_blocksc C s t | jkrtdt | j| jds | j| d S z | j d7 _ |
| | | | W | j
d S | j
w )Nz$My pid ({0:n}) is not last pid {1:n}Fr )r r rI r? formatrL acquirerU ra rW r| rx ry releaser{ r r r free s
z Heap.freec C s |dk r t d|tj|krtd|t | jkr"| | j D | j
d7 _
| | t
|d| j}| |\}}}|| }||k rS| |||f | j| ||f |||fW d S 1 slw Y d S )Nr Size {0:n} out of rangeSize {0:n} too larger )r? r} r= maxsize
OverflowErrorr r rI r! rL rV r| r[ r]
_alignmentrq rx rS add)r r rc ro rp real_stopr r r malloc( s"
$z Heap.mallocN)r' r( r) r re r_ r r^ r! staticmethodr[ rd rj rq rx rs ry r| r r r r r r rF s s
rF c @ s" e Zd Ze Zdd Zdd ZdS )r c C s^ |dk r t d|tj|krtd|tj|}||f| _t j
| tjj |fd d S )Nr r r )args) r? r} r= r r r _heapr r r r4 r )r r rn r r r r! F s
zBufferWrapper.__init__c C s& | j \\}}}}t|j||| S r# )r
memoryviewr )r rc ro rp r r r r create_memoryviewO s z BufferWrapper.create_memoryviewN)r' r( r) rF r r! r r r r r r B s )rk collectionsr r r r= r+ rJ contextr r r __all__r> r objectr rC r@ registerrF r r r r r
s(
$! P