update land energy/water conservation functions

Author: juliasloan25

experiments/ClimaEarth/components/land/climaland_integrated.jl

@@ -361,16 +361,36 @@ Interfacer.get_field(sim::ClimaLandSimulation, ::Val{:area_fraction}) = sim.area
 Interfacer.get_field(sim::ClimaLandSimulation, ::Val{:beta}) =
     CL.surface_evaporative_scaling(sim.model, sim.integrator.u, sim.integrator.p)
 Interfacer.get_field(sim::ClimaLandSimulation, ::Val{:emissivity}) = sim.integrator.p.ϵ_sfc
-Interfacer.get_field(sim::ClimaLandSimulation, ::Val{:energy}) =
-    CL.total_energy(sim.integrator.u, sim.integrator.p)
 Interfacer.get_field(sim::ClimaLandSimulation, ::Val{:surface_direct_albedo}) =
     CL.surface_albedo(sim.model, sim.integrator.u, sim.integrator.p)
 Interfacer.get_field(sim::ClimaLandSimulation, ::Val{:surface_diffuse_albedo}) =
     CL.surface_albedo(sim.model, sim.integrator.u, sim.integrator.p)
-Interfacer.get_field(sim::ClimaLandSimulation, ::Val{:water}) =
-    CL.total_water(sim.integrator.u, sim.integrator.p)
 Interfacer.get_field(sim::ClimaLandSimulation, ::Val{:surface_temperature}) =
     sim.integrator.p.T_sfc
+function Interfacer.get_field(sim::ClimaLandSimulation, ::Val{:energy}, level)
+    return sum(
+        CL.total_energy_per_area!(
+            sim.integrator.p.soil.sfc_scratch,
+            sim.model,
+            sim.integrator.u,
+            sim.integrator.p,
+            sim.integrator.t,
+        ),
+    )
+end
+function Interfacer.get_field(sim::ClimaLandSimulation, ::Val{:water}, level)
+    # Convert volumetric water content to mass water content
+    ρ_liq = LP.ρ_cloud_liq(sim.model.soil.parameters.earth_param_set)
+    return ρ_liq * sum(
+        CL.total_liq_water_vol_per_area!(
+            sim.integrator.p.soil.sfc_scratch,
+            sim.model,
+            sim.integrator.u,
+            sim.integrator.p,
+            sim.integrator.t,
+        ),
+    )
+end
 
 # Update fields stored in land drivers
 function Interfacer.update_field!(sim::ClimaLandSimulation, ::Val{:diffuse_fraction}, field)