diff options
Diffstat (limited to 'Source/Utils/CoarsenMR.H')
| -rw-r--r-- | Source/Utils/CoarsenMR.H | 154 |
1 files changed, 0 insertions, 154 deletions
diff --git a/Source/Utils/CoarsenMR.H b/Source/Utils/CoarsenMR.H deleted file mode 100644 index 1191b0c7e..000000000 --- a/Source/Utils/CoarsenMR.H +++ /dev/null @@ -1,154 +0,0 @@ -#ifndef WARPX_COARSEN_MR_H_ -#define WARPX_COARSEN_MR_H_ - -#include <AMReX_Array.H> -#include <AMReX_Array4.H> -#include <AMReX_Extension.H> -#include <AMReX_GpuQualifiers.H> -#include <AMReX_REAL.H> - -#include <AMReX_BaseFwd.H> - -#include <cstdlib> - -namespace CoarsenMR{ - - using namespace amrex; - - /** - * \brief Interpolates the floating point data contained in the source Array4 - * \c arr_src, extracted from a fine MultiFab, with weights defined in - * such a way that the total charge is preserved. - * - * \param[in] arr_src floating point data to be interpolated - * \param[in] sf staggering of the source fine MultiFab - * \param[in] sc staggering of the destination coarsened MultiFab - * \param[in] cr coarsening ratio along each spatial direction - * \param[in] i index along x of the coarsened Array4 to be filled - * \param[in] j index along y of the coarsened Array4 to be filled - * \param[in] k index along z of the coarsened Array4 to be filled - * \param[in] comp index along the fourth component of the Array4 \c arr_src - * containing the data to be interpolated - * - * \return interpolated field at cell (i,j,k) of a coarsened Array4 - */ - AMREX_GPU_DEVICE - AMREX_FORCE_INLINE - Real Interp ( Array4<Real const> const& arr_src, - GpuArray<int,3> const& sf, - GpuArray<int,3> const& sc, - GpuArray<int,3> const& cr, - const int i, - const int j, - const int k, - const int comp ) - { - // Indices of destination array (coarse) - const int ic[3] = { i, j, k }; - - // Number of points and starting indices of source array (fine) - int np[3], idx_min[3]; - - // Compute number of points - for ( int l = 0; l < 3; ++l ) { - if ( cr[l] == 1 ) np[l] = 1; // no coarsening - else np[l] = cr[l]*(1-sf[l])*(1-sc[l]) // cell-centered - +(2*(cr[l]-1)+1)*sf[l]*sc[l]; // nodal - } - - // Compute starting indices of source array (fine) - for ( int l = 0; l < 3; ++l ) { - if ( cr[l] == 1 ) idx_min[l] = ic[l]; // no coarsening - else idx_min[l] = ic[l]*cr[l]*(1-sf[l])*(1-sc[l]) // cell-centered - +(ic[l]*cr[l]-cr[l]+1)*sf[l]*sc[l]; // nodal - } - - // Auxiliary integer variables - const int numx = np[0]; - const int numy = np[1]; - const int numz = np[2]; - const int imin = idx_min[0]; - const int jmin = idx_min[1]; - const int kmin = idx_min[2]; - const int sfx = sf[0]; - const int sfy = sf[1]; - const int sfz = sf[2]; - const int scx = sc[0]; - const int scy = sc[1]; - const int scz = sc[2]; - const int crx = cr[0]; - const int cry = cr[1]; - const int crz = cr[2]; - int ii, jj, kk; - Real wx, wy, wz; - - // Add neutral elements (=0) beyond guard cells in source array (fine) - auto const arr_src_safe = [arr_src] - AMREX_GPU_DEVICE (int const ix, int const iy, int const iz, int const n) noexcept - { - return arr_src.contains( ix, iy, iz ) ? arr_src(ix,iy,iz,n) : 0.0_rt; - }; - - // Interpolate over points computed above. Weights are computed in order - // to guarantee total charge conservation for both cell-centered data - // (equal weights) and nodal data (weights depend on distance between - // points on fine and coarse grids). Terms multiplied by (1-sf)*(1-sc) - // are ON for cell-centered data and OFF for nodal data, while terms - // multiplied by sf*sc are ON for nodal data and OFF for cell-centered data. - // Python script Source/Utils/check_interp_points_and_weights.py can be - // used to check interpolation points and weights in 1D. - Real c = 0.0_rt; - for (int kref = 0; kref < numz; ++kref) { - for (int jref = 0; jref < numy; ++jref) { - for (int iref = 0; iref < numx; ++iref) { - ii = imin+iref; - jj = jmin+jref; - kk = kmin+kref; - wx = (1.0_rt/static_cast<Real>(numx))*(1-sfx)*(1-scx) // if cell-centered - +((amrex::Math::abs(crx-amrex::Math::abs(ii-i*crx)))/static_cast<Real>(crx*crx))*sfx*scx; // if nodal - wy = (1.0_rt/static_cast<Real>(numy))*(1-sfy)*(1-scy) // if cell-centered - +((amrex::Math::abs(cry-amrex::Math::abs(jj-j*cry)))/static_cast<Real>(cry*cry))*sfy*scy; // if nodal - wz = (1.0_rt/static_cast<Real>(numz))*(1-sfz)*(1-scz) // if cell-centered - +((amrex::Math::abs(crz-amrex::Math::abs(kk-k*crz)))/static_cast<Real>(crz*crz))*sfz*scz; // if nodal - c += wx*wy*wz*arr_src_safe(ii,jj,kk,comp); - } - } - } - return c; - } - - /** - * \brief Loops over the boxes of the coarsened MultiFab \c mf_dst and fills - * them by interpolating the data contained in the fine MultiFab \c mf_src. - * - * \param[in,out] mf_dst coarsened MultiFab containing the floating point data - * to be filled by interpolating the source fine MultiFab - * \param[in] mf_src fine MultiFab containing the floating point data to be interpolated - * \param[in] ncomp number of components to loop over for the coarsened - * Array4 extracted from the coarsened MultiFab \c mf_dst - * \param[in] ngrow number of guard cells to fill along each spatial direction - * \param[in] crse_ratio coarsening ratio between the fine MultiFab \c mf_src - * and the coarsened MultiFab \c mf_dst along each spatial direction - */ - void Loop ( MultiFab& mf_dst, - const MultiFab& mf_src, - const int ncomp, - const IntVect ngrow, - const IntVect crse_ratio ); - - /** - * \brief Stores in the coarsened MultiFab \c mf_dst the values obtained by - * interpolating the data contained in the fine MultiFab \c mf_src. - * - * \param[in,out] mf_dst coarsened MultiFab containing the floating point data - * to be filled by interpolating the fine MultiFab \c mf_src - * \param[in] mf_src fine MultiFab containing the floating point data to be interpolated - * \param[in] crse_ratio coarsening ratio between the fine MultiFab \c mf_src - * and the coarsened MultiFab \c mf_dst along each spatial direction - */ - void Coarsen ( MultiFab& mf_dst, - const MultiFab& mf_src, - const IntVect crse_ratio ); -} - -#endif // WARPX_COARSEN_MR_H_ |