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 "ADNodalKernel.h"
      11             : #include "SubProblem.h"
      12             : #include "SystemBase.h"
      13             : #include "MooseVariableFE.h"
      14             : #include "Assembly.h"
      15             : 
      16             : #include "metaphysicl/raw_type.h"
      17             : 
      18             : InputParameters
      19       57285 : ADNodalKernel::validParams()
      20             : {
      21       57285 :   auto params = NodalKernelBase::validParams();
      22       57285 :   params += ADFunctorInterface::validParams();
      23       57285 :   return params;
      24           0 : }
      25             : 
      26         120 : ADNodalKernel::ADNodalKernel(const InputParameters & parameters)
      27         120 :   : NodalKernelBase(parameters), ADFunctorInterface(this), _u(_var.adDofValues())
      28             : {
      29         120 :   if (getParam<std::vector<AuxVariableName>>("save_in").size())
      30           0 :     paramError("save_in",
      31             :                "ADNodalKernels do not support save_in. Please use the tagging system instead.");
      32         120 :   if (getParam<std::vector<AuxVariableName>>("diag_save_in").size())
      33           0 :     paramError(
      34             :         "diag_save_in",
      35             :         "ADNodalKernels do not support diag_save_in. Please use the tagging system instead.");
      36         120 : }
      37             : 
      38             : void
      39      235491 : ADNodalKernel::computeResidual()
      40             : {
      41      235491 :   if (_var.isNodalDefined())
      42             :   {
      43      235491 :     const auto dof_idx = _var.nodalDofIndex();
      44      235491 :     _qp = 0;
      45      235491 :     auto res = MetaPhysicL::raw_value(computeQpResidual());
      46      470982 :     addResiduals(_assembly,
      47           0 :                  std::array<Real, 1>{{res}},
      48      235491 :                  std::array<dof_id_type, 1>{{dof_idx}},
      49      235491 :                  _var.scalingFactor());
      50             :   }
      51      235491 : }
      52             : 
      53             : void
      54      115319 : ADNodalKernel::computeJacobian()
      55             : {
      56      115319 :   if (_var.isNodalDefined())
      57             :   {
      58      115319 :     const auto dof_idx = _var.nodalDofIndex();
      59      115319 :     _qp = 0;
      60      115319 :     const auto res = computeQpResidual();
      61      230638 :     addJacobian(_assembly,
      62      115319 :                 std::array<ADReal, 1>{{res}},
      63           0 :                 std::array<dof_id_type, 1>{{dof_idx}},
      64      115319 :                 _var.scalingFactor());
      65      115319 :   }
      66      230638 : }
      67             : 
      68             : void
      69      207431 : ADNodalKernel::computeOffDiagJacobian(const unsigned int jvar)
      70             : {
      71      207431 :   if (jvar == _var.number())
      72      115319 :     computeJacobian();
      73      207431 : }