Skip to content

Commit

Permalink
Use polymorphism for meshes/particlesPath in Python
Browse files Browse the repository at this point in the history
  • Loading branch information
franzpoeschel committed Nov 7, 2024
1 parent 6b47a35 commit 20f2cd5
Showing 1 changed file with 52 additions and 16 deletions.
68 changes: 52 additions & 16 deletions src/binding/python/Series.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -22,10 +22,12 @@
#include "openPMD/IO/Access.hpp"
#include "openPMD/IterationEncoding.hpp"
#include "openPMD/auxiliary/JSON.hpp"
#include "openPMD/auxiliary/Variant.hpp"
#include "openPMD/binding/python/Pickle.hpp"
#include "openPMD/config.hpp"

#include "openPMD/binding/python/Common.hpp"
#include <variant>

#if openPMD_HAVE_MPI
// re-implemented signatures:
Expand Down Expand Up @@ -282,26 +284,60 @@ this method.
&Series::openPMDextension,
&Series::setOpenPMDextension)
.def_property("base_path", &Series::basePath, &Series::setBasePath)
.def_property(
"meshes_path",
&Series::meshesPath,
py::overload_cast<std::string const &>(&Series::setMeshesPath))
.def("get_rank_table", &Series::rankTable, py::arg("collective"))
.def("set_rank_table", &Series::setRankTable, py::arg("my_rank_info"))
.def_property(
"particles_path",
&Series::particlesPath,
py::overload_cast<std::string const &>(&Series::setParticlesPath))
.def_property(
"meshes_paths",
&Series::meshesPath,
py::overload_cast<std::vector<std::string> const &>(
&Series::setMeshesPath))
"meshes_path",
[](Series &self)
-> std::variant<std::string, std::vector<std::string>> {
using res_t =
std::variant<std::string, std::vector<std::string>>;
auto res = self.meshesPaths();
if (res.size() == 1)
{
return res_t{std::move(res[0])};
}
else
{
return res_t{std::move(res)};
}
},
[](Series &self,
std::variant<std::string, std::vector<std::string>> const &arg)
-> Series & {
std::visit(
[&](auto const &arg_resolved) {
self.setMeshesPath(arg_resolved);
},
arg);
return self;
})
.def_property(
"particles_paths",
&Series::particlesPath,
py::overload_cast<std::vector<std::string> const &>(
&Series::setParticlesPath))
"particles_path",
[](Series &self)
-> std::variant<std::string, std::vector<std::string>> {
using res_t =
std::variant<std::string, std::vector<std::string>>;
auto res = self.particlesPaths();
if (res.size() == 1)
{
return res_t{std::move(res[0])};
}
else
{
return res_t{std::move(res)};
}
},
[](Series &self,
std::variant<std::string, std::vector<std::string>> const &arg)
-> Series & {
std::visit(
[&](auto const &arg_resolved) {
self.setParticlesPath(arg_resolved);
},
arg);
return self;
})
.def_property("author", &Series::author, &Series::setAuthor)
.def_property(
"machine",
Expand Down

0 comments on commit 20f2cd5

Please sign in to comment.