o Ge5O@sLdZddlZddlZddlZddlZddlmZm Z m Z m Z m Z m Z mZddlmZmZddlmZejdkZdZdZdZd Zd Zd Zed ed ededededdgdddiZeeeeeeeeeeed ZdZdZdZ ddddddZ!edede diZ"Gdd d Z#Gd!d"d"Z$Gd#d$d$eZ%d,d&d'Z&Gd(d)d)Z'Gd*d+d+Z(dS)-z1 Classes for read / write of matlab (TM) 4 files N) MatFileReader docfillermatdims read_dtypeconvert_dtypes arr_to_charsarr_dtype_number)squeeze_elementchars_to_strings)reducelittlef8f4i4i2u2u1header))moptr)mrowsr)ncolsr)imagfr)namlenrU1) rc32c24c16rc8rrrrS1<>z VAX D-floatz VAX G-floatCray)rrrrrdoublecharsparsec@seZdZdZdZddZdS) VarHeader4FcCs"||_||_||_||_||_dSN)namedtypemclassdims is_complex)selfr,r-r.r/r0r2^/var/www/html/SPIRAL.web.tool/spiral_venv/lib/python3.10/site-packages/scipy/io/matlab/mio4.py__init__Ps  zVarHeader4.__init__N)__name__ __module__ __qualname__ is_logical is_globalr4r2r2r2r3r*Ks r*c@sTeZdZdZddZddZdddZdd d Zd d Zd dZ ddZ ddZ dS) VarReader4z" Class to read matlab 4 variables cCs*||_|j|_|j|_|j|_|j|_dSr+) file_reader mat_streamdtypeschars_as_strings squeeze_me)r1r;r2r2r3r4`s  zVarReader4.__init__c Cst|j|jd}|jt|dd}|ddks"|ddkr&tdt|dd\}}|d vr=t d t |t t|d \}}|dkrLtd t|d \}}|}|d|df}|ddk} |j|} t || ||| S)z% Read and return header for variable rrrrz.Mat 4 mopt wrong format, byteswapping problem?)rrzBWe do not support byte ordering '%s'; returned data may be corruptdz,O in MOPT integer should be 0, wrong format? rrrr) rr<r=readintstrip ValueErrordivmodwarningswarn order_codes UserWarningr*) r1datar,MrestOPTr/r0r-r2r2r3 read_headergs4  zVarReader4.read_headerTcCst|j}|tkr ||}n"|tkr ||}|r|jrt|}n|tkr)||St d||r8|j r8t |S|S)NNo reader for class code %s) r. mxFULL_CLASSread_full_array mxCHAR_CLASSread_char_arrayr>r mxSPARSE_CLASSread_sparse_array TypeErrorr?r )r1hdrprocessr.arrr2r2r3array_from_headers      zVarReader4.array_from_headerc Csp|j}|j}|j}|D]}||9}q |jt|}t||kr'td|jt j |||dd}|r6| }|S)a Mat4 read using header `hdr` dtype and dims Parameters ---------- hdr : object object with attributes ``dtype``, ``dims``. dtype is assumed to be the correct endianness copy : bool, optional copies array before return if True (default True) (buffer is usually read only) Returns ------- arr : ndarray of dtype given by `hdr` ``dtype`` and shape given by `hdr` ``dims`` zNot enough bytes to read matrix '%s'; is this a badly-formed file? Consider listing matrices with `whosmat` and loading named matrices with `variable_names` kwarg to `loadmat`F)shaper-bufferorder) r-r/itemsizer<rErFlenrHr,npndarraycopy) r1r]ridtr/ num_bytesdrcr_r2r2r3read_sub_arrays$  zVarReader4.read_sub_arraycCs8|jr|j|dd}|j|dd}||dS||S)aM Full (rather than sparse) matrix getter Read matrix (array) can be real or complex Parameters ---------- hdr : ``VarHeader4`` instance Returns ------- arr : ndarray complex array if ``hdr.is_complex`` is True, otherwise a real numeric array F)riy?)r0rm)r1r]resres_jr2r2r3rWs   zVarReader4.read_full_arraycCsB||tj}|d}tj|jtdt |d 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-1rrbr-rc) rmastyperguint8tobytesdecoderhr/r-arrayri)r1r]r_Sr2r2r3rYs zVarReader4.read_char_arraycCs||}|ddddf}t|dt|df}tj|dddfdd}tj|dddfdd}|d8}|d8}|jdd krRtj|ddd fd d}ntj|ddd fd d}|ddd f|_tj|||ff|S) a Read and return sparse matrix type Parameters ---------- hdr : ``VarHeader4`` instance Returns ------- arr : ``scipy.sparse.coo_matrix`` with dtype ``float`` and shape read from the sparse matrix 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)rxr)rxrrintcr-rrrfloatcomplex) rmrFrgascontiguousarrayrbimagscipyr) coo_matrix)r1r]rntmpr/IJVr2r2r3r[s zVarReader4.read_sparse_arraycCs.|j}|tkrttt|j}ny|tkr&ttt|j}|jr%|dd}nc|tkr|j }|j}t |dkrB|ddkrB|ddksDdS|j |j |dddtjd||j |j d}|j |j |dddtjd||j |j d}t|t|f}ntd||jrtd d |D}|S) zwRead the shape of the array described by the header. The file position after this call is unspecified. Nrxrrrr2rqrUcSsg|]}|dkr|qS)rr2).0xr2r2r3 'sz0VarReader4.shape_from_header..)r.rVtuplemaprFr/rXr>rZr-rfr<seekrergrhrEr\r?)r1r]r.rbrjr/rowscolsr2r2r3shape_from_headers4 $   zVarReader4.shape_from_headerNT) r5r6r7__doc__r4rTr`rmrWrYr[rr2r2r2r3r:]s  $ *r:csXeZdZdZefddZddZddZdd Zdd d Z dddZ ddZ Z S)MatFile4Readerz Reader for Mat4 files cs$tj|g|Ri|d|_dS)zR Initialize matlab 4 file reader %(matstream_arg)s %(load_args)s N)superr4_matrix_reader)r1r<argskwargs __class__r2r3r4-s zMatFile4Reader.__init__cCs^|jdt|jtd}|jd|dkrdS|dks#|dkr)tr'dp(dStr-dp.dS)Nrrr$rAr%)r<rrrgr-SYS_LITTLE_ENDIAN)r1rr2r2r3guess_byte_order7s    zMatFile4Reader.guess_byte_ordercCstt|j|_t||_dS)za Run when beginning read of variables Sets up readers from parameters in `self` N)rmdtypes_template byte_orderr=r:r)r1r2r2r3initialize_readCszMatFile4Reader.initialize_readcCsV|j}tdd|jd}|jj|}|jr |jtks |d9}|j |}||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 cSs||Sr+r2)ryr2r2r3[sz0MatFile4Reader.read_var_header..rr) rrTr r/r-rer0r.rZr<tell)r1r]nremaining_bytes next_positionr2r2r3read_var_headerKs  zMatFile4Reader.read_var_headerTcCs|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`. )rr`)r1rr^r2r2r3read_var_arraybszMatFile4Reader.read_var_arrayNcCst|tr |g}n|durt|}|jd|i}|sf|\}}|jdur.dn|j d}|durC||vrC|j|q| |||<|j||durb| |t |dkrb |S|r!|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. NrNonelatin1) isinstancestrlistr<rr end_of_streamrr,rurremoverf)r1variable_namesmdictr]rr,r2r2r3 get_variablests,        zMatFile4Reader.get_variablescCs|jd|g}|sB|\}}|jdurdn|jd}|j|}t |j d}| |||f|j||r|S)z list variables from stream rNrrunknown) r<rrrrr,rurr mclass_infogetr.append)r1varsr]rr,rbinfor2r2r3list_variabless    zMatFile4Reader.list_variablesrr+) r5r6r7rrr4rrrrrr __classcell__r2r2rr3r+s    rrowcCs(t||}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 rz=Matlab 4 files cannot save arrays with more than 2 dimensions)rrfrHreshape)r_oned_asr/r2r2r3 arr_to_2ds   rc@sTeZdZddZddZddZeedfdd Zd d Z d d Z ddZ ddZ dS) VarWriter4cCs|j|_|j|_dSr+) file_streamr)r1 file_writerr2r2r3r4s zVarWriter4.__init__cCs|j|jdddS)Nra)rd)rwritert)r1r_r2r2r3 write_bytesszVarWriter4.write_bytescCs|j|dSr+)rr)r1sr2r2r3 write_stringszVarWriter4.write_stringrc Cstdtd}t }d}|d|d|d||d<|d|d<|d |d <||d <t|d |d <|||d } || 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 r2rrrBrCrDrrrrrrrN)rgemptyrrrfrrencode) r1r,rbrRrSrrrOrQrNr2r2r3 write_headers"   zVarWriter4.write_headercCstj|r|||dSt|}|j}|js!|| d}|j }|tj ur-t d|tj ur6t d|tjtjfvrF|||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_sparsergasarrayr-isnativerr newbyteordertypeobject_r\voidunicode_string_ write_char write_numeric)r1r_r,rjdttr2r2r3rs     zVarWriter4.writecCst||j}|jjdk}z t|jjdd}Wnty0|r'|d}n|d}t}Ynw|j ||j |t |d|rL| |j | |jdS| |dS)Ncrc128r)rRrSr)rrr-kind np_to_mtypesrKeyErrorrrmiDOUBLErrbrVrrealr~)r1r_r,rrRr2r2r3rs(      zVarWriter4.write_numericcCst|}t||j}|j}|j||ttd|jjdkr;t |}t j dt |||d}| d}t j |d|d}||dS)NrRrSUr2rqrpr#)rrrrbrmiUINT8rXr-rrgprodrhr itemrr)r1r_r,r/n_charsst_arrstr2r2r3rs$   zVarWriter4.write_charcCs|}|jjdk}tj|jdd|fdd}|j|dddf<|j|dddf<|dddd fd7<|rO|jj |ddd f<|jj |dddf<n |j|ddd f<|j |ddd f<|j ||j t td ||dS) zY Sparse matrices are 2-D See docstring for VarReader4.read_sparse_array rrrrrzNrxrrr)tocoor-rrgzerosnnzrcolrNrr~rbrrrZr)r1r_r,Arijvr2r2r3r.s$ zVarWriter4.write_sparseN) r5r6r7r4rrrrVrrrrrr2r2r2r3rs! rc@s$eZdZdZdddZdddZdS)MatFile4Writerz) Class for writing matlab 4 format files NcCs"||_|dur d}||_d|_dS)Nr)rr_matrix_writer)r1rrr2r2r3r4Is  zMatFile4Writer.__init__cCs.t||_|D] \}}|j||q 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)r1rrr,varr2r2r3 put_variablesPs zMatFile4Writer.put_variablesr+)r5r6r7rr4rr2r2r2r3rGs r)r))rsysrJnumpyrg scipy.sparsermiobaserrrrrrr mio_utilsr r functoolsr byteorderrrmiSINGLEmiINT32miINT16miUINT16rrrrVrXrZrLrr*r:rrrrr2r2r2r3sr$   O y