Merge pull request #3654 from CliMA/zs/aero_col_mass

add sea salt column mass diagnostics

Author: szy21

config/model_configs/aquaplanet_nonequil_allsky_gw_res.yml

@@ -20,10 +20,10 @@ orographic_gravity_wave: "gfdl_restart"
 surface_setup: "DefaultMoninObukhov"
 prescribe_ozone: true
 reproducibility_test: true
-prescribed_aerosols: ["CB1", "CB2", "SO4"]
+prescribed_aerosols: ["SO4", "CB1", "OC1", "DST01", "SSLT01"]
 toml: [toml/sphere_aquaplanet_1M.toml]
 diagnostics:
-  - short_name: [edt, evu, mmrso4, mmrbcpo, mmrbcpi, o3, od550aer, odsc550aer]
+  - short_name: [edt, evu, mmrso4, mmrbcpo, mmrocpo, mmrdust, mmrss, loadss, o3, od550aer, odsc550aer]
     reduction_time: average
     period: 1days
   - short_name: [reffclw, reffcli]

reproducibility_tests/ref_counter.jl

@@ -1,4 +1,4 @@
-215
+216
 
 # **README**
 #
@@ -21,6 +21,9 @@
 
 #=
 
+216
+- Change prescribed aerosols in `aquaplanet_nonequil_allsky_gw_res.yml`
+
 215
 - Update dependencies, including ClimaCore, which updated how metric terms are
   computed

src/diagnostics/tracer_diagnostics.jl

@@ -105,6 +105,48 @@ function compute_sea_salt!(out, state, cache, time)
     end
 end
 
+function compute_sea_salt_column!(out, state, cache, time)
+    :prescribed_aerosols_field in propertynames(cache.tracers) ||
+        error("Aerosols do not exist in the model")
+    any(
+        x -> x in propertynames(cache.tracers.prescribed_aerosols_field),
+        [:SSLT01, :SSLT02, :SSLT03, :SSLT04, :SSLT05],
+    ) || error("Sea salt does not exist in the model")
+    if isnothing(out)
+        out = zeros(axes(Fields.level(state.f, half)))
+        aero_conc = cache.scratch.ᶜtemp_scalar
+        @. aero_conc = 0
+        for prescribed_aerosol_name in
+            [:SSLT01, :SSLT02, :SSLT03, :SSLT04, :SSLT05]
+            if prescribed_aerosol_name in
+               propertynames(cache.tracers.prescribed_aerosols_field)
+                aerosol_field = getproperty(
+                    cache.tracers.prescribed_aerosols_field,
+                    prescribed_aerosol_name,
+                )
+                @. aero_conc += aerosol_field
+            end
+        end
+        Operators.column_integral_definite!(out, aero_conc)
+        return out
+    else
+        aero_conc = cache.scratch.ᶜtemp_scalar
+        @. aero_conc = 0
+        for prescribed_aerosol_name in
+            [:SSLT01, :SSLT02, :SSLT03, :SSLT04, :SSLT05]
+            if prescribed_aerosol_name in
+               propertynames(cache.tracers.prescribed_aerosols_field)
+                aerosol_field = getproperty(
+                    cache.tracers.prescribed_aerosols_field,
+                    prescribed_aerosol_name,
+                )
+                @. aero_conc += aerosol_field
+            end
+        end
+        Operators.column_integral_definite!(out, aero_conc)
+    end
+end
+
 ###
 # Ozone concentration (3d)
 ###
@@ -195,3 +237,15 @@ add_diagnostic_variable!(
     comments = "Prescribed dry mass fraction of hydrophilic organic carbon aerosol particles in air.",
     compute! = (out, u, p, t) -> compute_aerosol!(out, u, p, t, :OC2),
 )
+
+###
+# Sea salt column mass (2d)
+###
+add_diagnostic_variable!(
+    short_name = "loadss",
+    long_name = "Load of Sea-Salt Aerosol",
+    standard_name = "atmosphere_mass_content_of_sea_salt_dry_aerosol _particles",
+    units = "kg m^-2",
+    comments = "The total dry mass of sea salt aerosol particles per unit area.",
+    compute! = (out, u, p, t) -> compute_sea_salt_column!(out, u, p, t),
+)