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

gen/fhdl/namer: Add update method to HierarchyNode to replace update_hierarchy_node.

This commit is contained in:
Florent Kermarrec 2023-11-06 13:52:02 +01:00
parent c0057672d6
commit c8a96b8d79
1 changed files with 25 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
litex/gen/fhdl/namer.py
View File

@ -27,33 +27,32 @@ class HierarchyNode:
self.use_number = False
self.children = {}
def update_hierarchy_node(current, name, number, use_number, current_base):
def update(self, name, number, use_number, current_base=None):
"""
Updates or creates a hierarchy node based on the current position, name, and number.
If numbering is used, sorts and stores all numbers associated with the base node.
Parameters:
curren t (HierarchyNode): The current node in the hierarchy.
name (str): The name of the current hierarchy level.
number (int): The number associated with the current hierarchy level.
use_number (bool): Flag indicating whether to use the number in the hierarchy.
current_base (HierarchyNode): The base node for number usage information.
current_base (HierarchyNode, optional): The base node for number usage information.
Returns:
HierarchyNode: The updated or created node.
HierarchyNode: The updated or created child node.
"""
# Create the appropriate key for the node.
key = (name, number) if use_number else name
# Use setdefault to either get the existing child node or create a new one.
current = current.children.setdefault(key, HierarchyNode())
child = self.children.setdefault(key, HierarchyNode())
# Add the number to the set of numbers associated with this node.
current.numbers.add(number)
child.numbers.add(number)
# Increment the count of signals that have traversed this node.
current.signal_count += 1
child.signal_count += 1
# If numbering is used, sort and store all numbers associated with the base node.
if use_number and current_base:
current.all_numbers = sorted(current_base.numbers)
return current
child.all_numbers = sorted(current_base.numbers)
return child
def build_hierarchy_tree(signals, base_tree=None):
"""
@ -82,7 +81,7 @@ def build_hierarchy_tree(signals, base_tree=None):
use_number = current_base.use_number if current_base else False
# Update the current node in the hierarchy.
current = update_hierarchy_node(current, name, number, use_number, current_base)
current = current.update(name, number, use_number, current_base)
return root