www.mooseframework.org
ExplicitTVDRK2.C
Go to the documentation of this file.
1 //* This file is part of the MOOSE framework
2 //* https://www.mooseframework.org
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 "ExplicitTVDRK2.h"
11 #include "NonlinearSystemBase.h"
12 #include "FEProblem.h"
13 #include "PetscSupport.h"
14 
16 
17 template <>
20 {
22 
23  return params;
24 }
25 
27  : TimeIntegrator(parameters),
28  _stage(1),
29  _residual_old(_nl.addVector("residual_old", false, GHOSTED))
30 {
31  mooseInfo("ExplicitTVDRK2 and other multistage TimeIntegrators are known not to work with "
32  "Materials/AuxKernels that accumulate 'state' and should be used with caution.");
33 }
34 
35 void
37 {
38  if (_dt == _dt_old)
40  else
42 }
43 
44 void
46 {
47  // Since advanceState() is called in between stages 2 and 3, this
48  // changes the meaning of "_solution_old". In the second stage,
49  // "_solution_older" is actually the original _solution_old.
50  if (!_sys.solutionUDot())
51  mooseError("ExplicitTVDRK2: Time derivative of solution (`u_dot`) is not stored. Please set "
52  "uDotRequested() to true in FEProblemBase befor requesting `u_dot`.");
53 
54  NumericVector<Number> & u_dot = *_sys.solutionUDot();
55  u_dot = *_solution;
57 
58  _du_dot_du = 1. / _dt;
59  u_dot.close();
60 }
61 
62 void
63 ExplicitTVDRK2::computeADTimeDerivatives(DualReal & ad_u_dot, const dof_id_type & dof) const
64 {
66 }
67 
68 void
70 {
71  Real time_new = _fe_problem.time();
72  Real time_old = _fe_problem.timeOld();
73  Real time_stage2 = time_old + _dt;
74 
75  // Reset numbers of iterations
78 
79  // There is no work to do for the first stage (Y_1 = y_n). The
80  // first solve therefore happens in the second stage. Note that the
81  // non-time Kernels (which should be marked implicit=false) are
82  // evaluated at the old solution during this stage.
84  _console << "1st solve\n";
85  _stage = 2;
86  _fe_problem.timeOld() = time_old;
87  _fe_problem.time() = time_stage2;
91 
92  // Abort time step immediately on stage failure - see TimeIntegrator doc page
93  if (!_fe_problem.converged())
94  return;
95 
96  // Advance solutions old->older, current->old. Also moves Material
97  // properties and other associated state forward in time.
99 
100  // The "update" stage (which we call stage 3) requires an additional
101  // solve with the mass matrix.
103  _console << "2nd solve\n";
104  _stage = 3;
105  _fe_problem.timeOld() = time_stage2;
106  _fe_problem.time() = time_new;
110 
111  // Reset time at beginning of step to its original value
112  _fe_problem.timeOld() = time_old;
113 }
114 
115 void
116 ExplicitTVDRK2::postResidual(NumericVector<Number> & residual)
117 {
118  if (_stage == 1)
119  {
120  // If postResidual() is called before solve(), _stage==1 and we don't
121  // need to do anything.
122  }
123  else if (_stage == 2)
124  {
125  // In the standard RK notation, the stage 2 residual is given by:
126  //
127  // R := M*(Y_2 - y_n)/dt - f(t_n, Y_1) = 0
128  //
129  // where:
130  // .) M is the mass matrix.
131  // .) f(t_n, Y_1) is the residual we are currently computing,
132  // since this method is intended to be used with "implicit=false"
133  // kernels.
134  // .) M*(Y_2 - y_n)/dt corresponds to the residual of the time kernels.
135  // .) The minus signs are "baked in" to the non-time residuals, so
136  // they do not appear here.
137  // .) The current non-time residual is saved for the next stage.
139  _residual_old.close();
140 
141  residual.add(1.0, _Re_time);
142  residual.add(1.0, _residual_old);
143  residual.close();
144  }
145  else if (_stage == 3)
146  {
147  // In the standard RK notation, the update step residual is given by:
148  //
149  // R := M*(2*y_{n+1} - Y_2 - y_n)/(2*dt) - (1/2)*f(t_n+dt/2, Y_2) = 0
150  //
151  // where:
152  // .) M is the mass matrix.
153  // .) f(t_n+dt/2, Y_2) is the residual from stage 2.
154  // .) The minus sign is already baked in to the non-time
155  // residuals, so it does not appear here.
156  // .) Although this is an update step, we have to do a "solve"
157  // using the mass matrix.
158  residual.add(1.0, _Re_time);
159  residual.add(0.5, _Re_non_time);
160  residual.close();
161  }
162  else
163  mooseError(
164  "ExplicitTVDRK2::postResidual(): _stage = ", _stage, ", only _stage = 1-3 is allowed.");
165 }
virtual void computeTimeDerivatives() override
void computeTimeDerivativeHelper(T &u_dot, const T2 &u_old, const T3 &u_older) const
Helper function that actually does the math for computing the time derivative.
virtual NumericVector< Number > * solutionUDot()=0
void initPetscOutput()
Reinitialize petsc output for proper linear/nonlinear iteration display.
virtual Real & time() const
NonlinearSystemBase & getNonlinearSystemBase()
FEProblemBase & _fe_problem
unsigned int _stage
NumericVector< Number > & _Re_non_time
residual vector for non-time contributions
SystemBase & _sys
The main MOOSE class responsible for handling user-defined parameters in almost every MOOSE system...
DualNumber< Real, NumberArray< AD_MAX_DOFS_PER_ELEM, Real > > DualReal
Definition: DualReal.h:29
virtual void preSolve() override
void mooseError(Args &&... args) const
Definition: MooseObject.h:147
virtual void advanceState()
Advance all of the state holding vectors / datastructures so that we can move to the next timestep...
virtual bool converged() override
NumericVector< Number > & _residual_old
Buffer to store non-time residual from the first stage.
void computeADTimeDerivatives(DualReal &ad_u_dot, const dof_id_type &dof) const override
method for computing local automatic differentiation time derivatives
registerMooseObject("MooseApp", ExplicitTVDRK2)
InputParameters validParams< TimeIntegrator >()
unsigned int _n_linear_iterations
Total number of linear iterations over all stages of the time step.
const NumericVector< Number > & _solution_older
Real & _du_dot_du
solution vector for
const NumericVector< Number > *const & _solution
solution vectors
Base class for time integrators.
unsigned int getNumLinearIterationsLastSolve() const
Gets the number of linear iterations in the most recent solve.
void setConstJacobian(bool state)
Set flag that Jacobian is constant (for optimization purposes)
virtual System & system() override
Get the reference to the libMesh system.
ExplicitTVDRK2(const InputParameters &parameters)
NumericVector< Number > & _Re_time
residual vector for time contributions
virtual void postResidual(NumericVector< Number > &residual) override
Callback to the TimeIntegrator called immediately after the residuals are computed in NonlinearSystem...
virtual Real & timeOld() const
const NumericVector< Number > & _solution_old
void mooseInfo(Args &&... args) const
Definition: MooseObject.h:167
InputParameters validParams< ExplicitTVDRK2 >()
unsigned int _n_nonlinear_iterations
Total number of nonlinear iterations over all stages of the time step.
const ConsoleStream _console
An instance of helper class to write streams to the Console objects.
unsigned int getNumNonlinearIterationsLastSolve() const
Gets the number of nonlinear iterations in the most recent solve.
Explicit TVD (total-variation-diminishing) second-order Runge-Kutta time integration methods: ...
virtual void solve() override
Solves the time step and sets the number of nonlinear and linear iterations.