|
206
|
1 from asmnodes import ASymbol, AInstruction, ALabel, ANumber
|
|
200
|
2 from ppci import CompilerError
|
|
199
|
3
|
|
|
4 """
|
|
|
5 Base classes for defining a target
|
|
|
6 """
|
|
|
7
|
|
|
8 # Machine code interface:
|
|
|
9 class Operand:
|
|
|
10 """ Single machine operand """
|
|
|
11 pass
|
|
|
12
|
|
206
|
13 # standard immediates:
|
|
|
14
|
|
|
15 class Imm8:
|
|
|
16 def __init__(self, imm):
|
|
|
17 assert imm < 256
|
|
|
18 self.imm = imm
|
|
|
19
|
|
|
20 @classmethod
|
|
|
21 def Create(cls, vop):
|
|
|
22 if type(vop) is ANumber and vop.number < 256:
|
|
|
23 return cls(vop.number)
|
|
234
|
24
|
|
206
|
25 class Imm3:
|
|
|
26 def __init__(self, imm):
|
|
|
27 assert imm < 8
|
|
|
28 assert type(imm) is int
|
|
|
29 self.imm = imm
|
|
|
30
|
|
|
31 @classmethod
|
|
|
32 def Create(cls, vop):
|
|
|
33 if type(vop) is ANumber and vop.number < 8:
|
|
|
34 return cls(vop.number)
|
|
|
35
|
|
234
|
36 class Instruction:
|
|
|
37 def encode(self):
|
|
|
38 raise NotImplementedError('Instruction {0} has no encode yet, TODO'.format(type(self)))
|
|
|
39 def resolve(self, f):
|
|
|
40 pass
|
|
|
41
|
|
|
42
|
|
|
43 class PseudoInstruction(Instruction):
|
|
|
44 pass
|
|
|
45
|
|
|
46
|
|
|
47 class Label(PseudoInstruction):
|
|
206
|
48 def __init__(self, name):
|
|
|
49 self.name = name
|
|
234
|
50 self.address = 0
|
|
206
|
51
|
|
|
52 @classmethod
|
|
|
53 def Create(cls, vop):
|
|
|
54 if type(vop) is ASymbol:
|
|
|
55 name = vop.name
|
|
|
56 return cls(name)
|
|
|
57
|
|
234
|
58 class Comment(PseudoInstruction):
|
|
|
59 def __init__(self, txt):
|
|
|
60 self.txt = txt
|
|
|
61 def __repr__(self):
|
|
|
62 return '; {}'.format(self.txt)
|
|
|
63
|
|
|
64 class Alignment(PseudoInstruction):
|
|
|
65 def __init__(self, a):
|
|
|
66 self.align = a
|
|
|
67 def encode(self):
|
|
|
68 pad = []
|
|
|
69 address = self.address
|
|
|
70 while (address % i.align) != 0:
|
|
|
71 address += 1
|
|
|
72 pad.append(0)
|
|
|
73 return bytes(pad)
|
|
|
74
|
|
|
75
|
|
199
|
76 class Register(Operand):
|
|
|
77 def __init__(self, name):
|
|
|
78 self.name = name
|
|
|
79
|
|
|
80
|
|
|
81 class Target:
|
|
201
|
82 def __init__(self, name, desc=''):
|
|
|
83 self.name = name
|
|
|
84 self.desc = desc
|
|
200
|
85 self.registers = []
|
|
199
|
86 self.instructions = []
|
|
200
|
87
|
|
201
|
88 def instruction(self, cls):
|
|
|
89 """ Decorator function that registers an instruction to this target """
|
|
202
|
90 self.addInstruction(cls)
|
|
201
|
91 return cls
|
|
|
92
|
|
206
|
93 def check(self):
|
|
|
94 """ Check target """
|
|
|
95 for i in self.instructions:
|
|
|
96 assert hasattr(i, 'mnemonic')
|
|
|
97 assert hasattr(i, 'operands'), str(i)
|
|
|
98 assert type(i.mnemonic) is str
|
|
|
99 assert type(i.operands) is tuple, str(i)
|
|
|
100
|
|
202
|
101 def addInstruction(self, ins_class):
|
|
|
102 self.instructions.append(ins_class)
|
|
|
103
|
|
200
|
104 def mapOperand(self, operand):
|
|
|
105 """ Try to map an operand to a target type """
|
|
|
106 if type(operand) is ASymbol:
|
|
|
107 # Try to map to register:
|
|
|
108 regs = {}
|
|
|
109 for r in self.registers:
|
|
|
110 regs[r.name] = r
|
|
|
111 if operand.name in regs:
|
|
|
112 return regs[operand.name]
|
|
201
|
113 raise CompilerError('Cannot map {0}'.format(operand))
|
|
199
|
114
|
|
200
|
115 def mapInstruction(self, vi):
|
|
203
|
116 assert type(vi) is AInstruction
|
|
200
|
117 """ Map ast tree to real instruction for this target """
|
|
199
|
118
|
|
200
|
119 # map to real operands:
|
|
|
120
|
|
|
121 # look for a suitable instruction
|
|
|
122 for ic in self.instructions:
|
|
219
|
123 if ic.mnemonic.upper() == vi.mnemonic.upper() and len(ic.operands) == len(vi.operands):
|
|
203
|
124 # Try to map operands to the correct operand types:
|
|
|
125 rops = [roptype.Create(vop) for roptype, vop in zip(ic.operands, vi.operands)]
|
|
|
126
|
|
|
127 # Check if we succeeded:
|
|
|
128 optypes = tuple(map(type, rops))
|
|
|
129 if ic.operands == optypes:
|
|
|
130 return ic(*rops)
|
|
201
|
131 raise CompilerError('No suitable instruction found for "{0}"'.format(vi))
|
|
200
|
132
|
