diff options
Diffstat (limited to 'Source/Initialization')
| -rw-r--r-- | Source/Initialization/PlasmaInjector.cpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/Source/Initialization/PlasmaInjector.cpp b/Source/Initialization/PlasmaInjector.cpp index f9642d1b6..a7dbe728a 100644 --- a/Source/Initialization/PlasmaInjector.cpp +++ b/Source/Initialization/PlasmaInjector.cpp @@ -218,7 +218,7 @@ void RegularPosition::getPositionUnitBox(vec3& r, int i_part, int ref_fac) { int nx = ref_fac*_num_particles_per_cell_each_dim[0]; int ny = ref_fac*_num_particles_per_cell_each_dim[1]; -#if AMREX_SPACEDIM == 3 +#if (AMREX_SPACEDIM == 3) || (defined WARPX_RZ) int nz = ref_fac*_num_particles_per_cell_each_dim[2]; #else int nz = 1; @@ -296,9 +296,9 @@ PlasmaInjector::PlasmaInjector(int ispecies, const std::string& name) parseDensity(pp); parseMomentum(pp); } else if (part_pos_s == "nuniformpercell") { - num_particles_per_cell_each_dim.resize(3); + num_particles_per_cell_each_dim.assign(3, 1); pp.getarr("num_particles_per_cell_each_dim", num_particles_per_cell_each_dim); -#if ( AMREX_SPACEDIM == 2 ) +#if ( AMREX_SPACEDIM == 2 ) && !defined(WARPX_RZ) num_particles_per_cell_each_dim[2] = 1; #endif part_pos.reset(new RegularPosition(num_particles_per_cell_each_dim)); |