Mercurial > lcfOS
view python/testir.py @ 218:494828a7adf1
added some sort of cache to assembler
author | Windel Bouwman |
---|---|
date | Fri, 05 Jul 2013 15:30:22 +0200 |
parents | de3a68f677a5 |
children | 81752b0f85a5 |
line wrap: on
line source
import unittest, os import c3, ppci, ir, x86, transform class ConstantFolderTestCase(unittest.TestCase): def setUp(self): self.b = ir.Builder() self.cf = transform.ConstantFolder() def testBuilder(self): m = ir.Module('test') self.b.setModule(m) f = self.b.newFunction('test') self.b.setFunction(f) bb = self.b.newBB() self.b.setBB(bb) v1 = self.b.newTmp('t') v2 = self.b.newTmp('t') v3 = self.b.newTmp('t') self.b.addIns(ir.ImmLoad(v1, 5)) self.b.addIns(ir.ImmLoad(v2, 7)) self.b.addIns(ir.BinaryOperator(v3, '+', v1, v2)) self.assertEqual(3, len(m.Instructions)) self.cf.run(m) self.assertEqual(3, len(m.Instructions)) self.assertIsInstance(m.Instructions[-1], ir.ImmLoad) self.assertEqual(12, m.Instructions[-1].value) testsrc = """ package test2; function void tesssst(int henkie) { var int a, b, cee; a = 2 * 33 - 12; b = a * 2 + 13; a = b + a; cee = a; if (a > b and b *3 - a+8*b== 3*6-b) { var int x = a; x = b * 2 - a; a = x * x * add2(x, 22 - a); } else { a = b + a + add2(a, b); } var int y; y = a - b * 53; } function int add2(int x, int y) { var int res; res = x + y + 2 - 7 + 2; //if (y < 2) //{ // return y - 33; //} res = res + (x + 2 * y) + (x + 2 * y) + (2*8) + (2*8); if (x > 13) { while (y > 1337) { res = res + 2; y = y - 12; } } return res; } """ if __name__ == '__main__': unittest.main() sys.exit() diag = ppci.DiagnosticsManager() builder = c3.Builder(diag) cgenx86 = x86.X86CodeGenSimple(diag) ir = builder.build(testsrc) diag.printErrors(testsrc) #ir.dump() cf = transform.ConstantFolder() dcd = transform.DeadCodeDeleter() m2r = transform.Mem2RegPromotor() clr = transform.CleanPass() ir.check() cf.run(ir) dcd.run(ir) clr.run(ir) m2r.run(ir) #ir.dump() # Dump a graphiz file: with open('graaf.gv', 'w') as f: ir.dumpgv(f) os.system('dot -Tpdf -ograaf.pdf graaf.gv') asm = cgenx86.genBin(ir) #for a in asm: # print(a) with open('out.asm', 'w') as f: f.write('BITS 64\n') for a in asm: f.write(str(a) + '\n') print(a)