ADNodalBC.C

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://mooseframework.inl.gov
//*
//* All rights reserved, see COPYRIGHT for full restrictions
//* https://github.com/idaholab/moose/blob/master/COPYRIGHT
//*
//* Licensed under LGPL 2.1, please see LICENSE for details
//* https://www.gnu.org/licenses/lgpl-2.1.html

#include "ADNodalBC.h"
#include "NodalBCBase.h"
#include "ADDirichletBCBase.h"

// MOOSE includes
#include "Assembly.h"
#include "SubProblem.h"
#include "SystemBase.h"
#include "MooseVariableFE.h"
#include "MooseVariableScalar.h"
#include "FEProblemBase.h"

#include "libmesh/quadrature.h"

template <typename T, typename Base>
InputParameters
ADNodalBCTempl<T, Base>::validParams()
{
  return Base::validParams();
}

template <>
InputParameters
ADNodalBCTempl<RealVectorValue, NodalBCBase>::validParams()
{
  InputParameters params = NodalBCBase::validParams();
  // The below parameters are useful for vector Nodal BCs
  params.addParam<bool>("set_x_comp", true, "Whether to set the x-component of the variable");
  params.addParam<bool>("set_y_comp", true, "Whether to set the y-component of the variable");
  params.addParam<bool>("set_z_comp", true, "Whether to set the z-component of the variable");
  return params;
}

template <>
InputParameters
ADNodalBCTempl<RealVectorValue, ADDirichletBCBase>::validParams()
{
  InputParameters params = ADDirichletBCBase::validParams();
  // The below parameters are useful for vector Nodal BCs
  params.addParam<bool>("set_x_comp", true, "Whether to set the x-component of the variable");
  params.addParam<bool>("set_y_comp", true, "Whether to set the y-component of the variable");
  params.addParam<bool>("set_z_comp", true, "Whether to set the z-component of the variable");
  return params;
}

template <typename T, typename Base>
ADNodalBCTempl<T, Base>::ADNodalBCTempl(const InputParameters & parameters)
  : Base(parameters),
    MooseVariableInterface<T>(this,
                              true,
                              "variable",
                              Moose::VarKindType::VAR_SOLVER,
                              std::is_same<T, Real>::value ? Moose::VarFieldType::VAR_FIELD_STANDARD
                                                           : Moose::VarFieldType::VAR_FIELD_VECTOR),
    ADFunctorInterface(this),
    _var(*this->mooseVariable()),
    _current_node(_var.node()),
    _u(_var.adNodalValue()),
    _set_components(
        {std::is_same<T, RealVectorValue>::value ? this->template getParam<bool>("set_x_comp")
                                                 : true,
         std::is_same<T, RealVectorValue>::value ? this->template getParam<bool>("set_y_comp")
                                                 : true,
         std::is_same<T, RealVectorValue>::value ? this->template getParam<bool>("set_z_comp")
                                                 : true}),
    _undisplaced_assembly(_fe_problem.assembly(_tid, _sys.number()))
{
  _subproblem.haveADObjects(true);

  addMooseVariableDependency(this->mooseVariable());
}

namespace
{
const ADReal &
conversionHelper(const ADReal & value, const unsigned int)
{
  return value;
}

const ADReal &
conversionHelper(const libMesh::VectorValue<ADReal> & value, const unsigned int i)
{
  return value(i);
}
}

template <typename T, typename Base>
template <typename ADResidual>
void
ADNodalBCTempl<T, Base>::addResidual(const ADResidual & residual,
                                     const std::vector<dof_id_type> & dof_indices)
{
  mooseAssert(dof_indices.size() <= _set_components.size(),
              "The number of dof indices must be less than the number of settable components");

  for (const auto i : index_range(dof_indices))
    if (_set_components[i])
      setResidual(_sys, raw_value(conversionHelper(residual, i)), dof_indices[i]);
}

template <typename T, typename Base>
template <typename ADResidual>
void
ADNodalBCTempl<T, Base>::addJacobian(const ADResidual & residual,
                                     const std::vector<dof_id_type> & dof_indices)
{
  mooseAssert(dof_indices.size() <= _set_components.size(),
              "The number of dof indices must be less than the number of settable components");

  for (const auto i : index_range(dof_indices))
    if (_set_components[i])
      // If we store into the displaced assembly for nodal bc objects the data never actually makes
      // it into the global Jacobian
      addJacobian(_undisplaced_assembly,
                  std::array<ADReal, 1>{{conversionHelper(residual, i)}},
                  std::array<dof_id_type, 1>{{dof_indices[i]}},
                  /*scaling_factor=*/1);
}

template <typename T, typename Base>
void
ADNodalBCTempl<T, Base>::computeResidual()
{
  const std::vector<dof_id_type> & dof_indices = _var.dofIndices();
  if (dof_indices.empty())
    return;

  const auto residual = computeQpResidual();

  addResidual(residual, dof_indices);
}

template <typename T, typename Base>
void
ADNodalBCTempl<T, Base>::computeJacobian()
{
  const std::vector<dof_id_type> & dof_indices = _var.dofIndices();
  if (dof_indices.empty())
    return;

  const auto residual = computeQpResidual();

  addJacobian(residual, dof_indices);
}

template <typename T, typename Base>
void
ADNodalBCTempl<T, Base>::computeResidualAndJacobian()
{
  const std::vector<dof_id_type> & dof_indices = _var.dofIndices();
  if (dof_indices.empty())
    return;

  const auto residual = computeQpResidual();

  addResidual(residual, dof_indices);
  addJacobian(residual, dof_indices);
}

template <typename T, typename Base>
void
ADNodalBCTempl<T, Base>::computeOffDiagJacobian(const unsigned int jvar_num)
{
  // Only need to do this once because AD does all the derivatives at once
  if (jvar_num == _var.number())
    computeJacobian();
}

template <typename T, typename Base>
void
ADNodalBCTempl<T, Base>::computeOffDiagJacobianScalar(unsigned int)
{
  // scalar coupling will have been included in the all-at-once handling in computeOffDiagJacobian
}

template class ADNodalBCTempl<Real, NodalBCBase>;
template class ADNodalBCTempl<RealVectorValue, NodalBCBase>;
template class ADNodalBCTempl<Real, ADDirichletBCBase>;
template class ADNodalBCTempl<RealVectorValue, ADDirichletBCBase>;