Aluode/PerceptionLabPortable
0
1o
2 '�Yi � @ s� d Z ddlZddlZddlmZ ddlZddlZddlm Z ddl3m Z ddl Z
ddlmZ ddlmZ ddlmZ G d d4� d5�ZG d d � d e �ZG d
d� d�ZdS )zSome utility functions.� N)�Iterable)�Thread� )�logger)�
_check_option)�6get_configc @ sb e Zd ZdZ ddd�dd�Zdd � Zd7d � Zd d
� Zdd� Zdd� Z dd� Z8dd� Z dS )� ProgressBara� Generate a command-line progressbar.9 10 Parameters11 ----------12 iterable : iterable | int | None13 The iterable to use. Can also be an int for backward compatibility14 (acts like ``max_value``).15 initial_value : int16 Initial value of process, useful when resuming process from a specific17 value, defaults to 0.18 mesg : str19 Message to include at end of progress bar.20 max_total_width : int | str21 Maximum total message width. Can use "auto" (default) to try to set22 a sane value based on the current terminal width.23 max_value : int | None24 The max value. If None, the length of ``iterable`` will be used.25 which_tqdm : str | None26 Which tqdm module to use. Can be "tqdm", "tqdm.notebook", or "off".27 Defaults to ``None``, which uses the value of the MNE_TQDM environment28 variable, or ``"tqdm.auto"`` if that is not set.29 **kwargs : dict30 Additional keyword arguments for tqdm.31 Nr �auto)�32which_tqdmc 33 K s� dd l }|d u r
tdd�}td|d d� ddd� t�d|� �� |d vrPzt|� W n tyD } z
td34t|�� d | � ��d �d } ~ ww t ||�35d d
�d
�}|j }t ddd
ddd�}36|37� � D ]\} } | |vro|�
| | i� q`t|t�r�|| _|d u r�t|�| _n || _nt|�| _d | _|dkr�d }tjddd�� }
|
j| _W d � n1 s�w Y ~
d | _tjtjkp�|dk}|d| j|| j|||d�|��| _ d S )Nr ZMNE_TQDMz tqdm.auto� )�tqdmztqdm.�offZ beginning)�extrazUsing ProgressBar with )r r
zUnknown tqdm backend z, got: �.r Tg����Mb�?g�������?zg{percentage:3.0f}%|{bar}| {desc} : {n_fmt}/{total_fmt} [{elapsed}<{remaining}, {rate_fmt:>11}{postfix}])ZleaveZ minintervalZminitersZ smoothingZ38bar_formatr �wbZ tmp_mne_prog)�prefixr
)�iterable�desc�total�initialZncols�disable� ) r r r r �debug�39__import__� Exception�40ValueError�repr�getattr�split�dict�items�update�41isinstancer r �len� max_value�int�tempfile�NamedTemporaryFile�name� _mmap_fname�_mmap�level�logging�INFO�_tqdm)�selfr �
initial_valueZmesgZmax_total_widthr$ r42 �kwargsr �exc�defaults�key�val�tfr r r �E:\DocsHouse\542 percep lab latest\PerceptionLab\PerceptionLab_Portable\python_embed\Lib\site-packages\mne/utils/progressbar.py�__init__0 sp
43� ������44 4546 �� �zProgressBar.__init__c C s | � || jj � dS )aE Update progressbar with current value of process.47 48 Parameters49 ----------50 cur_value : number51 Current value of process. Should be <= max_value (but this is not52 enforced). The percent of the progressbar will be computed as53 ``(cur_value / max_value) * 100``.54 N)�update_with_increment_valuer. �n)r/ Z cur_valuer r r7 r! r s 55zProgressBar.updatec C s( z | j �|� W dS ty Y dS w )a( Update progressbar with an increment.56 57 Parameters58 ----------59 increment_value : int60 Value of the increment of process. The percent of the progressbar61 will be computed as62 ``(self.cur_value + increment_value / max_value) * 100``.63 N)r. r! � TypeError)r/ Zincrement_valuer r r7 r9 ~ s64 65 �z'ProgressBar.update_with_increment_valuec c s � | j E dH dS )z*Iterate to auto-increment the pbar with 1.N)r. �r/ r r r7 �__iter__� s �zProgressBar.__iter__c C s66 t | |�S )a Make a joblib-friendly index subset updater.67 68 Parameters69 ----------70 idx : ndarray71 List of indices for this subset.72 73 Returns74 -------75 updater : instance of PBSubsetUpdater76 Class with a ``.update(ii)`` method.77 )�_PBSubsetUpdater)r/ �idxr r r7 �subset� s 78
zProgressBar.subsetc C sX t �| j�r t�| j� tj| jtdt| j d�d�| _79| � d� t | �| _
| j
�� | S )Nzw+r )�shaper )�op�isfiler) �os�remove�npZmemmap�bool�maxr$ r* r! �
_UpdateThread�_thread�startr< r r r7 � __enter__� s �808182zProgressBar.__enter__c C sj | � | j�� � | j�� d| j_| j�� d | _t� | j83�r3z t � | j84� W d S t
y2 Y d S w d S )NF)r! r* �sumr. �closerJ �_mne_run�joinrB rC r) rD rE �PermissionError)r/ �type_�value� tracebackr r r7 �__exit__� s 8586 ��zProgressBar.__exit__c C s" t | dd�dur| j�� dS dS )zEnsure output completes.r. N)r r. rN r< r r r7 �__del__� s �zProgressBar.__del__)Nr Nr N) �__name__�87__module__� __qualname__�__doc__r8 r! r9 r= r@ rL rU rV r r r r7 r s" � �B r c s$ e Zd Z� fdd�Zdd� Z� ZS )rI c s t � jdd� d| _|| _d S )NT)�daemon)�superr8 rO �_mne_pb)r/ �pb�� __class__r r7 r8 � s 88z_UpdateThread.__init__c C s2 | j r| j�| jj�� � t�d� | j sd S d S )Ng�?)rO r] r! r* rM �time�sleepr< r r r7 �run� s 89�z_UpdateThread.run)rW rX rY r8 rc �
__classcell__r r r_ r7 rI � s rI c @ s e Zd Zdd� Zdd� ZdS )r> c C s |j | _|| _d S )N)r* �mmapr? )r/ r^ r? r r r7 r8 � s 90z_PBSubsetUpdater.__init__c C s d| j | j|d < d S )NTr )re r? )r/ �iir r r7 r! � s z_PBSubsetUpdater.updateN)rW rX rY r8 r! r r r r7 r> � s r> )rZ r, rD Zos.path�pathrB r& ra �collections.abcr � threadingr �numpyrF Z_loggingr �checkr �configr r rI r> r r r r7 �<module> s .