variational_smoother_system.h

Go to the documentation of this file.

// The libMesh Finite Element Library.
// Copyright (C) 2002-2026 Benjamin S. Kirk, John W. Peterson, Roy H. Stogner
 
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
 
// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// Lesser General Public License for more details.
 
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
#ifndef LIBMESH_VARIATIONAL_SMOOTHER_SYSTEM_H
#define LIBMESH_VARIATIONAL_SMOOTHER_SYSTEM_H
 
// libMesh includes
#include "libmesh/enum_fe_family.h"
#include "libmesh/fem_function_base.h"
#include "libmesh/fem_system.h"
#include "libmesh/libmesh_common.h"
 
// C++ includes
#include <map>
#include <memory>
 
namespace libMesh
{
 
struct MeshQualityInfo
{
  // dof_id_types will hold the id of the Elem where the metric occurs
  // IMPORTANT: the Real should be the first entry of the pair so that taking
  // the min/max accross processors will compare the numeric values instead of
  // the element ids.
 
  std::pair<Real, dof_id_type> max_elem_distortion{std::numeric_limits<Real>::lowest(),
                                                   DofObject::invalid_id};
  std::pair<Real, dof_id_type> min_elem_distortion{std::numeric_limits<Real>::max(),
                                                   DofObject::invalid_id};
  Real total_distortion = 0.;
 
  std::pair<Real, dof_id_type> max_elem_dilation{std::numeric_limits<Real>::lowest(),
                                                 DofObject::invalid_id};
  std::pair<Real, dof_id_type> min_elem_dilation{std::numeric_limits<Real>::max(),
                                                 DofObject::invalid_id};
  Real total_dilation = 0.;
 
  std::pair<Real, dof_id_type> max_elem_combined{std::numeric_limits<Real>::lowest(),
                                                 DofObject::invalid_id};
  std::pair<Real, dof_id_type> min_elem_combined{std::numeric_limits<Real>::max(),
                                                 DofObject::invalid_id};
  Real total_combined = 0.;
 
  std::pair<Real, dof_id_type> max_elem_det_S{std::numeric_limits<Real>::lowest(),
                                              DofObject::invalid_id};
  std::pair<Real, dof_id_type> min_elem_det_S{std::numeric_limits<Real>::max(),
                                              DofObject::invalid_id};
  Real total_det_S = 0.;
  Real max_qp_det_S = std::numeric_limits<Real>::lowest();
  Real min_qp_det_S = std::numeric_limits<Real>::max();
 
  bool mesh_is_tangled = false;
  bool initialized = false;
};
 
// FEMSystem, TimeSolver and  NewtonSolver will handle most tasks,
// but we must specify element residuals
class VariationalSmootherSystem : public libMesh::FEMSystem
{
public:
  VariationalSmootherSystem(libMesh::EquationSystems & es,
                            const std::string & name,
                            const unsigned int number)
    : libMesh::FEMSystem(es, name, number),
      _verbosity(0),
      _epsilon_squared(TOLERANCE),
      _epsilon_squared_assembly(0.),
      _ref_vol(0.),
      _dilation_weight(0.5),
      _untangling_solve(false)
  {}
 
  // Default destructor
  ~VariationalSmootherSystem() override;
 
  virtual void assembly (bool get_residual,
                         bool get_jacobian,
                         bool apply_heterogeneous_constraints = false,
                         bool apply_no_constraints = false) override;
 
  Real & get_dilation_weight() { return _dilation_weight; }
 
  virtual void solve() override;
 
  static std::pair<std::unique_ptr<Elem>, std::vector<std::unique_ptr<Node>>>
  get_target_elem(const ElemType & type);
 
  static void get_target_to_reference_jacobian(const Elem * const target_elem,
                                               const FEMContext & femcontext,
                                               std::vector<RealTensor> & jacobians,
                                               std::vector<Real> & jacobian_dets);
 
  const MeshQualityInfo & get_mesh_info();
 
  /*
   * Computes information about the mesh quality and sets the _mesh_info attribute.
   */
  void compute_mesh_quality_info();
 
  /*
   * Sets the verbosity of the object.
   */
  void set_verbosity(const unsigned int verbosity) { _verbosity = verbosity; }
 
protected:
 
  // System initialization
  virtual void init_data () override;
 
  // Context initialization
  virtual void init_context (libMesh::DiffContext & context) override;
 
  // Element residual and jacobian calculations
  // Time dependent parts
  virtual bool element_time_derivative (bool request_jacobian,
                                        libMesh::DiffContext & context) override;
 
  /* Computes the element reference volume used in the dilation metric
   * The reference value is set to the averaged value of all elements' average
   * |J|. Also computes any applicable target element inverse Jacobians. Target
   * elements are relavant when the reference element does not minimize the
   * distortion metric.
   */
  void prepare_for_smoothing();
 
  unsigned int _verbosity;
 
  const Real _epsilon_squared;
 
  Real _epsilon_squared_assembly;
 
  Real _ref_vol;
 
  Real _dilation_weight;
 
  /* Map to hold target qp-dependent element target-to-reference mapping
   * Jacobians, if any
   */
  std::map<ElemType, std::vector<RealTensor>> _target_jacobians;
 
  /*
   * Map to hold the determinants of _target_jacobians.
   */
  std::map<ElemType, std::vector<Real>> _target_jacobian_dets;
 
  MeshQualityInfo _mesh_info;
 
  bool _untangling_solve;
};
 
} // namespace libMesh
 
#endif // LIBMESH_VARIATIONAL_SMOOTHER_SYSTEM_H