3 Pd!O@sjdZddlZddlZddlZddlmZmZddlZ ddl m Z m Z m Z mZmZmZmZddlmZmZddlmZejdkZdZdZd Zd Zd Zd Zed edededededdd1d2d3d4d5gddiZeeeeeeeeeeed Z dZ!dZ"d Z#dddddd Z$e!d!e"d"e#d#iZ%Gd$d%d%e&Z'Gd&d'd'e&Z(Gd(d)d)e Z)d6d+d,Z*Gd-d.d.e&Z+Gd/d0d0e&Z,dS)7z1 Classes for read / write of matlab (TM) 4 files N)asbytesasstr) MatFileReader docfillermatdims read_dtypeconvert_dtypes arr_to_charsarr_dtype_number)squeeze_elementchars_to_strings)reducelittlef8f4i4i2u2u1headermoptmrowsncolsimagfnamlenU1) rZc32Zc24Zc16rZc8rrrrS1<>z VAX D-floatz VAX G-floatZCray)rrrrrdoublecharsparsec@seZdZdZdZddZdS) VarHeader4FcCs"||_||_||_||_||_dS)N)namedtypemclassdims is_complex)selfr(r)r*r+r,r.H/var/www/html/virt/lib64/python3.6/site-packages/scipy/io/matlab/mio4.py__init__Qs zVarHeader4.__init__N)__name__ __module__ __qualname__Z is_logical is_globalr0r.r.r.r/r'Lsr'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)r-r6r.r.r/r0as zVarReader4.__init__c Cst|j|jd}|jjt|djd}|ddksD|ddkrLtdt|dd\}}|dkrztj d t |t t|d \}}|dkrtd t|d \}}|}|d|df}|dd k} |j|} t || ||| S)z% Read and return header for variable rrrriz.Mat 4 mopt wrong format, byteswapping problem?irzBWe do not support byte ordering '%s'; returned data may be corruptdz,O in MOPT integer should be 0, wrong format? rrr)rr) rr7r8readintstrip ValueErrordivmodwarningswarn order_codes UserWarningr') r-datar(MrestOPTr+r,r)r.r.r/ read_headerhs.  zVarReader4.read_headerTcCst|j}|tkr|j|}nD|tkr@|j|}|r^|jr^t|}n|tkrR|j|St d||rp|j rpt |S|S)NzNo reader for class code %s) r* mxFULL_CLASSread_full_array mxCHAR_CLASSread_char_arrayr9r mxSPARSE_CLASSread_sparse_array TypeErrorr:r )r-hdrprocessr*arrr.r.r/array_from_headers       zVarReader4.array_from_headerc Cst|j}|j}|j}x|D] }||9}qW|jjt|}t||krRtd|jt j |||dd}|rp|j }|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+itemsizer7r>r?lenrAr(npndarraycopy) r-rUradtr+ num_bytesdr[rWr.r.r/read_sub_arrays     zVarReader4.read_sub_arraycCs8|jr.|j|dd}|j|dd}||dS|j|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)ray?)r,re)r-rUresZres_jr.r.r/rOs  zVarReader4.read_full_arraycCsB|j|jtj}|jjd}tj|jtjdtj |dj 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 )rZr)r[) reastyper_Zuint8tobytesdecoder`r+r)arrayra)r-rUrWSr.r.r/rQs zVarReader4.read_char_arraycCs|j|}|dd ddf}t|d t|df}tj|dddfdd}tj|dddfdd}|d8}|d8}|jddkrtj|dddfdd}n,tj|dddfd d}|dddf|_tjj|||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. NrrZintc)r)rrfloatcomplexrn)rnrrn)rnr) rer?r_ZascontiguousarrayrZimagscipyr&Z coo_matrix)r-rUrftmpr+IJVr.r.r/rSs zVarReader4.read_sparse_arraycCs2|j}|tkr ttt|j}n|tkrLttt|j}|jrJ|dd }n|tkr|j }|j}t |dko|ddko|ddksfS|j j |j |dddtjf||j j|j d}|j j |j |dddtjf||j j|j d}t|t|f}n td||jr.tdd|D}|S) zwRead the shape of the array described by the header. The file position after this call is unspecified. Nrrr)rZr)r[zNo reader for class code %scSsg|]}|dkr|qS)rr.).0xr.r.r/ (sz0VarReader4.shape_from_header..rn)r*rNtuplemapr?r+rPr9rRr)r^r7seekr]r_r`r>rTr:)r-rUr*rZrbr+rowscolsr.r.r/shape_from_header s. $ zVarReader4.shape_from_headerN)T)T) r1r2r3__doc__r0rMrXrerOrQrSr}r.r.r.r/r5^s  $*r5csXeZdZdZefddZddZddZdd 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)superrr0_matrix_reader)r-r7argskwargs) __class__r.r/r0.szMatFile4Reader.__init__cCs^|jjdt|jtjd}|jjd|dkr6dS|dksF|dkrRtrNdpPdStrZdp\dS)Nrrr"ir#)r7rzrr_r)SYS_LITTLE_ENDIAN)r-rr.r.r/guess_byte_order8s   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_orderr8r5r)r-r.r.r/initialize_readDszMatFile4Reader.initialize_readcCsX|jj}tdd|jd}|jj|}|jrB|jtk rB|d9}|j 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)Nr.)rvyr.r.r/\sz0MatFile4Reader.read_var_header..rr) rrMrr+r)r]r,r*rRr7tell)r-rUnZremaining_bytes next_positionr.r.r/read_var_headerLs  zMatFile4Reader.read_var_headerTcCs|jj||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)r-rrVr.r.r/read_var_arraycszMatFile4Reader.read_var_arrayNcCst|tr|g}n|dk r"t|}|jjd|ji}xz|js|j\}}t|j }|dk rx||krx|jj|q<|j |||<|jj||dk r<|j |t |dkrsr$  Oy