U
    ?hp  ã                   @   sR   d Z ddlZddl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)ÚMatFile5ReaderÚdatac                 C   s–   |   ¡  dg i}d}|  ¡ s’|  ¡ \}}|jd kr6dn
|j d¡}|dkrZd| }|d7 }| j|dd	}| j |¡ |||< |jr|d  	|¡ q|S )
NÚ__globals__r   ÚNoneÚlatin1Ú zvar_%dé   F)Úprocess)
Zinitialize_readZend_of_streamZread_var_headerÚnameÚdecodeZread_var_arrayÚ
mat_streamÚseekÚ	is_globalÚappend)ÚrdrÚmdictÚiÚhdrZnext_positionr
   Úres© r   úV/var/www/html/venv/lib/python3.8/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_jottings0   s    r-   )Ú__doc__Úos.pathr)   r   Zscipy.io.matlab._mio5r   r*   r+   ÚdirnameÚ__file__r,   r   r(   r-   r   r   r   r   Ú<module>   s   