B „Ž0d/ã @sØdZddlZddlZddlZddlmZddlmZ Gdd„de ƒZ Gdd „d e ƒZ Gd d „d e ƒZd d ddddddddœ Ze e¡Zdd„Zdd„Zdd„Zd'dd„ZGdd „d ƒZGd!d"„d"ƒZd#d$„Zd%d&„ZdS)(zf Base classes for MATLAB file stream reading. MATLAB is a registered trademark of the Mathworks inc. éN)Údocceré)Úbyteordercodesc@s eZdZdS)Ú MatReadErrorN)Ú__name__Ú __module__Ú __qualname__©r r úI/var/www/html/venv/lib/python3.7/site-packages/scipy/io/matlab/miobase.pyrsrc@s eZdZdS)Ú MatWriteErrorN)rrrr r r r r sr c@s eZdZdS)ÚMatReadWarningN)rrrr r r r r sr zfile_name : str Name of the mat file (do not need .mat extension if appendmat==True) Can also pass open file-like object.z|appendmat : bool, optional True to append the .mat extension to the end of the given filename, if not already present.až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).abstruct_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.zAmat_stream : file-like Object with file API, open for reading.along_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.6z[do_compression : bool, optional Whether to compress matrices on write. Default is False.z’oned_as : {'row', 'column'}, optional If 'column', write 1-D NumPy arrays as column vectors. If 'row', write 1D NumPy arrays as row vectors.zbunicode_strings : bool, optional If True, write strings as Unicode, else MATLAB usual encoding.) Zfile_argZ append_argZ load_argsZ struct_argZ matstream_argZ long_fieldsZdo_compressionÚoned_asZunicode_stringscCs2| ¡}x$|D]}t ||¡ |¡||<qW|S)až Convert dtypes in mapping to given order Parameters ---------- dtype_template : mapping mapping with values returning numpy dtype from ``np.dtype(val)`` order_code : str an order code suitable for using in ``dtype.newbyteorder()`` Returns ------- dtypes : mapping mapping where values have been replaced by ``np.dtype(val).newbyteorder(order_code)`` )ÚcopyÚnpÚdtypeZ newbyteorder)Zdtype_templateZ order_codeÚdtypesÚkr r r Úconvert_dtypes€s rcCs"|j}tjd|| |¡dd}|S)af Generic get of byte stream data of known type Parameters ---------- mat_stream : file_like object MATLAB (tm) mat file stream a_dtype : dtype dtype of array to read. `a_dtype` is assumed to be correct endianness. Returns ------- arr : ndarray Array of dtype `a_dtype` read from stream. r ÚF)ÚshaperÚbufferÚorder)ÚitemsizerÚndarrayÚread)Ú mat_streamZa_dtypeÚ num_bytesÚarrr r r Ú read_dtype—s rcCs¾| d¡| d¡}t|ƒdkr(tdƒ‚tjdtj|d}d|krP| d¡dS| d¡| d¡}| d¡t|dd kƒ}t||ƒ}t|d |ƒ}||f}|d kr®|Std |ƒ‚d S)a· Return major, minor tuple depending on apparent mat file type Where: #. 0,x -> 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 rézMat file appears to be empty)r)rrr)rré|ééIr)rr!z%Unknown mat file type, version %s, %sN) ÚseekrÚlenrrrZuint8ÚintÚ ValueError)ÚfileobjZ mopt_bytesZ mopt_intsZtst_strZmaj_indZmaj_valZmin_valÚretr r r Úget_matfile_version±s$"        r)ÚcolumncCs^|j}|dkrdSt|ƒdkrZ|ddkr.dS|dkr>|dS|dkrNd|Std |ƒ‚|S) aª 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])) # 1-D array, 1 element (1, 1) >>> matdims(np.array([1,2])) # 1-D array, 2 elements (2, 1) >>> matdims(np.array([[2],[3]])) # 2-D array, column vector (2, 1) >>> matdims(np.array([[2,3]])) # 2-D array, row vector (1, 2) >>> matdims(np.array([[[2,3]]])) # 3-D array, rowish vector (1, 1, 2) >>> matdims(np.array([])) # empty 1-D array (0, 0) >>> matdims(np.array([[]])) # empty 2-D array (0, 0) >>> matdims(np.array([[[]]])) # empty 3-D array (0, 0, 0) Optional argument flips 1-D shape behavior. >>> matdims(np.array([1,2]), 'row') # 1-D array, 2 elements (1, 2) The argument has to make sense though >>> matdims(np.array([1,2]), 'bizarre') Traceback (most recent call last): ... ValueError: 1-D option "bizarre" is strange r )rrrr)rrr*)rÚrowz1-D option "%s" is strange)rr$r&)rr rr r r Úmatdimsês=  r,c@s(eZdZdZdd„Zdd„Zdd„ZdS) Ú MatVarReaderz; Abstract class defining required interface for var readerscCsdS)Nr )ÚselfZ file_readerr r r Ú__init__9szMatVarReader.__init__cCsdS)z Returns header Nr )r.r r r Ú read_header<szMatVarReader.read_headercCsdS)z Reads array given header Nr )r.Úheaderr r r Úarray_from_header@szMatVarReader.array_from_headerN)rrrÚ__doc__r/r0r2r r r r r-7sr-c @s6eZdZdZed dd„ƒZdd„Zd d „Zd d „ZdS)Ú MatFileReadera  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 NFTc Csp||_i|_|s| ¡}n t |¡}||_||_|r>| ¡n||_||_ ||_ ||_ | |_ | rld|_d|_dS)z– Initializer for mat file reader mat_stream : file-like object with file API, open for reading %(load_args)s TFN) rrÚguess_byte_orderÚbocZ to_numpy_codeÚ byte_orderÚstruct_as_recordÚset_matlab_compatibleÚ squeeze_meÚchars_as_stringsÚ mat_dtypeÚ verify_compressed_data_integrityÚsimplify_cells) r.rr7r<r:r;Zmatlab_compatibler8r=r>r r r r/Os"   zMatFileReader.__init__cCsd|_d|_d|_dS)z4 Sets options to return arrays as MATLAB loads them TFN)r<r:r;)r.r r r r9usz#MatFileReader.set_matlab_compatiblecCstjS)z9 As we do not know what file type we have, assume native )r6Z native_code)r.r r r r5{szMatFileReader.guess_byte_ordercCs2|j d¡}|j ¡}|j |d¡t|ƒdkS)Nrr)rrÚtellr#r$)r.ÚbZcurposr r r Ú end_of_streams  zMatFileReader.end_of_stream)NFFTFTTF) rrrr3Ú docfillerr/r9r5rAr r r r r4Es r4cCst |jjdd…t|ƒ¡S)z3 Return dtype for given number of items per elementNr!)rrÚstr)rÚnumr r r Úarr_dtype_number†srEcCstt|jƒ}|sdg}| t|jjdd…ƒ¡tj|t|dƒ|d}|dkg}t  |¡s\|S|  ¡}d|t |ƒ<|S)z$ Convert string array to char array rr!N)rrrÚú ) ÚlistrÚappendr%rrCrrrEÚanyrÚtuple)rZdimsZemptiesr r r Ú arr_to_chars‹s    rL)r*)r3ÚoperatorÚ functoolsÚnumpyrZ scipy._librrFrr6Ú Exceptionrr Ú UserWarningr Zdoc_dictZfilldocrBrrr)r,r-r4rErLr r r r Ús4   /9 MA