Add Particle I/O: restart & restart_checkpoint

src/Particle/CMakeLists.txt

@@ -5,5 +5,6 @@ foreach(D IN LISTS AMReX_SPACEDIM)
         ParticleContainer_HiPACE.cpp
         ParticleContainer_ImpactX.cpp
         ParticleContainer_WarpX.cpp
+        ParticleContainer_FHDeX.cpp
     )
 endforeach()

src/Particle/ParticleContainer.H

@@ -312,8 +312,34 @@ void make_ParticleContainer_and_Iterators (py::module &m, std::string allocstr)
         //                  const Vector<std::string>& int_comp_names = Vector<std::string>()) const
         // void CheckpointPre ();
         // void CheckpointPost ();
+
         // void Restart (const std::string& dir, const std::string& file);
+        .def("restart",
+            [](
+                ParticleContainerType & pc,
+                std::string const & dir,
+                std::string const & file
+            ){
+                return pc.Restart(dir, file);
+            },
+            py::arg("dir"),
+            py::arg("file")
+        )
+
         // void Restart (const std::string& dir, const std::string& file, bool is_checkpoint);
+        .def("restart_checkpoint",
+            [](
+                ParticleContainerType & pc,
+                std::string const & dir,
+                std::string const & file,
+                bool is_checkpoint
+            ){
+                return pc.Restart(dir, file, is_checkpoint);
+            },
+            py::arg("dir"),
+            py::arg("file"),
+            py::arg("is_checkpoint")
+        )
 
         .def("write_plotfile",
              //py::overload_cast<std::string const &, std::string const &>(&ParticleContainerType::WritePlotFile, py::const_),
@@ -436,10 +462,12 @@ void make_ParticleContainer_and_Iterators (py::module &m)
             T_ParticleType::NReal,
             T_ParticleType::NInt
         >(m);
-        make_Particle<  // superparticle
-            T_ParticleType::NReal + T_NArrayReal,
-            T_ParticleType::NInt  + T_NArrayInt
-        >(m);
+        if (T_NArrayReal > 0 || T_NArrayInt > 0) {
+            make_Particle<  // superparticle
+                T_ParticleType::NReal + T_NArrayReal,
+                T_ParticleType::NInt  + T_NArrayInt
+            >(m);
+        }
 
         make_ArrayOfStructs<T_ParticleType::NReal, T_ParticleType::NInt> (m);
         make_StructOfArrays<T_NArrayReal, T_NArrayInt> (m);

src/Particle/ParticleContainer.cpp

@@ -36,6 +36,7 @@ namespace
 void init_ParticleContainer_HiPACE(py::module& m);
 void init_ParticleContainer_ImpactX(py::module& m);
 void init_ParticleContainer_WarpX(py::module& m);
+void init_ParticleContainer_FHDeX(py::module& m);
 
 void init_ParticleContainer(py::module& m) {
     using namespace amrex;
@@ -59,6 +60,7 @@ void init_ParticleContainer(py::module& m) {
     init_ParticleContainer_HiPACE(m);
     init_ParticleContainer_ImpactX(m);
     init_ParticleContainer_WarpX(m);
+    init_ParticleContainer_FHDeX(m);
 
     // for particle idcpu arrays
     m.def("unpack_ids", py::vectorize(unpack_id));

src/Particle/ParticleContainer_FHDeX.cpp

@@ -0,0 +1,12 @@
+
+#include "ParticleContainer.H"
+
+#include <AMReX_Particle.H>
+#include <AMReX_ParticleTile.H>
+
+
+void init_ParticleContainer_FHDeX(py::module& m) {
+    using namespace amrex;
+
+    make_ParticleContainer_and_Iterators<Particle<16, 4>, 0, 0>(m);
+}

tests/test_plotfileparticledata.py

@@ -0,0 +1,50 @@
+from dataclasses import dataclass
+
+import amrex.space3d as amr
+
+if not amr.initialized():
+    amr.initialize([])
+
+plt_file_name = "plt0000600"
+
+
+@dataclass
+class Particle:
+    x: float
+    y: float
+    z: float
+
+    ib_id: int
+    idx: int
+
+
+def load_data(plot_file_name):
+    plt = amr.PlotFileData(plt_file_name)
+
+    probDomain = plt.probDomain(0)
+    probLo = plt.probLo()
+    probHi = plt.probHi()
+    domain_box = amr.Box(probDomain.small_end, probDomain.big_end)
+    real_box = amr.RealBox(probLo, probHi)
+    std_geometry = amr.Geometry(domain_box, real_box, plt.coordSys(), [0, 0, 0])
+
+    pc = amr.ParticleContainer_16_4_0_0_default(
+        std_geometry,
+        plt.DistributionMap(plt.finestLevel()),
+        plt.boxArray(plt.finestLevel()),
+    )
+    pc.restart(plt_file_name, "immbdy_markers")
+
+    particles = list()
+    for pti in pc.iterator(pc, level=plt.finestLevel()):
+        aos = pti.aos()
+        for p in aos.to_numpy(copy=True):
+            particles.append(Particle(x=p[0], y=p[1], z=p[2], ib_id=p[-1], idx=p[-2]))
+
+    return particles
+
+
+particles = load_data(plt_file_name)
+for p in particles:
+    print(p)
+    print(p.x)