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
|
/* Copyright 2021 Revathi Jambunathan
*
* This file is part of WarpX.
*
* License: BSD-3-Clause-LBNL
*/
#ifndef WARPX_BACKTRANSFORMFUNCTOR_H_
#define WARPX_BACKTRANSFORMFUNCTOR_H_
#include "ComputeDiagFunctor.H"
#include <AMReX_Box.H>
#include <AMReX_IntVect.H>
#include <AMReX_REAL.H>
#include <AMReX_Vector.H>
#include <AMReX_BaseFwd.H>
#include <string>
/**
* \brief Functor to back-transform cell-centered data and store result in mf_out
*
* The cell-centered data is a ten-component multifab with field-data averaged-down
* from the finest to coarsest level, and stored as single-level data.
* For every i^th buffer, a z-slice corresponding to the z-boost location of the
* slice at the current timestep is extracted. This slice containing field-data
* in the boosted-frame is Lorentz-transformed to the lab-frame. The user-requested
* lab-frame field data is then stored in mf_dst.
*/
class
BackTransformFunctor final : public ComputeDiagFunctor
{
public:
/** Constructor description
*
* \param[in] mf_src cell-centered multifab containing all user-requested
fields in boosted-frame
* \param[in] lev mesh-refinement level of multifab.
* \param[in] ncomp number of components of mf_src to Lorentz-Transform
and store in destination multifab.
* \param[in] num_buffers number of user-defined snapshots in the back-transformed lab-frame
* \param[in] varnames names of the field-components as defined by the user for back-transformed diagnostics.
* \param[in] varnames_fields base names of field-components for the RZ modes
* \param[in] crse_ratio the coarsening ratio for fields
*/
BackTransformFunctor ( const amrex::MultiFab * mf_src, int lev,
int ncomp, int num_buffers,
amrex::Vector< std::string > varnames,
amrex::Vector< std::string > varnames_fields,
amrex::IntVect crse_ratio= amrex::IntVect(1));
/** \brief Lorentz-transform mf_src for the ith buffer and write the result in mf_dst.
*
* The source multifab, is a ten-component cell-centered multifab storing
* field-data in the boosted-frame. An z-slice is generated
* at the z-boost location for the ith buffer, stored in m_current_z_boost[i_buffer].
* The data is then lorentz-transformed in-place using LorenzTransformZ ().
* The user-requested fields are then copied to mf_dst.
*
* \param[out] mf_dst output MuliFab where the back-transformed data is written
* \param[in] dcomp first component of mf_dst in which the back-transformed
* lab-frame data for the user-request fields is written.
* \param[in] i_buffer buffer index for which the data is transformed.
*/
void operator ()(amrex::MultiFab& mf_dst, int dcomp, int i_buffer) const override;
/** \brief Prepare data required to back-transform fields for lab-frame snapshot, i_buffer
*
* \param[in] i_buffer index of the snapshot
* \param[in] z_slice_in_domain if the z-slice at current_z_boost is within
* the boosted-frame and lab-frame domain.
* The fields are sliced and back-transformed only if this value is true.
* \param[in] current_z_boost current z coordinate in the boosted-frame
* \param[in] buffer_box Box with index-space in lab-frame for the ith buffer
* \param[in] k_index_zlab k-index in the lab-frame corresponding to the
* current z co-ordinate in the lab-frame for the ith buffer.
* \param[in] snapshot_full if the current snapshot, with index, i_buffer, is full (1)
or not (0). If it is full, then Lorentz-transform is not performed
by setting m_perform_backtransform to 0.
*/
void PrepareFunctorData ( int i_buffer, bool z_slice_in_domain,
amrex::Real current_z_boost,
amrex::Box buffer_box, int k_index_zlab,
int snapshot_full ) override;
/** Allocate and initialize member variables and arrays required to back-transform
* field-data from boosted-frame to lab-frame.
*/
void InitData () override;
/** \brief In-place Lorentz-transform of MultiFab, data, from boosted-frame to the
* lab-frame for all fields, Ex, Ey, Ez, Bx, By, Bz, jx, jy, jz, and rho.
*
* \param[in] data z-slice field-data MultiFab to be back-transformed in-place
* from boosted-frame to lab-frame
* \param[in] gamma_boost The Lorentz factor of the boosted-frame in which
the simulation is run
* \param[in] beta_boost The ratio of boost velocity to the speed of light
*/
void LorentzTransformZ (amrex::MultiFab& data, amrex::Real gamma_boost,
amrex::Real beta_boost) const;
private:
/** pointer to source multifab (cell-centered multi-component multifab) */
amrex::MultiFab const * const m_mf_src = nullptr;
/** level at which m_mf_src is defined */
int const m_lev;
/** Number of buffers or snapshots */
int const m_num_buffers;
/** Vector of amrex::Box with index-space in the lab-frame */
amrex::Vector<amrex::Box> m_buffer_box;
/** Vector of current z co-ordinate in the boosted-frame for each buffer */
amrex::Vector<amrex::Real> m_current_z_boost;
/** Vector of 0s and 1s stored to check if back-transformation is to be performed
* for the ith buffer. The value is set 0 (false) or 1 (true) using the boolean
* z_slice_in_domain in PrepareFunctorData().
*/
amrex::Vector<int> m_perform_backtransform;
/** Vector of k-index correspoding to the current lab-frame z co-ordinate for each buffer */
amrex::Vector<int> m_k_index_zlab;
/** Vector of user-defined field names to be stored in the output multifab */
amrex::Vector< std::string > m_varnames;
/** Vector of user-defined field names without modifications for rz modes */
amrex::Vector< std::string > m_varnames_fields;
/** Indices that map user-defined fields to plot to the fields
* stored in the cell-centered MultiFab, m_mf_src.
* The cell-centered MultiFab stores Ex, Ey, Ez, Bx, By, Bz, jx, jy, jz, and rho.
*/
amrex::Vector<int> m_map_varnames;
};
#endif
|
