HOS-NWT capability (#1536)

* compiling with these changes, Waves2AMR commit a033db8

* fix rmodes initialization

* differentiate between dimensions for fft and for interpolations, goes with Waves2AMR commit 8a55cd1

* adapting to Waves2AMR ed0aba8, something is off with the 3d case

* update Waves2AMR submodule

* correct dimensions of additional mode pointers

* correct offset arguments

* correct readmodes type

* correct dx

* add vertical shift to ow_w2a case

* NWT cases for 2D and 3D data

* update Waves2AMR module

* update Waves2AMR module again

* formatting

* skip .dat files for spellcheck

* Update test/test_files/ow_w2a_nwt_3d/ow_w2a_nwt_3d.inp

* keep y offset for 3d case, remove from 2d case

* document offset arguments

Author: mbkuhn

.codespellrc

@@ -1,3 +1,3 @@
 [codespell]
-skip = .git,*.ipynb,*.bib,*.ps,*.patch,spack-build-*,*/build,*/submods,plt*,chk*,__pycache__,.ccls,.ccls-cache,*.pdf,DU21_A17.txt,inspect_abl_io.html
+skip = .git,*.ipynb,*.bib,*.ps,*.patch,spack-build-*,*/build,*/submods,plt*,chk*,__pycache__,.ccls,.ccls-cache,*.pdf,DU21_A17.txt,inspect_abl_io.html,*.dat
 ignore-words = .codespell-ignore-words

amr-wind/ocean_waves/relaxation_zones/W2AWaves.H

@@ -14,12 +14,19 @@ struct W2AWavesData : public RelaxZonesBaseData
 {
     // Prefix for HOS files
     std::string modes_file{"modes_HOS_SWENSE.dat"};
-    // Number of modes in each direction
+    // Number of modes in each direction (related to FFT)
     int n0{0};
     int n1{0};
+    int n2{0};
+    // Number of points in each direction (spatial output)
+    int n0_sp{0};
+    int n1_sp{0};
     // Spatial resolution in each direction
     amrex::Real dx0{0.};
     amrex::Real dx1{0.};
+    // Offsets for spatial data (between AMR-Wind domain and HOS domain)
+    amrex::Real xlo0{0.};
+    amrex::Real xlo1{0.};
     // Length to dimensionalize nondim data
     amrex::Real dimL{0.};
     // Timestep size of reference data
@@ -43,50 +50,80 @@ struct W2AWavesData : public RelaxZonesBaseData
     // Time (in AMR-Wind sim) of latest imported wave data
     amrex::Real t{0.0};
 
-    // Vectors to store modes
-    std::vector<std::complex<double>> mX, mY, mZ, mFS;
+    // Vectors to store complex or real modes
+    std::vector<std::complex<double>> c_mX, c_mY, c_mZ, c_mFS;
+    std::vector<double> r_mX, r_mY, r_mZ, r_mFS, r_mAdd;
 
 // Struct variables that have special types
 #ifdef AMR_WIND_USE_W2A
-    // FFTW plan
-    fftw_plan plan;
+    // FFTW plan vector
+    std::vector<fftw_plan> plan_vector;
     // FFTW pointers to store modes ready for ifft
-    fftw_complex *eta_mptr = nullptr, *u_mptr = nullptr, *v_mptr = nullptr,
-                 *w_mptr = nullptr;
-    // ReadModes object
-    ReadModes rmodes;
+    fftw_complex *c_eta_mptr = nullptr, *c_u_mptr = nullptr,
+                 *c_v_mptr = nullptr, *c_w_mptr = nullptr;
+    // FFTW pointers to store modes ready for ifft
+    double *r_eta_mptr = nullptr, *r_u_mptr = nullptr, *r_v_mptr = nullptr,
+           *r_w_mptr = nullptr, *au_mptr = nullptr, *av_mptr = nullptr,
+           *aw_mptr = nullptr;
+    // ReadModes objects
+    ReadModes<std::complex<double>> c_rmodes;
+    ReadModes<double> r_rmodes;
 #endif
 
     // Height vector (where velocity is sampled) stuff
     amrex::Vector<amrex::Real> hvec;
     // Index vector (where hvec overlaps with local boxes)
     amrex::Vector<int> indvec;
+    // Flag indicating whether the source simulation used HOS-Ocean or HOS-NWT
+    bool is_ocean{true};
     // Flag indicating interpolation should take place on this processor
     bool do_interp{true};
     // Flag indicating regrid has occurred since the last resizing of vecs
     bool resize_flag{false};
     // Vectors for spatial data from transformed modes
     amrex::Gpu::DeviceVector<amrex::Real> sp_eta_vec, sp_u_vec, sp_v_vec,
         sp_w_vec;
-    // Spacing of spatial data
-    amrex::Real sp_dx{0.}, sp_dy{0.};
 
     ~W2AWavesData()
     {
 #ifdef AMR_WIND_USE_W2A
-        if (eta_mptr) {
-            delete[] eta_mptr;
+        if (c_eta_mptr) {
+            delete[] c_eta_mptr;
+        }
+        if (c_u_mptr) {
+            delete[] c_u_mptr;
+        }
+        if (c_v_mptr) {
+            delete[] c_v_mptr;
+        }
+        if (c_w_mptr) {
+            delete[] c_w_mptr;
+        }
+        if (r_eta_mptr) {
+            delete[] r_eta_mptr;
+        }
+        if (r_u_mptr) {
+            delete[] r_u_mptr;
+        }
+        if (r_v_mptr) {
+            delete[] r_v_mptr;
+        }
+        if (r_w_mptr) {
+            delete[] r_w_mptr;
+        }
+        if (au_mptr) {
+            delete[] au_mptr;
         }
-        if (u_mptr) {
-            delete[] u_mptr;
+        if (av_mptr) {
+            delete[] av_mptr;
         }
-        if (v_mptr) {
-            delete[] v_mptr;
+        if (aw_mptr) {
+            delete[] aw_mptr;
         }
-        if (w_mptr) {
-            delete[] w_mptr;
+        for (int n = 0; n < (int)plan_vector.size(); ++n) {
+            fftw_destroy_plan(plan_vector[n]);
         }
-        fftw_destroy_plan(plan);
+        plan_vector.clear();
 #endif
     }
 };