3
2a @ s d dl Z d dlZd dlZd dlZd dlZd dlZddlmZmZ ddl m
Z
dgZ ej dkrvd dl
Z
G dd deZn,G d d deZd
d Zd d
Zejee G dd deZG dd deZdS ) N ) reductionassert_spawning)util
BufferWrapperwin32c @ s, e Zd Zej Zdd Zdd Zdd ZdS )Arenac C sz || _ xTtdD ]@}dtj t| jf }tjd||d}tj dkrHP |j qW t
d|| _ || _ | j | j f| _
d S )Nd z pym-%d-%sr )tagnamer z Cannot find name for new mmap)sizerangeosgetpidnext_randmmap_winapiZ GetLastErrorcloseFileExistsErrornamebuffer_state)selfr ir buf r )/opt/alt/python36/lib64/python3.6/heap.py__init__" s zArena.__init__c C s t | | jS )N)r r )r r r r __getstate__1 s zArena.__getstate__c C s, | \| _ | _| _tjd| j | jd| _d S )Nr )r
r )r r r r r )r stater r r __setstate__5 s zArena.__setstate__N) __name__
__module__ __qualname__tempfileZ_RandomNameSequencer r r r! r r r r r s r c @ s e Zd ZdddZdS )r r c C s || _ || _|d krtjdtj tj d\| _}tj| tj | tj
| jf t | jdddN}d
}||krd| }x t || D ]}|j
| qW ~|j
d|| W d Q R X tj| j| j | _d S ) Nr zpym-%d-)prefixdirwbF)closefdi r i )r fdr% Zmkstempr r r Z get_temp_dirunlinkFinalizer openr
writer r )r r r+ r fbsZzeros_r r r r @ s"
zArena.__init__Nr )r )r" r# r$ r r r r r r > s c C s( | j dkrtdt| jtj| j ffS )Nr zDArena is unpicklable because forking was enabled when it was createdr )r+
ValueError
rebuild_arenar r ZDupFd)ar r r reduce_arenaS s
r6 c C s t | |j S )N)r detach)r Zdupfdr r r r4 Y s r4 c @ sZ e Zd ZdZejfddZedd Zdd Z dd Z
d
d Z d d
Z dd Z
dd ZdS )Heap c C sJ t j | _tj | _|| _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_blockset_allocated_blocks_arenas_pending_free_blocks)r r r r r r f s
z
Heap.__init__c C s |d }| | | @ S )Nr r )nZ alignmentmaskr r r _roundups s z
Heap._roundupc C s t j| j|}|t| jkrj| jt| j|tj}| jd9 _t j
d| t |}| j j
| |d|fS | j| }| j| }|j }|s| j|= | j|= |\}}}| j||f= | j||f= |S )N z"allocating a new mmap of length %dr )bisectZ bisect_leftr> lenrH maxr= r PAGESIZEr infor rD appendr? popr@ rA ) r r r lengtharenaseqblockstartstopr r r _mallocy s"
z Heap._mallocc C s |\}}}y| j ||f }W n tk
r0 Y nX | j|\}}y| j||f }W n tk
rf Y nX | j|\}}|||f}|| }y| j| j| W n. tk
r |g| j|< tj| j| Y nX || j||f< || j ||f< d S )N) rA KeyError_absorbr@ r? rO rJ Zinsortr> ) r rT rR rU rV Z
prev_blockr2 Z
next_blockrQ r r r _free s(
z
Heap._freec C s^ |\}}}| j ||f= | j||f= || }| j| }|j| |sV| j|= | jj| ||fS )N)r@ rA r? remover> )r rT rR rU rV rQ rS r r r rY s
z Heap._absorbc C sF x@y| j j }W n tk
r& P Y nX | jj| | j| qW d S )N)rE rP
IndexErrorrC r[ rZ )r rT r r r _free_pending_blocks s zHeap._free_pending_blocksc
C sN | j jds| jj| n0z"| j | jj| | j| W d | j j X d S )NF) r< acquirerE rO r] rC r[ rZ release)r rT r r r free s z Heap.freec C s t j | jkr| j | jh | j | jt|d| j}| j |\}}}|| }||k rj| j
|||f |||f}| j j | |S Q R X d S )Nr )
r r r: r r< r] rH rL
_alignmentrW rZ rC add)r r rR rU rV Znew_stoprT r r r malloc s
z Heap.mallocN)r" r# r$ ra r rM r staticmethodrH rW rZ rY r] r` rc r r r r r8 b s
r8 c @ s" e Zd Ze Zdd Zdd ZdS )r c C s0 t jj|}||f| _tj| t jj|fd d S )N)args)r _heaprc r r r- r` )r r rT r r r r s
zBufferWrapper.__init__c C s& | j \\}}}}t|j||| S )N)r
memoryviewr )r rR rU rV r r r r create_memoryview s z BufferWrapper.create_memoryviewN)r" r# r$ r8 rf r rh r r r r r s )rJ r r sysr% r; contextr r r __all__platformr objectr r6 r4 registerr8 r r r r r
s$