mercurial-scm/hg-stable: mercurial/statprof.py annotate

annotate mercurial/statprof.py @ 51875:454feddab720

brancing: merge stable into default

author	Pierre-Yves David <pierre-yves.david@octobus.net>
date	Fri, 06 Sep 2024 02:12:19 +0200
parents	460e80488cf0 99632adff795
children	f4733654f144

rev	line source
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1 ## statprof.py
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	2 ## Copyright (C) 2012 Bryan O'Sullivan <bos@serpentine.com>
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	3 ## Copyright (C) 2011 Alex Fraser <alex at phatcore dot com>
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	4 ## Copyright (C) 2004,2005 Andy Wingo <wingo at pobox dot com>
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	5 ## Copyright (C) 2001 Rob Browning <rlb at defaultvalue dot org>
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	6
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	7 ## This library is free software; you can redistribute it and/or
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	8 ## modify it under the terms of the GNU Lesser General Public
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	9 ## License as published by the Free Software Foundation; either
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	10 ## version 2.1 of the License, or (at your option) any later version.
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	11 ##
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	12 ## This library is distributed in the hope that it will be useful,
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	13 ## but WITHOUT ANY WARRANTY; without even the implied warranty of
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	14 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	15 ## Lesser General Public License for more details.
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	16 ##
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	17 ## You should have received a copy of the GNU Lesser General Public
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	18 ## License along with this program; if not, contact:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	19 ##
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	20 ## Free Software Foundation Voice: +1-617-542-5942
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	21 ## 59 Temple Place - Suite 330 Fax: +1-617-542-2652
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	22 ## Boston, MA 02111-1307, USA gnu@gnu.org
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	23
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	24 """
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	25 statprof is intended to be a fairly simple statistical profiler for
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	26 python. It was ported directly from a statistical profiler for guile,
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	27 also named statprof, available from guile-lib [0].
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	28
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	29 [0] http://wingolog.org/software/guile-lib/statprof/
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	30
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	31 To start profiling, call statprof.start():
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	32 >>> start()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	33
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	34 Then run whatever it is that you want to profile, for example:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	35 >>> import test.pystone; test.pystone.pystones()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	36
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	37 Then stop the profiling and print out the results:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	38 >>> stop()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	39 >>> display()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	40 % cumulative self
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	41 time seconds seconds name
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	42 26.72 1.40 0.37 pystone.py:79:Proc0
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	43 13.79 0.56 0.19 pystone.py:133:Proc1
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	44 13.79 0.19 0.19 pystone.py:208:Proc8
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	45 10.34 0.16 0.14 pystone.py:229:Func2
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	46 6.90 0.10 0.10 pystone.py:45:__init__
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	47 4.31 0.16 0.06 pystone.py:53:copy
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	48 ...
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	49
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	50 All of the numerical data is statistically approximate. In the
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	51 following column descriptions, and in all of statprof, "time" refers
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	52 to execution time (both user and system), not wall clock time.
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	53
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	54 % time
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	55 The percent of the time spent inside the procedure itself (not
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	56 counting children).
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	57
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	58 cumulative seconds
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	59 The total number of seconds spent in the procedure, including
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	60 children.
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	61
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	62 self seconds
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	63 The total number of seconds spent in the procedure itself (not
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	64 counting children).
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	65
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	66 name
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	67 The name of the procedure.
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	68
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	69 By default statprof keeps the data collected from previous runs. If you
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	70 want to clear the collected data, call reset():
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	71 >>> reset()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	72
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	73 reset() can also be used to change the sampling frequency from the
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	74 default of 1000 Hz. For example, to tell statprof to sample 50 times a
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	75 second:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	76 >>> reset(50)
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	77
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	78 This means that statprof will sample the call stack after every 1/50 of
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	79 a second of user + system time spent running on behalf of the python
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	80 process. When your process is idle (for example, blocking in a read(),
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	81 as is the case at the listener), the clock does not advance. For this
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	82 reason statprof is not currently not suitable for profiling io-bound
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	83 operations.
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	84
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	85 The profiler uses the hash of the code object itself to identify the
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	86 procedures, so it won't confuse different procedures with the same name.
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	87 They will show up as two different rows in the output.
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	88
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	89 Right now the profiler is quite simplistic. I cannot provide
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	90 call-graphs or other higher level information. What you see in the
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	91 table is pretty much all there is. Patches are welcome :-)
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	92
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	93
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	94 Threading
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	95 ---------
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	96
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	97 Because signals only get delivered to the main thread in Python,
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	98 statprof only profiles the main thread. However because the time
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	99 reporting function uses per-process timers, the results can be
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	100 significantly off if other threads' work patterns are not similar to the
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	101 main thread's work patterns.
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	102 """
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	103 # no-check-code
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	104
30270 2ed0b3f9f79e statprof: use absolute_imports Gregory Szorc <gregory.szorc@gmail.com> parents: 30269 diff changeset	105 import collections
2ed0b3f9f79e statprof: use absolute_imports Gregory Szorc <gregory.szorc@gmail.com> parents: 30269 diff changeset	106 import contextlib
2ed0b3f9f79e statprof: use absolute_imports Gregory Szorc <gregory.szorc@gmail.com> parents: 30269 diff changeset	107 import getopt
2ed0b3f9f79e statprof: use absolute_imports Gregory Szorc <gregory.szorc@gmail.com> parents: 30269 diff changeset	108 import inspect
2ed0b3f9f79e statprof: use absolute_imports Gregory Szorc <gregory.szorc@gmail.com> parents: 30269 diff changeset	109 import json
2ed0b3f9f79e statprof: use absolute_imports Gregory Szorc <gregory.szorc@gmail.com> parents: 30269 diff changeset	110 import os
2ed0b3f9f79e statprof: use absolute_imports Gregory Szorc <gregory.szorc@gmail.com> parents: 30269 diff changeset	111 import signal
2ed0b3f9f79e statprof: use absolute_imports Gregory Szorc <gregory.szorc@gmail.com> parents: 30269 diff changeset	112 import sys
2ed0b3f9f79e statprof: use absolute_imports Gregory Szorc <gregory.szorc@gmail.com> parents: 30269 diff changeset	113 import threading
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	114 import time
30270 2ed0b3f9f79e statprof: use absolute_imports Gregory Szorc <gregory.szorc@gmail.com> parents: 30269 diff changeset	115
51858 460e80488cf0 typing: lock in correct changes from pytype 2023.04.11 -> 2023.06.16 Matt Harbison <matt_harbison@yahoo.com> parents: 51196 diff changeset	116 from typing import (
460e80488cf0 typing: lock in correct changes from pytype 2023.04.11 -> 2023.06.16 Matt Harbison <matt_harbison@yahoo.com> parents: 51196 diff changeset	117 List,
460e80488cf0 typing: lock in correct changes from pytype 2023.04.11 -> 2023.06.16 Matt Harbison <matt_harbison@yahoo.com> parents: 51196 diff changeset	118 )
460e80488cf0 typing: lock in correct changes from pytype 2023.04.11 -> 2023.06.16 Matt Harbison <matt_harbison@yahoo.com> parents: 51196 diff changeset	119
43085 eef9a2d67051 py3: manually import pycompat.open into files that need it Gregory Szorc <gregory.szorc@gmail.com> parents: 43077 diff changeset	120 from .pycompat import open
30583 c6ce11f2ee50 py3: make a bytes version of getopt.getopt() Pulkit Goyal <7895pulkit@gmail.com> parents: 30573 diff changeset	121 from . import (
30642 344e68882cd3 py3: replace os.environ with encoding.environ (part 4 of 5) Pulkit Goyal <7895pulkit@gmail.com> parents: 30583 diff changeset	122 encoding,
30583 c6ce11f2ee50 py3: make a bytes version of getopt.getopt() Pulkit Goyal <7895pulkit@gmail.com> parents: 30573 diff changeset	123 pycompat,
c6ce11f2ee50 py3: make a bytes version of getopt.getopt() Pulkit Goyal <7895pulkit@gmail.com> parents: 30573 diff changeset	124 )
c6ce11f2ee50 py3: make a bytes version of getopt.getopt() Pulkit Goyal <7895pulkit@gmail.com> parents: 30573 diff changeset	125
30270 2ed0b3f9f79e statprof: use absolute_imports Gregory Szorc <gregory.szorc@gmail.com> parents: 30269 diff changeset	126 defaultdict = collections.defaultdict
2ed0b3f9f79e statprof: use absolute_imports Gregory Szorc <gregory.szorc@gmail.com> parents: 30269 diff changeset	127 contextmanager = contextlib.contextmanager
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	128
51672 99632adff795 py3: fix type of some elements of __all__ lists Manuel Jacob <me@manueljacob.de> parents: 51196 diff changeset	129 __all__ = ['start', 'stop', 'reset', 'display', 'profile']
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	130
40196 2864f8d3fcd6 py3: use raw strings for stack names Gregory Szorc <gregory.szorc@gmail.com> parents: 40191 diff changeset	131 skips = {
43551 313e3a279828 cleanup: remove pointless r-prefixes on double-quoted strings Augie Fackler <augie@google.com> parents: 43509 diff changeset	132 "util.py:check",
313e3a279828 cleanup: remove pointless r-prefixes on double-quoted strings Augie Fackler <augie@google.com> parents: 43509 diff changeset	133 "extensions.py:closure",
313e3a279828 cleanup: remove pointless r-prefixes on double-quoted strings Augie Fackler <augie@google.com> parents: 43509 diff changeset	134 "color.py:colorcmd",
313e3a279828 cleanup: remove pointless r-prefixes on double-quoted strings Augie Fackler <augie@google.com> parents: 43509 diff changeset	135 "dispatch.py:checkargs",
313e3a279828 cleanup: remove pointless r-prefixes on double-quoted strings Augie Fackler <augie@google.com> parents: 43509 diff changeset	136 "dispatch.py:<lambda>",
313e3a279828 cleanup: remove pointless r-prefixes on double-quoted strings Augie Fackler <augie@google.com> parents: 43509 diff changeset	137 "dispatch.py:_runcatch",
313e3a279828 cleanup: remove pointless r-prefixes on double-quoted strings Augie Fackler <augie@google.com> parents: 43509 diff changeset	138 "dispatch.py:_dispatch",
313e3a279828 cleanup: remove pointless r-prefixes on double-quoted strings Augie Fackler <augie@google.com> parents: 43509 diff changeset	139 "dispatch.py:_runcommand",
313e3a279828 cleanup: remove pointless r-prefixes on double-quoted strings Augie Fackler <augie@google.com> parents: 43509 diff changeset	140 "pager.py:pagecmd",
313e3a279828 cleanup: remove pointless r-prefixes on double-quoted strings Augie Fackler <augie@google.com> parents: 43509 diff changeset	141 "dispatch.py:run",
313e3a279828 cleanup: remove pointless r-prefixes on double-quoted strings Augie Fackler <augie@google.com> parents: 43509 diff changeset	142 "dispatch.py:dispatch",
313e3a279828 cleanup: remove pointless r-prefixes on double-quoted strings Augie Fackler <augie@google.com> parents: 43509 diff changeset	143 "dispatch.py:runcommand",
313e3a279828 cleanup: remove pointless r-prefixes on double-quoted strings Augie Fackler <augie@google.com> parents: 43509 diff changeset	144 "hg.py:<module>",
313e3a279828 cleanup: remove pointless r-prefixes on double-quoted strings Augie Fackler <augie@google.com> parents: 43509 diff changeset	145 "evolve.py:warnobserrors",
32331 bd872f64a8ba cleanup: use set literals Martin von Zweigbergk <martinvonz@google.com> parents: 31091 diff changeset	146 }
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	147
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	148 ###########################################################################
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	149 ## Utils
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	150
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	151
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	152 def clock():
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	153 times = os.times()
38271 af402c6b90db statprof: also gather wall time Boris Feld <boris.feld@octobus.net> parents: 38197 diff changeset	154 return (times[0] + times[1], times[4])
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	155
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	156
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	157 ###########################################################################
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	158 ## Collection data structures
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	159
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	160
49037 642e31cb55f0 py3: use class X: instead of class X(object): Gregory Szorc <gregory.szorc@gmail.com> parents: 49030 diff changeset	161 class ProfileState:
51858 460e80488cf0 typing: lock in correct changes from pytype 2023.04.11 -> 2023.06.16 Matt Harbison <matt_harbison@yahoo.com> parents: 51196 diff changeset	162 samples: List["Sample"]
460e80488cf0 typing: lock in correct changes from pytype 2023.04.11 -> 2023.06.16 Matt Harbison <matt_harbison@yahoo.com> parents: 51196 diff changeset	163
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	164 def __init__(self, frequency=None):
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	165 self.reset(frequency)
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	166 self.track = b'cpu'
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	167
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	168 def reset(self, frequency=None):
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	169 # total so far
38271 af402c6b90db statprof: also gather wall time Boris Feld <boris.feld@octobus.net> parents: 38197 diff changeset	170 self.accumulated_time = (0.0, 0.0)
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	171 # start_time when timer is active
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	172 self.last_start_time = None
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	173 # a float
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	174 if frequency:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	175 self.sample_interval = 1.0 / frequency
50952 18c8c18993f0 pycompat: drop usage of hasattr/getattr/setattr/delatt proxy Pierre-Yves David <pierre-yves.david@octobus.net> parents: 50305 diff changeset	176 elif not hasattr(self, 'sample_interval'):
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	177 # default to 1000 Hz
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	178 self.sample_interval = 1.0 / 1000.0
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	179 else:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	180 # leave the frequency as it was
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	181 pass
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	182 self.remaining_prof_time = None
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	183 # for user start/stop nesting
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	184 self.profile_level = 0
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	185
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	186 self.samples = []
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	187
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	188 def accumulate_time(self, stop_time):
38271 af402c6b90db statprof: also gather wall time Boris Feld <boris.feld@octobus.net> parents: 38197 diff changeset	189 increment = (
af402c6b90db statprof: also gather wall time Boris Feld <boris.feld@octobus.net> parents: 38197 diff changeset	190 stop_time[0] - self.last_start_time[0],
af402c6b90db statprof: also gather wall time Boris Feld <boris.feld@octobus.net> parents: 38197 diff changeset	191 stop_time[1] - self.last_start_time[1],
af402c6b90db statprof: also gather wall time Boris Feld <boris.feld@octobus.net> parents: 38197 diff changeset	192 )
af402c6b90db statprof: also gather wall time Boris Feld <boris.feld@octobus.net> parents: 38197 diff changeset	193 self.accumulated_time = (
af402c6b90db statprof: also gather wall time Boris Feld <boris.feld@octobus.net> parents: 38197 diff changeset	194 self.accumulated_time[0] + increment[0],
af402c6b90db statprof: also gather wall time Boris Feld <boris.feld@octobus.net> parents: 38197 diff changeset	195 self.accumulated_time[1] + increment[1],
af402c6b90db statprof: also gather wall time Boris Feld <boris.feld@octobus.net> parents: 38197 diff changeset	196 )
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	197
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	198 def seconds_per_sample(self):
38272 15a1e37f80bd profiling: introduce a "profiling.time-track" option Boris Feld <boris.feld@octobus.net> parents: 38271 diff changeset	199 return self.accumulated_time[self.timeidx] / len(self.samples)
15a1e37f80bd profiling: introduce a "profiling.time-track" option Boris Feld <boris.feld@octobus.net> parents: 38271 diff changeset	200
15a1e37f80bd profiling: introduce a "profiling.time-track" option Boris Feld <boris.feld@octobus.net> parents: 38271 diff changeset	201 @property
15a1e37f80bd profiling: introduce a "profiling.time-track" option Boris Feld <boris.feld@octobus.net> parents: 38271 diff changeset	202 def timeidx(self):
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	203 if self.track == b'real':
38272 15a1e37f80bd profiling: introduce a "profiling.time-track" option Boris Feld <boris.feld@octobus.net> parents: 38271 diff changeset	204 return 1
15a1e37f80bd profiling: introduce a "profiling.time-track" option Boris Feld <boris.feld@octobus.net> parents: 38271 diff changeset	205 return 0
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	206
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	207
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	208 state = ProfileState()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	209
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	210
49037 642e31cb55f0 py3: use class X: instead of class X(object): Gregory Szorc <gregory.szorc@gmail.com> parents: 49030 diff changeset	211 class CodeSite:
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	212 cache = {}
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	213
43554 9f70512ae2cf cleanup: remove pointless r-prefixes on single-quoted strings Augie Fackler <augie@google.com> parents: 43551 diff changeset	214 __slots__ = ('path', 'lineno', 'function', 'source')
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	215
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	216 def __init__(self, path, lineno, function):
40534 93501a5fd62b statprof: add a couple of asserts to avoid storing unicodes Augie Fackler <augie@google.com> parents: 40385 diff changeset	217 assert isinstance(path, bytes)
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	218 self.path = path
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	219 self.lineno = lineno
40534 93501a5fd62b statprof: add a couple of asserts to avoid storing unicodes Augie Fackler <augie@google.com> parents: 40385 diff changeset	220 assert isinstance(function, bytes)
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	221 self.function = function
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	222 self.source = None
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	223
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	224 def __eq__(self, other):
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	225 try:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	226 return self.lineno == other.lineno and self.path == other.path
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	227 except:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	228 return False
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	229
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	230 def __hash__(self):
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	231 return hash((self.lineno, self.path))
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	232
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	233 @classmethod
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	234 def get(cls, path, lineno, function):
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	235 k = (path, lineno)
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	236 try:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	237 return cls.cache[k]
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	238 except KeyError:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	239 v = cls(path, lineno, function)
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	240 cls.cache[k] = v
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	241 return v
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	242
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	243 def getsource(self, length):
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	244 if self.source is None:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	245 try:
49561 aab3d4c05720 profile: prevent a crash when line number is unknown Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49037 diff changeset	246 lineno = self.lineno - 1 # lineno can be None
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	247 with open(self.path, b'rb') as fp:
42844 c085cb134b9e statprof: use context manager when reading source from file Martin von Zweigbergk <martinvonz@google.com> parents: 42843 diff changeset	248 for i, line in enumerate(fp):
c085cb134b9e statprof: use context manager when reading source from file Martin von Zweigbergk <martinvonz@google.com> parents: 42843 diff changeset	249 if i == lineno:
c085cb134b9e statprof: use context manager when reading source from file Martin von Zweigbergk <martinvonz@google.com> parents: 42843 diff changeset	250 self.source = line.strip()
c085cb134b9e statprof: use context manager when reading source from file Martin von Zweigbergk <martinvonz@google.com> parents: 42843 diff changeset	251 break
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	252 except:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	253 pass
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	254 if self.source is None:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	255 self.source = b''
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	256
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	257 source = self.source
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	258 if len(source) > length:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	259 source = source[: (length - 3)] + b"..."
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	260 return source
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	261
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	262 def filename(self):
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	263 return os.path.basename(self.path)
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	264
40385 cc4586749c8c statprof: fix overflow while skipping boilerplate parts Yuya Nishihara <yuya@tcha.org> parents: 40384 diff changeset	265 def skipname(self):
43554 9f70512ae2cf cleanup: remove pointless r-prefixes on single-quoted strings Augie Fackler <augie@google.com> parents: 43551 diff changeset	266 return '%s:%s' % (self.filename(), self.function)
40385 cc4586749c8c statprof: fix overflow while skipping boilerplate parts Yuya Nishihara <yuya@tcha.org> parents: 40384 diff changeset	267
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	268
49037 642e31cb55f0 py3: use class X: instead of class X(object): Gregory Szorc <gregory.szorc@gmail.com> parents: 49030 diff changeset	269 class Sample:
43554 9f70512ae2cf cleanup: remove pointless r-prefixes on single-quoted strings Augie Fackler <augie@google.com> parents: 43551 diff changeset	270 __slots__ = ('stack', 'time')
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	271
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	272 def __init__(self, stack, time):
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	273 self.stack = stack
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	274 self.time = time
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	275
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	276 @classmethod
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	277 def from_frame(cls, frame, time):
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	278 stack = []
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	279
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	280 while frame:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	281 stack.append(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	282 CodeSite.get(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	283 pycompat.sysbytes(frame.f_code.co_filename),
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	284 frame.f_lineno,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	285 pycompat.sysbytes(frame.f_code.co_name),
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	286 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	287 )
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	288 frame = frame.f_back
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	289
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	290 return Sample(stack, time)
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	291
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	292
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	293 ###########################################################################
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	294 ## SIGPROF handler
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	295
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	296
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	297 def profile_signal_handler(signum, frame):
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	298 if state.profile_level > 0:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	299 now = clock()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	300 state.accumulate_time(now)
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	301
38272 15a1e37f80bd profiling: introduce a "profiling.time-track" option Boris Feld <boris.feld@octobus.net> parents: 38271 diff changeset	302 timestamp = state.accumulated_time[state.timeidx]
15a1e37f80bd profiling: introduce a "profiling.time-track" option Boris Feld <boris.feld@octobus.net> parents: 38271 diff changeset	303 state.samples.append(Sample.from_frame(frame, timestamp))
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	304
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	305 signal.setitimer(signal.ITIMER_PROF, state.sample_interval, 0.0)
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	306 state.last_start_time = now
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	307
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	308
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	309 stopthread = threading.Event()
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	310
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	311
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	312 def samplerthread(tid):
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	313 while not stopthread.is_set():
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	314 now = clock()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	315 state.accumulate_time(now)
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	316
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	317 frame = sys._current_frames()[tid]
38272 15a1e37f80bd profiling: introduce a "profiling.time-track" option Boris Feld <boris.feld@octobus.net> parents: 38271 diff changeset	318
15a1e37f80bd profiling: introduce a "profiling.time-track" option Boris Feld <boris.feld@octobus.net> parents: 38271 diff changeset	319 timestamp = state.accumulated_time[state.timeidx]
15a1e37f80bd profiling: introduce a "profiling.time-track" option Boris Feld <boris.feld@octobus.net> parents: 38271 diff changeset	320 state.samples.append(Sample.from_frame(frame, timestamp))
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	321
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	322 state.last_start_time = now
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	323 time.sleep(state.sample_interval)
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	324
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	325 stopthread.clear()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	326
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	327
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	328 ###########################################################################
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	329 ## Profiling API
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	330
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	331
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	332 def is_active():
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	333 return state.profile_level > 0
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	334
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	335
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	336 lastmechanism = None
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	337
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	338
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	339 def start(mechanism=b'thread', track=b'cpu'):
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	340 '''Install the profiling signal handler, and start profiling.'''
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	341 state.track = track # note: nesting different mode won't work
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	342 state.profile_level += 1
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	343 if state.profile_level == 1:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	344 state.last_start_time = clock()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	345 rpt = state.remaining_prof_time
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	346 state.remaining_prof_time = None
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	347
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	348 global lastmechanism
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	349 lastmechanism = mechanism
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	350
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	351 if mechanism == b'signal':
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	352 signal.signal(signal.SIGPROF, profile_signal_handler)
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	353 signal.setitimer(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	354 signal.ITIMER_PROF, rpt or state.sample_interval, 0.0
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	355 )
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	356 elif mechanism == b'thread':
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	357 frame = inspect.currentframe()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	358 tid = [k for k, f in sys._current_frames().items() if f == frame][0]
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	359 state.thread = threading.Thread(
43509 5c9daf7df2b4 statprof: correctly always pass a str as the thread name Augie Fackler <augie@google.com> parents: 43106 diff changeset	360 target=samplerthread, args=(tid,), name="samplerthread"
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	361 )
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	362 state.thread.start()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	363
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	364
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	365 def stop():
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	366 '''Stop profiling, and uninstall the profiling signal handler.'''
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	367 state.profile_level -= 1
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	368 if state.profile_level == 0:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	369 if lastmechanism == b'signal':
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	370 rpt = signal.setitimer(signal.ITIMER_PROF, 0.0, 0.0)
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	371 signal.signal(signal.SIGPROF, signal.SIG_IGN)
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	372 state.remaining_prof_time = rpt[0]
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	373 elif lastmechanism == b'thread':
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	374 stopthread.set()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	375 state.thread.join()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	376
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	377 state.accumulate_time(clock())
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	378 state.last_start_time = None
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	379 statprofpath = encoding.environ.get(b'STATPROF_DEST')
30269 f42cd5434cc2 statprof: require paths to save or load profile data Gregory Szorc <gregory.szorc@gmail.com> parents: 30268 diff changeset	380 if statprofpath:
f42cd5434cc2 statprof: require paths to save or load profile data Gregory Szorc <gregory.szorc@gmail.com> parents: 30268 diff changeset	381 save_data(statprofpath)
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	382
30312 1e5346313963 statprof: return state from stop() Gregory Szorc <gregory.szorc@gmail.com> parents: 30272 diff changeset	383 return state
1e5346313963 statprof: return state from stop() Gregory Szorc <gregory.szorc@gmail.com> parents: 30272 diff changeset	384
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	385
30269 f42cd5434cc2 statprof: require paths to save or load profile data Gregory Szorc <gregory.szorc@gmail.com> parents: 30268 diff changeset	386 def save_data(path):
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	387 with open(path, b'w+') as file:
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	388 file.write(b"%f %f\n" % state.accumulated_time)
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	389 for sample in state.samples:
40197 113adb1b3f24 py3: use %d in a few places Gregory Szorc <gregory.szorc@gmail.com> parents: 40196 diff changeset	390 time = sample.time
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	391 stack = sample.stack
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	392 sites = [
51196 933551630b0d statprof: handle `lineno == None` in more cases Martin von Zweigbergk <martinvonz@google.com> parents: 50952 diff changeset	393 b'\1'.join([s.path, b'%d' % s.lineno or -1, s.function])
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	394 for s in stack
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	395 ]
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	396 file.write(b"%d\0%s\n" % (time, b'\0'.join(sites)))
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	397
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	398
30269 f42cd5434cc2 statprof: require paths to save or load profile data Gregory Szorc <gregory.szorc@gmail.com> parents: 30268 diff changeset	399 def load_data(path):
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	400 lines = open(path, b'rb').read().splitlines()
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	401
38286 99188a7c8717 statprof: fix save and load Boris Feld <boris.feld@octobus.net> parents: 38272 diff changeset	402 state.accumulated_time = [float(value) for value in lines[0].split()]
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	403 state.samples = []
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	404 for line in lines[1:]:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	405 parts = line.split(b'\0')
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	406 time = float(parts[0])
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	407 rawsites = parts[1:]
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	408 sites = []
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	409 for rawsite in rawsites:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	410 siteparts = rawsite.split(b'\1')
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	411 sites.append(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	412 CodeSite.get(siteparts[0], int(siteparts[1]), siteparts[2])
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	413 )
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	414
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	415 state.samples.append(Sample(sites, time))
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	416
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	417
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	418 def reset(frequency=None):
45957 89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 43554 diff changeset	419 """Clear out the state of the profiler. Do not call while the
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	420 profiler is running.
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	421
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	422 The optional frequency argument specifies the number of samples to
45957 89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 43554 diff changeset	423 collect per second."""
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	424 assert state.profile_level == 0, b"Can't reset() while statprof is running"
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	425 CodeSite.cache.clear()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	426 state.reset(frequency)
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	427
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	428
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	429 @contextmanager
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	430 def profile():
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	431 start()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	432 try:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	433 yield
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	434 finally:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	435 stop()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	436 display()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	437
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	438
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	439 ###########################################################################
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	440 ## Reporting API
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	441
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	442
49037 642e31cb55f0 py3: use class X: instead of class X(object): Gregory Szorc <gregory.szorc@gmail.com> parents: 49030 diff changeset	443 class SiteStats:
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	444 def __init__(self, site):
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	445 self.site = site
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	446 self.selfcount = 0
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	447 self.totalcount = 0
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	448
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	449 def addself(self):
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	450 self.selfcount += 1
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	451
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	452 def addtotal(self):
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	453 self.totalcount += 1
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	454
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	455 def selfpercent(self):
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	456 return self.selfcount / len(state.samples) * 100
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	457
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	458 def totalpercent(self):
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	459 return self.totalcount / len(state.samples) * 100
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	460
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	461 def selfseconds(self):
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	462 return self.selfcount * state.seconds_per_sample()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	463
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	464 def totalseconds(self):
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	465 return self.totalcount * state.seconds_per_sample()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	466
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	467 @classmethod
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	468 def buildstats(cls, samples):
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	469 stats = {}
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	470
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	471 for sample in samples:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	472 for i, site in enumerate(sample.stack):
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	473 sitestat = stats.get(site)
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	474 if not sitestat:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	475 sitestat = SiteStats(site)
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	476 stats[site] = sitestat
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	477
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	478 sitestat.addtotal()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	479
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	480 if i == 0:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	481 sitestat.addself()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	482
49026 2cce2fa5bcf7 py3: replace pycompat.itervalues(x) with x.values() Gregory Szorc <gregory.szorc@gmail.com> parents: 49004 diff changeset	483 return [s for s in stats.values()]
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	484
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	485
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	486 class DisplayFormats:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	487 ByLine = 0
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	488 ByMethod = 1
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	489 AboutMethod = 2
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	490 Hotpath = 3
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	491 FlameGraph = 4
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	492 Json = 5
30949 cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	493 Chrome = 6
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	494
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	495
30272 eea89068a98d statprof: pass data structure to display functions Gregory Szorc <gregory.szorc@gmail.com> parents: 30271 diff changeset	496 def display(fp=None, format=3, data=None, **kwargs):
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	497 '''Print statistics, either to stdout or the given file object.'''
38697 3c569172848d statprof: small if cleanup Boris Feld <boris.feld@octobus.net> parents: 38286 diff changeset	498 if data is None:
3c569172848d statprof: small if cleanup Boris Feld <boris.feld@octobus.net> parents: 38286 diff changeset	499 data = state
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	500
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	501 if fp is None:
48521 d2fff292c265 pytype: stop excluding statprof.py Matt Harbison <matt_harbison@yahoo.com> parents: 48520 diff changeset	502 from .utils import procutil
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	503
48521 d2fff292c265 pytype: stop excluding statprof.py Matt Harbison <matt_harbison@yahoo.com> parents: 48520 diff changeset	504 fp = procutil.stdout
30272 eea89068a98d statprof: pass data structure to display functions Gregory Szorc <gregory.szorc@gmail.com> parents: 30271 diff changeset	505 if len(data.samples) == 0:
40198 9d3034348c4f py3: switch from print(..., file=) to write() Gregory Szorc <gregory.szorc@gmail.com> parents: 40197 diff changeset	506 fp.write(b'No samples recorded.\n')
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	507 return
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	508
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	509 if format == DisplayFormats.ByLine:
30272 eea89068a98d statprof: pass data structure to display functions Gregory Szorc <gregory.szorc@gmail.com> parents: 30271 diff changeset	510 display_by_line(data, fp)
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	511 elif format == DisplayFormats.ByMethod:
30272 eea89068a98d statprof: pass data structure to display functions Gregory Szorc <gregory.szorc@gmail.com> parents: 30271 diff changeset	512 display_by_method(data, fp)
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	513 elif format == DisplayFormats.AboutMethod:
30272 eea89068a98d statprof: pass data structure to display functions Gregory Szorc <gregory.szorc@gmail.com> parents: 30271 diff changeset	514 display_about_method(data, fp, **kwargs)
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	515 elif format == DisplayFormats.Hotpath:
30272 eea89068a98d statprof: pass data structure to display functions Gregory Szorc <gregory.szorc@gmail.com> parents: 30271 diff changeset	516 display_hotpath(data, fp, **kwargs)
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	517 elif format == DisplayFormats.FlameGraph:
30272 eea89068a98d statprof: pass data structure to display functions Gregory Szorc <gregory.szorc@gmail.com> parents: 30271 diff changeset	518 write_to_flame(data, fp, **kwargs)
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	519 elif format == DisplayFormats.Json:
30272 eea89068a98d statprof: pass data structure to display functions Gregory Szorc <gregory.szorc@gmail.com> parents: 30271 diff changeset	520 write_to_json(data, fp)
30949 cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	521 elif format == DisplayFormats.Chrome:
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	522 write_to_chrome(data, fp, **kwargs)
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	523 else:
48520 c1fe758c1530 statprof: convert a few exception byte strings to str Matt Harbison <matt_harbison@yahoo.com> parents: 46033 diff changeset	524 raise Exception("Invalid display format")
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	525
30949 cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	526 if format not in (DisplayFormats.Json, DisplayFormats.Chrome):
40198 9d3034348c4f py3: switch from print(..., file=) to write() Gregory Szorc <gregory.szorc@gmail.com> parents: 40197 diff changeset	527 fp.write(b'---\n')
9d3034348c4f py3: switch from print(..., file=) to write() Gregory Szorc <gregory.szorc@gmail.com> parents: 40197 diff changeset	528 fp.write(b'Sample count: %d\n' % len(data.samples))
9d3034348c4f py3: switch from print(..., file=) to write() Gregory Szorc <gregory.szorc@gmail.com> parents: 40197 diff changeset	529 fp.write(b'Total time: %f seconds (%f wall)\n' % data.accumulated_time)
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	530
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	531
30272 eea89068a98d statprof: pass data structure to display functions Gregory Szorc <gregory.szorc@gmail.com> parents: 30271 diff changeset	532 def display_by_line(data, fp):
45957 89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 43554 diff changeset	533 """Print the profiler data with each sample line represented
89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 43554 diff changeset	534 as one row in a table. Sorted by self-time per line."""
30272 eea89068a98d statprof: pass data structure to display functions Gregory Szorc <gregory.szorc@gmail.com> parents: 30271 diff changeset	535 stats = SiteStats.buildstats(data.samples)
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	536 stats.sort(reverse=True, key=lambda x: x.selfseconds())
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	537
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	538 fp.write(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	539 b'%5.5s %10.10s %7.7s %-8.8s\n'
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	540 % (b'% ', b'cumulative', b'self', b'')
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	541 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	542 fp.write(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	543 b'%5.5s %9.9s %8.8s %-8.8s\n'
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	544 % (b"time", b"seconds", b"seconds", b"name")
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	545 )
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	546
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	547 for stat in stats:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	548 site = stat.site
50305 972f3e5c94b8 statprof: with Python 3.12, lineno is (more) often None Mads Kiilerich <mads@kiilerich.com> parents: 49901 diff changeset	549 sitelabel = b'%s:%d:%s' % (
972f3e5c94b8 statprof: with Python 3.12, lineno is (more) often None Mads Kiilerich <mads@kiilerich.com> parents: 49901 diff changeset	550 site.filename(),
972f3e5c94b8 statprof: with Python 3.12, lineno is (more) often None Mads Kiilerich <mads@kiilerich.com> parents: 49901 diff changeset	551 site.lineno or -1,
972f3e5c94b8 statprof: with Python 3.12, lineno is (more) often None Mads Kiilerich <mads@kiilerich.com> parents: 49901 diff changeset	552 site.function,
972f3e5c94b8 statprof: with Python 3.12, lineno is (more) often None Mads Kiilerich <mads@kiilerich.com> parents: 49901 diff changeset	553 )
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	554 fp.write(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	555 b'%6.2f %9.2f %9.2f %s\n'
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	556 % (
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	557 stat.selfpercent(),
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	558 stat.totalseconds(),
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	559 stat.selfseconds(),
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	560 sitelabel,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	561 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	562 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	563
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	564
30272 eea89068a98d statprof: pass data structure to display functions Gregory Szorc <gregory.szorc@gmail.com> parents: 30271 diff changeset	565 def display_by_method(data, fp):
45957 89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 43554 diff changeset	566 """Print the profiler data with each sample function represented
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	567 as one row in a table. Important lines within that function are
45957 89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 43554 diff changeset	568 output as nested rows. Sorted by self-time per line."""
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	569 fp.write(
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	570 b'%5.5s %10.10s %7.7s %-8.8s\n'
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	571 % (b'% ', b'cumulative', b'self', b'')
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	572 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	573 fp.write(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	574 b'%5.5s %9.9s %8.8s %-8.8s\n'
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	575 % (b"time", b"seconds", b"seconds", b"name")
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	576 )
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	577
30272 eea89068a98d statprof: pass data structure to display functions Gregory Szorc <gregory.szorc@gmail.com> parents: 30271 diff changeset	578 stats = SiteStats.buildstats(data.samples)
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	579
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	580 grouped = defaultdict(list)
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	581 for stat in stats:
40535 8664fdc1cfb3 statprof: clean up unicode/bytes a little Augie Fackler <augie@google.com> parents: 40534 diff changeset	582 grouped[stat.site.filename() + b":" + stat.site.function].append(stat)
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	583
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	584 # compute sums for each function
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	585 functiondata = []
49004 f254fc73d956 global: bulk replace simple pycompat.iteritems(x) with x.items() Gregory Szorc <gregory.szorc@gmail.com> parents: 48966 diff changeset	586 for fname, sitestats in grouped.items():
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	587 total_cum_sec = 0
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	588 total_self_sec = 0
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	589 total_percent = 0
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	590 for stat in sitestats:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	591 total_cum_sec += stat.totalseconds()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	592 total_self_sec += stat.selfseconds()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	593 total_percent += stat.selfpercent()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	594
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	595 functiondata.append(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	596 (fname, total_cum_sec, total_self_sec, total_percent, sitestats)
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	597 )
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	598
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	599 # sort by total self sec
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	600 functiondata.sort(reverse=True, key=lambda x: x[2])
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	601
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	602 for function in functiondata:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	603 if function[3] < 0.05:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	604 continue
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	605 fp.write(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	606 b'%6.2f %9.2f %9.2f %s\n'
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	607 % (
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	608 function[3], # total percent
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	609 function[1], # total cum sec
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	610 function[2], # total self sec
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	611 function[0],
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	612 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	613 ) # file:function
40198 9d3034348c4f py3: switch from print(..., file=) to write() Gregory Szorc <gregory.szorc@gmail.com> parents: 40197 diff changeset	614
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	615 function[4].sort(reverse=True, key=lambda i: i.selfseconds())
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	616 for stat in function[4]:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	617 # only show line numbers for significant locations (>1% time spent)
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	618 if stat.selfpercent() > 1:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	619 source = stat.site.getsource(25)
49030 37537a4d2695 statprof: remove superfluous sys.version_info check Gregory Szorc <gregory.szorc@gmail.com> parents: 49026 diff changeset	620 if not isinstance(source, bytes):
40201 7df42042636d py3: sprinkle statprof.py with utf-8 encoding Gregory Szorc <gregory.szorc@gmail.com> parents: 40199 diff changeset	621 source = pycompat.bytestr(source)
7df42042636d py3: sprinkle statprof.py with utf-8 encoding Gregory Szorc <gregory.szorc@gmail.com> parents: 40199 diff changeset	622
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	623 stattuple = (
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	624 stat.selfpercent(),
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	625 stat.selfseconds(),
50305 972f3e5c94b8 statprof: with Python 3.12, lineno is (more) often None Mads Kiilerich <mads@kiilerich.com> parents: 49901 diff changeset	626 stat.site.lineno or -1,
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	627 source,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	628 )
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	629
40198 9d3034348c4f py3: switch from print(..., file=) to write() Gregory Szorc <gregory.szorc@gmail.com> parents: 40197 diff changeset	630 fp.write(b'%33.0f%% %6.2f line %d: %s\n' % stattuple)
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	631
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	632
30272 eea89068a98d statprof: pass data structure to display functions Gregory Szorc <gregory.szorc@gmail.com> parents: 30271 diff changeset	633 def display_about_method(data, fp, function=None, **kwargs):
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	634 if function is None:
48520 c1fe758c1530 statprof: convert a few exception byte strings to str Matt Harbison <matt_harbison@yahoo.com> parents: 46033 diff changeset	635 raise Exception("Invalid function")
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	636
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	637 filename = None
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	638 if b':' in function:
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	639 filename, function = function.split(b':')
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	640
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	641 relevant_samples = 0
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	642 parents = {}
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	643 children = {}
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	644
30272 eea89068a98d statprof: pass data structure to display functions Gregory Szorc <gregory.szorc@gmail.com> parents: 30271 diff changeset	645 for sample in data.samples:
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	646 for i, site in enumerate(sample.stack):
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	647 if site.function == function and (
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	648 not filename or site.filename() == filename
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	649 ):
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	650 relevant_samples += 1
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	651 if i != len(sample.stack) - 1:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	652 parent = sample.stack[i + 1]
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	653 if parent in parents:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	654 parents[parent] = parents[parent] + 1
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	655 else:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	656 parents[parent] = 1
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	657
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	658 if site in children:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	659 children[site] = children[site] + 1
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	660 else:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	661 children[site] = 1
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	662
49004 f254fc73d956 global: bulk replace simple pycompat.iteritems(x) with x.items() Gregory Szorc <gregory.szorc@gmail.com> parents: 48966 diff changeset	663 parents = [(parent, count) for parent, count in parents.items()]
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	664 parents.sort(reverse=True, key=lambda x: x[1])
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	665 for parent, count in parents:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	666 fp.write(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	667 b'%6.2f%% %s:%s line %s: %s\n'
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	668 % (
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	669 count / relevant_samples * 100,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	670 pycompat.fsencode(parent.filename()),
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	671 pycompat.sysbytes(parent.function),
51196 933551630b0d statprof: handle `lineno == None` in more cases Martin von Zweigbergk <martinvonz@google.com> parents: 50952 diff changeset	672 parent.lineno or -1,
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	673 pycompat.sysbytes(parent.getsource(50)),
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	674 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	675 )
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	676
30272 eea89068a98d statprof: pass data structure to display functions Gregory Szorc <gregory.szorc@gmail.com> parents: 30271 diff changeset	677 stats = SiteStats.buildstats(data.samples)
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	678 stats = [
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	679 s
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	680 for s in stats
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	681 if s.site.function == function
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	682 and (not filename or s.site.filename() == filename)
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	683 ]
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	684
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	685 total_cum_sec = 0
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	686 total_self_sec = 0
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	687 total_self_percent = 0
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	688 total_cum_percent = 0
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	689 for stat in stats:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	690 total_cum_sec += stat.totalseconds()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	691 total_self_sec += stat.selfseconds()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	692 total_self_percent += stat.selfpercent()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	693 total_cum_percent += stat.totalpercent()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	694
40198 9d3034348c4f py3: switch from print(..., file=) to write() Gregory Szorc <gregory.szorc@gmail.com> parents: 40197 diff changeset	695 fp.write(
9d3034348c4f py3: switch from print(..., file=) to write() Gregory Szorc <gregory.szorc@gmail.com> parents: 40197 diff changeset	696 b'\n %s:%s Total: %0.2fs (%0.2f%%) Self: %0.2fs (%0.2f%%)\n\n'
9d3034348c4f py3: switch from print(..., file=) to write() Gregory Szorc <gregory.szorc@gmail.com> parents: 40197 diff changeset	697 % (
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	698 pycompat.sysbytes(filename or b'___'),
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	699 pycompat.sysbytes(function),
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	700 total_cum_sec,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	701 total_cum_percent,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	702 total_self_sec,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	703 total_self_percent,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	704 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	705 )
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	706
49004 f254fc73d956 global: bulk replace simple pycompat.iteritems(x) with x.items() Gregory Szorc <gregory.szorc@gmail.com> parents: 48966 diff changeset	707 children = [(child, count) for child, count in children.items()]
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	708 children.sort(reverse=True, key=lambda x: x[1])
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	709 for child, count in children:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	710 fp.write(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	711 b' %6.2f%% line %s: %s\n'
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	712 % (
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	713 count / relevant_samples * 100,
51196 933551630b0d statprof: handle `lineno == None` in more cases Martin von Zweigbergk <martinvonz@google.com> parents: 50952 diff changeset	714 child.lineno or -1,
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	715 pycompat.sysbytes(child.getsource(50)),
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	716 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	717 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	718
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	719
30272 eea89068a98d statprof: pass data structure to display functions Gregory Szorc <gregory.szorc@gmail.com> parents: 30271 diff changeset	720 def display_hotpath(data, fp, limit=0.05, **kwargs):
49037 642e31cb55f0 py3: use class X: instead of class X(object): Gregory Szorc <gregory.szorc@gmail.com> parents: 49030 diff changeset	721 class HotNode:
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	722 def __init__(self, site):
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	723 self.site = site
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	724 self.count = 0
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	725 self.children = {}
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	726
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	727 def add(self, stack, time):
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	728 self.count += time
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	729 site = stack[0]
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	730 child = self.children.get(site)
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	731 if not child:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	732 child = HotNode(site)
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	733 self.children[site] = child
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	734
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	735 if len(stack) > 1:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	736 i = 1
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	737 # Skip boiler plate parts of the stack
40385 cc4586749c8c statprof: fix overflow while skipping boilerplate parts Yuya Nishihara <yuya@tcha.org> parents: 40384 diff changeset	738 while i < len(stack) and stack[i].skipname() in skips:
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	739 i += 1
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	740 if i < len(stack):
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	741 child.add(stack[i:], time)
46032 068307b638f4 statprof: fix off-by-one-line error in output Kyle Lippincott <spectral@google.com> parents: 45957 diff changeset	742 else:
068307b638f4 statprof: fix off-by-one-line error in output Kyle Lippincott <spectral@google.com> parents: 45957 diff changeset	743 # Normally this is done by the .add() calls
068307b638f4 statprof: fix off-by-one-line error in output Kyle Lippincott <spectral@google.com> parents: 45957 diff changeset	744 child.count += time
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	745
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	746 root = HotNode(None)
30272 eea89068a98d statprof: pass data structure to display functions Gregory Szorc <gregory.szorc@gmail.com> parents: 30271 diff changeset	747 lasttime = data.samples[0].time
eea89068a98d statprof: pass data structure to display functions Gregory Szorc <gregory.szorc@gmail.com> parents: 30271 diff changeset	748 for sample in data.samples:
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	749 root.add(sample.stack[::-1], sample.time - lasttime)
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	750 lasttime = sample.time
43554 9f70512ae2cf cleanup: remove pointless r-prefixes on single-quoted strings Augie Fackler <augie@google.com> parents: 43551 diff changeset	751 showtime = kwargs.get('showtime', True)
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	752
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	753 def _write(node, depth, multiple_siblings):
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	754 site = node.site
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	755 visiblechildren = [
49026 2cce2fa5bcf7 py3: replace pycompat.itervalues(x) with x.values() Gregory Szorc <gregory.szorc@gmail.com> parents: 49004 diff changeset	756 c for c in node.children.values() if c.count >= (limit * root.count)
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	757 ]
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	758 if site:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	759 indent = depth * 2 - 1
46032 068307b638f4 statprof: fix off-by-one-line error in output Kyle Lippincott <spectral@google.com> parents: 45957 diff changeset	760 filename = (site.filename() + b':').ljust(15)
068307b638f4 statprof: fix off-by-one-line error in output Kyle Lippincott <spectral@google.com> parents: 45957 diff changeset	761 function = site.function
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	762
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	763 # lots of string formatting
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	764 listpattern = (
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	765 b''.ljust(indent)
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	766 + (b'\\' if multiple_siblings else b'\|')
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	767 + b' %4.1f%%'
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	768 + (b' %5.2fs' % node.count if showtime else b'')
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	769 + b' %s %s'
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	770 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	771 liststring = listpattern % (
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	772 node.count / root.count * 100,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	773 filename,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	774 function,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	775 )
46033 8b0a3ff5ed12 statprof: separate functions and "line", assume 4 digit line numbers Kyle Lippincott <spectral@google.com> parents: 46032 diff changeset	776 # 4 to account for the word 'line'
8b0a3ff5ed12 statprof: separate functions and "line", assume 4 digit line numbers Kyle Lippincott <spectral@google.com> parents: 46032 diff changeset	777 spacing_len = max(4, 55 - len(liststring))
8b0a3ff5ed12 statprof: separate functions and "line", assume 4 digit line numbers Kyle Lippincott <spectral@google.com> parents: 46032 diff changeset	778 prefix = b''
8b0a3ff5ed12 statprof: separate functions and "line", assume 4 digit line numbers Kyle Lippincott <spectral@google.com> parents: 46032 diff changeset	779 if spacing_len == 4:
8b0a3ff5ed12 statprof: separate functions and "line", assume 4 digit line numbers Kyle Lippincott <spectral@google.com> parents: 46032 diff changeset	780 prefix = b', '
8b0a3ff5ed12 statprof: separate functions and "line", assume 4 digit line numbers Kyle Lippincott <spectral@google.com> parents: 46032 diff changeset	781
8b0a3ff5ed12 statprof: separate functions and "line", assume 4 digit line numbers Kyle Lippincott <spectral@google.com> parents: 46032 diff changeset	782 codepattern = b'%s%s %d: %s%s'
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	783 codestring = codepattern % (
46033 8b0a3ff5ed12 statprof: separate functions and "line", assume 4 digit line numbers Kyle Lippincott <spectral@google.com> parents: 46032 diff changeset	784 prefix,
8b0a3ff5ed12 statprof: separate functions and "line", assume 4 digit line numbers Kyle Lippincott <spectral@google.com> parents: 46032 diff changeset	785 b'line'.rjust(spacing_len),
49561 aab3d4c05720 profile: prevent a crash when line number is unknown Pierre-Yves David <pierre-yves.david@octobus.net> parents: 49037 diff changeset	786 site.lineno if site.lineno is not None else -1,
46033 8b0a3ff5ed12 statprof: separate functions and "line", assume 4 digit line numbers Kyle Lippincott <spectral@google.com> parents: 46032 diff changeset	787 b''.ljust(max(0, 4 - len(str(site.lineno)))),
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	788 site.getsource(30),
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	789 )
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	790
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	791 finalstring = liststring + codestring
49026 2cce2fa5bcf7 py3: replace pycompat.itervalues(x) with x.values() Gregory Szorc <gregory.szorc@gmail.com> parents: 49004 diff changeset	792 childrensamples = sum([c.count for c in node.children.values()])
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	793 # Make frames that performed more than 10% of the operation red
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	794 if node.count - childrensamples > (0.1 * root.count):
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	795 finalstring = b'\033[91m' + finalstring + b'\033[0m'
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	796 # Make frames that didn't actually perform work dark grey
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	797 elif node.count - childrensamples == 0:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	798 finalstring = b'\033[90m' + finalstring + b'\033[0m'
40384 fc4c598dd4a0 statprof: fix indent level of fp.write() (issue6004) Yuya Nishihara <yuya@tcha.org> parents: 40381 diff changeset	799 fp.write(finalstring + b'\n')
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	800
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	801 newdepth = depth
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	802 if len(visiblechildren) > 1 or multiple_siblings:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	803 newdepth += 1
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	804
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	805 visiblechildren.sort(reverse=True, key=lambda x: x.count)
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	806 for child in visiblechildren:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	807 _write(child, newdepth, len(visiblechildren) > 1)
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	808
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	809 if root.count > 0:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	810 _write(root, 0, False)
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	811
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	812
30272 eea89068a98d statprof: pass data structure to display functions Gregory Szorc <gregory.szorc@gmail.com> parents: 30271 diff changeset	813 def write_to_flame(data, fp, scriptpath=None, outputfile=None, **kwargs):
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	814 if scriptpath is None:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	815 scriptpath = encoding.environ[b'HOME'] + b'/flamegraph.pl'
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	816 if not os.path.exists(scriptpath):
40198 9d3034348c4f py3: switch from print(..., file=) to write() Gregory Szorc <gregory.szorc@gmail.com> parents: 40197 diff changeset	817 fp.write(b'error: missing %s\n' % scriptpath)
9d3034348c4f py3: switch from print(..., file=) to write() Gregory Szorc <gregory.szorc@gmail.com> parents: 40197 diff changeset	818 fp.write(b'get it here: https://github.com/brendangregg/FlameGraph\n')
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	819 return
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	820
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	821 lines = {}
30272 eea89068a98d statprof: pass data structure to display functions Gregory Szorc <gregory.szorc@gmail.com> parents: 30271 diff changeset	822 for sample in data.samples:
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	823 sites = [s.function for s in sample.stack]
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	824 sites.reverse()
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	825 line = b';'.join(sites)
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	826 if line in lines:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	827 lines[line] = lines[line] + 1
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	828 else:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	829 lines[line] = 1
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	830
42845 db6d7cbda80b statprof: use context manager for file when writing flame graph Martin von Zweigbergk <martinvonz@google.com> parents: 42844 diff changeset	831 fd, path = pycompat.mkstemp()
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	832
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	833 with open(path, b"w+") as file:
49004 f254fc73d956 global: bulk replace simple pycompat.iteritems(x) with x.items() Gregory Szorc <gregory.szorc@gmail.com> parents: 48966 diff changeset	834 for line, count in lines.items():
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	835 file.write(b"%s %d\n" % (line, count))
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	836
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	837 if outputfile is None:
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	838 outputfile = b'~/flamegraph.svg'
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	839
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	840 os.system(b"perl ~/flamegraph.pl %s > %s" % (path, outputfile))
40198 9d3034348c4f py3: switch from print(..., file=) to write() Gregory Szorc <gregory.szorc@gmail.com> parents: 40197 diff changeset	841 fp.write(b'Written to %s\n' % outputfile)
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	842
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	843
30948 be3a4fde38eb statprof: add a path simplification function Bryan O'Sullivan <bryano@fb.com> parents: 30845 diff changeset	844 _pathcache = {}
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	845
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	846
30948 be3a4fde38eb statprof: add a path simplification function Bryan O'Sullivan <bryano@fb.com> parents: 30845 diff changeset	847 def simplifypath(path):
45957 89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 43554 diff changeset	848 """Attempt to make the path to a Python module easier to read by
30948 be3a4fde38eb statprof: add a path simplification function Bryan O'Sullivan <bryano@fb.com> parents: 30845 diff changeset	849 removing whatever part of the Python search path it was found
45957 89a2afe31e82 formating: upgrade to black 20.8b1 Augie Fackler <raf@durin42.com> parents: 43554 diff changeset	850 on."""
30948 be3a4fde38eb statprof: add a path simplification function Bryan O'Sullivan <bryano@fb.com> parents: 30845 diff changeset	851
be3a4fde38eb statprof: add a path simplification function Bryan O'Sullivan <bryano@fb.com> parents: 30845 diff changeset	852 if path in _pathcache:
be3a4fde38eb statprof: add a path simplification function Bryan O'Sullivan <bryano@fb.com> parents: 30845 diff changeset	853 return _pathcache[path]
42847 cde1c101ab8a py3: make statprof's chrome output work Martin von Zweigbergk <martinvonz@google.com> parents: 42846 diff changeset	854 hgpath = encoding.__file__.rsplit(os.sep, 2)[0]
30948 be3a4fde38eb statprof: add a path simplification function Bryan O'Sullivan <bryano@fb.com> parents: 30845 diff changeset	855 for p in [hgpath] + sys.path:
be3a4fde38eb statprof: add a path simplification function Bryan O'Sullivan <bryano@fb.com> parents: 30845 diff changeset	856 prefix = p + os.sep
be3a4fde38eb statprof: add a path simplification function Bryan O'Sullivan <bryano@fb.com> parents: 30845 diff changeset	857 if path.startswith(prefix):
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	858 path = path[len(prefix) :]
30948 be3a4fde38eb statprof: add a path simplification function Bryan O'Sullivan <bryano@fb.com> parents: 30845 diff changeset	859 break
be3a4fde38eb statprof: add a path simplification function Bryan O'Sullivan <bryano@fb.com> parents: 30845 diff changeset	860 _pathcache[path] = path
be3a4fde38eb statprof: add a path simplification function Bryan O'Sullivan <bryano@fb.com> parents: 30845 diff changeset	861 return path
be3a4fde38eb statprof: add a path simplification function Bryan O'Sullivan <bryano@fb.com> parents: 30845 diff changeset	862
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	863
30272 eea89068a98d statprof: pass data structure to display functions Gregory Szorc <gregory.szorc@gmail.com> parents: 30271 diff changeset	864 def write_to_json(data, fp):
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	865 samples = []
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	866
30272 eea89068a98d statprof: pass data structure to display functions Gregory Szorc <gregory.szorc@gmail.com> parents: 30271 diff changeset	867 for sample in data.samples:
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	868 stack = []
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	869
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	870 for frame in sample.stack:
40535 8664fdc1cfb3 statprof: clean up unicode/bytes a little Augie Fackler <augie@google.com> parents: 40534 diff changeset	871 stack.append(
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	872 (
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	873 pycompat.sysstr(frame.path),
51196 933551630b0d statprof: handle `lineno == None` in more cases Martin von Zweigbergk <martinvonz@google.com> parents: 50952 diff changeset	874 frame.lineno or -1,
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	875 pycompat.sysstr(frame.function),
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	876 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	877 )
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	878
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	879 samples.append((sample.time, stack))
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	880
40191 4b7eb862692e py3: encode json output to bytes and use write() Gregory Szorc <gregory.szorc@gmail.com> parents: 39129 diff changeset	881 data = json.dumps(samples)
4b7eb862692e py3: encode json output to bytes and use write() Gregory Szorc <gregory.szorc@gmail.com> parents: 39129 diff changeset	882 if not isinstance(data, bytes):
4b7eb862692e py3: encode json output to bytes and use write() Gregory Szorc <gregory.szorc@gmail.com> parents: 39129 diff changeset	883 data = data.encode('utf-8')
4b7eb862692e py3: encode json output to bytes and use write() Gregory Szorc <gregory.szorc@gmail.com> parents: 39129 diff changeset	884
4b7eb862692e py3: encode json output to bytes and use write() Gregory Szorc <gregory.szorc@gmail.com> parents: 39129 diff changeset	885 fp.write(data)
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	886
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	887
30949 cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	888 def write_to_chrome(data, fp, minthreshold=0.005, maxthreshold=0.999):
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	889 samples = []
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	890 laststack = collections.deque()
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	891 lastseen = collections.deque()
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	892
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	893 # The Chrome tracing format allows us to use a compact stack
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	894 # representation to save space. It's fiddly but worth it.
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	895 # We maintain a bijection between stack and ID.
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	896 stack2id = {}
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	897 id2stack = [] # will eventually be rendered
30949 cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	898
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	899 def stackid(stack):
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	900 if not stack:
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	901 return
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	902 if stack in stack2id:
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	903 return stack2id[stack]
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	904 parent = stackid(stack[1:])
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	905 myid = len(stack2id)
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	906 stack2id[stack] = myid
43554 9f70512ae2cf cleanup: remove pointless r-prefixes on single-quoted strings Augie Fackler <augie@google.com> parents: 43551 diff changeset	907 id2stack.append(dict(category=stack[0][0], name='%s %s' % stack[0]))
30949 cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	908 if parent is not None:
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	909 id2stack[-1].update(parent=parent)
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	910 return myid
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	911
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	912 # The sampling profiler can sample multiple times without
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	913 # advancing the clock, potentially causing the Chrome trace viewer
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	914 # to render single-pixel columns that we cannot zoom in on. We
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	915 # work around this by pretending that zero-duration samples are a
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	916 # millisecond in length.
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	917
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	918 clamp = 0.001
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	919
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	920 # We provide knobs that by default attempt to filter out stack
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	921 # frames that are too noisy:
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	922 #
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	923 # * A few take almost all execution time. These are usually boring
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	924 # setup functions, giving a stack that is deep but uninformative.
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	925 #
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	926 # * Numerous samples take almost no time, but introduce lots of
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	927 # noisy, oft-deep "spines" into a rendered profile.
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	928
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	929 blacklist = set()
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	930 totaltime = data.samples[-1].time - data.samples[0].time
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	931 minthreshold = totaltime * minthreshold
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	932 maxthreshold = max(totaltime * maxthreshold, clamp)
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	933
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	934 def poplast():
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	935 oldsid = stackid(tuple(laststack))
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	936 oldcat, oldfunc = laststack.popleft()
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	937 oldtime, oldidx = lastseen.popleft()
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	938 duration = sample.time - oldtime
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	939 if minthreshold <= duration <= maxthreshold:
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	940 # ensure no zero-duration events
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	941 sampletime = max(oldtime + clamp, sample.time)
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	942 samples.append(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	943 dict(
43554 9f70512ae2cf cleanup: remove pointless r-prefixes on single-quoted strings Augie Fackler <augie@google.com> parents: 43551 diff changeset	944 ph='E',
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	945 name=oldfunc,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	946 cat=oldcat,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	947 sf=oldsid,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	948 ts=sampletime * 1e6,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	949 pid=0,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	950 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	951 )
30949 cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	952 else:
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	953 blacklist.add(oldidx)
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	954
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	955 # Much fiddling to synthesize correctly(ish) nested begin/end
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	956 # events given only stack snapshots.
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	957
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	958 for sample in data.samples:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	959 stack = tuple(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	960 (
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	961 (
43554 9f70512ae2cf cleanup: remove pointless r-prefixes on single-quoted strings Augie Fackler <augie@google.com> parents: 43551 diff changeset	962 '%s:%d'
51196 933551630b0d statprof: handle `lineno == None` in more cases Martin von Zweigbergk <martinvonz@google.com> parents: 50952 diff changeset	963 % (
933551630b0d statprof: handle `lineno == None` in more cases Martin von Zweigbergk <martinvonz@google.com> parents: 50952 diff changeset	964 simplifypath(pycompat.sysstr(frame.path)),
933551630b0d statprof: handle `lineno == None` in more cases Martin von Zweigbergk <martinvonz@google.com> parents: 50952 diff changeset	965 frame.lineno or -1,
933551630b0d statprof: handle `lineno == None` in more cases Martin von Zweigbergk <martinvonz@google.com> parents: 50952 diff changeset	966 ),
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	967 pycompat.sysstr(frame.function),
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	968 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	969 for frame in sample.stack
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	970 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	971 )
30949 cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	972 qstack = collections.deque(stack)
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	973 if laststack == qstack:
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	974 continue
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	975 while laststack and qstack and laststack[-1] == qstack[-1]:
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	976 laststack.pop()
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	977 qstack.pop()
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	978 while laststack:
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	979 poplast()
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	980 for f in reversed(qstack):
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	981 lastseen.appendleft((sample.time, len(samples)))
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	982 laststack.appendleft(f)
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	983 path, name = f
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	984 sid = stackid(tuple(laststack))
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	985 samples.append(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	986 dict(
43554 9f70512ae2cf cleanup: remove pointless r-prefixes on single-quoted strings Augie Fackler <augie@google.com> parents: 43551 diff changeset	987 ph='B',
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	988 name=name,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	989 cat=path,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	990 ts=sample.time * 1e6,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	991 sf=sid,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	992 pid=0,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	993 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	994 )
30949 cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	995 laststack = collections.deque(stack)
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	996 while laststack:
cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	997 poplast()
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	998 events = [
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	999 sample for idx, sample in enumerate(samples) if idx not in blacklist
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	1000 ]
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	1001 frames = collections.OrderedDict(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	1002 (str(k), v) for (k, v) in enumerate(id2stack)
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	1003 )
42846 cd3b5be5515d py3: for statprof's Chrome output, write json to string, then encode to bytes Martin von Zweigbergk <martinvonz@google.com> parents: 42845 diff changeset	1004 data = json.dumps(dict(traceEvents=events, stackFrames=frames), indent=1)
cd3b5be5515d py3: for statprof's Chrome output, write json to string, then encode to bytes Martin von Zweigbergk <martinvonz@google.com> parents: 42845 diff changeset	1005 if not isinstance(data, bytes):
cd3b5be5515d py3: for statprof's Chrome output, write json to string, then encode to bytes Martin von Zweigbergk <martinvonz@google.com> parents: 42845 diff changeset	1006 data = data.encode('utf-8')
cd3b5be5515d py3: for statprof's Chrome output, write json to string, then encode to bytes Martin von Zweigbergk <martinvonz@google.com> parents: 42845 diff changeset	1007 fp.write(data)
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	1008 fp.write(b'\n')
30949 cb440e7af05d statprof: allow rendering in the Chrome trace viewer format Bryan O'Sullivan <bryano@fb.com> parents: 30948 diff changeset	1009
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	1010
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1011 def printusage():
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	1012 print(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	1013 r"""
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1014 The statprof command line allows you to inspect the last profile's results in
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1015 the following forms:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1016
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1017 usage:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1018 hotpath [-l --limit percent]
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1019 Shows a graph of calls with the percent of time each takes.
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1020 Red calls take over 10%% of the total time themselves.
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1021 lines
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1022 Shows the actual sampled lines.
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1023 functions
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1024 Shows the samples grouped by function.
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1025 function [filename:]functionname
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1026 Shows the callers and callees of a particular function.
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1027 flame [-s --script-path] [-o --output-file path]
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1028 Writes out a flamegraph to output-file (defaults to ~/flamegraph.svg)
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1029 Requires that ~/flamegraph.pl exist.
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	1030 (Specify alternate script path with --script-path.)"""
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	1031 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	1032
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1033
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1034 def main(argv=None):
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1035 if argv is None:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1036 argv = sys.argv
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1037
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1038 if len(argv) == 1:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1039 printusage()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1040 return 0
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1041
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1042 displayargs = {}
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1043
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1044 optstart = 2
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	1045 displayargs[b'function'] = None
43554 9f70512ae2cf cleanup: remove pointless r-prefixes on single-quoted strings Augie Fackler <augie@google.com> parents: 43551 diff changeset	1046 if argv[1] == 'hotpath':
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	1047 displayargs[b'format'] = DisplayFormats.Hotpath
43554 9f70512ae2cf cleanup: remove pointless r-prefixes on single-quoted strings Augie Fackler <augie@google.com> parents: 43551 diff changeset	1048 elif argv[1] == 'lines':
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	1049 displayargs[b'format'] = DisplayFormats.ByLine
43554 9f70512ae2cf cleanup: remove pointless r-prefixes on single-quoted strings Augie Fackler <augie@google.com> parents: 43551 diff changeset	1050 elif argv[1] == 'functions':
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	1051 displayargs[b'format'] = DisplayFormats.ByMethod
43554 9f70512ae2cf cleanup: remove pointless r-prefixes on single-quoted strings Augie Fackler <augie@google.com> parents: 43551 diff changeset	1052 elif argv[1] == 'function':
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	1053 displayargs[b'format'] = DisplayFormats.AboutMethod
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	1054 displayargs[b'function'] = argv[2]
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1055 optstart = 3
43554 9f70512ae2cf cleanup: remove pointless r-prefixes on single-quoted strings Augie Fackler <augie@google.com> parents: 43551 diff changeset	1056 elif argv[1] == 'flame':
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	1057 displayargs[b'format'] = DisplayFormats.FlameGraph
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1058 else:
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1059 printusage()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1060 return 0
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1061
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1062 # process options
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1063 try:
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	1064 opts, args = pycompat.getoptb(
49901 9eb69fa5a783 statprof: don't pass str `sys.argv` to a function expecting bytes Matt Harbison <matt_harbison@yahoo.com> parents: 49561 diff changeset	1065 pycompat.sysargv[optstart:],
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	1066 b"hl:f:o:p:",
687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	1067 [b"help", b"limit=", b"file=", b"output-file=", b"script-path="],
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	1068 )
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1069 except getopt.error as msg:
30271 7428223ed7c2 statprof: use print function Gregory Szorc <gregory.szorc@gmail.com> parents: 30270 diff changeset	1070 print(msg)
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1071 printusage()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1072 return 2
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1073
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	1074 displayargs[b'limit'] = 0.05
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1075 path = None
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1076 for o, value in opts:
43551 313e3a279828 cleanup: remove pointless r-prefixes on double-quoted strings Augie Fackler <augie@google.com> parents: 43509 diff changeset	1077 if o in ("-l", "--limit"):
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	1078 displayargs[b'limit'] = float(value)
43551 313e3a279828 cleanup: remove pointless r-prefixes on double-quoted strings Augie Fackler <augie@google.com> parents: 43509 diff changeset	1079 elif o in ("-f", "--file"):
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1080 path = value
43551 313e3a279828 cleanup: remove pointless r-prefixes on double-quoted strings Augie Fackler <augie@google.com> parents: 43509 diff changeset	1081 elif o in ("-o", "--output-file"):
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	1082 displayargs[b'outputfile'] = value
43551 313e3a279828 cleanup: remove pointless r-prefixes on double-quoted strings Augie Fackler <augie@google.com> parents: 43509 diff changeset	1083 elif o in ("-p", "--script-path"):
43077 687b865b95ad formatting: byteify all mercurial/ and hgext/ string literals Augie Fackler <augie@google.com> parents: 43076 diff changeset	1084 displayargs[b'scriptpath'] = value
43551 313e3a279828 cleanup: remove pointless r-prefixes on double-quoted strings Augie Fackler <augie@google.com> parents: 43509 diff changeset	1085 elif o in ("-h", "help"):
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1086 printusage()
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1087 return 0
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1088 else:
48520 c1fe758c1530 statprof: convert a few exception byte strings to str Matt Harbison <matt_harbison@yahoo.com> parents: 46033 diff changeset	1089 assert False, "unhandled option %s" % o
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1090
30845 262c2be8ea5a statprof: require input file Gregory Szorc <gregory.szorc@gmail.com> parents: 30642 diff changeset	1091 if not path:
43554 9f70512ae2cf cleanup: remove pointless r-prefixes on single-quoted strings Augie Fackler <augie@google.com> parents: 43551 diff changeset	1092 print('must specify --file to load')
30845 262c2be8ea5a statprof: require input file Gregory Szorc <gregory.szorc@gmail.com> parents: 30642 diff changeset	1093 return 1
262c2be8ea5a statprof: require input file Gregory Szorc <gregory.szorc@gmail.com> parents: 30642 diff changeset	1094
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1095 load_data(path=path)
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1096
35379 c5853c9a6545 py3: handle keyword arguments correctly in statprof.py Pulkit Goyal <7895pulkit@gmail.com> parents: 32331 diff changeset	1097 display(**pycompat.strkwargs(displayargs))
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1098
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1099 return 0
b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1100
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 43053 diff changeset	1101
43551 313e3a279828 cleanup: remove pointless r-prefixes on double-quoted strings Augie Fackler <augie@google.com> parents: 43509 diff changeset	1102 if __name__ == "__main__":
30267 b032a7b676c6 statprof: vendor statprof.py Gregory Szorc <gregory.szorc@gmail.com> parents: diff changeset	1103 sys.exit(main())

Mercurial > public > mercurial-scm > hg-stable

annotate mercurial/statprof.py @ 51875:454feddab720