annotate python/ir/instruction.py @ 175:a51b3c956386

Added function call in expressions
author Windel Bouwman
date Fri, 19 Apr 2013 22:15:54 +0200
parents 3eb06f5fb987
children 5fd02aa38b42
rev   line source
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
1 from .basicblock import BasicBlock
175
a51b3c956386 Added function call in expressions
Windel Bouwman
parents: 174
diff changeset
2 from .function import Function
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
3
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
4 class Value:
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
5 """ Temporary SSA value (value that is assigned only once! """
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
6 def __init__(self, name):
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
7 # TODO: add typing? for now only handle integers
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
8 self.name = name
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
9 self.interferes = set()
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
10 self.reg = None
174
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
11 self.used_by = []
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
12 def __repr__(self):
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
13 if self.reg:
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
14 n = self.reg
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
15 else:
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
16 n = self.name
174
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
17 return '{0}'.format(n) # + str(self.IsUsed)
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
18 @property
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
19 def IsUsed(self):
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
20 return len(self.used_by) > 0
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
21
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
22 class Use:
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
23 def __init__(self, user, val):
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
24 self.user = user
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
25 self.val = val
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
26 self.val.used_by.append(self.user)
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
27 def delete(self):
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
28 self.val.used_by.remove(self.user)
110
9e552d34bd60 Work on compiler
Windel Bouwman
parents: 104
diff changeset
29
155
b28a11c01dbe Simplified IR classes
Windel Bouwman
parents: 147
diff changeset
30 class Instruction:
104
ed230e947dc6 Added hexviewer
windel
parents: 99
diff changeset
31 """ Base class for all instructions. """
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
32 def __init__(self):
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
33 # live variables at this node:
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
34 self.live_in = set()
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
35 self.live_out = set()
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
36 # What variables this instruction uses and defines:
174
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
37 self.defs = []
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
38 self.uses = []
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
39 def delete(self):
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
40 for use in self.uses:
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
41 use.delete()
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
42 self.uses.clear()
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
43 def addUse(self, val):
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
44 self.uses.append(Use(self, val))
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
45 def addDef(self, v):
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
46 self.defs.append(v)
173
c1d2b6b9f9a7 Rework into passes
Windel Bouwman
parents: 171
diff changeset
47 def getParent(self):
c1d2b6b9f9a7 Rework into passes
Windel Bouwman
parents: 171
diff changeset
48 return self.parent
174
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
49 def setParent(self, p):
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
50 self.parent = p
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
51 Parent = property(getParent, setParent)
173
c1d2b6b9f9a7 Rework into passes
Windel Bouwman
parents: 171
diff changeset
52 @property
c1d2b6b9f9a7 Rework into passes
Windel Bouwman
parents: 171
diff changeset
53 def Targets(self):
c1d2b6b9f9a7 Rework into passes
Windel Bouwman
parents: 171
diff changeset
54 return self.getTargets()
170
4348da5ca307 Cleanup of ir dir
Windel Bouwman
parents: 160
diff changeset
55
158
9683a4cd848f Added some functions for code generation
Windel Bouwman
parents: 157
diff changeset
56 # Function calling:
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
57 class Call(Instruction):
175
a51b3c956386 Added function call in expressions
Windel Bouwman
parents: 174
diff changeset
58 def __init__(self, callee, arguments, result=None):
110
9e552d34bd60 Work on compiler
Windel Bouwman
parents: 104
diff changeset
59 super().__init__()
9e552d34bd60 Work on compiler
Windel Bouwman
parents: 104
diff changeset
60 self.callee = callee
175
a51b3c956386 Added function call in expressions
Windel Bouwman
parents: 174
diff changeset
61 assert type(callee) is Function
110
9e552d34bd60 Work on compiler
Windel Bouwman
parents: 104
diff changeset
62 self.arguments = arguments
175
a51b3c956386 Added function call in expressions
Windel Bouwman
parents: 174
diff changeset
63 for arg in arguments:
a51b3c956386 Added function call in expressions
Windel Bouwman
parents: 174
diff changeset
64 assert type(arg) is Value
a51b3c956386 Added function call in expressions
Windel Bouwman
parents: 174
diff changeset
65 self.addUse(arg)
a51b3c956386 Added function call in expressions
Windel Bouwman
parents: 174
diff changeset
66 self.result = result
a51b3c956386 Added function call in expressions
Windel Bouwman
parents: 174
diff changeset
67 if result:
a51b3c956386 Added function call in expressions
Windel Bouwman
parents: 174
diff changeset
68 assert type(result) is Value
a51b3c956386 Added function call in expressions
Windel Bouwman
parents: 174
diff changeset
69 self.addDef(result)
158
9683a4cd848f Added some functions for code generation
Windel Bouwman
parents: 157
diff changeset
70 def __repr__(self):
175
a51b3c956386 Added function call in expressions
Windel Bouwman
parents: 174
diff changeset
71 if self.result:
a51b3c956386 Added function call in expressions
Windel Bouwman
parents: 174
diff changeset
72 pfx = '{0} = '.format(self.result)
a51b3c956386 Added function call in expressions
Windel Bouwman
parents: 174
diff changeset
73 else:
a51b3c956386 Added function call in expressions
Windel Bouwman
parents: 174
diff changeset
74 pfx = ''
a51b3c956386 Added function call in expressions
Windel Bouwman
parents: 174
diff changeset
75 args = ','.join([str(arg) for arg in self.arguments])
a51b3c956386 Added function call in expressions
Windel Bouwman
parents: 174
diff changeset
76 return pfx + '{0}({1})'.format(self.callee.name, args)
110
9e552d34bd60 Work on compiler
Windel Bouwman
parents: 104
diff changeset
77
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
78 class Return(Instruction):
174
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
79 def __init__(self, value=None):
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
80 super().__init__()
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
81 self.value = value
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
82 if value:
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
83 self.addUse(value)
158
9683a4cd848f Added some functions for code generation
Windel Bouwman
parents: 157
diff changeset
84 def __repr__(self):
174
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
85 if self.value:
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
86 return 'Return {0}'.format(self.value)
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
87 else:
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
88 return 'Return'
173
c1d2b6b9f9a7 Rework into passes
Windel Bouwman
parents: 171
diff changeset
89 def getTargets(self):
c1d2b6b9f9a7 Rework into passes
Windel Bouwman
parents: 171
diff changeset
90 return []
70
35286e8abd03 Added some llvm classes
windel
parents:
diff changeset
91
174
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
92 class Alloc(Instruction):
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
93 """ Allocates space on the stack """
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
94 def __init__(self, value):
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
95 super().__init__()
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
96 self.value = value
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
97 self.addDef(value)
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
98 def __repr__(self):
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
99 return '{0} = alloc'.format(self.value)
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
100
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
101 class ImmLoad(Instruction):
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
102 def __init__(self, target, value):
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
103 super().__init__()
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
104 self.target = target
170
4348da5ca307 Cleanup of ir dir
Windel Bouwman
parents: 160
diff changeset
105 self.value = value
174
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
106 self.addDef(target)
170
4348da5ca307 Cleanup of ir dir
Windel Bouwman
parents: 160
diff changeset
107 def __repr__(self):
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
108 return '{0} = {1}'.format(self.target, self.value)
170
4348da5ca307 Cleanup of ir dir
Windel Bouwman
parents: 160
diff changeset
109
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
110 # Data operations
70
35286e8abd03 Added some llvm classes
windel
parents:
diff changeset
111 class BinaryOperator(Instruction):
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
112 def __init__(self, result, operation, value1, value2):
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
113 super().__init__()
157
8f3924b6076e Added some code generator things
Windel Bouwman
parents: 155
diff changeset
114 #print('operation is in binops:', operation in BinOps)
104
ed230e947dc6 Added hexviewer
windel
parents: 99
diff changeset
115 # Check types of the two operands:
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
116 self.result = result
174
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
117 self.addDef(result)
104
ed230e947dc6 Added hexviewer
windel
parents: 99
diff changeset
118 self.value1 = value1
ed230e947dc6 Added hexviewer
windel
parents: 99
diff changeset
119 self.value2 = value2
174
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
120 self.addUse(value1)
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
121 self.addUse(value2)
104
ed230e947dc6 Added hexviewer
windel
parents: 99
diff changeset
122 self.operation = operation
158
9683a4cd848f Added some functions for code generation
Windel Bouwman
parents: 157
diff changeset
123 def __repr__(self):
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
124 return '{0} = {2} {1} {3}'.format(self.result, self.operation, self.value1, self.value2)
70
35286e8abd03 Added some llvm classes
windel
parents:
diff changeset
125
170
4348da5ca307 Cleanup of ir dir
Windel Bouwman
parents: 160
diff changeset
126 # Memory functions:
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
127 class Load(Instruction):
174
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
128 def __init__(self, location, value):
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
129 super().__init__()
174
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
130 assert type(value) is Value
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
131 assert type(location) is Value, "Location must be a value"
157
8f3924b6076e Added some code generator things
Windel Bouwman
parents: 155
diff changeset
132 self.value = value
174
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
133 self.addDef(value)
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
134 self.location = location
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
135 self.addUse(self.location)
158
9683a4cd848f Added some functions for code generation
Windel Bouwman
parents: 157
diff changeset
136 def __repr__(self):
174
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
137 return '{1} = [{0}]'.format(self.location, self.value)
157
8f3924b6076e Added some code generator things
Windel Bouwman
parents: 155
diff changeset
138
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
139 class Store(Instruction):
174
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
140 def __init__(self, location, value):
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
141 super().__init__()
174
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
142 assert type(value) is Value
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
143 assert type(location) is Value, "Location must be a value"
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
144 self.location = location
160
10330be89bc2 Started from scratch with code edit
Windel Bouwman
parents: 158
diff changeset
145 self.value = value
174
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
146 self.addUse(value)
175
a51b3c956386 Added function call in expressions
Windel Bouwman
parents: 174
diff changeset
147 self.addUse(location)
160
10330be89bc2 Started from scratch with code edit
Windel Bouwman
parents: 158
diff changeset
148 def __repr__(self):
174
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
149 return '[{0}] = {1}'.format(self.location, self.value)
160
10330be89bc2 Started from scratch with code edit
Windel Bouwman
parents: 158
diff changeset
150
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
151 # Branching:
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
152 class Branch(Instruction):
170
4348da5ca307 Cleanup of ir dir
Windel Bouwman
parents: 160
diff changeset
153 def __init__(self, target):
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
154 super().__init__()
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
155 assert type(target) is BasicBlock
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
156 self.target = target
158
9683a4cd848f Added some functions for code generation
Windel Bouwman
parents: 157
diff changeset
157 def __repr__(self):
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
158 return 'BRANCH {0}'.format(self.target)
173
c1d2b6b9f9a7 Rework into passes
Windel Bouwman
parents: 171
diff changeset
159 def getTargets(self):
c1d2b6b9f9a7 Rework into passes
Windel Bouwman
parents: 171
diff changeset
160 return [self.target]
170
4348da5ca307 Cleanup of ir dir
Windel Bouwman
parents: 160
diff changeset
161
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
162 class ConditionalBranch(Instruction):
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
163 def __init__(self, a, cond, b, lab1, lab2):
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
164 super().__init__()
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
165 self.a = a
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
166 assert type(a) is Value
170
4348da5ca307 Cleanup of ir dir
Windel Bouwman
parents: 160
diff changeset
167 self.cond = cond
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
168 self.b = b
174
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
169 self.addUse(a)
3eb06f5fb987 Added memory alloc for locals
Windel Bouwman
parents: 173
diff changeset
170 self.addUse(b)
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
171 assert type(b) is Value
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
172 assert type(lab1) is BasicBlock
170
4348da5ca307 Cleanup of ir dir
Windel Bouwman
parents: 160
diff changeset
173 self.lab1 = lab1
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
174 assert type(lab2) is BasicBlock
170
4348da5ca307 Cleanup of ir dir
Windel Bouwman
parents: 160
diff changeset
175 self.lab2 = lab2
4348da5ca307 Cleanup of ir dir
Windel Bouwman
parents: 160
diff changeset
176 def __repr__(self):
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
177 return 'IF {0} {1} {2} THEN {3} ELSE {4}'.format(self.a, self.cond, self.b, self.lab1, self.lab2)
173
c1d2b6b9f9a7 Rework into passes
Windel Bouwman
parents: 171
diff changeset
178 def getTargets(self):
c1d2b6b9f9a7 Rework into passes
Windel Bouwman
parents: 171
diff changeset
179 return [self.lab1, self.lab2]
170
4348da5ca307 Cleanup of ir dir
Windel Bouwman
parents: 160
diff changeset
180
171
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
181 class PhiNode(Instruction):
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
182 def __init__(self):
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
183 super().__init__()
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
184 self.incBB = []
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
185 def addIncoming(self, bb):
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
186 self.incBB.append(bb)
3eb9b9e2958d Improved IR code
Windel Bouwman
parents: 170
diff changeset
187