Change _libwarpx.py functions to LibWarpX class methods (#2696)

* Initial attempt at moving function into LibWarpX

* Bugfix - undefined structs in get_particle_theta

This is untested, identified by IDE/visual inspection only

* Unify newlines

* Delete libwarpx.clight (duplicated picmi.py)

See picmi.constants.c

* Change function descriptive comment to docstring

Also remove extra newline in __init__ beginning.

* Replace pywarpx/_libwarpx calls appropriately

* Fix atexit register of finalize function

* Reorder WarpX.py imports for pre-commit

* Use picmi.Simulation.extension = libwarpx alias

This will hopefully preempt user script changes if we restructure things
to allow multiple WarpX runs to be done within a single Python script.

* Fix libwarpx ref in WarpX.getProbLo/getProbHi

* Update get_particle_boundary_buffer doc reference

1 files changed, 5 insertions, 6 deletions

diff --git a/Examples/Tests/pass_mpi_communicator/PICMI_inputs_2d.py b/Examples/Tests/pass_mpi_communicator/PICMI_inputs_2d.py
index fbb108a0e..b97905ee0 100755
--- a/Examples/Tests/pass_mpi_communicator/PICMI_inputs_2d.py
+++ b/Examples/Tests/pass_mpi_communicator/PICMI_inputs_2d.py
@@ -6,7 +6,6 @@
 # --- if the correct amount of processors are initialized in AMReX.
 
 from mpi4py import MPI
-import pywarpx
 from pywarpx import picmi
 
 constants = picmi.constants
@@ -128,14 +127,14 @@ new_comm_size = new_comm.size
 
 if color == 0:
     # verify that communicator contains correct number of procs (1)
-    assert pywarpx.getNProcs() == comm_world_size - 1
-    assert pywarpx.getNProcs() == new_comm_size
+    assert sim.extension.getNProcs() == comm_world_size - 1
+    assert sim.extension.getNProcs() == new_comm_size
 
 else:
     # verify that amrex initialized with 1 fewer proc than comm world
-    assert pywarpx.getNProcs() == comm_world_size - 1
-    assert pywarpx.getNProcs() == new_comm_size
+    assert sim.extension.getNProcs() == comm_world_size - 1
+    assert sim.extension.getNProcs() == new_comm_size
 
     # verify that amrex proc ranks are offset by -1 from
     # world comm proc ranks
-    assert pywarpx.getMyProc() == rank - 1
+    assert sim.extension.getMyProc() == rank - 1