Incremental GSC parsing (exact copy of the tutorials shared as a separate project) / tqdm / _tqdm_gui.pyc
1576 views
9�W�4 � @ s� d Z d d l m Z m Z d d l Z d d l m Z d d l m Z d d l m Z d d d
g i Z
d d g Z Gd
d � d e � Z d d � Z
d S)z�
GUI progressbar decorator for iterators.
Includes a default (x)range iterator printing to stderr.
Usage:
>>> from tqdm_gui import tgrange[, tqdm_gui]
>>> for i in tgrange(10): #same as: for i in tqdm_gui(xrange(10))
... ...
� )�division�absolute_importN)�time� )�_range)�tqdmz github.com/Z casperdclZlrq3000�tqdm_gui�tgrangec sj e Z d Z d Z e e j d d d � � Z � f d d � Z d d � Z d d
d � Z
d d
� Z � S)r z+
Experimental GUI version of tqdm!
�
c C s | j | � | j | � d S)zH
Print a message via tqdm_gui (just an alias for print)
N)�write)�cls�s�file�end� r �-lib/python3.5/site-packages/tqdm/_tqdm_gui.pyr s
ztqdm_gui.writec s� d d l } d d l j } d d l m } d | d <t t | � j | | � | j sb | d rf d S| j j
d � | | _ | | _ d | _
| j j d | _ d | j j d <t | j d � | _ | j d d$ � \ | _ } | j r g | _ g | _ g | _ n- | g � | _ | g � | _ | g � | _ | j | j | j d d
�\ | _ | j | j | j d d �\ | _ | j d d � | j r| j d d � | j d � | j j | j | j f d% d d �| j d d d d d d d d �| _ n: | j d d � | j! � | j d � | j d& d d �| j" � | j# | j$ ra| j$ n d d � | j% r�| j& d d d d! d" d' � | j' j( � j) d( � | j* � | _+ | j, � | | _- d S))Nr )�dequeTZguiz#Warning: GUI is experimental/alpha
�toolbar�Noneg �?Zfigsize� 皙����@�color�b�kg����MbP?�d Zpercent�cur�est�locz center right�xmin�xmax�g�< Zsecondsz
lower left�itz/sZstyleZsciZaxis�yZ scilimitsg333333�?)r r )r r )r r )r r g333333ÿ).Z
matplotlibZmatplotlib.pyplotZpyplot� collectionsr �superr �__init__�disable�fpr �mpl�pltZsp�rcParamsr �max� minintervalZsubplots�fig�total�xdata�ydata�zdataZplot�line1�line2�set_ylimZset_xlimZ
set_xlabelZlegend�axhspan�hspanZ invert_xaxisZgridZ
set_ylabel�unit�
unit_scaleZticklabel_formatZyaxisZget_offset_textZset_xZ
isinteractive�wasionZion�ax)�self�args�kwargsr) r* r r; )� __class__r r r& ' sZ
$$
#
ztqdm_gui.__init__c " # s~ | j } | j r, x | D] } | Vq Wd S| j } | j } | j } | j } | j } | j } | j } | j }
| j
} | j } | j }
| j
} | j } | j } | j } | j } | j } | j } | j } | j } | j } xr| D]j} | V|
d 7}
|
| } | | k r� t � � � | } | | k r� � |
} | r�| r�| d k rf| | n | | | d | | } | j } | | } |
| } | j | r�|
d | n � � | j | � | j | � | r | d k r | j � | j � | j � | j � \ } } | | k s3| | k r]d | } | j | | � | j j j � | rL| j | | � | j | | � y | j! j" � } WnH t# k
r�| j$ d d d d d d d d
�| _! | j! j" � } Yn X| | d <| | d <|
| | g | d <| d | g | d <t% | � d
k r9| | d <| j! j& | � n9 � f d d � | D� }! | j |! | � | j |! | � | j' | j( |
| | d | j) | | | | r�d | n d | �
d d d d �| j* d � | rN| r | | k r | | | } nC | r8| r8| | | | d | | } n | | d | | } |
} � } q� W| | _ |
| _ | j+ � d S)Nr g Y@�B g�������?r g����MbP?r r r r � � � c s g | ] } � | � q Sr r )�.0�i)�cur_tr r �
<listcomp>� s z%tqdm_gui.__iter__.<locals>.<listcomp>�fontnamezDejaVu Sans Mono�fontsize� g��&� .>)r r )r r )rA r )rC r ),�iterabler' r- � maxinterval�miniters�dynamic_minitersr8 r9 �ascii�start_t� last_print_t� last_print_n�n� smoothing�avg_time�
bar_formatr* r; r0 r1 r2 r3 r4 r r/ �append�popleft�get_ylimr5 �figure�canvas�draw�set_datar7 �get_xy�AttributeErrorr6 �len�set_xy� set_title� format_meter�desc�pause�close)"r< rK �objr- rL rM rN r8 r9 rO rP rQ rR rS rT rU rV r* r; r0 r1 r2 r3 r4 �delta_it�delta_t�elapsedr/ r# �z�ymin�ymax� poly_lims�t_agor )rF r �__iter__e s�
!
!
ztqdm_gui.__iter__r c
sD | j r
d S| d k r d } | j | 7_ | j | j } | | j k r@t � � � | j } | | j k r@� | j } | j r� | r� | j d k r� | | n | j | | d | j | j | _ | j
} | j } | | } | j | } | j j
| r| j d | n � � | j j
| � | j j
| � | rw| d k rw| j j � | j j � | j j � | j � \ } }
| |
k s�| |
k r�d | }
| j | |
� | j j j � | r�| j j | j | j � | j j | j | j � y | j j � } WnK t k
rc| j j d d d d d d d d
�| _ | j j � } Yn X| | d <|
| d <| j | |
g | d <| d | g | d <t | � d
k r�| | d <| j j | � nH � f d d � | j D� } | j j | | j � | j j | | j � | j | j! | j | | d | j" | j# | j$ | j% | j rcd | j n d | j& �
d d d d �| j j' d � | j( r+| j) r�| | j) k r�| j | j) | | _ na | j r | r | j | | j | d | j | j | _ n" | j | d | j | j | _ | j | _ � | _ d S)Nr r g Y@r@ g�������?g����MbP?r r r r rA rB rC c s g | ] } � | � q Sr r )rD rE )rF r r rG , s z#tqdm_gui.update.<locals>.<listcomp>rH zDejaVu Sans MonorI rJ g��&� .>)r r )r r )rA r )rC r )*r' rS rR rM r rQ r- rP rT rU r/ r; r0 rW r1 r2 rX rY r5 rZ r[ r\ r3 r] r4 r7 r^ r_ r* r6 r` ra rb rc rd rO r8 r9 rV re rN rL )
r< rS rh ri rj r/ r; r# rk rl rm rn ro r )rF r �update� s�
ztqdm_gui.updatec C so | j r
d Sd | _ | j j | � | j | j j d <| j sO | j j � | j sk | j j
| j � d S)NTr ) r' Z
_instances�remover r) r+ r: r* ZioffZleaverf r. )r< r r r rf M s
ztqdm_gui.close) �__name__�
__module__� __qualname__�__doc__� classmethod�sys�stdoutr r& rp rq rf r r )r? r r s > �dc O s t t | � | � S)zl
A shortcut for tqdm_gui(xrange(*args), **kwargs).
On Python3+ range is used instead of xrange.
)r r )r= r> r r r r ` s )rv Z
__future__r r rx r Z_utilsr Z_tqdmr �
__author__�__all__r r r r r r �<module> s � H