a BCCfP@sTdZddlZddlZddlZddlZddlmZm Z m Z m Z m Z m Z mZddlmZmZddlmZgdZejdkZdZdZd Zd Zd Zd Zed edededededdgdddiZeeeeeeeeeeed ZdZdZ d Z!ddddddZ"ede de!diZ#Gd d!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)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&Zc32Zc24Zc16r'Zc8r(r)r*r+S1<>z VAX D-floatz VAX G-floatZCray)rrr"r#r$doublecharsparsec@seZdZdZdZddZdS)rFcCs"||_||_||_||_||_dSN)namedtypemclassdims is_complex)selfr:r;r<r=r>r@Q/var/www/html/django/DPS/env/lib/python3.9/site-packages/scipy/io/matlab/_mio4.py__init__Ys zVarHeader4.__init__N)__name__ __module__ __qualname__Z is_logical is_globalrBr@r@r@rArTsrc@sTeZdZdZddZddZdddZdd d Zd d Zd dZ ddZ ddZ dS)rz" Class to read matlab 4 variables cCs*||_|j|_|j|_|j|_|j|_dSr9) file_reader mat_streamdtypeschars_as_strings squeeze_me)r?rGr@r@rArBis zVarReader4.__init__c Cst|j|jd}|jt|dd}|ddksD|ddkrLtdt|dd\}}|d vr~tj d t |t d d t|d \}}|dkrtdt|d\}}|}|d|df}|ddk} |j|} t || ||| S)z% Read and return header for variable r,r1r-rz.Mat 4 mopt wrong format, byteswapping problem?)rrzBWe do not support byte ordering '%s'; returned data may be corruptr#) stackleveldz,O in MOPT integer should be 0, wrong format? r.r/r0r) rrHrIreadintstrip ValueErrordivmodwarningswarnr UserWarningr) r?datar:MrestOPTr=r>r;r@r@rA read_headerps4  zVarReader4.read_headerTcCst|j}|tkr||}nD|tkr@||}|r^|jr^t|}n|tkrR||St d||rp|j rpt |S|S)NNo reader for class code %s) r<rread_full_arrayrread_char_arrayrJr rread_sparse_array TypeErrorrKr )r?hdrprocessr<arrr@r@rAarray_from_headers       zVarReader4.array_from_headerc Csp|j}|j}|j}|D] }||9}q|jt|}t||krNtd|jt j |||dd}|rl| }|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=itemsizerHrRrSlenrUr:npndarraycopy) r?rfrrdtr= num_bytesdrlrhr@r@rAread_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)rry?)r>rv)r?rfresZres_jr@r@rArbs  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-1r2rkr;rl) rvastyperpZuint8tobytesdecoderqr=r;arrayrr)r?rfrhSr@r@rArcs zVarReader4.read_char_arraycCs||}|ddddf}t|dt|df}tj|dddfdd}tj|dddfdd}|d8}|d8}|jdd krtj|ddd fd d}n,tj|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)rr)rrrZintcr;rr#r"floatcomplex) rvrSrpZascontiguousarrayrkimagscipyr8Z coo_matrix)r?rfrwtmpr=IJVr@r@rArds zVarReader4.read_sparse_arraycCs2|j}|tkr ttt|j}n|tkrLttt|j}|jrJ|dd}n|tkr|j }|j}t |dkr|ddkr|ddksdS|j |j |dddtjd||j |j d}|j |j |dddtjd||j |j d}t|t|f}n td||jr.td d |D}|S) zwRead the shape of the array described by the header. The file position after this call is unspecified. Nrr"rrr@ryracSsg|]}|dkr|qS)rr@).0xr@r@rA 0z0VarReader4.shape_from_header..)r<rtuplemaprSr=rrJrr;rorHseekrnrprqrRrerK)r?rfr<rkrsr=rowscolsr@r@rAshape_from_headers2 $   zVarReader4.shape_from_headerN)T)T) rCrDrE__doc__rBr`rirvrbrcrdrr@r@r@rArfs  $*rcsXeZdZdZefddZddZddZdd Zdd d Z dddZ ddZ Z S)r z Reader for Mat4 files cs$tj|g|Ri|d|_dS)zR Initialize matlab 4 file reader %(matstream_arg)s %(load_args)s N)superrB_matrix_reader)r?rHargskwargs __class__r@rArB6szMatFile4Reader.__init__cCs^|jdt|jtd}|jd|dkr6dS|dksF|dkrRtrNdpPdStrZdp\dS)Nrr(r4rMr5)rHrrrpr;r)r?r-r@r@rAguess_byte_order@s   zMatFile4Reader.guess_byte_ordercCstt|j|_t||_dS)za Run when beginning read of variables Sets up readers from parameters in `self` N)rrZ byte_orderrIrr)r?r@r@rAinitialize_readLszMatFile4Reader.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||Sr9r@)ryr@r@rAdrz0MatFile4Reader.read_var_header..rr") rr`r r=r;rnr>r<rrHtell)r?rfnZremaining_bytes next_positionr@r@rAread_var_headerTs  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`. )rri)r?r,rgr@r@rAread_var_arraykszMatFile4Reader.read_var_arrayNcCst|tr|g}n|dur"t|}|jd|i}|s|\}}|jdur\dn |j d}|dur||vr|j|q:| |||<|j||dur:| |t |dkr:qq:|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) isinstancestrlistrHrr end_of_streamrr:r|rremovero)r?Zvariable_namesmdictrfrr:r@r@rA get_variables}s(       zMatFile4Reader.get_variablescCs|jd|g}|s~|\}}|jdur:dn |jd}|j|}t |j d}| |||f|j|q|S)z list variables from stream rNrrunknown) rHrrrrr:r|rrrgetr<append)r?varsrfrr:rkinfor@r@rAlist_variabless   zMatFile4Reader.list_variables)T)N) rCrDrErrrBrrrrrr __classcell__r@r@rrAr 4s    r rowcCs(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 r"z=Matlab 4 files cannot save arrays with more than 2 dimensions)rrorUZreshape)rhoned_asr=r@r@rArs  rc@sTeZdZddZddZddZeedfdd Zd d Z d d Z ddZ ddZ dS)rcCs|j|_|j|_dSr9) file_streamr)r?Z file_writerr@r@rArBszVarWriter4.__init__cCs|j|jdddS)Nrj)rm)rwriter{)r?rhr@r@rA write_bytesszVarWriter4.write_bytescCs|j|dSr9)rr)r?sr@r@rA 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 r@r,rrNrPrQr-r.rr/r0r1rN)rpemptyrrrorrencode) r?r:rkr^r_r0r,r[r]rZr@r@rA write_headers"   zVarWriter4.write_headercCstj|r|||dSt|}|j}|jsB|| d}|j }|tj ur\t dn4|tj urpt dn |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)rr8issparse write_sparserpZasarrayr;ZisnativerzZ newbyteordertypeZobject_revoidstr_bytes_ write_char write_numeric)r?rhr:rsZdttr@r@rArs        zVarWriter4.writecCst||j}|jjdk}zt|jjdd}Wn0ty`|rN|d}n |d}t}Yn0|j ||j |t |d|r| |j | |jn | |dS)NcrZc128r&)r^r_r0)rrr;kindrrKeyErrorrzrrrkrrrealr)r?rhr:r0r^r@r@rArs&       zVarWriter4.write_numericcCs|jjtjkr*|jjtdjkr*t|}t||j}|j}|j ||t t d|jj dkrt |}tjdt|||d}|d}tj|d|d}||dS)Nr2r^r_Ur@ryrxr3)r;rrprrnrrrrkrrrrprodrqr itemrr)r?rhr:r=Zn_charsZst_arrstr@r@rAr$s&"   zVarWriter4.write_charcCs|}|jjdk}tj|jdd|fdd}|j|dddf<|j|dddf<|dddd fd7<|r|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 rrr#r&rNrrr"r)Ztocoor;rrpZzerosZnnzrcolrZrrrkrrrr)r?rhr:Ar0Zijvr@r@rAr8s$ zVarWriter4.write_sparseN) rCrDrErBrrrrrrrrrr@r@r@rArs!rc@s$eZdZdZdddZdddZdS) rz) Class for writing matlab 4 format files NcCs"||_|durd}||_d|_dS)Nr)rr_matrix_writer)r?rrr@r@rArBSs zMatFile4Writer.__init__cCs.t||_|D]\}}|j||qdS)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)r?rrr:varr@r@rA put_variablesZs zMatFile4Writer.put_variables)N)N)rCrDrErrBrr@r@r@rArQs r)r)*rsysrWnumpyrpZ scipy.sparserZ_miobaserrrrrrr Z _mio_utilsr r functoolsr __all__ byteorderrrrrrrrrrrrrr rrrr rrrr@r@r@rAsr$   Oy