ADNodalBC.h

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://www.mooseframework.org
//*
//* 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

#pragma once

#include "MooseVariableInterface.h"
#include "NodalBCBase.h"
#include "ADDirichletBCBase.h"
#include "ADFunctorInterface.h"

template <typename T, typename Base>
class ADNodalBCTempl : public Base, public MooseVariableInterface<T>, public ADFunctorInterface
{
public:
  static InputParameters validParams();

  ADNodalBCTempl(const InputParameters & parameters);

  const MooseVariableFE<T> & variable() const override { return _var; }

  bool shouldSetComp(unsigned short i) const { return _set_components[i]; }

protected:
  virtual typename Moose::ADType<T>::type computeQpResidual() = 0;

  MooseVariableFE<T> & _var;

  const Node * const & _current_node;

  const unsigned int _qp = 0;

  const typename Moose::ADType<T>::type & _u;

  const std::array<bool, 3> _set_components;

  using Base::_fe_problem;
  using Base::_subproblem;
  using Base::_sys;
  using Base::_tid;
  using Base::addJacobian;
  using Base::addMooseVariableDependency;
  using Base::setResidual;

private:
  void computeResidual() override final;
  void computeJacobian() override final;
  void computeResidualAndJacobian() override;
  void computeOffDiagJacobian(unsigned int jvar) override final;
  void computeOffDiagJacobianScalar(unsigned int jvar) override final;

  template <typename ADResidual>
  void addResidual(const ADResidual & residual, const std::vector<dof_id_type> & dof_indices);

  template <typename ADResidual>
  void addJacobian(const ADResidual & residual, const std::vector<dof_id_type> & dof_indices);

  Assembly & _undisplaced_assembly;
};

template <>
InputParameters ADNodalBCTempl<RealVectorValue, NodalBCBase>::validParams();

template <>
InputParameters ADNodalBCTempl<RealVectorValue, ADDirichletBCBase>::validParams();

using ADNodalBC = ADNodalBCTempl<Real, NodalBCBase>;
using ADVectorNodalBC = ADNodalBCTempl<RealVectorValue, NodalBCBase>;