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
|
#ifndef WARPX_FLUSHFORMATSENSEI_H_
#define WARPX_FLUSHFORMATSENSEI_H_
#include "FlushFormat.H"
#include <AMReX_AmrMesh.H>
#if defined(AMREX_USE_SENSEI_INSITU)
# include <AMReX_AmrMeshParticleInSituBridge.H>
#else
namespace amrex {
using AmrMeshParticleInSituBridge = void;
}
#endif
/**
* \brief This class aims at dumping performing in-situ diagnostics with
* SENSEI. In particular, function WriteToFile takes fields and particles as
* input arguments, and calls amrex functions to do the in-situ visualization.
*
* To use the SENSEI flush format one must compile WarpX with the GNUMakefile
* setting: USE_SENSEI_INSITU=TRUE
*
* In addition to the standard controls, the following inputs file fields are
* supported:
*
* sensei_config - the path to a SENSEI XML configuration (required)
* sensei_pin_mesh - integer 0 or 1 forcing the moving mesh to be fixed
* at 0,0,0 (optional)
*
* Fri 29 May 2020 11:19:38 AM PDT : Tested with SENSEI version 3.2.0
*/
class FlushFormatSensei : public FlushFormat
{
public:
FlushFormatSensei();
~FlushFormatSensei();
FlushFormatSensei(const FlushFormatSensei &) = delete;
void operator=(const FlushFormatSensei &) = delete;
/** Construct.
*
* \param[in] amr_mesh an AmrMesh instance
* \param[in] diag_name ParmParse scope string.
*/
FlushFormatSensei (amrex::AmrMesh *amr_mesh, std::string diag_name);
/** Do in-situ visualization for field and particle data */
virtual void WriteToFile (
amrex::Vector<std::string> varnames,
const amrex::Vector<amrex::MultiFab>& mf,
amrex::Vector<amrex::Geometry>& geom,
amrex::Vector<int> iteration, double time,
const amrex::Vector<ParticleDiag>& particle_diags, int nlev,
std::string prefix, int file_min_digits,
bool plot_raw_fields,
bool plot_raw_fields_guards,
bool use_pinned_pc = false,
bool isBTD = false, int snapshotID = -1,
int bufferID = 1, int numBuffers = 1,
const amrex::Geometry& full_BTD_snapshot = amrex::Geometry(),
bool isLastBTDFlush = false,
const amrex::Vector<int>& totalParticlesFlushedAlready = amrex::Vector<int>() ) const override;
/** \brief Do in-situ visualization for particle data.
* \param[in] particle_diags Each element of this vector handles output of 1 species.
*/
void WriteParticles(const amrex::Vector<ParticleDiag>& particle_diags) const;
private:
std::string m_insitu_config;
int m_insitu_pin_mesh = 0;
amrex::AmrMeshParticleInSituBridge * m_insitu_bridge = nullptr;
amrex::AmrMesh * m_amr_mesh = nullptr;
};
#endif // WARPX_FLUSHFORMATSENSEI_H_
|
