1 files changed, 94 insertions, 0 deletions
diff --git a/Source/Parallelization/WarpXCommUtil.H b/Source/Parallelization/WarpXCommUtil.H
new file mode 100644
index 000000000..2a5ec5390
--- /dev/null
+++ b/Source/Parallelization/WarpXCommUtil.H
@@ -0,0 +1,94 @@
+/* Copyright 2019 Andrew Myers
+ *
+ * This file is part of WarpX.
+ *
+ * License: BSD-3-Clause-LBNL
+ */
+#ifndef WARPX_COMMUTIL_H_
+#define WARPX_COMMUTIL_H_
+
+#include <AMReX_FabArray.H>
+#include <AMReX_Gpu.H>
+#include <AMReX_iMultiFab.H>
+#include <AMReX_MultiFab.H>
+#include <AMReX_Periodicity.H>
+#include <AMReX_TypeTraits.H>
+
+#include "WarpX.H"
+
+namespace WarpXCommUtil
+{
+
+using comm_float_type = float;
+
+template <class FAB1, class FAB2>
+void
+mixedCopy (amrex::FabArray<FAB1>& dst, amrex::FabArray<FAB2> const& src, int srccomp, int dstcomp, int numcomp, const amrex::IntVect& nghost)
+{
+    auto const& srcma = src.const_arrays();
+    auto const& dstma = dst.arrays();
+    ParallelFor(dst, nghost, numcomp,
+    [=] AMREX_GPU_DEVICE (int box_no, int i, int j, int k, int n) noexcept
+    {
+        dstma[box_no](i,j,k,dstcomp+n) = (typename FAB1::value_type) srcma[box_no](i,j,k,srccomp+n);
+    });
+    amrex::Gpu::synchronize();
+}
+
+void ParallelCopy (amrex::MultiFab&            dst,
+                   const amrex::MultiFab&      src,
+                   int                         src_comp,
+                   int                         dst_comp,
+                   int                         num_comp,
+                   const amrex::IntVect&       src_nghost,
+                   const amrex::IntVect&       dst_nghost,
+                   const amrex::Periodicity&   period = amrex::Periodicity::NonPeriodic(),
+                   amrex::FabArrayBase::CpOp   op = amrex::FabArrayBase::COPY);
+
+void ParallelAdd (amrex::MultiFab&            dst,
+                  const amrex::MultiFab&      src,
+                  int                         src_comp,
+                  int                         dst_comp,
+                  int                         num_comp,
+                  const amrex::IntVect&       src_nghost,
+                  const amrex::IntVect&       dst_nghost,
+                  const amrex::Periodicity&   period = amrex::Periodicity::NonPeriodic());
+
+void FillBoundary (amrex::MultiFab&          mf,
+                   const amrex::Periodicity& period = amrex::Periodicity::NonPeriodic());
+
+void FillBoundary (amrex::MultiFab&          mf,
+                   amrex::IntVect            ng,
+                   const amrex::Periodicity& period = amrex::Periodicity::NonPeriodic());
+
+void FillBoundary (amrex::iMultiFab&         mf,
+                   const amrex::Periodicity& period = amrex::Periodicity::NonPeriodic());
+
+void FillBoundary (amrex::iMultiFab&         mf,
+                   amrex::IntVect            ng,
+                   const amrex::Periodicity& period = amrex::Periodicity::NonPeriodic());
+
+void
+FillBoundary (amrex::Vector<amrex::MultiFab*> const& mf, const amrex::Periodicity& period);
+
+void SumBoundary (amrex::MultiFab&          mf,
+                  const amrex::Periodicity& period = amrex::Periodicity::NonPeriodic());
+
+void SumBoundary (amrex::MultiFab&          mf,
+                  int                       start_comp,
+                  int                       num_comps,
+                  amrex::IntVect            ng,
+                  const amrex::Periodicity& period = amrex::Periodicity::NonPeriodic());
+
+void SumBoundary (amrex::MultiFab&          mf,
+                  int                       start_comp,
+                  int                       num_comps,
+                  amrex::IntVect            src_ng,
+                  amrex::IntVect            dst_ng,
+                  const amrex::Periodicity& period = amrex::Periodicity::NonPeriodic());
+
+void OverrideSync (amrex::MultiFab&          mf,
+                   const amrex::Periodicity& period = amrex::Periodicity::NonPeriodic());
+}
+
+#endif