Initial commit; basic structure.

Author: cemitch99

src/initialization/Algorithms.H

@@ -23,7 +23,8 @@ namespace impactx
         True_3D,  /**< 3D beam distribution */
         Gauss3D,  /**< Assume a 3D Gaussian beam distribution */
         Gauss2p5D,  /**< Assume a transverse 2D Gaussian beam distribution */
-        True_2D   /**< Averaged 2D transverse beam distribution with a current along s */
+        True_2D,   /**< Averaged 2D transverse beam distribution */
+        True_2p5D   /**< Averaged 2D transverse beam distribution with a current along s */
     );
 
     /** Return the currently active space charge algorithm */

src/initialization/Algorithms.cpp

@@ -74,6 +74,10 @@ namespace impactx
         {
             return SpaceChargeAlgo::True_2D;
         }
+        else if (space_charge == "2p5D")
+        {
+            return SpaceChargeAlgo::True_2p5D;
+        }
         else
         {
             throw std::runtime_error("algo.space_charge = " + space_charge + " is not a valid option");

src/particles/spacecharge/GatherAndPush.cpp

@@ -82,7 +82,7 @@ namespace impactx::particles::spacecharge
                 amrex::ParticleReal const push_consts = dt * charge * inv_gamma2 / pz_ref_SI;
 
                 // gather to each particle and push momentum
-                if (space_charge == SpaceChargeAlgo::True_2D) {
+                if (space_charge == SpaceChargeAlgo::True_2D || space_charge == SpaceChargeAlgo::True_2p5D) {
                     // flatten 3rd dimension
                     auto prob_lo_2D = gm.ProbLoArray();
                     prob_lo_2D[2] = 0.0_rt;

src/particles/spacecharge/HandleSpacecharge.cpp

@@ -43,7 +43,7 @@ namespace impactx::particles::spacecharge
         // turn off if less than 2 particles
         if (amr_data->track_particles.m_particle_container->TotalNumberOfParticles(true, false) < 2) { return; }
 
-        if (space_charge != SpaceChargeAlgo::True_2D)
+        if (space_charge != SpaceChargeAlgo::True_2D && space_charge != SpaceChargeAlgo::True_2p5D)
         {
             // transform from x',y',t to x,y,z
             transformation::CoordinateTransformation(
@@ -60,7 +60,7 @@ namespace impactx::particles::spacecharge
         {
             Gauss2p5dPush(*amr_data->track_particles.m_particle_container, slice_ds);
         }
-        else if (space_charge == SpaceChargeAlgo::True_3D || space_charge == SpaceChargeAlgo::True_2D)
+        else if (space_charge == SpaceChargeAlgo::True_3D || space_charge == SpaceChargeAlgo::True_2D || space_charge == SpaceChargeAlgo::True_2p5D)
         {
             // Note: The following operations assume that
             // the particles are in x, y, z coordinates.
@@ -107,7 +107,7 @@ namespace impactx::particles::spacecharge
             );
         }
 
-        if (space_charge != SpaceChargeAlgo::True_2D)
+        if (space_charge != SpaceChargeAlgo::True_2D && space_charge != SpaceChargeAlgo::True_2p5D)
         {
             // transform from x,y,z to x',y',t
             transformation::CoordinateTransformation(