Slopes test (AMReX-Codes#1859)

Slope computation on cell centroids using the Least Square method. These tests are based from the LeastSquares Test (AMReX-Codes#1707) but instead of calculating scalar values on face centroids they are calculated on cell centroids using a linear function rather than a second order polynomial (AMReX-Codes#1707).

Author: vzendejasl

Tests/Slopes/GNUmakefile

@@ -0,0 +1,26 @@
+TEST = TRUE
+USE_ASSERTION = TRUE
+
+USE_EB = TRUE
+DEBUG = FALSE
+USE_MPI  = TRUE
+USE_OMP  = FALSE
+
+COMP = gnu
+
+DIM = 3
+
+AMREX_HOME ?= ../..
+
+include $(AMREX_HOME)/Tools/GNUMake/Make.defs
+include ./Make.package
+
+Pdirs := Base Boundary AmrCore
+Pdirs += EB
+
+Ppack	+= $(foreach dir, $(Pdirs), $(AMREX_HOME)/Src/$(dir)/Make.package)
+
+include $(Ppack)
+
+include $(AMREX_HOME)/Tools/GNUMake/Make.rules
+

Tests/Slopes/Make.package

@@ -0,0 +1,4 @@
+
+CEXE_sources += main.cpp
+CEXE_sources += MyTest.cpp initEB.cpp initData.cpp initLinearData.cpp initLinearDataFor2D.cpp initLinearDataFor3D.cpp
+CEXE_headers += MyTest.H MyEB.H

Tests/Slopes/MyEB.H

@@ -0,0 +1,59 @@
+#ifndef MY_EB_H_
+#define MY_EB_H_
+
+#include <AMReX_Array.H>
+#include <AMReX_Config.H>
+#include <AMReX_EB2_IF_Base.H>
+
+#include <cmath>
+#include <algorithm>
+
+// For all implicit functions, >0: body; =0: boundary; <0: fluid
+
+class FlowerIF
+    : amrex::GPUable
+{
+public:
+
+    FlowerIF (amrex::Real a_radius, amrex::Real a_delta, int a_npetals,
+              const amrex::RealArray& a_center, bool a_inside)
+        : m_r(a_radius),
+          m_dr(a_delta),
+          m_npetals(a_npetals),
+          m_center(amrex::makeXDim3(a_center)),
+          m_inside(a_inside),
+          m_sign(a_inside ? 1.0 : -1.0)
+        {}
+
+    FlowerIF (const FlowerIF& rhs) noexcept = default;
+    FlowerIF (FlowerIF&& rhs) noexcept = default;
+    FlowerIF& operator= (const FlowerIF& rhs) = delete;
+    FlowerIF& operator= (FlowerIF&& rhs) = delete;
+
+    AMREX_GPU_HOST_DEVICE inline
+    amrex::Real operator() (AMREX_D_DECL(amrex::Real x, amrex::Real y, amrex::Real z))
+        const noexcept
+    {
+        amrex::Real posx = x - m_center.x;
+        amrex::Real posy = y - m_center.y;
+        amrex::Real r = std::hypot(posx, posy);
+        amrex::Real theta = std::atan2(posy, posx);
+        return m_sign*(r - m_r - m_dr * std::cos(m_npetals*theta));
+    }
+
+    inline amrex::Real operator() (const amrex::RealArray& p) const noexcept
+    {
+        return this->operator() (AMREX_D_DECL(p[0], p[1], p[2]));
+    }
+
+protected:
+    amrex::Real      m_r;
+    amrex::Real      m_dr;
+    amrex::Real      m_npetals;
+    amrex::XDim3     m_center;
+    bool             m_inside;
+    //
+    amrex::Real      m_sign;
+};
+
+#endif

Tests/Slopes/MyTest.H

@@ -0,0 +1,72 @@
+#ifndef MY_TEST_H_
+#define MY_TEST_H_
+
+#include <AMReX_Config.H>
+#include <AMReX_EBFabFactory.H>
+#include <AMReX_Array.H>
+
+class MyTest
+{
+public:
+
+    MyTest ();
+    ~MyTest ();
+
+    void compute_gradient ();
+    void writePlotfile ();
+    void initData ();
+
+private:
+
+    void initializeEB ();
+    void readParameters ();
+    void initGrids ();
+
+    void initializeLinearData (int ilev);
+    void initializeLinearDataFor2D (int ilev);
+    void initializeLinearDataFor3D (int ilev);
+
+    int max_level = 0;
+    int ref_ratio = 2;
+    int n_cell = 128;
+    int max_grid_size = 64;
+    amrex::Vector<int> is_periodic = amrex::Vector<int>(3,0);
+    int eb_is_dirichlet = 0;
+    int eb_is_homog_dirichlet = 0;
+    amrex::Vector<amrex::Real> prob_lo = amrex::Vector<amrex::Real>(3,0.0);
+    amrex::Vector<amrex::Real> prob_hi = amrex::Vector<amrex::Real>(3,1.0);
+
+    std::string plot_file_name{"plot"};
+
+    amrex::Vector<amrex::Real> scalars;
+
+    int max_coarsening_level = 0;
+
+    bool use_linear_1d = false;
+    bool linear_1d_askew = false;
+    amrex::Vector<amrex::Real> linear_1d_pt_on_top_wall = amrex::Vector<amrex::Real>(3,0.0);
+    amrex::Real linear_1d_height = 1.0;
+    amrex::Real linear_1d_rotation = 0.;
+    int linear_1d_flow_dir = 0;
+    int linear_1d_height_dir = 1;
+    amrex::Real linear_1d_bottom = 0.0;
+    amrex::Vector<amrex::Real> linear_1d_askew_rotation = amrex::Vector<amrex::Real>(2,0.0);
+    int linear_1d_no_flow_dir = 0;
+
+    amrex::Vector<amrex::Geometry> geom;
+    amrex::Vector<amrex::BoxArray> grids;
+    amrex::Vector<amrex::DistributionMapping> dmap;
+    amrex::Vector<std::unique_ptr<amrex::EBFArrayBoxFactory> > factory;
+
+    amrex::Vector<amrex::MultiFab> phi;
+    amrex::Vector<amrex::MultiFab> grad_x;
+    amrex::Vector<amrex::MultiFab> grad_x_analytic;
+    amrex::Vector<amrex::MultiFab> grad_y;
+    amrex::Vector<amrex::MultiFab> grad_y_analytic;
+    amrex::Vector<amrex::MultiFab> grad_z;
+    amrex::Vector<amrex::MultiFab> grad_z_analytic;
+    amrex::Vector<amrex::MultiFab> ccentr;
+
+};
+
+#endif