pyHDL: CLI Update for DOM (#1808)

1 files changed, 13 insertions, 0 deletions

diff --git a/pyGHDL/dom/NonStandard.py b/pyGHDL/dom/NonStandard.py
index 1524b549f..bf48db900 100644
--- a/pyGHDL/dom/NonStandard.py
+++ b/pyGHDL/dom/NonStandard.py
@@ -42,6 +42,7 @@ from typing import Any
 
 from pydecor import export
 
+from pyGHDL.dom.PSL import VerificationUnit, VerificationProperty, VerificationMode
 from pyVHDLModel.VHDLModel import (
     Design as VHDLModel_Design,
     Library as VHDLModel_Library,
@@ -198,6 +199,18 @@ class Document(VHDLModel_Document):
                 configuration = Configuration.parse(libraryUnit)
                 self.Configurations.append(configuration)
 
+            elif nodeKind == nodes.Iir_Kind.Vunit_Declaration:
+                vunit = VerificationUnit.parse(libraryUnit)
+                self.VerificationUnits.append(vunit)
+
+            elif nodeKind == nodes.Iir_Kind.Vprop_Declaration:
+                vprop = VerificationProperty.parse(libraryUnit)
+                self.VerificationProperties.append(vprop)
+
+            elif nodeKind == nodes.Iir_Kind.Vmode_Declaration:
+                vmod = VerificationMode.parse(libraryUnit)
+                self.VerificationModes.append(vmod)
+
             else:
                 raise DOMException(
                     "Unknown design unit kind '{kindName}'({kind}).".format(