Mercurial > lcfOS
view python/ir/function.py @ 268:5ec7580976d9
Op naar tree-IR
author | Windel Bouwman |
---|---|
date | Wed, 14 Aug 2013 20:12:40 +0200 |
parents | 225f444019b1 |
children | 5f8c04a8d26b |
line wrap: on
line source
from .basicblock import Block class Function: def __init__(self, name): self.name = name self.entry = Block('entry') def __repr__(self): return 'Function {0}'.format(self.name) def addBB(self, bb): self.bbs.append(bb) bb.parent = self addBasicBlock = addBB def removeBasicBlock(self, bb): self.bbs.remove(bb) bb.parent = None def getBBs(self): bbs = [self.entry] worklist = [self.entry] while worklist: b = worklist.pop() for sb in b.Successors: if sb not in bbs: bbs.append(sb) worklist.append(sb) return bbs def findBasicBlock(self, name): for bb in self.bbs: if bb.name == name: return bb raise KeyError(name) BasicBlocks = property(getBBs) @property def Entry(self): return self.BasicBlocks[0] def check(self): for bb in self.BasicBlocks: bb.check() def call(self, *args): varmap = {} bb = self.Entry ip = 0 while True: i = bb.Instructions[ip] ip += 1 return