mesh.hpp

Go to the documentation of this file.

// Copyright (c) Lawrence Livermore National Security, LLC and
// other Smith Project Developers. See the top-level LICENSE file for
// details.
//
// SPDX-License-Identifier: (BSD-3-Clause)
 
#pragma once
 
#include <memory>
#include <string>
#include <functional>
#include <map>
#include <vector>
 
#include "mpi.h"
 
#include "mfem.hpp"
#include "smith/numerics/functional/tensor.hpp"
#include "smith/numerics/functional/domain.hpp"
 
namespace smith {
 
// Forward declare
struct Domain;
class FiniteElementState;
class FiniteElementDual;
 
class Mesh {
 public:
  Mesh(mfem::Mesh&& mesh, const std::string& meshtag, int serial_refine = 0, int parallel_refine = 0,
       MPI_Comm comm = MPI_COMM_WORLD);
 
  Mesh(mfem::ParMesh&& mesh, const std::string& meshtag);
 
  Mesh(const std::string& meshfile, const std::string& meshtag, int serial_refine = 0, int parallel_refine = 0,
       MPI_Comm comm = MPI_COMM_WORLD);
 
  const std::string& tag() const { return mesh_tag_; }
 
  const mfem::ParMesh& mfemParMesh() const { return *mfem_mesh_; }
 
  mfem::ParMesh& mfemParMesh() { return *mfem_mesh_; }
 
  MPI_Comm getComm() const;
 
  static std::string entireBodyName() { return "entire_body"; }
 
  smith::Domain& entireBody() const;
 
  static std::string entireBoundaryName() { return "entire_boundary"; }
 
  smith::Domain& entireBoundary() const;
 
  static std::string internalBoundaryName() { return "internal_boundary"; }
 
  smith::Domain& internalBoundary() const;
 
  void insertDomain(const std::string& domain_name, const Domain& domain);
 
  smith::Domain& domain(const std::string& domain_name) const;
 
  smith::Domain& addDomainOfBoundaryElements(const std::string& domain_name,
                                             std::function<bool(std::vector<vec3>, int)> func);
 
  smith::Domain& addDomainOfBoundaryElements(const std::string& domain_name,
                                             std::function<bool(std::vector<vec2>, int)> func);
 
  smith::Domain& addDomainOfInternalBoundaryElements(const std::string& domain_name,
                                                     std::function<bool(std::vector<vec3>, int)> func);
 
  smith::Domain& addDomainOfInternalBoundaryElements(const std::string& domain_name,
                                                     std::function<bool(std::vector<vec2>, int)> func);
 
  smith::Domain& addDomainOfBodyElements(const std::string& domain_name,
                                         std::function<bool(std::vector<vec3>, int)> func);
 
  smith::Domain& addDomainOfBodyElements(const std::string& domain_name,
                                         std::function<bool(std::vector<vec2>, int)> func);
 
  const mfem::ParFiniteElementSpace& shapeDisplacementSpace();
 
  smith::FiniteElementState newShapeDisplacement();
 
  smith::FiniteElementDual newShapeDisplacementDual();
 
 private:
  void notifyIfRankHasNoElements() const;
 
  void createDomains();
 
  void errorIfDomainExists(const std::string& domain_name) const;
 
  std::string mesh_tag_;
 
  mfem::ParMesh* mfem_mesh_;
 
  mutable std::map<std::string, smith::Domain> domains_;
};
 
}  // namespace smith