Block-Structured AMR Software Framework
 
Loading...
Searching...
No Matches
AMReX_COORDSYS_2D_C.H
Go to the documentation of this file.
1#ifndef AMREX_COORDSYS_2D_C_H_
2#define AMREX_COORDSYS_2D_C_H_
3#include <AMReX_Config.H>
4
5#include <AMReX_Gpu.H>
6#include <AMReX_Array.H>
7#include <AMReX_FArrayBox.H>
8#include <cmath>
9
10namespace amrex {
11
12AMREX_GPU_HOST_DEVICE
13inline
14void
15amrex_setvol (Box const& bx, Array4<Real> const& vol,
16 GpuArray<Real,2> const& offset,
17 GpuArray<Real,2> const& dx, const int coord) noexcept
18{
19 const auto lo = amrex::lbound(bx);
20 const auto hi = amrex::ubound(bx);
21
22 if (coord == 0) // Cartesian
23 {
24 Real dv = dx[0]*dx[1];
25 for (int j = lo.y; j <= hi.y; ++j) {
26 AMREX_PRAGMA_SIMD
27 for (int i = lo.x; i <= hi.x; ++i) {
28 vol(i,j,0) = dv;
29 }
30 }
31 }
32 else if (coord == 1) // r-z
33 {
34 const Real pi = Real(3.1415926535897932);
35 for (int j = lo.y; j <= hi.y; ++j) {
36 AMREX_PRAGMA_SIMD
37 for (int i = lo.x; i <= hi.x; ++i) {
38 Real ri = offset[0] + dx[0]*(i);
39 Real ro = ri + dx[0];
40 Real v = pi*dx[1]*dx[0]*(ro + ri);
41 vol(i,j,0) = std::abs(v);
42 }
43 }
44 }
45 else // r-theta
46 {
47 const Real pi = Real(3.1415926535897932);
48 for (int j = lo.y; j <= hi.y; ++j) {
49 Real ti = offset[1] + dx[1]*(j);
50 Real to = ti + dx[1];
51 Real tmp = (Real(2.)*pi)*(std::cos(ti)-std::cos(to))/Real(3.0);
52 AMREX_PRAGMA_SIMD
53 for (int i = lo.x; i <= hi.x; ++i) {
54 Real ri = offset[0] + dx[0]*(i);
55 Real ro = ri + dx[0];
56 Real v = tmp*(ro-ri)*(ro*ro+ro*ri+ri*ri);
57 vol(i,j,0) = std::abs(v);
58 }
59 }
60 }
61}
62
63AMREX_GPU_HOST_DEVICE
64inline
65void
66amrex_setarea (Box const& bx, Array4<Real> const& area,
67 GpuArray<Real,2> const& offset,
68 GpuArray<Real,2> const& dx, const int dir, const int coord) noexcept
69{
70 const auto lo = amrex::lbound(bx);
71 const auto hi = amrex::ubound(bx);
72
73 if (coord == 0)
74 {
75 Real a = (dir == 0) ? dx[1] : dx[0];
76 for (int j = lo.y; j <= hi.y; ++j) {
77 AMREX_PRAGMA_SIMD
78 for (int i = lo.x; i <= hi.x; ++i) {
79 area(i,j,0) = a;
80 }
81 }
82 }
83 else if (coord == 1)
84 {
85 const Real pi = Real(3.1415926535897932);
86 if (dir == 0)
87 {
88 for (int j = lo.y; j <= hi.y; ++j) {
89 AMREX_PRAGMA_SIMD
90 for (int i = lo.x; i <= hi.x; ++i) {
91 Real ri = offset[0] + dx[0]*(i);
92 Real a = std::abs((Real(2.)*pi)*ri*dx[1]);
93 area(i,j,0) = a;
94 }
95 }
96 }
97 else
98 {
99 for (int j = lo.y; j <= hi.y; ++j) {
100 AMREX_PRAGMA_SIMD
101 for (int i = lo.x; i <= hi.x; ++i) {
102 Real rc = offset[0] + dx[0]*(i+Real(0.5));
103 Real a = std::abs(dx[0]*(Real(2.)*pi)*rc);
104 area(i,j,0) = a;
105 }
106 }
107 }
108 }
109 else
110 {
111 const Real pi = Real(3.1415926535897932);
112 if (dir == 0)
113 {
114 for (int j = lo.y; j <= hi.y; ++j) {
115 Real ti = offset[1] + dx[1]*(j);
116 Real to = ti + dx[1];
117 Real tmp = (Real(2.)*pi)*(std::cos(ti)-std::cos(to));
118 AMREX_PRAGMA_SIMD
119 for (int i = lo.x; i <= hi.x; ++i) {
120 Real ri = offset[0] + dx[0]*(i);
121 Real a = std::abs(tmp*ri*ri);
122 area(i,j,0) = a;
123 }
124 }
125 }
126 else
127 {
128 for (int j = lo.y; j <= hi.y; ++j) {
129 Real ti = offset[1] + dx[1]*(j);
130 Real tmp = pi*std::sin(ti);
131 AMREX_PRAGMA_SIMD
132 for (int i = lo.x; i <= hi.x; ++i) {
133 Real ri = offset[0] + dx[0]*(i);
134 Real ro = ri + dx[0];
135 Real a = std::abs(tmp*(ro-ri)*(ro+ri));
136 area(i,j,0) = a;
137 }
138 }
139 }
140 }
141}
142
143AMREX_GPU_HOST_DEVICE
144inline
145void
146amrex_setdloga (Box const& bx, Array4<Real> const& dloga,
147 GpuArray<Real,2> const& offset,
148 GpuArray<Real,2> const& dx, const int dir, const int coord) noexcept
149{
150 const auto lo = amrex::lbound(bx);
151 const auto hi = amrex::ubound(bx);
152
153 if (coord == 0)
154 {
155 for (int j = lo.y; j <= hi.y; ++j) {
156 AMREX_PRAGMA_SIMD
157 for (int i = lo.x; i <= hi.x; ++i) {
158 dloga(i,j,0) = Real(0.0);
159 }
160 }
161 }
162 else if (coord == 1)
163 {
164 if (dir == 0)
165 {
166 for (int j = lo.y; j <= hi.y; ++j) {
167 AMREX_PRAGMA_SIMD
168 for (int i = lo.x; i <= hi.x; ++i) {
169 Real rc = offset[0] + dx[0]*(i+Real(0.5));
170 dloga(i,j,0) = Real(1.0)/rc;
171 }
172 }
173 }
174 else
175 {
176 for (int j = lo.y; j <= hi.y; ++j) {
177 AMREX_PRAGMA_SIMD
178 for (int i = lo.x; i <= hi.x; ++i) {
179 dloga(i,j,0) = Real(0.0);
180 }
181 }
182 }
183 }
184 else
185 {
186 if (dir == 0)
187 {
188 for (int j = lo.y; j <= hi.y; ++j) {
189 AMREX_PRAGMA_SIMD
190 for (int i = lo.x; i <= hi.x; ++i) {
191 Real rc = offset[0] + dx[0]*(i+Real(0.5));
192 dloga(i,j,0) = Real(2.0)/rc;
193 }
194 }
195 }
196 else
197 {
198 for (int j = lo.y; j <= hi.y; ++j) {
199 Real ti = offset[1] + dx[1]*(j);
200 Real to = ti + dx[1];
201 Real tmp = Real(1.0)/std::tan(Real(0.5)*(ti+to));
202 AMREX_PRAGMA_SIMD
203 for (int i = lo.x; i <= hi.x; ++i) {
204 Real rc = offset[0] + dx[0]*(i+Real(0.5) );
205 dloga(i,j,0) = tmp/rc;
206 }
207 }
208 }
209 }
210}
211
212}
213
214#endif
AMReX_Array.H
AMREX_PRAGMA_SIMD
#define AMREX_PRAGMA_SIMD
Definition AMReX_Extension.H:80
AMReX_FArrayBox.H
AMREX_GPU_HOST_DEVICE
#define AMREX_GPU_HOST_DEVICE
Definition AMReX_GpuQualifiers.H:20
AMReX_Gpu.H
offset
Array4< int const > offset
Definition AMReX_HypreMLABecLap.cpp:1089
amrex::Real
amrex_real Real
Floating Point Type for Fields.
Definition AMReX_REAL.H:79
amrex
Definition AMReX_Amr.cpp:49
amrex::ubound
__host__ __device__ Dim3 ubound(Array4< T > const &a) noexcept
Definition AMReX_Array4.H:319
amrex::amrex_setvol
__host__ __device__ void amrex_setvol(Box const &bx, Array4< Real > const &vol, GpuArray< Real, 1 > const &offset, GpuArray< Real, 1 > const &dx, const int coord) noexcept
Definition AMReX_COORDSYS_1D_C.H:15
amrex::amrex_setarea
__host__ __device__ void amrex_setarea(Box const &bx, Array4< Real > const &area, GpuArray< Real, 1 > const &offset, GpuArray< Real, 1 > const &dx, const int, const int coord) noexcept
Definition AMReX_COORDSYS_1D_C.H:57
amrex::amrex_setdloga
__host__ __device__ void amrex_setdloga(Box const &bx, Array4< Real > const &dloga, GpuArray< Real, 1 > const &offset, GpuArray< Real, 1 > const &dx, const int, const int coord) noexcept
Definition AMReX_COORDSYS_1D_C.H:96
amrex::lbound
__host__ __device__ Dim3 lbound(Array4< T > const &a) noexcept
Definition AMReX_Array4.H:312
amrex::Array4
Definition AMReX_Array4.H:61
amrex::GpuArray
Fixed-size array that can be used on GPU.
Definition AMReX_Array.H:40