| Just van Rossum | 40f9b7b | 1999-01-30 22:39:17 +0000 | [diff] [blame] | 1 | import W |
| Jack Jansen | 5a6fdcd | 2001-08-25 12:15:04 +0000 | [diff] [blame^] | 2 | from Carbon import Evt |
| Just van Rossum | 40f9b7b | 1999-01-30 22:39:17 +0000 | [diff] [blame] | 3 | |
| 4 | import sys |
| 5 | import StringIO |
| 6 | import string |
| 7 | import pstats, fpformat |
| 8 | |
| 9 | # increase precision |
| 10 | def f8(x): |
| 11 | return string.rjust(fpformat.fix(x, 4), 8) |
| 12 | pstats.f8 = f8 |
| 13 | |
| 14 | # hacking around a hack |
| 15 | if sys.version[:3] > '1.4': |
| 16 | timer = Evt.TickCount |
| 17 | else: |
| 18 | def timer(TickCount = Evt.TickCount): |
| 19 | return TickCount() / 60.0 |
| 20 | |
| 21 | class ProfileBrowser: |
| 22 | |
| 23 | def __init__(self, stats = None): |
| 24 | self.sortkeys = ('calls',) |
| 25 | self.setupwidgets() |
| 26 | self.setstats(stats) |
| 27 | |
| 28 | def setupwidgets(self): |
| 29 | self.w = W.Window((580, 400), "Profile Statistics", minsize = (200, 100), tabbable = 0) |
| 30 | self.w.divline = W.HorizontalLine((0, 20, 0, 0)) |
| 31 | self.w.titlebar = W.TextBox((4, 4, 40, 12), 'Sort by:') |
| 32 | self.buttons = [] |
| 33 | self.w.button_calls = W.RadioButton((54, 4, 45, 12), 'calls', self.buttons, self.setsort) |
| 34 | self.w.button_time = W.RadioButton((104, 4, 40, 12), 'time', self.buttons, self.setsort) |
| 35 | self.w.button_cumulative = W.RadioButton((154, 4, 75, 12), 'cumulative', self.buttons, self.setsort) |
| 36 | self.w.button_stdname = W.RadioButton((234, 4, 60, 12), 'stdname', self.buttons, self.setsort) |
| 37 | self.w.button_calls.set(1) |
| 38 | self.w.button_file = W.RadioButton((304, 4, 40, 12), 'file', self.buttons, self.setsort) |
| 39 | self.w.button_line = W.RadioButton((354, 4, 50, 12), 'line', self.buttons, self.setsort) |
| 40 | self.w.button_name = W.RadioButton((404, 4, 50, 12), 'name', self.buttons, self.setsort) |
| 41 | ## self.w.button_nfl = W.RadioButton((4, 4, 12, 12), 'nfl', self.buttons, self.setsort) |
| 42 | ## self.w.button_pcalls = W.RadioButton((4, 4, 12, 12), 'pcalls', self.buttons, self.setsort) |
| 43 | self.w.text = W.TextEditor((0, 21, -15, -15), inset = (6, 5), |
| 44 | readonly = 1, wrap = 0, fontsettings = ('Monaco', 0, 9, (0, 0, 0))) |
| 45 | self.w._bary = W.Scrollbar((-15, 20, 16, -14), self.w.text.vscroll, max = 32767) |
| 46 | self.w._barx = W.Scrollbar((-1, -15, -14, 16), self.w.text.hscroll, max = 32767) |
| 47 | self.w.open() |
| 48 | |
| 49 | def setstats(self, stats): |
| 50 | self.stats = stats |
| 51 | self.stats.strip_dirs() |
| 52 | self.displaystats() |
| 53 | |
| 54 | def setsort(self): |
| 55 | # Grmpf. The callback doesn't give us the button:-( |
| 56 | for b in self.buttons: |
| 57 | if b.get(): |
| 58 | if b._title == self.sortkeys[0]: |
| 59 | return |
| 60 | self.sortkeys = (b._title,) + self.sortkeys[:3] |
| 61 | break |
| 62 | self.displaystats() |
| 63 | |
| 64 | def displaystats(self): |
| 65 | W.SetCursor('watch') |
| 66 | apply(self.stats.sort_stats, self.sortkeys) |
| 67 | saveout = sys.stdout |
| 68 | try: |
| 69 | s = sys.stdout = StringIO.StringIO() |
| 70 | self.stats.print_stats() |
| 71 | finally: |
| 72 | sys.stdout = saveout |
| 73 | text = string.join(string.split(s.getvalue(), '\n'), '\r') |
| 74 | self.w.text.set(text) |
| 75 | |
| 76 | |
| 77 | def main(): |
| 78 | import pstats |
| 79 | args = sys.argv[1:] |
| 80 | for i in args: |
| 81 | stats = pstats.Stats(i) |
| 82 | browser = ProfileBrowser(stats) |
| 83 | else: |
| 84 | import macfs |
| 85 | fss, ok = macfs.PromptGetFile('Profiler data') |
| 86 | if not ok: sys.exit(0) |
| 87 | stats = pstats.Stats(fss.as_pathname()) |
| 88 | browser = ProfileBrowser(stats) |
| 89 | |
| 90 | if __name__ == '__main__': |
| 91 | main() |