SingleTraceLineSourceTest.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 "SingleTraceLineSourceTest.h"
#include "NonlinearSystemBase.h"
#include "SystemBase.h"
 
registerMooseObject("RayTracingTestApp", SingleTraceLineSourceTest);
 
InputParameters
SingleTraceLineSourceTest::validParams()
{
  auto params = RepeatableRayStudy::validParams();
  params.addParam<TagName>("residual_vector_tag",
                           "the vector tag for the residual tag you will accumulate into");
  params.set<bool>("allow_other_flags_with_prekernels") = true;
  return params;
}
 
SingleTraceLineSourceTest::SingleTraceLineSourceTest(const InputParameters & parameters)
  : RepeatableRayStudy(parameters),
    _residual_tag_name(getParam<TagName>("residual_vector_tag")),
    _has_traced(false)
{
}
 
void
SingleTraceLineSourceTest::execute()
{
  if (_current_execute_flag == EXEC_TIMESTEP_BEGIN)
  {
    _has_traced = false;
    return;
  }
 
  if (_current_execute_flag == EXEC_NONLINEAR)
  {
    mooseAssert(_fe_problem.currentlyComputingJacobian(),
                "Should be computing jacobian but is not.");
    return;
  }
 
  const auto contribution_tag_id = _fe_problem.getVectorTagID(_residual_tag_name);
  auto & nl = _fe_problem.getNonlinearSystemBase(_sys.number());
  auto & contribution_vec = nl.getVector(contribution_tag_id);
  if (!_has_traced)
  {
    contribution_vec.zero();
    RayTracingStudy::execute();
    contribution_vec.close();
    _has_traced = true;
  }
 
  auto & residual_vec = nl.getVector(nl.residualVectorTag());
  residual_vec.close();
  residual_vec += contribution_vec;
}