1 files changed, 16 insertions, 4 deletions
diff --git a/Source/FieldSolver/SpectralSolver/SpectralAlgorithms/PsatdAlgorithm.H b/Source/FieldSolver/SpectralSolver/SpectralAlgorithms/PsatdAlgorithm.H
index 0487e5226..34743525e 100644
--- a/Source/FieldSolver/SpectralSolver/SpectralAlgorithms/PsatdAlgorithm.H
+++ b/Source/FieldSolver/SpectralSolver/SpectralAlgorithms/PsatdAlgorithm.H
@@ -1,6 +1,8 @@
 #ifndef WARPX_PSATD_ALGORITHM_H_
 #define WARPX_PSATD_ALGORITHM_H_
 
+#include <SpectralKSpace.H>
+#include <SpectralFieldData.H>
 #include <SpectralBaseAlgorithm.H>
 
 /* \brief Class that updates the field in spectral space
@@ -8,16 +10,26 @@
  */
 class PsatdAlgorithm : public SpectralBaseAlgorithm
 {
+    using SpectralCoefficients = amrex::FabArray< amrex::BaseFab <amrex::Real> >;
+
     public:
         PsatdAlgorithm(const SpectralKSpace& spectral_kspace,
                          const amrex::DistributionMapping& dm,
                          const int norder_x, const int norder_y,
-                         const int norder_z, const bool nodal,
-                         const amrex::Real dt);
-        // Redefine update equation from base class
-        virtual void pushSpectralFields(SpectralFieldData& f) const override final;
+                         const int norder_z, const bool nodal, const amrex::Real dt);
+        PsatdAlgorithm() = default; // Default constructor
+        PsatdAlgorithm& operator=(PsatdAlgorithm&& algorithm) = default;
+        void pushSpectralFields(SpectralFieldData& f) const;
+        void InitializeCoefficience(const SpectralKSpace& spectral_kspace,
+                                    const amrex::DistributionMapping& dm, 
+                                    const amrex::Real dt);
 
     private:
+        // Modified finite-order vectors
+        KVectorComponent modified_kx_vec, modified_kz_vec;
+#if (AMREX_SPACEDIM==3)
+        KVectorComponent modified_ky_vec;
+#endif
         SpectralCoefficients C_coef, S_ck_coef, X1_coef, X2_coef, X3_coef;
 };