blob: 4b02d1b7e3e0ff756dde3adeb8ae957b00b31f47 (
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
51
52
53
|
/* Copyright 2019 David Grote
*
* This file is part of WarpX.
*
* License: BSD-3-Clause-LBNL
*/
#include "Utils/TextMsg.H"
#include "Utils/WarpXConst.H"
#include "SpectralKSpaceRZ.H"
#include <cmath>
/* \brief Initialize k space object.
*
* \param realspace_ba Box array that corresponds to the decomposition
* of the fields in real space (cell-centered ; includes guard cells)
* \param dm Indicates which MPI proc owns which box, in realspace_ba.
* \param realspace_dx Cell size of the grid in real space
*/
SpectralKSpaceRZ::SpectralKSpaceRZ (const amrex::BoxArray& realspace_ba,
const amrex::DistributionMapping& dm,
const amrex::RealVect realspace_dx)
{
dx = realspace_dx; // Store the cell size as member `dx`
WARPX_ALWAYS_ASSERT_WITH_MESSAGE(
realspace_ba.ixType() == amrex::IndexType::TheCellType(),
"SpectralKSpaceRZ expects a cell-centered box.");
// Create the box array that corresponds to spectral space
amrex::BoxList spectral_bl; // Create empty box list
// Loop over boxes and fill the box list
for (int i=0; i < realspace_ba.size(); i++ ) {
// For local FFTs, boxes in spectral space start at 0 in
// each direction and have the same number of points as the
// (cell-centered) real space box
amrex::Box realspace_bx = realspace_ba[i];
amrex::IntVect fft_size = realspace_bx.length();
amrex::IntVect spectral_bx_size = fft_size;
// Define the corresponding box
amrex::Box spectral_bx = amrex::Box(amrex::IntVect::TheZeroVector(),
spectral_bx_size - amrex::IntVect::TheUnitVector() );
spectral_bl.push_back(spectral_bx);
}
spectralspace_ba.define(spectral_bl);
// Allocate the components of the kz vector
const int i_dim = 1;
const bool only_positive_k = false;
k_vec[i_dim] = getKComponent(dm, realspace_ba, i_dim, only_positive_k);
}
|
