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
|
/* Copyright 2020 David Grote
*
* This file is part of WarpX.
*
* License: BSD-3-Clause-LBNL
*/
#include "SpectralBinomialFilter.H"
#include <AMReX_REAL.H>
#include <cmath>
using namespace amrex::literals;
/* \brief Initialize the general filter array */
void
SpectralBinomialFilter::InitFilterArray (HankelTransform::RealVector const & kvec,
amrex::Real const dels,
int const npasses,
bool const compensation,
KFilterArray & filter)
{
const int N = kvec.size();
filter.resize(N);
amrex::Real* p_filter = filter.data();
amrex::Real const* p_kvec = kvec.data();
amrex::ParallelFor(N, [=] AMREX_GPU_DEVICE (int i) noexcept
{
amrex::Real const ss = std::sin(0.5_rt*p_kvec[i]*dels);
amrex::Real const ss2 = ss*ss;
amrex::Real filt = std::pow(1._rt - ss2, npasses);
if (compensation) {
filt *= (1._rt + npasses*ss2);
}
p_filter[i] = filt;
});
}
/* \brief Initialize the radial and longitudinal filter arrays */
void
SpectralBinomialFilter::InitFilterArray (RealKVector const & kr,
RealKVector const & kz,
amrex::RealVect const dx,
amrex::IntVect const filter_npass_each_dir,
bool const compensation)
{
// Note that this includes the kr values for all modes
InitFilterArray(kr, dx[0], filter_npass_each_dir[0], compensation, filter_r);
InitFilterArray(kz, dx[1], filter_npass_each_dir[1], compensation, filter_z);
}
|
