EAMxx: Move the unit conversion of O3 from wet mm rto dry vmr. Add a namelist parameter in RRTMGP so that one can decide whether to use prog ozone from the chemistry model or not.

Author: odiazib

components/eamxx/cime_config/namelist_defaults_eamxx.xml

@@ -307,7 +307,6 @@ be lost if SCREAM_HACK_XML is not enabled.
       <mam4_do_rename type="logical" doc="Switch to enable aerosol microphysics rename process">true</mam4_do_rename>
       <!-- LINOZ parameters -->
       <mam4_compute_linoz    type="logical"    doc="turn on/off linoz computation">true</mam4_compute_linoz>
-      <mam4_use_ozone_linoz_in_rrtmgp    type="logical"    doc="Use OZONE (o3_volume_mix_ratio) computed by MAM4 in the radiation model.">false</mam4_use_ozone_linoz_in_rrtmgp>
       <mam4_o3_tau    type="real"    doc="Linoz tau parameter">172800.0</mam4_o3_tau>
       <mam4_o3_sfc    type="real"    doc="Linoz surface parameter">3.0E-008</mam4_o3_sfc>
       <mam4_o3_lbl    type="integer" doc="Linoz lbl parameter">4</mam4_o3_lbl>
@@ -558,6 +557,8 @@ be lost if SCREAM_HACK_XML is not enabled.
       </do_subcol_sampling>
       <pool_size_multiplier type="real">1.0</pool_size_multiplier>
       <force_run_after_restart type="logical" doc="Force rad to run on first step after restart, regardless of rad frequency">false</force_run_after_restart>
+      <prognostic_ozone    type="logical"    doc="Use OZONE computed by a chemistry model, e.g, mam4xx.">false</prognostic_ozone>
+
     </rrtmgp>
 
     <mac_aero_mic inherit="atm_proc_group">

components/eamxx/src/physics/mam/eamxx_mam_microphysics_process_interface.cpp

@@ -36,7 +36,7 @@ MAMMicrophysics::MAMMicrophysics(const ekat::Comm &comm,
 
   // LINOZ namelist parameters
   config_.linoz.compute = m_params.get<bool>("mam4_compute_linoz", true);
-  m_use_ozone_linoz_in_rrtmgp =m_params.get<bool>("mam4_use_ozone_linoz_in_rrtmgp",false);
+
   if (config_.linoz.compute) {
     config_.linoz.o3_lbl = m_params.get<int>("mam4_o3_lbl");
     config_.linoz.o3_tau = m_params.get<double>("mam4_o3_tau");
@@ -380,11 +380,6 @@ void MAMMicrophysics::set_grids(
                                            index_season_lai_);
   }
 
-  if (config_.linoz.compute and m_use_ozone_linoz_in_rrtmgp) {
-     // dry O3 vmr [mol/mol]
-    add_field<Computed>("o3_volume_mix_ratio", scalar3d_mid, mol / mol, grid_name);
-  }
-
 }  // set_grids
 
 // ================================================================
@@ -562,12 +557,7 @@ void MAMMicrophysics::initialize_impl(const RunType run_type) {
     scream::mam_coupling::create_linoz_chlorine_reader(
         linoz_chlorine_file, ts, chlorine_loading_ymd, chlorine_values_,
         chlorine_time_secs_);
-  } else{
-    EKAT_REQUIRE_MSG(
-        !m_use_ozone_linoz_in_rrtmgp,
-        "Error! mam4_compute_linoz is false. "
-        "Linoz will no compute o3_volume_mix_ratio. \n");
-  }  // LINOZ
+  }
 
   init_temporary_views();
   // FIXME : why are we only using nlev_ instead of ncol_xnlev?
@@ -1058,19 +1048,6 @@ void MAMMicrophysics::run_impl(const double dt) {
   // postprocess output
   post_process(wet_aero_, dry_aero_, dry_atm_);
   Kokkos::fence();
-  if (config_.linoz.compute and m_use_ozone_linoz_in_rrtmgp) {
-    // 0 corresponds to O3
-    constexpr int O3_idx=0;
-    constexpr Real O3_gas_mol_weights = mam4::gas_chemistry::adv_mass[O3_idx];
-    auto dry_o3_vmr = get_field_out("o3_volume_mix_ratio").get_view<Real**>();
-    Kokkos::parallel_for(policy, KOKKOS_LAMBDA(const ThreadTeam& team) {
-      const int icol = team.league_rank();
-      Kokkos::parallel_for(Kokkos::TeamVectorRange(team, nlev), [&] (const int& k) {
-        dry_o3_vmr(icol,k) = PF::calculate_vmr_from_mmr(O3_gas_mol_weights,
-                          dry_atm.qv(icol,k),dry_aero.gas_mmr[O3_idx](icol,k));
-      });
-    });
-  }
 
 }  // MAMMicrophysics::run_impl
 

components/eamxx/src/physics/mam/eamxx_mam_microphysics_process_interface.hpp

@@ -158,8 +158,6 @@ class MAMMicrophysics final : public MAMGenericInterface {
   void init_temporary_views();
   int len_temporary_views_{0};
 
-  // if linoz is turn on, mam4 will update o3 in rrtmgp
-  bool m_use_ozone_linoz_in_rrtmgp;
 
 };  // MAMMicrophysics
 

