1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
#!/usr/bin/python
# emacs: -*- mode: python-mode; py-indent-offset: 4; indent-tabs-mode: nil -*-
# ex: set sts=4 ts=4 sw=4 et:
### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ##
#
# See COPYING file distributed along with the NiBabel package for the
# copyright and license terms.
#
### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ##
""""""
__docformat__ = 'restructuredtext'
import os
import sys
from os import path
if __name__ == '__main__':
usage = (
"""Usage: %s [options] <python module> ...
"""
% sys.argv[0]
)
# default options
convert2kcache = True
displaykcachegrinder = True
printstats = False
pfilename = None
pstatsfilename = None
profilelines = True
profilelevel = 10 # how many most hungry to list in stats
run = True # either to run profiling at all
removed = sys.argv.pop(0)
if not len(sys.argv):
print(usage)
sys.exit(1)
while sys.argv[0].startswith('-'):
if sys.argv[0] in ['-l', '--level']:
profilelevel = int(sys.argv[1])
sys.argv.pop(0)
elif sys.argv[0] in ['-o', '--output-file']:
pfilename = sys.argv[1]
sys.argv.pop(0)
elif sys.argv[0] in ['-O', '--output-statsfile']:
pstatsfilename = sys.argv[1]
sys.argv.pop(0)
elif sys.argv[0] in ['-s', '--stats']:
printstats = True
convert2kcache = False
displaykcachegrinder = False
elif sys.argv[0] in ['-n', '--no-run']:
run = False
elif sys.argv[0] in ['-P', '--no-profilelines']:
profilelines = False
elif sys.argv[0] in ['-K', '--no-kcache']:
convert2kcache = False
displaykcachegrinder = False
else:
print(usage)
sys.exit(1)
sys.argv.pop(0)
cmdname = sys.argv[0]
dirname = path.dirname(cmdname)
(root, ext) = path.splitext(path.basename(cmdname))
sys.path.append(dirname)
# now do profiling
try:
import hotshot
except ImportError:
raise RuntimeError('No hotshot')
if pfilename is None:
pfilename = cmdname + '.prof'
if run:
exec(f'import {root} as runnable')
if not 'main' in runnable.__dict__:
print(f'OOPS: file/module {cmdname} has no function main defined')
sys.exit(1)
prof = hotshot.Profile(pfilename, lineevents=profilelines)
try:
# actually return values are never setup
# since unittest.main sys.exit's
results = prof.runcall(runnable.main)
except SystemExit:
pass
print(f'Saving profile data into {pfilename}')
prof.close()
if printstats or pstatsfilename:
import hotshot.stats
print('Loading profile file to print statistics')
stats = hotshot.stats.load(pfilename)
if printstats:
stats.strip_dirs()
stats.sort_stats('time', 'calls')
stats.print_stats(profilelevel)
if pstatsfilename:
stats.dump_stats(pstatsfilename)
kfilename = pfilename + '.kcache'
if convert2kcache:
cmd = 'hotshot2calltree -o %s %s' % (kfilename, pfilename)
if os.system(cmd):
print('!!! Make sure to install kcachegrind-converters ;-)')
sys.exit(1)
if displaykcachegrinder:
if os.system('kcachegrind %s' % kfilename):
print('!!! Make sure to install kcachegrind ;-)')
sys.exit(1)
else:
print('Go away -- nothing to look here for as a module')
|
