diff options
author | 2023-08-12 11:17:38 -0700 | |
---|---|---|
committer | 2023-08-12 11:17:38 -0700 | |
commit | 6c93d9fc13830d574c69ac7b166f5fbdb0809731 (patch) | |
tree | 8742df6045aa2bfdccb5a7991eae436e886e47d1 /Source/Python/WarpXWrappers.cpp | |
parent | f6760c8e6d64605f73476f9bc8292dc9d85df454 (diff) | |
download | WarpX-6c93d9fc13830d574c69ac7b166f5fbdb0809731.tar.gz WarpX-6c93d9fc13830d574c69ac7b166f5fbdb0809731.tar.zst WarpX-6c93d9fc13830d574c69ac7b166f5fbdb0809731.zip |
Transition to pyAMReX (#3474)
* pyAMReX: Build System
* CI Updates (Changed Options)
* Callback modernization (#4)
* refactor callbacks.py
* added binding code in `pyWarpX.cpp` to add or remove keys from the callback dictionary
* minor PR cleanups
Co-authored-by: Axel Huebl <axel.huebl@plasma.ninja>
* Added Python level reference to fetch the multifabs (#3)
* pyAMReX: Build System
* Added Python level reference to Ex_aux
* Now uses the multifab map
* Fix typo
Co-authored-by: Axel Huebl <axel.huebl@plasma.ninja>
* Add initialization and finalize routines (#5)
A basic PICMI input file will now run to completion.
* Regression Tests: WarpX_PYTHON=ON
* Update Imports to nD pyAMReX
* IPO/LTO Control
Although pybind11 relies heavily on IPO/LTO to create low-latency,
small-binary bindings, some compilers will have troubles with that.
Thus, we add a compile-time option to optionally disable it when
needed.
* Fix: Link Legacy WarpXWrappers.cpp
* Wrap WarpX instance and start multi particle container
* Fix test Python_pass_mpi_comm
* Start wrapper for multiparticle container
* Add return policy
Co-authored-by: Axel Huebl <axel.huebl@plasma.ninja>
* Update fields to use MultiFabs directly
Remove EOL white space
Removed old routines accessing MultiFabs
Update to use "node_centered"
* Fix compilation with Python
* Update fields.py to use modified MultiFab tag names
* Remove incorrect, unused code
* Add function to extract the WarpX MultiParticleContainer
* Complete class WarpXParticleContainer
* Wrap functions getNprocs / getMyProc
* restore `install___` callback API - could remove later if we want but should maintain backward compatibility for now
* add `gett_new` and `getistep` functions wrappers; fix typos in `callbacks.py`; avoid error in getting `rho` from `fields.py`
* Update callback call and `getNproc`/`getMyProc` function
* Replace function add_n_particles
* Fix setitem in fields.py for 1d and 2d
* also update `gett_new()` in `_libwarpx.py` in case we want to keep that API
* added binding for `WarpXParIter` - needed to port `libwarpx.depositChargeDensity()` which is an ongoing effort
* Wrap function num_real_comp
* added binding for `TotalNumberOfParticles` and continue progress on enabling 1d MCC test to run
* add `SyncRho()` binding and create helper function in `libwarpx.depositChargeDensity` to manage scope of the particle iter
* Clean up issues in fields.py
* update bindings for `get_particle_structs`
* Fix setitem in fields.py
* switch order of initialization for particle container and particle iterator
* switch deposit_charge loop to C++ code; bind `ApplyInverseVolumeScalingToChargeDensity`
* move `WarpXParticleContainer.cpp` and `MultiParticleContainer.cpp` to new Particles folder
* added binding for `ParticleBoundaryBuffer`
* More fixes for fields.py
* Fix: Backtraces from Python
Add the Python executable name with an absolute path, so backtraces
in AMReX work. See linked AMReX issue for details.
* Cleaning
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Fix: Backtraces from Python Part II
Do not add Python script name - it confuses the AMReX ParmParser to
build its table.
* Fix: CMake Dependencies for Wheel
This fixes a racecondition during `pip_install`: it was possible
that not all dimensions where yet build from pybind before we
start packing them in the wheel for pip install.
* MCC Test: Install Callbacks before Run
Otherwise hangs in aquiring the gil during shutdown.
* addition of `Python/pywarpx/particle_containers.py` and various associated bindings
* Fix: CMake Superbuild w/ Shared AMReX
We MUST build AMReX as a shared (so/dll/dylib) library, otherwise
all the global state in it will cause split-brain situations, where
our Python modules operate on different stacks.
* add `clear_all()` to callbacks in order to remove all callbacks at finalize
* add `-DWarpX_PYTHON=ON` to CI tests that failed to build
* add `get_comp_index` and continue to port particle data bindings
* Add AMReX Module as `libwarpx_so.amr`
Attribute to pass through the already loaded AMReX module with the
matching dimensionality to the simulation.
* Fix for fields accounting for guard cells
* Fix handling of ghost cells in fields
* Update & Test: Particle Boundary Scraping
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* CI: Python Updates
- modernize Python setups
- drop CUDA 11.0 for good and go 11.3+ as documented already
```
Error #3246: Internal Compiler Error (codegen): "there was an error in verifying the lgenfe output!"
```
* CI: Python Updates (chmod)
* Add support for cupy in fields.py
* Add MultiFab reduction routines
* CI: CUDA 11.3 is <= Ubuntu 20.04
* changed `AddNParticles` to take `amrex::Vector` arguments
* setup.py: WarpX_PYTHON=ON
* update various 2d and rz tests with new APIs
* add `-DWarpX_PYTHON_IPO=OFF` to compile string for 2d and 3d Python CI tests to speed up linking
* CI: -DpyAMReX_IPO=OFF
* CI: -DpyAMReX_IPO=OFF
actually adding `=OFF`
* CI: Intel Python
* CI: macOS Python Executable
Ensure we always use the same `python3` executable, as specified
by the `PATH` priority.
* CMake: Python Multi-Config Build
Add support for multi-config generators, especially on Windows.
* __init__.py: Windows DLL Support
Python 3.8+ on Windows: DLL search paths for dependent
shared libraries
Refs.:
- https://github.com/python/cpython/issues/80266
- https://docs.python.org/3.8/library/os.html#os.add_dll_directory
* CI: pywarpx Update
our setup.py cannot install pyamrex yet as a dependency.
* ABLASTR: `ablastr/export.H`
Add a new header to export public globals that are not covered by
`WINDOWS_EXPORT_ALL_SYMBOLS`.
https://stackoverflow.com/questions/54560832/cmake-windows-export-all-symbols-does-not-cover-global-variables/54568678#54568678
* WarpX: EXPORT Globals in `.dll` files
WarpX still uses a lot of globals:
- `static` member variables
- `extern` global variables
These globals cannot be auto-exported with CMake's
`WINDOWS_EXPORT_ALL_SYMBOLS` helper and thus we need to mark them
manually for DLL export (and import) via the new ABLASTR
`ablastr/export.H` helper macros.
This starts to export symbols in the:
- WarpX and particle container classes
- callback hook database map
- ES solver
* CI: pywarpx Clang CXXFLAGS Down
Move CXXFLAGS (`-Werror ...`) down until deps are installed.
* GNUmake: Generate `ablastr/export.H`
* CMake: More Symbol Exports for Windows
* `WarpX-tests.ini`: Simplify Python no-IPO
Also avoids subtle differences in compilation that increase
compile time.
* Update PICMI_inputs_EB_API.py for embedded_boundary_python_API CI test
* Fix Python_magnetostatic_eb_3d
* Update: Python_restart_runtime_components
New Python APIs
* Windows: no dllimport for now
* CI: Skip `PYINSTALLOPTIONS` For Now
* CMake: Dependency Bump Min-Versions
for external packages picked up by `find_package`.
* Fix F and G_fp names in fields.py
* Tests: Disable `Python_pass_mpi_comm`
* Wrappers: Cleanup
* pyWarpX: Include Cleaning
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* fields.py: Fix F and G Wrappers
Correct MultiFab names (w/o components).
* Remove unused in fields.py
* Windows: New Export Headers
- ABLASTR: `ablastr/export.H`
- WarpX: `Utils/export.H`
* removed `WarpInterface.py` since that functionality is now in `particle_containers.py`; removed parts of `WarpXWrappers.cpp` that have been ported to pyamrex
* CMake: Link OBJECT Target PRIVATE
* CMake: Remove OBJECT Target
Simplify and make `app` link `lib` (default: static). Remove OBJECT
target.
* Fix in fields.py for the components index
* Update get_particle_id/cpu
As implemented in pyAMReX with
https://github.com/AMReX-Codes/pyamrex/pull/165
* WarpX: Update for Private Constructor
* Import AMReX Before pyd DLL Call
Importing AMReX will add the `add_dll_directory` to a potentially
shared amrex DLL on Windows.
* Windows CI: Set PATH to amrex_Nd.dll
* CMake: AMReX_INSTALL After Python
In superbuild, Python can modify `AMReX_BUILD_SHARED_LIBS`.
* Clang Win CI: Manually Install requirements
Sporadic error is:
```
...
Installing collected packages: pyparsing, numpy, scipy, periodictable, picmistandard
ERROR: Could not install packages due to an OSError: [WinError 32] The process cannot access the file because it is being used by another process: 'C:\\hostedtoolcache\\windows\\Python\\3.11.4\\x64\\Lib\\site-packages\\numpy\\polynomial\\__init__.py'
Consider using the `--user` option or check the permissions.
```
* Hopefully final fixes to fields.py
* Update getProbLo/getProbHi
* Set plasma length strength
Co-authored-by: Remi Lehe <remi.lehe@normalesup.org>
* Fix fields method to remove CodeQL notice
* Update Comments & Some Finalize
* Move: set_plasma_lens_strength
to MPC
---------
Co-authored-by: Roelof Groenewald <40245517+roelof-groenewald@users.noreply.github.com>
Co-authored-by: David Grote <dpgrote@lbl.gov>
Co-authored-by: Remi Lehe <remi.lehe@normalesup.org>
Co-authored-by: Dave Grote <grote1@llnl.gov>
Co-authored-by: Roelof Groenewald <regroenewald@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Diffstat (limited to 'Source/Python/WarpXWrappers.cpp')
-rw-r--r-- | Source/Python/WarpXWrappers.cpp | 626 |
1 files changed, 2 insertions, 624 deletions
diff --git a/Source/Python/WarpXWrappers.cpp b/Source/Python/WarpXWrappers.cpp index 02cee52e0..44820dd1f 100644 --- a/Source/Python/WarpXWrappers.cpp +++ b/Source/Python/WarpXWrappers.cpp @@ -39,71 +39,6 @@ #include <array> #include <cstdlib> -namespace -{ - amrex::Real** getMultiFabPointers (amrex::MultiFab& mf, int *num_boxes, int *ncomps, int **ngrowvect, int **shapes) - { - *ncomps = mf.nComp(); - *num_boxes = mf.local_size(); - int shapesize = AMREX_SPACEDIM; - *ngrowvect = static_cast<int*>(malloc(sizeof(int)*shapesize)); - for (int j = 0; j < AMREX_SPACEDIM; ++j) { - (*ngrowvect)[j] = mf.nGrow(j); - } - if (mf.nComp() > 1) shapesize += 1; - *shapes = static_cast<int*>(malloc(sizeof(int)*shapesize * (*num_boxes))); - auto data = - static_cast<amrex::Real**>(malloc((*num_boxes) * sizeof(amrex::Real*))); - -#ifdef AMREX_USE_OMP -#pragma omp parallel if (amrex::Gpu::notInLaunchRegion()) -#endif - for ( amrex::MFIter mfi(mf, false); mfi.isValid(); ++mfi ) { - const int i = mfi.LocalIndex(); - data[i] = mf[mfi].dataPtr(); - for (int j = 0; j < AMREX_SPACEDIM; ++j) { - (*shapes)[shapesize*i+j] = mf[mfi].box().length(j); - } - if (mf.nComp() > 1) (*shapes)[shapesize*i+AMREX_SPACEDIM] = mf.nComp(); - } - return data; - } - int* getMultiFabLoVects (const amrex::MultiFab& mf, int *num_boxes, int **ngrowvect) - { - const int shapesize = AMREX_SPACEDIM; - *ngrowvect = static_cast<int*>(malloc(sizeof(int)*shapesize)); - for (int j = 0; j < AMREX_SPACEDIM; ++j) { - (*ngrowvect)[j] = mf.nGrow(j); - } - *num_boxes = mf.local_size(); - auto loVects = static_cast<int*>(malloc((*num_boxes)*AMREX_SPACEDIM * sizeof(int))); - - int i = 0; - for ( amrex::MFIter mfi(mf, false); mfi.isValid(); ++mfi, ++i ) { - const int* loVect = mf[mfi].loVect(); - for (int j = 0; j < AMREX_SPACEDIM; ++j) { - loVects[AMREX_SPACEDIM*i+j] = loVect[j]; - } - } - return loVects; - } - // Copy the nodal flag data and return the copy: - // the nodal flag data should not be modifiable from Python. - int* getFieldNodalFlagData ( const amrex::MultiFab* mf ) - { - if (mf == nullptr) return nullptr; - const amrex::IntVect nodal_flag( mf->ixType().toIntVect() ); - auto *nodal_flag_data = static_cast<int*>(malloc(AMREX_SPACEDIM * sizeof(int))); - - constexpr int NODE = amrex::IndexType::NODE; - - for (int i=0 ; i < AMREX_SPACEDIM ; i++) { - nodal_flag_data[i] = (nodal_flag[i] == NODE ? 1 : 0); - } - return nodal_flag_data; - } -} - int warpx_Real_size() { return (int)sizeof(amrex::Real); @@ -130,85 +65,9 @@ namespace return WarpX::galerkin_interpolation; } - int warpx_nComps() - { - return PIdx::nattribs; - } - - int warpx_nCompsSpecies(const char* char_species_name) - { - auto & mypc = WarpX::GetInstance().GetPartContainer(); - const std::string species_name(char_species_name); - auto & myspc = mypc.GetParticleContainerFromName(species_name); - return myspc.NumRealComps(); - } - - int warpx_SpaceDim() - { - return AMREX_SPACEDIM; - } - - void amrex_init (int argc, char* argv[]) - { - warpx::initialization::amrex_init(argc, argv); - } - - void amrex_init_with_inited_mpi (int argc, char* argv[], MPI_Comm mpicomm) - { - warpx::initialization::amrex_init(argc, argv, true, mpicomm); - } - - void amrex_finalize (int /*finalize_mpi*/) - { - amrex::Finalize(); - } - - void warpx_init () - { - WarpX& warpx = WarpX::GetInstance(); - warpx.InitData(); - ExecutePythonCallback("afterinit"); - ExecutePythonCallback("particleloader"); - } - - void warpx_finalize () + void amrex_init_with_inited_mpi (int argc, char* argv[], MPI_Comm /* mpicomm */) { - WarpX::Finalize(); - } - - void warpx_set_callback_py ( - const char* char_callback_name, WARPX_CALLBACK_PY_FUNC_0 callback) - { - const std::string callback_name(char_callback_name); - warpx_callback_py_map[callback_name] = callback; - } - - void warpx_clear_callback_py (const char* char_callback_name) - { - const std::string callback_name(char_callback_name); - warpx_callback_py_map.erase(callback_name); - } - - void warpx_evolve (int numsteps) - { - WarpX& warpx = WarpX::GetInstance(); - warpx.Evolve(numsteps); - } - - void warpx_addNParticles( - const char* char_species_name, int lenx, amrex::ParticleReal const * x, - amrex::ParticleReal const * y, amrex::ParticleReal const * z, - amrex::ParticleReal const * vx, amrex::ParticleReal const * vy, - amrex::ParticleReal const * vz, const int nattr_real, - amrex::ParticleReal const * attr_real, const int nattr_int, - int const * attr_int, int uniqueparticles) - { - auto & mypc = WarpX::GetInstance().GetPartContainer(); - const std::string species_name(char_species_name); - auto & myspc = mypc.GetParticleContainerFromName(species_name); - const int lev = 0; - myspc.AddNParticles(lev, lenx, x, y, z, vx, vy, vz, nattr_real, attr_real, - nattr_int, attr_int, uniqueparticles); + warpx::initialization::amrex_init(argc, argv, true); } void warpx_ConvertLabParamsToBoost() @@ -226,340 +85,11 @@ namespace CheckGriddingForRZSpectral(); } - amrex::Real warpx_getProbLo(int dir) - { - WarpX& warpx = WarpX::GetInstance(); - const amrex::Geometry& geom = warpx.Geom(0); - return geom.ProbLo(dir); - } - - amrex::Real warpx_getProbHi(int dir) - { - WarpX& warpx = WarpX::GetInstance(); - const amrex::Geometry& geom = warpx.Geom(0); - return geom.ProbHi(dir); - } - amrex::Real warpx_getCellSize(int dir, int lev) { const std::array<amrex::Real,3>& dx = WarpX::CellSize(lev); return dx[dir]; } - long warpx_getNumParticles(const char* char_species_name, const bool local) { - const auto & mypc = WarpX::GetInstance().GetPartContainer(); - const std::string species_name(char_species_name); - auto & myspc = mypc.GetParticleContainerFromName(species_name); - // the first argument below is to only count valid particles - return myspc.TotalNumberOfParticles(true, local); - } - -#define WARPX_GET_FIELD(FIELD, GETTER) \ - amrex::Real** FIELD(int lev, int direction, \ - int *return_size, int *ncomps, int **ngrowvect, int **shapes) { \ - auto * mf = GETTER(lev, direction); \ - if (mf != nullptr) { \ - return getMultiFabPointers(*mf, return_size, ncomps, ngrowvect, shapes); \ - } else { \ - return nullptr; \ - } \ - } - -#define WARPX_GET_LOVECTS(FIELD, GETTER) \ - int* FIELD(int lev, int direction, \ - int *return_size, int **ngrowvect) { \ - auto * mf = GETTER(lev, direction); \ - if (mf != nullptr) { \ - return getMultiFabLoVects(*mf, return_size, ngrowvect); \ - } else { \ - return nullptr; \ - } \ - } - - WARPX_GET_FIELD(warpx_getEfield, WarpX::GetInstance().get_pointer_Efield_aux) - WARPX_GET_FIELD(warpx_getEfieldCP, WarpX::GetInstance().get_pointer_Efield_cp) - WARPX_GET_FIELD(warpx_getEfieldFP, WarpX::GetInstance().get_pointer_Efield_fp) - - WARPX_GET_FIELD(warpx_getBfield, WarpX::GetInstance().get_pointer_Bfield_aux) - WARPX_GET_FIELD(warpx_getBfieldCP, WarpX::GetInstance().get_pointer_Bfield_cp) - WARPX_GET_FIELD(warpx_getBfieldFP, WarpX::GetInstance().get_pointer_Bfield_fp) - - WARPX_GET_FIELD(warpx_getEdgeLengths, WarpX::GetInstance().get_pointer_edge_lengths) - WARPX_GET_FIELD(warpx_getFaceAreas, WarpX::GetInstance().get_pointer_face_areas) - - WARPX_GET_FIELD(warpx_getCurrentDensity, WarpX::GetInstance().get_pointer_current_fp) - WARPX_GET_FIELD(warpx_getCurrentDensityCP, WarpX::GetInstance().get_pointer_current_cp) - WARPX_GET_FIELD(warpx_getCurrentDensityFP, WarpX::GetInstance().get_pointer_current_fp) - - WARPX_GET_FIELD(warpx_getCurrentDensityFP_Ampere, WarpX::GetInstance().GetHybridPICModel().get_pointer_current_fp_ampere) - - WARPX_GET_FIELD(warpx_getVectorPotentialFP, WarpX::GetInstance().get_pointer_vector_potential_fp) - - WARPX_GET_LOVECTS(warpx_getEfieldLoVects, WarpX::GetInstance().get_pointer_Efield_aux) - WARPX_GET_LOVECTS(warpx_getEfieldCPLoVects, WarpX::GetInstance().get_pointer_Efield_cp) - WARPX_GET_LOVECTS(warpx_getEfieldFPLoVects, WarpX::GetInstance().get_pointer_Efield_fp) - - WARPX_GET_LOVECTS(warpx_getBfieldLoVects, WarpX::GetInstance().get_pointer_Bfield_aux) - WARPX_GET_LOVECTS(warpx_getBfieldCPLoVects, WarpX::GetInstance().get_pointer_Bfield_cp) - WARPX_GET_LOVECTS(warpx_getBfieldFPLoVects, WarpX::GetInstance().get_pointer_Bfield_fp) - - WARPX_GET_LOVECTS(warpx_getCurrentDensityLoVects, WarpX::GetInstance().get_pointer_current_fp) - WARPX_GET_LOVECTS(warpx_getCurrentDensityCPLoVects, WarpX::GetInstance().get_pointer_current_cp) - WARPX_GET_LOVECTS(warpx_getCurrentDensityFPLoVects, WarpX::GetInstance().get_pointer_current_fp) - WARPX_GET_LOVECTS(warpx_getVectorPotentialFPLoVects, WarpX::GetInstance().get_pointer_vector_potential_fp) - - WARPX_GET_LOVECTS(warpx_getEdgeLengthsLoVects, WarpX::GetInstance().get_pointer_edge_lengths) - WARPX_GET_LOVECTS(warpx_getFaceAreasLoVects, WarpX::GetInstance().get_pointer_face_areas) - - int* warpx_getEx_nodal_flag() {return getFieldNodalFlagData( WarpX::GetInstance().get_pointer_Efield_aux(0,0) );} - int* warpx_getEy_nodal_flag() {return getFieldNodalFlagData( WarpX::GetInstance().get_pointer_Efield_aux(0,1) );} - int* warpx_getEz_nodal_flag() {return getFieldNodalFlagData( WarpX::GetInstance().get_pointer_Efield_aux(0,2) );} - int* warpx_getBx_nodal_flag() {return getFieldNodalFlagData( WarpX::GetInstance().get_pointer_Bfield_aux(0,0) );} - int* warpx_getBy_nodal_flag() {return getFieldNodalFlagData( WarpX::GetInstance().get_pointer_Bfield_aux(0,1) );} - int* warpx_getBz_nodal_flag() {return getFieldNodalFlagData( WarpX::GetInstance().get_pointer_Bfield_aux(0,2) );} - int* warpx_getJx_nodal_flag() {return getFieldNodalFlagData( WarpX::GetInstance().get_pointer_current_fp(0,0) );} - int* warpx_getJy_nodal_flag() {return getFieldNodalFlagData( WarpX::GetInstance().get_pointer_current_fp(0,1) );} - int* warpx_getJz_nodal_flag() {return getFieldNodalFlagData( WarpX::GetInstance().get_pointer_current_fp(0,2) );} - int* warpx_getAx_nodal_flag() {return getFieldNodalFlagData( WarpX::GetInstance().get_pointer_vector_potential_fp(0,0) );} - int* warpx_getAy_nodal_flag() {return getFieldNodalFlagData( WarpX::GetInstance().get_pointer_vector_potential_fp(0,1) );} - int* warpx_getAz_nodal_flag() {return getFieldNodalFlagData( WarpX::GetInstance().get_pointer_vector_potential_fp(0,2) );} - int* warpx_getRho_nodal_flag() {return getFieldNodalFlagData( WarpX::GetInstance().get_pointer_rho_fp(0) );} - int* warpx_getPhi_nodal_flag() {return getFieldNodalFlagData( WarpX::GetInstance().get_pointer_phi_fp(0) );} - int* warpx_getF_nodal_flag() {return getFieldNodalFlagData( WarpX::GetInstance().get_pointer_F_fp(0) );} - int* warpx_getG_nodal_flag() {return getFieldNodalFlagData( WarpX::GetInstance().get_pointer_G_fp(0) );} - int* warpx_get_edge_lengths_x_nodal_flag() {return getFieldNodalFlagData( WarpX::GetInstance().get_pointer_edge_lengths(0, 0) );} - int* warpx_get_edge_lengths_y_nodal_flag() {return getFieldNodalFlagData( WarpX::GetInstance().get_pointer_edge_lengths(0, 1) );} - int* warpx_get_edge_lengths_z_nodal_flag() {return getFieldNodalFlagData( WarpX::GetInstance().get_pointer_edge_lengths(0, 2) );} - int* warpx_get_face_areas_x_nodal_flag() {return getFieldNodalFlagData( WarpX::GetInstance().get_pointer_face_areas(0, 0) );} - int* warpx_get_face_areas_y_nodal_flag() {return getFieldNodalFlagData( WarpX::GetInstance().get_pointer_face_areas(0, 1) );} - int* warpx_get_face_areas_z_nodal_flag() {return getFieldNodalFlagData( WarpX::GetInstance().get_pointer_face_areas(0, 2) );} - -#define WARPX_GET_SCALAR(SCALAR, GETTER) \ - amrex::Real** SCALAR(int lev, \ - int *return_size, int *ncomps, int **ngrowvect, int **shapes) { \ - auto * mf = GETTER(lev); \ - if (mf != nullptr) { \ - return getMultiFabPointers(*mf, return_size, ncomps, ngrowvect, shapes); \ - } else { \ - return nullptr; \ - } \ - } - -#define WARPX_GET_LOVECTS_SCALAR(SCALAR, GETTER) \ - int* SCALAR(int lev, \ - int *return_size, int **ngrowvect) { \ - auto * mf = GETTER(lev); \ - if (mf != nullptr) { \ - return getMultiFabLoVects(*mf, return_size, ngrowvect); \ - } else { \ - return nullptr; \ - } \ - } - - WARPX_GET_SCALAR(warpx_getChargeDensityCP, WarpX::GetInstance().get_pointer_rho_cp) - WARPX_GET_SCALAR(warpx_getChargeDensityFP, WarpX::GetInstance().get_pointer_rho_fp) - - WARPX_GET_LOVECTS_SCALAR(warpx_getChargeDensityCPLoVects, WarpX::GetInstance().get_pointer_rho_cp) - WARPX_GET_LOVECTS_SCALAR(warpx_getChargeDensityFPLoVects, WarpX::GetInstance().get_pointer_rho_fp) - - WARPX_GET_SCALAR(warpx_getPhiFP, WarpX::GetInstance().get_pointer_phi_fp) - - WARPX_GET_LOVECTS_SCALAR(warpx_getPhiFPLoVects, WarpX::GetInstance().get_pointer_phi_fp) - - // F and G - WARPX_GET_SCALAR(warpx_getFfieldCP, WarpX::GetInstance().get_pointer_F_cp) - WARPX_GET_SCALAR(warpx_getFfieldFP, WarpX::GetInstance().get_pointer_F_fp) - WARPX_GET_LOVECTS_SCALAR(warpx_getFfieldCPLoVects, WarpX::GetInstance().get_pointer_F_cp) - WARPX_GET_LOVECTS_SCALAR(warpx_getFfieldFPLoVects, WarpX::GetInstance().get_pointer_F_fp) - WARPX_GET_SCALAR(warpx_getGfieldCP, WarpX::GetInstance().get_pointer_G_cp) - WARPX_GET_SCALAR(warpx_getGfieldFP, WarpX::GetInstance().get_pointer_G_fp) - WARPX_GET_LOVECTS_SCALAR(warpx_getGfieldCPLoVects, WarpX::GetInstance().get_pointer_G_cp) - WARPX_GET_LOVECTS_SCALAR(warpx_getGfieldFPLoVects, WarpX::GetInstance().get_pointer_G_fp) - -#define WARPX_GET_FIELD_PML(FIELD, GETTER) \ - amrex::Real** FIELD(int lev, int direction, \ - int *return_size, int *ncomps, int **ngrowvect, int **shapes) { \ - auto * pml = WarpX::GetInstance().GetPML(lev); \ - if (!pml) return nullptr; \ - auto * mf = (pml->GETTER()[direction]); \ - if (!mf) return nullptr; \ - return getMultiFabPointers(*mf, return_size, ncomps, ngrowvect, shapes); \ - } - -#define WARPX_GET_LOVECTS_PML(FIELD, GETTER) \ - int* FIELD(int lev, int direction, \ - int *return_size, int **ngrowvect) { \ - auto * pml = WarpX::GetInstance().GetPML(lev); \ - if (!pml) return nullptr; \ - auto * mf = (pml->GETTER()[direction]); \ - if (!mf) return nullptr; \ - return getMultiFabLoVects(*mf, return_size, ngrowvect); \ - } - - WARPX_GET_FIELD_PML(warpx_getEfieldCP_PML, GetE_cp) - WARPX_GET_FIELD_PML(warpx_getEfieldFP_PML, GetE_fp) - WARPX_GET_FIELD_PML(warpx_getBfieldCP_PML, GetB_cp) - WARPX_GET_FIELD_PML(warpx_getBfieldFP_PML, GetB_fp) - WARPX_GET_FIELD_PML(warpx_getCurrentDensityCP_PML, Getj_cp) - WARPX_GET_FIELD_PML(warpx_getCurrentDensityFP_PML, Getj_fp) - WARPX_GET_LOVECTS_PML(warpx_getEfieldCPLoVects_PML, GetE_cp) - WARPX_GET_LOVECTS_PML(warpx_getEfieldFPLoVects_PML, GetE_fp) - WARPX_GET_LOVECTS_PML(warpx_getBfieldCPLoVects_PML, GetB_cp) - WARPX_GET_LOVECTS_PML(warpx_getBfieldFPLoVects_PML, GetB_fp) - WARPX_GET_LOVECTS_PML(warpx_getCurrentDensityCPLoVects_PML, Getj_cp) - WARPX_GET_LOVECTS_PML(warpx_getCurrentDensityFPLoVects_PML, Getj_fp) - -#define WARPX_GET_SCALAR_PML(SCALAR, GETTER) \ - amrex::Real** SCALAR(int lev, \ - int *return_size, int *ncomps, int **ngrowvect, int **shapes) { \ - auto * pml = WarpX::GetInstance().GetPML(lev); \ - if (!pml) return nullptr; \ - auto * mf = pml->GETTER(); \ - if (!mf) return nullptr; \ - return getMultiFabPointers(*mf, return_size, ncomps, ngrowvect, shapes); \ - } - -#define WARPX_GET_LOVECTS_PML_SCALAR(SCALAR, GETTER) \ - int* SCALAR(int lev, \ - int *return_size, int **ngrowvect) { \ - auto * pml = WarpX::GetInstance().GetPML(lev); \ - if (!pml) return nullptr; \ - auto * mf = pml->GETTER(); \ - if (!mf) return nullptr; \ - return getMultiFabLoVects(*mf, return_size, ngrowvect); \ - } - - // F and G - WARPX_GET_SCALAR_PML(warpx_getFfieldCP_PML, GetF_cp) - WARPX_GET_SCALAR_PML(warpx_getFfieldFP_PML, GetF_fp) - WARPX_GET_LOVECTS_PML_SCALAR(warpx_getFfieldCPLoVects_PML, GetF_cp) - WARPX_GET_LOVECTS_PML_SCALAR(warpx_getFfieldFPLoVects_PML, GetF_fp) - WARPX_GET_SCALAR_PML(warpx_getGfieldCP_PML, GetG_cp) - WARPX_GET_SCALAR_PML(warpx_getGfieldFP_PML, GetG_fp) - WARPX_GET_LOVECTS_PML_SCALAR(warpx_getGfieldCPLoVects_PML, GetG_cp) - WARPX_GET_LOVECTS_PML_SCALAR(warpx_getGfieldFPLoVects_PML, GetG_fp) - - int* warpx_getF_pml_nodal_flag () - { - auto * pml = WarpX::GetInstance().GetPML(0); - if (!pml) return nullptr; - return getFieldNodalFlagData(pml->GetF_fp()); - } - - int* warpx_getG_pml_nodal_flag () - { - auto * pml = WarpX::GetInstance().GetPML(0); - if (!pml) return nullptr; - return getFieldNodalFlagData(pml->GetG_fp()); - } - - amrex::ParticleReal** warpx_getParticleStructs( - const char* char_species_name, int lev, - int* num_tiles, int** particles_per_tile) { - const auto & mypc = WarpX::GetInstance().GetPartContainer(); - const std::string species_name(char_species_name); - auto & myspc = mypc.GetParticleContainerFromName(species_name); - - *num_tiles = myspc.numLocalTilesAtLevel(lev); - *particles_per_tile = static_cast<int*>(malloc(*num_tiles*sizeof(int))); - memset(*particles_per_tile, 0, *num_tiles*sizeof(int)); - - auto data = static_cast<amrex::ParticleReal**>(malloc(*num_tiles*sizeof(typename WarpXParticleContainer::ParticleType*))); - int i = 0; - for (WarpXParIter pti(myspc, lev); pti.isValid(); ++pti, ++i) { - auto& aos = pti.GetArrayOfStructs(); - data[i] = (amrex::ParticleReal*) aos.data(); - (*particles_per_tile)[i] = pti.numParticles(); - } - return data; - } - - amrex::ParticleReal** warpx_getParticleArrays ( - const char* char_species_name, const char* char_comp_name, - int lev, int* num_tiles, int** particles_per_tile ) { - - const auto & mypc = WarpX::GetInstance().GetPartContainer(); - const std::string species_name(char_species_name); - auto & myspc = mypc.GetParticleContainerFromName(species_name); - - const int comp = warpx_getParticleCompIndex(char_species_name, char_comp_name); - - *num_tiles = myspc.numLocalTilesAtLevel(lev); - *particles_per_tile = static_cast<int*>(malloc(*num_tiles*sizeof(int))); - memset(*particles_per_tile, 0, *num_tiles*sizeof(int)); - - auto data = static_cast<amrex::ParticleReal**>(malloc(*num_tiles*sizeof(amrex::ParticleReal*))); - int i = 0; - for (WarpXParIter pti(myspc, lev); pti.isValid(); ++pti, ++i) { - auto& soa = pti.GetStructOfArrays(); - data[i] = (amrex::ParticleReal*) soa.GetRealData(comp).dataPtr(); - (*particles_per_tile)[i] = pti.numParticles(); - } - return data; - } - - void warpx_convert_id_to_long (amrex::Long* ids, const WarpXParticleContainer::ParticleType* pstructs, int size) - { - amrex::Long* d_ptr = nullptr; -#ifdef AMREX_USE_GPU - amrex::Gpu::DeviceVector<amrex::Long> d_ids(size); - d_ptr = d_ids.data(); -#else - d_ptr = ids; -#endif - amrex::ParallelFor(size, [=] AMREX_GPU_DEVICE (int i) noexcept - { - d_ptr[i] = pstructs[i].id(); - }); -#ifdef AMREX_USE_GPU - amrex::Gpu::dtoh_memcpy(ids, d_ptr, size*sizeof(amrex::Long)); -#endif - } - - void warpx_convert_cpu_to_int (int* cpus, const WarpXParticleContainer::ParticleType* pstructs, int size) - { - int* d_ptr = nullptr; -#ifdef AMREX_USE_GPU - amrex::Gpu::DeviceVector<int> d_cpus(size); - d_ptr = d_cpus.data(); -#else - d_ptr = cpus; -#endif - amrex::ParallelFor(size, [=] AMREX_GPU_DEVICE (int i) noexcept - { - d_ptr[i] = pstructs[i].cpu(); - }); -#ifdef AMREX_USE_GPU - amrex::Gpu::dtoh_memcpy(cpus, d_ptr, size*sizeof(int)); -#endif - } - - int warpx_getParticleCompIndex ( - const char* char_species_name, const char* char_comp_name ) - { - const auto & mypc = WarpX::GetInstance().GetPartContainer(); - - const std::string species_name(char_species_name); - auto & myspc = mypc.GetParticleContainerFromName(species_name); - - const std::string comp_name(char_comp_name); - auto particle_comps = myspc.getParticleComps(); - - return particle_comps.at(comp_name); - } - - void warpx_addRealComp(const char* char_species_name, - const char* char_comp_name, bool comm=true) - { - auto & mypc = WarpX::GetInstance().GetPartContainer(); - const std::string species_name(char_species_name); - auto & myspc = mypc.GetParticleContainerFromName(species_name); - - const std::string comp_name(char_comp_name); - myspc.AddRealComp(comp_name, comm); - - mypc.defineAllParticleTiles(); - } - amrex::Real warpx_sumParticleCharge(const char* char_species_name, const bool local) { auto & mypc = WarpX::GetInstance().GetPartContainer(); @@ -568,117 +98,6 @@ namespace return myspc.sumParticleCharge(local); } - int warpx_getParticleBoundaryBufferSize(const char* species_name, int boundary, bool local) - { - const std::string name(species_name); - auto& particle_buffers = WarpX::GetInstance().GetParticleBoundaryBuffer(); - return particle_buffers.getNumParticlesInContainer(species_name, boundary, local); - } - - int** warpx_getParticleBoundaryBufferScrapedSteps(const char* species_name, int boundary, int lev, - int* num_tiles, int** particles_per_tile) - { - const std::string name(species_name); - auto& particle_buffers = WarpX::GetInstance().GetParticleBoundaryBuffer(); - auto& particle_buffer = particle_buffers.getParticleBuffer(species_name, boundary); - - const int comp = particle_buffer.NumIntComps() - 1; - - *num_tiles = particle_buffer.numLocalTilesAtLevel(lev); - *particles_per_tile = static_cast<int*>(malloc(*num_tiles*sizeof(int))); - memset(*particles_per_tile, 0, *num_tiles*sizeof(int)); - - auto data = static_cast<int**>(malloc(*num_tiles*sizeof(int*))); - int i = 0; - for (amrex::ParIter<0,0,PIdx::nattribs, 0, amrex::PinnedArenaAllocator> pti(particle_buffer, lev); pti.isValid(); ++pti, ++i) { - auto& soa = pti.GetStructOfArrays(); - data[i] = (int*) soa.GetIntData(comp).dataPtr(); - (*particles_per_tile)[i] = pti.numParticles(); - } - - return data; - } - - amrex::ParticleReal** warpx_getParticleBoundaryBuffer(const char* species_name, int boundary, int lev, - int* num_tiles, int** particles_per_tile, const char* comp_name) - { - const std::string name(species_name); - auto& particle_buffers = WarpX::GetInstance().GetParticleBoundaryBuffer(); - auto& particle_buffer = particle_buffers.getParticleBuffer(species_name, boundary); - - const int comp = warpx_getParticleCompIndex(species_name, comp_name); - - *num_tiles = particle_buffer.numLocalTilesAtLevel(lev); - *particles_per_tile = static_cast<int*>(malloc(*num_tiles*sizeof(int))); - memset(*particles_per_tile, 0, *num_tiles*sizeof(int)); - - auto data = static_cast<amrex::ParticleReal**>(malloc(*num_tiles*sizeof(amrex::ParticleReal*))); - int i = 0; - for (amrex::ParIter<0,0,PIdx::nattribs, 0, amrex::PinnedArenaAllocator> pti(particle_buffer, lev); pti.isValid(); ++pti, ++i) { - auto& soa = pti.GetStructOfArrays(); - data[i] = (amrex::ParticleReal*) soa.GetRealData(comp).dataPtr(); - (*particles_per_tile)[i] = pti.numParticles(); - } - - return data; - } - - amrex::ParticleReal** warpx_getParticleBoundaryBufferStructs(const char* species_name, int boundary, int lev, - int* num_tiles, int** particles_per_tile) - { - const std::string name(species_name); - auto& particle_buffers = WarpX::GetInstance().GetParticleBoundaryBuffer(); - auto& particle_buffer = particle_buffers.getParticleBuffer(species_name, boundary); - - *num_tiles = particle_buffer.numLocalTilesAtLevel(lev); - *particles_per_tile = static_cast<int*>(malloc(*num_tiles*sizeof(int))); - memset(*particles_per_tile, 0, *num_tiles*sizeof(int)); - - auto data = static_cast<amrex::ParticleReal**>(malloc(*num_tiles*sizeof(typename WarpXParticleContainer::ParticleType*))); - int i = 0; - for (amrex::ParIter<0,0,PIdx::nattribs, 0, amrex::PinnedArenaAllocator> pti(particle_buffer, lev); pti.isValid(); ++pti, ++i) { - auto& aos = pti.GetArrayOfStructs(); - data[i] = (amrex::ParticleReal*) aos.data(); - (*particles_per_tile)[i] = pti.numParticles(); - } - return data; - } - - void warpx_clearParticleBoundaryBuffer () { - auto& particle_buffers = WarpX::GetInstance().GetParticleBoundaryBuffer(); - particle_buffers.clearParticles(); - } - - void warpx_depositChargeDensity (const char* char_species_name, int lev) { - // this function is used to deposit a given species' charge density - // in the rho_fp multifab which can then be accessed from python via - // pywarpx.fields.RhoFPWrapper() - WarpX& warpx = WarpX::GetInstance(); - const auto & mypc = warpx.GetPartContainer(); - const std::string species_name(char_species_name); - auto & myspc = mypc.GetParticleContainerFromName(species_name); - auto * rho_fp = warpx.get_pointer_rho_fp(lev); - - if (rho_fp == nullptr) { - ablastr::warn_manager::WMRecordWarning( - "WarpXWrappers", "rho_fp is not allocated", - ablastr::warn_manager::WarnPriority::low - ); - return; - } - - for (WarpXParIter pti(myspc, lev); pti.isValid(); ++pti) - { - const long np = pti.numParticles(); - auto& wp = pti.GetAttribs(PIdx::w); - // Do this unconditionally, ignoring myspc.do_not_deposit, to support diagnostic uses - myspc.DepositCharge(pti, wp, nullptr, rho_fp, 0, 0, np, 0, lev, lev); - } -#ifdef WARPX_DIM_RZ - warpx.ApplyInverseVolumeScalingToChargeDensity(rho_fp, lev); -#endif - } - void warpx_ComputeDt () { WarpX& warpx = WarpX::GetInstance(); warpx.ComputeDt(); @@ -704,10 +123,6 @@ namespace WarpX& warpx = WarpX::GetInstance(); warpx.FillBoundaryB(warpx.getngEB()); } - void warpx_SyncRho () { - WarpX& warpx = WarpX::GetInstance(); - warpx.SyncRho(); - } void warpx_SyncCurrent ( const amrex::Vector<std::array<std::unique_ptr<amrex::MultiFab>,3>>& J_fp, const amrex::Vector<std::array<std::unique_ptr<amrex::MultiFab>,3>>& J_cp, @@ -724,18 +139,10 @@ namespace warpx.PushParticlesandDepose(cur_time); } - int warpx_getistep (int lev) { - const WarpX& warpx = WarpX::GetInstance(); - return warpx.getistep(lev); - } void warpx_setistep (int lev, int ii) { WarpX& warpx = WarpX::GetInstance(); warpx.setistep(lev, ii); } - amrex::Real warpx_gett_new (int lev) { - const WarpX& warpx = WarpX::GetInstance(); - return warpx.gett_new(lev); - } void warpx_sett_new (int lev, amrex::Real time) { WarpX& warpx = WarpX::GetInstance(); warpx.sett_new(lev, time); @@ -759,35 +166,6 @@ namespace return warpx.finestLevel(); } - int warpx_getMyProc () { - return amrex::ParallelDescriptor::MyProc(); - } - - int warpx_getNProcs () { - return amrex::ParallelDescriptor::NProcs(); - } - - void warpx_setPotentialEB (const char * char_potential) { - WarpX& warpx = WarpX::GetInstance(); - const std::string potential(char_potential); - warpx.m_poisson_boundary_handler.setPotentialEB(potential); - } - - void warpx_setPlasmaLensStrength (const int i_lens, const amrex::Real strength_E, const amrex::Real strength_B ) { - auto & mypc = WarpX::GetInstance().GetPartContainer(); - WARPX_ALWAYS_ASSERT_WITH_MESSAGE( i_lens >= mypc.h_repeated_plasma_lens_strengths_E.size(), - "i_lens is larger than the number of plasma lenses."); - mypc.h_repeated_plasma_lens_strengths_E[i_lens] = strength_E; - mypc.h_repeated_plasma_lens_strengths_B[i_lens] = strength_B; - amrex::Gpu::copyAsync(amrex::Gpu::hostToDevice, - mypc.h_repeated_plasma_lens_strengths_E.begin(), mypc.h_repeated_plasma_lens_strengths_E.end(), - mypc.d_repeated_plasma_lens_strengths_E.begin()); - amrex::Gpu::copyAsync(amrex::Gpu::hostToDevice, - mypc.h_repeated_plasma_lens_strengths_B.begin(), mypc.h_repeated_plasma_lens_strengths_B.end(), - mypc.d_repeated_plasma_lens_strengths_B.begin()); - amrex::Gpu::synchronize(); - } - void mypc_Redistribute () { auto & mypc = WarpX::GetInstance().GetPartContainer(); mypc.Redistribute(); |