_MhRdZddlZddlZddlZddlmZddlZddlZ ddl m Z m Z m Z mZmZmZmZddlmZmZddlmZgdZejd kZdZdZd Zd Zd Zd ZededededededdgdddiZ eeeeeeeeeeed Z!dZ"dZ#d Z$ddddddZ%e"de#de$d iZ&ej'ej(j)Z*Gd!d"Z+Gd#d$Z,Gd%d&e Z-d-d(Z.Gd)d*Z/Gd+d,Z0dS).z1 Classes for read / write of matlab (TM) 4 files N)mul) MatFileReader docfillermatdims read_dtypeconvert_dtypes arr_to_charsarr_dtype_number)squeeze_elementchars_to_strings)reduce)MatFile4ReaderMatFile4WriterSYS_LITTLE_ENDIAN VarHeader4 VarReader4 VarWriter4 arr_to_2d mclass_infomdtypes_templatemiDOUBLEmiINT16miINT32miSINGLEmiUINT16miUINT8 mxCHAR_CLASS mxFULL_CLASSmxSPARSE_CLASS np_to_mtypes order_codeslittlef8f4i4i2u2u1header))moptr*)mrowsr*)ncolsr*)imagfr*)namlenr*U1) r(c32c24c16r)c8r*r+r,r-S1<>z VAX D-floatz VAX G-floatCray)rrr$r%r&doublecharsparseceZdZdZdZdZdS)rFcL||_||_||_||_||_dSN)namedtypemclassdims is_complex)selfrCrDrErFrGs U/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/scipy/io/matlab/_mio4.py__init__zVarHeader4.__init__^s+     $N)__name__ __module__ __qualname__ is_logical is_globalrJrKrIrrYs-JI % % % % %rKrcFeZdZdZdZdZd dZd dZdZdZ d Z d Z d S) rz" Class to read matlab 4 variables ct||_|j|_|j|_|j|_|j|_dSrB) file_reader mat_streamdtypeschars_as_strings squeeze_me)rHrTs rIrJzVarReader4.__init__ns7&%0!( + <%0rKct|j|jd}|jt |dd}|ddks |ddkrt dt|dd\}}|d vr+tj d t|d td t|d\}}|dkrt dt|d\}}|}|d|df}|ddk} |j|} t|| ||| S)z% Read and return header for variable r.r3r/rz.Mat 4 mopt wrong format, byteswapping problem?)rrz!We do not support byte ordering 'z'; returned data may be corruptr%) stackleveldz,O in MOPT integer should be 0, wrong format? r0r1r2r) rrUrVreadintstrip ValueErrordivmodwarningswarnr" UserWarningr) rHdatarCMrestOPTrFrGrDs rI read_headerzVarReader4.read_headerus^$/4;x+@AA##CX$7$788>>wGG  &%%c%66C''%'88E%"*% %""3'''rKcX||tj}|d}tj|jtjdtj | S)z latin-1 text matrix (char matrix) reader Parameters ---------- hdr : ``VarHeader4`` instance Returns ------- arr : ndarray with dtype 'U1', shape given by `hdr` ``dims`` latin-1r4r|rDr}) rastyperuint8tobytesrrrFrDarrayr)rHrurwSs rIrrzVarReader4.read_char_arrays~!!#&&--bh77 KKMM  + +z "!#!....2dff 5rKc>||}|ddddf}t|dt|df}tj|dddfd}tj|dddfd}|dz}|dz}|jdd kr!tj|ddd fd }n1tj|ddd fd }|ddd f|_t j|||ff|S) a Read and return sparse matrix type Parameters ---------- hdr : ``VarHeader4`` instance Returns ------- arr : coo_array with dtype ``float`` and shape read from the sparse array data Notes ----- MATLAB 4 real sparse arrays are saved in a N+1 by 3 array format, where N is the number of non-zero values. Column 1 values [0:N] are the (1-based) row indices of the each non-zero value, column 2 [0:N] are the column indices, column 3 [0:N] are the (real) values. The last values [-1,0:2] of the rows, column indices are shape[0] and shape[1] respectively of the output matrix. The last value for the values column is a padding 0. mrows and ncols values from the header give the shape of the stored matrix, here [N+1, 3]. Complex data are saved as a 4 column matrix, where the fourth column contains the imaginary component; the last value is again 0. Complex sparse data do *not* have the header ``imagf`` field set to True; the fact that the data are complex is only detectable because there are 4 storage columns. N)rr)rrrintcrDrr%r$floatcomplex) rrarascontiguousarrayr|imagscipyr? coo_array)rHrurtmprFIJVs rIrszVarReader4.read_sparse_arrays&6!!#&&#2#aaa%jCICI/  QQQqS 7 7 7  QQQqS 7 7 7 Q Q 9Q<1  $S1XG<<>>A1XAF|%%q!Ai666rKc|j}|tkr)ttt|j}n}|t kr:ttt|j}|jr |dd}n8|tkr|j }|j}t|dkr|ddkr |ddksdS|j |j |ddz zdtjd||j |j }|j |j |ddz zdtjd||j |j }t |t |f}nt#d||jrtd |D}|S) zwRead the shape of the array described by the header. The file position after this call is unspecified. Nrr$rrrQrrpcg|] }|dk| S)rrQ).0xs rI z0VarReader4.shape_from_header..8s666qAvv1vvvrK)rErtuplemaprarFrrWr rDrrUseekrrrr`rtrX)rHrurEr|rrFrowscolss rIshape_from_headerzVarReader4.shape_from_headers \ ! !#c38,,--EE | # ##c38,,--E$ #crc  ~ % %B8DIINNtAw!||Q1 r O Q! !s: 5d555f555"rKc |jdt|jtjd}|jd|dkrdS|dks|dkr t rdpdSt rdpdS)Nrr*r:r[r;)rUrrrrDr)rHr/s rIguess_byte_orderzMatFile4Reader.guess_byte_orderHs Q$/28D>>:: Q 1993 !88td{{$,3 3 (S/C/rKcltt|j|_t ||_dS)za Run when beginning read of variables Sets up readers from parameters in `self` N)r r byte_orderrVrr)rHs rIinitialize_readzMatFile4Reader.initialize_readTs. %%5tGG (..rKc|j}tt|jt j|jj}|j r|j tks|dz}|j |z}||fS)ab Read and return header, next position Parameters ---------- None Returns ------- header : object object that can be passed to self.read_var_array, and that has attributes ``name`` and ``is_global`` next_position : int position in stream of next variable r$)rrnrrrFrrrDrrGrEr rUtell)rHruremaining_bytes next_positions rIread_var_headerzMatFile4Reader.read_var_header\s}!--// ch9K0L0LMM > !#*">"> q O,,..@ M!!rKTc8|j||S)a Read array, given `header` Parameters ---------- header : header object object with fields defining variable header process : {True, False}, optional If True, apply recursive post-processing during loading of array. Returns ------- arr : array array with post-processing applied or not according to `process`. )rrx)rHr.rvs rIread_var_arrayzMatFile4Reader.read_var_arrayss "44VWEEErKNczt|tr|g}n|t|}|jd|i}|s|\}}|jdn|j d}|||vr|j|o| |||<|j||)| |t|dkrn||S)a, get variables from stream as dictionary Parameters ---------- variable_names : None or str or sequence of str, optional variable name, or sequence of variable names to get from Mat file / file stream. If None, then get all variables in file. NrNonerz) isinstancestrlistrUrr end_of_streamrrCrrremover)rHvariable_namesmdictrurrCs rI get_variableszMatFile4Reader.get_variablessK nc * * 2,-NN  '!.11N Q $$&& !%!5!5!7!7 C X-6638??83L3LD)d..H.H$$]333--c22E$K O  / / /)%%d+++~&&!++$$&&  rKc|jd|g}|s|\}}|jdn|jd}|j|}t |j d}| |||f|j|||S)z list variables from stream rNrrzunknown) rUrrrrrCrrrrgetrEappend)rHvarsrurrCr|infos rIlist_variableszMatFile4Reader.list_variabless Q $$&& 0!%!5!5!7!7 C X-6638??83L3LD'99#>>E??3:y99D KKud+ , , , O  / / /$$&& 0 rKrrB) rLrMrNrrrJrrrrrr __classcell__)rs@rIrr<s!!####Y# 0 0 0///""".FFFF$>rKrrowct||}t|dkrtd||S)a Make ``arr`` exactly two dimensional If `arr` has more than 2 dimensions, raise a ValueError Parameters ---------- arr : array oned_as : {'row', 'column'}, optional Whether to reshape 1-D vectors as row vectors or column vectors. See documentation for ``matdims`` for more detail Returns ------- arr2d : array 2-D version of the array r$z=Matlab 4 files cannot save arrays with more than 2 dimensions)rrrcreshape)rwoned_asrFs rIrrsK" 3 D 4yy1}}()) ) ;;t  rKcFeZdZdZdZdZeedfdZdZ dZ dZ d Z d S) rc6|j|_|j|_dSrB) file_streamr)rH file_writers rIrJzVarWriter4.__init__s&2"* rKcb|j|ddS)Nr{)r~)rwriter)rHrws rI write_byteszVarWriter4.write_bytess- s{{{5566666rKc:|j|dSrB)rr)rHss rI write_stringzVarWriter4.write_strings q!!!!!rKrcltjdtd}t }d}|dz|dzz|dzz|z|d<|d|d<|d |d <||d <t |d z|d <|||d z} || ddS)a) Write header for given data options Parameters ---------- name : str name of variable shape : sequence Shape of array as it will be read in matlab P : int, optional code for mat4 data type, one of ``miDOUBLE, miSINGLE, miINT32, miINT16, miUINT16, miUINT8`` T : int, optional code for mat4 matrix class, one of ``mxFULL_CLASS, mxCHAR_CLASS, mxSPARSE_CLASS`` imagf : int, optional flag indicating complex rQr.rr\r^r_r/r0rr1r2r3rzN)remptyrrrrrencode) rHrCr|rlrmr2r.rirkrhs rI write_headerzVarWriter4.write_headers$".x899! ! d(c'"b&!v (w(wwt99q=x    d{ $++h//00000rKctj|r|||dSt j|}|j}|js(|| d}|j }|tj urtd|tj urtd|tjtjfvr|||dS|||dS)z Write matrix `arr`, with name `name` Parameters ---------- arr : array_like array to write name : str name in matlab workspace N=z!Cannot save object arrays in Mat4zCannot save void type arrays)rr?issparse write_sparserasarrayrDisnativer newbyteordertypeobject_rtvoidstr_bytes_ write_char write_numeric)rHrwrCrdtts rIrzVarWriter4.writes <  % %    c4 ( ( ( Fjoo Y{ 3**R__S1122Cg "*  ?@@ @ BG^^:;; ; RWbi( ( ( OOC & & & F 3%%%%%rKct||j}|jjdk} t|jjdd}nD#t $r7|r|d}n|d}t}YnwxYw| ||j |t||r6| |j | |jdS| |dS)Ncrc128r()rlrmr2)rrrDkindr!rKeyErrorrrrr|rrrealr)rHrwrCr2rls rIrzVarWriter4.write_numericsT\** #% SY]122./AA    'jj((jj&&AAA   $)( %  ' ' '  "   SX & & &   SX & & & & &   S ! ! ! ! !sA>BBc\|jjtjkr6|jjtjdjkrt |}t ||j}|j}| ||tt|jj dkrwtj|}tjdt!|||}|d}tj|d|}||dS)Nr4rlrmUrQrrr9)rDrrrrr rrr|rrrrmathprodrr itemrr)rHrwrCrFn_charsst_arrsts rIrzVarWriter4.write_char,s 9>RW $ $);rx~~?V)V)Vs##CT\**y        9>S iooGZb&6sG&D&D'*,,,F%%i00B*4tB???C rKc|}|jjdk}tj|jdzd|zfd}|j|dddf<|j|dddf<|dddd fxxdz cc<|r'|jj |ddd f<|jj |dddf<n|j|ddd f<|j |ddd f<| ||j tt ||dS) zY Sparse matrices are 2-D See docstring for VarReader4.read_sparse_array rrr%r(rNrrr$r)tocoorDrrzerosnnzrcolrhrrr|rrr r)rHrwrCAr2ijvs rIrzVarWriter4.write_sparse@s+ IIKK #h 1U7+4888UCRCE UCRCE  CRC!G   CAJCAJJCAJgBqsF    I     rKN) rLrMrNrJrrrrrrrrrrQrKrIrrs+++777"""+3l!1111B&&&:""",(rKrc"eZdZdZddZddZdS)rz) Class for writing matlab 4 format files Nc8||_|d}||_d|_dS)Nr)rr_matrix_writer)rHrrs rIrJzMatFile4Writer.__init__[s)& ?G "rKct||_|D] \}}|j||!dS)a Write variables in `mdict` to stream Parameters ---------- mdict : mapping mapping with method ``items`` return name, contents pairs where ``name`` which will appeak in the matlab workspace in file load, and ``contents`` is something writeable to a matlab file, such as a NumPy array. write_header : {None, True, False} If True, then write the matlab file header before writing the variables. If None (the default) then write the file header if we are at position 0 in the stream. By setting False here, and setting the stream position to the end of the file, you can append variables to a matlab file N)rritemsr)rHrrrCvars rI put_variableszMatFile4Writer.put_variablesbsU().. 1 1ID#   % %c4 0 0 0 0 1 1rKrB)rLrMrNrrJrrQrKrIrrYsB33####111111rKr)r)1rsysreroperatorrnumpyr scipy.sparser_miobaserrrrr r r _mio_utilsr r functoolsr__all__ byteorderrrrrrrrrr!rrr r"riinfointpmaxrrrrrrrrQrKrIr s  FFFFFFFFFFFFFFFFFF:9999999   MX-       d d T T d T !!!  $                     (&H  BHRW   ! %%%%%%%%$NNNNNNNNbvvvvv]vvvr0IIIIIIIIX1111111111rK