3
\ 4 � @ s� d dl Z d dlZd dlZd dlZddddgZdZdZdZejZ e
Z d Z d
d � Z
G d d� d�Zd
d
e edfdd�Zd
d
e eedfdd�Zddd�dd�Zedkr�eje� � dS )� N�Timer�timeit�repeat�
default_timerz <timeit-src>i@B � z�
def inner(_it, _timer{init}):
{setup}
_t0 = _timer()
for _i in _it:
{stmt}
_t1 = _timer()
return _t1 - _t0
c C s | j ddd| �S )N�
� )�replace)�src�indent� r � /usr/lib64/python3.6/timeit.py�reindentP s r c @ sL e Zd Zddedfdd�Zd
dd�Zefdd�Zeefd d
�Z dd d �Z
dS )r �passNc C s� || _ i }|d krt� n|}d}t|t�rJt|td� |d }t|d�}n*t|�rl||d< |d7 }d}d}ntd��t|t�r�t|| td� t|d �}n&t|�r�||d
< |d 7 }d }ntd
��t j
|||d�} | | _ t| td�}
t |
||� |d | _
d S )N� �execr � �_setupz, _setup=_setupz_setup()z&setup is neither a string nor callable� Z_stmtz
, _stmt=_stmtz_stmt()z%stmt is neither a string nor callable)�stmt�setup�init�inner)�timer�_globals�
isinstance�str�compile�dummy_src_namer �callable�
ValueError�template�formatr
r r ) �selfr r r �globalsZlocal_nsZ global_nsr Z
stmtprefixr
�coder r r
�__init__f s6
zTimer.__init__c C sJ dd l }dd l}| jd k r:t| j�d | jjd�tf|jt< |j|d� d S )Nr r )�file)� linecache� tracebackr
�len�splitr �cache� print_exc)r# r' r( r) r r r
r- � s
zTimer.print_excc
C sB t jd |�}tj� }tj� z| j|| j�}W d |r<tj� X |S )N)� itertoolsr �gc� isenabled�disabler r �enable)r# �number�itZgcoldZtimingr r r
r � s
z Timer.timeitc C s. g }x$t |�D ]}| j|�}|j|� qW |S )N)�ranger �append)r# r r3 �r�i�tr r r
r � s
z Timer.repeatc C sF x<t dd�D ].}d| }| j|�}|r0|||� |dkr P q W ||fS )N� �
g�������?)r5 r )r# �callbackr8 r3 �
time_takenr r r
� autorange� s
zTimer.autorange)N)N) �__name__�
__module__� __qualname__r r& r- �default_numberr �default_repeatr r> r r r r
r T s "
r c C s t | |||�j|�S )N)r r )r r r r3 r$ r r r
r � s c C s t | |||�j||�S )N)r r )r r r r r3 r$ r r r
r � s )� _wrap_timerc s^ | d krt jdd � } dd l}y(|j| ddddddd d
d d g �\}} W n2 |jk
rx } zt|� td
� dS d }~X nX t}dj| �p�d}d}g }t} d}
d } dddd�} d� � x�|D � ]�\}
}|
d8kr�t|�}|
d9kr�|j |� |
d:k�r|| k� r�|} ntdt j
d � dS |
d;k�r0t|�} | dk�r0d} |
d<k�r@t j }|
d=k�rPt j }|
d>k�r`t j
}|
d?k�r�|
�rx� d7 � |
d7 }
|
d@kr�ttd)d*� dS q�W dj|��p�d}dd l}t jjd|j� |d k �r�||�}t|||�}|dk�r(d }|
�r� fd+d,�}y|j|�\}}W n |j� dS y|j| |�}W n |j� dS t|�}|
�rxtd-d)j� fd.d/�|D ��� td0| d)d*� |d | }| d k �r�| | }n>d1d/� | j� D �}|jd2d3� x |D ]\}} ||k�r�P �q�W td4| � || | f � t|�}|d | }t|�}||d5 k�rZ|d | }dd l}|j d6� || | f t d7d� d S )ANr: r z
n:u:s:r:tcpvhznumber=zsetup=zrepeat=�time�clockZprocess�verbosezunit=�helpz#use -h/--help for command line help� r r g @�@g ��.A)�usecZmsecZsecr �-n�--number�-s�--setup�-u�--unitz4Unrecognized unit. Please select usec, msec, or sec.)r' �-r�--repeat�-t�--time�-c�--clock�-p� --process�-v� --verbose�-h�--helpr )�endc s d}t |j| |� d�� d S )Nz#{num} loops -> {secs:.{prec}g} secs)ZnumZsecsZprec)�printr" )r3 r= �msg)� precisionr r
r< @ s zmain.<locals>.callbackz
raw times:c s g | ]}d � |f �qS )z%.*gr )�.0�x)r` r r
�
<listcomp>O s zmain.<locals>.<listcomp>z %d loops,c S s g | ]\}}||f�qS r r )ra Zunit�scaler r r
rc U s T)�reversez best of %d: %.*g %s per loopr ztThe test results are likely unreliable. The worst
time (%.*g %s) was more than four times slower than the best time.r )rK rL )rM rN )rO rP )rQ rR )rS rT )rU rV )rW rX )rY rZ )r[ r\ ) �sys�argv�getopt�errorr^ r �joinrC �intr6 �stderrrE rF Z process_time�__doc__�os�path�insert�curdirr r> r- r �min�items�sort�max�warnings�
warn_explicit� UserWarning)�argsrD rh Zopts�errr r r3 r r rG Z time_unitZunits�o�arn r9 r< �_r7 ZbestrJ rd ZscalesZworstrv r )r` r
�main� s�
r~ �__main__)N)r/ rf rE r. �__all__r rB rC Z perf_counterr r$ r r! r r r r r~ r? �exitr r r r
�<module>5 s(
y