ó ˽÷Xc@`södZddlmZmZmZddlZddlmZddl m Z m Z ddl m Z mZddlmZmZd d d d gZd „Ze ed„ƒZe ded„ƒZe edeedd„ƒZe ed„ƒZdS(s7 Module for reading and writing matlab (TM) .mat files i(tdivisiontprint_functiontabsolute_importN(t string_typesi(tget_matfile_versiont docfiller(tMatFile4ReadertMatFile4Writer(tMatFile5ReadertMatFile5Writertmat_reader_factorytloadmattsavemattwhosmatcC`s¹t|tƒr„yt|dƒSWq„tk r€}|rz|jdƒ rz|d7}yt|dƒSWqztk rvqzXn‚q„Xny|jdƒWntk r´tdƒ‚nX|S(s1 Open `file_like` and return as file-like object trbs.matis/Reader needs file name or open file-like object(t isinstanceRtopentIOErrortendswithtreadtAttributeError(t file_liket appendmatte((s2/tmp/pip-build-7oUkmx/scipy/scipy/io/matlab/mio.pyt _open_files     cK`s‚t||ƒ}t|ƒ\}}|dkr:t||S|dkrSt||S|dkrntdƒ‚ntd|ƒ‚dS(sH Create reader for matlab .mat format files. Parameters ---------- %(file_arg)s %(append_arg)s %(load_args)s %(struct_arg)s Returns ------- matreader : MatFileReader object Initialized instance of MatFileReader class matching the mat file type detected in `filename`. iiis+Please use HDF reader for matlab v7.3 filessDid not recognize version %sN(RRRRtNotImplementedErrort TypeError(t file_nameRtkwargst byte_streamtmjvtmnv((s2/tmp/pip-build-7oUkmx/scipy/scipy/io/matlab/mio.pyR (s     cK`sx|jddƒ}t|||}|j|ƒ}|dk rO|j|ƒn|}t|tƒrt|jjƒn|S(su Load MATLAB file. Parameters ---------- file_name : str Name of the mat file (do not need .mat extension if appendmat==True). Can also pass open file-like object. mdict : dict, optional Dictionary in which to insert matfile variables. appendmat : bool, optional True to append the .mat extension to the end of the given filename, if not already present. 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). struct_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. verify_compressed_data_integrity : bool, optional Whether the length of compressed sequences in the MATLAB file should be checked, to ensure that they are not longer than we expect. It is advisable to enable this (the default) because overlong compressed sequences in MATLAB files generally indicate that the files have experienced some sort of corruption. variable_names : None or sequence If None (the default) - read all variables in file. Otherwise `variable_names` should be a sequence of strings, giving names of the matlab variables to read from the file. The reader will skip any variable with a name not in this sequence, possibly saving some read processing. Returns ------- mat_dict : dict dictionary with variable names as keys, and loaded matrices as values. Notes ----- v4 (Level 1.0), v6 and v7 to 7.2 matfiles are supported. You will need an HDF5 python library to read matlab 7.3 format mat files. Because scipy does not supply one, we do not implement the HDF5 / 7.3 interface here. tvariable_namesN( tpoptNoneR t get_variablestupdateRRt mat_streamtclose(RtmdictRRR tMRt matfile_dict((s2/tmp/pip-build-7oUkmx/scipy/scipy/io/matlab/mio.pyR Fs@ t5trowc C`st|tƒ}|rJ|r8|ddkr8|d}nt|dƒ}n$t|dƒshtdƒ‚n|}|dkr¡|rtdƒ‚nt||ƒ} n?|dkrÔt|d |d td |d |ƒ} n td ƒ‚| j |ƒ|r|j ƒndS(sZ Save a dictionary of names and arrays into a MATLAB-style .mat file. This saves the array objects in the given dictionary to a MATLAB- style .mat file. Parameters ---------- file_name : str or file-like object Name of the .mat file (.mat extension not needed if ``appendmat == True``). Can also pass open file_like object. mdict : dict Dictionary from which to save matfile variables. appendmat : bool, optional True (the default) to append the .mat extension to the end of the given filename, if not already present. format : {'5', '4'}, string, optional '5' (the default) for MATLAB 5 and up (to 7.2), '4' for MATLAB 4 .mat files. long_field_names : bool, optional False (the default) - maximum field name length in a structure is 31 characters which is the documented maximum length. True - maximum field name length in a structure is 63 characters which works for MATLAB 7.6+. do_compression : bool, optional Whether or not to compress matrices on write. Default is False. oned_as : {'row', 'column'}, optional If 'column', write 1-D numpy arrays as column vectors. If 'row', write 1-D numpy arrays as row vectors. See also -------- mio4.MatFile4Writer mio5.MatFile5Writer iüÿÿÿs.mattwbtwrites4Writer needs file name or writeable file-like objectt4s6Long field names are not available for version 4 filesR*tdo_compressiontunicode_stringstlong_field_namestoned_assFormat should be '4' or '5'N( RRRthasattrRt ValueErrorRR tTruet put_variablesR&( RR'RtformatR1R/R2tfile_is_stringt file_streamtMW((s2/tmp/pip-build-7oUkmx/scipy/scipy/io/matlab/mio.pyR ’s,+       cK`s>t||}|jƒ}t|tƒr:|jjƒn|S(sA List variables inside a MATLAB file. Parameters ---------- %(file_arg)s %(append_arg)s %(load_args)s %(struct_arg)s Returns ------- variables : list of tuples A list of tuples, where each tuple holds the matrix name (a string), its shape (tuple of ints), and its data class (a string). Possible data classes are: int8, uint8, int16, uint16, int32, uint32, int64, uint64, single, double, cell, struct, object, char, sparse, function, opaque, logical, unknown. Notes ----- v4 (Level 1.0), v6 and v7 to 7.2 matfiles are supported. You will need an HDF5 python library to read matlab 7.3 format mat files. Because scipy does not supply one, we do not implement the HDF5 / 7.3 interface here. .. versionadded:: 0.12.0 (R tlist_variablesRRR%R&(RRRtMLt variables((s2/tmp/pip-build-7oUkmx/scipy/scipy/io/matlab/mio.pyR Øs  (t__doc__t __future__RRRtnumpytnptscipy._lib.sixRtmiobaseRRtmio4RRtmio5RR t__all__RR5R R"R tFalseR R (((s2/tmp/pip-build-7oUkmx/scipy/scipy/io/matlab/mio.pyts(  K@