aboutsummaryrefslogtreecommitdiff
path: root/Source/Diagnostics/ReducedDiags
diff options
context:
space:
mode:
Diffstat (limited to 'Source/Diagnostics/ReducedDiags')
-rw-r--r--Source/Diagnostics/ReducedDiags/ParticleExtrema.H11
-rw-r--r--Source/Diagnostics/ReducedDiags/ParticleExtrema.cpp142
2 files changed, 77 insertions, 76 deletions
diff --git a/Source/Diagnostics/ReducedDiags/ParticleExtrema.H b/Source/Diagnostics/ReducedDiags/ParticleExtrema.H
index e58138dd8..b6bfb7c5e 100644
--- a/Source/Diagnostics/ReducedDiags/ParticleExtrema.H
+++ b/Source/Diagnostics/ReducedDiags/ParticleExtrema.H
@@ -10,6 +10,7 @@
#include "ReducedDiags.H"
+#include <map>
#include <string>
/**
@@ -36,6 +37,16 @@ public:
*/
void ComputeDiags(int step) override final;
+private:
+ /// auxiliary structure to store headers and indices of the reduced diagnostics
+ struct aux_header_index
+ {
+ std::string header;
+ int idx;
+ };
+
+ /// map to store header texts and indices of the reduced diagnostics
+ std::map<std::string, aux_header_index> m_headers_indices;
};
#endif
diff --git a/Source/Diagnostics/ReducedDiags/ParticleExtrema.cpp b/Source/Diagnostics/ReducedDiags/ParticleExtrema.cpp
index 3658a8c16..ed2d2ea13 100644
--- a/Source/Diagnostics/ReducedDiags/ParticleExtrema.cpp
+++ b/Source/Diagnostics/ReducedDiags/ParticleExtrema.cpp
@@ -82,18 +82,45 @@ ParticleExtrema::ParticleExtrema (std::string rd_name)
// get WarpXParticleContainer class object
auto & myspc = mypc.GetParticleContainer(i_s);
- if (myspc.DoQED())
- {
- // resize data array for QED species
- const int num_quantities = 18;
- m_data.resize(num_quantities,0.0);
- } else
- {
- // resize data array for regular species
- const int num_quantities = 16;
- m_data.resize(num_quantities,0.0);
+ auto all_diag_names = std::vector<std::string> {};
+ auto add_diag = [&,c=0] (
+ const std::string& name, const std::string& header) mutable {
+ m_headers_indices[name] = aux_header_index{header, c++};
+ all_diag_names.push_back(name);
+ };
+
+ add_diag("xmin", "xmin(m)");
+ add_diag("xmax", "xmax(m)");
+ add_diag("ymin", "ymin(m)");
+ add_diag("ymax", "ymax(m)");
+ add_diag("zmin", "zmin(m)");
+ add_diag("zmax", "zmax(m)");
+ add_diag("pxmin", "pxmin(kg*m/s)");
+ add_diag("pxmax", "pxmax(kg*m/s)");
+ add_diag("pymin", "pymin(kg*m/s)");
+ add_diag("pymax", "pymax(kg*m/s)");
+ add_diag("pzmin", "pzmin(kg*m/s)");
+ add_diag("pzmax", "pzmax(kg*m/s)");
+ add_diag("gmin", "gmin()");
+ add_diag("gmax", "gmax()");
+
+#if (defined WARPX_DIM_3D)
+ add_diag("wmin", "wmin()");
+ add_diag("wmax", "wmax()");
+#elif (defined(WARPX_DIM_XZ) || defined(WARPX_DIM_RZ))
+ add_diag("wmin", "wmin(1/m)");
+ add_diag("wmax", "wmax(1/m)");
+#else
+ add_diag("wmin", "wmin(1/m^2)");
+ add_diag("wmax", "wmax(1/m^2)");
+#endif
+ if (myspc.DoQED()){
+ add_diag("chimin", "chimin()");
+ add_diag("chimax", "chimax()");
}
+ m_data.resize(all_diag_names.size());
+
if (ParallelDescriptor::IOProcessor())
{
if ( m_IsNotRestart )
@@ -103,55 +130,14 @@ ParticleExtrema::ParticleExtrema (std::string rd_name)
ofs.open(m_path + m_rd_name + "." + m_extension,
std::ofstream::out | std::ofstream::app);
// write header row
- int c = 0;
+ int off = 0;
ofs << "#";
- ofs << "[" << c++ << "]step()";
- ofs << m_sep;
- ofs << "[" << c++ << "]time(s)";
- ofs << m_sep;
- ofs << "[" << c++ << "]xmin(m)";
- ofs << m_sep;
- ofs << "[" << c++ << "]xmax(m)";
- ofs << m_sep;
- ofs << "[" << c++ << "]ymin(m)";
- ofs << m_sep;
- ofs << "[" << c++ << "]ymax(m)";
- ofs << m_sep;
- ofs << "[" << c++ << "]zmin(m)";
- ofs << m_sep;
- ofs << "[" << c++ << "]zmax(m)";
- ofs << m_sep;
- ofs << "[" << c++ << "]pxmin(kg*m/s)";
+ ofs << "[" << off++ << "]step()";
ofs << m_sep;
- ofs << "[" << c++ << "]pxmax(kg*m/s)";
- ofs << m_sep;
- ofs << "[" << c++ << "]pymin(kg*m/s)";
- ofs << m_sep;
- ofs << "[" << c++ << "]pymax(kg*m/s)";
- ofs << m_sep;
- ofs << "[" << c++ << "]pzmin(kg*m/s)";
- ofs << m_sep;
- ofs << "[" << c++ << "]pzmax(kg*m/s)";
- ofs << m_sep;
- ofs << "[" << c++ << "]gmin()";
- ofs << m_sep;
- ofs << "[" << c++ << "]gmax()";
- ofs << m_sep;
-#if (defined WARPX_DIM_3D)
- ofs << "[" << c++ << "]wmin()";
- ofs << m_sep;
- ofs << "[" << c++ << "]wmax()";
-#else
- ofs << "[" << c++ << "]wmin(1/m)";
- ofs << m_sep;
- ofs << "[" << c++ << "]wmax(1/m)";
-#endif
- if (myspc.DoQED())
- {
- ofs << m_sep;
- ofs << "[" << c++ << "]chimin()";
- ofs << m_sep;
- ofs << "[" << c++ << "]chimax()";
+ ofs << "[" << off++ << "]time(s)";
+ for (const auto& name : all_diag_names){
+ const auto& el = m_headers_indices[name];
+ ofs << m_sep << "[" << el.idx + off << "]" << el.header;
}
ofs << std::endl;
// close file
@@ -424,7 +410,6 @@ void ParticleExtrema::ComputeDiags (int step)
// declare external fields
const int offset = 0;
const auto getExternalEB = GetExternalEBField(pti, offset);
-
// define variables in preparation for field gathering
amrex::Box box = pti.tilebox();
box.grow(ngEB);
@@ -487,27 +472,32 @@ void ParticleExtrema::ComputeDiags (int step)
ParallelDescriptor::ReduceRealMax(chimax_f);
}
#endif
- m_data[0] = xmin;
- m_data[1] = xmax;
- m_data[2] = ymin;
- m_data[3] = ymax;
- m_data[4] = zmin;
- m_data[5] = zmax;
- m_data[6] = uxmin*m;
- m_data[7] = uxmax*m;
- m_data[8] = uymin*m;
- m_data[9] = uymax*m;
- m_data[10] = uzmin*m;
- m_data[11] = uzmax*m;
- m_data[12] = gmin;
- m_data[13] = gmax;
- m_data[14] = wmin;
- m_data[15] = wmax;
+
+ const auto get_idx = [&](const std::string& name){
+ return m_headers_indices.at(name).idx;
+ };
+
+ m_data[get_idx("xmin")] = xmin;
+ m_data[get_idx("xmax")] = xmax;
+ m_data[get_idx("ymin")] = ymin;
+ m_data[get_idx("ymax")] = ymax;
+ m_data[get_idx("zmin")] = zmin;
+ m_data[get_idx("zmax")] = zmax;
+ m_data[get_idx("pxmin")] = uxmin*m;
+ m_data[get_idx("pxmax")] = uxmax*m;
+ m_data[get_idx("pymin")] = uymin*m;
+ m_data[get_idx("pymax")] = uymax*m;
+ m_data[get_idx("pzmin")] = uzmin*m;
+ m_data[get_idx("pzmax")] = uzmax*m;
+ m_data[get_idx("gmin")] = gmin;
+ m_data[get_idx("gmax")] = gmax;
+ m_data[get_idx("wmin")] = wmin;
+ m_data[get_idx("wmax")] = wmax;
#if (defined WARPX_QED)
if (myspc.DoQED())
{
- m_data[16] = chimin_f;
- m_data[17] = chimax_f;
+ m_data[get_idx("chimin")] = chimin_f;
+ m_data[get_idx("chimax")] = chimax_f;
}
#endif
}
generated by cgit v1.2.3 (git 2.46.0) at 2025-08-10 11:51:31 +0000