AMReX_distFcnElement.H

Go to the documentation of this file.

#ifndef AMREX_DISTFCNELEMENT_H_
#define AMREX_DISTFCNELEMENT_H_
#include <AMReX_Config.H>
 
#include <AMReX_RealVect.H>
#include <AMReX.H>
#include <AMReX_Vector.H>
 
namespace amrex {
 
class distFcnElement2d { // NOLINT(cppcoreguidelines-special-member-functions)
 public:
  distFcnElement2d() = default;
  virtual ~distFcnElement2d() = default;
 
  [[nodiscard]] virtual distFcnElement2d* newDistFcnElement2d() const = 0;
 
  [[nodiscard]] virtual amrex::Real cpdist(amrex::RealVect pt, amrex::RealVect& cp) const = 0;
  [[nodiscard]] virtual amrex::Real cpside(amrex::RealVect pt, amrex::RealVect& cp) const = 0;
  static int solve_thomas (const std::vector<amrex::Real> &a,
                           std::vector<amrex::Real> &b,
                           const std::vector<amrex::Real> &c,
                           std::vector<amrex::Real> &d,
                           std::vector<amrex::Real> &X);
};
 
 
class LineDistFcnElement2d: public distFcnElement2d { // NOLINT(cppcoreguidelines-special-member-functions)
 public:
  LineDistFcnElement2d() = default;
  ~LineDistFcnElement2d() override = default;
 
  [[nodiscard]] distFcnElement2d* newDistFcnElement2d() const override;
 
  void set_control_points(std::vector<amrex::RealVect>  pts);
 
  [[nodiscard]] amrex::Real cpdist(amrex::RealVect pt, amrex::RealVect& cp) const override;
  [[nodiscard]] amrex::Real cpside(amrex::RealVect pt, amrex::RealVect& cp) const override;
 
  void print_control_points();
 
 protected:
  static void single_seg_cpdist(amrex::RealVect pt,
                         amrex::Real x0, amrex::Real x1,
                         amrex::Real y0, amrex::Real y1,
                         amrex::RealVect& cp,
                         amrex::Real& dist);
 
 private:
  std::vector<amrex::Real> control_points_x;
  std::vector<amrex::Real> control_points_y;
 
 
};
 
 
class SplineDistFcnElement2d: public distFcnElement2d { // NOLINT(cppcoreguidelines-special-member-functions)
 public:
  SplineDistFcnElement2d() = default;
  ~SplineDistFcnElement2d() override = default;
 
  [[nodiscard]] distFcnElement2d* newDistFcnElement2d() const override;
 
 
  void set_control_points(std::vector<amrex::RealVect>  pts);
  void set_bc_points(amrex::RealVect start, amrex::RealVect end);
 
  void print_control_points() const;
  void print_spline() const;
 
  void calc_D(bool clamped_bc = false);
 
  [[nodiscard]] amrex::Real cpdist(amrex::RealVect pt, amrex::RealVect& cp) const override;
  [[nodiscard]] amrex::Real cpside(amrex::RealVect pt, amrex::RealVect& cp) const override;
 
 protected:
  static amrex::Real eval(amrex::Real t, amrex::Real y0, amrex::Real y1,
                   amrex::Real D0, amrex::Real D1);
  static void  dxbydt(amrex::Real t, amrex::Real y0, amrex::Real y1,
               amrex::Real D0, amrex::Real D1, amrex::Real& dyf,
               amrex::Real& d2yf);
 
  static void single_spline_cpdist(amrex::RealVect pt,
                            amrex::Real x0, amrex::Real x1,
                            amrex::Real Dx0, amrex::Real Dx1,
                            amrex::Real y0, amrex::Real y1,
                            amrex::Real Dy0, amrex::Real Dy1,
                            amrex::Real& t, amrex::RealVect& cp,
                            amrex::Real& dist);
 
  static amrex::Real dist(amrex::RealVect pt,
                   amrex::Real x0, amrex::Real x1,
                   amrex::Real Dx0, amrex::Real Dx1,
                   amrex::Real y0, amrex::Real y1,
                   amrex::Real Dy0, amrex::Real Dy1,
                   amrex::Real& t,
                   amrex::RealVect& spt);
 
 private:
  std::vector<amrex::Real> control_points_x;
  std::vector<amrex::Real> control_points_y;
 
  amrex::RealVect bc_pt_start;
  amrex::RealVect bc_pt_end;
 
  std::vector<amrex::Real> Dx;
  std::vector<amrex::Real> Dy;
};
 
}
 
#endif