usb-tools/amaranth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

hdl.rec: implement slicing by component names.

Browse source 
Fixes #121.
This commit is contained in:
whitequark 2019-07-02 17:44:55 +00:00
parent 34f110100a
commit 6b843b5be6
2 changed files with 34 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

17
nmigen/hdl/rec.py
View file

@ -56,8 +56,15 @@ class Layout:
shape = (shape, False)
self.fields[name] = (shape, direction)
def __getitem__(self, name):
return self.fields[name]
def __getitem__(self, item):
if isinstance(item, tuple):
return Layout([
(name, shape, dir)
for (name, (shape, dir)) in self.fields.items()
if name in item
])
return self.fields[item]
def __iter__(self):
for name, (shape, dir) in self.fields.items():
@ -121,6 +128,12 @@ class Record(Value):
reference = "Record '{}'".format(self.name)
raise AttributeError("{} does not have a field '{}'. Did you mean one of: {}?"
.format(reference, item, ", ".join(self.fields))) from None
elif isinstance(item, tuple):
return Record(self.layout[item], fields={
field_name: field_value
for field_name, field_value in self.fields.items()
if field_name in item
})
else:
return super().__getitem__(item)