Hardware
/
litex
mirror of https://github.com/enjoy-digital/litex.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

gen/genlib/record: rename leave_out by omit and add keep parameter to Record.connect

This commit is contained in:
Florent Kermarrec 2016-04-21 08:08:47 +02:00
parent ee378b2557
commit c330e7be49
1 changed files with 18 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
litex/gen/genlib/record.py
View File

@ -131,15 +131,24 @@ class Record:
def raw_bits(self):
return Cat(*self.flatten())
def connect(self, *slaves, leave_out=set()):
if isinstance(leave_out, str):
leave_out = {leave_out}
def connect(self, *slaves, keep=None, omit=None):
if keep is None:
keep = set([f[0] for f in self.layout])
if isinstance(keep, list):
keep = set(keep)
if omit is None:
omit = set()
if isinstance(omit, list):
omit = set(omit)
keep = keep - omit
r = []
for f in self.layout:
field = f[0]
if field not in leave_out:
self_e = getattr(self, field)
if isinstance(self_e, Signal):
if field in keep:
direction = f[2]
if direction == DIR_M_TO_S:
r += [getattr(slave, field).eq(self_e) for slave in slaves]
@ -149,7 +158,7 @@ class Record:
raise TypeError
else:
for slave in slaves:
r += self_e.connect(getattr(slave, field), leave_out=leave_out)
r += self_e.connect(getattr(slave, field), keep=keep, omit=omit)
return r
def connect_flat(self, *slaves):