Adding two vertical axes for the snow in NOAH-MP

RegIPSL/modeles/WRF/frame/module_xios.F

@@ -293,6 +293,31 @@ CONTAINS
       CALL wrf_debug(nlevdbg1, 'soil_layer definition for '//tfo)
       CALL wrf_debug(nlevdbg1, 'soil_layer definition ok '//clwrfdom)
 
+
+      kpn=config_flags%num_snow_layers
+#if defined key_cpp_xios
+      CALL xios_set_axis_attr("snow_layers", size=kpn, VALUE=(/ (REAL(ji,KIND=8), ji = 1, kpn) /), &
+      &                               long_name="Snow Layers", unit="m", positive="up")
+#else
+      CALL xios_set_axis_attr("snow_layers", n_glo=kpn, VALUE=(/ (REAL(ji,KIND=8), ji = 1, kpn) /), & 
+      &                               long_name="Snow Layers", unit="m", positive="up")
+#endif
+      tfo = TRIM(i2str(kpn))//' levels'
+      CALL wrf_debug(nlevdbg1, 'snow_layers definition for '//tfo)
+      CALL wrf_debug(nlevdbg1, 'snow_layers definition ok '//clwrfdom)
+
+      kpn=config_flags%num_snso_layers
+#if defined key_cpp_xios
+      CALL xios_set_axis_attr("snso_layers", size=kpn, VALUE=(/ (REAL(ji,KIND=8), ji = 1, kpn) /), &
+      &                               long_name="snso Layers", unit="m", positive="up")
+#else
+      CALL xios_set_axis_attr("snso_layers", n_glo=kpn, VALUE=(/ (REAL(ji,KIND=8), ji = 1, kpn) /), & 
+      &                               long_name="snso Layers", unit="m", positive="up")
+#endif
+      tfo = TRIM(i2str(kpn))//' levels'
+      CALL wrf_debug(nlevdbg1, 'snso_layers definition for '//tfo)
+      CALL wrf_debug(nlevdbg1, 'snso_layers definition ok '//clwrfdom)
+      
       kpn=config_flags%num_ext_model_couple_dom
 #if defined key_cpp_xios
       CALL xios_set_axis_attr("num_ext_model", size=kpn, VALUE=(/ (REAL(ji,KIND=8), ji = 1, kpn) /), &
@@ -487,6 +512,10 @@ CONTAINS
                claxe = "num_ext_model"
             ELSE IF ( INDEX(vdimname,'lagday') /= 0 ) THEN
                claxe = "lagday"
+            ELSE IF ( INDEX(vdimname,'snow_layers_stag') /= 0 ) THEN
+               claxe = "snow_layers"
+            ELSE IF ( INDEX(vdimname,'snso_layers_stag') /= 0 ) THEN
+               claxe = "snso_layers"
             ELSE
                CALL wrf_debug(nlevdbg0, 'xios_def_field 3D: for '//cl20//' missing vertical axis name : '//TRIM(vdimname))
                CALL wrf_error_fatal ( "ERROR: xios_def_field unknown vertical axis" )

RegIPSL/modeles/WRF/xml/context_wrf.xml

@@ -31,6 +31,8 @@
       <axis id="soil_layer" />
       <axis id="num_ext_model" />
       <axis id="lagday" />
+      <axis id="snow_layers" />
+      <axis id="snso_layers" />
     </axis_definition>
  
     <domain_definition>