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
|
from pypy.interpreter.error import OperationError
from pypy.interpreter.gateway import unwrap_spec
from pypy.interpreter.pyframe import PyFrame
from pypy.interpreter.pycode import PyCode
from pypy.interpreter.baseobjspace import W_Root
from rpython.rlib import rvmprof, jit
from pypy.interpreter.error import oefmt
# ____________________________________________________________
_get_code = lambda frame, w_arg_or_err: frame.pycode
_decorator = rvmprof.vmprof_execute_code("pypy", _get_code, W_Root)
my_execute_frame = _decorator(PyFrame.execute_frame)
class __extend__(PyFrame):
def execute_frame(self, w_arg_or_err=None):
# indirection for the optional arguments
return my_execute_frame(self, w_arg_or_err)
def _safe(s):
if len(s) > 110:
s = s[:107] + '...'
return s.replace(':', ';')
def _get_full_name(pycode):
# careful, must not have extraneous ':' or be longer than 255 chars
return "py:%s:%d:%s" % (_safe(pycode.co_name), pycode.co_firstlineno,
_safe(pycode.co_filename))
rvmprof.register_code_object_class(PyCode, _get_full_name)
def _init_ready(pycode):
rvmprof.register_code(pycode, _get_full_name)
PyCode._init_ready = _init_ready
# ____________________________________________________________
class Cache:
def __init__(self, space):
self.w_VMProfError = space.new_exception_class("_vmprof.VMProfError")
def VMProfError(space, e):
w_VMProfError = space.fromcache(Cache).w_VMProfError
return OperationError(w_VMProfError, space.newtext(e.msg))
@unwrap_spec(fileno=int, period=float, memory=int, lines=int, native=int, real_time=int)
def enable(space, fileno, period, memory, lines, native, real_time):
"""Enable vmprof. Writes go to the given 'fileno', a file descriptor
opened for writing. *The file descriptor must remain open at least
until disable() is called.*
'interval' is a float representing the sampling interval, in seconds.
Must be smaller than 1.0
"""
try:
rvmprof.enable(fileno, period, memory, native, real_time)
except rvmprof.VMProfError as e:
raise VMProfError(space, e)
def disable(space):
"""Disable vmprof. Remember to close the file descriptor afterwards
if necessary.
"""
try:
rvmprof.disable()
except rvmprof.VMProfError as e:
raise VMProfError(space, e)
def is_enabled(space):
return space.newbool(rvmprof.is_enabled())
def get_profile_path(space):
path = rvmprof.get_profile_path(space)
if path is None:
# profiling is not enabled
return space.w_None
if path == "":
# Indicates an error! Assume platform does not implement the function call
raise oefmt(space.w_NotImplementedError, "platform not implemented")
return space.newtext(path)
def stop_sampling(space):
return space.newint(rvmprof.stop_sampling())
def start_sampling(space):
rvmprof.start_sampling()
return space.w_None
|
