def wrap(funcname, retcode, signature):
argcount = len(signature)
ourcode = MyPPCAssembler()
ourcode.mflr(r0)
ourcode.stmw(r31, r1, -4)
ourcode.stw(r0, r1, 8)
ourcode.stwu(r1, r1, -80)
ourcode.lwz(r3, r4, 8)
ourcode.cmpwi(r3, argcount)
ourcode.bne("argserror")
assert argcount < 9
if argcount > 0:
load_arg(ourcode, 0, signature[0])
for i in range(2, argcount):
load_arg(ourcode, i, signature[i])
if argcount > 1:
load_arg(ourcode, 1, signature[1])
ourcode.load_word(r0, lookup(funcname))
ourcode.mtctr(r0)
ourcode.bctrl()
if retcode == 'i':
s = lookup("PyInt_FromLong")
ourcode.load_word(r0, s)
ourcode.mtctr(r0)
ourcode.bctrl()
elif retcode == 'f':
s = lookup("PyFloat_FromDouble")
ourcode.load_word(r0, s)
ourcode.mtctr(r0)
ourcode.bctrl()
ourcode.label("epilogue")
ourcode.lwz(r0, r1, 88)
ourcode.addi(r1, r1, 80)
ourcode.mtlr(r0)
ourcode.lmw(r31, r1, -4)
ourcode.blr()
err_set = lookup("PyErr_SetObject")
exc = lookup("PyExc_TypeError")
ourcode.label("argserror")
ourcode.load_word(r5, err_set)
ourcode.mtctr(r5)
ourcode.load_from(r3, exc)
ourcode.mr(r4, r3)
ourcode.bctrl()
ourcode.li(r3, 0)
ourcode.b("epilogue")
return ourcode.assemble()
def test_intcheck(self):
a = MyPPCAssembler()
a.lwz(r5, r4, pystructs.PyVarObject.ob_size)
a.cmpwi(r5, 1)
a.bne("not_one")
a.lwz(r5, r4, pystructs.PyTupleObject.ob_item + 0*4)
a.lwz(r5, r5, 4)
a.load_word(r6, lookup("PyInt_Type"))
a.cmpw(r5, r6)
a.bne("not_int")
a.li(r3, 1)
a.b("exit")
a.label("not_int")
a.li(r3, 0)
a.b("exit")
a.label("not_one")
a.li(r3, 2)
a.label("exit")
a.load_word(r5, lookup("PyInt_FromLong"))
a.mtctr(r5)
a.bctr()
f = a.assemble()
assert f() == 2
assert f("", "") == 2
assert f("") == 0
assert f(1) == 1
def make_func(code, retcode, signature, localwords=0):
"""code shouldn't contain prologue/epilogue (or touch r31)"""
stacksize = 80 + 4*localwords
argcount = len(signature)
ourcode = MyPPCAssembler()
ourcode.mflr(r0)
ourcode.stmw(r31, r1, -4)
ourcode.stw(r0, r1, 8)
ourcode.stwu(r1, r1, -stacksize)
ourcode.lwz(r3, r4, 8)
ourcode.cmpwi(r3, argcount)
ourcode.bne("argserror")
assert argcount < 9
if argcount > 0:
load_arg(ourcode, 0, signature[0])
for i in range(2, argcount):
load_arg(ourcode, i, signature[i])
if argcount > 1:
load_arg(ourcode, 1, signature[1])
ourcode.bl(FAST_ENTRY_LABEL)
if retcode == 'i':
s = lookup("PyInt_FromLong")
ourcode.load_word(r0, s)
ourcode.mtctr(r0)
ourcode.bctrl()
elif retcode == 'f':
s = lookup("PyFloat_FromDouble")
ourcode.load_word(r0, s)
ourcode.mtctr(r0)
ourcode.bctrl()
ourcode.label("epilogue")
ourcode.lwz(r0, r1, stacksize + 8)
ourcode.addi(r1, r1, stacksize)
ourcode.mtlr(r0)
ourcode.lmw(r31, r1, -4)
ourcode.blr()
err_set = lookup("PyErr_SetObject")
exc = lookup("PyExc_TypeError")
ourcode.label("argserror")
ourcode.load_word(r5, err_set)
ourcode.mtctr(r5)
ourcode.load_from(r3, exc)
ourcode.mr(r4, r3)
ourcode.bctrl()
ourcode.li(r3, 0)
ourcode.b("epilogue")
ourcode.label(FAST_ENTRY_LABEL)
# err, should be an Assembler method:
l = {}
for k in code.labels:
l[k] = code.labels[k] + 4*len(ourcode.insts)
r = code.rlabels.copy()
for k in code.rlabels:
r[k + 4*len(ourcode.insts)] = code.rlabels[k]
ourcode.insts.extend(code.insts)
ourcode.labels.update(l)
ourcode.rlabels.update(r)
r = ourcode.assemble()
r.FAST_ENTRY_LABEL = ourcode.labels[FAST_ENTRY_LABEL]
return r
