ERF
Energy Research and Forecasting: An Atmospheric Modeling Code
•All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Pages
Functions
ERF_Interpolation_Bilinear.H File Reference
#include "ERF_DataStruct.H"
Include dependency graph for ERF_Interpolation_Bilinear.H:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Functions

AMREX_FORCE_INLINE AMREX_GPU_HOST_DEVICE int get_single_index (int i, int j, int k, int nx, int ny)
 
AMREX_FORCE_INLINE AMREX_GPU_HOST_DEVICE void bilinear_interpolation (const amrex::Real *xvec, const amrex::Real *yvec, const amrex::Real *zvec, const amrex::Real dxvec, const amrex::Real dyvec, const int nx, const int ny, const int nz, const amrex::Real x, const amrex::Real y, const amrex::Real z, const amrex::Real *varvec, amrex::Real &tmp_var)
 
AMREX_GPU_DEVICE amrex::Real interpolate_from_coarse (const amrex::Array4< const amrex::Real > &crse, int n, amrex::Real x, amrex::Real y, amrex::Real z, const amrex::Real *prob_lo_crse, const amrex::Real *dx_crse)
 

Function Documentation

◆ bilinear_interpolation()

AMREX_FORCE_INLINE AMREX_GPU_HOST_DEVICE void bilinear_interpolation ( const amrex::Real *  xvec,
const amrex::Real *  yvec,
const amrex::Real *  zvec,
const amrex::Real  dxvec,
const amrex::Real  dyvec,
const int  nx,
const int  ny,
const int  nz,
const amrex::Real  x,
const amrex::Real  y,
const amrex::Real  z,
const amrex::Real *  varvec,
amrex::Real &  tmp_var 
)
29 {
30  int iloc=-1, jloc=-1, kloc=-1;
31  for(int k=0;k<nz;k++){
32  if(zvec[k] > z){
33  kloc = k-1;
34  break;
35  }
36  else if (zvec[k] == z) {
37  kloc = k;
38  }
39 
40  }
41 
42  iloc = static_cast<int>(std::floor((x - xvec[0]) / dxvec));
43  jloc = static_cast<int>(std::floor((y - yvec[0]) / dyvec));
44 
45  if(iloc > nx-1 or iloc < 0 or
46  jloc > ny-1 or iloc < 0 or
47  kloc > nz-1 or kloc < 0){
48  }
49 
50  amrex::Real xlo = xvec[0] + iloc*dxvec;
51  amrex::Real ylo = yvec[0] + jloc*dyvec;
52  amrex::Real zlo = zvec[kloc];
53 
54  amrex::Real w_x = (x - xlo)/dxvec;
55  amrex::Real w_y = (y - ylo)/dyvec;
56  amrex::Real w_z = (z - zlo)/(zvec[kloc+1] - zvec[kloc]);
57 
58  int ind0 = get_single_index(iloc,jloc,kloc,nx,ny);
59  int ind1 = get_single_index(iloc+1,jloc,kloc,nx,ny);
60  int ind2 = get_single_index(iloc+1,jloc+1,kloc,nx,ny);
61  int ind3 = get_single_index(iloc,jloc+1,kloc,nx,ny);
62  int ind4 = get_single_index(iloc,jloc,kloc+1,nx,ny);
63  int ind5 = get_single_index(iloc+1,jloc,kloc+1,nx,ny);
64  int ind6 = get_single_index(iloc+1,jloc+1,kloc+1,nx,ny);
65  int ind7 = get_single_index(iloc,jloc+1,kloc+1,nx,ny);
66 
67  tmp_var = (1-w_x)*(1-w_y)*(1-w_z)*varvec[ind0] + w_x*(1-w_y)*(1-w_z)*varvec[ind1] +
68  w_x*w_y*(1-w_z)*varvec[ind2] + (1-w_x)*w_y*(1-w_z)*varvec[ind3] +
69  (1-w_x)*(1-w_y)*w_z*varvec[ind4] + w_x*(1-w_y)*w_z*varvec[ind5] +
70  w_x*w_y*w_z*varvec[ind6] + (1-w_x)*w_y*w_z*varvec[ind7];
71 
72  //std::cout << "Variable value is " << tmp_var << "\n";
73 }
Coord::y
@ y
Coord::x
@ x
Coord::z
@ z
get_single_index
AMREX_FORCE_INLINE AMREX_GPU_HOST_DEVICE int get_single_index(int i, int j, int k, int nx, int ny)
Definition: ERF_Interpolation_Bilinear.H:13
Here is the call graph for this function:

◆ get_single_index()

AMREX_FORCE_INLINE AMREX_GPU_HOST_DEVICE int get_single_index ( int  i,
int  j,
int  k,
int  nx,
int  ny 
)

Bilinear interpolation routines. There are different versions for different scenarios that take in a different argument list

15 {
16 
17  int si = k*nx*ny + j*nx + i;
18  return si;
19 }

Referenced by bilinear_interpolation(), and fill_weather_data_multifab().

Here is the caller graph for this function:

◆ interpolate_from_coarse()

AMREX_GPU_DEVICE amrex::Real interpolate_from_coarse ( const amrex::Array4< const amrex::Real > &  crse,
int  n,
amrex::Real  x,
amrex::Real  y,
amrex::Real  z,
const amrex::Real *  prob_lo_crse,
const amrex::Real *  dx_crse 
)
inline
82 {
83  // Convert to coarse grid index space
84  amrex::Real ic_real = (x - prob_lo_crse[0]) / dx_crse[0];
85  amrex::Real jc_real = (y - prob_lo_crse[1]) / dx_crse[1];
86  amrex::Real kc_real = (z - prob_lo_crse[2]) / dx_crse[2];
87 
88  int ic = static_cast<int>(amrex::Math::floor(ic_real));
89  int jc = static_cast<int>(amrex::Math::floor(jc_real));
90  int kc = static_cast<int>(amrex::Math::floor(kc_real));
91 
92  amrex::Real dx = ic_real - ic;
93  amrex::Real dy = jc_real - jc;
94  amrex::Real dz = kc_real - kc;
95 
96  // Trilinear interpolation
97  amrex::Real v000 = crse(ic, jc, kc, n);
98  amrex::Real v100 = crse(ic + 1, jc, kc, n);
99  amrex::Real v010 = crse(ic, jc + 1, kc, n);
100  amrex::Real v110 = crse(ic + 1, jc + 1, kc, n);
101  amrex::Real v001 = crse(ic, jc, kc + 1, n);
102  amrex::Real v101 = crse(ic + 1, jc, kc + 1, n);
103  amrex::Real v011 = crse(ic, jc + 1, kc + 1, n);
104  amrex::Real v111 = crse(ic + 1, jc + 1, kc + 1, n);
105 
106  amrex::Real v = v000 * (1 - dx) * (1 - dy) * (1 - dz) +
107  v100 * dx * (1 - dy) * (1 - dz) +
108  v010 * (1 - dx) * dy * (1 - dz) +
109  v110 * dx * dy * (1 - dz) +
110  v001 * (1 - dx) * (1 - dy) * dz +
111  v101 * dx * (1 - dy) * dz +
112  v011 * (1 - dx) * dy * dz +
113  v111 * dx * dy * dz;
114 
115  return v;
116 }

Referenced by ERF::interp_weather_data_onto_mesh().

Here is the caller graph for this function: