boundary_condition_manager.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 <set>
#include <optional>
#include <vector>
 
#include "smith/physics/boundary_conditions/boundary_condition.hpp"
#include "smith/physics/state/finite_element_state.hpp"
 
namespace smith {
 
class BoundaryConditionManager {
 public:
  explicit BoundaryConditionManager(const mfem::ParMesh& mesh) : num_attrs_(mesh.bdr_attributes.Max()) {}
 
  void addEssential(const std::set<int>& ess_bdr, smith::GeneralCoefficient ess_bdr_coef,
                    mfem::ParFiniteElementSpace& space, const std::optional<int> component = {});
 
  void addEssential(const mfem::Array<int>& local_dofs, std::shared_ptr<mfem::Coefficient> ess_bdr_coef,
                    mfem::ParFiniteElementSpace& space, std::optional<int> component = {});
 
  void addEssentialByTrueDofs(const mfem::Array<int>& true_dofs, std::shared_ptr<mfem::VectorCoefficient> ess_bdr_coef,
                              mfem::ParFiniteElementSpace& space);
 
  const mfem::Array<int>& allEssentialTrueDofs() const
  {
    if (!all_dofs_valid_) {
      updateAllDofs();
    }
    return all_true_dofs_;
  }
 
  const mfem::Array<int>& allEssentialLocalDofs() const
  {
    if (!all_dofs_valid_) {
      updateAllDofs();
    }
    return all_local_dofs_;
  }
 
  std::unique_ptr<mfem::HypreParMatrix> eliminateAllEssentialDofsFromMatrix(mfem::HypreParMatrix& matrix) const
  {
    return std::unique_ptr<mfem::HypreParMatrix>(matrix.EliminateRowsCols(allEssentialTrueDofs()));
  }
 
  std::vector<BoundaryCondition>& essentials() { return ess_bdr_; }
 
  const std::vector<BoundaryCondition>& essentials() const { return ess_bdr_; }
 
 private:
  void updateAllDofs() const;
 
  const int num_attrs_;
 
  std::vector<BoundaryCondition> ess_bdr_;
 
  std::set<int> attrs_in_use_;
 
  mutable mfem::Array<int> all_true_dofs_;
 
  mutable mfem::Array<int> all_local_dofs_;
 
  mutable bool all_dofs_valid_ = false;
};
 
}  // namespace smith