mirror of
https://github.com/enjoy-digital/litex.git
synced 2025-01-04 09:52:26 -05:00
corelogic: support reverse in displacer/chooser
This commit is contained in:
parent
264be80f2d
commit
1662e1b3bc
1 changed files with 14 additions and 4 deletions
|
@ -40,17 +40,27 @@ def split(v, *counts):
|
||||||
offset += n
|
offset += n
|
||||||
return tuple(r)
|
return tuple(r)
|
||||||
|
|
||||||
def displacer(signal, shift, output, n=None):
|
def displacer(signal, shift, output, n=None, reverse=False):
|
||||||
if n is None:
|
if n is None:
|
||||||
n = 2**shift.bv.width
|
n = 2**shift.bv.width
|
||||||
w = signal.bv.width
|
w = signal.bv.width
|
||||||
l = [Replicate(shift == i, w) & signal for i in range(n)]
|
if reverse:
|
||||||
|
r = reversed(range(n))
|
||||||
|
else:
|
||||||
|
r = range(n)
|
||||||
|
l = [Replicate(shift == i, w) & signal for i in r]
|
||||||
return output.eq(Cat(*l))
|
return output.eq(Cat(*l))
|
||||||
|
|
||||||
def chooser(signal, shift, output, n=None):
|
def chooser(signal, shift, output, n=None, reverse=False):
|
||||||
if n is None:
|
if n is None:
|
||||||
n = 2**shift.bv.width
|
n = 2**shift.bv.width
|
||||||
w = output.bv.width
|
w = output.bv.width
|
||||||
cases = [[Constant(i, shift.bv), output.eq(signal[i*w:(i+1)*w])] for i in range(n)]
|
cases = []
|
||||||
|
for i in range(n):
|
||||||
|
if reverse:
|
||||||
|
s = n - i - 1
|
||||||
|
else:
|
||||||
|
s = i
|
||||||
|
cases.append([Constant(i, shift.bv), output.eq(signal[s*w:(s+1)*w])])
|
||||||
cases[n-1][0] = Default()
|
cases[n-1][0] = Default()
|
||||||
return Case(shift, *cases)
|
return Case(shift, *cases)
|
||||||
|
|
Loading…
Reference in a new issue