diff --git a/litex/gen/fhdl/hierarchy.py b/litex/gen/fhdl/hierarchy.py index 17ff922dc..720fe0207 100644 --- a/litex/gen/fhdl/hierarchy.py +++ b/litex/gen/fhdl/hierarchy.py @@ -26,7 +26,8 @@ class LiteXHierarchyExplorer: r = "" names = set() names.add(None) - # Modules / SubModules. + + # Modules / Sub-Modules. for name, mod in module._submodules: if name is None: n = 0 @@ -51,10 +52,13 @@ class LiteXHierarchyExplorer: r += f"{self.tree_ident*ident}{self.tree_entry}{self._colorer(f'[{s.of}]', 'yellow')}\n" return r - def __repr__(self): - r = "\n" + def get_hierarchy(self): + r = "" r += f"{self._colorer(self.top.__class__.__name__, 'underline')}\n" r += self.get_tree(self.top) - r += f"{self._colorer('* ', 'cyan')}: Generated name.\n" + r += f"{self._colorer('* ', 'cyan')}: Generated name.\n" r += f"{self._colorer('[]', 'yellow')}: BlackBox.\n" return r + + def __repr__(self): + return f"\n{self.get_hierarchy()}" diff --git a/litex/gen/fhdl/verilog.py b/litex/gen/fhdl/verilog.py index bce9ce3cb..ce52fa355 100644 --- a/litex/gen/fhdl/verilog.py +++ b/litex/gen/fhdl/verilog.py @@ -89,11 +89,11 @@ def _generate_timescale(time_unit="1ns", time_precision="1ps"): # ------------------------------------------------------------------------------------------------ # def _generate_hierarchy(top): - hierarchy = LiteXHierarchyExplorer(top=top, depth=None, with_colors=False).__repr__() + hierarchy_explorer = LiteXHierarchyExplorer(top=top, depth=None, with_colors=False) r = "/*\n" - for l in hierarchy.split("\n"): + for l in hierarchy_explorer.get_hierarchy().split("\n"): r += l + "\n" - #r += "//" + l + "\n" + r = r[:-1] r += "*/\n" return r