Block-Structured AMR Software Framework
 
•All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
Loading...
Searching...
No Matches
AMReX_MLPoisson_1D_K.H
Go to the documentation of this file.
1#ifndef AMREX_MLPOISSON_1D_K_H_
2#define AMREX_MLPOISSON_1D_K_H_
3#include <AMReX_Config.H>
4
5namespace amrex {
6
7template <typename T>
8AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
9void mlpoisson_adotx (int i, Array4<T> const& y,
10 Array4<T const> const& x,
11 T dhx) noexcept
12{
13 y(i,0,0) = dhx * (x(i-1,0,0) - T(2.0)*x(i,0,0) + x(i+1,0,0));
14}
15
16template <typename T>
17AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
18void mlpoisson_adotx_os (int i, Array4<T> const& y,
19 Array4<T const> const& x,
20 Array4<int const> const& osm,
21 T dhx) noexcept
22{
23 if (osm(i,0,0) == 0) {
24 y(i,0,0) = T(0.0);
25 } else {
26 y(i,0,0) = dhx * (x(i-1,0,0) - T(2.0)*x(i,0,0) + x(i+1,0,0));
27 }
28}
29
30template <typename T>
31AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
32void mlpoisson_adotx_m (int i, Array4<T> const& y,
33 Array4<T const> const& x,
34 T dhx, T dx, T probxlo) noexcept
35{
36 T rel = (probxlo + i *dx) * (probxlo + i *dx);
37 T rer = (probxlo +(i+1)*dx) * (probxlo +(i+1)*dx);
38 y(i,0,0) = dhx * (rel*x(i-1,0,0) - (rel+rer)*x(i,0,0) + rer*x(i+1,0,0));
39}
40
41template <typename T>
42AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
43void mlpoisson_flux_x (Box const& box, Array4<T> const& fx,
44 Array4<T const> const& sol, T dxinv) noexcept
45{
46 const auto lo = amrex::lbound(box);
47 const auto hi = amrex::ubound(box);
48
49 AMREX_PRAGMA_SIMD
50 for (int i = lo.x; i <= hi.x; ++i) {
51 fx(i,0,0) = dxinv*(sol(i,0,0)-sol(i-1,0,0));
52 }
53}
54
55template <typename T>
56AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
57void mlpoisson_flux_x_m (Box const& box, Array4<T> const& fx,
58 Array4<T const> const& sol, T dxinv,
59 T dx, T probxlo) noexcept
60{
61 const auto lo = amrex::lbound(box);
62 const auto hi = amrex::ubound(box);
63
64 AMREX_PRAGMA_SIMD
65 for (int i = lo.x; i <= hi.x; ++i) {
66 T re = (probxlo + i*dx) * (probxlo + i*dx);
67 fx(i,0,0) = dxinv*re*(sol(i,0,0)-sol(i-1,0,0));
68 }
69}
70
71template <typename T>
72AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
73void mlpoisson_flux_xface (Box const& box, Array4<T> const& fx,
74 Array4<T const> const& sol, T dxinv, int xlen) noexcept
75{
76 const auto lo = amrex::lbound(box);
77
78 int i = lo.x;
79 fx(i,0,0) = dxinv*(sol(i,0,0)-sol(i-1,0,0));
80 i += xlen;
81 fx(i,0,0) = dxinv*(sol(i,0,0)-sol(i-1,0,0));
82}
83
84template <typename T>
85AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
86void mlpoisson_flux_xface_m (Box const& box, Array4<T> const& fx,
87 Array4<T const> const& sol, T dxinv, int xlen,
88 T dx, T probxlo) noexcept
89{
90 const auto lo = amrex::lbound(box);
91
92 int i = lo.x;
93 T re = (probxlo + i*dx) * (probxlo + i*dx);
94 fx(i,0,0) = dxinv*re*(sol(i,0,0)-sol(i-1,0,0));
95 i += xlen;
96 re = (probxlo + i*dx) * (probxlo + i*dx);
97 fx(i,0,0) = dxinv*re*(sol(i,0,0)-sol(i-1,0,0));
98}
99
100template <typename T>
101AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
102void mlpoisson_gsrb (int i, int, int, Array4<T> const& phi, Array4<T const> const& rhs,
103 T dhx,
104 Array4<T const> const& f0, Array4<int const> const& m0,
105 Array4<T const> const& f1, Array4<int const> const& m1,
106 Box const& vbox, int redblack) noexcept
107{
108 const auto vlo = amrex::lbound(vbox);
109 const auto vhi = amrex::ubound(vbox);
110
111 T gamma = -dhx*T(2.0);
112
113 if ((i+redblack)%2 == 0) {
114 T cf0 = (i == vlo.x && m0(vlo.x-1,0,0) > 0)
115 ? f0(vlo.x,0,0) : T(0.0);
116 T cf1 = (i == vhi.x && m1(vhi.x+1,0,0) > 0)
117 ? f1(vhi.x,0,0) : T(0.0);
118
119 T g_m_d = gamma + dhx*(cf0+cf1);
120
121 T res = rhs(i,0,0) - gamma*phi(i,0,0)
122 - dhx*(phi(i-1,0,0) + phi(i+1,0,0));
123
124 phi(i,0,0) = phi(i,0,0) + res /g_m_d;
125 }
126}
127
128template <typename T>
129AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
130void mlpoisson_gsrb_os (int i, int, int, Array4<T> const& phi, Array4<T const> const& rhs,
131 Array4<int const> const& osm, T dhx,
132 Array4<T const> const& f0, Array4<int const> const& m0,
133 Array4<T const> const& f1, Array4<int const> const& m1,
134 Box const& vbox, int redblack) noexcept
135{
136 const auto vlo = amrex::lbound(vbox);
137 const auto vhi = amrex::ubound(vbox);
138
139 T gamma = -dhx*T(2.0);
140
141 if ((i+redblack)%2 == 0) {
142 if (osm(i,0,0) == 0) {
143 phi(i,0,0) = T(0.0);
144 } else {
145 T cf0 = (i == vlo.x && m0(vlo.x-1,0,0) > 0)
146 ? f0(vlo.x,0,0) : T(0.0);
147 T cf1 = (i == vhi.x && m1(vhi.x+1,0,0) > 0)
148 ? f1(vhi.x,0,0) : T(0.0);
149
150 T g_m_d = gamma + dhx*(cf0+cf1);
151
152 T res = rhs(i,0,0) - gamma*phi(i,0,0)
153 - dhx*(phi(i-1,0,0) + phi(i+1,0,0));
154
155 phi(i,0,0) = phi(i,0,0) + res /g_m_d;
156 }
157 }
158}
159
160template <typename T>
161AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
162void mlpoisson_gsrb_m (int i, int, int, Array4<T> const& phi, Array4<T const> const& rhs,
163 T dhx,
164 Array4<T const> const& f0, Array4<int const> const& m0,
165 Array4<T const> const& f1, Array4<int const> const& m1,
166 Box const& vbox, int redblack, T dx, T probxlo) noexcept
167{
168 const auto vlo = amrex::lbound(vbox);
169 const auto vhi = amrex::ubound(vbox);
170
171 if ((i+redblack)%2 == 0) {
172 T cf0 = (i == vlo.x && m0(vlo.x-1,0,0) > 0)
173 ? f0(vlo.x,0,0) : T(0.0);
174 T cf1 = (i == vhi.x && m1(vhi.x+1,0,0) > 0)
175 ? f1(vhi.x,0,0) : T(0.0);
176
177 T rel = (probxlo + i *dx) * (probxlo + i *dx);
178 T rer = (probxlo +(i+1)*dx) * (probxlo +(i+1)*dx);
179
180 T gamma = -dhx*(rel+rer);
181
182 T g_m_d = gamma + dhx*(rel*cf0+rer*cf1);
183
184 T res = rhs(i,0,0) - gamma*phi(i,0,0)
185 - dhx*(rel*phi(i-1,0,0) + rer*phi(i+1,0,0));
186
187 phi(i,0,0) = phi(i,0,0) + res /g_m_d;
188 }
189}
190
191template <typename T>
192AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
193void mlpoisson_jacobi (int i, int, int, Array4<T> const& phi, Array4<T const> const& rhs,
194 Array4<T const> const& Ax, T dhx,
195 Array4<T const> const& f0, Array4<int const> const& m0,
196 Array4<T const> const& f1, Array4<int const> const& m1,
197 Box const& vbox) noexcept
198{
199 const auto vlo = amrex::lbound(vbox);
200 const auto vhi = amrex::ubound(vbox);
201
202 T gamma = -dhx*T(2.0);
203
204 T cf0 = (i == vlo.x && m0(vlo.x-1,0,0) > 0)
205 ? f0(vlo.x,0,0) : T(0.0);
206 T cf1 = (i == vhi.x && m1(vhi.x+1,0,0) > 0)
207 ? f1(vhi.x,0,0) : T(0.0);
208
209 T g_m_d = gamma + dhx*(cf0+cf1);
210
211 phi(i,0,0) += T(2.0/3.0) * (rhs(i,0,0) - Ax(i,0,0)) / g_m_d;
212}
213
214template <typename T>
215AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
216void mlpoisson_jacobi_os (int i, int, int, Array4<T> const& phi, Array4<T const> const& rhs,
217 Array4<T const> const& Ax, Array4<int const> const& osm, T dhx,
218 Array4<T const> const& f0, Array4<int const> const& m0,
219 Array4<T const> const& f1, Array4<int const> const& m1,
220 Box const& vbox) noexcept
221{
222 const auto vlo = amrex::lbound(vbox);
223 const auto vhi = amrex::ubound(vbox);
224
225 if (osm(i,0,0) == 0) {
226 phi(i,0,0) = T(0.0);
227 } else {
228 T cf0 = (i == vlo.x && m0(vlo.x-1,0,0) > 0)
229 ? f0(vlo.x,0,0) : T(0.0);
230 T cf1 = (i == vhi.x && m1(vhi.x+1,0,0) > 0)
231 ? f1(vhi.x,0,0) : T(0.0);
232
233 T gamma = -dhx*T(2.0);
234 T g_m_d = gamma + dhx*(cf0+cf1);
235
236 phi(i,0,0) += T(2.0/3.0) * (rhs(i,0,0) - Ax(i,0,0)) / g_m_d;
237 }
238}
239
240template <typename T>
241AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
242void mlpoisson_jacobi_m (int i, int, int, Array4<T> const& phi, Array4<T const> const& rhs,
243 Array4<T const> const& Ax, T dhx,
244 Array4<T const> const& f0, Array4<int const> const& m0,
245 Array4<T const> const& f1, Array4<int const> const& m1,
246 Box const& vbox, T dx, T probxlo) noexcept
247{
248 const auto vlo = amrex::lbound(vbox);
249 const auto vhi = amrex::ubound(vbox);
250
251 T cf0 = (i == vlo.x && m0(vlo.x-1,0,0) > 0)
252 ? f0(vlo.x,0,0) : T(0.0);
253 T cf1 = (i == vhi.x && m1(vhi.x+1,0,0) > 0)
254 ? f1(vhi.x,0,0) : T(0.0);
255
256 T rel = (probxlo + i *dx) * (probxlo + i *dx);
257 T rer = (probxlo +(i+1)*dx) * (probxlo +(i+1)*dx);
258
259 T gamma = -dhx*(rel+rer);
260
261 T g_m_d = gamma + dhx*(rel*cf0+rer*cf1);
262
263 phi(i,0,0) += T(2.0/3.0) * (rhs(i,0,0) - Ax(i,0,0)) / g_m_d;
264}
265
266template <typename T>
267AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
268void mlpoisson_normalize (int i, int, int, Array4<T> const& x,
269 T dhx, T dx, T probxlo) noexcept
270{
271 T rel = (probxlo + i *dx) * (probxlo + i *dx);
272 T rer = (probxlo +(i+1)*dx) * (probxlo +(i+1)*dx);
273 x(i,0,0) /= (-dhx*(rel+rer));
274}
275
276}
277
278#endif
AMREX_PRAGMA_SIMD
#define AMREX_PRAGMA_SIMD
Definition AMReX_Extension.H:80
AMREX_FORCE_INLINE
#define AMREX_FORCE_INLINE
Definition AMReX_Extension.H:119
AMREX_GPU_HOST_DEVICE
#define AMREX_GPU_HOST_DEVICE
Definition AMReX_GpuQualifiers.H:20
amrex
Definition AMReX_Amr.cpp:49
amrex::mlpoisson_jacobi
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void mlpoisson_jacobi(int i, int, int, Array4< T > const &phi, Array4< T const > const &rhs, Array4< T const > const &Ax, T dhx, Array4< T const > const &f0, Array4< int const > const &m0, Array4< T const > const &f1, Array4< int const > const &m1, Box const &vbox) noexcept
Definition AMReX_MLPoisson_1D_K.H:193
amrex::mlpoisson_adotx_m
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void mlpoisson_adotx_m(int i, Array4< T > const &y, Array4< T const > const &x, T dhx, T dx, T probxlo) noexcept
Definition AMReX_MLPoisson_1D_K.H:32
amrex::mlpoisson_gsrb_m
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void mlpoisson_gsrb_m(int i, int, int, Array4< T > const &phi, Array4< T const > const &rhs, T dhx, Array4< T const > const &f0, Array4< int const > const &m0, Array4< T const > const &f1, Array4< int const > const &m1, Box const &vbox, int redblack, T dx, T probxlo) noexcept
Definition AMReX_MLPoisson_1D_K.H:162
amrex::mlpoisson_flux_x
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void mlpoisson_flux_x(Box const &box, Array4< T > const &fx, Array4< T const > const &sol, T dxinv) noexcept
Definition AMReX_MLPoisson_1D_K.H:43
amrex::mlpoisson_adotx
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void mlpoisson_adotx(int i, Array4< T > const &y, Array4< T const > const &x, T dhx) noexcept
Definition AMReX_MLPoisson_1D_K.H:9
amrex::mlpoisson_flux_xface_m
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void mlpoisson_flux_xface_m(Box const &box, Array4< T > const &fx, Array4< T const > const &sol, T dxinv, int xlen, T dx, T probxlo) noexcept
Definition AMReX_MLPoisson_1D_K.H:86
amrex::mlpoisson_jacobi_os
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void mlpoisson_jacobi_os(int i, int, int, Array4< T > const &phi, Array4< T const > const &rhs, Array4< T const > const &Ax, Array4< int const > const &osm, T dhx, Array4< T const > const &f0, Array4< int const > const &m0, Array4< T const > const &f1, Array4< int const > const &m1, Box const &vbox) noexcept
Definition AMReX_MLPoisson_1D_K.H:216
amrex::mlpoisson_flux_x_m
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void mlpoisson_flux_x_m(Box const &box, Array4< T > const &fx, Array4< T const > const &sol, T dxinv, T dx, T probxlo) noexcept
Definition AMReX_MLPoisson_1D_K.H:57
amrex::mlpoisson_gsrb
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void mlpoisson_gsrb(int i, int, int, Array4< T > const &phi, Array4< T const > const &rhs, T dhx, Array4< T const > const &f0, Array4< int const > const &m0, Array4< T const > const &f1, Array4< int const > const &m1, Box const &vbox, int redblack) noexcept
Definition AMReX_MLPoisson_1D_K.H:102
amrex::ubound
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE Dim3 ubound(Array4< T > const &a) noexcept
Definition AMReX_Array4.H:315
amrex::lbound
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE Dim3 lbound(Array4< T > const &a) noexcept
Definition AMReX_Array4.H:308
amrex::mlpoisson_flux_xface
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void mlpoisson_flux_xface(Box const &box, Array4< T > const &fx, Array4< T const > const &sol, T dxinv, int xlen) noexcept
Definition AMReX_MLPoisson_1D_K.H:73
amrex::mlpoisson_normalize
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void mlpoisson_normalize(int i, int, int, Array4< T > const &x, T dhx, T dx, T probxlo) noexcept
Definition AMReX_MLPoisson_1D_K.H:268
amrex::mlpoisson_gsrb_os
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void mlpoisson_gsrb_os(int i, int, int, Array4< T > const &phi, Array4< T const > const &rhs, Array4< int const > const &osm, T dhx, Array4< T const > const &f0, Array4< int const > const &m0, Array4< T const > const &f1, Array4< int const > const &m1, Box const &vbox, int redblack) noexcept
Definition AMReX_MLPoisson_1D_K.H:130
amrex::mlpoisson_adotx_os
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void mlpoisson_adotx_os(int i, Array4< T > const &y, Array4< T const > const &x, Array4< int const > const &osm, T dhx) noexcept
Definition AMReX_MLPoisson_1D_K.H:18
amrex::mlpoisson_jacobi_m
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE void mlpoisson_jacobi_m(int i, int, int, Array4< T > const &phi, Array4< T const > const &rhs, Array4< T const > const &Ax, T dhx, Array4< T const > const &f0, Array4< int const > const &m0, Array4< T const > const &f1, Array4< int const > const &m1, Box const &vbox, T dx, T probxlo) noexcept
Definition AMReX_MLPoisson_1D_K.H:242
amrex::Array4
Definition AMReX_Array4.H:61