Add a plot header file for sampling particles

amr-wind/utilities/sampling/Sampling.H

@@ -8,6 +8,7 @@
 #include "amr-wind/utilities/PostProcessing.H"
 #include "amr-wind/utilities/sampling/SamplerBase.H"
 #include "amr-wind/utilities/sampling/SamplingContainer.H"
+#include <AMReX_PlotFileUtil.H>
 
 /**
  *  \defgroup sampling Data-sampling utilities
@@ -109,11 +110,19 @@
      */
     virtual void write_ascii();
 
+    //! Output extra information for certain formats
+    void write_header_file(const std::string& fname);
+
     //! Output extra information for certain formats
     void write_info_file(const std::string& fname);
 
     SamplingContainer& sampling_container() { return *m_scontainer; }
 
+    amrex::Vector<std::string> int_var_names()
+    {
+        return {"uid", "set_id", "probe_id"};
+    };
+
 private:
     CFDSim& m_sim;
 

amr-wind/utilities/sampling/Sampling.cpp

@@ -390,16 +390,18 @@ void Sampling::impl_write_native()
     const std::string sampling_name =
         amrex::Concatenate(m_label, m_sim.time().time_index());
     const std::string name(post_dir + "/" + sampling_name);
-    amrex::Vector<std::string> int_var_names{"uid", "set_id", "probe_id"};
 
     m_scontainer->WritePlotFile(
-        name, "particles", m_var_names, int_var_names,
+        name, "particles", m_var_names, int_var_names(),
         [=] AMREX_GPU_DEVICE(const SamplingContainer::SuperParticleType& p) {
             return p.id() > 0;
         });
 
     const std::string info_name = name + "/sampling_info";
     write_info_file(info_name);
+
+    const std::string header_name = name + "/Header";
+    write_header_file(header_name);
 }
 
 void Sampling::write_ascii()
@@ -443,6 +445,43 @@ void Sampling::write_info_file(const std::string& fname)
     fh.close();
 }
 
+void Sampling::write_header_file(const std::string& fname)
+{
+    BL_PROFILE("amr-wind::Sampling::write_header_file");
+
+    // Only I/O processor writes the info file
+    if (!amrex::ParallelDescriptor::IOProcessor()) {
+        return;
+    }
+
+    if ((m_out_fmt != "native")) {
+        amrex::Abort(
+            "write_header_file is implemented only for native formats");
+    }
+
+    amrex::VisMF::IO_Buffer io_buffer(amrex::VisMF::IO_Buffer_Size);
+    std::ofstream hdr;
+    hdr.rdbuf()->pubsetbuf(io_buffer.dataPtr(), io_buffer.size());
+    hdr.open(
+        fname.c_str(),
+        std::ofstream::out | std::ofstream::trunc | std::ofstream::binary);
+    if (!hdr.good()) {
+        amrex::FileOpenFailed(fname);
+    }
+
+    const int nlevels = m_sim.repo().num_active_levels();
+    const auto& bas = m_sim.mesh().boxArray();
+    const auto& geoms = m_sim.mesh().Geom();
+    const auto& ref_ratio = m_sim.mesh().refRatio();
+    const amrex::Vector<int> level_steps(nlevels, m_sim.time().time_index());
+    auto vnames = m_var_names;
+    const auto inames = int_var_names();
+    vnames.insert(vnames.end(), inames.begin(), inames.end());
+    amrex::WriteGenericPlotfileHeader(
+        hdr, nlevels, bas, vnames, geoms, m_sim.time().new_time(), level_steps,
+        ref_ratio, "HyperCLaw-V1.1", "Level_", "Cell");
+}
+
 void Sampling::prepare_netcdf_file()
 {
 #ifdef AMR_WIND_USE_NETCDF