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       57271 : ADNodalKernel::validParams()
      20             : {
      21       57271 :   auto params = NodalKernelBase::validParams();
      22       57271 :   params += ADFunctorInterface::validParams();
      23       57271 :   return params;
      24           0 : }
      25             : 
      26         113 : ADNodalKernel::ADNodalKernel(const InputParameters & parameters)
      27         113 :   : NodalKernelBase(parameters), ADFunctorInterface(this), _u(_var.adDofValues())
      28             : {
      29         113 :   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         113 :   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         113 : }
      37             : 
      38             : void
      39      207288 : ADNodalKernel::computeResidual()
      40             : {
      41      207288 :   if (_var.isNodalDefined())
      42             :   {
      43      207288 :     const auto dof_idx = _var.nodalDofIndex();
      44      207288 :     _qp = 0;
      45      207288 :     auto res = MetaPhysicL::raw_value(computeQpResidual());
      46      414576 :     addResiduals(_assembly,
      47           0 :                  std::array<Real, 1>{{res}},
      48      207288 :                  std::array<dof_id_type, 1>{{dof_idx}},
      49      207288 :                  _var.scalingFactor());
      50             :   }
      51      207288 : }
      52             : 
      53             : void
      54      101295 : ADNodalKernel::computeJacobian()
      55             : {
      56      101295 :   if (_var.isNodalDefined())
      57             :   {
      58      101295 :     const auto dof_idx = _var.nodalDofIndex();
      59      101295 :     _qp = 0;
      60      101295 :     const auto res = computeQpResidual();
      61      202590 :     addJacobian(_assembly,
      62      101295 :                 std::array<ADReal, 1>{{res}},
      63           0 :                 std::array<dof_id_type, 1>{{dof_idx}},
      64      101295 :                 _var.scalingFactor());
      65      101295 :   }
      66      202590 : }
      67             : 
      68             : void
      69      182095 : ADNodalKernel::computeOffDiagJacobian(const unsigned int jvar)
      70             : {
      71      182095 :   if (jvar == _var.number())
      72      101295 :     computeJacobian();
      73      182095 : }