Implemented fft_periodic_single_box for RZ spectral (#1301)

* Implemented fft_periodic_single_box for RZ spectral

For RZ psatd, simplified copy for forward transform

* Apply review's suggestions and clean up

* Add few comments, fix warnings, apply style conventions

Co-authored-by: Dave Grote <dpgrote@lbl.gov>
Co-authored-by: Edoardo Zoni <ezoni@lbl.gov>

1 files changed, 24 insertions, 24 deletions

diff --git a/Source/FieldSolver/SpectralSolver/SpectralFieldDataRZ.H b/Source/FieldSolver/SpectralSolver/SpectralFieldDataRZ.H
index a2ab2b9cb..a073ec483 100644
--- a/Source/FieldSolver/SpectralSolver/SpectralFieldDataRZ.H
+++ b/Source/FieldSolver/SpectralSolver/SpectralFieldDataRZ.H
@@ -34,31 +34,32 @@ class SpectralFieldDataRZ
 
         using BinomialFilter = amrex::LayoutData<SpectralBinomialFilter>;
 
-        SpectralFieldDataRZ(const amrex::BoxArray& realspace_ba,
-                            const SpectralKSpaceRZ& k_space,
-                            const amrex::DistributionMapping& dm,
-                            const int n_field_required,
-                            const int n_modes,
-                            const int lev);
-        SpectralFieldDataRZ() = default; // Default constructor
+        SpectralFieldDataRZ (const amrex::BoxArray& realspace_ba,
+                             const SpectralKSpaceRZ& k_space,
+                             const amrex::DistributionMapping& dm,
+                             const int n_field_required,
+                             const int n_modes,
+                             const int lev);
+        SpectralFieldDataRZ () = default; // Default constructor
         SpectralFieldDataRZ& operator=(SpectralFieldDataRZ&& field_data) = default;
-        ~SpectralFieldDataRZ();
+        ~SpectralFieldDataRZ ();
 
-        void ForwardTransform(const amrex::MultiFab& mf,
-                              const int field_index, const int i_comp);
-        void ForwardTransform(const amrex::MultiFab& mf_r, const int field_index_r,
-                              const amrex::MultiFab& mf_t, const int field_index_t);
-        void BackwardTransform(amrex::MultiFab& mf,
+        void ForwardTransform (const amrex::MultiFab& mf,
                                const int field_index, const int i_comp);
-        void BackwardTransform(amrex::MultiFab& mf_r, const int field_index_r,
-                               amrex::MultiFab& mf_t, const int field_index_t);
-
-        void FABZForwardTransform(amrex::MFIter const & mfi,
-                                  amrex::MultiFab const & tempHTransformedSplit,
-                                  int field_index, const bool is_nodal_z);
-        void FABZBackwardTransform(amrex::MFIter const & mfi, const int field_index,
-                                   amrex::MultiFab & tempHTransformedSplit,
-                                   const bool is_nodal_z);
+        void ForwardTransform (const amrex::MultiFab& mf_r, const int field_index_r,
+                               const amrex::MultiFab& mf_t, const int field_index_t);
+        void BackwardTransform (amrex::MultiFab& mf,
+                                const int field_index, const int i_comp);
+        void BackwardTransform (amrex::MultiFab& mf_r, const int field_index_r,
+                                amrex::MultiFab& mf_t, const int field_index_t);
+
+        void FABZForwardTransform (amrex::MFIter const & mfi, amrex::Box const & realspace_bx,
+                                   amrex::MultiFab const & tempHTransformedSplit,
+                                   int field_index, const bool is_nodal_z);
+        void FABZBackwardTransform (amrex::MFIter const & mfi, amrex::Box const & realspace_bx,
+                                    const int field_index,
+                                    amrex::MultiFab & tempHTransformedSplit,
+                                    const bool is_nodal_z);
 
         void InitFilter (amrex::IntVect const & filter_npass_each_dir, bool const compensation,
                          SpectralKSpaceRZ const & k_space);
@@ -67,7 +68,7 @@ class SpectralFieldDataRZ
         void ApplyFilter (int const field_index1, int const field_index2, int const field_index3);
 
         // Returns an array that holds the kr for all of the modes
-        HankelTransform::RealVector const & getKrArray(amrex::MFIter const & mfi) const {
+        HankelTransform::RealVector const & getKrArray (amrex::MFIter const & mfi) const {
             return multi_spectral_hankel_transformer[mfi].getKrArray();
         }
 
@@ -88,7 +89,6 @@ class SpectralFieldDataRZ
         SpectralShiftFactor zshift_FFTfromCell, zshift_FFTtoCell;
         MultiSpectralHankelTransformer multi_spectral_hankel_transformer;
         BinomialFilter binomialfilter;
-
 };
 
 #endif // WARPX_SPECTRAL_FIELD_DATA_RZ_H_