components/eamxx/src/physics/rrtmgp/eamxx_rrtmgp_process_interface.cpp

@@ -82,6 +82,9 @@ RRTMGPRadiation (const ekat::Comm& comm, const ekat::ParameterList& params)
 
   // Determine rad timestep, specified as number of atm steps
   m_rad_freq_in_steps = m_params.get<Int>("rad_frequency", 1);
+  // use ozone computed by a chemistry model, e.g., mam4xx
+  m_prognostic_ozone =m_params.get<bool>("prognostic_ozone",false);
+
 }
 
 void RRTMGPRadiation::set_grids(const std::shared_ptr<const GridsManager> grids_manager) {
@@ -383,6 +386,9 @@ void RRTMGPRadiation::init_buffers(const ATMBufferManager &buffer_manager)
   mem += m_buffer.d_tint.size();
   m_buffer.d_dz  = decltype(m_buffer.d_dz)(mem, m_col_chunk_size, m_nlay);
   mem += m_buffer.d_dz.size();
+  //
+  m_buffer.o3_prog  = decltype(m_buffer.o3_prog)(mem, m_col_chunk_size, m_nlay);
+  mem += m_buffer.o3_prog.size();
   // 3d arrays
   m_buffer.sw_flux_up_k = decltype(m_buffer.sw_flux_up_k)(mem, m_col_chunk_size, m_nlay+1);
   mem += m_buffer.sw_flux_up_k.size();
@@ -549,6 +555,19 @@ void RRTMGPRadiation::initialize_impl(const RunType run_type) {
 
   m_force_run_on_next_step = run_type==RunType::Initial or
     m_params.get("force_run_after_restart",false);
+
+  // If m_prognostic_ozone is true, we should not use O3 as it is currently employed in RRTMGP.
+  if (m_prognostic_ozone)
+  {
+   for (int igas = 0; igas < m_ngas; igas++) {
+    auto name = m_gas_names[igas];
+    EKAT_REQUIRE_MSG(
+        m_gas_names[igas] !="o3",
+        "Error! m_prognostic_ozone is set to true,"
+        "remove o3 from list of active gases.. \n");
+   }
+  }
+
 }
 
 // =========================================================================================
@@ -955,6 +974,28 @@ void RRTMGPRadiation::run_impl (const double dt) {
         m_gas_concs_k.set_vmr(name, tmp2d_k);
       }
 
+      if (m_prognostic_ozone)
+      {
+        auto o3_dry_vmr = m_buffer.o3_prog;
+        // get O3 from FM, here we assume that O3 is wet_mmr
+        auto o3_wet_mmr = get_field_out("O3").get_view<Real**>();
+        const Real O3_molecular_weigth= PC::get_gas_mol_weight("o3");
+        const auto policy = ekat::ExeSpaceUtils<ExeSpace>::get_default_team_policy(ncol, m_nlay);
+        const int nlev_local = m_nlay;
+        Kokkos::parallel_for(policy, KOKKOS_LAMBDA(const MemberType& team) {
+        const int icol = team.league_rank();
+          Kokkos::parallel_for(Kokkos::TeamVectorRange(team, nlev_local), [&] (const int& k) {
+          // from wet to dry
+          const Real o3_dry_mmr = PF::calculate_drymmr_from_wetmmr(o3_wet_mmr(icol,k),d_qv(icol,k));
+          // Question: Do I need to use qv_dry?
+          // from mmr to vmr
+          o3_dry_vmr(icol,k) = PF::calculate_vmr_from_mmr(O3_molecular_weigth,
+                          d_qv(icol,k),o3_dry_mmr);
+        });
+      });
+      m_gas_concs_k.set_vmr("o3", o3_dry_vmr);
+      }
+
       // Set layer cloud fraction.
       //
       // If not doing subcolumn sampling for mcica, we want to make sure we use grid-mean

components/eamxx/src/physics/rrtmgp/eamxx_rrtmgp_process_interface.hpp

@@ -133,7 +133,7 @@ class RRTMGPRadiation : public AtmosphereProcess {
   // Structure for storing local variables initialized using the ATMBufferManager
   struct Buffer {
     static constexpr int num_1d_ncol        = 8;
-    static constexpr int num_2d_nlay        = 16;
+    static constexpr int num_2d_nlay        = 17;
     static constexpr int num_2d_nlay_p1     = 23;
     static constexpr int num_2d_nswbands    = 2;
     static constexpr int num_3d_nlev_nswbands = 4;
@@ -171,6 +171,10 @@ class RRTMGPRadiation : public AtmosphereProcess {
     ureal2dk sw_heating_k;
     ureal2dk lw_heating_k;
 
+    // ozone prog
+    ureal2dk o3_prog;
+
+
     // 2d size (ncol, nlay+1)
     ureal2dk d_tint;
     ureal2dk p_lev_k;
@@ -240,6 +244,10 @@ class RRTMGPRadiation : public AtmosphereProcess {
   Buffer m_buffer;
 
   bool m_force_run_on_next_step = false;
+  // use a ozone prognostic variable computed by chemistry
+  bool m_prognostic_ozone = false;
+
+
 };  // class RRTMGPRadiation
 
 }  // namespace scream