domain.hpp

Go to the documentation of this file.

// Copyright (c) 2019-2024, Lawrence Livermore National Security, LLC and
// other Serac Project Developers. See the top-level LICENSE file for
// details.
//
// SPDX-License-Identifier: (BSD-3-Clause)
 
#pragma once
 
#include <vector>
#include "mfem.hpp"
 
#include "serac/numerics/functional/tensor.hpp"
 
namespace serac {
 
struct Domain {
  enum Type
  {
    Elements,
    BoundaryElements
  };
 
  static constexpr int num_types = 2;  
 
  const mfem::Mesh& mesh_;
 
  int dim_;
 
  Type type_;
 
  std::vector<int> vertex_ids_;
  std::vector<int> edge_ids_;
  std::vector<int> tri_ids_;
  std::vector<int> quad_ids_;
  std::vector<int> tet_ids_;
  std::vector<int> hex_ids_;
 
  Domain(const mfem::Mesh& m, int d, Type type = Domain::Type::Elements) : mesh_(m), dim_(d), type_(type) {}
 
  static Domain ofVertices(const mfem::Mesh& mesh, std::function<bool(vec2)> func);
 
  static Domain ofVertices(const mfem::Mesh& mesh, std::function<bool(vec3)> func);
 
  static Domain ofEdges(const mfem::Mesh& mesh, std::function<bool(std::vector<vec2>, int)> func);
 
  static Domain ofEdges(const mfem::Mesh& mesh, std::function<bool(std::vector<vec3>)> func);
 
  static Domain ofFaces(const mfem::Mesh& mesh, std::function<bool(std::vector<vec2>, int)> func);
 
  static Domain ofFaces(const mfem::Mesh& mesh, std::function<bool(std::vector<vec3>, int)> func);
 
  static Domain ofElements(const mfem::Mesh& mesh, std::function<bool(std::vector<vec2>, int)> func);
 
  static Domain ofElements(const mfem::Mesh& mesh, std::function<bool(std::vector<vec3>, int)> func);
 
  static Domain ofBoundaryElements(const mfem::Mesh& mesh, std::function<bool(std::vector<vec2>, int)> func);
 
  static Domain ofBoundaryElements(const mfem::Mesh& mesh, std::function<bool(std::vector<vec3>, int)> func);
 
  const std::vector<int>& get(mfem::Geometry::Type geom) const
  {
    if (geom == mfem::Geometry::POINT) return vertex_ids_;
    if (geom == mfem::Geometry::SEGMENT) return edge_ids_;
    if (geom == mfem::Geometry::TRIANGLE) return tri_ids_;
    if (geom == mfem::Geometry::SQUARE) return quad_ids_;
    if (geom == mfem::Geometry::TETRAHEDRON) return tet_ids_;
    if (geom == mfem::Geometry::CUBE) return hex_ids_;
 
    exit(1);
  }
};
 
Domain EntireDomain(const mfem::Mesh& mesh);
 
Domain EntireBoundary(const mfem::Mesh& mesh);
 
Domain operator|(const Domain& a, const Domain& b);
 
Domain operator&(const Domain& a, const Domain& b);
 
Domain operator-(const Domain& a, const Domain& b);
 
template <int dim>
inline auto by_attr(int value)
{
  return [value](std::vector<tensor<double, dim> >, int attr) { return attr == value; };
}
 
}  // namespace serac