a
R�fnY � @ sr d dl Z d dlZd dlZd dlZg d�ZG dd� d�Zddd�Zddd �ZG d
d � d �Zd d
� Z e
dkrne � dS )� N)�run�runctx�Profilec @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�_Utilsc C s
|| _ d S �N)�profiler)�selfr � r �,/opt/alt/python39/lib64/python3.9/profile.py�__init__0 s z_Utils.__init__c C sR | � � }z4z|�|� W n ty* Y n0 W | �|||� n| �|||� 0 d S r )r r �
SystemExit�_show)r � statement�filename�sort�profr r r
r 3 s z
_Utils.runc C sV | � � }z8z|�|||� W n ty. Y n0 W | �|||� n| �|||� 0 d S r )r r r r
)r r �globals�localsr r r r r r
r < s z
_Utils.runctxc C s" |d ur|� |� n
|�|� d S r )�
dump_stats� print_stats)r r r r r r r
r
E s z _Utils._showN)�__name__�
__module__� __qualname__r r r r
r r r r
r * s r ���c C s t t��| ||�S r )r r r )r r r r r r
r Q s r c C s t t��| ||||�S r )r r r )r r r r r r r r
r ^ s r c @ s� e Zd ZdZd3dd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z e
e e e e e d�Z
dd� ZG dd� d�ZG dd� d�Zd d � Zd d � Zd4d!d"�Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd5d/d0�Zd1d2� ZdS )6r r Nc C s� i | _ d | _d| _d| _|d u r&| j}|| _|sHtj | _| _| j | _
nj|| _| �� }z t |�}W n t y� || _| j | _
Y n00 |dkr�| j
| _
n| j| _
|tfdd�}|| _| �� | _| �d� d S )N� � c S s
|| � �S r r )�timer�sumr r r
�get_time_timer� s z(Profile.__init__.<locals>.get_time_timerr )�timings�cur�cmd� c_func_name�bias�time� process_timer �get_time�trace_dispatch_i�
dispatcher�len� TypeError�trace_dispatch�trace_dispatch_lr �t�
simulate_call)r r r# r- Zlengthr r r r
r � s0
zProfile.__init__c C s� | j }|� }|d |d | j | j }|dkr8|j| _| j| | ||�rd|� }|d |d | _n |� }|d |d | | _d S )Nr � �c_call�r r- r# r r" �dispatch)r �frame�event�argr r- �rr r r
r+ � s zProfile.trace_dispatchc C sT | j }|� | j | j }|dkr(|j| _| j| | ||�rD|� | _n |� | | _d S �Nr0 r1 �r r3 r4 r5 r r- r r r
r' � s
zProfile.trace_dispatch_ic C s` | j }|� d | j | j }|dkr,|j| _| j| | ||�rL|� d | _n|� d | | _d S )Ng N@r0 r1 r8 r r r
�trace_dispatch_mac� s zProfile.trace_dispatch_macc C sT | j }|� | j | j }|dkr(|j| _| j| | ||�rD|� | _n |� | | _d S r7 )r& r- r# r r" r2 )r r3 r4 r5 r& r- r r r
r, � s
zProfile.trace_dispatch_lc C sD | j \}}}}}}||ur*|r*| �||�S ||| ||||f| _ dS �Nr/ )r �trace_dispatch_return) r r3 r- �rpt�rit�ret�rfn�rframe�rcurr r r
�trace_dispatch_exception� s
z Profile.trace_dispatch_exceptionc C s� | j r@|j| j d ur@| j \}}}}}}t|tj�s@| �|d� |j} | j| j| j f}
|dd|
|| j f| _ | j
} |
| v r�| |
\} }
}}}| |
d |||f| |
< nddddi f| |
< dS �N���r r/ ) r �f_back�
isinstancer �
fake_framer; �f_code� co_filename�co_firstlineno�co_namer )r r3 r- r< r= r> r? r@ rA Zfcode�fnr �cc�ns�tt�ct�callersr r r
�trace_dispatch_call s zProfile.trace_dispatch_callc
C sn dd| j f}|dd||| jf| _| j}||v rX|| \}}}}} ||d ||| f||< nddddi f||< dS )Nr r r/ )r" r r )
r r3 r- rL r rM rN rO rP rQ r r r
�trace_dispatch_c_call s z Profile.trace_dispatch_c_callc C s� || j d ur | �| j d d� | j \}}}}}}|| }|| }|\} }
} } }
}| |
| | | | |
|f| _ | j}|| \}}}}}|s�|| }|d }| |v r�|| d || < nd|| < ||d || ||f||< dS rC )r r; r )r r3 r- r< r= r> r? rA Z frame_totalZpptZpitZpetZpfn�pframeZpcurr rM rN rO rP rQ r r r
r; " s" z Profile.trace_dispatch_return)�callZ exception�returnr0 Z c_exceptionZc_returnc C s" | j d rd S || _| �|� d S )Nr )r r! r. )r r! r r r
�set_cmdX s zProfile.set_cmdc @ s e Zd Zdd� Zdd� ZdS )zProfile.fake_codec C s || _ || _|| _d| _d S �Nr )rI �co_linerK rJ )r r �line�namer r r
r ^ s zProfile.fake_code.__init__c C s t | j| j| jf�S r )�reprrI rY rK �r r r r
�__repr__d s zProfile.fake_code.__repr__N)r r r r r^ r r r r
� fake_code] s r_ c @ s e Zd Zdd� ZdS )zProfile.fake_framec C s || _ || _d S r )rH rE )r �codeZpriorr r r
r h s zProfile.fake_frame.__init__N)r r r r r r r r
rG g s rG c C sF | � dd|�}| jr | jd }nd }| �||�}| jd | |d� d S )NZprofiler rD rU )r_ r rG r2 )r r[ r` rT r3 r r r
r. l s zProfile.simulate_callc C sJ | j }|� | j }| jd r:| jd | | jd |� d}q|� | | _d S )Nr rV rD r )r& r- r r2 )r r& r- r r r
�simulate_cmd_completex s
z Profile.simulate_cmd_completer c C s$ dd l }|�| ��� �|��� d S rX )�pstatsZStatsZ
strip_dirsZ
sort_statsr )r r rb r r r
r � s zProfile.print_statsc C sD t |d��&}| �� t�| j|� W d � n1 s60 Y d S )N�wb)�open� create_stats�marshal�dump�stats)r �file�fr r r
r � s zProfile.dump_statsc C s | � � | �� d S r )ra �snapshot_statsr] r r r
re � s zProfile.create_statsc C s^ i | _ | j�� D ]H\}\}}}}}|�� }d}|�� D ] }||7 }q6|||||f| j |< qd S rX )rh r �items�copy�values) r �funcrM rN rO rP rQ ZncZcallcntr r r
rk � s
zProfile.snapshot_statsc C s dd l }|j}| �|||�S rX )�__main__�__dict__r )r r! rp �dictr r r
r � s z Profile.runc C sB | � |� t�| j� zt|||� W t�d � n t�d � 0 | S r )rW �sys�
setprofiler( �exec)r r! r r r r r
r � s
zProfile.runctxc O sF | � t|�� t�| j� z||i |��W t�d � S t�d � 0 d S r )rW r\ rs rt r( )r ro �args�kwr r r
�runcall� s
�zProfile.runcallc C s@ | j turtd��| j}d| _z| �||�W || _S || _0 d S )Nz&Subclasses must override .calibrate().r )� __class__r r* r# �_calibrate_inner)r �m�verboseZ
saved_biasr r r
� calibrate� s
�zProfile.calibratec C s | j }dd� }|fdd�}||� |� }||� |� }|| }|rLtd|� t� } |� }| �dt� t� � |� }|| }
|r�td|
� d} d} | j�� D ]0\\}
}}\}}}}}|d v r�| |7 } | |7 } q�|r�td
| � td | � | |d kr�td
| ��| | d | }|�rtd|� |S )Nc S s t | �D ]}d}qd S r: ��range)�n�i�xr r r
�f1� s z$Profile._calibrate_inner.<locals>.f1c S s t | �D ] }|d� qd S )N�d r~ )r{ r� r� r r r
rj � s z#Profile._calibrate_inner.<locals>.fz elapsed time without profiling =zf(m)z elapsed time with profiling =g )rj r� z!'CPU seconds' profiler reported =ztotal # calls =r/ z internal error: total calls = %dg @z+mean stopwatch overhead per profile event =) r& �printr r r r r rl �
ValueError)r r{ r| r& r� rj Zt0�t1Zelapsed_noprofile�pZelapsed_profileZ total_callsZ
reported_timer rZ �funcnamerM rN rO rP rQ Zmeanr r r
rz � sB
�
zProfile._calibrate_inner)NN)r )r ) r r r r# r r+ r' r9 r, rB rR rS r; r2 rW r_ rG r. ra r r re rk r r rx r} rz r r r r
r g s> '
' '�
1
r c
C s� dd l } ddlm} d}||d�}d|_|jdddd d d
� |jd d d
ddd� |jdddddd
� tjdd � s�|�� t�d� |� � \}}|tjd d �< |j
d ur�| j � |j
�|_
t
|�dk�r�|jr�dd l}d}|j|d d�}nh|d } tj �d| j �| �� t�| �� }
t|
�� | d�}W d � n1 �s:0 Y | dd d d�}zt||d |j
|j� W n8 t�y� } z d t_t�| j� W Y d } ~ n
d } ~ 0 0 n|�� |S ) Nr )� OptionParserzMprofile.py [-o output_file_path] [-s sort] [-m module | scriptfile] [arg] ...)�usageFz-oz --outfile�outfilezSave stats to <outfile>)�dest�help�defaultz-m�module�
store_truezProfile a library module.)r� �actionr� r� z-sz--sortr z?Sort order when printing to stdout, based on pstats.Stats classr r/ r z(run_module(modname, run_name='__main__'))�
run_module�modnameru rp )�__file__r � __package__�
__cached__) �osZoptparser� Zallow_interspersed_argsZ
add_optionrs �argvZ print_usage�exit�
parse_argsr� �path�abspathr) r� �runpyr� �insert�dirname�io� open_code�compile�readr r �BrokenPipeError�stdout�errno) r� r� r� �parserZoptionsrv r� r` ZglobsZprogname�fp�excr r r
�main) sZ
�
�
�
� 0�$r� rp )Nr )Nr ) r� rs r$ rf �__all__r r r r r� r r r r r
�<module> s
'
E9