File: //home/arjun/projects/env/lib64/python3.10/site-packages/billiard/__pycache__/heap.cpython-310.pyc
o
!we�# � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZddlm Z ddlm
Z
ddlm Z ddl m
Z
mZ dgZejd dkZejd krQG d
d � d �ZnG 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 ]!}dt�� t| j�f }tjd||d�}t�� dkr$ n|� � 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 � GetLastError�close�IOError�errno�EEXIST�name�buffer�_state)�selfr �ir �buf�exc� r" �F/home/arjun/projects/env/lib/python3.10/site-packages/billiard/heap.py�__init__' s
�Arena.__init__c C s t �| � | jS �N)r �assert_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 sD || _ || _|dkr�trptjdt�� f t�� d�\| _}t� |� t�
| tj | jf� t j
| jddd��2}d}||krOd| }t|| �D ]}|�|� qF~|�d|| � |�� |ks`J �W d � n1 sjw Y n'tjdt�� 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 � get_temp_dir�unlink�Finalizer �io�openr �write�tell� ftruncater r )r r r6 r �f�bs�zeros�_r" r" r# r$ G s: �
��
�
r% N)r )r, r- r. r$ r" r" r" r# r
E s c C s( | j dkr td��t| jt�| j �ffS )Nr zCArena is unpicklable becauseforking was enabled when it was created)r6 �
ValueError�
rebuild_arenar r �DupFd)�ar" r" r# � reduce_arenaf s
rI c C s t | |�� �S r& )r
�detach)r �dupfdr" r" r# rF l s rF 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�kr5| �t| j|�tj�}| jd9 _t �
d|� t |�}| j �
|� |d|fS | j| }| j| }|�� }|sM| j|= | j|= |\}}}| j||f= | j||f= |S )N� z"allocating a new mmap of length %dr )�bisect� bisect_leftrS �lenr_ �maxrR r �PAGESIZEr �infor
rY �appendrT �poprU rV ) 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 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& ) rV �KeyError�_absorbrU rT rg ra �insortrS ) r rl rj rm rn �
prev_blockrD �
next_blockri r" r" r# �_free� s.
� �
�z
Heap._freec C s^ |\}}}| j ||f= | j||f= || }| j| }|�|� |s+| j|= | j�|� ||fS r& )rU rV rT �removerS )r rl rj rm rn ri rk r" r" r# rq � s
z Heap._absorbc C s> z| j �� }W n
ty Y d S w | j�|� | �|� qr& )rZ rh �
IndexErrorrX rv ru �r rl r" r" r# �_free_pending_blocks� s �
�zHeap._free_pending_blocksc C sj t �� | jks J �| j�d�s| j�|� d S z| �� | j� |� | �
|� W | j� � d S | j� � w )NF) r r rN rQ �acquirerZ rg ry rX rv ru �releaserx r" r" r# �free� s z Heap.freec C s� d| kr t jk sJ � J �t�� | jkr| �� | j�; | �� | �t |d�| j
�}| � |�\}}}|| }||k rD| � |||f� |||f}| j
�|� |W d � S 1 s[w Y d S )Nr r )�sys�maxsizer r rN r$ rQ ry r_ rd �
_alignmentro ru rX �add)r r rj rm rn �new_stoprl r" r" r# �malloc� s
$�z Heap.mallocN)r, r- r. r r re r$ � staticmethodr_ ro ru rq ry r| r� r" r" r" r# rL v s
rL c @ s2 e Zd Ze� Zdd� Zdd� Zdd� Zdd� Zd S )
r c C sN d| kr t jk sJ � J �tj�|�}||f| _tj| tjj|fd� d S )Nr )�args) r} r~ r �_heapr� r r r; r| )r r rl r" r" r# r$ s
zBufferWrapper.__init__c C s4 | j \\}}}}t�|j�\}}||ksJ �|| S r& )r r �address_of_bufferr )r rj rm rn r �addressri r" r" r# � get_address s zBufferWrapper.get_addressc C s
| j d S r[ )r r( r" r" r# �get_size s
zBufferWrapper.get_sizec C s& | j \\}}}}t|j�||| � S r& )r �
memoryviewr )r rj rm rn r r" r" r# �create_memoryview s z BufferWrapper.create_memoryviewN) r, r- r. rL r� r$ r� r� r� r" r" r" r# r s )ra r r< r r r} rO r/ � r r r �_extr r �__all__� version_infor7 �platformr
rI rF �registerrL r r" r" r" r# �<module> s.
"!