Incremental GSC parsing (exact copy of the tutorials shared as a separate project) / tqdm / __pycache__ / _tqdm_gui.cpython-35.pyc
1576 views
9T#X�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/� casperdcl�lrq3000�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 �S/projects/382bb574-6d4d-4998-962a-4389b3ceddc1/vanguard_tutorials/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 )�dequeT�guiz#Warning: GUI is experimental/alpha
�toolbar�Noneg �?�figsize� 皙����@�color�b�kg����MbP?�d �percent�cur�est�locz center right�xmin�xmax�g�< �secondsz
lower left�itz/s�style�sci�axis�y� scilimitsg333333�?)r r )�curr! )r. r! )r r g333333ÿ).�
matplotlib�matplotlib.pyplot�pyplot� collectionsr �superr
�__init__�disable�fpr
�mpl�plt�sp�rcParamsr �max� mininterval�subplots�fig�total�xdata�ydata�zdata�plot�line1�line2�set_ylim�set_xlim�
set_xlabel�legend�axhspan�hspan� invert_xaxis�grid�
set_ylabel�unit�
unit_scale�ticklabel_format�yaxis�get_offset_text�set_x�
isinteractive�wasion�ion�ax)�self�args�kwargsr7 r8 r rX )� __class__r r r4 ' 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 )r^ r )r` r ),�iterabler5 r< � maxinterval�miniters�dynamic_minitersrO rP �ascii�start_t� last_print_t� last_print_n�n� smoothing�avg_time�
bar_formatr8 rX r@ rA rB rD rE r r? �append�popleft�get_ylimrF �figure�canvas�draw�set_datarK �get_xy�AttributeErrorrJ �len�set_xy� set_title� format_meter�desc�pause�close)"rY rh �objr< ri rj rk rO rP rl rm rn ro rp rq rr rs r8 rX r@ rA rB rD rE �delta_it�delta_t�elapsedr? r, �z�ymin�ymax� poly_lims�t_agor )rc 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% r^ r_ r` c s g | ] } � | � q Sr r )ra rb )rc r r rd , s z#tqdm_gui.update.<locals>.<listcomp>re zDejaVu Sans Monorf rg g��&� .>)r r )r r )r^ r )r` r )*r5 rp ro rj r rn r< rm rq rr r? rX r@ rt rA rB ru rv rF rw rx ry rD rz rE rK r{ r| r8 rJ r} r~ r r� r� rl rO rP rs r� rk ri )
rY rp r� r� r� r? rX r, r� r� r� r� r� r )rc 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 ) r5 �
_instances�remover r7 r: rV r8 �ioff�leaver� r> )rY r r r r� M s
ztqdm_gui.close) �__name__�
__module__� __qualname__�__doc__� classmethod�sys�stdoutr
r4 r� r� r� 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 )rZ r[ r r r r ` s )r� �
__future__r r r� r �_utilsr �_tqdmr �
__author__�__all__r
r r r r r �<module> s � H