Refactoring_code_structure

PyPIC3D/__init__.py

@@ -6,17 +6,28 @@
 import matplotlib.pyplot as plt
 # import external libraries
 
-jax.config.update('jax_platform_name', 'cpu')
-
 from . import errors
 from . import boundaryconditions
 from . import initialization
-from . import particle
-from .diagnostics import plotting
 from . import utils
-from .solvers import pstd
-from .solvers import fdtd
 from . import boris
-from . import rho
 from . import evolve
-from .solvers import vector_potential
+
+from .solvers import vector_potential
+from .solvers import pstd
+from .solvers import fdtd
+from .solvers import first_order_yee
+from .solvers import electrostatic_yee
+
+from .particles import particle_initialization
+from .particles import species_class
+
+from .deposition import shapes
+from .deposition import Esirkepov
+from .deposition import J_from_rhov
+from .deposition import rho
+
+from .diagnostics import plotting
+from .diagnostics import fluid_quantities
+from .diagnostics import openPMD
+from .diagnostics import vtk

PyPIC3D/__main__.py

@@ -40,7 +40,7 @@
     compute_mass_density
 )
 
-from PyPIC3D.rho import compute_rho
+from PyPIC3D.deposition.rho import compute_rho
 
 
 # Importing functions from the PyPIC3D package

PyPIC3D/boris.py

@@ -2,7 +2,7 @@
 from jax import jit
 import jax.numpy as jnp
 
-from PyPIC3D.shapes import get_first_order_weights, get_second_order_weights
+from PyPIC3D.deposition.shapes import get_first_order_weights, get_second_order_weights
 from PyPIC3D.utils import wrap_around
 
 @jit