o
    EbP                     @   sT  d Z ddlZddlZddlZddlZddlmZm	Z	m
Z
mZmZmZmZ ddlmZmZ ddlmZ g dZejdkZdZdZd	Zd
ZdZdZededededededdg dddiZeeeeeeeeeeedZdZdZ d	Z!ddddddZ"ede de!diZ#G d d! d!Z$G d"d# d#Z%G d$d% d%eZ&d-d'd(Z'G d)d* d*Z(G d+d, d,Z)dS ).z1 Classes for read / write of matlab (TM) 4 files
    N   )MatFileReader	docfillermatdims
read_dtypeconvert_dtypesarr_to_charsarr_dtype_number)squeeze_elementchars_to_strings)reduce)MatFile4ReaderMatFile4WriterSYS_LITTLE_ENDIAN
VarHeader4
VarReader4
VarWriter4	arr_to_2dmclass_infomdtypes_templatemiDOUBLEmiINT16miINT32miSINGLEmiUINT16miUINT8mxCHAR_CLASSmxFULL_CLASSmxSPARSE_CLASSnp_to_mtypesorder_codeslittle            f8f4i4i2u2u1header))moptr(   )mrowsr(   )ncolsr(   )imagfr(   )namlenr(   U1)r&   Zc32Zc24Zc16r'   Zc8r(   r)   r*   r+   S1<>zVAX D-floatzVAX G-floatZCray)r   r   r"   r#   r$   Zdoublecharsparsec                   @   s   e Zd ZdZdZdd ZdS )r   Fc                 C   s"   || _ || _|| _|| _|| _d S N)namedtypemclassdims
is_complex)selfr9   r:   r;   r<   r=    r?   7/usr/lib/python3/dist-packages/scipy/io/matlab/_mio4.py__init__Y   s
   
zVarHeader4.__init__N)__name__
__module____qualname__Z
is_logicalZ	is_globalrA   r?   r?   r?   r@   r   T   s    r   c                   @   sT   e Zd ZdZdd Zdd ZdddZdd	d
Zdd Zdd Z	dd Z
dd ZdS )r   z" Class to read matlab 4 variables c                 C   s*   || _ |j| _|j| _|j| _|j| _d S r8   )file_reader
mat_streamdtypeschars_as_strings
squeeze_me)r>   rE   r?   r?   r@   rA   i   s
   zVarReader4.__init__c                 C   s   t | j| jd }| jt|d d}|d dk s"|d dkr&tdt|d d\}}|d	vr=t	d
t
|  t t|d\}}|dkrLtdt|d\}}|}|d |d f}|d dk}	| j| }
t||
|||	S )z% Read and return header for variable r,   r1       r-   r     z.Mat 4 mopt wrong format, byteswapping problem?  )r   r   zBWe do not support byte ordering '%s'; returned data may be corruptd   z,O in MOPT integer should be 0, wrong format?
   r.   r/   r0   r   )r   rF   rG   readintstrip
ValueErrordivmodwarningswarnr    UserWarningr   )r>   datar9   MrestOPTr<   r=   r:   r?   r?   r@   read_headerp   s4   
zVarReader4.read_headerTc                 C   st   |j }|tkr| |}n"|tkr | |}|r| jrt|}n|tkr)| |S t	d| |r8| j
r8t|S |S )NNo reader for class code %s)r;   r   read_full_arrayr   read_char_arrayrH   r   r   read_sparse_array	TypeErrorrI   r
   )r>   hdrprocessr;   arrr?   r?   r@   array_from_header   s   



zVarReader4.array_from_headerc           	      C   sp   |j }|j}|j}|D ]}||9 }q| jt|}t||kr'td|j t	j
|||dd}|r6| }|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:   bufferorder)r:   r<   itemsizerF   rO   rP   lenrR   r9   npndarraycopy)	r>   rc   ro   dtr<   Z	num_bytesdri   re   r?   r?   r@   read_sub_array   s$   
zVarReader4.read_sub_arrayc                 C   s8   |j r| j|dd}| j|dd}||d  S | |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)ro   y              ?)r=   rr   )r>   rc   resZres_jr?   r?   r@   r_      s
   
