_Mh2 FUdZddlmZddlZddlmZddlmZ gdZ Gdd e Z Gd d e Z Gd d eZdddddddddd ZejeZeed< dZdZdddZeZdZdZd(d!ZGd"d#ZGd$d%Zd&Zd'ZdS))zf Base classes for MATLAB file stream reading. MATLAB is a registered trademark of the Mathworks inc. )FinalN)doccer)_byteordercodes) MatReadErrorMatReadWarning MatWriteErrorceZdZdZdS)rz"Exception indicating a read issue.N__name__ __module__ __qualname____doc__X/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/scipy/io/matlab/_miobase.pyrrs,,,,rrceZdZdZdS)r z#Exception indicating a write issue.Nr rrrr r s----rr ceZdZdZdS)rzWarning class for read issues.Nr rrrrrs((((rrzfile_name : str Name of the mat file (do not need .mat extension if appendmat==True) Can also pass open file-like object.zappendmat : bool, optional True to append the .mat extension to the end of the given filename, if not already present. Default is True.abyte_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.zoned_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.) file_arg append_arg load_args struct_arg matstream_arg long_fieldsdo_compressiononed_asunicode_strings docfillerc|}|D]2}tj|||||<3|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)`` )copynpdtype newbyteorder)dtype_template order_codedtypesks rconvert_dtypesr(sR" " "F AAHVAY''44Z@@q Mrcj|j}tjd|||d}|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. rF)shaper"bufferorder)itemsizer!ndarrayread) mat_streama_dtype num_bytesarrs r read_dtyper5s@$ I *2"&OOI66   C JrT appendmatczddlm}|||5}t|cdddS#1swxYwYdS)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 ---------- file_name : str Name of the mat file (do not need .mat extension if appendmat==True). Can also pass open file-like object. appendmat : bool, optional True to append the .mat extension to the end of the given filename, if not already present. Default is True. 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)_open_file_contextr6N)_mior9_get_matfile_version) file_namer7r9fileobjs rmatfile_versionr>sJ)(((((  I ; ; ;-w#G,,------------------s 044c|d|t}t|tkrt d|dtkrt dtdt jdt j|dd}d|vr|ddS|d |d}|dt|d d k}t||}t|d |z }||f}|d vr|Stdj |)Nrz Mat file appears to be truncatedz&Mat file appears to be corrupt (first z bytes == 0))rAr+r"r,rr|Ir)rrEz%Unknown mat file type, version {}, {}) seekr0 _HDR_N_BYTESlenrcountr!r/uint8int ValueErrorformat)r= hdr_bytes mopt_intstst_strmaj_indmaj_valmin_valrets rr;r;sj LLOOO \**I 9~~ $$=>>>q\))@%1@@@AA A RXimLLLII~~ Qu LLll1ooG LLOOO'!*'((G''"##G'!g+&''G G C& C<CSI J JJrcolumnc|j}|dkrdSt|dkr7|ddkrdS|dkr|dzS|dkrd|zStd |d |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 -------- >>> import numpy as np >>> from scipy.io.matlab._miobase import matdims >>> 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)rrrrrCrV)rrowz 1-D option "z " is strange)r+rIrM)r4rr+s rmatdimsrYs~ IE {{v 5zzQ 8q==6  4<    %< AGAAABB B Lrc$eZdZdZdZdZdZdS) MatVarReaderz; Abstract class defining required interface for var readerscdS)Nr)self file_readers r__init__zMatVarReader.__init__Ps rcdS)z Returns header Nrr]s r read_headerzMatVarReader.read_headerS rcdS)z Reads array given header Nr)r]headers rarray_from_headerzMatVarReader.array_from_headerWrcrN)r r rrr_rbrfrrrr[r[NsGEE           rr[cLeZdZdZe d dZdZdZdZdS) 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 ,||_i|_|s|}ntj|}||_||_|r|n||_||_ ||_ ||_ | |_ | rd|_d|_dSdS)z Initializer for mat file reader mat_stream : file-like object with file API, open for reading %(load_args)s TFN) r1r&guess_byte_orderboc to_numpy_code byte_orderstruct_as_recordset_matlab_compatible squeeze_mechars_as_strings mat_dtype verify_compressed_data_integritysimplify_cells) r]r1rmrrrprqmatlab_compatiblernrsrts rr_zMatFileReader.__init__fs$%  7..00JJ*:66J$ 0  '  & & ( ( ( ((DO$4D !&DN0P-,  *"DO$)D ! ! ! * *rc0d|_d|_d|_dS)z4 Sets options to return arrays as MATLAB loads them TFN)rrrprqras rroz#MatFileReader.set_matlab_compatibles %rctjS)z9 As we do not know what file type we have, assume native )rk native_coderas rrjzMatFileReader.guess_byte_orders rc|jd}|j}|j|dz t |dkS)Nrr)r1r0tellrGrI)r]bcurposs r end_of_streamzMatFileReader.end_of_streamsT O  # #%%'' VAX&&&1vv{r)NFFTFTTF) r r rrrr_rorjr}rrrrhrh\s  !"&#("&26 %#*#*#*Y#*J&&& rrhcntj|jjddt|zS)z3 Return dtype for given number of items per elementNrE)r!r"str)r4nums rarr_dtype_numberrs* 8CIM"1"%C0 1 11rct|j}|sdg}|t|jjddt j|t|d|}|t j d|jkg}t j |s|S| }d|t|<|S)z$ Convert string array to char array rrENrB)r" ) listr+appendrLr"rr!r/rarrayanyr tuple)r4dimsemptiess r arr_to_charsrs  ??D sKKCIM!""%&&''' *4+C33 ! ! !Cbhr33334G 6'?? ((**CCg Jr)rV)rtypingrnumpyr! scipy._librrrrk__all__ Exceptionrr UserWarningrdoc_dictfilldocr__annotations__r(r5r>get_matfile_versionrHr;rYr[rhrrrrrrs< $$$$$$   -----9---.....I...)))))[)))  < 9   > . #  ? 6 Ea1F1F h"6>(++ 5+++*Z.4-1'-'-'-'-'-T& KKK8KKKK\         >>>>>>>>B222 r