From: Florent Kermarrec <florent@enjoy-digital.fr>
Date: Thu, 21 Apr 2016 10:16:26 +0000 (+0200)
Subject: gen/genlib/record: fix connect
X-Git-Tag: 24jan2021_ls180~1964
X-Git-Url: https://git.libre-soc.org/?a=commitdiff_plain;h=e80cfedd7fb8a004b0661212f9223000c24bc0c5;p=litex.git

gen/genlib/record: fix connect
---

diff --git a/litex/gen/genlib/record.py b/litex/gen/genlib/record.py
index 9b4ce3db..e98922ff 100644
--- a/litex/gen/genlib/record.py
+++ b/litex/gen/genlib/record.py
@@ -133,22 +133,22 @@ class Record:
 
     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)
+            _keep = set([f[0] for f in self.layout])
+        elif isinstance(keep, list):
+            _keep = set(keep)
         if omit is None:
-            omit = set()
-        if isinstance(omit, list):
-            omit = set(omit)
+            _omit = set()
+        elif isinstance(omit, list):
+            _omit = set(omit)
 
-        keep = keep - omit
+        _keep = _keep - _omit
 
         r = []
         for f in self.layout:
             field = f[0]
             self_e = getattr(self, field)
             if isinstance(self_e, Signal):
-                if field in keep:
+                if field in _keep:
                     direction = f[2]
                     if direction == DIR_M_TO_S:
                         r += [getattr(slave, field).eq(self_e) for slave in slaves]