typed records

Author: wpbonelli

flopy4/mf6/codec/reader/grammar/__init__.py

@@ -4,7 +4,7 @@
 import jinja2
 from modflow_devtools.dfn import Dfn, get_blocks, get_fields
 
-from flopy4.mf6.codec.reader.grammar.filters import field_type
+from flopy4.mf6.codec.reader.grammar.filters import field_type, record_child_type
 
 
 def _get_template_env():
@@ -16,6 +16,7 @@ def _get_template_env():
         keep_trailing_newline=True,
     )
     env.filters["field_type"] = field_type
+    env.filters["record_child_type"] = record_child_type
     return env
 
 

flopy4/mf6/codec/reader/grammar/filters.py

@@ -13,3 +13,18 @@ def field_type(field: Field) -> str:
             return "record"
         case _:
             return field["type"]
+
+
+def record_child_type(field: Field) -> str:
+    """Get the grammar type for a field within a record context."""
+    match field["type"]:
+        case "string":
+            return "string"
+        case "integer":
+            return "integer"
+        case "double precision":
+            return "double"
+        case "keyword":
+            return ""
+        case _:
+            return field["type"]

flopy4/mf6/codec/reader/grammar/templates/component.lark.jinja

@@ -13,5 +13,11 @@ block: {% for block_name in blocks.keys() %}{{ block_name }}_block{% if not loop
 {% endfor %}
 {% for field_name, field in fields.items() %}
 {% set field_type = field|field_type %}
+{% if field['type'] == 'record' and field.get('children') %}
+{{ field_name }}: "{{ field['name'] }}"i {{ field_name }}_record
+{{ field_name }}_record: {% for child_name, child in field['children'].items() %}{% set child_type = child|record_child_type %}{% if child['type'] == 'keyword' %}"{{ child['name'] }}"i{% else %}{{ child_type }}{% endif %}{% if not loop.last %} {% endif %}{% endfor %}
+
+{% else %}
 {{ field_name }}: "{{ field['name'] }}"i {{ field_type }}
+{% endif %}
 {% endfor %}