Line data    Source code

       1             : //* This file is part of the MOOSE framework
       2             : //* https://mooseframework.inl.gov
       3             : //*
       4             : //* All rights reserved, see COPYRIGHT for full restrictions
       5             : //* https://github.com/idaholab/moose/blob/master/COPYRIGHT
       6             : //*
       7             : //* Licensed under LGPL 2.1, please see LICENSE for details
       8             : //* https://www.gnu.org/licenses/lgpl-2.1.html
       9             : 
      10             : #include "ElementSideNeighborLayers.h"
      11             : #include "MooseMesh.h"
      12             : #include "Conversion.h"
      13             : #include "MooseApp.h"
      14             : #include "Executioner.h"
      15             : #include "FEProblemBase.h"
      16             : #include "NonlinearSystem.h"
      17             : 
      18             : #include "libmesh/default_coupling.h"
      19             : #include "libmesh/point_neighbor_coupling.h"
      20             : #include "libmesh/dof_map.h"
      21             : 
      22             : using namespace libMesh;
      23             : 
      24             : registerMooseObject("MooseApp", ElementSideNeighborLayers);
      25             : 
      26             : InputParameters
      27      756478 : ElementSideNeighborLayers::validParams()
      28             : {
      29      756478 :   InputParameters params = FunctorRelationshipManager::validParams();
      30             : 
      31     3782390 :   params.addRangeCheckedParam<unsigned short>(
      32             :       "layers",
      33     1512956 :       1,
      34             :       "element_side_neighbor_layers>=1 & element_side_neighbor_layers<=10",
      35             :       "The number of additional geometric elements to make available when "
      36             :       "using distributed mesh. No effect with replicated mesh.");
      37     1512956 :   params.addParam<bool>("use_point_neighbors",
      38     1512956 :                         false,
      39             :                         "Whether to use point neighbors, which introduces additional ghosting to "
      40             :                         "that used for simple face neighbors.");
      41             : 
      42      756478 :   return params;
      43           0 : }
      44             : 
      45      340173 : ElementSideNeighborLayers::ElementSideNeighborLayers(const InputParameters & parameters)
      46             :   : FunctorRelationshipManager(parameters),
      47      340169 :     _layers(getParam<unsigned short>("layers")),
      48     1020511 :     _use_point_neighbors(getParam<bool>("use_point_neighbors"))
      49             : {
      50      340169 : }
      51             : 
      52       61871 : ElementSideNeighborLayers::ElementSideNeighborLayers(const ElementSideNeighborLayers & other)
      53             :   : FunctorRelationshipManager(other),
      54       61871 :     _layers(other._layers),
      55       61871 :     _use_point_neighbors(other._use_point_neighbors)
      56             : {
      57       61871 : }
      58             : 
      59             : std::unique_ptr<GhostingFunctor>
      60       61871 : ElementSideNeighborLayers::clone() const
      61             : {
      62       61871 :   return _app.getFactory().copyConstruct(*this);
      63             : }
      64             : 
      65             : std::string
      66         142 : ElementSideNeighborLayers::getInfo() const
      67             : {
      68         142 :   std::ostringstream oss;
      69         142 :   std::string layers = _layers == 1 ? "layer" : "layers";
      70             : 
      71         142 :   oss << "ElementSideNeighborLayers (" << _layers << " " << layers << ')';
      72             : 
      73         284 :   return oss.str();
      74         142 : }
      75             : 
      76             : // the LHS ("this" object) in MooseApp::addRelationshipManager is the existing RelationshipManager
      77             : // object to which we are comparing the rhs to determine whether it should get added
      78             : bool
      79      602797 : ElementSideNeighborLayers::operator>=(const RelationshipManager & rhs) const
      80             : {
      81      602797 :   const auto * rm = dynamic_cast<const ElementSideNeighborLayers *>(&rhs);
      82      602797 :   if (!rm)
      83      286237 :     return false;
      84             :   else
      85             :     // We use a >= comparison instead of == for _layers because if we already have more ghosting
      86             :     // than the new RM provides, then that's an indication that we should *not* add the new one
      87      616831 :     return (_layers >= rm->_layers) && (_use_point_neighbors == rm->_use_point_neighbors) &&
      88      616831 :            baseGreaterEqual(*rm);
      89             : }
      90             : 
      91             : template <typename Functor>
      92             : void
      93       69512 : ElementSideNeighborLayers::initFunctor(Functor & functor)
      94             : {
      95       69512 :   functor.set_n_levels(_layers);
      96             : 
      97       69512 :   if (_dof_map)
      98             :   {
      99             :     // Need to see if there are periodic BCs - if so we need to dig them out
     100       61323 :     auto periodic_boundaries_ptr = _dof_map->get_periodic_boundaries();
     101             : 
     102             :     mooseAssert(periodic_boundaries_ptr, "Periodic Boundaries Pointer is nullptr");
     103             : 
     104       61323 :     functor.set_periodic_boundaries(periodic_boundaries_ptr);
     105       61323 :     functor.set_dof_coupling(_dof_map->_dof_coupling);
     106             :   }
     107       69512 : }
     108             : 
     109             : void
     110       69512 : ElementSideNeighborLayers::internalInitWithMesh(const MeshBase &)
     111             : {
     112       69512 :   if (_use_point_neighbors)
     113             :   {
     114          39 :     auto functor = std::make_unique<PointNeighborCoupling>();
     115          39 :     initFunctor(*functor);
     116          39 :     _functor = std::move(functor);
     117          39 :   }
     118             :   else
     119             :   {
     120       69473 :     auto functor = std::make_unique<DefaultCoupling>();
     121       69473 :     initFunctor(*functor);
     122       69473 :     _functor = std::move(functor);
     123       69473 :   }
     124       69512 : }
     125             : 
     126             : void
     127       87213 : ElementSideNeighborLayers::dofmap_reinit()
     128             : {
     129       87213 :   if (_dof_map)
     130             :   {
     131       87213 :     if (_use_point_neighbors)
     132          75 :       static_cast<PointNeighborCoupling *>(_functor.get())
     133          75 :           ->set_dof_coupling(_dof_map->_dof_coupling);
     134             :     else
     135       87138 :       static_cast<DefaultCoupling *>(_functor.get())->set_dof_coupling(_dof_map->_dof_coupling);
     136             :   }
     137       87213 : }