zVarReader4.read_full_arrayc                 C   sB   |  |tj}| d}tj|jtdt	|d
 S )z latin-1 text matrix (char matrix) reader

        Parameters
        ----------
        hdr : ``VarHeader4`` instance

        Returns
        -------
        arr : ndarray
            with dtype 'U1', shape given by `hdr` ``dims``
        latin-1r2   rh   r:   ri   )rr   astyperm   Zuint8tobytesdecodern   r<   r:   Zarrayro   )r>   rc   re   Sr?   r?   r@   r`      s   zVarReader4.read_char_arrayc                 C   s   |  |}|ddddf }t|d t|d f}tj|dddf dd}tj|dddf dd}|d8 }|d8 }|jd d	krRtj|ddd
f dd}ntj|ddd
f dd}|ddd	f |_tj|||ff|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.
        N)rz   r   )rz   r   r   Zintcr:   r   r#   r"   floatcomplex)	rr   rP   rm   Zascontiguousarrayrh   imagscipyr7   Z
coo_matrix)r>   rc   rs   tmpr<   IJVr?   r?   r@   ra      s   
zVarReader4.read_sparse_arrayc                 C   s.  |j }|tkrttt|j}ny|tkr&ttt|j}| jr%|dd }nc|tkr|j	}|j}t
|dkrB|d dkrB|d dksDdS | j|j|d d  d tjd|| j|jd}| j|j|d d  d tjd|| j|jd}t|t|f}ntd| | jrtd	d
 |D }|S )zwRead the shape of the array described by the header.
        The file position after this call is unspecified.
        Nrz   r"   r   r   r?   ru   r^   c                 S   s   g | ]}|d kr|qS )r   r?   ).0xr?   r?   r@   
<listcomp>0  s    z0VarReader4.shape_from_header.<locals>.<listcomp>)r;   r   tuplemaprP   r<   r   rH   r   r:   rl   rF   seekrk   rm   rn   rO   rb   rI   )r>   rc   r;   rh   rp   r<   ZrowsZcolsr?   r?   r@   shape_from_header  s4   $zVarReader4.shape_from_headerNT)rB   rC   rD   __doc__rA   r]   rf   rr   r_   r`   ra   r   r?   r?   r?   r@   r   f   s    

$*r   c                       sX   e Zd ZdZe fddZdd Zdd Zdd	 ZdddZ	dddZ
dd Z  ZS )r   z Reader for Mat4 files c                    s$   t  j|g|R i | d| _dS )zR Initialize matlab 4 file reader

    %(matstream_arg)s
    %(load_args)s
        N)superrA   _matrix_reader)r>   rF   argskwargs	__class__r?   r@   rA   6  s   
zMatFile4Reader.__init__c                 C   s^   | j d t| j td}| j d |dkrdS |dk s#|dkr)tr'dp(dS tr-dp.dS )Nr   r(   r4   rK   r5   )rF   r   r   rm   r:   r   )r>   r-   r?   r?   r@   guess_byte_order@  s   zMatFile4Reader.guess_byte_orderc                 C   s   t t| j| _t| | _dS )za Run when beginning read of variables

        Sets up readers from parameters in `self`
        N)r   r   Z
byte_orderrG   r   r   )r>   r?   r?   r@   initialize_readL  s   zMatFile4Reader.initialize_readc                 C   sV   | j  }tdd |jd}|jj| }|jr |jtks |d9 }| j	
 | }||fS )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
        c                 S   s   | | S r8   r?   )r   yr?   r?   r@   <lambda>d  s    z0MatFile4Reader.read_var_header.<locals>.<lambda>r   r"   )r   r]   r   r<   r:   rk   r=   r;   r   rF   tell)r>   rc   nZremaining_bytesnext_positionr?   r?   r@   read_var_headerT  s   
