From 4fa339e5951df30d06cd6084c5d7e47b7b3fa73c Mon Sep 17 00:00:00 2001 From: Roelof Groenewald <40245517+roelof-groenewald@users.noreply.github.com> Date: Tue, 20 Jul 2021 17:55:42 -0700 Subject: Python - add new particle attributes at runtime (#2112) * exposes AddRealComp to Python to allow extra particle attributes to be added at runtime; also includes a new function to grab a particle data array from the name of the component rather than the index * added function to get index of a particle component given the PID name * changed new get component index and get_particle_arrays_from_comp_name functions to take species name as argument rather than species id * changed warpx_addRealComp to accept a species name as input and only add the new component for that species --- Source/Python/WarpXWrappers.h | 15 +++++++++++++++ 1 file changed, 15 insertions(+) (limited to 'Source/Python/WarpXWrappers.h') diff --git a/Source/Python/WarpXWrappers.h b/Source/Python/WarpXWrappers.h index f98600707..1672b01ae 100644 --- a/Source/Python/WarpXWrappers.h +++ b/Source/Python/WarpXWrappers.h @@ -8,6 +8,7 @@ #ifndef WARPX_WRAPPERS_H_ #define WARPX_WRAPPERS_H_ +#include "Particles/WarpXParticleContainer.H" #include "Evolve/WarpXDtType.H" #include #include @@ -92,6 +93,20 @@ extern "C" { amrex::ParticleReal** warpx_getParticleArrays(int speciesnumber, int comp, int lev, int* num_tiles, int** particles_per_tile); + amrex::ParticleReal** warpx_getParticleArraysFromCompName( + const char* char_species_name, const char* char_comp_name, int lev, + int* num_tiles, int** particles_per_tile); + + amrex::ParticleReal** warpx_getParticleArraysUsingPC( + WarpXParticleContainer& myspc, int comp, + int lev, int* num_tiles, int** particles_per_tile); + + int warpx_getParticleCompIndex( + const char* char_species_name, const char* char_comp_name); + + void warpx_addRealComp( + const char* char_species_name, const char* char_comp_name, bool comm); + void warpx_ComputeDt (); void warpx_MoveWindow (int step, bool move_j); -- cgit v1.2.3