B
    ²ô`e  ã               @   s^   d Z ddlZddlZddlmZ ddlmZ ej 	e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Š   |   ¡  dg i}d}xp|  ¡ s„|  ¡ \}}t|jƒ}|dkrLd| }|d7 }| j|dd}| j |¡ |||< |jr|d  	|¡ 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   úS/home/dcms/DCMS/lib/python3.7/site-packages/scipy/io/matlab/tests/test_mio_funcs.pyÚread_minimat_vars   s    

r   c             C   s‚   t | dƒ}t|dd}| ¡ }|d }t | ¡ ¡}| d¡ ||_|j d¡}|dkr\dp^d|_	|j d	¡ t
|ƒ}| ¡  |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 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   