|
268
|
1 from .basicblock import Block
|
|
172
|
2
|
|
|
3 class Function:
|
|
230
|
4 def __init__(self, name):
|
|
269
|
5 self.name = name
|
|
|
6 self.entry = Block('{}_entry'.format(name))
|
|
|
7 self.epiloog = Block('{}_epilog'.format(name))
|
|
205
|
8
|
|
230
|
9 def __repr__(self):
|
|
|
10 return 'Function {0}'.format(self.name)
|
|
205
|
11
|
|
230
|
12 def addBB(self, bb):
|
|
|
13 self.bbs.append(bb)
|
|
|
14 bb.parent = self
|
|
|
15 addBasicBlock = addBB
|
|
205
|
16
|
|
230
|
17 def removeBasicBlock(self, bb):
|
|
|
18 self.bbs.remove(bb)
|
|
|
19 bb.parent = None
|
|
|
20
|
|
|
21 def getBBs(self):
|
|
268
|
22 bbs = [self.entry]
|
|
|
23 worklist = [self.entry]
|
|
|
24 while worklist:
|
|
|
25 b = worklist.pop()
|
|
|
26 for sb in b.Successors:
|
|
|
27 if sb not in bbs:
|
|
|
28 bbs.append(sb)
|
|
|
29 worklist.append(sb)
|
|
269
|
30 bbs.remove(self.entry)
|
|
270
|
31 if self.epiloog in bbs:
|
|
|
32 bbs.remove(self.epiloog)
|
|
269
|
33 bbs.insert(0, self.entry)
|
|
|
34 bbs.append(self.epiloog)
|
|
268
|
35 return bbs
|
|
205
|
36
|
|
230
|
37 def findBasicBlock(self, name):
|
|
|
38 for bb in self.bbs:
|
|
|
39 if bb.name == name:
|
|
|
40 return bb
|
|
|
41 raise KeyError(name)
|
|
|
42
|
|
|
43 BasicBlocks = property(getBBs)
|
|
172
|
44
|
|
253
|
45 @property
|
|
|
46 def Entry(self):
|
|
269
|
47 return self.entry
|
|
253
|
48
|
|
239
|
49 def check(self):
|
|
|
50 for bb in self.BasicBlocks:
|
|
|
51 bb.check()
|
|
253
|
52
|
|
|
53 def call(self, *args):
|
|
|
54 varmap = {}
|
|
|
55 bb = self.Entry
|
|
|
56 ip = 0
|
|
|
57 while True:
|
|
|
58 i = bb.Instructions[ip]
|
|
|
59 ip += 1
|
|
|
60 return
|
|
|
61
|
|
|
62
|
