File: //usr/local/lib/python3.9/site-packages/billiard/__pycache__/heap.cpython-39.pyc
a
X>h�# � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZddlm Z ddlm
Z
ddlm Z ddl m
Z
mZ dgZejd dkZejd kr�G d
d � d �Zn*G d d � d �Zd
d� Zdd� Ze
�ee� G dd� d�ZG dd� d�ZdS )� N� )�context)� reduction)�util)� _billiard�win32�
BufferWrapper� r c @ s, e Zd Ze�� Zdd� Zdd� Zdd� ZdS )�Arenac C s� || _ td�D ]B}dt�� t| j�f }tjd||d�}t�� dkrH qf|� � qt
d�}t j |_ |�|| _
|| _| j | j
f| _d S )N�d z pym-%d-%s�����tagnamer z Cannot find name for new mmap)�size�range�os�getpid�next�_rand�mmapr Z GetLastError�close�IOError�errno�EEXIST�name�buffer�_state)�selfr �ir �buf�exc� r! �7/usr/local/lib/python3.9/site-packages/billiard/heap.py�__init__' s
�Arena.__init__c C s t �| � | jS �N)r Zassert_spawningr �r r! r! r" � __getstate__8 s
zArena.__getstate__c C s, | \| _ | _| _tjd| j | jd�| _d S )Nr r
)r r r r r )r �stater! r! r" � __setstate__< s zArena.__setstate__N) �__name__�
__module__� __qualname__�tempfile�_RandomNameSequencer r# r' r) r! r! r! r" r
# s r
c @ s e Zd Zddd�ZdS )r
r c C sF || _ || _|dk�r0tr�tjdt�� f t�� d�\| _}t� |� t�
| tj | jf� t j
| jddd��d}d}||kr�d| }t|| �D ]}|�|� q�~|�d|| � |�� |ks�J �W d � n1 s�0 Y nNtjdt�� f t�� d�\| _}t� |� t�
| tj | jf� t�| j|� t�| j| j �| _d S ) Nr zpym-%d-)�prefix�dir�wbF)�closefdi � )r �fd�PY3r- �mkstempr r r Z get_temp_dir�unlink�Finalizer �io�openr �write�tell� ftruncater r )r r r4 r �f�bsZzeros�_r! r! r" r# G s6
�
0 �
r$ N)r )r* r+ r, r# r! r! r! r" r
E s c C s( | j dkrtd��t| jt�| j �ffS )Nr zCArena is unpicklable becauseforking was enabled when it was created)r4 �
ValueError�
rebuild_arenar r ZDupFd)�ar! r! r" � reduce_arenaf s
rD c C s t | |�� �S r% )r
�detach)r Zdupfdr! r! r" rB l s rB c @ sZ e Zd ZdZejfdd�Zedd� �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 r% )r r �_lastpid� threading�Lock�_lock�_size�_lengths� _len_to_seq�_start_to_block�_stop_to_block�set�_allocated_blocks�_arenas�_pending_free_blocks)r r r! r! r" r# z s
z
Heap.__init__c C s |d }| | | @ S �Nr r! )�n� alignment�maskr! r! r" �_roundup� s z
Heap._roundupc C s� t �| j|�}|t| j�krj| �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 )�bisect� bisect_leftrM �lenrY �maxrL r �PAGESIZEr �infor
rS �appendrN �poprO rP ) r r r �length�arena�seq�block�start�stopr! r! r" �_malloc� s"
z Heap._mallocc C s� |\}}}z| j ||f }W n ty. Y n0 | �|�\}}z| j||f }W n tyb Y n0 | �|�\}}|||f}|| }z| j| �|� W n, ty� |g| j|< t�| j|� Y n0 || j||f< || j ||f< d S r% ) rP �KeyError�_absorbrO rN ra r[ �insortrM ) r rf rd rg rh Z
prev_blockr@ Z
next_blockrc r! r! r" �_free� s(
z
Heap._freec C s^ |\}}}| j ||f= | j||f= || }| j| }|�|� |sV| j|= | j�|� ||fS r% )rO rP rN �removerM )r rf rd rg rh rc re r! r! r" rk � s
z Heap._absorbc C sB z| j �� }W n ty$ Y q>Y n0 | j�|� | �|� q d S r% )rT rb �
IndexErrorrR rn rm �r rf r! r! r" �_free_pending_blocks� s
zHeap._free_pending_blocksc C sj t �� | jksJ �| j�d�s,| j�|� n:z,| �� | j� |� | �
|� W | j� � n | j� � 0 d S )NF) r r rH rK �acquirerT ra rq rR rn rm �releaserp r! r! r" �free� s z Heap.freec C s� d| krt jk s n J �t�� | jkr2| �� | j�v | �� | �t |d�| j
�}| � |�\}}}|| }||k r�| � |||f� |||f}| j
�|� |W d � S 1 s�0 Y d S )Nr r )�sys�maxsizer r rH r# rK rq rY r^ �
_alignmentri rm rR �add)r r rd rg rh Znew_stoprf r! r! r" �malloc� s
z Heap.mallocN)r* r+ r, rw r r_ r# � staticmethodrY ri rm rk rq rt ry r! r! r! r" rF v s
rF c @ s2 e Zd Ze� Zdd� Zdd� Zdd� Zdd� Zd S )
r c C sL d| krt jk s n J �tj�|�}||f| _tj| tjj|fd� d S )Nr )�args) ru rv r �_heapry r r r8 rt )r r rf r! r! r" r# s
zBufferWrapper.__init__c C s4 | j \\}}}}t�|j�\}}||ks,J �|| S r% )r r Zaddress_of_bufferr )r rd rg rh r �addressrc r! r! r" � get_address s zBufferWrapper.get_addressc C s
| j d S rU )r r& r! r! r" �get_size s zBufferWrapper.get_sizec C s& | j \\}}}}t|j�||| � S r% )r �
memoryviewr )r rd rg rh r r! r! r" �create_memoryview s z BufferWrapper.create_memoryviewN) r* r+ r, rF r| r# r~ r r� r! r! r! r" r s
)r[ r r9 r r ru rI r- � r r r Z_extr r �__all__� version_infor5 �platformr
rD rB �registerrF r r! r! r! r" �<module>
s,
"!