EAMxx: Fixing after rebasing against master.

Author: odiazib

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

@@ -236,28 +236,19 @@ void MAMMicrophysics::set_grids(
     add_field<Computed>("mam4_microphysics_tendency_renaming_cloud_borne", vector3d_num_gas_aerosol_constituents, nondim, grid_name);
   }
 
-<<<<<<< HEAD
   // Creating a Linoz reader and setting Linoz parameters involves reading data
   // from a file and configuring the necessary parameters for the Linoz model.
-#ifdef USE_OLD_LINOZ_FILE_READ
-  if (config_.linoz.compute) {
 
-=======
->>>>>>> b16a6664e4 (EAMxx: Fixing variable names and add comments.)
-  // NOTE: order matches mam4xx:
-  // names of variables in oxid file.
-  var_names_oxi_={"O3", "OH", "NO3", "HO2"};
   // names of variables for linoz
-  var_names_linoz_ = {
+  if (config_.linoz.compute) {
+    var_names_linoz_ = {
         "o3_clim",  "o3col_clim", "t_clim",      "PmL_clim",
         "dPmL_dO3", "dPmL_dT",    "dPmL_dO3col", "cariolle_pscs"};
-<<<<<<< HEAD
-    std::cout << "Using former reader..." << "\n";
-=======
-
+  }
 #ifdef USE_OLD_LINOZ_FILE_READ
-  {
->>>>>>> b16a6664e4 (EAMxx: Fixing variable names and add comments.)
+  if (config_.linoz.compute) {
+
+    std::cout << "Using former reader..." << "\n";
     linoz_file_name_ = m_params.get<std::string>("mam4_linoz_file_name");
     const std::string linoz_map_file =
         m_params.get<std::string>("aero_microphys_remap_file", "");
@@ -281,7 +272,9 @@ void MAMMicrophysics::set_grids(
     linoz_data_.init(num_cols_io_linoz, num_levs_io_linoz, nvars);
     linoz_data_.allocate_temporary_views();
   }  // LINOZ reader
-
+  // NOTE: order matches mam4xx:
+  // names of variables in oxid file.
+  var_names_oxi_={"O3", "OH", "NO3", "HO2"};
   {
     oxid_file_name_ = m_params.get<std::string>("mam4_oxid_file_name");
     const std::string oxid_map_file =
@@ -311,9 +304,11 @@ void MAMMicrophysics::set_grids(
   }  // oxid file reader
 
 #else
-  // The DataInterpolation class uses Field. We save these fields in FM.
-  for(const auto &field_name : var_names_linoz_) {
+  if (config_.linoz.compute) {
+    // The DataInterpolation class uses Field. We save these fields in FM.
+    for(const auto &field_name : var_names_linoz_) {
       add_field<Computed>(field_name, scalar3d_mid, nondim, grid_name);
+    }
   }
   for(const auto &field_name : var_names_oxi_) {
       add_field<Computed>(field_name, scalar3d_mid, nondim, grid_name);
@@ -735,7 +730,7 @@ void MAMMicrophysics::initialize_impl(const RunType run_type) {
   add_postcondition_check<Interval>(get_field_out("cldfrac_liq"),m_grid,0.0,1.0,false);
   add_postcondition_check<LowerBound>(get_field_out("tke"),m_grid,0);
   */
-
+#ifdef USE_OLD_LINOZ_FILE_READ
   if (config_.linoz.compute) {
     // climatology data for linear stratospheric chemistry
     auto linoz_o3_clim = buffer_.scratch[0];  // ozone (climatology) [vmr]
@@ -753,7 +748,7 @@ void MAMMicrophysics::initialize_impl(const RunType run_type) {
                                                   // overhead O3 column [vmr/DU]
     auto linoz_cariolle_pscs =
         buffer_.scratch[7];  // Cariolle parameter for PSC loss of ozone [1/s]
-#endif
+
     auto ts = start_of_step_ts();
     std::string linoz_chlorine_file =
         m_params.get<std::string>("mam4_linoz_chlorine_file");
@@ -762,7 +757,7 @@ void MAMMicrophysics::initialize_impl(const RunType run_type) {
         linoz_chlorine_file, ts, chlorine_loading_ymd, chlorine_values_,
         chlorine_time_secs_);
   }
-
+#endif
   init_temporary_views();
   // FIXME : why are we only using nlev_ instead of ncol_xnlev?
   cmfdqr_ = view_1d("cmfdqr_", nlev_);
@@ -779,7 +774,9 @@ void MAMMicrophysics::initialize_impl(const RunType run_type) {
       TracerDataReader_, curr_month, *TracerHorizInterp_, tracer_data_);
 #else
   set_oxid_reader();
-  set_linoz_reader();
+  if (config_.linoz.compute) {
+    set_linoz_reader();
+  }
 #endif
 
 #ifdef USE_OLD_VERTICAL_FILE_READ
@@ -915,8 +912,8 @@ void MAMMicrophysics::run_impl(const double dt) {
   }
 
 
-  view_2d linoz_output[8];
-  if (config_.linoz.compute) {
+
+#ifdef USE_OLD_LINOZ_FILE_READ
   // climatology data for linear stratospheric chemistry
   // ozone (climatology) [vmr]
   view_2d linoz_o3_clim;
@@ -934,39 +931,20 @@ void MAMMicrophysics::run_impl(const double dt) {
   view_2d linoz_dPmL_dO3col;
   // Cariolle parameter for PSC loss of ozone [1/s]
   view_2d linoz_cariolle_pscs;
-
-
-  linoz_output[0] = linoz_o3_clim;
-  linoz_output[1] = linoz_o3col_clim;
-  linoz_output[2] = linoz_t_clim;
-  linoz_output[3] = linoz_PmL_clim;
-  linoz_output[4] = linoz_dPmL_dO3;
-  linoz_output[5] = linoz_dPmL_dT;
-  linoz_output[6] = linoz_dPmL_dO3col;
-  linoz_output[7] = linoz_cariolle_pscs;
-  }
 #else
-  //FIXME: I will need to modify perform_atmospheric_chemistry_and_microphysics in MAM4xx.
-  // Let's do that after we have tested the new interface for DataInterpolation.
-  // const auto linoz_o3_clim = get_field_in("o3_clim").get_view<const Real **>();
-  // const auto linoz_o3col_clim = get_field_in("o3col_clim").get_view<const Real **>();
-  // const auto linoz_t_clim = get_field_in("t_clim").get_view<const Real **>();
-  // const auto linoz_PmL_clim = get_field_in("PmL_clim").get_view<const Real **>();
-  // const auto linoz_dPmL_dO3 = get_field_in("dPmL_dO3").get_view<const Real **>();
-  // const auto linoz_dPmL_dT = get_field_in("dPmL_dT").get_view<const Real **>();
-  // const auto linoz_dPmL_dO3col = get_field_in("dPmL_dO3col").get_view<const Real **>();
-  // const auto linoz_cariolle_pscs = get_field_in("cariolle_pscs").get_view<const Real **>();
   data_interp_oxid_->run(end_of_step_ts());
-  data_interp_linoz_->run(end_of_step_ts());
 
-  const auto linoz_o3_clim = get_field_out("o3_clim").get_view<Real **>();
-  const auto linoz_o3col_clim = get_field_out("o3col_clim").get_view<Real **>();
-  const auto linoz_t_clim = get_field_out("t_clim").get_view<Real **>();
-  const auto linoz_PmL_clim = get_field_out("PmL_clim").get_view<Real **>();
-  const auto linoz_dPmL_dO3 = get_field_out("dPmL_dO3").get_view<Real **>();
-  const auto linoz_dPmL_dT = get_field_out("dPmL_dT").get_view<Real **>();
-  const auto linoz_dPmL_dO3col = get_field_out("dPmL_dO3col").get_view<Real **>();
-  const auto linoz_cariolle_pscs = get_field_out("cariolle_pscs").get_view<Real **>();
+  if (config_.linoz.compute) {
+    data_interp_linoz_->run(end_of_step_ts());
+    linoz_o3_clim = get_field_out("o3_clim").get_view<Real **>();
+    linoz_o3col_clim = get_field_out("o3col_clim").get_view<Real **>();
+    linoz_t_clim = get_field_out("t_clim").get_view<Real **>();
+    linoz_PmL_clim = get_field_out("PmL_clim").get_view<Real **>();
+    linoz_dPmL_dO3 = get_field_out("dPmL_dO3").get_view<Real **>();
+    linoz_dPmL_dT = get_field_out("dPmL_dT").get_view<Real **>();
+    linoz_dPmL_dO3col = get_field_out("dPmL_dO3col").get_view<Real **>();
+    linoz_cariolle_pscs = get_field_out("cariolle_pscs").get_view<Real **>();
+  }
 
   const auto oxid_O3 = get_field_out("O3").get_view<Real **>();
   const auto oxid_OH = get_field_out("OH").get_view<Real **>();
@@ -988,6 +966,7 @@ void MAMMicrophysics::run_impl(const double dt) {
   }
 
 #ifdef USE_OLD_LINOZ_FILE_READ
+  const auto &cnst_offline = cnst_offline_;
   // Update the TracerTimeState to reflect the current time
   trace_time_state_.t_now = ts.frac_of_year_in_days();
   scream::mam_coupling::advance_tracer_data(
@@ -998,7 +977,6 @@ void MAMMicrophysics::run_impl(const double dt) {
       dry_atm_.p_mid, dry_atm_.z_iface,  // in
       cnst_offline_);                    // out
   Kokkos::fence();
-  const auto &cnst_offline = cnst_offline_;
 
   if (config_.linoz.compute) {
     view_2d linoz_output[8];
@@ -1010,7 +988,6 @@ void MAMMicrophysics::run_impl(const double dt) {
     linoz_output[5] = linoz_dPmL_dT;
     linoz_output[6] = linoz_dPmL_dO3col;
     linoz_output[7] = linoz_cariolle_pscs;
-
     scream::mam_coupling::advance_tracer_data(
       LinozDataReader_,                  // in
       *LinozHorizInterp_,                // out
@@ -1021,6 +998,7 @@ void MAMMicrophysics::run_impl(const double dt) {
     Kokkos::fence();
   }
 #endif
+
 #ifdef USE_OLD_VERTICAL_FILE_READ
   int i                            = 0;
   for(const auto &var_name : extfrc_lst_) {
@@ -1188,6 +1166,12 @@ void MAMMicrophysics::run_impl(const double dt) {
         for(int i = 0; i < mam4::mo_setinv::num_tracer_cnst; ++i) {
           cnst_offline_icol[i] = ekat::subview(cnst_offline[i], icol);
         }
+ #else
+       cnst_offline_icol[0] = ekat::subview(oxid_O3, icol);;
+       cnst_offline_icol[1] = ekat::subview(oxid_OH, icol);;
+       cnst_offline_icol[2] = ekat::subview(oxid_NO3, icol);;
+       cnst_offline_icol[3] = ekat::subview(oxid_HO2, icol);;
+ #endif
         // calculate o3 column densities (first component of col_dens in Fortran
         // code)
         auto o3_col_dens_i = ekat::subview(o3_col_dens, icol);