diff options
Diffstat (limited to 'Source/FieldSolver/SpectralSolver/SpectralSolver.cpp')
-rw-r--r-- | Source/FieldSolver/SpectralSolver/SpectralSolver.cpp | 32 |
1 files changed, 30 insertions, 2 deletions
diff --git a/Source/FieldSolver/SpectralSolver/SpectralSolver.cpp b/Source/FieldSolver/SpectralSolver/SpectralSolver.cpp index c24a7af69..9df981528 100644 --- a/Source/FieldSolver/SpectralSolver/SpectralSolver.cpp +++ b/Source/FieldSolver/SpectralSolver/SpectralSolver.cpp @@ -9,7 +9,8 @@ #include <PsatdAlgorithm.H> #include <GalileanAlgorithm.H> #include <PMLPsatdAlgorithm.H> - +#include "WarpX.H" +#include "WarpXProfilerWrapper.H" /* \brief Initialize the spectral Maxwell solver * @@ -62,4 +63,31 @@ SpectralSolver::SpectralSolver( field_data = SpectralFieldData( realspace_ba, k_space, dm, algorithm->getRequiredNumberOfFields() ); -}; +} + +void +SpectralSolver::ForwardTransform( const amrex::MultiFab& mf, + const int field_index, + const int i_comp ) +{ + WARPX_PROFILE("SpectralSolver::ForwardTransform"); + field_data.ForwardTransform( mf, field_index, i_comp ); +} + +void +SpectralSolver::BackwardTransform( amrex::MultiFab& mf, + const int field_index, + const int i_comp ) +{ + WARPX_PROFILE("SpectralSolver::BackwardTransform"); + field_data.BackwardTransform( mf, field_index, i_comp ); +} + +void +SpectralSolver::pushSpectralFields(){ + WARPX_PROFILE("SpectralSolver::pushSpectralFields"); + // Virtual function: the actual function used here depends + // on the sub-class of `SpectralBaseAlgorithm` that was + // initialized in the constructor of `SpectralSolver` + algorithm->pushSpectralFields( field_data ); +} |