ilvdZddlmZmZmZmZddlZddlZddlZddl Z ddl m Z ddl mZGddeZdS)z mrcmemmap --------- Module which exports the :class:`MrcMemmap` class. Classes: :class:`MrcMemmap`: An MrcFile subclass that uses a memory-mapped data array. )absolute_importdivisionprint_functionunicode_literalsN)MrcFilecFeZdZdZdZfdZdZdZdZdZ dZ xZ S) MrcMemmapaMrcFile subclass that uses a :class:`numpy memmap array ` for the data. Using a memmap means that the disk access is done lazily: the data array will only be read or written in small chunks when required. To access the contents of the array, use the array slice operator. Usage is the same as for :class:`~mrcfile.mrcfile.MrcFile`. Note that memmap arrays use a fairly small chunk size and so performance could be poor on file systems that are optimised for infrequent large I/O operations. If required, it is possible to create a very large empty file by creating a new MrcMemmap and then calling :meth:`_open_memmap` to create the memmap array, which can then be filled slice-by-slice. Be aware that the contents of a new, empty memmap array depend on your platform: the data values could be garbage or zeros. cLd|jj|jS)NzMrcMemmap('{0}', mode='{1}'))format _iostreamname_modeselfs [/srv/www/vhosts/g4struct/public_html/venv/lib/python3.11/site-packages/mrcfile/mrcmemmap.py__repr__zMrcMemmap.__repr__4s)-44T^5H59ZAA Ac>|jj}tt|||j|kr|j}|||_|j|j_ |jj|jz}||jz}|jdkr|tj zdkr|dz }|j |||j|jt#j|j|dSdS)a[Replace the file's extended header. Note that the file's entire data block must be moved if the extended header size changes. Setting a new extended header can therefore be very time consuming with large files, if the new extended header occupies a different number of bytes than the previous one. rrN)_extended_headernbytessuperr set_extended_header_datacopy _close_dataheadernsymbtmmapALLOCATIONGRANULARITYr truncate _open_memmapdtypeshapenpcopyto)rextended_headerold_ext_header_size data_copy header_nbytes total_nbytes __class__s rrzMrcMemmap.set_extended_header8s#3: i22?CCC  !%8 8 8 ))I      $3D !!0!7DK  K.1GGM(9+;;L1$$8R)RVW)W)W! N # #L 1 1 1   ioy ? ? ? Idj) , , , , , 9 8rc|js|jd|j|j|j|j|j|j|j|j|jjtj dSdS)z4Flush the header and data arrays to the file buffer.rN) _read_onlyr seekwriterr'flushrrosSEEK_CURrs rr1zMrcMemmap.flushSs @ N   " " " N  - - - N !5 6 6 6 N " " " J      N " " " N   12; ? ? ? ? ? @ @rcB tj|j}nT#t$rG}|jr:t jd|td|_ Yd}~dSd}~wwxYwtj |j}| ||dS)zRead the data block from the file. This method first calculates the parameters needed to read the data (block start position, endian-ness, file mode, array shape) and then opens the data as a numpy memmap array. z{0} - data block not readN) utilsdata_dtype_from_headerr ValueError _permissivewarningswarnr RuntimeWarningrdata_shape_from_headerr")rr#errr$s r _read_datazMrcMemmap._read_datacs 0==EE     9@@EE,...!  ,T[99 %'''''s A-;A('A((A-cn|jrdnd}|jjt|jjz}|j tj|j|||||_ nB#t$r5}|j r"tj dtd|_ n|Yd}~nd}~wwxYw|jz|}||z }|jj}|dkr|t"jzdkrd}||kr6d||z } tj | tdSdSdS) z:Open a new memmap array pointing at the file's data block.rzr+)r#modeoffsetr$zError opening memmapNrrz*MRC file is {0} bytes larger than expected)r.rrintrr r1r%memmapr Exceptionr8r9r:r;data_get_file_sizerr r ) rr#r$acc_moder*ex file_sizeremaining_file_size data_sizemsgs rr"zMrcMemmap._open_memmapyso/333t *S1C-D-DD   4>).(0*7). 000DJJ      4nEEE!    9 ++--I"+m";  (IA~~-$2L"LPQ"Q"Q ...C2Y>?? c>22222 ! /.s #A00 B/:+B**B/cz|j3|jd|jj_d|_dSdS)aDelete the existing memmap array, if it exists. The array is flagged as read-only before deletion, so if a reference to it has been kept elsewhere, changes to it should no longer be able to change the file contents. NF)rr1flags writeablers rrzMrcMemmap._close_datas@ : ! J     ).DJ  &DJJJ " !rc|jjt|jjz|jz}|j|||j|jtj |j |ddS)zbOverride of :meth:`_set_new_data` to handle opening a new memmap and copying data into it.no)castingN) rrrCrr r!r"r#r$r%r&r)rrFrJs r _set_new_datazMrcMemmap._set_new_datasv K&T[-?)@)@@4;N   *** $*dj111 $*dD111111r) __name__ __module__ __qualname____doc__rrr1r>r"rrT __classcell__)r,s@rr r s*AAA-----6@@@ (((,$3$3$3L    2 2 2 2 2 2 2rr )rX __future__rrrrrr2r9numpyr% mrcfile.utilsr5mrcfilerr rrr_s  ************ W2W2W2W2W2W2W2W2W2W2r