Current File : //snap/core/17212/usr/lib/python3.5/multiprocessing/__pycache__/heap.cpython-35.pyc
|�g� � @ s# d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z d d l m Z d d l m Z d g Z
e j d k r� d d l Z Gd d � d e
� Z n> Gd
d � d e
� Z d d � Z d
d � Z e j e e � Gd d � d e
� Z Gd d � d e
� Z d S)� N� )�context)� reduction)�util�
BufferWrapperZwin32c @ s@ e Z d Z e j � Z d d � Z d d � Z d d � Z d S)�Arenac C s� | | _ xt t d � D]Z } d t j � t | j � f } t j d | d | �} t j � d k rf P| j � q Wt
d � � | | _ | | _ | j | j f | _
d S)N�d z pym-%d-%sr �tagnamer z Cannot find name for new mmap���)�size�range�os�getpid�next�_rand�mmap�_winapiZ GetLastError�close�FileExistsError�name�buffer�_state)�selfr �ir Zbuf� r �*/usr/lib/python3.5/multiprocessing/heap.py�__init__# s zArena.__init__c C s t j | � | j S)N)r Zassert_spawningr )r r r r � __getstate__2 s
zArena.__getstate__c C s> | \ | _ | _ | _ t j d | j d | j �| _ d S)Nr r r
)r r r r r )r �stater r r � __setstate__6 s zArena.__setstate__N) �__name__�
__module__� __qualname__�tempfileZ_RandomNameSequencer r r r r r r r r s r c @ s e Z d Z d d d � Z d S)r r c C s- | | _ | | _ | d
k rt j d d t j � d t j � � \ | _ } t j | � t j | t j
| j f � t | j d d d ��{ } d } | | k r� d | } x% t | | � D] } | j
| � q� W~ | j
d | | � | j � | k st � Wd QRXt j | j | j � | _ d S) Nr �prefixzpym-%d-�dir�wb�closefdFi s r
i )r �fdr# Zmkstempr
r r Z get_temp_dir�unlink�Finalizer �openr �write�tell�AssertionErrorr r )r r r( r �fZbsZzeros�_r r r r A s" (
zArena.__init__Nr
)r r! r" r r r r r r ? s c C s: | j d k r t d � � t | j t j | j � f f S)Nr zDArena is unpicklable because forking was enabled when it was createdr
)r( �
ValueError�
rebuild_arenar r ZDupFd)�ar r r � reduce_arenaT s r4 c C s t | | j � � S)N)r �detach)r Zdupfdr r r r2 Z s r2 c @ s� e Z d Z d Z e j d d � Z e d d � � Z d d � Z d d � Z
d
d � Z d d
� Z d d � Z
d d � Z d S)�Heap� c C sm t j � | _ t j � | _ | | _ 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_block�set�_allocated_blocks�_arenas�_pending_free_blocks)r r r r r r g s z
Heap.__init__c C s | d } | | | @S)Nr r )�nZ alignment�maskr r r �_roundupt s
z
Heap._roundupc C s t j | j | � } | t | j � k r� | j t | j | � t j � } | j d 9_ t j
d | � t | � } | j j
| � | d | f S| j | } | j | } | j � } | s� | j | =| j | =| \ } } } | j | | f =| j | | f =| S)N� z"allocating a new mmap of length %dr )�bisectZ bisect_leftr< �lenrF �maxr; r �PAGESIZEr �infor rB �appendr= �popr> r? ) r r r �length�arena�seq�block�start�stopr r r �_mallocz s" !
z Heap._mallocc C s$ | \ } } } y | j | | f } Wn t k
r: Yn X| j | � \ } } y | j | | f } Wn t k
r{ Yn X| j | � \ } } | | | f } | | } y | j | j | � Wn5 t k
r� | g | j | <t j | j | � Yn X| | j | | f <| | j | | f <d S)N) r? �KeyError�_absorbr> r= rM rH Zinsortr< ) r rR rP rS rT Z
prev_blockr0 Z
next_blockrO r r r �_free� s(
z
Heap._freec C s} | \ } } } | j | | f =| j | | f =| | } | j | } | j | � | ss | j | =| j j | � | | f S)N)r> r? r= �remover< )r rR rP rS rT rO rQ r r r rW � s
z Heap._absorbc C sQ xJ y | j j � } Wn t k
r+ PYn X| j j | � | j | � q Wd S)N)rC rN �
IndexErrorrA rY rX )r rR r r r �_free_pending_blocks� s
zHeap._free_pending_blocksc
C s� t j � | j k s t � | j j d � s@ | j j | � n< z+ | j � | j j
| � | j | � Wd | j j � Xd S)NF)
r
r r8 r. r: �acquirerC rM r[ rA rY rX �release)r rR r r r �free� s
z Heap.freec C s� d | k o t j k n s% t � t j � | j k rD | j � | j �� | j � | j t
| d � | j � } | j | � \ } } } | | } | | k r� | j
| | | f � | | | f } | j j | � | SWd QRXd S)Nr r )�sys�maxsizer. r
r r8 r r: r[ rF rJ �
_alignmentrU rX rA �add)r r rP rS rT Znew_stoprR r r r �malloc� s %
z Heap.mallocN)r r! r" ra r rK r � staticmethodrF rU rX rW r[ r^ rc r r r r r6 c s
r6 c @ s1 e Z d Z e � Z d d � Z d d � Z d S)r c C si d | k o t j k n s% t � t j j | � } | | f | _ t j | t j j d | f �d S)Nr �args)
r_ r` r. r �_heaprc r r r* r^ )r r rR r r r r � s %zBufferWrapper.__init__c C s3 | j \ \ } } } } t | j � | | | � S)N)r �
memoryviewr )r rP rS rT r r r r �create_memoryview� s z BufferWrapper.create_memoryviewN)r r! r" r6 rf r rh r r r r r � s )rH r r
r_ r# r9 � r r r �__all__�platformr �objectr r4 r2 �registerr6 r r r r r �<module>
s$ �
Mini Shell