1 from functools
import wraps
2 from soc
.decoder
.orderedset
import OrderedSet
3 from soc
.decoder
.selectable_int
import (FieldSelectableInt
, SelectableInt
,
5 from soc
.decoder
.power_enums
import spr_dict
, XER_bits
6 from soc
.decoder
.helpers
import exts
7 from collections
import namedtuple
10 instruction_info
= namedtuple('instruction_info',
11 'func read_regs uninit_regs write_regs ' + \
12 'special_regs op_fields form asmregs')
22 def create_args(reglist
, extra
=None):
34 def __init__(self
, bytes_per_word
=8):
36 self
.bytes_per_word
= bytes_per_word
37 self
.word_log2
= math
.ceil(math
.log2(bytes_per_word
))
39 def _get_shifter_mask(self
, width
, remainder
):
40 shifter
= ((self
.bytes_per_word
- width
) - remainder
) * \
42 mask
= (1 << (width
* 8)) - 1
45 # TODO: Implement ld/st of lesser width
46 def ld(self
, address
, width
=8):
47 remainder
= address
& (self
.bytes_per_word
- 1)
48 address
= address
>> self
.word_log2
49 assert remainder
& (width
- 1) == 0, "Unaligned access unsupported!"
50 if address
in self
.mem
:
51 val
= self
.mem
[address
]
55 if width
!= self
.bytes_per_word
:
56 shifter
, mask
= self
._get
_shifter
_mask
(width
, remainder
)
57 val
= val
& (mask
<< shifter
)
59 print("Read {:x} from addr {:x}".format(val
, address
))
62 def st(self
, address
, value
, width
=8):
63 remainder
= address
& (self
.bytes_per_word
- 1)
64 address
= address
>> self
.word_log2
65 assert remainder
& (width
- 1) == 0, "Unaligned access unsupported!"
66 print("Writing {:x} to addr {:x}".format(value
, address
))
67 if width
!= self
.bytes_per_word
:
68 if address
in self
.mem
:
69 val
= self
.mem
[address
]
72 shifter
, mask
= self
._get
_shifter
_mask
(width
, remainder
)
73 val
&= ~
(mask
<< shifter
)
74 val |
= value
<< shifter
75 self
.mem
[address
] = val
77 self
.mem
[address
] = value
79 def __call__(self
, addr
, sz
):
80 val
= self
.ld(addr
.value
, sz
)
81 print ("memread", addr
, sz
, val
)
82 return SelectableInt(val
, sz
*8)
84 def memassign(self
, addr
, sz
, val
):
85 print ("memassign", addr
, sz
, val
)
86 self
.st(addr
.value
, val
.value
, sz
)
90 def __init__(self
, decoder
, regfile
):
94 self
[i
] = SelectableInt(regfile
[i
], 64)
96 def __call__(self
, ridx
):
99 def set_form(self
, form
):
102 def getz(self
, rnum
):
103 #rnum = rnum.value # only SelectableInt allowed
104 print("GPR getzero", rnum
)
106 return SelectableInt(0, 64)
109 def _get_regnum(self
, attr
):
110 getform
= self
.sd
.sigforms
[self
.form
]
111 rnum
= getattr(getform
, attr
)
114 def ___getitem__(self
, attr
):
115 print("GPR getitem", attr
)
116 rnum
= self
._get
_regnum
(attr
)
117 return self
.regfile
[rnum
]
120 for i
in range(0, len(self
), 8):
123 s
.append("%08x" % self
[i
+j
].value
)
125 print("reg", "%2d" % i
, s
)
128 def __init__(self
, pc_init
=0):
129 self
.CIA
= SelectableInt(pc_init
, 64)
130 self
.NIA
= self
.CIA
+ SelectableInt(4, 64)
132 def update(self
, namespace
):
133 self
.CIA
= namespace
['NIA'].narrow(64)
134 self
.NIA
= self
.CIA
+ SelectableInt(4, 64)
135 namespace
['CIA'] = self
.CIA
136 namespace
['NIA'] = self
.NIA
140 def __init__(self
, dec2
, initial_sprs
={}):
143 self
.update(initial_sprs
)
145 def __getitem__(self
, key
):
146 # if key in special_sprs get the special spr, otherwise return key
147 if isinstance(key
, SelectableInt
):
149 key
= special_sprs
.get(key
, key
)
151 return dict.__getitem
__(self
, key
)
154 dict.__setitem
__(self
, key
, SelectableInt(0, info
.length
))
155 return dict.__getitem
__(self
, key
)
157 def __setitem__(self
, key
, value
):
158 if isinstance(key
, SelectableInt
):
160 key
= special_sprs
.get(key
, key
)
161 dict.__setitem
__(self
, key
, value
)
163 def __call__(self
, ridx
):
169 # decoder2 - an instance of power_decoder2
170 # regfile - a list of initial values for the registers
171 def __init__(self
, decoder2
, regfile
, initial_sprs
={}, initial_cr
=0):
172 self
.gpr
= GPR(decoder2
, regfile
)
175 self
.spr
= SPR(decoder2
, initial_sprs
)
177 # FPR (same as GPR except for FP nums)
178 # 4.2.2 p124 FPSCR (definitely "separate" - not in SPR)
179 # note that mffs, mcrfs, mtfsf "manage" this FPSCR
180 # 2.3.1 CR (and sub-fields CR0..CR6 - CR0 SO comes from XER.SO)
181 # note that mfocrf, mfcr, mtcr, mtocrf, mcrxrx "manage" CRs
183 # 2.3.2 LR (actually SPR #8) -- Done
184 # 2.3.3 CTR (actually SPR #9) -- Done
185 # 2.3.4 TAR (actually SPR #815)
186 # 3.2.2 p45 XER (actually SPR #1) -- Done
187 # 3.2.3 p46 p232 VRSAVE (actually SPR #256)
189 # create CR then allow portions of it to be "selectable" (below)
190 self
._cr
= SelectableInt(initial_cr
, 64) # underlying reg
191 self
.cr
= FieldSelectableInt(self
._cr
, list(range(32,64)))
193 # "undefined", just set to variable-bit-width int (use exts "max")
194 self
.undefined
= SelectableInt(0, 256) # TODO, not hard-code 256!
196 self
.namespace
= {'GPR': self
.gpr
,
199 'memassign': self
.memassign
,
203 'undefined': self
.undefined
,
204 'mode_is_64bit': True,
208 # field-selectable versions of Condition Register TODO check bitranges?
211 bits
= tuple(range(i
*4, (i
+1)*4))# errr... maybe?
212 _cr
= FieldSelectableInt(self
.cr
, bits
)
214 self
.namespace
["CR%d" % i
] = _cr
216 self
.decoder
= decoder2
.dec
219 def memassign(self
, ea
, sz
, val
):
220 self
.mem
.memassign(ea
, sz
, val
)
222 def prep_namespace(self
, formname
, op_fields
):
223 # TODO: get field names from form in decoder*1* (not decoder2)
224 # decoder2 is hand-created, and decoder1.sigform is auto-generated
226 # then "yield" fields only from op_fields rather than hard-coded
228 fields
= self
.decoder
.sigforms
[formname
]
229 for name
in op_fields
:
231 sig
= getattr(fields
, name
.upper())
233 sig
= getattr(fields
, name
)
235 if name
in ['BF', 'BFA']:
236 self
.namespace
[name
] = val
238 self
.namespace
[name
] = SelectableInt(val
, sig
.width
)
240 self
.namespace
['XER'] = self
.spr
['XER']
241 self
.namespace
['CA'] = self
.spr
['XER'][XER_bits
['CA']].value
243 def handle_carry_(self
, inputs
, outputs
):
244 inv_a
= yield self
.dec2
.e
.invert_a
246 inputs
[0] = ~inputs
[0]
248 imm_ok
= yield self
.dec2
.e
.imm_data
.ok
250 imm
= yield self
.dec2
.e
.imm_data
.data
251 inputs
.append(SelectableInt(imm
, 64))
252 assert len(outputs
) >= 1
254 gts
= [(x
> output
) for x
in inputs
]
256 cy
= 1 if any(gts
) else 0
257 self
.spr
['XER'][XER_bits
['CA']] = cy
261 gts
= [(x
[32:64] > output
[32:64]) == SelectableInt(1, 1)
263 cy32
= 1 if any(gts
) else 0
264 self
.spr
['XER'][XER_bits
['CA32']] = cy32
266 def handle_overflow(self
, inputs
, outputs
):
267 inv_a
= yield self
.dec2
.e
.invert_a
269 inputs
[0] = ~inputs
[0]
271 imm_ok
= yield self
.dec2
.e
.imm_data
.ok
273 imm
= yield self
.dec2
.e
.imm_data
.data
274 inputs
.append(SelectableInt(imm
, 64))
275 assert len(outputs
) >= 1
277 input_sgn
= [exts(x
.value
, x
.bits
) < 0 for x
in inputs
]
278 output_sgn
= exts(output
.value
, output
.bits
) < 0
279 ov
= 1 if input_sgn
[0] == input_sgn
[1] and \
280 output_sgn
!= input_sgn
[0] else 0
282 self
.spr
['XER'][XER_bits
['OV']] = ov
283 so
= self
.spr
['XER'][XER_bits
['SO']]
285 self
.spr
['XER'][XER_bits
['SO']] = so
289 def handle_comparison(self
, outputs
):
291 out
= exts(out
.value
, out
.bits
)
292 zero
= SelectableInt(out
== 0, 1)
293 positive
= SelectableInt(out
> 0, 1)
294 negative
= SelectableInt(out
< 0, 1)
295 SO
= self
.spr
['XER'][XER_bits
['SO']]
296 cr_field
= selectconcat(negative
, positive
, zero
, SO
)
297 self
.crl
[0].eq(cr_field
)
299 def set_pc(self
, pc_val
):
300 self
.namespace
['NIA'] = SelectableInt(pc_val
, 64)
301 self
.pc
.update(self
.namespace
)
304 def call(self
, name
):
305 # TODO, asmregs is from the spec, e.g. add RT,RA,RB
306 # see http://bugs.libre-riscv.org/show_bug.cgi?id=282
307 info
= self
.instrs
[name
]
308 yield from self
.prep_namespace(info
.form
, info
.op_fields
)
310 # preserve order of register names
311 input_names
= create_args(list(info
.read_regs
) + list(info
.uninit_regs
))
314 # main registers (RT, RA ...)
316 for name
in input_names
:
317 regnum
= yield getattr(self
.decoder
, name
)
319 self
.namespace
[regname
] = regnum
320 print('reading reg %d' % regnum
)
321 inputs
.append(self
.gpr(regnum
))
323 # "special" registers
324 for special
in info
.special_regs
:
325 if special
in special_sprs
:
326 inputs
.append(self
.spr
[special
])
328 inputs
.append(self
.namespace
[special
])
331 results
= info
.func(self
, *inputs
)
334 carry_en
= yield self
.dec2
.e
.output_carry
336 yield from self
.handle_carry_(inputs
, results
)
337 ov_en
= yield self
.dec2
.e
.oe
339 yield from self
.handle_overflow(inputs
, results
)
340 rc_en
= yield self
.dec2
.e
.rc
.data
342 self
.handle_comparison(results
)
344 # any modified return results?
346 output_names
= create_args(info
.write_regs
)
347 for name
, output
in zip(output_names
, results
):
348 if isinstance(output
, int):
349 output
= SelectableInt(output
, 256)
350 if name
in info
.special_regs
:
351 print('writing special %s' % name
, output
)
352 if name
in special_sprs
:
353 self
.spr
[name
] = output
355 self
.namespace
[name
].eq(output
)
357 regnum
= yield getattr(self
.decoder
, name
)
358 print('writing reg %d %s' % (regnum
, str(output
)))
360 output
= SelectableInt(output
.value
, 64)
361 self
.gpr
[regnum
] = output
363 # update program counter
364 self
.pc
.update(self
.namespace
)
368 """ Decorator factory. """
369 def variable_injector(func
):
371 def decorator(*args
, **kwargs
):
373 func_globals
= func
.__globals
__ # Python 2.6+
374 except AttributeError:
375 func_globals
= func
.func_globals
# Earlier versions.
377 context
= args
[0].namespace
378 saved_values
= func_globals
.copy() # Shallow copy of dict.
379 func_globals
.update(context
)
380 result
= func(*args
, **kwargs
)
381 args
[0].namespace
= func_globals
382 #exec (func.__code__, func_globals)
385 # func_globals = saved_values # Undo changes.
391 return variable_injector