a
    Ad                     @   s&   d dl Z d dlmZ G dd dZdS )    N)sharedc                   @   s4   e Zd Zdd Zdd Zdd Zdd Zd	d
 ZdS )	TotalTQDMc                 C   s
   d | _ d S N)_tqdmself r   A/var/www/html/stable-diffusion-webui/modules/shared_total_tqdm.py__init__   s    zTotalTQDM.__init__c                 C   s&   t j dtjjtjj dtjd| _d S )NzTotal progress   )desctotalpositionfile)tqdmr   state	job_countsampling_stepsprogress_print_outr   r   r   r   r	   reset
   s    zTotalTQDM.resetc                 C   s4   t jjrt jjrd S | jd u r&|   | j  d S r   )r   optsmultiple_tqdmcmd_optsdisable_console_progressbarsr   r   updater   r   r   r	   r      s
    
zTotalTQDM.updatec                 C   s2   t jjrt jjrd S | jd u r&|   || j_d S r   )r   r   r   r   r   r   r   r   )r   Z	new_totalr   r   r	   updateTotal   s
    
zTotalTQDM.updateTotalc                 C   s(   | j d ur$| j   | j   d | _ d S r   )r   refreshcloser   r   r   r	   clear    s    


zTotalTQDM.clearN)__name__
__module____qualname__r
   r   r   r   r   r   r   r   r	   r      s
   r   )r   modulesr   r   r   r   r   r	   <module>   s   