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
|
from rpython.jit.metainterp.history import ConstInt, FLOAT
from rpython.jit.backend.ppc.locations import imm
def check_imm_box(arg, lower_bound=-2**15, upper_bound=2**15-1):
if isinstance(arg, ConstInt):
i = arg.getint()
return lower_bound <= i <= upper_bound
return False
def _check_imm_arg(i):
return (-2**15) <= i <= (2**15-1)
def _prepare_cmp_op(signed):
lower_bound = -2**15 if signed else 0
upper_bound = 2**15-1 if signed else 2**16-1
def f(self, op):
l0 = self.ensure_reg(op.getarg(0))
a1 = op.getarg(1)
if check_imm_box(a1, lower_bound, upper_bound):
l1 = imm(a1.getint())
else:
l1 = self.ensure_reg(a1)
self.free_op_vars()
res = self.force_allocate_reg_or_cc(op)
return [l0, l1, res]
return f
prepare_cmp_op = _prepare_cmp_op(signed=True)
prepare_cmp_op_unsigned = _prepare_cmp_op(signed=False)
def prepare_unary_cmp(self, op):
l0 = self.ensure_reg(op.getarg(0))
l1 = imm(0)
self.free_op_vars()
res = self.force_allocate_reg_or_cc(op)
return [l0, l1, res]
def prepare_float_cmp(self, op):
l0 = self.ensure_reg(op.getarg(0))
l1 = self.ensure_reg(op.getarg(1))
self.free_op_vars()
res = self.force_allocate_reg_or_cc(op)
return [l0, l1, res]
def prepare_unary_op(self, op):
l0 = self.ensure_reg(op.getarg(0))
self.free_op_vars()
res = self.force_allocate_reg(op)
return [l0, res]
def prepare_binary_op(self, op):
reg1 = self.ensure_reg(op.getarg(0))
reg2 = self.ensure_reg(op.getarg(1))
self.free_op_vars()
res = self.force_allocate_reg(op)
return [reg1, reg2, res]
def prepare_int_add_or_mul(self, op):
a0 = op.getarg(0)
a1 = op.getarg(1)
if check_imm_box(a0):
a0, a1 = a1, a0
l0 = self.ensure_reg(a0)
if check_imm_box(a1):
l1 = imm(a1.getint())
else:
l1 = self.ensure_reg(a1)
self.free_op_vars()
res = self.force_allocate_reg(op)
return [l0, l1, res]
def prepare_int_sub(self, op):
l0 = self.ensure_reg(op.getarg(0))
a1 = op.getarg(1)
if check_imm_box(a1, -2**15+1, 2**15):
l1 = imm(a1.getint())
else:
l1 = self.ensure_reg(a1)
self.free_op_vars()
res = self.force_allocate_reg(op)
return [l0, l1, res]
|