&^ d dgZ ddlmZ ddlZddlZddlZddlZddlZddlZej dk rddl
Z
dZ nddl Z dZ dd l
mZ ej ej z Zd
Ze rd Znd Zd
Z G d d ZdZ G d d ZdS ) SharedMemory
ShareableList )partialNntFT )resource_tracker z/psm_wnsm_c ~ t t t z
dz } t t j | z }|S )N )_SHM_SAFE_NAME_LENGTHlen_SHM_NAME_PREFIXsecrets token_hex)nbytesnames D/opt/alt/python311/lib64/python3.11/multiprocessing/shared_memory.py_make_filenamer ( s7 <#c*:&;&;;
AF g / 7 7 7D K c e Zd Z dZdZdZdZej Z dZ
e rdndZ ddZ
d Zd Zd
Zed Zed Zed
Zd Zd ZdS )r Ni TFr c x |dk st d |r.t t j z | _ |dk rt d |#| j t j z st d t r*|O t } t j || j | j
| _ n# t $ r Y Cw xY w|| _
n;| j rd|z n|}t j || j | j
| _ || _
|r |rt j | j | t j | j }|j }t% j | j | | _ n## t( $ r | w xY wt- j | j
d n|r# |t n|}t1 j t0 j t0 j t0 j |d z d
z |d
z | } t1 j }|t0 j k rZ|Bt t> j t j! t> j |t0 j t1 j" | t% j d || | _ t1 j" | n# t1 j" | w xY w|| _
n|| _
t1 j# t0 j$ d
| } t1 j% |t0 j$ ddd }t1 j" | n# t1 j" | w xY w t1 j& | }t1 j' | n# t1 j' | w xY wt% j d || | _ || _( tS | j | _* d S )Nr z!'size' must be a positive integerz4'size' must be a positive number different from zeroz&'name' can only be None if create=TrueT)mode/
shared_memory l r )tagnameF)+
ValueError_O_CREXosO_RDWR_flagsO_EXCL
_USE_POSIXr _posixshmemshm_open_mode_fdFileExistsError_name_prepend_leading_slash ftruncatefstatst_sizemmap_mmapOSErrorunlinkr register_winapiCreateFileMappingINVALID_HANDLE_VALUENULLPAGE_READWRITE GetLastErrorERROR_ALREADY_EXISTSerrnoEEXISTstrerror CloseHandleOpenFileMapping
FILE_MAP_READ
MapViewOfFileVirtualQuerySizeUnmapViewOfFile_size
memoryview_buf) selfr createsizestats temp_nameh_maplast_error_codep_bufs r __init__zSharedMemory.__init__K s qyy @AA A Y!BI-DKqyy !WXXX < bi 7< EFF F _ ? | )++D!#.#7 K!%$ $ $
+ ! ! ! !!%DJ%)%@JsTzzd&/K
"
1d 1L4000** } !Ytx 6 6
%dj/ B B B B 6
? 48L 0 0 0dI $ 5 4 . z1 z)!
E3*1*>*@*@*g.JJJ # /&5$)L$&K $=$=$($+$@ '" '" !" !) +E2222 &*Yr4%K%K%K
+E2222+E2222!*DJ "
/)
/ # 1 - E '....G'....3"3E::D+E2222G+E2222 !Yr4 > > >
tz** sJ &B, ,
B98B9 A E E>4A'J" 1 J" "J8+"L" "L8<M% %M;c R | d S # t $ r Y d S w xY wN)closer2 rH s r __del__zSharedMemory.__del__ s: JJLLLLL DD s
&&c . | j | j d| j ffS )NF) __class__r rJ rT s r
__reduce__zSharedMemory.__reduce__ s# N
r c @ | j j d| j d| j dS )N(z, size=))rW __name__r rJ rT s r __repr__zSharedMemory.__repr__ s* .)LLDILL LLLLr c | j S rR )rG rT s r bufzSharedMemory.buf s
>yr c | j }t r0| j r)| j d r| j dd }|S )Nr r )r+ r% r,
startswith)rH
reported_names r r zSharedMemory.name sK >
/$5 /z$$S))
/ $
122
r c | j S rR )rE rT s r rJ zSharedMemory.size s
zr c | j | j d | _ | j | j d | _ t r-| j dk r$t
j | j d| _ d S d S d S )Nr r )rG releaser1 rS r% r) r! rT s r rS zSharedMemory.close s 0 9 I DI : ! J DJ $(a-- HTX DHHH --r c t r<| j r7t j | j t j | j d d S d S d S )Nr )r% r+ r&
shm_unlinkr
unregisterrT s r r3 zSharedMemory.unlink sb '
E$* E "4: . . . '
O D D D D D E E E Er )NFr )r\
__module__ __qualname__r+ r) r1 rG r! r" r# r( r% r, rP rU rX r] propertyr_ r rJ rS r3 r r r r 2 s
B
E
C E D
YF E%/ :TTUl+ l+ l+ l+\
M M M X X X E E E E Er utf8c j e Zd Z edededededdj diZ dZ
d d d
d d Z e d
Z
d!dddZd Zd Zd Zd Zd Zd Zd Zd Zed Zed Zed Zed Zed Z ed Z ed Z d Z d Z e!e"j# Z$dS )"r qdzxxxxxxx?z%dsNzxxxxxx?x c | S rR rl values r zShareableList. s r c \ | d t S N )rstripdecode _encodingrs s r ru zShareableList. s g..55i@@ r c , | d S rw )ry rs s r ru zShareableList. s g.. r c d S rR rl )_values r ru zShareableList. s $ r )r r r c t | t t d j f sdS t | t rdS t | t rdS dS )Nr r r r )
isinstancestrbytesrW rs s r _extract_recreation_codez&ShareableList._extract_recreation_code s\ 2 %#udn!=>> 1
s
#
# 1
u
%
% 11r r c L |||pd} fd|D }t | _ d}dg _ |D ]I}||d dk r j nt |d d z
} j | J fd|D }t
j d j z d | z j
z j z }t |d |
_
nt | _
|t t
j d j z j
j d j g j R t
j d | j
j j gfd |D R t
j j
j
j j gfd |D R t
j j j
j j g|R d S t _ t) t
j j j
j d
_ d S )Nrl c g | ]v}t |t t f sj t | n=j t | j t
| j z d z z fz wS )r )r r r _types_mappingtype
_alignmentr .0itemrH s r
z*ShareableList.__init__..- s &dS%L99#DJJ// ,T$ZZ8 3t99+G!+KL< r r r sc : g | ]} | S rl )r r s r r z*ShareableList.__init__..? s4 ! ! !8<--d33! ! !r ro T)rI rJ c 3 p K | ]0}t |t r| n|V 1d S rR )r r encoder v_encs r z)ShareableList.__init__..Z s@ PP!Jq#$6$6=!((4...APPPPPPr c 3 B K | ]}| V d S rR )r r s r r z)ShareableList.__init__..` s- 33Q!((4..333333r rq )r _list_len_allocated_offsetsr intappendstructcalcsize_format_size_metainfojoin_format_packing_metainfo _format_back_transform_codesr shmr{ pack_intor_ _offset_data_start_offset_packing_formats _offset_back_transform_codeslist unpack_from) rH sequencer _formatsoffsetfmt_recreation_codesrequested_sizer s ` @r rP zShareableList.__init__* s <8/ ~2H %
H !]]DNF ()cD #
7
7SW^^$//S"X N'..v6666! ! ! !@H! ! ! $_d00!!"-. 12 N $DNKKKDHH#D))DH D d00
)
!! '
QPPPxPPP
- ,
4333(333
1 1
$
!YYDN&*".HL ' 'D # # #r c 6 |dk r|n || j z }|| j k s | j dk rt d t j d| j j | j |dz z d }| d }| t }|S )Nr Requested position out of range.8srq rx )
r
IndexErrorr r r r_ r ry rz r{ )rH positionr r
fmt_as_strs r _get_packing_formatz!ShareableList._get_packing_formats s H '1}}88(T^2K & &DNQ,>,> ?@@ @ HL (8a< 7
hhw ZZ **
r c || j k s | j dk rt d t j d| j j | j |z d }| j | }|S )Nr r b)r r r r r r_ r _back_transforms_mapping)rH r transform_codetransform_functions r _get_back_transformz!ShareableList._get_back_transform st C & &DNQ,>,> ?@@ @ + HL - 8
" :> J!!r c V || j k s | j dk rt d t j d| j j | j |dz z | t | | }t j d| j j | j
|z | d S )Nr r r rq r ) r r r r r r_ r r r{ r r )rH r r rt r s r !_set_packing_format_and_transformz/ShareableList._set_packing_format_and_transform s ?
& &DNQ,>,> ?@@ @ HL (8a< 7 i ( (
66u== HL - 8
r c < |dk r|n || j z } | j | j | z }t j | | | j j | \ }n # t $ r t d w xY w| | } || }|S )Nr zindex out of range)
r r r r r r r r_ r r )rH r r r back_transforms r __getitem__zShareableList.__getitem__ s '1}}88(T^2K 3,t/Fx/PPF%((22 DQQ
3 3 3 122 2 3 11(;; N1 s A
A A9c |dk r|n || j z } | j | }| j |z }| | }n # t $ r t d w xY wt |t t f s | j t | }|}n| j |dz |z
}t |t r|
t n|}t | |k rt d |d dk r|}n| j t |fz }| ||| t j || j j || d S )Nr z assignment index out of ranger z(bytes/str item exceeds available storager r )r r r r r r r r r r r r{ r r r r r r r_ ) rH r rt item_offsetr current_format
new_format
encoded_valueallocated_lengths r __setitem__zShareableList.__setitem__ s '1}}88(T^2K > 1(;K,{:F ! 5 5h ? ?NN > > > <== = > %#u .. ,T%[[9J !MM # 6x!| D{ R !+5# 6 6 BU\\) 4 4 4>r c N t j d| j j d d S )Nro r )r r r r_ rT s r __len__zShareableList.__len__ s !#tx|Q77::r c Z | j j dt | d| j j dS )NrZ z, name=r[ )rW r\ r r r rT s r r] zShareableList.__repr__ s0 .)QQDJJQQtx}QQQQr c l d fdt j D S )Nr c 3 B K | ]} | V d S rR )r )r irH s r r z'ShareableList.format.. sB
,-D $ $Q ' '
r )r ranger rT s `r formatzShareableList.format sL Gww
16t~1F1F
r c d| j dz z S )Nro r r rT s r r z#ShareableList._format_size_metainfo s Hdnq())r c d| j z S )Nr r rT s r r z&ShareableList._format_packing_metainfo s Hdn$$r c d| j z S )Nr r rT s r r z*ShareableList._format_back_transform_codes s HT^##r c | j dz dz S )Nr rq r rT s r r z ShareableList._offset_data_start s "a''r c , | j | j d z S )Nr )r r rT s r r z%ShareableList._offset_packing_formats s &)@)DDDr c & | j | j dz z S )Nrq )r r rT s r r z*ShareableList._offset_back_transform_codes s +dnq.@@@r c < t fd| D S )Nc 3 $ K | ]
}|k V d S rR rl )r entryrt s r r z&ShareableList.count..
s' 44e5E>444444r )sum)rH rt s `r countzShareableList.count s) M4444t444444r c f t | D ]\ }}||k r|c S t |d )Nz not in this container) enumerater )rH rt r r s r indexzShareableList.index sT : ) A AOHe~~ ? ? ?@@ @r rR )%r\ ri rj r floatboolr r rW r r r staticmethodr rP r r r r r rX r r] rk r r r r r r r r r classmethodtypes GenericAlias__class_getitem__rl r r r r s$ *& S
s j U
u
N J @ @ . . \ Gd G G G G GR
"
"
"
, " J J JB? ? ?; ; ;R R R
X
* * X* % % X% $ $ X$ ( ( X(
E E XE A A XA5 5 5
A A A $ E$677r )__all__ functoolsr r0 r! r< r r r r r5 r% r& r r O_CREATr$ r r
r r r r{ r rl r r r sX
O
-
7d??NNNJJJ
*ry
BE BE BE BE BE BE BE BEJ
]8 ]8 ]8 ]8 ]8 ]8 ]8 ]8 ]8 ]8r