From b5057a44cd8dba90960df04a923da27171593322 Mon Sep 17 00:00:00 2001
From: Dave Grote <grote1@llnl.gov>
Date: Fri, 1 Feb 2019 18:16:25 -0800
Subject: RZ: setup theta for particles

---
 Source/PhysicalParticleContainer.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'Source/PhysicalParticleContainer.cpp')

diff --git a/Source/PhysicalParticleContainer.cpp b/Source/PhysicalParticleContainer.cpp
index a21285a17..564724f49 100644
--- a/Source/PhysicalParticleContainer.cpp
+++ b/Source/PhysicalParticleContainer.cpp
@@ -155,7 +155,7 @@ PhysicalParticleContainer::AddGaussianBeam(Real x_m, Real y_m, Real z_m,
        std::array<Real, 3> u;
        Real weight;
        for (long i = 0; i < npart; ++i) {
-#if ( AMREX_SPACEDIM == 3 )
+#if ( AMREX_SPACEDIM == 3 | WARPX_RZ)
             weight = q_tot/npart/charge;
             Real x = distx(mt);
             Real y = disty(mt);
-- 
cgit v1.2.3