3
\g @ sb d dl Z d dlZd dlZd dlZd dlZd dlmZ dgZG dd dZG dd dZ dd Z
d d
Z d d Z d
d Z
dd Zdd Zdd Zedkr^d dlZy d dlZW n ek
r Y nX G dd dejZee jdkre jd ZndZyTeeZx"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_keyStatsc @ s e Zd ZddddZdd 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 Zdqd8d9Zd:d; Zd r r r
r r r )r ZoldstatsZnewstatsr r6 r7 r8 r9 r: r; ZnewfuncZ
newcallersfunc2callerZold_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 |trpytj|}W n& tjk
rB |d| 7 }||fS X g }x|D ] }|jt|rN|j| qNW nzt|}t |t rd| kodk n rt
|| d }|d | }n2t |t
rd| ko|k n r|}|d | }t|t|kr|dt|t||f 7 }||fS )Nz#
g g ?g ?r z6 List reduced from %r to %r due to restriction <%r>
) r% r& recompileerrorsearchr5 rb r floatr` )r ZsellistmsgZnew_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| jk rd}x(|D ] }tt
||krtt
|}qW |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_listwidthZ 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| tr| \}
} } }
|
| krd|
| 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|jdd| jd tt|d| jd |dkrxtdd| jd ntt|| d| jd tt|d| jd |dkrtd 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 leftrightindex directionlrr 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* pathbasename) 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 dr<|jdrz{%s}r z %s:%d(%s))r r r )
startswithendswith)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|trhtdd t||| D ||< q|| |7 < q,|||< q,W |S )Nc S s g | ]}|d |d qS )r r r ).0ir 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 Zd6ddZdd 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 )7ProfileBrowserNc C s6 t jj| d| _d | _tj| _|d k r2| j| d S )Nz% ) cmdCmdr 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 | jrt| 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 |ryt || _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 fddt 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 ZcmdloopKeyboardInterruptr r r r sL