311 amrex::Vector<amrex::Real> m_xterrain,m_yterrain,m_zterrain;
313 int nx = 0;
int ny = 0;
315 if (amrex::ParallelDescriptor::IOProcessor()) {
317 amrex::Print()<<
"Reading terrain file: "<< fname<< std::endl;
318 std::ifstream file(fname);
320 if (!file.is_open()) {
321 amrex::Abort(
"Error: Could not open the file " + fname+
"\n");
325 if (file.peek() == std::ifstream::traits_type::eof()) {
326 amrex::Abort(
"Error: The file " + fname+
" is empty.\n");
329 amrex::Real value1,value2,value3;
332 amrex::Real lat_min, lon_min;
334 file >> lon_min >> lat_min;
335 if(std::fabs(lon_min) > 180.0) {
336 amrex::Error(
"The value of longitude for entry in the first line in " + fname
337 +
" should not exceed 180. It is " + std::to_string(lon_min));
339 if(std::fabs(lat_min) > 90.0) {
340 amrex::Error(
"The value of latitude for entry in the first line in " + fname
341 +
" should not exceed 90. It is " + std::to_string(lat_min));
347 while (file >> value1 >> value2 >> value3) {
348 m_xterrain.push_back(value1);
350 m_yterrain.push_back(value2);
352 m_zterrain.push_back(value3);
355 AMREX_ASSERT(m_xterrain.size() ==
static_cast<long int>(nx*ny));
356 AMREX_ASSERT(m_yterrain.size() ==
static_cast<long int>(ny));
357 AMREX_ASSERT(m_zterrain.size() ==
static_cast<long int>(nx*ny));
361 nx = erf_get_single_value<int>(file,cnt); cnt++;
362 ny = erf_get_single_value<int>(file,cnt); cnt++;
363 amrex::Print()<<
"Expecting " << nx <<
" values of x, " <<
364 ny <<
" values of y, and " <<
365 nx*ny <<
" values of z" << std::endl;
366 AMREX_ALWAYS_ASSERT(nx > 0);
367 AMREX_ALWAYS_ASSERT(ny > 0);
368 m_xterrain.resize(nx);
369 m_yterrain.resize(ny);
370 m_zterrain.resize(nx * ny);
371 for (
int n = 0; n < nx; n++) {
372 m_xterrain[n] = erf_get_single_value<amrex::Real>(file,cnt);
375 for (
int n = 0; n < ny; n++) {
376 m_yterrain[n] = erf_get_single_value<amrex::Real>(file,cnt);
379 for (
int n = 0; n < nx * ny; n++) {
380 m_zterrain[n] = erf_get_single_value<amrex::Real>(file,cnt);
389 amrex::ParallelDescriptor::Bcast(&nx,1,amrex::ParallelDescriptor::IOProcessorNumber());
390 amrex::ParallelDescriptor::Bcast(&ny,1,amrex::ParallelDescriptor::IOProcessorNumber());
394 m_xterrain.resize(nx);
395 m_yterrain.resize(ny);
396 m_zterrain.resize(nz);
398 amrex::ParallelDescriptor::Bcast(m_xterrain.data(),nx,amrex::ParallelDescriptor::IOProcessorNumber());
399 amrex::ParallelDescriptor::Bcast(m_yterrain.data(),ny,amrex::ParallelDescriptor::IOProcessorNumber());
400 amrex::ParallelDescriptor::Bcast(m_zterrain.data(),nz,amrex::ParallelDescriptor::IOProcessorNumber());
403 amrex::Gpu::DeviceVector<amrex::Real> d_xterrain(nx),d_yterrain(ny),d_zterrain(nz);
404 amrex::Gpu::copy(amrex::Gpu::hostToDevice, m_xterrain.begin(), m_xterrain.end(), d_xterrain.begin());
405 amrex::Gpu::copy(amrex::Gpu::hostToDevice, m_yterrain.begin(), m_yterrain.end(), d_yterrain.begin());
406 amrex::Gpu::copy(amrex::Gpu::hostToDevice, m_zterrain.begin(), m_zterrain.end(), d_zterrain.begin());
408 amrex::Real* d_xt = d_xterrain.data();
409 amrex::Real* d_yt = d_yterrain.data();
410 amrex::Real* d_zt = d_zterrain.data();
412 auto dx = geom.CellSizeArray();
413 auto ProbLoArr = geom.ProbLoArray();
415 int ilo = geom.Domain().smallEnd(0);
416 int jlo = geom.Domain().smallEnd(1);
417 int klo = geom.Domain().smallEnd(2);
418 int ihi = geom.Domain().bigEnd(0) + 1;
419 int jhi = geom.Domain().bigEnd(1) + 1;
421 amrex::Box
zbx = terrain_fab.box();
422 amrex::Array4<amrex::Real>
const& z_arr = terrain_fab.array();
424 amrex::ParallelFor(
zbx, [=] AMREX_GPU_DEVICE (
int i,
int j,
int )
427 int ii = amrex::min(amrex::max(i,ilo),ihi);
428 int jj = amrex::min(amrex::max(j,jlo),jhi);
431 amrex::Real
x = ProbLoArr[0] + ii * dx[0] + 1e-3;
432 amrex::Real
y = ProbLoArr[1] + jj * dx[1] + 1e-3;
434 int ind11, ind12, ind21, ind22;
435 amrex::Real x1, x2, y1, y2;
437 int iindex_terrain=-1;
438 int jindex_terrain=-1;
444 for (
int it = 0; it < ny && jindex_terrain == -1; it++) {
446 jindex_terrain = it-1;
449 if (jindex_terrain == -1) {
450 jindex_terrain = ny-1;
453 int gstart = (jindex_terrain )*nx;
454 int gend = (jindex_terrain+1)*nx-1;
455 for (
int it = gstart; it <= gend && iindex_terrain == -1; it++) {
457 iindex_terrain = it-gstart;
462 ind11 = jindex_terrain*nx + iindex_terrain;
469 y1 = d_yt[jindex_terrain];
470 y2 = d_yt[jindex_terrain+1];
472 amrex::Real denom = (x2-x1)*(y2-y1);
473 amrex::Real w_11 = (x2-
x)*(y2-
y)/denom;
474 amrex::Real w_12 = (x2-
x)*(
y-y1)/denom;
475 amrex::Real w_21 = (
x-x1)*(y2-
y)/denom;
476 amrex::Real w_22 = (
x-x1)*(
y-y1)/denom;
479 z_arr(i,j,klo) = w_11*d_zt[ind11] + w_12*d_zt[ind12] + w_21*d_zt[ind21] + w_22*d_zt[ind22];
483 for (
int it = 0; it < ny && jindex_terrain == -1; it++) {
485 jindex_terrain = it-1;
488 if (jindex_terrain == -1) {
489 jindex_terrain = ny-1;
492 for (
int it = 0; it < nx && iindex_terrain == -1; it++) {
494 iindex_terrain = it-1;
497 if (iindex_terrain == -1) {
498 iindex_terrain = nx-1;
502 x1 = d_xt[iindex_terrain];
503 x2 = d_xt[iindex_terrain+1];
504 y1 = d_yt[jindex_terrain];
505 y2 = d_yt[jindex_terrain+1];
511 ind11 = iindex_terrain * ny + jindex_terrain;
512 ind21 = std::min(iindex_terrain+1,nx-1) * ny + jindex_terrain;
514 ind12 = iindex_terrain * ny + std::min(jindex_terrain+1,ny-1);
515 ind22 = std::min(iindex_terrain+1,nx-1) * ny + std::min(jindex_terrain+1,ny-1);
521 ind11 = jindex_terrain * nx + iindex_terrain;
522 ind12 = std::min(jindex_terrain+1,ny-1) * nx + iindex_terrain;
524 ind21 = jindex_terrain * nx + std::min(iindex_terrain+1,nx-1);
525 ind22 = std::min(jindex_terrain+1,ny-1) * nx + std::min(iindex_terrain+1,nx-1);
528 if (iindex_terrain == nx-1 && jindex_terrain == ny-1)
530 z_arr(i,j,klo) = d_zt[ind11];
532 else if (iindex_terrain != nx-1 && jindex_terrain == ny-1)
534 amrex::Real w_11 = (x2-
x);
535 amrex::Real w_21 = (
x-x1);
536 amrex::Real denom = (x2-x1);
537 z_arr(i,j,klo) = (w_11*d_zt[ind11] + w_21*d_zt[ind21])/denom;
539 else if (iindex_terrain == nx-1 && jindex_terrain != ny-1)
541 amrex::Real w_11 = (y2-
y);
542 amrex::Real w_12 = (
y-y1);
543 amrex::Real denom = (y2-y1);
544 z_arr(i,j,klo) = (w_11*d_zt[ind11] + w_12*d_zt[ind12])/denom;
548 amrex::Real w_11 = (x2-
x)*(y2-
y);
549 amrex::Real w_21 = (
x-x1)*(y2-
y);
550 amrex::Real w_12 = (x2-
x)*(
y-y1);
551 amrex::Real w_22 = (
x-x1)*(
y-y1);
552 amrex::Real denom = (x2-x1)*(y2-y1);
553 z_arr(i,j,klo) = (w_11*d_zt[ind11] + w_12*d_zt[ind12] + w_21*d_zt[ind21] + w_22*d_zt[ind22]) / denom;
const Box zbx
Definition: ERF_DiffSetup.H:23