8#include <AMReX_buildInfo.H>
46#ifdef REMORA_USE_NETCDF
67 BL_PROFILE(
"REMORA::REMORA()");
69 if (ParallelDescriptor::IOProcessor()) {
72 const char* buildgithash = amrex::buildInfoGetBuildGitHash();
73 const char* buildgitname = amrex::buildInfoGetBuildGitName();
75 if (strlen(remora_hash) > 0) {
76 amrex::Print() <<
"\n"
77 <<
"REMORA git hash: " << remora_hash <<
"\n";
79 if (strlen(amrex_hash) > 0) {
80 amrex::Print() <<
"AMReX git hash: " << amrex_hash <<
"\n";
82 if (strlen(buildgithash) > 0) {
83 amrex::Print() << buildgitname <<
" git hash: " << buildgithash <<
"\n";
86 amrex::Print() <<
"\n";
101 int nlevs_max = max_level + 1;
103 istep.resize(nlevs_max, 0);
105 for (
int lev = 1; lev <= max_level; ++lev) {
111 t_new.resize(nlevs_max, 0.0_rt);
112 t_old.resize(nlevs_max, -1.e100_rt);
113 dt.resize(nlevs_max, 1.e100_rt);
129 IntVect cum_ref_ratio = IntVect(1,1,0);
133 for (
int lev = 0; lev < max_level; ++lev)
135 amrex::Print() <<
"Refinement ratio at level " << lev <<
" set to be " <<
136 ref_ratio[lev][0] <<
" " << ref_ratio[lev][1] <<
" " << ref_ratio[lev][2] << std::endl;
138 if (ref_ratio[lev][2] != 1)
140 amrex::Print() <<
"********************************************************************************" << std::endl;
141 amrex::Print() <<
"We don't allow refinement in the vertical -- make sure to set ref_ratio = 1 in z" << std::endl;
142 amrex::Print() <<
"It's possible you set amr.ref_ratio when you meant to set amr.ref_ratio_vect " << std::endl;
143 amrex::Print() <<
"********************************************************************************" << std::endl;
147 cum_ref_ratio[0] *= ref_ratio[lev][0];
148 cum_ref_ratio[1] *= ref_ratio[lev][1];
153REMORA::REMORA (
const amrex::RealBox& rb,
int max_level_in,
const amrex::Vector<int>& n_cell_in,
int coord,
const amrex::Vector<amrex::IntVect>& ref_ratio_in,
const amrex::Array<int,AMREX_SPACEDIM>& is_per, std::string prefix)
154 :
amrex::AmrCore (rb, max_level_in, n_cell_in, coord, ref_ratio_in, is_per)
156 BL_PROFILE(
"REMORA::REMORA(explicit)");
159 if (ParallelDescriptor::IOProcessor()) {
162 const char* buildgithash = amrex::buildInfoGetBuildGitHash();
163 const char* buildgitname = amrex::buildInfoGetBuildGitName();
165 if (strlen(remora_hash) > 0) {
166 amrex::Print() <<
"\n"
167 <<
"REMORA git hash: " << remora_hash <<
"\n";
169 if (strlen(amrex_hash) > 0) {
170 amrex::Print() <<
"AMReX git hash: " << amrex_hash <<
"\n";
172 if (strlen(buildgithash) > 0) {
173 amrex::Print() << buildgitname <<
" git hash: " << buildgithash <<
"\n";
176 amrex::Print() <<
"\n";
186 int nlevs_max = max_level + 1;
188 istep.resize(nlevs_max, 0);
190 for (
int lev = 1; lev <= max_level; ++lev) {
196 t_new.resize(nlevs_max, 0.0_rt);
197 t_old.resize(nlevs_max, -1.e100_rt);
198 dt.resize(nlevs_max, 1.e100_rt);
213 for (
int lev = 0; lev < max_level; ++lev)
215 amrex::Print() <<
"Refinement ratio at level " << lev <<
" set to be " <<
216 ref_ratio[lev][0] <<
" " << ref_ratio[lev][1] <<
" " << ref_ratio[lev][2] << std::endl;
218 if (ref_ratio[lev][2] != 1)
220 amrex::Print() <<
"********************************************************************************" << std::endl;
221 amrex::Print() <<
"We don't allow refinement in the vertical -- make sure to set ref_ratio = 1 in z" << std::endl;
222 amrex::Print() <<
"It's possible you set amr.ref_ratio when you meant to set amr.ref_ratio_vect " << std::endl;
223 amrex::Print() <<
"********************************************************************************" << std::endl;
241 for (
int i = 1; i <
nscalar; ++i) {
242 cons_names.emplace_back(
"tracer_" + std::to_string(i));
249 BL_PROFILE_VAR(
"REMORA::Evolve()",evolve);
250 Real cur_time =
t_new[0];
256 amrex::Print() <<
"\nCoarse STEP " << step+1 <<
" starts ..." << std::endl;
262 auto dEvolveTime0 = amrex::second();
264 if (max_level == 0) {
273 amrex::Print() <<
"Coarse STEP " << step+1 <<
" ends." <<
" TIME = " << cur_time
274 <<
" DT = " <<
dt[0] << std::endl;
278 auto dEvolveTime = amrex::second() - dEvolveTime0;
279 ParallelDescriptor::ReduceRealMax(dEvolveTime,ParallelDescriptor::IOProcessorNumber());
280 amrex::Print() <<
"Timestep time = " << dEvolveTime <<
" seconds." <<
'\n';
287#ifdef AMREX_MEM_PROFILING
289 std::ostringstream ss;
290 ss <<
"[STEP " << step+1 <<
"]";
291 MemProfiler::report(ss.str());
298 BL_PROFILE_VAR_STOP(evolve);
346 BL_PROFILE(
"REMORA::post_timestep()");
348#ifdef REMORA_USE_PARTICLES
349 particleData.Redistribute();
354 for (
int lev = finest_level-1; lev >= 0; lev--)
378 BL_PROFILE(
"REMORA::InitData()");
408#ifdef REMORA_USE_MOAB
414 for (
int lev = 1; lev <= finest_level; lev++)
416 advflux_reg[lev].reset(
new YAFluxRegister(grids[lev], grids[lev-1],
417 dmap[lev], dmap[lev-1],
418 geom[lev], geom[lev-1],
419 ref_ratio[lev-1], lev,
ncons));
424 for (
int lev = 0; lev <= finest_level; ++lev)
431 FillPatch(lev,
t_new[lev], *
cons_new[lev],
cons_new,
BCVars::cons_bc,
BdyVars::t, 0,
true,
false,0,0,0.0,*
cons_new[lev]);
432 FillPatch(lev,
t_new[lev], *
xvel_new[lev],
xvel_new,
xvel_bc(),
BdyVars::u, 0,
true,
false,0,0,0.0,*
xvel_new[lev]);
433 FillPatch(lev,
t_new[lev], *
yvel_new[lev],
yvel_new,
yvel_bc(),
BdyVars::v, 0,
true,
false,0,0,0.0,*
yvel_new[lev]);
483 BL_PROFILE(
"REMORA::Construct_REMORAFillPatchers()");
484 amrex::Print() <<
":::Construct_REMORAFillPatchers " << lev << std::endl;
486 auto& ba_fine =
cons_new[lev ]->boxArray();
487 auto& ba_crse =
cons_new[lev-1]->boxArray();
488 auto& dm_fine =
cons_new[lev ]->DistributionMap();
489 auto& dm_crse =
cons_new[lev-1]->DistributionMap();
491 BoxList bl2d_fine = ba_fine.boxList();
492 for (
auto& b : bl2d_fine) {
495 BoxArray ba2d_fine(std::move(bl2d_fine));
497 BoxList bl2d_crse = ba_crse.boxList();
498 for (
auto& b : bl2d_crse) {
501 BoxArray ba2d_crse(std::move(bl2d_crse));
505 FPr_c.emplace_back(ba_fine, dm_fine, geom[lev] ,
506 ba_crse, dm_crse, geom[lev-1],
508 FPr_u.emplace_back(convert(ba_fine, IntVect(1,0,0)), dm_fine, geom[lev] ,
509 convert(ba_crse, IntVect(1,0,0)), dm_crse, geom[lev-1],
511 FPr_v.emplace_back(convert(ba_fine, IntVect(0,1,0)), dm_fine, geom[lev] ,
512 convert(ba_crse, IntVect(0,1,0)), dm_crse, geom[lev-1],
514 FPr_w.emplace_back(convert(ba_fine, IntVect(0,0,1)), dm_fine, geom[lev] ,
515 convert(ba_crse, IntVect(0,0,1)), dm_crse, geom[lev-1],
518 FPr_ubar.emplace_back(convert(ba2d_fine, IntVect(1,0,0)), dm_fine, geom[lev] ,
519 convert(ba2d_crse, IntVect(1,0,0)), dm_crse, geom[lev-1],
521 FPr_vbar.emplace_back(convert(ba2d_fine, IntVect(0,1,0)), dm_fine, geom[lev] ,
522 convert(ba2d_crse, IntVect(0,1,0)), dm_crse, geom[lev-1],
532 BL_PROFILE(
"REMORA::Define_REMORAFillPatchers()");
533 amrex::Print() <<
":::Define_REMORAFillPatchers " << lev << std::endl;
535 auto& ba_fine =
cons_new[lev ]->boxArray();
536 auto& ba_crse =
cons_new[lev-1]->boxArray();
537 auto& dm_fine =
cons_new[lev ]->DistributionMap();
538 auto& dm_crse =
cons_new[lev-1]->DistributionMap();
540 BoxList bl2d_fine = ba_fine.boxList();
541 for (
auto& b : bl2d_fine) {
544 BoxArray ba2d_fine(std::move(bl2d_fine));
546 BoxList bl2d_crse = ba_crse.boxList();
547 for (
auto& b : bl2d_crse) {
550 BoxArray ba2d_crse(std::move(bl2d_crse));
555 FPr_c[lev-1].Define(ba_fine, dm_fine, geom[lev] ,
556 ba_crse, dm_crse, geom[lev-1],
558 FPr_u[lev-1].Define(convert(ba_fine, IntVect(1,0,0)), dm_fine, geom[lev] ,
559 convert(ba_crse, IntVect(1,0,0)), dm_crse, geom[lev-1],
561 FPr_v[lev-1].Define(convert(ba_fine, IntVect(0,1,0)), dm_fine, geom[lev] ,
562 convert(ba_crse, IntVect(0,1,0)), dm_crse, geom[lev-1],
564 FPr_w[lev-1].Define(convert(ba_fine, IntVect(0,0,1)), dm_fine, geom[lev] ,
565 convert(ba_crse, IntVect(0,0,1)), dm_crse, geom[lev-1],
568 FPr_ubar[lev-1].Define(convert(ba2d_fine, IntVect(1,0,0)), dm_fine, geom[lev] ,
569 convert(ba2d_crse, IntVect(1,0,0)), dm_crse, geom[lev-1],
571 FPr_vbar[lev-1].Define(convert(ba2d_fine, IntVect(0,1,0)), dm_fine, geom[lev] ,
572 convert(ba2d_crse, IntVect(0,1,0)), dm_crse, geom[lev-1],
579 BL_PROFILE(
"REMORA::restart()");
592 BL_PROFILE(
"REMORA::set_zeta()");
598#ifdef REMORA_USE_NETCDF
599 amrex::Print() <<
"Calling init_zeta_from_netcdf on level " << lev << std::endl;
601 amrex::Print() <<
"Sea surface height loaded from netcdf file \n " << std::endl;
604 amrex::Abort(
"Unknown IC_Type");
609 vec_zeta[lev]->FillBoundary(geom[lev].periodicity());
615 Real dummy_time = 0.0_rt;
619 vec_zeta[lev]->FillBoundary(geom[lev].periodicity());
631 BL_PROFILE(
"REMORA::bathymetry()");
643#ifdef REMORA_USE_NETCDF
644 amrex::Print() <<
"Calling init_bathymetry_from_netcdf " << std::endl;
646 amrex::Print() <<
"Bathymetry loaded from netcdf file \n " << std::endl;
647 amrex::Print() <<
"Calling init_grid_vars_from_netcdf " << std::endl;
649 amrex::Print() <<
"Grid variables loaded from netcdf file \n " << std::endl;
652 amrex::Abort(
"Unknown IC_Type");
660 vec_h[lev]->FillBoundary(geom[lev].periodicity());
661 vec_h[lev]->EnforcePeriodicity(geom[lev].periodicity());
667 Real dummy_time = 0.0_rt;
671 vec_h[lev]->FillBoundary(geom[lev].periodicity());
672 vec_h[lev]->EnforcePeriodicity(geom[lev].periodicity());
683 Real dummy_time = 0.0_rt;
700 Real dummy_time = 0.0_rt;
721 0,
false,
false,0,0,0.0,*
vec_zeta[lev]);
736 FillPatch(lev,
t_new[lev], *
cons_new[lev],
cons_new,
BCVars::cons_bc,
BdyVars::t, 0,
true,
false,0,0,0.0,*
cons_new[lev]);
737 FillPatch(lev,
t_new[lev], *
xvel_new[lev],
xvel_new,
xvel_bc(),
BdyVars::u, 0,
true,
false,0,0,0.0,*
xvel_new[lev]);
738 FillPatch(lev,
t_new[lev], *
yvel_new[lev],
yvel_new,
yvel_bc(),
BdyVars::v, 0,
true,
false,0,0,0.0,*
yvel_new[lev]);
746 BL_PROFILE(
"REMORA::set_coriolis()");
752#ifdef REMORA_USE_NETCDF
755 amrex::Print() <<
"Calling init_coriolis_from_netcdf " << std::endl;
757 amrex::Print() <<
"Coriolis loaded from netcdf file \n" << std::endl;
759 Real dummy_time = 0.0_rt;
764 Abort(
"Don't know this coriolis_type!");
769 vec_fcor[lev]->EnforcePeriodicity(geom[lev].periodicity());
775 BL_PROFILE(
"REMORA::init_set_vmix()");
783 Abort(
"Don't know this vertical mixing type");
792 BL_PROFILE(
"REMORA::set_analytic_vmix()");
798 for (
int n = 0; n <
ncons; n++) {
813#ifdef REMORA_USE_NETCDF
815 amrex::Print() <<
"Calling init_masks_from_netcdf level " << lev << std::endl;
817 amrex::Print() <<
"Masks loaded from netcdf file \n " << std::endl;
819 Real dummy_time = 0.0_rt;
835 BL_PROFILE(
"REMORA::set_hmixcoef()");
840 Real lev_scale = 1.0_rt;
843 for (
int l = 0; l < lev; ++l) {
844 rf *= std::sqrt(
static_cast<Real
>(ref_ratio[l][0]) *
static_cast<Real
>(ref_ratio[l][1]));
846 lev_scale = 1.0_rt / rf;
856 for (
int n = 0; n <
ncons; n++) {
889 for (
int n = 0; n <
ncons; n++) {
897 amrex::Gpu::LaunchSafeGuard lsg(
true);
898 Real denom_min = amrex::ReduceMin(*
vec_pm[lev], *
vec_pn[lev], 0,
899 [=] AMREX_GPU_HOST_DEVICE (Box
const& bx,
900 Array4<Real const>
const& pm,
901 Array4<Real const>
const& pn) -> Real
903 Real local_min = 1.0e200_rt;
904 amrex::Loop(bx, [=,&local_min] (
int i,
int j,
int)
noexcept
906 local_min = amrex::min(local_min, pm(i,j,0) * pn(i,j,0));
911 ParallelDescriptor::ReduceRealMin(denom_min);
912 if (denom_min <= 0.0_rt) {
913 Abort(
"scaled_to_grid: found non-positive pm*pn (grid metrics must be > 0)");
916 Real grdmax = amrex::ReduceMax(*
vec_pm[lev], *
vec_pn[lev], 0,
917 [=] AMREX_GPU_HOST_DEVICE (Box
const& bx,
918 Array4<Real const>
const& pm,
919 Array4<Real const>
const& pn) -> Real
921 Real local_max = 0.0_rt;
922 amrex::Loop(bx, [=,&local_max] (
int i,
int j,
int)
noexcept
924 Real denom = pm(i,j,0) * pn(i,j,0);
925 if (denom > 0.0_rt) {
926 Real G = std::sqrt(1.0_rt / denom);
927 local_max = amrex::max(local_max, G);
933 ParallelDescriptor::ReduceRealMax(grdmax);
934 if (grdmax <= 0.0_rt) {
935 Abort(
"scaled_to_grid: grdmax <= 0");
944 for (
int l = 0; l < lev; ++l) {
945 rf *= std::sqrt(
static_cast<Real
>(ref_ratio[l][0]) *
static_cast<Real
>(ref_ratio[l][1]));
947 lev_scale = 1.0_rt / rf;
951 Real cff = visc0 / grdmax;
956 amrex::Gpu::DeviceVector<Real> diff0_d(
ncons);
957 amrex::Gpu::copy(amrex::Gpu::hostToDevice,
960 Real
const* diff0_ptr = diff0_d.data();
962 for (MFIter mfi(*
vec_visc2_r[lev], TilingIfNotGPU()); mfi.isValid(); ++mfi)
964 const Box& bx = mfi.validbox();
965 auto pm =
vec_pm[lev]->const_array(mfi);
966 auto pn =
vec_pn[lev]->const_array(mfi);
970 int ncons_local =
ncons;
971 ParallelFor(makeSlab(bx,2,0), [=] AMREX_GPU_DEVICE (
int i,
int j,
int)
noexcept
973 Real denom = pm(i,j,0) * pn(i,j,0);
974 Real grdscl = (denom > 0.0_rt) ? std::sqrt(1.0_rt / denom) : 0.0_rt;
975 visc2_r(i,j,0) = cff * grdscl;
977 for (
int n = 0; n < ncons_local; n++) {
978 diff2(i,j,0,n) = ((diff0_ptr[n] * lev_scale) / grdmax) * grdscl;
990 for (MFIter mfi(*
vec_visc2_p[lev], TilingIfNotGPU()); mfi.isValid(); ++mfi)
992 const Box& bx = mfi.validbox();
996 ParallelFor(makeSlab(bx,2,0), [=] AMREX_GPU_DEVICE (
int i,
int j,
int)
noexcept
998 visc2_p(i,j,0) = 0.25_rt * (
1015 Real visc_min_all =
vec_visc2_r[lev]->min(0,0,
false);
1016 Real visc_max_all =
vec_visc2_r[lev]->max(0,0,
false);
1019 amrex::Gpu::LaunchSafeGuard lsg_diag(
true);
1021 [=] AMREX_GPU_HOST_DEVICE (Box
const& bx,
1022 Array4<Real const>
const& visc2,
1023 Array4<Real const>
const& mskr) -> Real
1025 Real local_min = 1.0e200_rt;
1026 amrex::Loop(bx, [=,&local_min] (
int i,
int j,
int)
noexcept
1028 if (mskr(i,j,0) > 0.0_rt) {
1029 local_min = amrex::min(local_min, visc2(i,j,0));
1034 ParallelDescriptor::ReduceRealMin(visc_min_wet);
1037 [=] AMREX_GPU_HOST_DEVICE (Box
const& bx,
1038 Array4<Real const>
const& visc2,
1039 Array4<Real const>
const& mskr) -> Real
1041 Real local_max = -1.0e200_rt;
1042 amrex::Loop(bx, [=,&local_max] (
int i,
int j,
int)
noexcept
1044 if (mskr(i,j,0) > 0.0_rt) {
1045 local_max = amrex::max(local_max, visc2(i,j,0));
1050 ParallelDescriptor::ReduceRealMax(visc_max_wet);
1054 [=] AMREX_GPU_HOST_DEVICE (Box
const& bx,
1055 Array4<Real const>
const& visc2,
1056 Array4<Real const>
const& mskr) -> Real
1058 Real local_min = 1.0e200_rt;
1059 amrex::Loop(bx, [=,&local_min] (
int i,
int j,
int)
noexcept
1061 const Real v = (mskr(i,j,0) > 0.0_rt) ? visc2(i,j,0) : 0.0_rt;
1062 local_min = amrex::min(local_min, v);
1066 ParallelDescriptor::ReduceRealMin(visc_min_mask0);
1069 [=] AMREX_GPU_HOST_DEVICE (Box
const& bx,
1070 Array4<Real const>
const& visc2,
1071 Array4<Real const>
const& mskr) -> Real
1073 Real local_max = -1.0e200_rt;
1074 amrex::Loop(bx, [=,&local_max] (
int i,
int j,
int)
noexcept
1076 const Real v = (mskr(i,j,0) > 0.0_rt) ? visc2(i,j,0) : 0.0_rt;
1077 local_max = amrex::max(local_max, v);
1081 ParallelDescriptor::ReduceRealMax(visc_max_mask0);
1082 if (ParallelDescriptor::IOProcessor() && lev == 0)
1084 Print() <<
"\nHorizontal mixing scaled by grid metric\n";
1085 Print() <<
"grdmax = " << grdmax <<
"\n";
1087 Print() <<
"AMR scaling (linear) lev_scale = " << lev_scale <<
"\n";
1089 Print() <<
"visc2(all) min/max = "
1090 << visc_min_all <<
" / "
1091 << visc_max_all <<
"\n";
1092 Print() <<
"visc2(wet,k=0) min/max = "
1093 << visc_min_wet <<
" / "
1094 << visc_max_wet <<
"\n";
1095 Print() <<
"visc2(mask->0) min/max = "
1096 << visc_min_mask0 <<
" / "
1097 << visc_max_mask0 <<
"\n";
1101 Abort(
"Don't know this horizontal mixing type");
1108 for (
int n = 0; n <
ncons; n++) {
1120 BL_PROFILE(
"REMORA::init_flat_bathymetry()");
1121 vec_h[lev]->setVal(-geom[0].ProbLo()[2]);
1130 BL_PROFILE(
"REMORA::set_smflux()");
1134#ifdef REMORA_USE_NETCDF
1149 BL_PROFILE(
"REMORA::set_wind()");
1156 if (!(driver_has_uwind && driver_has_vwind)) {
1159 if (
vec_uwind[lev] !=
nullptr) {
vec_uwind[lev]->FillBoundary(geom[lev].periodicity()); }
1160 if (
vec_vwind[lev] !=
nullptr) {
vec_vwind[lev]->FillBoundary(geom[lev].periodicity()); }
1162#ifdef REMORA_USE_NETCDF
1163 if (!driver_has_uwind) {
1168 if (
vec_uwind[lev] !=
nullptr) {
vec_uwind[lev]->FillBoundary(geom[lev].periodicity()); }
1170 if (!driver_has_vwind) {
1175 if (
vec_vwind[lev] !=
nullptr) {
vec_vwind[lev]->FillBoundary(geom[lev].periodicity()); }
1184 vec_Tair[lev]->FillBoundary(geom[lev].periodicity());
1196 vec_qair[lev]->FillBoundary(geom[lev].periodicity());
1199 vec_qair[lev]->FillBoundary(geom[lev].periodicity());
1206 vec_Pair[lev]->FillBoundary(geom[lev].periodicity());
1213 vec_srflx[lev]->FillBoundary(geom[lev].periodicity());
1227 vec_rain[lev]->FillBoundary(geom[lev].periodicity());
1234 vec_cloud[lev]->FillBoundary(geom[lev].periodicity());
1241 vec_EminusP[lev]->FillBoundary(geom[lev].periodicity());
1245 amrex::Abort(
"Unknown wind_type in REMORA::set_wind()");
1256 BL_PROFILE(
"REMORA::init_only()");
1269 vec_ru[lev]->setVal(0.0_rt);
1270 vec_rv[lev]->setVal(0.0_rt);
1280#ifdef REMORA_USE_NETCDF
1286 amrex::Error(
"NetCDF climatology file name must be provided via input");
1316 amrex::Print() <<
"Calling init_bdry_from_netcdf at level " << lev << std::endl;
1318 amrex::Print() <<
"Boundary data loaded from netcdf file \n " << std::endl;
1325 amrex::Error(
"NetCDF forcing file name must be provided via input for winds");
1338 amrex::Error(
"NetCDF forcing file name must be provided via input for surface momentum fluxes");
1406 amrex::Error(
"NetCDF forcing file name must be provided via input for longwave radiation");
1420 amrex::Error(
"NetCDF river file name must be provided via input for rivers");
1422 auto dom = geom[0].Domain();
1423 int nz = dom.length(2);
1445 amrex::Print() <<
"Reading high resolution bathymetry and grid data" << std::endl;
1449 amrex::Print() <<
"Done reading in high resolution bathymetry and grid data" << std::endl;
1452 amrex::Print() <<
"Reading high resolution initial data" << std::endl;
1456 amrex::Print() <<
"Done reading in high resolution initial data" << std::endl;
1460 Abort(
"Not compiled with NetCDF, but selected boundary conditions require NetCDF");
1463 Abort(
"Not compiled with NetCDF, but using river sources requires NetCDF");
1490#ifdef REMORA_USE_NETCDF
1491 amrex::Print() <<
"Calling init_data_from_netcdf " << std::endl;
1494 amrex::Print() <<
"Initial data loaded from netcdf file \n " << std::endl;
1497 amrex::Abort(
"Unknown IC_Type");
1525 xvel_new[lev]->OverrideSync(geom[lev].periodicity());
1526 yvel_new[lev]->OverrideSync(geom[lev].periodicity());
1527 zvel_new[lev]->OverrideSync(geom[lev].periodicity());
1546 BL_PROFILE(
"REMORA::ReadParameters()");
1549 bool noprefix_max_step = pp.queryAdd(
"max_step",
max_step);
1550 bool noprefix_stop_time = pp.queryAdd(
"stop_time",
stop_time);
1551 bool remora_max_step = pp.queryAdd(
"remora.max_step",
max_step);
1552 bool remora_stop_time = pp.queryAdd(
"remora.stop_time",
stop_time);
1553 if (remora_max_step and noprefix_max_step) {
1554 Abort(
"remora.max_step and max_step are both specified. Please use only one!");
1556 if (remora_stop_time and noprefix_stop_time) {
1557 Abort(
"remora.stop_time and stop_time are both specified. Please use only one!");
1564 pp.queryAdd(
"nscalar",
nscalar);
1566 amrex::Abort(
"remora.nscalar must be at least 1");
1586 if (pp.contains(
"data_log")) {
1587 int num_datalogs = pp.countval(
"data_log");
1590 pp.queryarr(
"data_log",
datalogname, 0, num_datalogs);
1591 for (
int i = 0; i < num_datalogs; i++)
1597 pp.queryAdd(
"sum_period",
sum_per);
1601 amrex::Abort(
"remora.file_min_digits must be non-negative");
1604 pp.queryAdd(
"cfl",
cfl);
1612 amrex::Abort(
"Dt is over-specfied");
1632 std::string plotfile_type_str =
"amrex";
1633 pp.queryAdd(
"plotfile_type", plotfile_type_str);
1634 if (plotfile_type_str ==
"amrex") {
1636 }
else if (plotfile_type_str ==
"netcdf" || plotfile_type_str ==
"NetCDF") {
1638#ifdef REMORA_USE_NETCDF
1643 auto dom = geom[0].Domain();
1644 int nx = dom.length(0) + 2;
1645 int ny = dom.length(1) + 2;
1646 int nz = dom.length(2);
1650 / (nx * ny * 64.0_rt * (
NC3D*nz +
NC2D)));
1655 amrex::Warning(
"NetCDF output for a single timestep appears to exceed 2GB. NetCDF output may not work. See Documentation for information about tested MPICH versions.");
1661 Real est_hist_file_size =
NCH2D * nx * ny * 64.0_rt + nt_out * nx * ny * 64.0_rt * (
NC3D*nz +
NC2D);
1662 if (est_hist_file_size > 1.6e10) {
1663 amrex::Warning(
"WARNING: NetCDF history file may be larger than 2GB limit. Consider setting remora.chunk_history_file=true");
1667 Print() <<
"NetCDF history files will have " <<
steps_per_history_file <<
" steps per file." << std::endl;
1671 amrex::Print() <<
"User selected plotfile_type = " << plotfile_type_str << std::endl;
1672 amrex::Abort(
"Dont know this plotfile_type");
1674#ifndef REMORA_USE_NETCDF
1677 amrex::Abort(
"Please compile with NetCDF in order to enable NetCDF plotfiles");
1681#ifdef REMORA_USE_NETCDF
1691 for (
int lev = 0; lev <= max_level; lev++)
1693 const std::string nc_file_names = amrex::Concatenate(
"nc_init_file_",lev,1);
1694 const std::string nc_bathy_file_names = amrex::Concatenate(
"nc_grid_file_",lev,1);
1696 if (pp.contains(nc_file_names.c_str()))
1698 int num_files = pp.countval(nc_file_names.c_str());
1699 int num_bathy_files = pp.countval(nc_bathy_file_names.c_str());
1700 if (num_files != num_bathy_files) {
1701 amrex::Error(
"Must have same number of netcdf files for grid info as for solution");
1708 pp.queryarr(nc_file_names.c_str() ,
nc_init_file[lev] ,0,num_files);
1709 pp.queryarr(nc_bathy_file_names.c_str(),
nc_grid_file[lev],0,num_files);
1720 if (pp.contains(
"nc_frc_file")) {
1721 int num_files = pp.countval(
"nc_frc_file");
1723 pp.queryarr(
"nc_frc_file",
nc_frc_file, 0, num_files);
1727 if (pp.contains(
"nc_river_file")) {
1728 int num_files = pp.countval(
"nc_river_file");
1730 pp.queryarr(
"nc_river_file",
nc_riv_file, 0, num_files);
1734 if (pp.contains(
"nc_clim_his_file")) {
1735 int num_files = pp.countval(
"nc_clim_his_file");
1774 amrex::Abort(
"hires_grid_level must be less than or equal to amr.max_level");
1778 amrex::Abort(
"hires_init_level must be less than or equal to amr.max_level");
1780#ifdef REMORA_USE_PARTICLES
1781 readTracersParams();
1785 ParmParse pp_amr(
"amr");
1789 amrex::Abort(
"Time substepping is not yet implemented. amr.do_substep must be 0");
1802 amrex::Abort(
"Cannot do high-resolution initialization for analytic initial conditions. Not yet implemented");
1811 BL_PROFILE(
"REMORA::AverageDown()");
1812 for (
int lev = finest_level-1; lev >= 0; --lev)
1824 BL_PROFILE(
"REMORA::AverageDownTo()");
1826 0,
cons_new[crse_lev]->nComp(), refRatio(crse_lev));
1828 0,
vec_Zt_avg1[crse_lev]->nComp(), refRatio(crse_lev));
1830 Array<MultiFab*,AMREX_SPACEDIM> faces_crse;
1831 Array<MultiFab*,AMREX_SPACEDIM> faces_fine;
1832 faces_crse[0] =
xvel_new[crse_lev];
1833 faces_crse[1] =
yvel_new[crse_lev];
1834 faces_crse[2] =
zvel_new[crse_lev];
1836 faces_fine[0] =
xvel_new[crse_lev+1];
1837 faces_fine[1] =
yvel_new[crse_lev+1];
1838 faces_fine[2] =
zvel_new[crse_lev+1];
1840 average_down_faces(GetArrOfConstPtrs(faces_fine), faces_crse,
1841 refRatio(crse_lev),geom[crse_lev]);
1852 auto const& crsema = vec_mf[crse_lev]->arrays();
1853 auto const& finema = vec_mf[crse_lev+1]->const_arrays();
1854 auto ref_ratio_crse = refRatio(crse_lev);
1855 auto index_type = (vec_mf[crse_lev]->boxArray().ixType()).toIntVect();
1857 if (index_type[0]==0 and index_type[1]==0) {
1858 ParallelFor(*vec_mf[crse_lev], nghost_crse, vec_mf[crse_lev]->nComp(),
1859 [=] AMREX_GPU_DEVICE (
int box_no,
int i,
int j,
int k,
int n)
noexcept
1861 amrex_avgdown(i,j,k,n,crsema[box_no],finema[box_no],0,0,ref_ratio_crse);
1863 }
else if (index_type[0]==1 and index_type[1]==0) {
1864 ParallelFor(*vec_mf[crse_lev], nghost_crse, 1,
1865 [=] AMREX_GPU_DEVICE (
int box_no,
int i,
int j,
int k,
int n)
noexcept
1867 amrex_avgdown_faces(i,j,k,n,crsema[box_no],finema[box_no],0,0,ref_ratio_crse,0);
1869 }
else if (index_type[0]==0 and index_type[1]==1) {
1870 ParallelFor(*vec_mf[crse_lev], nghost_crse, 1,
1871 [=] AMREX_GPU_DEVICE (
int box_no,
int i,
int j,
int k,
int n)
noexcept
1873 amrex_avgdown_faces(i,j,k,n,crsema[box_no],finema[box_no],0,0,ref_ratio_crse,1);
1876 amrex::Abort(
"Unexpected nodality in average_down_with_grow_cells");
1878 Gpu::streamSynchronize();
PlotfileType
plotfile format
std::unique_ptr< ProblemBase > amrex_probinit(const amrex_real *problo, const amrex_real *probhi) AMREX_ATTRIBUTE_WEAK
Function to init the physical bounds of the domain and instantiate a Problem derived from ProblemBase...
A class to hold and interpolate time series data read from a NetCDF file.
static PlotfileType plotfile_type
Native or NetCDF plotfile output.
std::string nc_grid_file_hires
Grid file for high resolution bathymetry.
void FillCoarsePatchPC(int lev, amrex::Real time, amrex::MultiFab *mf_fine, amrex::MultiFab *mf_crse, const int bccomp, const int bdy_var_type=BdyVars::null, const int icomp=0, const bool fill_all=true, const int n_not_fill=0, const int icomp_calc=0, const amrex::Real dt=amrex::Real(0.0), const amrex::MultiFab &mf_calc=amrex::MultiFab())
fill an entire multifab by interpolating from the coarser level using the piecewise constant interpol...
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_EminusP
evaporation minus precipitation [kg/m^2/s], defined at rho-points
amrex::Vector< std::string > nc_riv_file
NetCDF river file(s)
void set_grid_vars_averaged_down(int lev)
Set pm/pn by averaging down from higher-resolution grid.
std::string riv_time_varname
Name of time field for river time.
int foextrap_periodic_bc() const noexcept
amrex::Vector< std::string > nc_clim_his_file
NetCDF climatology history file(s)
int ncons
Number of conserved scalars in the state (temperature + salt + passive scalars)
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_zeta_full_domain
high resolution initial free surface height (2D)
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_rv2d
v velocity RHS (2D, includes horizontal and vertical advection)
std::string nc_init_file_hires
Init file for high resolution.
static amrex::Real fixed_dt
User specified fixed baroclinic time step.
amrex::Real last_plot_file_time
Simulation time when we last output a plotfile.
int zvel_bc() const noexcept
void init_full_domain_zeta_from_analytic()
Initialize high resolution initial sea surface height from analytic functions.
static bool plot_staggered_vels
Whether to write the staggered velocities (not averaged to cell centers)
void init_bathymetry_from_netcdf(int lev)
Bathymetry data initialization from NetCDF file.
void init_bcs()
Read in boundary parameters from input file and set up data structures.
int xvel_bc() const noexcept
void set_zeta_averaged_down(int lev)
Copy over zeta data that has been averaged down from high res.
void calculate_nodal_masks(int lev)
Calculate u-, v-, and psi-point masks based on rho-point masks after analytic initialization.
std::unique_ptr< NCTimeSeries > qair_data_from_file
Data container for specific humidity read from file.
static amrex::Real previousCPUTimeUsed
Accumulator variable for CPU time used thusfar.
amrex::Vector< std::string > cons_names
Names of scalars for plotfile output.
amrex::Vector< std::unique_ptr< amrex::YAFluxRegister > > advflux_reg
array of flux registers for refluxing in multilevel
std::unique_ptr< NCTimeSeries > sustr_data_from_file
Data container for u-component surface momentum flux read from file.
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_fcor
coriolis factor (2D)
void allocate_init_full_domain()
Allocate multifabs for storing full-domain high resolution initial data.
void init_gls_vmix(int lev, SolverChoice solver_choice)
Initialize GLS variables.
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_xvel_full_domain
multilevel data container for high res initial x velocities (u in ROMS)
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_h
multilevel data container for current step's z velocities (largely unused; W stored separately)
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_pm
horizontal scaling factor: 1 / dx (2D)
void set2DPlotVariables(const std::string &pp_plot_var_names_2d)
amrex::Vector< REMORAFillPatcher > FPr_v
Vector over levels of FillPatchers for v (3D)
void init_zeta_full_domain_from_netcdf()
Full-domain high res sea-surface height data initialization from NetCDF file.
amrex::Vector< amrex::MultiFab * > cons_new
multilevel data container for current step's scalar data: temperature, salinity, passive tracer
static bool write_history_file
Whether to output NetCDF files as a single history file with several time steps.
void FillCoarsePatch(int lev, amrex::Real time, amrex::MultiFab *mf_fine, amrex::MultiFab *mf_crse, const int bccomp, const int bdy_var_type=BdyVars::null, const int icomp=0, const bool fill_all=true, const int n_not_fill=0, const int icomp_calc=0, const amrex::Real dt=amrex::Real(0.0), const amrex::MultiFab &mf_calc=amrex::MultiFab())
fill an entire multifab by interpolating from the coarser level
void stretch_transform(int lev)
Calculate vertical stretched coordinates.
std::unique_ptr< NCTimeSeries > rain_data_from_file
Data container for precipitation rate read from file.
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_vwind
Wind in the v direction, defined at rho-points.
std::unique_ptr< ProblemBase > prob
Pointer to container of analytical functions for problem definition.
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_mskr
land/sea mask at cell centers (2D)
void Construct_REMORAFillPatchers(int lev)
Construct FillPatchers.
void init_grid_vars_from_netcdf(int lev)
Grid variable initialization from NetCDF file.
static int sum_interval
Diagnostic sum output interval in number of steps.
int history_count
Counter for which time index we are writing to in the netcdf history file.
amrex::Real stop_time
Time to stop.
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_rain
precipitation rate [kg/m^2/s]
int do_substep
Whether to substep fine levels in time.
void Evolve()
Advance solution to final time.
std::string bdry_time_varname
Default name of time field for boundary data.
amrex::Real plotfile_fill_value
fill value for masked arrays in amrex plotfiles
void ReadCheckpointFile()
read checkpoint file from disk
bool chunk_history_file
Whether to chunk netcdf history file.
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_sustr
Surface stress in the u direction.
amrex::Real get_t_old(int lev) const
Accessor method for t_old to expose to outside classes.
int yvel_bc() const noexcept
std::unique_ptr< NCTimeSeries > longwave_down_data_from_file
Data container for downward longwave radiation flux read from file.
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_ru2d
u velocity RHS (2D, includes horizontal and vertical advection)
amrex::Vector< std::string > datalogname
amrex::Vector< amrex::MultiFab * > zvel_new
multilevel data container for current step's z velocities (largely unused; W stored separately)
void WriteAtIntermediateTime(int step, amrex::Real cur_time)
Write checkpoint and plotfiles at intermediate point of simulation, if needed.
void init_only(int lev, amrex::Real time)
Init (NOT restart or regrid)
void init_set_vmix(int lev)
Initialize vertical mixing coefficients from file or analytic.
std::unique_ptr< NCTimeSeries > v_clim_data_from_file
Data container for v-velocity climatology data read from file.
std::string clim_u_time_varname
Name of time field for u climatology data.
void set_grid_scale(int lev)
Set pm and pn arrays and x/y coords on level lev.
void set_coriolis(int lev)
Initialize Coriolis factor from file or analytic.
int foextrap_bc() const noexcept
amrex::Vector< REMORAFillPatcher > FPr_u
Vector over levels of FillPatchers for u (3D)
std::string clim_temp_time_varname
Name of time field for temperature climatology data.
amrex::Vector< amrex::Vector< amrex::Box > > boxes_at_level
the boxes specified at each level by tagging criteria
static amrex::Vector< amrex::AMRErrorTag > ref_tags
Holds info for dynamically generated tagging criteria.
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_Akt
Vertical diffusion coefficient (3D)
std::unique_ptr< NCTimeSeriesRiver > river_source_transportbar
Data container for vertically integrated momentum transport in rivers.
std::array< bool, AtmosState::NumTypes > driver_atmos_state_from_driver
provenance flags for driver-supplied atmospheric forcing lanes
std::string clim_ubar_time_varname
Name of time field for ubar climatology data.
std::unique_ptr< NCTimeSeries > u_clim_data_from_file
Data container for u-velocity climatology data read from file.
std::string check_file
Checkpoint file prefix.
static amrex::Real startCPUTime
Variable for CPU timing.
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_pm_full_domain
horizontal scaling factor: 1 / dx (2D) on whole domain
amrex::Vector< amrex::MultiFab * > xvel_old
multilevel data container for last step's x velocities (u in ROMS)
amrex::Real start_time
Time of the start of the simulation, in seconds.
void init_data_from_netcdf(int lev)
Problem initialization from NetCDF file.
void init_masks_from_netcdf(int lev)
Mask data initialization from NetCDF file.
amrex::Vector< amrex::MultiFab * > yvel_new
multilevel data container for current step's y velocities (v in ROMS)
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_uwind
Wind in the u direction, defined at rho-points.
static bool plot_nodal_data
Whether to write nodal data (Nu_nd) to plotfiles.
static amrex::Real fixed_fast_dt
User specified fixed barotropic time step.
int regrid_int
how often each level regrids the higher levels of refinement (after a level advances that many time s...
amrex::Real check_int_time
Checkpoint output interval in seconds.
void init_scalar_metadata()
Build runtime scalar names after nscalar is known.
int zeta_bc() const noexcept
void Define_REMORAFillPatchers(int lev)
Define FillPatchers.
amrex::Vector< amrex::IntVect > cum_ref_ratios
Cumulative refinement ratio between level 0 and level i.
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_visc2_p
Harmonic viscosity defined on the psi points (corners of horizontal grid cells)
amrex::Real plot_int_time
Plotfile output interval in seconds.
amrex::Vector< int > num_files_at_level
how many netcdf input files specified at each level
amrex::Vector< REMORAFillPatcher > FPr_vbar
Vector over levels of FillPatchers for vbar (2D)
void AverageDownTo(int crse_lev)
more flexible version of AverageDown() that lets you average down across multiple levels
int steps_per_history_file
Number of time steps per netcdf history file.
void post_timestep(int nstep, amrex::Real time, amrex::Real dt_lev)
Called after every level 0 timestep.
int max_step
maximum number of steps
amrex::Vector< amrex::MultiFab * > zvel_old
multilevel data container for last step's z velocities (largely unused; W stored separately)
std::unique_ptr< NCTimeSeries > svstr_data_from_file
Data container for v-component surface momentum flux read from file.
amrex::Vector< std::string > nc_frc_file
NetCDF forcing file(s)
amrex::Vector< int > num_boxes_at_level
how many boxes specified at each level by tagging criteria
amrex::Vector< amrex::MultiFab * > xvel_new
multilevel data container for current step's x velocities (u in ROMS)
void refinement_criteria_setup()
Set refinement criteria.
int last_check_file_step
Step when we last output a checkpoint file.
void init_beta_plane_coriolis(int lev)
Calculate Coriolis parameters from beta plane parametrization.
std::string clim_vbar_time_varname
Name of time field for vbar climatology data.
amrex::Vector< int > nsubsteps
How many substeps on each level?
amrex::Vector< std::unique_ptr< REMORAPhysBCFunct > > physbcs
Vector (over level) of functors to apply physical boundary conditions.
void ComputeDt()
a wrapper for estTimeStep()
void fill_3d_masks(int lev)
Copy maskr to all z levels.
std::unique_ptr< NCTimeSeries > EminusP_data_from_file
Data container for evaporation minus precipitation read from file.
int plot_int
Plotfile output interval in iterations.
void FillPatch(int lev, amrex::Real time, amrex::MultiFab &mf_to_be_filled, amrex::Vector< amrex::MultiFab * > const &mfs, const int bccomp, const int bdy_var_type=BdyVars::null, const int icomp=0, const bool fill_all=true, const bool fill_set=true, const int n_not_fill=0, const int icomp_calc=0, const amrex::Real dt=amrex::Real(0.0), const amrex::MultiFab &mf_calc=amrex::MultiFab())
Fill a new MultiFab by copying in phi from valid region and filling ghost cells.
std::unique_ptr< NCTimeSeries > cloud_data_from_file
Data container for cloud cover fraction read from file.
void WriteAtFinalTime()
Write checkpoint and plotfiles at end of simulation.
void InitData()
Initialize multilevel data.
void set3DPlotVariables(const std::string &pp_plot_var_names_3d)
amrex::Vector< int > istep
which step?
void WriteCheckpointFile()
write checkpoint file to disk
std::string nc_clim_coeff_file
NetCDF climatology coefficient file.
void setRecordDataInfo(int i, const std::string &filename)
void set_analytic_vmix(int lev)
Set vertical mixing coefficients from analytic.
void init_flat_bathymetry(int lev)
Initialize flat bathymetry to value from problo.
std::unique_ptr< NCTimeSeries > temp_clim_data_from_file
Data container for temperature climatology data read from file.
amrex::Vector< std::string > bdry_time_name_byvar
Name of time fields for boundary data.
static int file_min_digits
Minimum number of digits in plotfile name or chunked history file.
void init_riv_pos_from_netcdf(int lev)
static amrex::Vector< std::string > nc_bdry_file
NetCDF boundary data.
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_visc2_r
Harmonic viscosity defined on the rho points (centers)
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_yvel_full_domain
multilevel data container for high res initial y velocities (v in ROMS)
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_svstr
Surface stress in the v direction.
std::unique_ptr< NCTimeSeries > srflx_data_from_file
Data container for shortwave radiation flux read from file.
void set_zeta(int lev)
Initialize zeta from file or analytic.
static amrex::Real change_max
Fraction maximum change in subsequent time steps.
void init_zeta_from_netcdf(int lev)
Sea-surface height data initialization from NetCDF file.
void set_zeta_average(int lev)
Set Zt_avg1 to zeta.
void init_coriolis_from_netcdf(int lev)
Coriolis parameter data initialization from NetCDF file.
std::string pp_prefix
default prefix for input file parameters
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_h_full_domain
Bathymetry data on the whole domain at each potential level.
void set_bathymetry(int lev)
Initialize bathymetry from file or analytic.
amrex::Vector< amrex::MultiFab * > yvel_old
multilevel data container for last step's y velocities (v in ROMS)
std::unique_ptr< NCTimeSeries > ubar_clim_data_from_file
Data container for ubar climatology data read from file.
void init_full_domain_from_analytic()
Initialize high resolution initial problem data from analytic functions.
void init_data_full_domain_from_netcdf()
High resolution roblem initialization from NetCDF file.
amrex::Vector< REMORAFillPatcher > FPr_c
Vector over levels of FillPatchers for scalars.
int hires_init_level
Which level the high resolution initialization data is at.
std::unique_ptr< NCTimeSeries > Tair_data_from_file
Data container for air temperature read from file.
int nscalar
Number of passive scalars carried in the state.
std::string clim_v_time_varname
Name of time field for v climatology data.
amrex::Vector< REMORAFillPatcher > FPr_w
Vector over levels of FillPatchers for w.
void average_down_with_grow_cells(int lev, amrex::Vector< std::unique_ptr< amrex::MultiFab > > &mf)
Average down from level lev+1 to lev in mf, including grow cells.
std::string clim_salt_time_varname
Name of time field for salinity climatology data.
std::unique_ptr< NCTimeSeries > Uwind_data_from_file
Data container for u-direction wind read from file.
std::unique_ptr< NCTimeSeries > Pair_data_from_file
Data container for air pressure read from file.
amrex::Vector< amrex::Real > t_new
new time at each level
void init_stretch_coeffs()
initialize and calculate stretch coefficients
void init_bdry_from_netcdf(int lev)
Boundary data initialization from NetCDF file.
static SolverChoice solverChoice
Container for algorithmic choices.
void set_masks(int lev)
Initialize land-sea masks from file or analytic.
amrex::Vector< amrex::Vector< std::unique_ptr< NCTimeSeriesBoundary > > > boundary_series
Vector over BdyVars of boundary series data containers.
int cf_set_width
Width for fixing values at coarse-fine interface.
void ReadParameters()
read in some parameters from inputs file
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_ru
u velocity RHS (3D, includes horizontal and vertical advection)
void sum_integrated_quantities(amrex::Real time)
Integrate conserved quantities for diagnostics.
static int total_nc_plot_file_step
static amrex::Vector< amrex::Vector< std::string > > nc_grid_file
NetCDF grid file.
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_longwave_down
Downward longwave radiation.
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_zeta
free surface height (2D)
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_vbar
barotropic y velocity (2D)
void set_zeta_to_Ztavg(int lev)
Set zeta components to be equal to time-averaged Zt_avg1.
bool expand_plotvars_to_unif_rr
whether plotfile variables should be expanded to a uniform refinement ratio
int plot_file_on_restart
Whether to output a plotfile on restart from checkpoint.
void set_2darrays(int lev)
Set 2D momentum arrays from 3D momentum.
void init_analytic(int lev)
Initialize initial problem data from analytic functions.
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_ubar
barotropic x velocity (2D)
amrex::Vector< amrex::MultiFab * > cons_old
multilevel data container for last step's scalar data: temperature, salinity, passive tracer
std::string frc_time_varname
Name of time field for forcing data.
void set_wind(int lev)
Initialize or calculate wind speed from file or analytic.
amrex::Vector< REMORAFillPatcher > FPr_ubar
Vector over levels of FillPatchers for ubar (2D)
bool is_it_time_for_action(int nstep, amrex::Real time, amrex::Real dt, int action_interval, amrex::Real action_per)
Decide if it is time to take an action.
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_cons_full_domain
multilevel data container for high res initial data: temperature, salinity, passive tracer
std::unique_ptr< NCTimeSeries > Vwind_data_from_file
Data container for v-direction wind read from file.
void init_bathymetry_full_domain_from_netcdf()
Full domain high-res bathymetry data initialization from NetCDF file.
void set_hmixcoef(int lev)
Initialize horizontal mixing coefficients.
amrex::Vector< std::unique_ptr< NCTimeSeriesRiver > > river_source_cons
Vector of data containers for scalar data in rivers.
void timeStep(int lev, amrex::Real time, int iteration)
advance a level by dt, includes a recursive call for finer levels
std::unique_ptr< NCTimeSeriesRiver > river_source_transport
Data container for momentum transport in rivers.
void init_grid_vars_full_domain_from_netcdf()
Full domain high-res grid variable initialization from NetCDF file.
void AverageDown()
set covered coarse cells to be the average of overlying fine cells
static int fixed_ndtfast_ratio
User specified, number of barotropic steps per baroclinic step.
amrex::Real netcdf_fill_value
fill value for masked arrays in netcdf output
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_pn_full_domain
horizontal scaling factor: 1 / dy (2D) on whole domain
void timeStepML(amrex::Real time, int iteration)
advance all levels by dt, loops over finer levels
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_pn
horizontal scaling factor: 1 / dy (2D)
amrex::Vector< std::unique_ptr< std::fstream > > datalog
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_Akv
Vertical viscosity coefficient (3D)
static amrex::Real cfl
CFL condition.
void append3DPlotVariables(const std::string &pp_plot_var_names_3d)
void allocate_bathymetry_grid_vars_full_domain()
Allocate multifabs for storing full-domain bathymetry and grid vars data.
void set_init_data_averaged_down(int lev)
Problem initialization from averaged-down high resolution data.
static int verbose
Verbosity level of output.
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_cloud
cloud cover fraction [0-1], defined at rho-points
std::string plot_file_name
Plotfile prefix.
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_Zt_avg1
Average of the free surface, zeta (2D)
std::unique_ptr< NCTimeSeries > salt_clim_data_from_file
Data container for salinity climatology data read from file.
int check_int
Checkpoint output interval in iterations.
void set_smflux(int lev)
Initialize or calculate surface momentum flux from file or analytic.
void WritePlotFile(int istep)
main driver for writing AMReX plotfiles
std::string restart_chkfile
If set, restart from this checkpoint file.
void init_clim_nudg_coeff(int lev)
Wrapper to initialize climatology nudging coefficient.
void init_bathymetry_full_domain_from_analytic()
Full domain bathymetry data initialization from analytic.
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_rv
v velocity RHS (3D, includes horizontal and vertical advection)
int cf_width
Nudging width at coarse-fine interface.
static amrex::Vector< amrex::Vector< std::string > > nc_init_file
NetCDF initialization file.
int last_plot_file_step
Step when we last output a plotfile.
amrex::Vector< amrex::Real > t_old
old time at each level
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_srflx
Shortwave radiation flux [W/m²], defined at rho-points.
amrex::Real last_check_file_time
Simulation time when we last output a checkpoint file.
void append2DPlotVariables(const std::string &pp_plot_var_names_2d)
void set_bathymetry_averaged_down(int lev)
Copy over bathymetry data that has been averaged down from high resolution input netcdf file.
amrex::Vector< amrex::Real > dt
time step at each level
static amrex::Real sum_per
Diagnostic sum output interval in time.
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_Pair
Air pressure [mb], defined at rho-points.
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_qair
Specific humidity [kg/kg], defined at rho-points.
int hires_grid_level
Which level the high resolution bathymetry is at.
std::unique_ptr< NCTimeSeries > vbar_clim_data_from_file
Data container for vbar climatology data read from file.
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_Tair
Air temperature [°C], defined at rho-points.
amrex::Vector< std::unique_ptr< amrex::MultiFab > > vec_diff2
Harmonic diffusivity for temperature / salinity.
static constexpr int cons_bc
static constexpr int Temp_bc_comp
const char * buildInfoGetGitHash(int i)
HorizMixingType horiz_mixing_type
bool longwave_down_from_netcdf
amrex::Vector< amrex::Real > tnu2
std::string longwave_netcdf_varname
amrex::Vector< int > do_rivers_cons
ScaledToGridAMRScaling scaled_to_grid_amr_scaling
bool boundary_from_netcdf
void init_params(int ncons)
read in and initialize parameters
VertMixingType vert_mixing_type
CouplingType coupling_type