Fred Drake | 47cada7 | 2001-10-15 22:18:53 +0000 | [diff] [blame] | 1 | """Statistics analyzer for HotShot.""" |
| 2 | |
| 3 | import profile |
| 4 | import pstats |
| 5 | |
| 6 | import hotshot.log |
| 7 | |
| 8 | from hotshot.log import ENTER, EXIT |
| 9 | |
| 10 | |
| 11 | def load(filename): |
| 12 | return StatsLoader(filename).load() |
| 13 | |
| 14 | |
| 15 | class StatsLoader: |
| 16 | def __init__(self, logfn): |
| 17 | self._logfn = logfn |
| 18 | self._code = {} |
| 19 | self._stack = [] |
| 20 | self.pop_frame = self._stack.pop |
| 21 | |
| 22 | def load(self): |
| 23 | # The timer selected by the profiler should never be used, so make |
| 24 | # sure it doesn't work: |
| 25 | p = Profile() |
| 26 | p.get_time = _brokentimer |
| 27 | log = hotshot.log.LogReader(self._logfn) |
| 28 | taccum = 0 |
| 29 | for event in log: |
| 30 | what, (filename, lineno, funcname), tdelta = event |
| 31 | if tdelta > 0: |
| 32 | taccum += tdelta |
| 33 | |
| 34 | # We multiply taccum to convert from the microseconds we |
| 35 | # have to the seconds that the profile/pstats module work |
| 36 | # with; this allows the numbers to have some basis in |
| 37 | # reality (ignoring calibration issues for now). |
| 38 | |
| 39 | if what == ENTER: |
| 40 | frame = self.new_frame(filename, lineno, funcname) |
| 41 | p.trace_dispatch_call(frame, taccum * .000001) |
| 42 | taccum = 0 |
| 43 | |
| 44 | elif what == EXIT: |
| 45 | frame = self.pop_frame() |
| 46 | p.trace_dispatch_return(frame, taccum * .000001) |
| 47 | taccum = 0 |
| 48 | |
| 49 | # no further work for line events |
| 50 | |
| 51 | assert not self._stack |
| 52 | return pstats.Stats(p) |
| 53 | |
| 54 | def new_frame(self, *args): |
| 55 | # args must be filename, firstlineno, funcname |
| 56 | # our code objects are cached since we don't need to create |
| 57 | # new ones every time |
| 58 | try: |
| 59 | code = self._code[args] |
| 60 | except KeyError: |
| 61 | code = FakeCode(*args) |
| 62 | self._code[args] = code |
| 63 | # frame objects are create fresh, since the back pointer will |
| 64 | # vary considerably |
| 65 | if self._stack: |
| 66 | back = self._stack[-1] |
| 67 | else: |
| 68 | back = None |
| 69 | frame = FakeFrame(code, back) |
| 70 | self._stack.append(frame) |
| 71 | return frame |
| 72 | |
| 73 | |
| 74 | class Profile(profile.Profile): |
| 75 | def simulate_cmd_complete(self): |
| 76 | pass |
| 77 | |
| 78 | |
| 79 | class FakeCode: |
| 80 | def __init__(self, filename, firstlineno, funcname): |
| 81 | self.co_filename = filename |
| 82 | self.co_firstlineno = firstlineno |
| 83 | self.co_name = self.__name__ = funcname |
| 84 | |
| 85 | |
| 86 | class FakeFrame: |
| 87 | def __init__(self, code, back): |
| 88 | self.f_back = back |
| 89 | self.f_code = code |
| 90 | |
| 91 | |
| 92 | def _brokentimer(): |
Collin Winter | 865ea89 | 2007-08-22 21:45:02 +0000 | [diff] [blame] | 93 | raise RuntimeError("this timer should not be called") |