Merge pull request #10 from RemiLehe/charge_buffers

Implement charge buffers

1 files changed, 6 insertions, 0 deletions

diff --git a/Source/WarpX.cpp b/Source/WarpX.cpp
index ea54db2b7..9ca17c759 100644
--- a/Source/WarpX.cpp
+++ b/Source/WarpX.cpp
@@ -161,6 +161,7 @@ WarpX::WarpX ()
     current_buffer_masks.resize(nlevs_max);
     gather_buffer_masks.resize(nlevs_max);
     current_buf.resize(nlevs_max);
+    charge_buf.resize(nlevs_max);
 
     pml.resize(nlevs_max);
 
@@ -463,6 +464,8 @@ WarpX::ClearLevel (int lev)
         current_buf[lev][i].reset();
     }
 
+    charge_buf[lev].reset();
+    
     current_buffer_masks[lev].reset();
     gather_buffer_masks[lev].reset();
 
@@ -644,6 +647,9 @@ WarpX::AllocLevelData (int lev, const BoxArray& ba, const DistributionMapping& d
             current_buf[lev][0].reset( new MultiFab(amrex::convert(cba,jx_nodal_flag),dm,1,ngJ));
             current_buf[lev][1].reset( new MultiFab(amrex::convert(cba,jy_nodal_flag),dm,1,ngJ));
             current_buf[lev][2].reset( new MultiFab(amrex::convert(cba,jz_nodal_flag),dm,1,ngJ));
+            if (do_dive_cleaning) {
+                charge_buf[lev].reset( new MultiFab(amrex::convert(cba,IntVect::TheUnitVector()),dm,2,ngRho));
+            }
             current_buffer_masks[lev].reset( new iMultiFab(ba, dm, 1, 0) );
         }
     }