ADNodalKernel.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 "ADNodalKernel.h"
#include "SubProblem.h"
#include "SystemBase.h"
#include "MooseVariableFE.h"
#include "Assembly.h"

#include "metaphysicl/raw_type.h"

InputParameters
ADNodalKernel::validParams()
{
  auto params = NodalKernelBase::validParams();
  params += ADFunctorInterface::validParams();
  return params;
}

ADNodalKernel::ADNodalKernel(const InputParameters & parameters)
  : NodalKernelBase(parameters),
    ADFunctorInterface(this),
    MooseVariableInterface<Real>(this,
                                 true,
                                 "variable",
                                 Moose::VarKindType::VAR_SOLVER,
                                 Moose::VarFieldType::VAR_FIELD_STANDARD),
    _var(*mooseVariable()),
    _u(_var.adDofValues())
{
  addMooseVariableDependency(mooseVariable());
}

void
ADNodalKernel::computeResidual()
{
  if (_var.isNodalDefined())
  {
    const auto dof_idx = _var.nodalDofIndex();
    _qp = 0;
    auto res = MetaPhysicL::raw_value(computeQpResidual());
    addResiduals(_assembly,
                 std::array<Real, 1>{{res}},
                 std::array<dof_id_type, 1>{{dof_idx}},
                 _var.scalingFactor());
  }
}

void
ADNodalKernel::computeJacobian()
{
  if (_var.isNodalDefined())
  {
    const auto dof_idx = _var.nodalDofIndex();
    _qp = 0;
    const auto res = computeQpResidual();
    addJacobian(_assembly,
                std::array<ADReal, 1>{{res}},
                std::array<dof_id_type, 1>{{dof_idx}},
                _var.scalingFactor());
  }
}

void
ADNodalKernel::computeOffDiagJacobian(const unsigned int jvar)
{
  if (jvar == _var.number())
    computeJacobian();
}