zMatFile4Reader.read_var_headerTc                 C   s   | j ||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`.
        )r   rf   )r>   r,   rd   r?   r?   r@   read_var_arrayk  s   zMatFile4Reader.read_var_arrayNc                 C   s   t |tr	|g}n|durt|}| jd |   i }|  sf|  \}}|jdu r.dn|j	d}|durC||vrC| j| q| 
|||< | j| |durb|| t|dkrb	 |S |  r!|S )a,   get variables from stream as dictionary

        Parameters
        ----------
        variable_names : None or str or sequence of str, optional
            variable name, or sequence of variable names to get from Mat file /
            file stream. If None, then get all variables in file.
        Nr   Nonelatin1)
isinstancestrlistrF   r   r   end_of_streamr   r9   rx   r   removerl   )r>   Zvariable_namesmdictrc   r   r9   r?   r?   r@   get_variables}  s,   
	
zMatFile4Reader.get_variablesc                 C   s   | j d |   g }|  sB|  \}}|jdu rdn|jd}| j|}t	
|jd}||||f | j | |  r|S )z list variables from stream r   Nr   r   unknown)rF   r   r   r   r   r9   rx   r   r   r   getr;   append)r>   varsrc   r   r9   rh   infor?   r?   r@   list_variables  s   zMatFile4Reader.list_variablesr   r8   )rB   rC   rD   r   r   rA   r   r   r   r   r   r   __classcell__r?   r?   r   r@   r   4  s    	

r   rowc                 C   s(   t | |}t|dkrtd| |S )a   Make ``arr`` exactly two dimensional

    If `arr` has more than 2 dimensions, raise a ValueError

    Parameters
    ----------
    arr : array
    oned_as : {'row', 'column'}, optional
       Whether to reshape 1-D vectors as row vectors or column vectors.
       See documentation for ``matdims`` for more detail

    Returns
    -------
    arr2d : array
       2-D version of the array
    r"   z=Matlab 4 files cannot save arrays with more than 2 dimensions)r   rl   rR   Zreshape)re   oned_asr<   r?   r?   r@   r     s   

r   c                   @   sT   e Zd Zdd Zdd Zdd Zeedfdd	Zd
d Z	dd Z
dd Zdd ZdS )r   c                 C   s   |j | _ |j| _d S r8   )file_streamr   )r>   Zfile_writerr?   r?   r@   rA     s   zVarWriter4.__init__c                 C   s   | j |jdd d S )Nrg   )rj   )r   writerw   )r>   re   r?   r?   r@   write_bytes  s   zVarWriter4.write_bytesc                 C   s   | j | d S r8   )r   r   )r>   sr?   r?   r@   write_string  s   zVarWriter4.write_stringr   c           
      C   s   t dtd }t }d}|d |d  |d  | |d< |d |d< |d	 |d
< ||d< t|d	 |d< | | |d }	| |	d dS )a(   Write header for given data options

        Parameters
        ----------
        name : str
            name of variable
        shape : sequence
           Shape of array as it will be read in matlab
        P : int, optional
            code for mat4 data type, one of ``miDOUBLE, miSINGLE, miINT32,
            miINT16, miUINT16, miUINT8``
        T : int, optional
            code for mat4 matrix class, one of ``mxFULL_CLASS, mxCHAR_CLASS,
            mxSPARSE_CLASS``
        imagf : int, optional
            flag indicating complex
        r?   r,   r   rL   rM   rN   r-   r.   r   r/   r0   r1    r   N)rm   emptyr   r   rl   r   r   encode)
r>   r9   rh   r[   r\   r0   r,   rX   rZ   rW   r?   r?   r@   write_header  s"   
zVarWriter4.write_headerc                 C   s   t j|r| || dS t|}|j}|js!||	d}|j
}|tju r-td|tju r6td|tjtjfv rF| || dS | || dS )z Write matrix `arr`, with name `name`

        Parameters
        ----------
        arr : array_like
           array to write
        name : str
           name in matlab workspace
        N=z!Cannot save object arrays in Mat4zCannot save void type arrays)r   r7   Zissparsewrite_sparserm   Zasarrayr:   Zisnativerv   ZnewbyteordertypeZobject_rb   ZvoidZunicode_Zstring_
write_charwrite_numeric)r>   re   r9   rp   Zdttr?   r?   r@   r     s    


zVarWriter4.writec                 C   s   t || j}|jjdk}zt|jjdd   }W n ty0   |r'|d}n|d}t}Y nw | j	||j
|t|d |rL| |j | |j d S | | d S )Ncr   Zc128r&   )r[   r\   r0   )r   r   r:   kindr   r   KeyErrorrv   r   r   rh   r   r   realr~   )r>   re   r9   r0   r[   r?   r?   r@   r     s(   
zVarWriter4.write_numericc                 C   s   t |}t|| j}|j}| j||ttd |jjdkr;t	
|}t	jdt|||d}| d}t	j|d|d}| | d S )Nr[   r\   Ur?   ru   rt   r3   )r   r   r   rh   r   r   r   r:   r   rm   Zprodrn   r	   itemr   r   )r>   re   r9   r<   Zn_charsZst_arrstr?   r?   r@   r   $  s$   
zVarWriter4.write_charc                 C   s   |  }|jjdk}tj|jd d| fdd}|j|dddf< |j|dddf< |dddd	f  d7  < |rO|jj	|ddd	f< |jj
|dddf< n	|j|ddd	f< |j|ddd	f< | j||jttd
 | | dS )zY Sparse matrices are 2-D

        See docstring for VarReader4.read_sparse_array
        r   r   r#   r&   r{   Nrz   r   r"   r   )Ztocoor:   r   rm   ZzerosZnnzr   colrW   r   r~   rh   r   r   r   r   )r>   re   r9   Ar0   Zijvr?   r?   r@   r   7  s$   zVarWriter4.write_sparseN)rB   rC   rD   rA   r   r   r   r   r   r   r   r   r   r?   r?   r?   r@   r     s    !r   c                   @   s$   e Zd ZdZdddZdddZdS )r   z) Class for writing matlab 4 format files Nc                 C   s"   || _ |d u r	d}|| _d | _d S )Nr   )r   r   _matrix_writer)r>   r   r   r?   r?   r@   rA   R  s
   
zMatFile4Writer.__init__c                 C   s.   t | | _| D ]\}}| j|| q	dS )a   Write variables in `mdict` to stream

        Parameters
        ----------
        mdict : mapping
           mapping with method ``items`` return name, contents pairs
           where ``name`` which will appeak in the matlab workspace in
           file load, and ``contents`` is something writeable to a
           matlab file, such as a NumPy array.
        write_header : {None, True, False}
           If True, then write the matlab file header before writing the
           variables. If None (the default) then write the file header
           if we are at position 0 in the stream. By setting False
           here, and setting the stream position to the end of the file,
           you can append variables to a matlab file
        N)r   r   itemsr   )r>   r   r   r9   varr?   r?   r@   put_variablesY  s   
zMatFile4Writer.put_variablesr8   )rB   rC   rD   r   rA   r   r?   r?   r?   r@   r   P  s    
r   )r   )*r   sysrT   Znumpyrm   Zscipy.sparser   Z_miobaser   r   r   r   r   r   r	   Z
_mio_utilsr
   r   	functoolsr   __all__	byteorderr   r   r   r   r   r   r   r   r   r   r   r   r    r   r   r   r   r   r   r   r?   r?   r?   r@   <module>   st    $
		 O
y 