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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
|
from __future__ import print_function
# Copyright (C) 2003 CAMP
# Please see the accompanying LICENSE file for further information.
import sys
import time
import functools
def function_timer(func, *args, **kwargs):
out = kwargs.pop('timeout', sys.stdout)
t1 = time.time()
r = func(*args, **kwargs)
t2 = time.time()
print(t2 - t1, file=out)
return r
class Timer:
"""Timer object.
Use like this::
timer = Timer()
timer.start('description')
# do something
timer.stop()
or::
with timer('description'):
# do something
To get a summary call::
timer.write()
"""
def __init__(self, print_levels=1000):
self.timers = {}
self.t0 = time.time()
self.running = []
self.print_levels = print_levels
def print_info(self, calc):
"""Override to get to write info during calculator's initialize()."""
pass
def start(self, name):
names = tuple(self.running + [name])
self.timers[names] = self.timers.get(names, 0.0) - time.time()
self.running.append(name)
def stop(self, name=None):
if name is None:
name = self.running[-1]
names = tuple(self.running)
running = self.running.pop()
if name != running:
raise RuntimeError('Must stop timers by stack order. '
'Requested stopping of %s but topmost is %s'
% (name, running))
self.timers[names] += time.time()
return names
def __call__(self, name):
"""Context manager for timing a block of code.
Example (t is a timer object)::
with t('Add two numbers'):
x = 2 + 2
# same as this:
t.start('Add two numbers')
x = 2 + 2
t.stop()
"""
self.start(name)
return self
def __enter__(self):
pass
def __exit__(self, *args):
self.stop()
def get_time(self, *names):
return self.timers[names]
def write(self, out=sys.stdout):
were_running = list(self.running)
while self.running:
self.stop()
if len(self.timers) == 0:
return
t0 = time.time()
tot = t0 - self.t0
n = max([len(names[-1]) + len(names) for names in self.timers]) + 1
line = '-' * (n + 26) + '\n'
out.write('%-*s incl. excl.\n' % (n, 'Timing:'))
out.write(line)
tother = tot
inclusive = self.timers.copy()
exclusive = self.timers.copy()
keys = sorted(exclusive.keys())
for names in keys:
t = exclusive[names]
if len(names) > 1:
if len(names) < self.print_levels + 1:
exclusive[names[:-1]] -= t
else:
tother -= t
exclusive[('Other',)] = tother
inclusive[('Other',)] = tother
keys.append(('Other',))
for names in keys:
t = exclusive[names]
tinclusive = inclusive[names]
r = t / tot
p = 100 * r
i = int(40 * r + 0.5)
if i == 0:
bar = '|'
else:
bar = '|%s|' % ('-' * (i - 1))
level = len(names)
if level > self.print_levels:
continue
name = (level - 1) * ' ' + names[-1] + ':'
out.write('%-*s%9.3f %9.3f %5.1f%% %s\n' %
(n, name, tinclusive, t, p, bar))
out.write(line)
out.write('%-*s%9.3f %5.1f%%\n\n' % (n + 10, 'Total:', tot, 100.0))
for name in were_running:
self.start(name)
def add(self, timer):
for name, t in timer.timers.items():
self.timers[name] = self.timers.get(name, 0.0) + t
class timer:
"""Decorator for timing a method call.
Example::
from ase.utils.timing import timer, Timer
class A:
def __init__(self):
self.timer = Timer()
@timer('Add two numbers')
def add(self, x, y):
return x + y
"""
def __init__(self, name):
self.name = name
def __call__(self, method):
@functools.wraps(method)
def new_method(slf, *args, **kwargs):
slf.timer.start(self.name)
x = method(slf, *args, **kwargs)
try:
slf.timer.stop()
except IndexError:
pass
return x
return new_method
|
