contact_data.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 <functional>
#include <memory>
#include <optional>
#include <set>
#include <vector>
 
#include "mfem.hpp"
 
#include "smith/smith_config.hpp"
#include "smith/physics/contact/contact_config.hpp"
#include "smith/physics/state/finite_element_dual.hpp"
#ifdef SMITH_USE_TRIBOL
#include "smith/physics/contact/contact_interaction.hpp"
#endif
 
namespace smith {
 
namespace contact {
 
const ContactOptions default_contact_options = {.method = ContactMethod::SingleMortar,
                                                .enforcement = ContactEnforcement::Penalty,
                                                .type = ContactType::Frictionless,
                                                .penalty = 1.0e3};
 
}  // namespace contact
 
class ContactData {
 public:
  ContactData(const mfem::ParMesh& mesh);
 
  ~ContactData();
 
  void addContactInteraction(int interaction_id, const std::set<int>& bdry_attr_surf1,
                             const std::set<int>& bdry_attr_surf2, ContactOptions contact_opts);
 
  void updateGeometry(int cycle, double time, double& dt,
                      std::optional<std::reference_wrapper<const mfem::Vector>> u_shape = std::nullopt,
                      std::optional<std::reference_wrapper<const mfem::Vector>> u = std::nullopt,
                      bool eval_jacobian = false);
 
  void updateForcesAndJacobian(int cycle, double time, double& dt,
                               std::optional<std::reference_wrapper<const mfem::Vector>> u_shape = std::nullopt,
                               std::optional<std::reference_wrapper<const mfem::Vector>> u = std::nullopt,
                               std::optional<std::reference_wrapper<const mfem::Vector>> p = std::nullopt);
 
  void reset();
 
  FiniteElementDual forces() const;
 
  mfem::HypreParVector mergedPressures() const;
 
  mfem::HypreParVector mergedGaps(bool zero_inactive = false) const;
 
  std::unique_ptr<mfem::BlockOperator> mergedJacobian() const;
 
  void residualFunction(const mfem::Vector& u_shape, const mfem::Vector& u, mfem::Vector& r);
 
  std::unique_ptr<mfem::BlockOperator> jacobianFunction(std::unique_ptr<mfem::HypreParMatrix> orig_J) const;
 
  std::unique_ptr<mfem::HypreParMatrix> contactSubspaceTransferOperator();
 
  bool haveContactInteractions() const
  {
#ifdef SMITH_USE_TRIBOL
    return !interactions_.empty();
#else
    return false;
#endif
  }
 
#ifdef SMITH_USE_TRIBOL
  const std::vector<ContactInteraction>& getContactInteractions() const { return interactions_; }
#endif
 
  bool haveLagrangeMultipliers() const { return have_lagrange_multipliers_; }
 
  int numPressureDofs() const { return num_pressure_dofs_; };
 
 protected:
  void setPressures(const mfem::Vector& merged_pressures) const;
 
  void setDisplacements(const mfem::Vector& u_shape, const mfem::Vector& u);
 
 private:
#ifdef SMITH_USE_TRIBOL
  void updateDofOffsets() const;
 
  const mfem::ParMesh& mesh_;
#endif
 
  const mfem::ParGridFunction* reference_nodes_;
 
#ifdef SMITH_USE_TRIBOL
  mfem::ParGridFunction current_coords_;
 
  std::vector<ContactInteraction> interactions_;
#endif
 
  bool have_lagrange_multipliers_;
 
  int num_pressure_dofs_;
 
  mutable bool offsets_up_to_date_;
 
  mutable mfem::Array<int> jacobian_offsets_;
 
  mutable mfem::Array<int> pressure_dof_offsets_;
 
  mutable mfem::Array<HYPRE_BigInt> global_pressure_dof_offsets_;
 
  mfem::Array<int> contact_dofs_;
 
  int cycle_{0};
  double time_{0.0};
  double dt_{1.0};
};
 
}  // namespace smith