Merge pull request #95 from ECP-WarpX/new_spectral_structures

Implement PSATD solver in C++

1 files changed, 32 insertions, 0 deletions

diff --git a/Source/FieldSolver/SpectralSolver/PsatdAlgorithm.H b/Source/FieldSolver/SpectralSolver/PsatdAlgorithm.H
new file mode 100644
index 000000000..acefcc466
--- /dev/null
+++ b/Source/FieldSolver/SpectralSolver/PsatdAlgorithm.H
@@ -0,0 +1,32 @@
+#ifndef WARPX_PSATD_ALGORITHM_H_
+#define WARPX_PSATD_ALGORITHM_H_
+
+#include <SpectralKSpace.H>
+#include <SpectralFieldData.H>
+
+/* \brief Class that updates the field in spectral space
+ * and stores the coefficients of the corresponding update equation.
+ */
+class PsatdAlgorithm
+{
+    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);
+        PsatdAlgorithm() = default; // Default constructor
+        PsatdAlgorithm& operator=(PsatdAlgorithm&& algorithm) = default;
+        void pushSpectralFields(SpectralFieldData& f) const;
+
+    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;
+};
+
+#endif // WARPX_PSATD_ALGORITHM_H_