32 const amrex::Real tol,
36 const amrex::Real omega = 2.0_rt *
static_cast<amrex::Real
>(M_PI) / T;
37 const amrex::Real depsdk = H / 2.0_rt;
42 amrex::Real k = (omega * omega) / g;
44 amrex::Real f = tol + 1.0_rt;
45 while (std::abs(f) > tol && iter < iter_max) {
49 const amrex::Real S = 2.0_rt * std::exp(2.0_rt * k * d) /
50 (std::exp(4.0_rt * k * d) + 1.0_rt);
51 const amrex::Real C = 1.0_rt - S;
52 const amrex::Real eps = k * H / 2.0_rt;
53 const amrex::Real C0 = std::sqrt(std::tanh(k * d));
54 const amrex::Real C2 =
55 C0 * (2.0_rt + 7.0_rt * S * S) / (4.0_rt * C * C);
56 const amrex::Real numterm_C4 =
57 (4.0_rt + 32.0_rt * S - 116.0_rt * std::pow(S, 2.0_rt) -
58 400.0_rt * std::pow(S, 3.0_rt) - 71.0_rt * std::pow(S, 4.0_rt) +
59 146.0_rt * std::pow(S, 5.0_rt));
60 const amrex::Real C4 =
61 C0 * numterm_C4 / (32.0_rt * std::pow(C, 5.0_rt));
63 const amrex::Real dSdk = -2.0_rt * d * std::sinh(2.0_rt * k * d) /
64 std::pow(std::cosh(2.0_rt * k * d), 2.0_rt);
65 const amrex::Real dCdk = -dSdk;
66 const amrex::Real dC0dk =
67 d / (2.0_rt * C0 * std::pow(std::cosh(k * d), 2.0_rt));
69 const amrex::Real dC2dk =
70 (4.0_rt * std::pow(C, 2.0_rt) *
71 (dC0dk * (2.0_rt + 7.0_rt * std::pow(S, 2.0_rt)) +
72 C0 * 14.0_rt * S * dSdk) -
73 C0 * (2.0_rt + 7.0_rt * std::pow(S, 2.0_rt)) * 8.0_rt * C * dCdk) /
74 (16.0_rt * std::pow(C, 4.0_rt));
75 const amrex::Real dC4dk =
76 (32.0_rt * std::pow(C, 5.0_rt) *
78 C0 * (32.0_rt * dSdk - 232.0_rt * S * dSdk -
79 1200.0_rt * std::pow(S, 2.0_rt) * dSdk -
80 284.0_rt * std::pow(S, 3.0_rt) * dSdk +
81 730.0_rt * std::pow(S, 4.0_rt))) -
82 C0 * numterm_C4 * 160.0_rt * std::pow(C, 4.0_rt) * dCdk) /
83 (1024.0_rt * std::pow(C, 10.0_rt));
87 g * std::pow(C0, 2.0_rt) + g * k * 2.0_rt * C0 * dC0dk;
90 dfdk += g * (2.0_rt * C0 * std::pow(eps, 2.0_rt) * C2 +
91 std::pow(eps, 4.0_rt) * std::pow(C2, 2.0_rt)) +
93 (2.0_rt * dC0dk * std::pow(eps, 2.0_rt) * C2 +
95 (2.0_rt * eps * depsdk * C2 +
96 std::pow(eps, 2.0_rt) * dC2dk) +
97 4.0_rt * std::pow(eps, 3.0_rt) * std::pow(C2, 2.0_rt) *
99 std::pow(eps, 4.0_rt) * 2.0_rt * C2 * dC2dk);
102 dfdk += g * (2.0_rt * std::pow(eps, 4.0_rt) * C0 * C4 +
103 2.0_rt * std::pow(eps, 6.0_rt) * C2 * C4 +
104 std::pow(eps, 8.0_rt) * std::pow(C4, 2.0_rt)) +
106 (8.0_rt * std::pow(eps, 3.0_rt) * depsdk * C0 * C4 +
107 2.0_rt * std::pow(eps, 4.0_rt) *
108 (C0 * dC4dk + C4 * dC0dk) +
109 12.0_rt * std::pow(eps, 5.0_rt) * depsdk * C2 * C4 +
110 2.0_rt * std::pow(eps, 6.0_rt) *
111 (C2 * dC4dk + C4 * dC2dk) +
112 8.0_rt * std::pow(eps, 7.0_rt) * depsdk *
113 std::pow(C4, 2.0_rt) +
114 std::pow(eps, 8.0_rt) * 2.0_rt * C4 * dC4dk);
119 f = g * k * std::pow(C0, 2.0_rt);
123 (2.0_rt * C0 * std::pow(eps, 2.0_rt) * C2 +
124 std::pow(eps, 4.0_rt) * std::pow(C2, 2.0_rt));
128 (2.0_rt * std::pow(eps, 4.0_rt) * C0 * C4 +
129 2.0_rt * std::pow(eps, 6.0_rt) * C2 * C4 +
130 std::pow(eps, 8.0_rt) * std::pow(C4, 2.0_rt));
143 if (iter == iter_max) {
147 return 2.0_rt *
static_cast<amrex::Real
>(M_PI) / k;
152 amrex::Real wavenumber,
153 amrex::Real water_depth,
174 amrex::Real kd = wavenumber * water_depth;
175 if (kd > 50.0_rt *
static_cast<amrex::Real
>(M_PI)) {
176 kd = 50.0_rt *
static_cast<amrex::Real
>(M_PI);
181 2.0_rt * std::exp(2.0_rt * kd) / (std::exp(4.0_rt * kd) + 1.0_rt);
182 amrex::Real C = 1.0_rt - S;
183 amrex::Real Sh = std::sinh(kd);
184 amrex::Real Th = std::tanh(kd);
187 (1.0_rt + std::exp(-2.0_rt * kd)) / (1.0_rt - std::exp(-2 * kd));
190 a11 = 1.0_rt / std::sinh(kd);
192 a22 = 3.0_rt * std::pow(S, 2.0_rt) / (2.0_rt * std::pow(C, 2.0_rt));
193 b22 = CTh * (1.0_rt + 2.0_rt * S) / (2.0_rt * C);
194 c2 = std::sqrt(Th) * (2.0_rt + 7.0_rt * std::pow(S, 2.0_rt)) /
195 (4.0_rt * std::pow(C, 2.0_rt));
196 if (stokes_order == 2) {
201 a31 = (-4.0_rt - 20.0_rt * S + 10.0_rt * std::pow(S, 2.0_rt) -
202 13.0_rt * std::pow(S, 3.0_rt)) /
203 (8.0_rt * Sh * std::pow(C, 3.0_rt));
204 a33 = (-2.0_rt * std::pow(S, 2.0_rt) + 11.0_rt * std::pow(S, 3.0_rt)) /
205 (8.0_rt * Sh * std::pow(C, 3.0_rt));
207 (1.0_rt + 3.0_rt * S + 3.0_rt * std::pow(S, 2.0_rt) +
208 2.0_rt * std::pow(S, 3.0_rt)) /
209 (8.0_rt * std::pow(C, 3.0_rt));
210 if (stokes_order == 3) {
215 a42 = (12.0_rt * S - 14.0_rt * std::pow(S, 2.0_rt) -
216 264.0_rt * std::pow(S, 3.0_rt) - 45.0_rt * std::pow(S, 4.0_rt) -
217 13.0_rt * std::pow(S, 5.0_rt)) /
218 (24.0_rt * std::pow(C, 5.0_rt));
219 a44 = (10.0_rt * std::pow(S, 3.0_rt) - 174.0_rt * std::pow(S, 4.0_rt) +
220 291.0_rt * std::pow(S, 5.0_rt) + 278.0_rt * std::pow(S, 6.0_rt)) /
221 (48.0_rt * (3.0_rt + 2.0_rt * S) * std::pow(C, 5.0_rt));
223 (6.0_rt - 26.0_rt * S - 182.0_rt * std::pow(S, 2.0_rt) -
224 204.0_rt * std::pow(S, 3.0_rt) - 25.0_rt * std::pow(S, 4.0_rt) +
225 26.0_rt * std::pow(S, 5.0_rt)) /
226 (6.0_rt * (3.0_rt + 2.0_rt * S) * std::pow(C, 4.0_rt));
228 (24.0_rt + 92.0_rt * S + 122.0_rt * std::pow(S, 2.0_rt) +
229 66.0_rt * std::pow(S, 3.0_rt) + 67.0_rt * std::pow(S, 4.0_rt) +
230 34.0_rt * std::pow(S, 5.0_rt)) /
231 (24.0_rt * (3.0_rt + 2.0_rt * S) * std::pow(C, 4.0_rt));
233 (4.0_rt + 32.0_rt * S - 116.0_rt * std::pow(S, 2.0_rt) -
234 400.0_rt * std::pow(S, 3.0_rt) - 71.0_rt * std::pow(S, 4.0_rt) +
235 146.0_rt * std::pow(S, 5.0_rt)) /
236 (32.0_rt * std::pow(C, 5.0_rt));
237 if (stokes_order == 4) {
242 a51 = (-1184.0_rt + 32.0_rt * S + 13232.0_rt * std::pow(S, 2.0_rt) +
243 21712.0_rt * std::pow(S, 3.0_rt) + 20940.0_rt * std::pow(S, 4.0_rt) +
244 12554.0_rt * std::pow(S, 5.0_rt) - 500.0_rt * std::pow(S, 6.0_rt) -
245 3341.0_rt * std::pow(S, 7.0_rt) - 670.0_rt * std::pow(S, 8.0_rt)) /
246 (64.0_rt * Sh * (3.0_rt + 2.0_rt * S) * (4.0_rt + S) *
247 std::pow(C, 6.0_rt));
248 a53 = (4.0_rt * S + 105.0_rt * std::pow(S, 2.0_rt) +
249 198.0_rt * std::pow(S, 3.0_rt) - 1376.0_rt * std::pow(S, 4.0_rt) -
250 1302.0_rt * std::pow(S, 5.0_rt) - 117.0_rt * std::pow(S, 6.0_rt) +
251 58.0_rt * std::pow(S, 7.0_rt)) /
252 (32.0_rt * Sh * (3.0_rt + 2.0_rt * S) * std::pow(C, 6.0_rt));
253 a55 = (-6.0_rt * std::pow(S, 3.0_rt) + 272.0_rt * std::pow(S, 4.0_rt) -
254 1552.0_rt * std::pow(S, 5.0_rt) + 852.0_rt * std::pow(S, 6.0_rt) +
255 2029.0_rt * std::pow(S, 7.0_rt) + 430.0_rt * std::pow(S, 8.0_rt)) /
256 (64.0_rt * Sh * (3.0_rt + 2.0_rt * S) * (4.0_rt + S) *
257 std::pow(C, 6.0_rt));
260 (132.0_rt + 17.0_rt * S - 2216.0_rt * std::pow(S, 2.0_rt) -
261 5897.0_rt * std::pow(S, 3.0_rt) - 6292.0_rt * std::pow(S, 4.0_rt) -
262 2687.0_rt * std::pow(S, 5.0_rt) + 194.0_rt * std::pow(S, 6.0_rt) +
263 467.0_rt * std::pow(S, 7.0_rt) + 82.0_rt * std::pow(S, 8.0_rt)) /
264 (128.0_rt * (3.0_rt + 2.0_rt * S) * (4.0_rt + S) * std::pow(C, 6.0_rt));
267 (300.0_rt + 1579.0_rt * S + 3176.0_rt * std::pow(S, 2.0_rt) +
268 2949.0_rt * std::pow(S, 3.0_rt) + 1188.0_rt * std::pow(S, 4.0_rt) +
269 675.0_rt * std::pow(S, 5.0_rt) + 1326.0_rt * std::pow(S, 6.0_rt) +
270 827.0_rt * std::pow(S, 7.0_rt) + 130.0_rt * std::pow(S, 8.0_rt)) /
271 (384.0_rt * (3.0_rt + 2.0_rt * S) * (4.0_rt + S) * std::pow(C, 6.0_rt));
272 if (stokes_order == 5) {
276 if (stokes_order > 5 || stokes_order < 2) {
278 "invalid stokes order specified. It should be between 2,3,4 or 5 ");
285 amrex::Real wavelength,
286 amrex::Real water_depth,
287 amrex::Real wave_height,
293 amrex::Real phase_offset,
299 const amrex::Real wavenumber =
300 2.0_rt *
static_cast<amrex::Real
>(M_PI) / wavelength;
303 amrex::Real c0{0.0_rt};
304 amrex::Real a11{0.0_rt}, a22{0.0_rt}, b22{0.0_rt}, c2{0.0_rt};
305 amrex::Real a31{0.0_rt}, a33{0.0_rt}, b31{0.0_rt}, a42{0.0_rt}, a44{0.0_rt};
306 amrex::Real b42{0.0_rt}, b44{0.0_rt}, c4{0.0_rt};
307 amrex::Real a51{0.0_rt}, a53{0.0_rt}, a55{0.0_rt}, b53{0.0_rt}, b55{0.0_rt};
310 stokes_order, wavenumber, water_depth, c0, a11, a22, b22, c2, a31, a33,
311 b31, a42, a44, b42, b44, c4, a51, a53, a55, b53, b55);
313 const amrex::Real eps = wavenumber * wave_height / 2.0_rt;
314 const amrex::Real c =
315 (c0 + std::pow(eps, 2.0_rt) * c2 + std::pow(eps, 4.0_rt) * c4) *
316 std::sqrt(g / wavenumber);
318 const amrex::Real omega = c * wavenumber;
319 const amrex::Real phase = wavenumber * x - omega * time - phase_offset;
321 eta = (eps * std::cos(phase)
322 + std::pow(eps, 2.0_rt) * b22 *
323 std::cos(2.0_rt * phase)
324 + std::pow(eps, 3.0_rt) * b31 *
325 (std::cos(phase) - std::cos(3.0_rt * phase)) +
326 std::pow(eps, 4.0_rt) * (b42 * std::cos(2.0_rt * phase) +
327 b44 * std::cos(4.0_rt * phase)) +
328 std::pow(eps, 5.0_rt) * (-(b53 + b55) * std::cos(phase) +
329 b53 * std::cos(3.0_rt * phase) +
330 b55 * std::cos(5.0_rt * phase))) /
338 const int MAX_ORDER = 5;
339 amrex::GpuArray<amrex::Real, MAX_ORDER> a;
340 if (stokes_order == 2) {
344 if (stokes_order == 3) {
345 a[0] = a11 + (eps * eps) * a31;
349 if (stokes_order == 4) {
350 a[0] = a11 + (eps * eps) * a31;
351 a[1] = a22 + (eps * eps) * a42;
355 if (stokes_order == 5) {
356 a[0] = a11 + (eps * eps) * a31 + std::pow(eps, 4.0_rt) * a51;
357 a[1] = a22 + (eps * eps) * a42;
358 a[2] = a33 + (eps * eps) * a53;
366 for (
int n = 1; n <= stokes_order; ++n) {
369 amrex::Real nkdz = n * wavenumber * (water_depth + (z - zsl));
370 if (nkdz > 50.0_rt *
static_cast<amrex::Real
>(M_PI)) {
371 nkdz = 50.0_rt *
static_cast<amrex::Real
>(M_PI);
373 u_w += std::pow(eps,
static_cast<amrex::Real
>(n)) *
374 static_cast<amrex::Real
>(n) * a[n - 1] * std::cosh(nkdz) *
375 std::cos(
static_cast<amrex::Real
>(n) * phase);
376 w_w += std::pow(eps,
static_cast<amrex::Real
>(n)) *
377 static_cast<amrex::Real
>(n) * a[n - 1] * std::sinh(nkdz) *
378 std::sin(
static_cast<amrex::Real
>(n) * phase);
380 u_w *= (c0 * std::sqrt(g / wavenumber));
381 w_w *= (c0 * std::sqrt(g / wavenumber));
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void stokes_coefficients(int stokes_order, amrex::Real wavenumber, amrex::Real water_depth, amrex::Real &c0, amrex::Real &a11, amrex::Real &a22, amrex::Real &b22, amrex::Real &c2, amrex::Real &a31, amrex::Real &a33, amrex::Real &b31, amrex::Real &a42, amrex::Real &a44, amrex::Real &b42, amrex::Real &b44, amrex::Real &c4, amrex::Real &a51, amrex::Real &a53, amrex::Real &a55, amrex::Real &b53, amrex::Real &b55)
Definition stokes_waves_K.H:150