SolidMechanicsAugmentedLagrangianContact.hpp

/*
 * ------------------------------------------------------------------------------------------------------------
 * SPDX-License-Identifier: LGPL-2.1-only
 *
 * Copyright (c) 2016-2024 Lawrence Livermore National Security LLC
 * Copyright (c) 2018-2024 Total, S.A
 * Copyright (c) 2018-2024 The Board of Trustees of the Leland Stanford Junior University
 * Copyright (c) 2023-2024 Chevron
 * Copyright (c) 2019-     GEOS/GEOSX Contributors
 * All rights reserved
 *
 * See top level LICENSE, COPYRIGHT, CONTRIBUTORS, NOTICE, and ACKNOWLEDGEMENTS files for details.
 * ------------------------------------------------------------------------------------------------------------
 */
 
/*
 * SolidMechanicsAugmentedLagrangianContact.hpp
 *
 */
 
#ifndef GEOS_PHYSICSSOLVERS_CONTACT_SOLIDMECHANICSAUGMENTEDLAGRANGIANCONTACT_HPP_
#define GEOS_PHYSICSSOLVERS_CONTACT_SOLIDMECHANICSAUGMENTEDLAGRANGIANCONTACT_HPP_
 
#include "physicsSolvers/contact/ContactSolverBase.hpp"
 
namespace geos
{
 
class SolidMechanicsAugmentedLagrangianContact : public ContactSolverBase
{
public:
  SolidMechanicsAugmentedLagrangianContact( const string & name,
                                            Group * const parent );
 
  ~SolidMechanicsAugmentedLagrangianContact() override;
 
  static string catalogName()
  {
    return "SolidMechanicsAugmentedLagrangianContact";
  }
  string getCatalogName() const override { return catalogName(); }
 
  virtual void registerDataOnMesh( dataRepository::Group & meshBodies ) override final;
 
  virtual void setupDofs( DomainPartition const & domain,
                          DofManager & dofManager ) const override;
 
  virtual void setupSystem( DomainPartition & domain,
                            DofManager & dofManager,
                            CRSMatrix< real64, globalIndex > & localMatrix,
                            ParallelVector & rhs,
                            ParallelVector & solution,
                            bool const setSparsity = true ) override final;
 
  virtual void implicitStepSetup( real64 const & time_n,
                                  real64 const & dt,
                                  DomainPartition & domain ) override final;
 
  virtual void implicitStepComplete( real64 const & time_n,
                                     real64 const & dt,
                                     DomainPartition & domain ) override final;
 
  virtual void assembleSystem( real64 const time,
                               real64 const dt,
                               DomainPartition & domain,
                               DofManager const & dofManager,
                               CRSMatrixView< real64, globalIndex const > const & localMatrix,
                               arrayView1d< real64 > const & localRhs ) override;
 
  virtual real64 calculateResidualNorm( real64 const & time_n,
                                        real64 const & dt,
                                        DomainPartition const & domain,
                                        DofManager const & dofManager,
                                        arrayView1d< real64 const > const & localRhs ) override;
 
  virtual void applySystemSolution( DofManager const & dofManager,
                                    arrayView1d< real64 const > const & localSolution,
                                    real64 const scalingFactor,
                                    real64 const dt,
                                    DomainPartition & domain ) override;
 
  void updateState( DomainPartition & domain ) override final;
 
  virtual bool updateConfiguration( DomainPartition & domain ) override final;
 
 
  template< typename LAMBDA >
  void forFiniteElementOnFractureSubRegions( string const & meshName, LAMBDA && lambda ) const
  {
 
    std::map< string,
              array1d< localIndex > > const & faceTypesToFaceElements = m_faceTypesToFaceElements.at( meshName );
 
    for( const auto & [finiteElementName, faceElementList] : faceTypesToFaceElements )
    {
      arrayView1d< localIndex const > const faceElemList = faceElementList.toViewConst();
 
      finiteElement::FiniteElementBase const & subRegionFE = *(m_faceTypeToFiniteElements.at( finiteElementName ));
 
      lambda( finiteElementName, subRegionFE, faceElemList );
    }
 
  }
 
  template< typename LAMBDA >
  void forFiniteElementOnStickFractureSubRegions( string const & meshName, LAMBDA && lambda ) const
  {
 
    bool const isStickState = true;
 
    std::map< string, array1d< localIndex > > const &
    faceTypesToFaceElements = m_faceTypesToFaceElementsStick.at( meshName );
 
    for( const auto & [finiteElementName, faceElementList] : faceTypesToFaceElements )
    {
      arrayView1d< localIndex const > const faceElemList = faceElementList.toViewConst();
 
      finiteElement::FiniteElementBase const & subRegionFE = *(m_faceTypeToFiniteElements.at( finiteElementName ));
 
      lambda( finiteElementName, subRegionFE, faceElemList, isStickState );
    }
 
  }
 
  template< typename LAMBDA >
  void forFiniteElementOnSlipFractureSubRegions( string const & meshName, LAMBDA && lambda ) const
  {
 
    bool const isStickState = false;
 
    std::map< string, array1d< localIndex > > const &
    faceTypesToFaceElements = m_faceTypesToFaceElementsSlip.at( meshName );
 
    for( const auto & [finiteElementName, faceElementList] : faceTypesToFaceElements )
    {
      arrayView1d< localIndex const > const faceElemList = faceElementList.toViewConst();
 
      finiteElement::FiniteElementBase const & subRegionFE = *(m_faceTypeToFiniteElements.at( finiteElementName ));
 
      lambda( finiteElementName, subRegionFE, faceElemList, isStickState );
    }
 
  }
 
  void updateStickSlipList( DomainPartition const & domain );
 
  void createFaceTypeList( DomainPartition const & domain );
 
  void createBubbleCellList( DomainPartition & domain ) const;
 
private:
 
  void addCouplingNumNonzeros( DomainPartition & domain,
                               DofManager & dofManager,
                               arrayView1d< localIndex > const & rowLengths ) const;
 
  void addCouplingSparsityPattern( DomainPartition const & domain,
                                   DofManager const & dofManager,
                                   SparsityPatternView< globalIndex > const & pattern ) const;
 
  void computeTolerances( DomainPartition & domain ) const;
 
  std::map< string, std::map< string, array1d< localIndex > > > m_faceTypesToFaceElements;
 
  std::map< string, std::map< string, array1d< localIndex > > > m_faceTypesToFaceElementsStick;
 
  std::map< string, std::map< string, array1d< localIndex > > > m_faceTypesToFaceElementsSlip;
 
  std::map< string, std::unique_ptr< geos::finiteElement::FiniteElementBase > > m_faceTypeToFiniteElements;
 
  struct viewKeyStruct : ContactSolverBase::viewKeyStruct
  {
 
    constexpr static char const * normalDisplacementToleranceString() { return "normalDisplacementTolerance"; }
 
    constexpr static char const * normalTractionToleranceString() { return "normalTractionTolerance"; }
 
    constexpr static char const * slidingToleranceString() { return "slidingTolerance"; }
 
    constexpr static char const * dispJumpUpdPenaltyString() { return "dispJumpUpdPenalty"; }
  };
 
  real64 const m_slidingCheckTolerance = 0.05;
 
  bool m_simultaneous = true;
 
  bool m_symmetric = true;
 
};
 
} /* namespace geos */
 
#endif /* GEOS_PHYSICSSOLVERS_CONTACT_SOLIDMECHANICSAUGMENTEDLAGRANGIANCONTACT_HPP_ */