From cbd23a8e7c6e2970a3f26e072d67173cf533e94d Mon Sep 17 00:00:00 2001 From: Remi Lehe Date: Wed, 13 May 2020 13:26:27 -0700 Subject: Allow to pass component in stencil templates (#1012) --- .../CartesianYeeAlgorithm.H | 28 +++++++++++----------- 1 file changed, 14 insertions(+), 14 deletions(-) (limited to 'Source/FieldSolver/FiniteDifferenceSolver/FiniteDifferenceAlgorithms/CartesianYeeAlgorithm.H') diff --git a/Source/FieldSolver/FiniteDifferenceSolver/FiniteDifferenceAlgorithms/CartesianYeeAlgorithm.H b/Source/FieldSolver/FiniteDifferenceSolver/FiniteDifferenceAlgorithms/CartesianYeeAlgorithm.H index f1778a525..3a98bea0e 100644 --- a/Source/FieldSolver/FiniteDifferenceSolver/FiniteDifferenceAlgorithms/CartesianYeeAlgorithm.H +++ b/Source/FieldSolver/FiniteDifferenceSolver/FiniteDifferenceAlgorithms/CartesianYeeAlgorithm.H @@ -42,10 +42,10 @@ struct CartesianYeeAlgorithm { static amrex::Real UpwardDx ( amrex::Array4 const& F, amrex::Real const * const coefs_x, int const n_coefs_x, - int const i, int const j, int const k ) { + int const i, int const j, int const k, int const ncomp=0 ) { amrex::Real const inv_dx = coefs_x[0]; - return inv_dx*( F(i+1,j,k) - F(i,j,k) ); + return inv_dx*( F(i+1,j,k,ncomp) - F(i,j,k,ncomp) ); }; /** @@ -54,10 +54,10 @@ struct CartesianYeeAlgorithm { static amrex::Real DownwardDx ( amrex::Array4 const& F, amrex::Real const * const coefs_x, int const n_coefs_x, - int const i, int const j, int const k ) { + int const i, int const j, int const k, int const ncomp=0 ) { amrex::Real const inv_dx = coefs_x[0]; - return inv_dx*( F(i,j,k) - F(i-1,j,k) ); + return inv_dx*( F(i,j,k,ncomp) - F(i-1,j,k,ncomp) ); }; /** @@ -66,12 +66,12 @@ struct CartesianYeeAlgorithm { static amrex::Real UpwardDy ( amrex::Array4 const& F, amrex::Real const * const coefs_y, int const n_coefs_y, - int const i, int const j, int const k ) { + int const i, int const j, int const k, int const ncomp=0 ) { using namespace amrex; #if defined WARPX_DIM_3D Real const inv_dy = coefs_y[0]; - return inv_dy*( F(i,j+1,k) - F(i,j,k) ); + return inv_dy*( F(i,j+1,k,ncomp) - F(i,j,k,ncomp) ); #elif (defined WARPX_DIM_XZ) return 0._rt; // 2D Cartesian: derivative along y is 0 #endif @@ -83,12 +83,12 @@ struct CartesianYeeAlgorithm { static amrex::Real DownwardDy ( amrex::Array4 const& F, amrex::Real const * const coefs_y, int const n_coefs_y, - int const i, int const j, int const k ) { + int const i, int const j, int const k, int const ncomp=0 ) { using namespace amrex; #if defined WARPX_DIM_3D Real const inv_dy = coefs_y[0]; - return inv_dy*( F(i,j,k) - F(i,j-1,k) ); + return inv_dy*( F(i,j,k,ncomp) - F(i,j-1,k,ncomp) ); #elif (defined WARPX_DIM_XZ) return 0._rt; // 2D Cartesian: derivative along y is 0 #endif @@ -100,14 +100,14 @@ struct CartesianYeeAlgorithm { static amrex::Real UpwardDz ( amrex::Array4 const& F, amrex::Real const * const coefs_z, int const n_coefs_z, - int const i, int const j, int const k ) { + int const i, int const j, int const k, int const ncomp=0 ) { using namespace amrex; Real const inv_dz = coefs_z[0]; #if defined WARPX_DIM_3D - return inv_dz*( F(i,j,k+1) - F(i,j,k) ); + return inv_dz*( F(i,j,k+1,ncomp) - F(i,j,k,ncomp) ); #elif (defined WARPX_DIM_XZ) - return inv_dz*( F(i,j+1,k) - F(i,j,k) ); + return inv_dz*( F(i,j+1,k,ncomp) - F(i,j,k,ncomp) ); #endif }; @@ -117,14 +117,14 @@ struct CartesianYeeAlgorithm { static amrex::Real DownwardDz ( amrex::Array4 const& F, amrex::Real const * const coefs_z, int const n_coefs_z, - int const i, int const j, int const k ) { + int const i, int const j, int const k, int const ncomp=0 ) { using namespace amrex; Real const inv_dz = coefs_z[0]; #if defined WARPX_DIM_3D - return inv_dz*( F(i,j,k) - F(i,j,k-1) ); + return inv_dz*( F(i,j,k,ncomp) - F(i,j,k-1,ncomp) ); #elif (defined WARPX_DIM_XZ) - return inv_dz*( F(i,j,k) - F(i,j-1,k) ); + return inv_dz*( F(i,j,k,ncomp) - F(i,j-1,k,ncomp) ); #endif }; -- cgit v1.2.3