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
|
from rpython.flowspace.model import Constant
from rpython.flowspace.operation import op
from rpython.rtyper.error import TyperError
from rpython.rtyper.rmodel import Repr
from rpython.tool.pairtype import pairtype
class ControlledInstanceRepr(Repr):
def __init__(self, rtyper, s_real_obj, controller):
self.rtyper = rtyper
self.s_real_obj = s_real_obj
self.r_real_obj = rtyper.getrepr(s_real_obj)
self.controller = controller
self.lowleveltype = self.r_real_obj.lowleveltype
def convert_const(self, value):
real_value = self.controller.convert(value)
return self.r_real_obj.convert_const(real_value)
def reveal(self, r):
if r is not self:
raise TyperError("expected %r, got %r" % (self, r))
return self.s_real_obj, self.r_real_obj
def rtype_getattr(self, hop):
return self.controller.rtype_getattr(hop)
def rtype_setattr(self, hop):
return self.controller.rtype_setattr(hop)
def rtype_bool(self, hop):
return self.controller.rtype_bool(hop)
def rtype_simple_call(self, hop):
return self.controller.rtype_call(hop)
class __extend__(pairtype(ControlledInstanceRepr, Repr)):
def rtype_getitem((r_controlled, r_key), hop):
return r_controlled.controller.rtype_getitem(hop)
def rtype_setitem((r_controlled, r_key), hop):
return r_controlled.controller.rtype_setitem(hop)
def rtype_delitem((r_controlled, r_key), hop):
return r_controlled.controller.rtype_delitem(hop)
def rtypedelegate(callable, hop, revealargs=[0], revealresult=False):
bk = hop.rtyper.annotator.bookkeeper
c_meth = Constant(callable)
s_meth = bk.immutablevalue(callable)
hop2 = hop.copy()
for index in revealargs:
r_controlled = hop2.args_r[index]
if not isinstance(r_controlled, ControlledInstanceRepr):
raise TyperError("args_r[%d] = %r, expected ControlledInstanceRepr"
% (index, r_controlled))
s_new, r_new = r_controlled.s_real_obj, r_controlled.r_real_obj
hop2.args_s[index], hop2.args_r[index] = s_new, r_new
v = hop2.args_v[index]
if isinstance(v, Constant):
real_value = r_controlled.controller.convert(v.value)
hop2.args_v[index] = Constant(real_value)
if revealresult:
r_controlled = hop2.r_result
if not isinstance(r_controlled, ControlledInstanceRepr):
raise TyperError("r_result = %r, expected ControlledInstanceRepr"
% (r_controlled,))
s_new, r_new = r_controlled.s_real_obj, r_controlled.r_real_obj
hop2.s_result, hop2.r_result = s_new, r_new
hop2.v_s_insertfirstarg(c_meth, s_meth)
spaceop = op.simple_call(*hop2.args_v)
spaceop.result = hop2.spaceop.result
hop2.spaceop = spaceop
return hop2.dispatch()
|
