Store SAVE attribute with OMNI+OFP (fix #164)

ecmwf-ifs · Oct 6, 2023 · 6bc5d8e · 6bc5d8e
1 parent fe25d74
commit 6bc5d8e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/loki/frontend/ofp.py b/loki/frontend/ofp.py
@@ -917,6 +917,10 @@ def visit_declaration(self, o, **kwargs):
             target = self.visit(o.find('attribute-target'), **kwargs)
             attrs.update((target,))
 
+        if o.find('attribute-save') is not None:
+            save = self.visit(o.find('attribute-save'), **kwargs)
+            attrs.update((save,))
+
         if o.find('access-spec') is not None:
             access_spec = self.visit(o.find('access-spec'), **kwargs)
             attrs.update((access_spec,))

diff --git a/loki/frontend/omni.py b/loki/frontend/omni.py
@@ -830,6 +830,8 @@ def visit_FbasicType(self, o, **kwargs):
             _type.private = True
         if o.get('is_public') == 'true':
             _type.public = True
+        if o.get('is_save') == 'true':
+            _type.save = True
         return _type
 
     def visit_FfunctionType(self, o, **kwargs):