From 6e2719c66ce430ed1d9ec24273a2f54035f93d87 Mon Sep 17 00:00:00 2001 From: Tristan Gingold Date: Thu, 8 Feb 2018 03:54:12 +0100 Subject: Improve python binding. --- src/vhdl/python/libghdl/thinutils.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) (limited to 'src/vhdl/python/libghdl/thinutils.py') diff --git a/src/vhdl/python/libghdl/thinutils.py b/src/vhdl/python/libghdl/thinutils.py index 93f00b8ba..85edafbff 100644 --- a/src/vhdl/python/libghdl/thinutils.py +++ b/src/vhdl/python/libghdl/thinutils.py @@ -194,6 +194,7 @@ def declarations_iter(n): if n1 != Null_Iir: for n2 in declarations_iter(n1): yield n2 + # All these nodes are handled: if k in [iirs.Iir_Kind.Entity_Declaration, iirs.Iir_Kind.Architecture_Body, iirs.Iir_Kind.Package_Declaration, @@ -248,7 +249,11 @@ def declarations_iter(n): def concurrent_stmts_iter(n): """Iterator on concurrent statements in n.""" k = iirs.Get_Kind(n) - if k == iirs.Iir_Kind.Design_Unit: + if k == iirs.Iir_Kind.Design_File: + for n1 in chain_iter(iirs.Get_First_Design_Unit(n)): + for n2 in concurrent_stmts_iter(n1): + yield n2 + elif k == iirs.Iir_Kind.Design_Unit: for n1 in concurrent_stmts_iter(iirs.Get_Library_Unit(n)): yield n1 elif k == iirs.Iir_Kind.Entity_Declaration \ -- cgit v1.2.3