Block-Structured AMR Software Framework
AMReX_COORDSYS_1D_C.H
Go to the documentation of this file.
1 #ifndef AMREX_COORDSYS_1D_C_H_
2 #define AMREX_COORDSYS_1D_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 
10 namespace amrex {
11 
12 AMREX_GPU_HOST_DEVICE
13 inline
14 void
15 amrex_setvol (Box const& bx, Array4<Real> const& vol,
16  GpuArray<Real,1> const& offset,
17  GpuArray<Real,1> 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];
25  AMREX_PRAGMA_SIMD
26  for (int i = lo.x; i <= hi.x; ++i) {
27  vol(i,0,0) = dv;
28  }
29  }
30  else if (coord == 1)
31  {
32  const Real pi = Real(3.1415926535897932);
33  AMREX_PRAGMA_SIMD
34  for (int i = lo.x; i <= hi.x; ++i) {
35  Real ri = offset[0] + dx[0]*i;
36  Real ro = ri + dx[0];
37  Real v = pi*(ro-ri)*(ro + ri);
38  vol(i,0,0) = std::abs(v);
39  }
40  }
41  else
42  {
43  const Real pi = Real(3.1415926535897932);
44  AMREX_PRAGMA_SIMD
45  for (int i = lo.x; i <= hi.x; ++i) {
46  Real ri = offset[0] + dx[0]*i;
47  Real ro = ri + dx[0];
48  Real v = (Real(4./3.)*pi)*(ro-ri)*(ro*ro+ro*ri+ri*ri);
49  vol(i,0,0) = std::abs(v);
50  }
51  }
52 }
53 
54 AMREX_GPU_HOST_DEVICE
55 inline
56 void
57 amrex_setarea (Box const& bx, Array4<Real> const& area,
58  GpuArray<Real,1> const& offset,
59  GpuArray<Real,1> const& dx, const int /*dir*/, const int coord) noexcept
60 {
61  const auto lo = amrex::lbound(bx);
62  const auto hi = amrex::ubound(bx);
63 
64  if (coord == 0)
65  {
66  AMREX_PRAGMA_SIMD
67  for (int i = lo.x; i <= hi.x; ++i) {
68  area(i,0,0) = 1.0;
69  }
70  }
71  else if (coord == 1)
72  {
73  const Real pi = Real(3.1415926535897932);
74  AMREX_PRAGMA_SIMD
75  for (int i = lo.x; i <= hi.x; ++i) {
76  Real ri = offset[0] + dx[0]*i;
77  Real a = (Real(2.)*pi)*ri;
78  area(i,0,0) = std::abs(a);
79  }
80  }
81  else
82  {
83  const Real pi = Real(3.1415926535897932);
84  AMREX_PRAGMA_SIMD
85  for (int i = lo.x; i <= hi.x; ++i) {
86  Real ri = offset[0] + dx[0]*i;
87  Real a = (Real(4.0)*pi)*ri*ri;
88  area(i,0,0) = std::abs(a);
89  }
90  }
91 }
92 
93 AMREX_GPU_HOST_DEVICE
94 inline
95 void
96 amrex_setdloga (Box const& bx, Array4<Real> const& dloga,
97  GpuArray<Real,1> const& offset,
98  GpuArray<Real,1> const& dx, const int /*dir*/, const int coord) noexcept
99 {
100  const auto lo = amrex::lbound(bx);
101  const auto hi = amrex::ubound(bx);
102 
103  if (coord == 0)
104  {
105  AMREX_PRAGMA_SIMD
106  for (int i = lo.x; i <= hi.x; ++i) {
107  dloga(i,0,0) = 0.0;
108  }
109  }
110  else if (coord == 1)
111  {
112  AMREX_PRAGMA_SIMD
113  for (int i = lo.x; i <= hi.x; ++i) {
114  Real rc = offset[0] + dx[0]*(i+Real(0.5));
115  dloga(i,0,0) = Real(1.0)/rc;
116  }
117  }
118  else
119  {
120  AMREX_PRAGMA_SIMD
121  for (int i = lo.x; i <= hi.x; ++i) {
122  Real ri = offset[0] + dx[0]*(i+Real(0.5));
123  dloga(i,0,0) = Real(2.0)/ri;
124  }
125  }
126 }
127 
128 }
129 
130 #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::Math::pi
constexpr std::enable_if_t< std::is_floating_point_v< T >, T > pi()
Definition: AMReX_Math.H:62
amrex
Definition: AMReX_Amr.cpp:49
amrex::abs
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE T abs(const GpuComplex< T > &a_z) noexcept
Return the absolute value of a complex number.
Definition: AMReX_GpuComplex.H:356
amrex::amrex_setvol
AMREX_GPU_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
AMREX_GPU_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::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::amrex_setdloga
AMREX_GPU_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::Array4
Definition: AMReX_Array4.H:61
amrex::GpuArray
Definition: AMReX_Array.H:34