NAG Library Routine Document
D03UAF
1 Purpose
D03UAF performs at each call one iteration of the Strongly Implicit Procedure. It is used to calculate on successive calls a sequence of approximate corrections to the current estimate of the solution when solving a system of simultaneous algebraic equations for which the iterative update matrix is of five-point molecule form on a two-dimensional topologically-rectangular mesh. (‘Topological’ means that a polar grid , for example, can be used as it is equivalent to a rectangular box.)
2 Specification
SUBROUTINE D03UAF ( |
N1, N2, LDA, A, B, C, D, E, APARAM, IT, R, WRKSP1, WRKSP2, IFAIL) |
INTEGER |
N1, N2, LDA, IT, IFAIL |
REAL (KIND=nag_wp) |
A(LDA,N2), B(LDA,N2), C(LDA,N2), D(LDA,N2), E(LDA,N2), APARAM, R(LDA,N2), WRKSP1(LDA,N2), WRKSP2(LDA,N2) |
|
3 Description
Given a set of simultaneous equations
(which could be nonlinear) derived, for example, from a finite difference representation of a two-dimensional elliptic partial differential equation and its boundary conditions, the solution
may be obtained iteratively from a starting approximation
by the formulae
Thus
is the residual of the
th approximate solution
, and
is the update change vector.
D03UAF determines the approximate change vector
corresponding to a given residual
, i.e., it determines an approximate solution to a set of equations
where
is a square
by
matrix and
is a known vector of length
. The set of equations
(2) must be of five-diagonal form
for
and
, provided that
. Indeed, if
, then the equation is assumed to be
For example, if
and
, the equations take the form
The calling program supplies the current residual
at each iteration and the coefficients of the five-point molecule system of equations on which the update procedure is based. The routine performs one iteration, using the approximate
factorization of the Strongly Implicit Procedure with the necessary acceleration parameter adjustment, to calculate the approximate solution
of the set of equations
(2). The change
overwrites the residual array for return to the calling program. The calling program must combine this change stored in
with the old approximation to obtain the new approximate solution for
. It must then recalculate the residuals and, if the accuracy requirements have not been satisfied, commence the next iterative cycle.
Clearly there is no requirement that the iterative update matrix passed in the form of the five-diagonal element arrays
A,
B,
C,
D and
E is the same as that used to calculate the residuals, and therefore the one governing the problem. However, the convergence may be impaired if they are not equal. Indeed, if the system of equations
(1) is not precisely of the five-diagonal form illustrated above but has a few additional terms, then the methods of deferred or defect correction can be employed. The residual is calculated by the calling program using the full system of equations, but the update formula is based on a five-diagonal system
(2) of the form given above. For example, the solution of a system of nine-diagonal equations each involving the combination of terms with
and
could use the five-diagonal coefficients on which to base the update, provided these incorporate the major features of the equations.
Problems in topologically non-rectangular regions can be solved using the routine by surrounding the region with a circumscribing topological rectangle. The equations for the nodal values external to the region of interest are set to zero (i.e., ) and the boundary conditions are incorporated into the equations for the appropriate nodes.
If there is no better initial approximation when starting the iterative cycle, one can use an array of all zeros as the initial approximation from which the first set of residuals are determined.
The routine can be used to solve linear elliptic equations in which case the arrays
A,
B,
C,
D,
E and the quantities
will be unchanged during the iterative cycles, or for solving nonlinear elliptic equations in which case some or all of these arrays may require updating as each new approximate solution is derived. Depending on the nonlinearity, some under-relaxation of the coefficients and/or source terms may be needed during their recalculation using the new estimates of the solution (see
Jacobs (1972)).
The routine can also be used to solve each step of a time-dependent parabolic equation in two space dimensions. The solution at each time step can be expressed in terms of an elliptic equation if the Crank–Nicolson or other form of implicit time integration is used.
Neither diagonal dominance, nor positive-definiteness, of the matrix
or of the update matrix formed from the arrays
A,
B,
C,
D and
E is necessary to ensure convergence.
For problems in which the solution is not unique, in the sense that an arbitrary constant can be added to the solution (for example Laplace's equation with all Neumann boundary conditions), the calling program should subtract a typical nodal value from the whole solution at every iteration to keep rounding errors to a minimum.
4 References
Ames W F (1977) Nonlinear Partial Differential Equations in Engineering (2nd Edition) Academic Press
Jacobs D A H (1972) The strongly implicit procedure for the numerical solution of parabolic and elliptic partial differential equations Note RD/L/N66/72 Central Electricity Research Laboratory
Stone H L (1968) Iterative solution of implicit approximations of multi-dimensional partial differential equations SIAM J. Numer. Anal. 5 530–558
5 Parameters
- 1: – INTEGERInput
-
On entry: the number of nodes in the first coordinate direction, .
Constraint:
.
- 2: – INTEGERInput
-
On entry: the number of nodes in the second coordinate direction, .
Constraint:
.
- 3: – INTEGERInput
-
On entry: the first dimension of the arrays
A,
B,
C,
D,
E,
R,
WRKSP1 and
WRKSP2 as declared in the (sub)program from which D03UAF is called.
Constraint:
.
- 4: – REAL (KIND=nag_wp) arrayInput
-
On entry:
must contain the coefficient of the ‘southerly’ term involving
in the
th equation of the system
(2), for
and
. The elements of
A, for
, must be zero after incorporating the boundary conditions, since they involve nodal values from outside the rectangle.
- 5: – REAL (KIND=nag_wp) arrayInput
-
On entry:
must contain the coefficient of the ‘westerly’ term involving
in the
th equation of the system
(2), for
and
. The elements of
B, for
, must be zero after incorporating the boundary conditions, since they involve nodal values from outside the rectangle.
- 6: – REAL (KIND=nag_wp) arrayInput
-
On entry:
must contain the coefficient of the ‘central’ term involving
in the
th equation of the system
(2), for
and
. The elements of
C are checked to ensure that they are nonzero. If any element is found to be zero, the corresponding algebraic equation is assumed to be
. This feature can be used to define the equations for nodes at which, for example, Dirichlet boundary conditions are applied, or for nodes external to the problem of interest, by setting
at appropriate points. The corresponding value of
is set equal to the appropriate value, namely the difference between the prescribed value of
and the current value of
in the Dirichlet case, or zero at an external point.
- 7: – REAL (KIND=nag_wp) arrayInput
-
On entry:
must contain the coefficient of the ‘easterly’ term involving
in the
th equation of the system
(2), for
and
. The elements of
D, for
, must be zero after incorporating the boundary conditions, since they involve nodal values from outside the rectangle.
- 8: – REAL (KIND=nag_wp) arrayInput
-
On entry:
must contain the coefficient of the ‘northerly’ term involving
in the
th equation of the system
(2), for
and
. The elements of
E, for
, must be zero after incorporating the boundary conditions, since they involve nodal values from outside the rectangle.
- 9: – REAL (KIND=nag_wp)Input
-
On entry: the iteration acceleration factor. A value of is adequate for most typical problems. However, if convergence is slow, the value can be reduced, typically to or . If divergence is obtained, the value can be increased, typically to , or .
Constraint:
.
- 10: – INTEGERInput
-
On entry: the iteration number. It must be initialized, but not necessarily to
, before the first call, and must be incremented by one in the calling program for each subsequent call. D03UAF uses the counter to select the appropriate acceleration parameter from a sequence of nine, each one being used twice in succession. (Note that the acceleration parameter depends on the value of
APARAM.)
- 11: – REAL (KIND=nag_wp) arrayInput/Output
-
On entry:
must contain the current residual
on the right-hand side of the
th equation of the system
(2), for
and
.
On exit: these residuals are overwritten by the corresponding components of solution
to the system
(2), i.e., the changes to be made to the vector
to reduce the residuals supplied.
- 12: – REAL (KIND=nag_wp) arrayWorkspace
- 13: – REAL (KIND=nag_wp) arrayWorkspace
-
- 14: – INTEGERInput/Output
-
On entry:
IFAIL must be set to
,
. If you are unfamiliar with this parameter you should refer to
Section 3.3 in the Essential Introduction for details.
For environments where it might be inappropriate to halt program execution when an error is detected, the value
is recommended. If the output of error messages is undesirable, then the value
is recommended. Otherwise, if you are not familiar with this parameter, the recommended value is
.
When the value is used it is essential to test the value of IFAIL on exit.
On exit:
unless the routine detects an error or a warning has been flagged (see
Section 6).
6 Error Indicators and Warnings
If on entry
or
, explanatory error messages are output on the current error message unit (as defined by
X04AAF).
Errors or warnings detected by the routine:
-
-
-
-
On entry, | . |
An unexpected error has been triggered by this routine. Please
contact
NAG.
See
Section 3.8 in the Essential Introduction for further information.
Your licence key may have expired or may not have been installed correctly.
See
Section 3.7 in the Essential Introduction for further information.
Dynamic memory allocation failed.
See
Section 3.6 in the Essential Introduction for further information.
7 Accuracy
The improvement in accuracy for each iteration, i.e., on each call, depends on the size of the system and on the condition of the update matrix characterised by the five-diagonal coefficient arrays. The ultimate accuracy obtainable depends on the above factors and on the machine precision. However, since D03UAF works with residuals and the update vector, the calling program can, in most cases where at each iteration all the residuals are usually of about the same size, calculate the residuals from extended precision values of the function, source term and equation coefficients if greater accuracy is required. The rate of convergence obtained with the Strongly Implicit Procedure is not always smooth because of the cyclic use of nine acceleration parameters. The convergence may become slow with very large problems. The final accuracy obtained can be judged approximately from the rate of convergence determined from the changes to the dependent variable and in particular the change on the last iteration.
8 Parallelism and Performance
Not applicable.
The time taken is approximately proportional to for each call.
When used with deferred or defect correction, the residual is calculated in the calling program from a different system of equations to those represented by the five-point molecule coefficients used by D03UAF as the basis of the iterative update procedure. When using deferred correction the overall rate of convergence depends not only on the items detailed in
Section 7 but also on the difference between the two coefficient matrices used.
Convergence may not always be obtained when the problem is very large and/or the coefficients of the equations have widely disparate values. The latter case may be associated with an ill-conditioned matrix.
10 Example
This example solves Laplace's equation in a rectangle with a non-uniform grid spacing in the and coordinate directions and with Dirichlet boundary conditions specifying the function on the perimeter of the rectangle equal to .
10.1 Program Text
Program Text (d03uafe.f90)
10.2 Program Data
Program Data (d03uafe.d)
10.3 Program Results
Program Results (d03uafe.r)