Handle context references.

1 files changed, 22 insertions, 4 deletions

diff --git a/pyGHDL/dom/DesignUnit.py b/pyGHDL/dom/DesignUnit.py
index c905a439d..7f4d8e964 100644
--- a/pyGHDL/dom/DesignUnit.py
+++ b/pyGHDL/dom/DesignUnit.py
@@ -47,6 +47,7 @@ from pydecor import export
 from pyVHDLModel.SyntaxModel import (
     LibraryClause as VHDLModel_LibraryClause,
     UseClause as VHDLModel_UseClause,
+    ContextReference as VHDLModel_ContextReference,
     Entity as VHDLModel_Entity,
     Architecture as VHDLModel_Architecture,
     Package as VHDLModel_Package,
@@ -82,16 +83,16 @@ __all__ = []
 
 @export
 class LibraryClause(VHDLModel_LibraryClause, DOMMixin):
-    def __init__(self, node: Iir, names: Iterable[Name]):
+    def __init__(self, libraryNode: Iir, names: Iterable[Name]):
         super().__init__(names)
-        DOMMixin.__init__(self, node)
+        DOMMixin.__init__(self, libraryNode)
 
 
 @export
 class UseClause(VHDLModel_UseClause, DOMMixin):
-    def __init__(self, node: Iir, names: Iterable[Name]):
+    def __init__(self, useNode: Iir, names: Iterable[Name]):
         super().__init__(names)
-        DOMMixin.__init__(self, node)
+        DOMMixin.__init__(self, useNode)
 
     @classmethod
     def parse(cls, useNode: Iir):
@@ -105,6 +106,23 @@ class UseClause(VHDLModel_UseClause, DOMMixin):
 
 
 @export
+class ContextReference(VHDLModel_ContextReference, DOMMixin):
+    def __init__(self, contextNode: Iir, names: Iterable[Name]):
+        super().__init__(names)
+        DOMMixin.__init__(self, contextNode)
+
+    @classmethod
+    def parse(cls, contextNode: Iir):
+        from pyGHDL.dom._Translate import GetNameFromNode
+
+        contexts = [GetNameFromNode(nodes.Get_Selected_Name(contextNode))]
+        for context in utils.chain_iter(nodes.Get_Context_Reference_Chain(contextNode)):
+            contexts.append(GetNameFromNode(nodes.Get_Selected_Name(context)))
+
+        return cls(contextNode, contexts)
+
+
+@export
 class Entity(VHDLModel_Entity, DOMMixin):
     def __init__(
         self,