Mercurial > lcfOS
annotate python/canon.py @ 284:05184b95fa16
Moved tests to seperate folder
| author | Windel Bouwman |
|---|---|
| date | Fri, 15 Nov 2013 13:43:22 +0100 |
| parents | 02385f62f250 |
| children |
| rev | line source |
|---|---|
| 275 | 1 import ir |
| 2 from itertools import chain | |
| 3 | |
| 4 def make(function, frame): | |
| 5 """ | |
| 6 Create canonicalized version of the IR-code. This means: | |
| 7 - Calls out of expressions. | |
| 8 - Other things? | |
| 9 """ | |
| 10 # Change the tree. This modifies the IR-tree! | |
| 11 # Move all parameters into registers | |
| 12 parmoves = [] | |
| 13 for p in function.arguments: | |
| 14 pt = newTemp() | |
| 15 frame.parMap[p] = pt | |
| 16 parmoves.append(ir.Move(pt, frame.argLoc(p.num))) | |
| 17 function.entry.instructions = parmoves + function.entry.instructions | |
| 18 | |
| 19 for block in function.Blocks: | |
| 20 for stmt in block.instructions: | |
| 21 rewriteStmt(stmt, frame) | |
| 22 linearize(block) | |
|
280
02385f62f250
Rework from str interface to Instruction interface
Windel Bouwman
parents:
279
diff
changeset
|
23 # TODO: schedule here? |
| 275 | 24 |
| 25 # Visit all nodes with some function: | |
| 26 # TODO: rewrite into visitor. | |
| 27 | |
| 28 # Rewrite rewrites call instructions into Eseq instructions. | |
| 29 | |
| 30 def rewriteStmt(stmt, frame): | |
| 31 if isinstance(stmt, ir.Jump): | |
| 32 pass | |
| 33 elif isinstance(stmt, ir.CJump): | |
| 34 stmt.a = rewriteExp(stmt.a, frame) | |
| 35 stmt.b = rewriteExp(stmt.b, frame) | |
| 36 elif isinstance(stmt, ir.Move): | |
| 37 stmt.src = rewriteExp(stmt.src, frame) | |
| 38 stmt.dst = rewriteExp(stmt.dst, frame) | |
| 39 elif isinstance(stmt, ir.Terminator): | |
| 40 pass | |
| 41 elif isinstance(stmt, ir.Exp): | |
| 42 stmt.e = rewriteExp(stmt.e, frame) | |
| 43 else: | |
| 44 raise NotImplementedError('STMT NI: {}'.format(stmt)) | |
| 45 | |
| 46 newTemp = ir.NamedClassGenerator('canon_reg', ir.Temp).gen | |
| 47 | |
| 48 def rewriteExp(exp, frame): | |
| 49 if isinstance(exp, ir.Binop): | |
| 50 exp.a = rewriteExp(exp.a, frame) | |
| 51 exp.b = rewriteExp(exp.b, frame) | |
| 52 return exp | |
| 53 elif isinstance(exp, ir.Const): | |
| 54 return exp | |
| 55 elif isinstance(exp, ir.Temp): | |
| 56 return exp | |
| 57 elif isinstance(exp, ir.Parameter): | |
| 58 return frame.parMap[exp] | |
| 59 elif isinstance(exp, ir.LocalVariable): | |
| 60 offset = frame.allocVar(exp) | |
| 279 | 61 return ir.Add(frame.fp, ir.Const(offset)) |
| 275 | 62 elif isinstance(exp, ir.Mem): |
| 63 exp.e = rewriteExp(exp.e, frame) | |
| 64 return exp | |
| 65 elif isinstance(exp, ir.Call): | |
| 66 exp.arguments = [rewriteExp(p, frame) for p in exp.arguments] | |
| 67 # Rewrite call into eseq: | |
| 68 t = newTemp() | |
| 69 return ir.Eseq(ir.Move(t, exp), t) | |
| 70 else: | |
| 71 raise NotImplementedError('NI: {}'.format(exp)) | |
| 72 | |
| 73 # The flatten functions pull out seq instructions to the sequence list. | |
| 74 | |
| 75 def flattenExp(exp): | |
| 76 if isinstance(exp, ir.Binop): | |
| 77 exp.a, sa = flattenExp(exp.a) | |
| 78 exp.b, sb = flattenExp(exp.b) | |
| 79 return exp, sa + sb | |
| 80 elif isinstance(exp, ir.Temp): | |
| 81 return exp, [] | |
| 82 elif isinstance(exp, ir.Const): | |
| 83 return exp, [] | |
| 84 elif isinstance(exp, ir.Mem): | |
| 85 exp.e, s = flattenExp(exp.e) | |
| 86 return exp, s | |
| 87 elif isinstance(exp, ir.Eseq): | |
| 88 s = flattenStmt(exp.stmt) | |
| 89 exp.e, se = flattenExp(exp.e) | |
| 90 return exp.e, s + se | |
| 91 elif isinstance(exp, ir.Call): | |
| 92 sp = [] | |
| 93 p = [] | |
| 94 for p_, sp_ in (flattenExp(p) for p in exp.arguments): | |
| 95 p.append(p_) | |
| 96 sp.extend(sp_) | |
| 97 exp.arguments = p | |
| 98 return exp, sp | |
| 99 else: | |
| 100 raise NotImplementedError('NI: {}'.format(exp)) | |
| 101 | |
| 102 def flattenStmt(stmt): | |
| 103 if isinstance(stmt, ir.Jump): | |
| 104 return [stmt] | |
| 105 elif isinstance(stmt, ir.CJump): | |
| 106 stmt.a, sa = flattenExp(stmt.a) | |
| 107 stmt.b, sb = flattenExp(stmt.b) | |
| 108 return sa + sb + [stmt] | |
| 109 elif isinstance(stmt, ir.Move): | |
| 110 stmt.dst, sd = flattenExp(stmt.dst) | |
| 111 stmt.src, ss = flattenExp(stmt.src) | |
| 112 return sd + ss + [stmt] | |
| 113 elif isinstance(stmt, ir.Terminator): | |
| 114 return [stmt] | |
| 115 elif isinstance(stmt, ir.Exp): | |
| 116 stmt.e, se = flattenExp(stmt.e) | |
| 117 return se + [stmt] | |
| 118 else: | |
| 119 raise NotImplementedError('STMT NI: {}'.format(stmt)) | |
| 120 | |
| 121 | |
| 122 def linearize(block): | |
| 123 """ | |
| 124 Move seq instructions to top and flatten these in an instruction list | |
| 125 """ | |
| 126 i = list(flattenStmt(s) for s in block.instructions) | |
| 127 block.instructions = list(chain.from_iterable(i)) | |
| 128 |