feat(params): support real(stp) declarations in Fortran codegen

Author: sbryngelson

toolchain/mfc/params/definitions.py

@@ -489,7 +489,7 @@ def get_value_label(param_name: str, value: int) -> str:
 }
 
 
-def _r(name, ptype, tags=None, desc=None, hint=None, math=None, str_len=None):
+def _r(name, ptype, tags=None, desc=None, hint=None, math=None, str_len=None, storage_precision=False):
     """Register a parameter with optional feature tags and description."""
     if hint is None:
         hint = _lookup_hint(name)
@@ -514,6 +514,7 @@ def _r(name, ptype, tags=None, desc=None, hint=None, math=None, str_len=None):
             hint=hint,
             math_symbol=math or "",
             str_len=str_len if str_len is not None else "name_len",
+            storage_precision=storage_precision,
         )
     )
 

toolchain/mfc/params/generators/fortran_gen.py

@@ -21,11 +21,11 @@
 _FORTRAN_TYPES = {
     ParamType.INT: "integer",
     ParamType.ANALYTIC_INT: "integer",
-    ParamType.REAL: "real(wp)",
-    ParamType.ANALYTIC_REAL: "real(wp)",
     ParamType.LOG: "logical",
 }
 
+_REAL_TYPES = (ParamType.REAL, ParamType.ANALYTIC_REAL)
+
 _VALID_TARGETS = ("pre", "sim", "post")
 
 
@@ -46,6 +46,8 @@ def get_namelist_var(name: str) -> str:
 def fortran_type_decl(param: ParamDef) -> str:
     if param.param_type == ParamType.STR:
         return f"character(LEN={param.str_len})"
+    if param.param_type in _REAL_TYPES:
+        return f"real({'stp' if param.storage_precision else 'wp'})"
     return _FORTRAN_TYPES[param.param_type]
 
 

toolchain/mfc/params/schema.py

@@ -65,6 +65,7 @@ class ParamDef:
     hint: str = ""  # Constraint/usage hint for docs (e.g. "Used with grcbc_in")
     math_symbol: str = ""  # LaTeX math symbol (Doxygen format, e.g. "\\f$\\gamma_k\\f$")
     str_len: str = "name_len"  # For STR type: Fortran character length constant ("path_len" for case_dir)
+    storage_precision: bool = False  # For REAL types: declare as real(stp) instead of real(wp)
 
     def __post_init__(self):
         # Validate name

toolchain/mfc/params_tests/test_fortran_gen.py

@@ -33,6 +33,16 @@ def test_fortran_type_real():
     assert fortran_type_decl(ParamDef(name="x", param_type=ParamType.REAL)) == "real(wp)"
 
 
+def test_fortran_type_real_storage_precision():
+    from mfc.params.generators.fortran_gen import fortran_type_decl
+    from mfc.params.schema import ParamDef, ParamType
+
+    p = ParamDef(name="x", param_type=ParamType.REAL, storage_precision=True)
+    assert fortran_type_decl(p) == "real(stp)"
+    a = ParamDef(name="x", param_type=ParamType.ANALYTIC_REAL, storage_precision=True)
+    assert fortran_type_decl(a) == "real(stp)"
+
+
 def test_fortran_type_log():
     from mfc.params.generators.fortran_gen import fortran_type_decl
     from mfc.params.schema import ParamDef, ParamType