σ Λ½χXc@`s—dZddlmZmZmZddlZddlZddlmZddl Z ej ddkrre Z neZ ddlmZddlmZd efd „ƒYZd efd „ƒYZd efd„ƒYZi dd6dd6dd6dd6dd6dd6dd6dd6dd 6ZejeƒZd!„Zd"„Zd#„Zd$d%„Zd&e fd'„ƒYZ!d(e fd)„ƒYZ"d*„Z#d+„Z$dS(,sf Base classes for MATLAB file stream reading. MATLAB is a registered trademark of the Mathworks inc. i(tdivisiontprint_functiontabsolute_importN(treducei(tdocceri(tbyteordercodest MatReadErrorcB`seZRS((t__name__t __module__(((s6/tmp/pip-build-7oUkmx/scipy/scipy/io/matlab/miobase.pyRst MatWriteErrorcB`seZRS((RR(((s6/tmp/pip-build-7oUkmx/scipy/scipy/io/matlab/miobase.pyR stMatReadWarningcB`seZRS((RR(((s6/tmp/pip-build-7oUkmx/scipy/scipy/io/matlab/miobase.pyR #ssfile_name : str Name of the mat file (do not need .mat extension if appendmat==True) Can also pass open file-like object.tfile_args|appendmat : bool, optional True to append the .mat extension to the end of the given filename, if not already present.t append_argsžbyte_order : str or None, optional None by default, implying byte order guessed from mat file. Otherwise can be one of ('native', '=', 'little', '<', 'BIG', '>'). mat_dtype : bool, optional If True, return arrays in same dtype as would be loaded into MATLAB (instead of the dtype with which they are saved). squeeze_me : bool, optional Whether to squeeze unit matrix dimensions or not. chars_as_strings : bool, optional Whether to convert char arrays to string arrays. matlab_compatible : bool, optional Returns matrices as would be loaded by MATLAB (implies squeeze_me=False, chars_as_strings=False, mat_dtype=True, struct_as_record=True).t load_argssdstruct_as_record : bool, optional Whether to load MATLAB structs as numpy record arrays, or as old-style numpy arrays with dtype=object. Setting this flag to False replicates the behavior of scipy version 0.7.x (returning numpy object arrays). The default setting is True, because it allows easier round-trip load and save of MATLAB files.t struct_argsAmat_stream : file-like Object with file API, open for reading.t matstream_argslong_field_names : bool, optional * False - maximum field name length in a structure is 31 characters which is the documented maximum length. This is the default. * True - maximum field name length in a structure is 63 characters which works for MATLAB 7.6t long_fieldss[do_compression : bool, optional Whether to compress matrices on write. Default is False.tdo_compressions’oned_as : {'row', 'column'}, optional If 'column', write 1-D numpy arrays as column vectors. If 'row', write 1D numpy arrays as row vectors.toned_assbunicode_strings : bool, optional If True, write strings as Unicode, else MATLAB usual encoding.tunicode_stringscC`sA|jƒ}x.|D]&}tj||ƒj|ƒ|| version 4 format mat files #. 1,x -> version 5 format mat files #. 2,x -> version 7.3 format mat files (HDF format) Parameters ---------- fileobj : file_like object implementing seek() and read() Returns ------- major_version : {0, 1, 2} major MATLAB File format version minor_version : int minor MATLAB file format version Raises ------ MatReadError If the file is empty. ValueError The matfile version is unknown. Notes ----- Has the side effect of setting the file read pointer to 0 iisMat file appears to be emptyRRRi|itIis%Unknown mat file type, version %s, %sN(i(iiR)(ii( tseekR#tlenRRR"tuint8tinttbyteordt ValueError(tfileobjt mopt_bytest mopt_intsttst_strtmaj_indtmaj_valtmin_valtret((s6/tmp/pip-build-7oUkmx/scipy/scipy/io/matlab/miobase.pytget_matfile_version»s$" !      tcolumncC`sœ|j}|dkrdSttj|ƒdkrKd tj|jdgƒSt|ƒdkr˜|dkrq|d S|dkr…d |Std|ƒ‚n|S( s– Determine equivalent MATLAB dimensions for given array Parameters ---------- arr : ndarray Input array oned_as : {'column', 'row'}, optional Whether 1-D arrays are returned as MATLAB row or column matrices. Default is 'column'. Returns ------- dims : tuple Shape tuple, in the form MATLAB expects it. Notes ----- We had to decide what shape a 1 dimensional array would be by default. ``np.atleast_2d`` thinks it is a row vector. The default for a vector in MATLAB (e.g. ``>> 1:12``) is a row vector. Versions of scipy up to and including 0.11 resulted (accidentally) in 1-D arrays being read as column vectors. For the moment, we maintain the same tradition here. Examples -------- >>> matdims(np.array(1)) # numpy scalar (1, 1) >>> matdims(np.array([1])) # 1d array, 1 element (1, 1) >>> matdims(np.array([1,2])) # 1d array, 2 elements (2, 1) >>> matdims(np.array([[2],[3]])) # 2d array, column vector (2, 1) >>> matdims(np.array([[2,3]])) # 2d array, row vector (1, 2) >>> matdims(np.array([[[2,3]]])) # 3d array, rowish vector (1, 1, 2) >>> matdims(np.array([])) # empty 1d array (0, 0) >>> matdims(np.array([[]])) # empty 2d (0, 0) >>> matdims(np.array([[[]]])) # empty 3d (0, 0, 0) Optional argument flips 1-D shape behavior. >>> matdims(np.array([1,2]), 'row') # 1d array, 2 elements (1, 2) The argument has to make sense though >>> matdims(np.array([1,2]), 'bizarre') Traceback (most recent call last): ... ValueError: 1D option "bizarre" is strange iiiR9trows1D option "%s" is strange((ii(i(i(i( RRtoperatortmulRtmaxtndimR+R/(R'RR((s6/tmp/pip-build-7oUkmx/scipy/scipy/io/matlab/miobase.pytmatdimsτs=     t MatVarReadercB`s)eZdZd„Zd„Zd„ZRS(s; Abstract class defining required interface for var readerscC`sdS(N((tselft file_reader((s6/tmp/pip-build-7oUkmx/scipy/scipy/io/matlab/miobase.pyt__init__CscC`sdS(s Returns header N((RA((s6/tmp/pip-build-7oUkmx/scipy/scipy/io/matlab/miobase.pyt read_headerFscC`sdS(s Reads array given header N((RAtheader((s6/tmp/pip-build-7oUkmx/scipy/scipy/io/matlab/miobase.pytarray_from_headerJs(RRt__doc__RCRDRF(((s6/tmp/pip-build-7oUkmx/scipy/scipy/io/matlab/miobase.pyR@As  t MatFileReaderc B`sMeZdZedeeeeeed„ƒZd„Zd„Z d„Z RS(s  Base object for reading mat files To make this class functional, you will need to override the following methods: matrix_getter_factory - gives object to fetch next matrix from stream guess_byte_order - guesses file byte order from file c C`sƒ||_i|_|s'|jƒ}ntj|ƒ}||_||_|r[|jƒn||_||_ ||_ ||_ dS(s– Initializer for mat file reader mat_stream : file-like object with file API, open for reading %(load_args)s N( R$Rtguess_byte_ordertboct to_numpy_codet byte_ordertstruct_as_recordtset_matlab_compatiblet squeeze_metchars_as_stringst mat_dtypet verify_compressed_data_integrity( RAR$RLRQRORPtmatlab_compatibleRMRR((s6/tmp/pip-build-7oUkmx/scipy/scipy/io/matlab/miobase.pyRCYs        cC`st|_t|_t|_dS(s4 Sets options to return arrays as MATLAB loads them N(tTrueRQtFalseRORP(RA((s6/tmp/pip-build-7oUkmx/scipy/scipy/io/matlab/miobase.pyRN{s  cC`stjS(s9 As we do not know what file type we have, assume native (RJt native_code(RA((s6/tmp/pip-build-7oUkmx/scipy/scipy/io/matlab/miobase.pyRIscC`sE|jjdƒ}|jjƒ}|jj|dƒt|ƒdkS(Nii(R$R#ttellR*R+(RAtbtcurpos((s6/tmp/pip-build-7oUkmx/scipy/scipy/io/matlab/miobase.pyt end_of_stream…sN( RRRGt docfillertNoneRURTRCRNRIRZ(((s6/tmp/pip-build-7oUkmx/scipy/scipy/io/matlab/miobase.pyRHOs  cC`s!tj|jjd t|ƒƒS(s3 Return dtype for given number of items per elementi(RRtstr(R'tnum((s6/tmp/pip-build-7oUkmx/scipy/scipy/io/matlab/miobase.pytarr_dtype_numberŒscC`s‘t|jƒ}|s!dg}n|jt|jjdƒƒtjd|dt|dƒd|ƒ}|dkg}tj |ƒs‡|S|j ƒ}d||<|S(s$ Convert string array to char array iiRRRtt ( tlistRtappendR-RR]RR"R_tanyR(R'tdimstempties((s6/tmp/pip-build-7oUkmx/scipy/scipy/io/matlab/miobase.pyt arr_to_chars‘s    (%RGt __future__RRRtsysR;tscipy._lib.sixRtnumpyRt version_infoR-R.tordt scipy.miscRR`RRJt ExceptionRR t UserWarningR tdoc_dicttfilldocR[RR(R8R?tobjectR@RHR_Rg(((s6/tmp/pip-build-7oUkmx/scipy/scipy/io/matlab/miobase.pyts@     /   9 M=