blob: 5a87031c40c8a1918c7c2a1ca6e4e7473acfd602 (
plain) (
blame)
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
|
/* Copyright 2019-2020 David Grote
*
* This file is part of WarpX.
*
* License: BSD-3-Clause-LBNL
*/
#ifndef WARPX_HANKEL_TRANSFORM_H_
#define WARPX_HANKEL_TRANSFORM_H_
#include <AMReX_FArrayBox.H>
/* \brief This defines the class that performs the Hankel transform.
* Original authors: Remi Lehe, Manuel Kirchen
*
*
* Definition of the Hankel forward and backward transform of order p:
* g(kr) = \int_0^\infty f(r) J_p(kr r) r dr
* f(r ) = \int_0^\infty g(kr) J_p(kr r) kr dkr
*/
class HankelTransform
{
public:
using RealVector = amrex::Gpu::DeviceVector<amrex::Real>;
// Constructor
HankelTransform(const int hankel_order,
const int azimuthal_mode,
const int nr,
const amrex::Real rmax);
const RealVector & getSpectralWavenumbers() {return m_kr;}
void HankelForwardTransform(amrex::FArrayBox const& F, int const F_icomp,
amrex::FArrayBox & G, int const G_icomp);
void HankelInverseTransform(amrex::FArrayBox const& G, int const G_icomp,
amrex::FArrayBox & F, int const F_icomp);
private:
// Even though nk == nr always, use a seperate variable for clarity.
int m_nr, m_nk;
RealVector m_kr;
RealVector m_invM;
RealVector m_M;
};
#endif
|
