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
|
#include "FlushFormatAscent.H"
#include "WarpX.H"
using namespace amrex;
void
FlushFormatAscent::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
{
#ifdef AMREX_USE_ASCENT
auto & warpx = WarpX::GetInstance();
// wrap mesh data
conduit::Node bp_mesh;
amrex::MultiLevelToBlueprint(
nlev, amrex::GetVecOfConstPtrs(mf), varnames, geom, time, iteration, warpx.refRatio(), bp_mesh);
WriteParticles(particle_diags, bp_mesh);
// If you want to save blueprint HDF5 files w/o using an Ascent
// extract, you can call the following AMReX helper:
// const auto step = istep[0];
// WriteBlueprintFiles(bp_mesh,"bp_export",step,"hdf5");
ascent::Ascent ascent;
conduit::Node opts;
opts["exceptions"] = "catch";
opts["mpi_comm"] = MPI_Comm_c2f(ParallelDescriptor::Communicator());
ascent.open(opts);
ascent.publish(bp_mesh);
conduit::Node actions;
ascent.execute(actions);
ascent.close();
#endif // AMREX_USE_ASCENT
}
#ifdef AMREX_USE_ASCENT
void
FlushFormatAscent::WriteParticles(const amrex::Vector<ParticleDiag>& particle_diags, conduit::Node& a_bp_mesh) const
{
// wrap particle data for each species
// we prefix the fields with "particle_{species_name}" b/c we
// want to to uniquely name all the fields that can be plotted
for (unsigned i = 0, n = particle_diags.size(); i < n; ++i) {
Vector<std::string> particle_varnames;
Vector<std::string> particle_int_varnames;
std::string prefix = "particle_" + particle_diags[i].getSpeciesName();
// Get pc for species
// auto& pc = mypc->GetParticleContainer(i);
WarpXParticleContainer* pc = particle_diags[i].getParticleContainer();
// get names of real comps
std::map<std::string, int> real_comps_map = pc->getParticleComps();
std::map<std::string, int>::const_iterator r_itr = real_comps_map.begin();
// TODO: Looking at other code paths, I am not sure compile time
// QED field is included in getParticleComps()?
while (r_itr != real_comps_map.end())
{
// get next real particle name
std::string varname = r_itr->first;
particle_varnames.push_back(prefix + "_" + varname);
r_itr++;
}
// get names of int comps
std::map<std::string, int> int_comps_map = pc->getParticleiComps();
std::map<std::string, int>::const_iterator i_itr = int_comps_map.begin();
while (i_itr != int_comps_map.end())
{
// get next real particle name
std::string varname = i_itr->first;
particle_int_varnames.push_back(prefix + "_" + varname);
i_itr++;
}
// wrap pc for current species into a blueprint topology
amrex::ParticleContainerToBlueprint(*pc,
particle_varnames,
particle_int_varnames,
a_bp_mesh,
prefix);
}
}
#endif // AMREX_USE_ASCENT
|
