3
Pde                 @   s^   d Z ddlZddlZddlmZ ddlmZ ejj	ejj
edZdd Zdd	 Zd
d ZdS )zT Jottings to work out format for __function_workspace__ matrix at end
of mat file.

    N)asstr)MatFile5Readerdatac             C   s   | j   dg i}d}xp| j s| j \}}t|j}|dkrLd| }|d7 }| j|dd}| jj| |||< |jr|d j	| qW |S )N__globals__r    zvar_%d   F)process)
Zinitialize_readZend_of_streamZread_var_headerr   nameZread_var_array
mat_streamseek	is_globalappend)rdrmdictihdrZnext_positionr	   res r   X/var/www/html/virt/lib64/python3.6/site-packages/scipy/io/matlab/tests/test_mio_funcs.pyread_minimat_vars   s    

r   c             C   s   t | d}t|dd}|j }|d }tj|j }|jd ||_|jjd}|dkr\dp^d|_	|jjd	 t
|}|j  |S )
NrbT)Zstruct_as_recordZ__function_workspace__   s   IM<>   )openr   Zget_variablesioBytesIOtobytesr   r
   readZ
byte_orderr   close)fnamefpr   varsfwsZws_bsmir   r   r   r   read_workspace_vars!   s    

r&   c              C   s   t jjtd} t|  d S )Nzparabola.mat)ospathjointest_data_pathr&   )r!   r   r   r   test_jottings2   s    r+   )__doc__os.pathr'   r   Znumpy.compatr   Zscipy.io.matlab.mio5r   r(   r)   dirname__file__r*   r   r&   r+   r   r   r   r   <module>   s   