switch the MHD / MOL hydro to use the flatten header (#2906)

this eliminates the duplicate flatten.cpp version
AMReX-Astro · Jul 10, 2024 · dd89b8e · dd89b8e
1 parent 04f8622
commit dd89b8e
Show file tree

Hide file tree

Showing 5 changed files with 19 additions and 199 deletions.
diff --git a/Source/hydro/Castro_hydro.H b/Source/hydro/Castro_hydro.H
@@ -118,19 +118,6 @@
                  amrex::Array4<amrex::Real> const& q_arr,
                  amrex::Array4<amrex::Real> const& qaux_arr);
 
-///
-/// compute the flattening coefficient.  This is 0 if we are in a shock and
-/// 1 if we are not applying any flattening
-///
-/// @param bx         the box to operate over
-/// @param q_arr      the primitive variable state
-/// @param flatn      the flattening coefficient
-/// @param pres_comp  index into q_arr of the pressure component
-///
-    static void uflatten(const amrex::Box& bx,
-                  amrex::Array4<amrex::Real const> const& q_arr,
-                  amrex::Array4<amrex::Real> const& flatn,
-                  const int pres_comp);
 
 ///
 /// A multidimensional shock detection algorithm

diff --git a/Source/hydro/Castro_mol_hydro.cpp b/Source/hydro/Castro_mol_hydro.cpp
@@ -8,6 +8,7 @@
 #include <advection_util.H>
 
 #include <fourth_center_average.H>
+#include <flatten.H>
 
 using namespace amrex;
 
@@ -117,19 +118,23 @@ Castro::construct_mol_hydro_source(Real time, Real dt, MultiFab& A_update)
         Array4<Real> const flatn_arr = flatn.array();
 
         if (first_order_hydro == 1) {
-          amrex::ParallelFor(obx,
-          [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept
-          {
-            flatn_arr(i,j,k) = 0.0;
-          });
+            amrex::ParallelFor(obx,
+            [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept
+            {
+                flatn_arr(i,j,k) = 0.0;
+            });
         } else if (use_flattening == 1) {
-          uflatten(obx, q_arr, flatn_arr, QPRES);
+            amrex::ParallelFor(obx,
+            [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept
+            {
+                flatn_arr(i,j,k) = hydro::flatten(i, j, k, q_arr, QPRES);
+            });
         } else {
-          amrex::ParallelFor(obx,
-          [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept
-          {
-            flatn_arr(i,j,k) = 1.0;
-          });
+            amrex::ParallelFor(obx,
+            [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept
+            {
+                flatn_arr(i,j,k) = 1.0;
+            });
         }
 
 

diff --git a/Source/hydro/Make.package b/Source/hydro/Make.package
@@ -10,7 +10,6 @@ CEXE_sources += Castro_mol.cpp
 CEXE_headers += advection_util.H
 CEXE_sources += advection_util.cpp
 CEXE_headers += flatten.H
-CEXE_sources += flatten.cpp
 
 ifeq ($(USE_TRUE_SDC),TRUE)
   CEXE_sources += Castro_mol_hydro.cpp

diff --git a/Source/hydro/flatten.cpp b/Source/hydro/flatten.cpp
diff --git a/Source/mhd/Castro_mhd.cpp b/Source/mhd/Castro_mhd.cpp
@@ -1,6 +1,7 @@
 #include <Castro.H>
 
 #include <advection_util.H>
+#include <flatten.H>
 
 using namespace amrex;
 
@@ -196,10 +197,6 @@ Castro::construct_ctu_mhd_source(Real time, Real dt)
           auto flatn_arr = flatn.array();
           auto elix_flatn = flatn.elixir();
 
-          flatg.resize(bxi, 1);
-          auto flatg_arr = flatg.array();
-          auto elix_flatg = flatg.elixir();
-
           if (use_flattening == 0) {
             amrex::ParallelFor(bxi,
             [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept
@@ -209,13 +206,11 @@ Castro::construct_ctu_mhd_source(Real time, Real dt)
 
           } else {
 
-            uflatten(bxi, q_arr, flatn_arr, QPRES);
-            uflatten(bxi, q_arr, flatg_arr, QPTOT);
-
             amrex::ParallelFor(bxi,
             [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept
             {
-              flatn_arr(i,j,k) = flatn_arr(i,j,k) * flatg_arr(i,j,k);
+                flatn_arr(i,j,k) = hydro::flatten(i, j, k, q_arr, QPRES) *
+                                   hydro::flatten(i, j, k, q_arr, QPTOT);
             });
 
           }