diff options
Diffstat (limited to 'Source/FieldSolver/FiniteDifferenceSolver/FiniteDifferenceAlgorithms')
3 files changed, 55 insertions, 8 deletions
diff --git a/Source/FieldSolver/FiniteDifferenceSolver/FiniteDifferenceAlgorithms/CartesianCKCAlgorithm.H b/Source/FieldSolver/FiniteDifferenceSolver/FiniteDifferenceAlgorithms/CartesianCKCAlgorithm.H index ef6d53416..b688110b7 100644 --- a/Source/FieldSolver/FiniteDifferenceSolver/FiniteDifferenceAlgorithms/CartesianCKCAlgorithm.H +++ b/Source/FieldSolver/FiniteDifferenceSolver/FiniteDifferenceAlgorithms/CartesianCKCAlgorithm.H @@ -143,9 +143,10 @@ struct CartesianCKCAlgorithm { /** * Perform derivative along x on a nodal grid, from a cell-centered field `F` */ + template< typename T_Field> AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE static amrex::Real DownwardDx ( - amrex::Array4<amrex::Real> const& F, + T_Field const& F, amrex::Real const * const coefs_x, int const /*n_coefs_x*/, int const i, int const j, int const k, int const ncomp=0 ) { @@ -189,15 +190,16 @@ struct CartesianCKCAlgorithm { /** * Perform derivative along y on a nodal grid, from a cell-centered field `F` */ + template< typename T_Field> AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE static amrex::Real DownwardDy ( - amrex::Array4<amrex::Real> const& F, + T_Field const& F, amrex::Real const * const coefs_y, int const n_coefs_y, 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]; + amrex::Real const inv_dy = coefs_y[0]; return inv_dy*( F(i,j,k,ncomp) - F(i,j-1,k,ncomp) ); amrex::ignore_unused(n_coefs_y); #elif (defined WARPX_DIM_XZ) @@ -248,11 +250,12 @@ struct CartesianCKCAlgorithm { /** * Perform derivative along z on a nodal grid, from a cell-centered field `F` */ + template< typename T_Field> AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE static amrex::Real DownwardDz ( - amrex::Array4<amrex::Real> const& F, + T_Field const& F, amrex::Real const * const coefs_z, int const /*n_coefs_z*/, - int const i, int const j, int const k, int const ncomp=0 ) { + int const i, int const j, int const k, int const ncomp=0) { amrex::Real const inv_dz = coefs_z[0]; #if defined WARPX_DIM_3D diff --git a/Source/FieldSolver/FiniteDifferenceSolver/FiniteDifferenceAlgorithms/CartesianYeeAlgorithm.H b/Source/FieldSolver/FiniteDifferenceSolver/FiniteDifferenceAlgorithms/CartesianYeeAlgorithm.H index 305fb3507..2ec6a3d22 100644 --- a/Source/FieldSolver/FiniteDifferenceSolver/FiniteDifferenceAlgorithms/CartesianYeeAlgorithm.H +++ b/Source/FieldSolver/FiniteDifferenceSolver/FiniteDifferenceAlgorithms/CartesianYeeAlgorithm.H @@ -9,6 +9,7 @@ #define WARPX_FINITE_DIFFERENCE_ALGORITHM_CARTESIAN_YEE_H_ #include "Utils/WarpXConst.H" +#include "FieldAccessorFunctors.H" #include <AMReX.H> #include <AMReX_REAL.H> @@ -68,9 +69,10 @@ struct CartesianYeeAlgorithm { /** * Perform derivative along x on a nodal grid, from a cell-centered field `F`*/ + template< typename T_Field> AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE static amrex::Real DownwardDx ( - amrex::Array4<amrex::Real> const& F, + T_Field const& F, amrex::Real const * const coefs_x, int const /*n_coefs_x*/, int const i, int const j, int const k, int const ncomp=0 ) { @@ -103,9 +105,10 @@ struct CartesianYeeAlgorithm { /** * Perform derivative along y on a nodal grid, from a cell-centered field `F`*/ + template< typename T_Field> AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE static amrex::Real DownwardDy ( - amrex::Array4<amrex::Real> const& F, + T_Field const& F, amrex::Real const * const coefs_y, int const n_coefs_y, int const i, int const j, int const k, int const ncomp=0 ) { @@ -143,9 +146,10 @@ struct CartesianYeeAlgorithm { /** * Perform derivative along z on a nodal grid, from a cell-centered field `F`*/ + template< typename T_Field> AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE static amrex::Real DownwardDz ( - amrex::Array4<amrex::Real> const& F, + T_Field const& F, amrex::Real const * const coefs_z, int const /*n_coefs_z*/, int const i, int const j, int const k, int const ncomp=0 ) { diff --git a/Source/FieldSolver/FiniteDifferenceSolver/FiniteDifferenceAlgorithms/FieldAccessorFunctors.H b/Source/FieldSolver/FiniteDifferenceSolver/FiniteDifferenceAlgorithms/FieldAccessorFunctors.H new file mode 100644 index 000000000..3ba6de335 --- /dev/null +++ b/Source/FieldSolver/FiniteDifferenceSolver/FiniteDifferenceAlgorithms/FieldAccessorFunctors.H @@ -0,0 +1,40 @@ +#ifndef WARPX_FIELD_ACCESSOR_FUNCTORS_H +#define WARPX_FIELD_ACCESSOR_FUNCTORS_H +#include "WarpX.H" +/** + * \brief Functor that returns the division of the source m_field Array4 value + by macroparameter, m_parameter value at the respective (i,j,k,ncomp). + */ +struct FieldAccessorMacroscopic +{ + AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE + FieldAccessorMacroscopic ( amrex::Array4<amrex::Real const> const a_field, + amrex::Array4<amrex::Real const> const a_parameter ) + : m_field(a_field), m_parameter(a_parameter) {} + + /** + * \brief return field value at (i,j,k,ncomp) scaled by (1/m_parameters(i,j,k,ncomp)) + * \param[in] i index along x of the Array4, m_field and m_parameter. + * \param[in] j index along y of the Array4, m_field and m_parameter. + * \param[in] k index along z of the Array4, m_field and m_parameter. + * \param[in] ncomp index along fourth component of the Array4, containing field-data + to be returned after diving with zero-th component + of m_paramter. + * + * \return m_field/m_paramter at (i,j,k,ncomp) + */ + AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE + amrex::Real operator() (int const i, int const j, + int const k, int const ncomp) const noexcept + { + return ( m_field(i, j, k, ncomp) / m_parameter(i, j, k, 0) ) ; + } +private: + /** Array4 of the source field to be scaled and returned by the operator() */ + amrex::Array4<amrex::Real const> const m_field; + /** Array4 of the macroscopic parameter used to divide m_field in the operator() */ + amrex::Array4<amrex::Real const> const m_parameter; +}; + + +#endif |