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
|
#include "FlushFormatSensei.H"
#include "WarpX.H"
#ifdef BL_USE_SENSEI_INSITU
# include <AMReX_AmrMeshInSituBridge.H>
#endif
FlushFormatSensei::FlushFormatSensei () :
m_insitu_config(), m_insitu_pin_mesh(0), m_insitu_bridge(nullptr),
m_amr_mesh(nullptr)
{}
FlushFormatSensei::FlushFormatSensei (amrex::AmrMesh *amr_mesh,
std::string diag_name) :
m_insitu_config(), m_insitu_pin_mesh(0), m_insitu_bridge(nullptr),
m_amr_mesh(amr_mesh)
{
#ifndef BL_USE_SENSEI_INSITU
(void)amr_mesh;
(void)diag_name;
#else
amrex::ParmParse pp(diag_name);
pp.query("sensei_config", m_insitu_config);
pp.query("sensei_pin_mesh", m_insitu_pin_mesh);
m_insitu_bridge = new amrex::AmrMeshInSituBridge;
m_insitu_bridge->setEnabled(true);
m_insitu_bridge->setConfig(m_insitu_config);
m_insitu_bridge->setPinMesh(m_insitu_pin_mesh);
if (!m_amr_mesh || m_insitu_bridge->initialize())
{
amrex::ErrorStream() << "FlushFormtSensei::FlushFormatSensei : "
"Failed to initialize the in situ bridge." << std::endl;
amrex::Abort();
}
m_insitu_bridge->setFrequency(1);
#endif
}
FlushFormatSensei::~FlushFormatSensei ()
{
#ifdef BL_USE_SENSEI_INSITU
delete m_insitu_bridge;
#endif
}
void
FlushFormatSensei::WriteToFile (
const amrex::Vector<std::string> varnames,
const amrex::Vector<amrex::MultiFab>& mf,
amrex::Vector<amrex::Geometry>& geom,
const amrex::Vector<int> iteration, const double time,
const amrex::Vector<ParticleDiag>& particle_diags, int nlev,
const std::string prefix, bool plot_raw_fields,
bool plot_raw_fields_guards, bool plot_raw_rho, bool plot_raw_F) const
{
#ifndef BL_USE_SENSEI_INSITU
(void)varnames;
(void)mf;
(void)geom;
(void)iteration;
(void)time;
(void)particle_diags;
(void)nlev;
(void)prefix;
(void)plot_raw_fields;
(void)plot_raw_fields_guards;
(void)plot_raw_rho;
(void)plot_raw_F;
#else
amrex::Vector<amrex::MultiFab> *mf_ptr =
const_cast<amrex::Vector<amrex::MultiFab>*>(&mf);
if (m_insitu_bridge->update(iteration[0], time, m_amr_mesh,
{mf_ptr}, {varnames}))
{
amrex::ErrorStream() << "FlushFormatSensei::WriteToFile : "
"Failed to update the in situ bridge." << std::endl;
amrex::Abort();
}
#endif
}
void
FlushFormatSensei::WriteParticles (
const amrex::Vector<ParticleDiag>& particle_diags) const
{
#ifndef BL_USE_SENSEI_INSITU
(void)particle_diags;
#else
amrex::ErrorStream() << "FlushFormatSensei::WriteParticles : "
"Not yet implemented." << std::endl;
amrex::Abort();
#endif
}
|
