o
6��fpY � @ sx d dl Z d dlZd dlZd dlZg d�ZG dd� d�Zddd�Zddd �ZG d
d � d �Zd d
� Z e
dkr:e � dS dS )� N)�run�runctx�Profilec @ s. e Zd Z dd� Zdd� Zdd� Zdd� Zd S )
�_Utilsc C s
|| _ d S �N)�profiler)�selfr � r �./opt/alt/python310/lib64/python3.10/profile.py�__init__0 �
z_Utils.__init__c C sd | � � }z%z|�|� W n ty Y n w W | �|||� d S W | �|||� d S | �|||� w r )r r �
SystemExit�_show)r � statement�filename�sort�profr r r
r 3 s ���"z
_Utils.runc C sh | � � }z'z |�|||� W n ty Y n w W | �|||� d S W | �|||� d S | �|||� w r )r r r
r )r r �globals�localsr r r r r r
r < s ���"z
_Utils.runctxc C s$ |d ur |� |� d S |�|� 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 Z dZd3dd�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}|| _|s$tj | _| _| j | _
n5|| _| �� }zt |�}W n t yA || _| j | _
Y nw |dkrK| j
| _
n| j| _
|tfdd�}|| _| �� | _| �d� d S )N� � c S s
|| � �S r r )�timer�sumr r r
�get_time_timer� r 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 � s2
�
zProfile.__init__c C s� | j }|� }|d |d | j | j }|dkr |j| _| j| | ||�r3|� }|d |d | _d S |� }|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 �V | j }|� | j | j }|dkr|j| _| j| | ||�r#|� | _d S |� | | _d S �Nr1 r2 �r r4 r5 r6 r r. r r r
r( � � zProfile.trace_dispatch_ic C sb | j }|� d | j | j }|dkr|j| _| j| | ||�r'|� d | _d S |� d | | _d S )Ng N@r1 r2 r: r r r
�trace_dispatch_mac� s zProfile.trace_dispatch_macc C r8 r9 )r' r. r$ r r# r3 )r r4 r5 r6 r' r. r r r
r- � r; zProfile.trace_dispatch_lc C sD | j \}}}}}}||ur|r| �||�S ||| ||||f| _ dS �Nr0 )r! �trace_dispatch_return) r r4 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 rR| |
\} }
}}}| |
d |||f| |
< dS ddddi f| |
< dS �N���r r0 ) r! �f_back�
isinstancer �
fake_framer> �f_code� co_filename�co_firstlineno�co_namer )r r4 r. r? r@ rA rB rC rD Zfcode�fnr �cc�ns�tt�ct�callersr r r
�trace_dispatch_call s �zProfile.trace_dispatch_callc
C sp dd| j f}|dd||| jf| _| j}||v r-|| \}}}}} ||d ||| f||< dS ddddi f||< dS )Nr r r0 )r# r! r )
r r4 r. rO r rP rQ rR rS rT 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}|| \}}}}}|sL|| }|d }| |v rY|| d || < nd|| < ||d || ||f||< dS rF )r! r> r )r r4 r. r? r@ rA rB rD Z frame_totalZpptZpitZpetZpfn�pframeZpcurr rP rQ rR rS rT r r r
r> " s" z Profile.trace_dispatch_return)�callZ exception�returnr1 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 )rL �co_linerN rM )r r �line�namer r r
r ^ s
zProfile.fake_code.__init__c C s t | j| j| jf�S r )�reprrL r] rN �r r r r
�__repr__d s zProfile.fake_code.__repr__N)r r r r rb r r r r
� fake_code] s rc c @ s e Zd Zdd� ZdS )zProfile.fake_framec C s || _ || _d S r )rK rH )r �codeZpriorr r r
r h s
zProfile.fake_frame.__init__N)r r r r r r r r
rJ g s rJ c C sF | � dd|�}| jr| jd }nd }| �||�}| jd | |d� d S )NZprofiler rG rX )rc r! rJ r3 )r r_ rd rW r4 r r r
r/ l s zProfile.simulate_callc C sR | j }|� | j }| jd r!| jd | | jd |� d}| jd s|� | | _d S )Nr rY rG r )r' r. r! r3 )r r' r. r r r
�simulate_cmd_completex s
�z Profile.simulate_cmd_completer c C s$ dd l }|�| ��� �|��� d S r\ )�pstatsZStatsZ
strip_dirsZ
sort_statsr )r r rf r r r
r � s
zProfile.print_statsc C sF t |d��}| �� t�| j|� W d � d S 1 s w 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 )re �snapshot_statsra r r r
ri � s zProfile.create_statsc C s^ i | _ | j�� D ]$\}\}}}}}|�� }d}|�� D ]}||7 }q|||||f| j |< qd S r\ )rl r �items�copy�values) r �funcrP rQ rR rS rT ZncZcallcntr r r
ro � s
�zProfile.snapshot_statsc C s dd l }|j}| �|||�S r\ )�__main__�__dict__r )r r" rt �dictr r r
r � s z Profile.runc C s@ | � |� t�| j� zt|||� W t�d � | S t�d � w r )r[ �sys�
setprofiler) �exec)r r" r r r r r
r � s
�zProfile.runctxc O sB | � t|�� t�| j� z
||i |��W t�d � S t�d � w r )r[ r` rw rx r) )r rs �args�kwr r r
�runcall� s
zProfile.runcallc C s< | j tur td��| j}d| _z
| �||�W || _S || _w )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�}||� |� }||� |� }|| }|r&td|� t� } |� }| �dt� t� � |� }|| }
|rCtd|
� d} d} | j�� D ]\\}
}}\}}}}}|d v rd| |7 } | |7 } qL|rqtd
| � td | � | |d kr}td
| ��| | d | }|r�td|� |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
rn � s
�z#Profile._calibrate_inner.<locals>.fz elapsed time without profiling =zf(m)z elapsed time with profiling =g )rn r� z!'CPU seconds' profiler reported =ztotal # calls =r0 z internal error: total calls = %dg @z+mean stopwatch overhead per profile event =) r' �printr r r r r rp �
ValueError)r r r� r' r� rn Zt0�t1Zelapsed_noprofile�pZelapsed_profileZ total_callsZ
reported_timer r^ �funcnamerP rQ rR rS rT Zmeanr r r
r~ � sD
��
zProfile._calibrate_inner)NN)r )r ) r r r r$ r r, r( r< r- rE rU rV r> r3 r[ rc rJ r/ re r r ri ro r r r| r� r~ r r r r
r g sB &
' '�
1 r c
C s� dd l } ddlm} d}||d�}d|_|jdddd d d
� |jd d d
ddd� |jdddddd
� tjdd � sB|�� t�d� |� � \}}|tjd d �< |j
d ur\| j � |j
�|_
t
|�dkr�|jrtdd l}d}|j|d d�}n3|d } tj �d| j �| �� t�| ��}
t|
�� | d�}W d � n1 s�w Y | dd d d�}z
t||d |j
|j� W |S ty� } zd t_t�| j� W Y d } ~ |S d } ~ ww |�� |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 r0 r z(run_module(modname, run_name='__main__'))�
run_module�modnamery rt )�__file__r � __package__�
__cached__) �osZoptparser� Zallow_interspersed_argsZ
add_optionrw �argvZ print_usage�exit�
parse_argsr� �path�abspathr* r� �runpyr� �insert�dirname�io� open_code�compile�readr r �BrokenPipeError�stdout�errno) r� r� r� �parserZoptionsrz r� rd ZglobsZprogname�fp�excr r r
�main) sd
�
�
�
� �����r� rt rZ ) r� rw r% rj �__all__r r r r r� r r r r r
�<module> s"
'
E9
