jan/masque
1
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Drop ports by default

Browse Source
This commit is contained in:
Jan Petykiewicz 2023-02-08 09:26:44 -08:00
parent 38a7ba6434
commit 1d720b6577
2 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
masque/library.py
View File

@ -207,6 +207,7 @@ class Library(Mapping[str, 'Pattern'], metaclass=ABCMeta):
def flatten(
self,
tops: Union[str, Sequence[str]],
flatten_ports: bool = False, # TODO document
) -> Dict[str, 'Pattern']:
"""
Removes all refs and adds equivalent shapes.
@ -238,6 +239,8 @@ class Library(Mapping[str, 'Pattern'], metaclass=ABCMeta):
raise PatternError(f'Circular reference in {name} to {target}')
p = ref.as_pattern(pattern=flattened[target])
if not flatten_ports:
p.ports.clear()
pat.append(p)
pat.refs.clear()

3
masque/pattern.py
View File

@ -561,6 +561,7 @@ class Pattern(PortList, AnnotatableImpl, Mirrorable):
def flatten(
self: P,
library: Mapping[str, P],
flatten_ports: bool = False, # TODO document
) -> 'Pattern':
"""
Removes all refs (recursively) and adds equivalent shapes.
@ -592,6 +593,8 @@ class Pattern(PortList, AnnotatableImpl, Mirrorable):
raise PatternError(f'Circular reference in {name} to {target}')
p = ref.as_pattern(pattern=flattened[target])
if not flatten_ports:
p.ports.clear()
pat.append(p)
pat.refs.clear()