1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
|
#ifndef WARPX_COARSEN_IO_H_
#define WARPX_COARSEN_IO_H_
#include "WarpX.H"
#include <AMReX_Math.H>
namespace CoarsenIO{
using namespace amrex;
/**
* \brief Interpolates the floating point data contained in the source Array4
* \c arr_src, extracted from a fine MultiFab, by averaging over either
* 1 point or 2 equally distant points.
*
* \param[in] arr_src floating point data to be interpolated
* \param[in] sf staggering of the source fine MultiFab
* \param[in] sc staggering of the destination coarsened MultiFab
* \param[in] cr coarsening ratio along each spatial direction
* \param[in] i index along x of the coarsened Array4 to be filled
* \param[in] j index along y of the coarsened Array4 to be filled
* \param[in] k index along z of the coarsened Array4 to be filled
* \param[in] comp index along the fourth component of the Array4 \c arr_src
* containing the data to be interpolated
*
* \return interpolated field at cell (i,j,k) of a coarsened Array4
*/
AMREX_GPU_DEVICE
AMREX_FORCE_INLINE
Real Interp ( Array4<Real const> const& arr_src,
GpuArray<int,3> const& sf,
GpuArray<int,3> const& sc,
GpuArray<int,3> const& cr,
const int i,
const int j,
const int k,
const int comp )
{
// Indices of destination array (coarse)
const int ic[3] = { i, j, k };
// Number of points and starting indices of source array (fine)
int np[3], idx_min[3];
// Compute number of points
for ( int l = 0; l < 3; ++l ) {
if ( cr[l] == 1 ) np[l] = 1+amrex::Math::abs(sf[l]-sc[l]); // no coarsening
else np[l] = 2-sf[l];
}
// Compute starting indices of source array (fine)
for ( int l = 0; l < 3; ++l ) {
if ( cr[l] == 1 ) idx_min[l] = ic[l]-sc[l]*(1-sf[l]); // no coarsening
else idx_min[l] = ic[l]*cr[l]+static_cast<int>(cr[l]/2)*(1-sc[l])-(1-sf[l]);
}
// Auxiliary integer variables
const int numx = np[0];
const int numy = np[1];
const int numz = np[2];
const int imin = idx_min[0];
const int jmin = idx_min[1];
const int kmin = idx_min[2];
int ii, jj, kk;
Real wx, wy, wz;
// Interpolate over points computed above
Real c = 0.0_rt;
for (int kref = 0; kref < numz; ++kref) {
for (int jref = 0; jref < numy; ++jref) {
for (int iref = 0; iref < numx; ++iref) {
ii = imin+iref;
jj = jmin+jref;
kk = kmin+kref;
wx = 1.0_rt/static_cast<Real>(numx);
wy = 1.0_rt/static_cast<Real>(numy);
wz = 1.0_rt/static_cast<Real>(numz);
c += wx*wy*wz*arr_src(ii,jj,kk,comp);
}
}
}
return c;
}
/**
* \brief Loops over the boxes of the coarsened MultiFab \c mf_dst and fills
* them by interpolating the data contained in the fine MultiFab \c mf_src.
*
* \param[in,out] mf_dst coarsened MultiFab containing the floating point data
* to be filled by interpolating the source fine MultiFab
* \param[in] mf_src fine MultiFab containing the floating point data to be interpolated
* \param[in] dcomp offset for the fourth component of the coarsened Array4
* object, extracted from its MultiFab, where the interpolated
* values will be stored
* \param[in] scomp offset for the fourth component of the fine Array4
* object, extracted from its MultiFab, containing the
* data to be interpolated
* \param[in] ncomp number of components to loop over for the coarsened
* Array4 extracted from the coarsened MultiFab \c mf_dst
* \param[in] ngrow number of guard cells to fill
* \param[in] crse_ratio coarsening ratio between the fine MultiFab \c mf_src
* and the coarsened MultiFab \c mf_dst along each spatial direction
*/
void Loop ( MultiFab& mf_dst,
const MultiFab& mf_src,
const int dcomp,
const int scomp,
const int ncomp,
const IntVect ngrow,
const IntVect crse_ratio=IntVect(1) );
/**
* \brief Stores in the coarsened MultiFab \c mf_dst the values obtained by
* interpolating the data contained in the fine MultiFab \c mf_src.
*
* \param[in,out] mf_dst coarsened MultiFab containing the floating point data
* to be filled by interpolating the fine MultiFab \c mf_src
* \param[in] mf_src fine MultiFab containing the floating point data to be interpolated
* \param[in] dcomp offset for the fourth component of the coarsened Array4
* object, extracted from its MultiFab, where the interpolated
* values will be stored
* \param[in] scomp offset for the fourth component of the fine Array4
* object, extracted from its MultiFab, containing the
* data to be interpolated
* \param[in] ncomp number of components to loop over for the coarsened
* Array4 extracted from the coarsened MultiFab \c mf_dst
* \param[in] ngrow number of guard cells to fill
* \param[in] crse_ratio coarsening ratio between the fine MultiFab \c mf_src
* and the coarsened MultiFab \c mf_dst along each spatial direction
*/
void Coarsen ( MultiFab& mf_dst,
const MultiFab& mf_src,
const int dcomp,
const int scomp,
const int ncomp,
const int ngrow,
const IntVect crse_ratio=IntVect(1) );
void Coarsen ( MultiFab& mf_dst,
const MultiFab& mf_src,
const int dcomp,
const int scomp,
const int ncomp,
const IntVect ngrowvect,
const IntVect crse_ratio=IntVect(1) );
}
#endif // WARPX_COARSEN_IO_H_
|
