B &]\O@sdZddlmZmZmZddlZddlZddlZddl m Z m Z ddl Z ddlmZddlmZmZmZmZmZmZmZddlmZmZdd lmZejd kZdZdZ d Z!d Z"d Z#dZ$ede de!de"de#de$dddddddgddiZ%eeeee e e!e"e#e$e$d Z&dZ'dZ(d Z)dddd d!d"Z*e'd#e(d$e)d%iZ+Gd&d'd'e,Z-Gd(d)d)e,Z.Gd*d+d+eZ/d3d-d.Z0Gd/d0d0e,Z1Gd1d2d2e,Z2dS)4z1 Classes for read / write of matlab (TM) 4 files )divisionprint_functionabsolute_importN)asbytesasstr) string_types) MatFileReader docfillermatdims read_dtypeconvert_dtypes arr_to_charsarr_dtype_number)squeeze_elementchars_to_strings)reducelittlef8f4i4i2u2u1header)moptr)mrowsr)ncolsr)imagfr)namlenrU1) rZc32Zc24Zc16rZc8rrrrS1<>z VAX D-floatz VAX G-floatZCray)rrrrrZdoublecharsparsec@seZdZdZdZddZdS) VarHeader4FcCs"||_||_||_||_||_dS)N)namedtypemclassdims is_complex)selfr+r,r-r.r/r13lib/python3.7/site-packages/scipy/io/matlab/mio4.py__init__Us zVarHeader4.__init__N)__name__ __module__ __qualname__Z is_logicalZ is_globalr3r1r1r1r2r*Psr*c@sTeZdZdZddZddZdddZdd d Zd d Zd dZ ddZ ddZ dS) VarReader4z" Class to read matlab 4 variables cCs*||_|j|_|j|_|j|_|j|_dS)N) file_reader mat_streamdtypeschars_as_strings squeeze_me)r0r8r1r1r2r3es zVarReader4.__init__c Cst|j|jd}|jt|dd}|ddksD|ddkrLtdt|dd\}}|d krzt d t |t t|d \}}|dkrtd t|d \}}|}|d|df}|ddk} |j|} t || ||| S)z% Read and return header for variable rr#rriz.Mat 4 mopt wrong format, byteswapping problem?i)rrzBWe do not support byte ordering '%s'; returned data may be corruptdz,O in MOPT integer should be 0, wrong format? r r!r"r) r r9r:readintstrip ValueErrordivmodwarningswarn order_codes UserWarningr*) r0datar+MrestOPTr.r/r,r1r1r2 read_headerls.  zVarReader4.read_headerTcCst|j}|tkr||}nD|tkr@||}|r^|jr^t|}n|tkrR||St d||rp|j rpt |S|S)NzNo reader for class code %s) r- mxFULL_CLASSread_full_array mxCHAR_CLASSread_char_arrayr;rmxSPARSE_CLASSread_sparse_array TypeErrorr<r)r0hdrprocessr-arrr1r1r2array_from_headers       zVarReader4.array_from_headerc Cst|j}|j}|j}x|D] }||9}qW|jt|}t||krRtd|jt j |||dd}|rp| }|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 givem by `hdr` ``dtype`` and shape givem 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.itemsizer9r@rAlenrCr+npndarraycopy) r0rWrcdtr.Z num_bytesdr]rYr1r1r2read_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)rcy?)r/rf)r0rWresZres_jr1r1r2rQs  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`` zlatin-1r$)r\r,r]) rfastyperaZuint8tostringdecoderbr.r,Zarrayrc)r0rWrYSr1r1r2rSs 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 is 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)rlr)rlrrZintc)r,rrrfloatcomplex) rfrAraZascontiguousarrayr\imagscipyr)Z coo_matrix)r0rWrgZtmpr.IJVr1r1r2rUs 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. Nrlrrrr1)r)r\r,r]zNo reader for class code %scSsg|]}|dkr|qS)rr1).0xr1r1r2 ,sz0VarReader4.shape_from_header..)r-rPtuplemaprAr.rRr;rTr,r`r9seekr_rarbr@rVr<)r0rWr-r\rdr.ZrowsZcolsr1r1r2shape_from_header s. $ zVarReader4.shape_from_headerN)T)T) r4r5r6__doc__r3rOrZrfrQrSrUrzr1r1r1r2r7bs  $*r7csXeZdZdZefddZddZddZdd Zdd d Z dddZ ddZ Z S)MatFile4Readerz Reader for Mat4 files cs"tt|j|f||d|_dS)zR Initialize matlab 4 file reader %(matstream_arg)s %(load_args)s N)superr|r3_matrix_reader)r0r9argskwargs) __class__r1r2r32szMatFile4Reader.__init__cCs^|jdt|jtd}|jd|dkr6dS|dksF|dkrRtrNdpPdStrZdp\dS)Nrrr&ir')r9ryr rar,SYS_LITTLE_ENDIAN)r0rr1r1r2guess_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)r mdtypes_templateZ byte_orderr:r7r~)r0r1r1r2initialize_readHszMatFile4Reader.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||S)Nr1)ruyr1r1r2`sz0MatFile4Reader.read_var_header..rr) r~rOrr.r,r_r/r-rTr9tell)r0rWnZremaining_bytes next_positionr1r1r2read_var_headerPs  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`. )r~rZ)r0rrXr1r1r2read_var_arraygszMatFile4Reader.read_var_arrayNcCst|tr|g}n|dk r"t|}|jd|i}xz|s|\}}t|j }|dk rx||krx|j|q<| |||<|j||dk r<| |t |dkrr?rr rr!r"r#N)raemptyrrr`rrr) r0r+r\rMrNr"rrJrLr1r1r2 write_headers    zVarWriter4.write_headercCstj|r|||dSt|}|j}|jsB|| d}|j }|tj kr\t dn4|tj krpt dn |tjtjfkr|||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)rpr)Zissparse write_sparseraZasarrayr,ZisnativerhZ newbyteordertypeZobject_rVZvoidZunicode_Zstring_ write_char write_numeric)r0rYr+rdZdttr1r1r2rs        zVarWriter4.writecCst||j}|jjdk}yt|jjdd}Wn2tk rb|rP|d}n |d}t}YnX|j ||j |t |d|r| |j | |jn | |dS)NcrZc128r)rMrNr")rrr,kind np_to_mtypesstrKeyErrorrhmiDOUBLErr\rPrrealro)r0rYr+r"rMr1r1r2r s$      zVarWriter4.write_numericcCst|}t||j}|j}|j||ttd|jjdkrvt |}t j dt |||d}| d}t j |d|d}||dS)N)rMrNUr1)r\r,r]zlatin-1r%)rrrr\rmiUINT8rRr,rraproductrbritemencoder)r0rYr+r.Zn_charsZst_arrstr1r1r2rs    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) zX Sparse matrices are 2D See docstring for VarReader4.read_sparse_array rrrr)r,Nrlrr)rMrN)Ztocoor,rraZzerosZnnzrcolrIrror\rrrTr)r0rYr+Ar"Zijvr1r1r2r2s" zVarWriter4.write_sparseN) r4r5r6r3rrrrPrrrrrr1r1r1r2rs rc@s$eZdZdZdddZdddZdS) MatFile4Writerz) Class for writing matlab 4 format files NcCs"||_|dkrd}||_d|_dS)Nr)rr_matrix_writer)r0rrr1r1r2r3Ms zMatFile4Writer.__init__cCs2t||_x"|D]\}}|j||qWdS)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)r0rrr+varr1r1r2 put_variablesTs zMatFile4Writer.put_variables)N)N)r4r5r6r{r3rr1r1r1r2rKs r)r)3r{Z __future__rrrsysrEZnumpyraZ numpy.compatrrZ scipy.sparserpZscipy._lib.sixrZmiobaser r r r r rrZ mio_utilsrr functoolsr byteorderrrZmiSINGLEZmiINT32ZmiINT16ZmiUINT16rrrrPrRrTrGrobjectr*r7r|rrrr1r1r1r2sv $  Oy