← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/lsprof.py

  • Committer: John Arbash Meinel
  • Date: 2008-09-05 02:29:34 UTC
  • mto: (3697.7.4 1.7)
  • mto: This revision was merged to the branch mainline in revision 3748.
  • Revision ID: john@arbash-meinel.com-20080905022934-s8692mbwpkdwi106
Cleanups to the algorithm documentation.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/lsprof.py
27
27
 
28
28
 
29
29
def profile(f, *args, **kwds):
30
 
    """Run a function profile.
31
 
 
32
 
    Exceptions are not caught: If you need stats even when exceptions are to be
33
 
    raised, passing in a closure that will catch the exceptions and transform
34
 
    them appropriately for your driver function.
35
 
 
36
 
    :return: The functions return value and a stats object.
37
 
    """
 
30
    """XXX docstring"""
38
31
    global _g_threadmap
39
32
    p = Profiler()
40
33
    p.enable(subcalls=True)
41
34
    threading.setprofile(_thread_profile)
 
35
    # Note: The except clause is needed below so that profiling data still
 
36
    # gets dumped even when exceptions are encountered. The except clause code
 
37
    # is taken straight from run_bzr_catch_errrors() in commands.py and ought
 
38
    # to be kept in sync with it.
42
39
    try:
43
 
        ret = f(*args, **kwds)
 
40
        try:
 
41
            ret = f(*args, **kwds)
 
42
        except (KeyboardInterrupt, Exception), e:
 
43
            import bzrlib.trace
 
44
            bzrlib.trace.report_exception(sys.exc_info(), sys.stderr)
 
45
            ret = 3
44
46
    finally:
45
47
        p.disable()
46
48
        for pp in _g_threadmap.values():
47
49
            pp.disable()
48
50
        threading.setprofile(None)
49
 
 
 
51
    
50
52
    threads = {}
51
53
    for tid, pp in _g_threadmap.items():
52
54
        threads[tid] = Stats(pp.getstats(), {})
151
153
 
152
154
    This code is taken from http://ddaa.net/blog/python/lsprof-calltree
153
155
    with the changes made by J.P. Calderone and Itamar applied. Note that
154
 
    isinstance(code, str) needs to be used at times to determine if the code
 
156
    isinstance(code, str) needs to be used at times to determine if the code 
155
157
    object is actually an external code object (with a filename, etc.) or
156
158
    a Python built-in.
157
159
    """
161
163
        self.out_file = None
162
164
 
163
165
    def output(self, out_file):
164
 
        self.out_file = out_file
 
166
        self.out_file = out_file        
165
167
        out_file.write('events: Ticks\n')
166
168
        self._print_summary()
167
169
        for entry in self.data: