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
|
from cProfile import Profile
from enthought.traits.api import Any, Dict, Event, HasTraits, Str
class ProfileThis(HasTraits):
""" Control profiling of different parts of the code.
"""
# Mapping of str name to Profile instance.
profilers = Dict()
# The active Profile instance.
active_profile = Any()
active_profile_name = Str()
# An event with the profiler just ending.
profile_ended = Event()
def start(self, name):
""" Start a particular profile.
"""
if self.active_profile is None:
if name not in self.profilers:
self.profilers[name] = Profile()
self.active_profile = self.profilers[name]
self.active_profile_name = name
self.active_profile.clear()
self.active_profile.enable()
def stop(self):
""" Stop the running profile.
"""
if self.active_profile is not None:
p = self.active_profile
name = self.active_profile_name
p.disable()
self.active_profile = None
self.active_profile_name = ''
self.profile_ended = (name, p)
|
