Skip to content

Commit

Permalink
Python: Pure SoA Particle
Browse files Browse the repository at this point in the history
  • Loading branch information
ax3l committed Dec 21, 2023
1 parent 0d3fde5 commit 991fd06
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 6 deletions.
2 changes: 1 addition & 1 deletion Source/Python/Particles/PinnedMemoryParticleContainer.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,6 @@ void init_PinnedMemoryParticleContainer (py::module& m)
{
py::class_<
PinnedMemoryParticleContainer,
amrex::ParticleContainer<0,0,PIdx::nattribs,0,amrex::PinnedArenaAllocator>
amrex::ParticleContainerPureSoA<PIdx::nattribs, 0, amrex::PinnedArenaAllocator>
> pmpc (m, "PinnedMemoryParticleContainer");
}
10 changes: 5 additions & 5 deletions Source/Python/Particles/WarpXParticleContainer.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -12,11 +12,11 @@
void init_WarpXParIter (py::module& m)
{
py::class_<
WarpXParIter, amrex::ParIter<0,0,PIdx::nattribs>
WarpXParIter, amrex::ParIterSoA<PIdx::nattribs, 0>
>(m, "WarpXParIter")
.def(py::init<amrex::ParIter<0,0,PIdx::nattribs>::ContainerType&, int>(),
.def(py::init<amrex::ParIterSoA<PIdx::nattribs, 0>::ContainerType&, int>(),
py::arg("particle_container"), py::arg("level"))
.def(py::init<amrex::ParIter<0,0,PIdx::nattribs>::ContainerType&, int, amrex::MFItInfo&>(),
.def(py::init<amrex::ParIterSoA<PIdx::nattribs, 0>::ContainerType&, int, amrex::MFItInfo&>(),
py::arg("particle_container"), py::arg("level"),
py::arg("info"))
;
Expand All @@ -26,11 +26,11 @@ void init_WarpXParticleContainer (py::module& m)
{
py::class_<
WarpXParticleContainer,
amrex::ParticleContainer<0, 0, PIdx::nattribs, 0>
amrex::ParticleContainerPureSoA<PIdx::nattribs, 0>
> wpc (m, "WarpXParticleContainer");
wpc
.def("add_real_comp",
[](WarpXParticleContainer& pc, const std::string& name, bool const comm) { pc.AddRealComp(name, comm); },
[](WarpXParticleContainer& pc, const std::string& name, bool comm) { pc.AddRealComp(name, comm); },
py::arg("name"), py::arg("comm")
)
.def("add_n_particles",
Expand Down

0 comments on commit 991fd06

Please sign in to comment.