From 6b5e08c5373607115cb0fec38b67a76e2f7f8927 Mon Sep 17 00:00:00 2001
From: Patrick Lehmann <Patrick.Lehmann@plc2.de>
Date: Tue, 22 Jun 2021 23:30:06 +0200
Subject: Handle shared variables.

---
 pyGHDL/dom/Object.py                 | 17 +++++++++++++++++
 pyGHDL/dom/_Translate.py             |  8 ++++++++
 pyGHDL/dom/formatting/prettyprint.py | 12 +++++++++++-
 3 files changed, 36 insertions(+), 1 deletion(-)

(limited to 'pyGHDL')

diff --git a/pyGHDL/dom/Object.py b/pyGHDL/dom/Object.py
index 953fddc41..c8777f722 100644
--- a/pyGHDL/dom/Object.py
+++ b/pyGHDL/dom/Object.py
@@ -39,6 +39,7 @@ from pyVHDLModel.VHDLModel import (
     Constant as VHDLModel_Constant,
     DeferredConstant as VHDLModel_DeferredConstant,
     Variable as VHDLModel_Variable,
+    SharedVariable as VHDLModel_SharedVariable,
     Signal as VHDLModel_Signal,
     Expression,
     SubTypeOrSymbol,
@@ -105,6 +106,22 @@ class Variable(VHDLModel_Variable):
         return cls(name, subTypeIndication, defaultExpression)
 
 
+@export
+class SharedVariable(VHDLModel_SharedVariable):
+    def __init__(self, name: str, subType: SubTypeOrSymbol):
+        super().__init__(name)
+
+        self._name = name
+        self._subType = subType
+
+    @classmethod
+    def parse(cls, node):
+        name = GetNameOfNode(node)
+        subTypeIndication = GetSubtypeIndicationFromNode(node, "variable", name)
+
+        return cls(name, subTypeIndication)
+
+
 @export
 class Signal(VHDLModel_Signal):
     def __init__(
diff --git a/pyGHDL/dom/_Translate.py b/pyGHDL/dom/_Translate.py
index 2b2a44e60..18b4160b8 100644
--- a/pyGHDL/dom/_Translate.py
+++ b/pyGHDL/dom/_Translate.py
@@ -366,6 +366,14 @@ def GetDeclaredItemsFromChainedNodes(nodeChain: Iir, entity: str, name: str):
             from pyGHDL.dom.Object import Constant
 
             result.append(Constant.parse(item))
+
+        elif kind == nodes.Iir_Kind.Variable_Declaration:
+            from pyGHDL.dom.Object import SharedVariable
+
+            if nodes.Get_Shared_Flag(item):
+                result.append(SharedVariable.parse(item))
+            else:
+                raise DOMException("Found non-shared variable.")
         elif kind == nodes.Iir_Kind.Signal_Declaration:
             from pyGHDL.dom.Object import Signal
 
diff --git a/pyGHDL/dom/formatting/prettyprint.py b/pyGHDL/dom/formatting/prettyprint.py
index 0509b826d..c4f76acaa 100644
--- a/pyGHDL/dom/formatting/prettyprint.py
+++ b/pyGHDL/dom/formatting/prettyprint.py
@@ -24,7 +24,7 @@ from pyGHDL.dom.DesignUnit import (
     Context,
     Component,
 )
-from pyGHDL.dom.Object import Constant, Signal
+from pyGHDL.dom.Object import Constant, Signal, SharedVariable
 from pyGHDL.dom.InterfaceItem import (
     GenericConstantInterfaceItem,
     PortSignalInterfaceItem,
@@ -302,6 +302,16 @@ class PrettyPrint:
                     expr=str(item.DefaultExpression),
                 )
             )
+        elif isinstance(item, SharedVariable):
+            buffer.append(
+                "{prefix}- shared variable {name} : {subtype}".format(
+                    prefix=prefix,
+                    name=item.Name,
+                    subtype=self.formatSubtypeIndication(
+                        item.SubType, "shared variable", item.Name
+                    ),
+                )
+            )
         elif isinstance(item, Signal):
             buffer.append(
                 "{prefix}- signal {name} : {subtype}{initValue}".format(
-- 
cgit v1.2.3