3
\�g � @ sb d dl Z d dlZd dlZd dlZd dlZd dlmZ dgZG dd� d�ZG dd� d�Z dd� Z
d d
� Z d d � Z d
d� Z
dd� Zdd� Zdd� Zedk�r^d dlZy d dlZW n ek
r� Y nX G dd� dej�Zee j�dkr�e jd ZndZyTee�Zx"e jdd� D ]Zej e� �q W e dej d� ej � e d ej d� W n e k
�r\ Y nX dS ) � N)�
cmp_to_key�Statsc @ s� e Zd Zdd�dd�Zdd� Zdd� Zd d
� Zd d � Zd
d� ZdAdEdIdMdPdSdVdYd\dadedhdldpd �Z d!d"� Z
d#d$� Z d%d&� Z d'd(� Z
d)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Zd5d6� Zdqd8d9�Zd:d;� Zd r r r
r r r )r ZoldstatsZnewstatsr r6 r7 r8 r9 r: r; ZnewfuncZ
newcallers�func2�callerZold_topZnew_topr r r �
strip_dirs� s.
zStats.strip_dirsc
C s| | j r
d S i | _ }xb| jj� D ]T\}\}}}}}||krBi ||< x0|j� D ]$\}} ||krdi ||< | || |< qLW q W d S )N)r r r4 )
r r r6 r7 r8 r9 r: r; rk rl r r r � calc_callees s
zStats.calc_calleesc
C s |}t |t�rpytj|�}W n& tjk
rB |d| 7 }||fS X g }x�|D ] }|jt|��rN|j|� qNW nzt|�}t |t �r�d| ko�dk n r�t
|| d �}|d |� }n2t |t
�r�d| ko�|k n r�|}|d |� }t|�t|�k�r|dt|�t|�|f 7 }||fS )Nz#
g g �?g �?r z6 List reduced from %r to %r due to restriction <%r>
) r% r& �re�compile�error�searchr5 rb r �floatr` )r Zsel�list�msgZnew_listZrexr6 �countr r r �eval_print_amount s,
"" zStats.eval_print_amountc C s� | j }| jr*| jd d � }d| j d }nt| jj� �}d}x |D ]}| j|||�\}}qBW t|�}|spd|fS t|| j d� |t| j�k r�d}x(|D ] }tt
|��|kr�tt
|��}q�W |d |fS )Nz Ordered by: �
z! Random listing order was used
r )r rR ) r r ra rt r �keysrw r r" r r5 )r Zsel_list�widthZ stat_listru Z selectionrv r6 r r r �get_print_list2 s$
zStats.get_print_listc G s� x | j D ]}t|| jd� qW | j r0t| jd� d}x"| jD ]}t|t|�| jd� qc C s t t|�j|�| d| jd� |s2t | jd� d S t|j� �}d}x�|D ]�}t|�}|| } t| t�r�| \}
} } }
|
| kr�d|
| f }n
d|
f }d|jdd t |� �t
| �t
|
�|f }|d
}n$d || t
| j | d �f }|d }t || | | jd� d}qHW d S )
Nr| )r~ r )r r z%d/%dz%dz %s %s %s %srP rR r z %s(%r) %srF ) r" r5 r� r �sortedry r% rf �rjustr �f8r )r r� �sourceZ call_dictZarrowZclistr� r6 rX r� r8 r7 r9 r: ZsubstatsZ
left_widthr r r r� � s*
zStats.print_call_linec C s" t dd| jd� t d| jd� d S )Nz- ncalls tottime percall cumtime percallr| )r~ r zfilename:lineno(function))r )r" r )r r r r r� � s zStats.print_titlec C s� | j | \}}}}}t|�}||kr4|d t|� }t|jd�d| jd� tt|�d| jd� |dkrxtdd| jd� ntt|| �d| jd� tt|�d| jd� |dkr�td d| jd� ntt|| �d| jd� tt|�| jd� d S )
N�/� r| )r~ r r r} )r z z )r r& r" r� r r� r5 )r r6 r7 r8 r9 r: r; �cr r r r� � s zStats.print_liner �r r �r� )r� rE r �r r �r� )r� rE r �rF r �r� )r� rG r �rF r �r� )r� rG �rH r �r� )r� rI �rH r �r� )r� rI �rJ r �r� )r� rK �rH r �r� )r� rI �rL r �r� )r� rM �rL r �rH r �rJ r �r� r� r� )r� rN r �r r �r� )r� rO �rP r �r� )r� rQ r �rR r �r� )r� rS r �rR r �r� )r� rS )r� )�__name__�
__module__� __qualname__r r r r r
rD rZ r^ rg ri rm rn rw r{ r� r� r� r� r� r� r� r r r r r sD !
c @ s e Zd Zdd� Zdd� ZdS )rd c C s
|| _ d S )N)�comp_select_list)r r� r r r r � s zTupleComp.__init__c C sB x<| j D ]2\}}|| }|| }||k r.| S ||kr|S qW dS )Nr )r� )r �left�right�index� direction�l�rr r r re � s zTupleComp.compareN)r� r� r� r re r r r r rd � s rd c C s | \}}}t jj|�||fS )N)r* �path�basename)� func_namerC rV rX r r r rj � s
rj c C s | d S )NrR r )r6 r r r r � s r c C sN | d d� d krB| d }|j d�r<|jd�rz{%s}r z %s:%d(%s))r� r r )�
startswith�endswith)r� rX r r r r5 � s r5 c C s@ |\}}}}}| \}}} }
} || || || ||
t | |�fS )N)� add_callers) �targetr� r7 r8 r9 r: r; Zt_ccZt_ncZt_ttZt_ctZ t_callersr r r r> � s r> c C s� i }x | j � D ]\}}|||< qW xb|j � D ]V\}}||krzt|t�rhtdd� t||| �D ��||< q�|| |7 < q,|||< q,W |S )Nc S s g | ]}|d |d �qS )r r r )�.0�ir r r �
� s z add_callers..)r4 r% rf �zip)r� r� Z new_callersr6 rl r r r r� � s
r� c C s" d}x| j � D ] }||7 }qW |S )Nr )r� )r; r8 rT r r r � count_calls s r� c C s d| S )Nz%8.3fr )�xr r r r� s r� �__main__c @ s� e Zd Zd6dd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Zd4d5� Z dS )7�ProfileBrowserNc C s6 t jj| � d| _d | _tj| _|d k r2| j|� d S )Nz% ) �cmd�Cmdr �promptr r r r �do_read)r �profiler r r r s zProfileBrowser.__init__c C s� |j � }g }x�|D ]�}y|jt|�� wW n tk
r> Y nX y8t|�}|dksZ|dk rjtd| jd� w|j|� wW n tk
r� Y nX |j|� qW | jr�t| j|�|� ntd| jd� dS )Nr r z#Fraction argument must be in [0, 1])r z No statistics object is loaded.) �splitrb r` �
ValueErrorrs r" r r �getattr)r �fnrV r Z processedZtermZfracr r r �generic" s,
zProfileBrowser.genericc C sX t d| jd� t d| jd� t d| jd� t d| jd� t d| jd� t d| jd� d S )NzArguments may be:)r z0* An integer maximum number of entries to print.z:* A decimal fractional number between 0 and 1, controllingz- what fraction of selected entries to print.z8* A regular expression; only entries with function namesz that match it are printed.)r" r )r r r r � generic_help: s zProfileBrowser.generic_helpc C sd | j rRy| j j|� W q` tk
rN } z td||f | jd� W Y d d }~X q`X ntd| jd� dS )Nz$Failed to load statistics for %s: %s)r z No statistics object is loaded.r )r r
�IOErrorr" r )r rV �er r r �do_addB s *zProfileBrowser.do_addc C s t d| jd� d S )Nz>Add profile info from given file to current statistics object.)r )r" r )r r r r �help_addK s zProfileBrowser.help_addc C s | j d|�S )Nr� )r� )r rV r r r �
do_calleesN s zProfileBrowser.do_calleesc C s t d| jd� | j� d S )Nz6Print callees statistics from the current stat object.)r )r" r r� )r r r r � help_calleesP s zProfileBrowser.help_calleesc C s | j d|�S )Nr� )r� )r rV r r r �
do_callersT s zProfileBrowser.do_callersc C s t d| jd� | j� d S )Nz6Print callers statistics from the current stat object.)r )r" r r� )r r r r � help_callersV s zProfileBrowser.help_callersc C s t d| jd� dS )Nr )r r )r" r )r rV r r r �do_EOFZ s zProfileBrowser.do_EOFc C s t d| jd� d S )NzLeave the profile brower.)r )r" r )r r r r �help_EOF] s zProfileBrowser.help_EOFc C s dS )Nr r )r rV r r r �do_quit` s zProfileBrowser.do_quitc C s t d| jd� d S )NzLeave the profile brower.)r )r" r )r r r r � help_quitb s zProfileBrowser.help_quitc C s� |r�yt |�| _W nj tk
rF } zt|jd | jd� d S d }~X n8 tk
r| } z t|jjd || jd� d S d }~X nX |d | _ n6t
| j �dkr�| j d d� }| j |� ntd| jd� dS ) Nr )r �:z% rR z1No statistics object is current -- cannot reload.r �����) r r �OSErrorr" r r r! r0 r� r� r r� )r rV �errr r r r� e s zProfileBrowser.do_readc C s t d| jd� t d| jd� d S )Nz+Read in profile data from a specified file.)r z*Without argument, reload the current file.)r" r )r r r r � help_readv s zProfileBrowser.help_readc C s$ | j r| j j� ntd| jd� dS )Nz No statistics object is loaded.)r r )r ri r" r )r rV r r r �
do_reversez s zProfileBrowser.do_reversec C s t d| jd� d S )Nz/Reverse the sort order of the profiling report.)r )r" r )r r r r � help_reverse� s zProfileBrowser.help_reversec s� | j std| jd� d S | j j� � |rRt� fdd�|j� D ��rR| j j|j� � n@td| jd� x0tjj � D ]"\}}td||d f | jd� qlW dS ) Nz No statistics object is loaded.)r c 3 s | ]}|� kV qd S )Nr )r� r� )�abbrevsr r � � s z)ProfileBrowser.do_sort..z/Valid sort keys (unique prefixes are accepted):z%s -- %sr r )
r r" r r^ �allr� rg r rZ r4 )r rV r_ r� r )r� r �do_sort� s
zProfileBrowser.do_sortc C s t d| jd� t d| jd� d S )Nz.Sort profile data according to specified keys.)r z3(Typing `sort' without arguments lists valid keys.))r" r )r r r r � help_sort� s zProfileBrowser.help_sortc s � fdd�t jD �S )Nc s g | ]}|j � �r|�qS r )r� )r� �a)�textr r r� � s z0ProfileBrowser.complete_sort..)r rZ )r r� r r )r� r �
complete_sort� s z ProfileBrowser.complete_sortc C s | j d|�S )Nr� )r� )r rV r r r �do_stats� s zProfileBrowser.do_statsc C s t d| jd� | j� d S )Nz.Print statistics from the current stat object.)r )r" r r� )r r r r �
help_stats� s zProfileBrowser.help_statsc C s$ | j r| j j� ntd| jd� d S )Nz No statistics object is loaded.)r )r rm r" r )r rV r r r �do_strip� s zProfileBrowser.do_stripc C s t d| jd� d S )Nz r� r� r� r� r� �readline� ImportErrorr� r� r �argvZ initprofileZbrowserr� r� r" r Zcmdloop�KeyboardInterruptr r r r � sL