|
PROGRAM TEST
|
IMPLICIT NONE
|
C
|
INTEGER LDA, LDAF, LDB, LDIWRK, LDWORK, LDX
|
INTEGER N, NRHS
|
PARAMETER (N = 4)
|
PARAMETER (LDA = N)
|
PARAMETER (LDAF = N)
|
PARAMETER (LDB = N)
|
PARAMETER (LDIWRK = N)
|
PARAMETER (LDWORK = 3 * N)
|
PARAMETER (LDX = LDB)
|
PARAMETER (NRHS = 1)
|
C
|
DOUBLE PRECISION A(LDA,N), AF(LDAF,N), B(LDB,NRHS)
|
DOUBLE PRECISION B1(LDB,NRHS), BERR(NRHS), FERR(NRHS)
|
DOUBLE PRECISION WORK(LDWORK), X(LDX,NRHS)
|
INTEGER ICOL, INFO, IROW, IWORK(LDIWRK)
|
C
|
EXTERNAL DCOPY, DPORFS, DPOTRF, DPOTRS, DSYMV
|
INTRINSIC ABS
|
C
|
C Initialize the array A to store in symmetric form the
|
C coefficient matrix A shown below. Initialize the array B
|
C to store the right hand side vector b shown below.
|
C
|
C 10**5 10**-5 10**-6 10**-7 10**4
|
C A = 10**-5 10**5 10**-6 10**-7 b = 10**4
|
C 10**-6 10**-6 10**5 10**-7 10**4
|
C 10**-7 10**-7 10**-7 10**5 10**4
|
C
|
DATA A / 1.0D5, 1.0D-5, 1.0D-6, 1.0D-7,
|
$ 8D180, 1.0D5, 1.0D-6, 1.0D-7,
|
$ 8D180, 8D180, 1.0D5, 1.0D-7,
|
$ 8D180, 8D180, 8D180, 1.0D5 /
|
DATA B / 1.0D4, 1.0D4, 1.0D4, 1.0D4 /
|
C
|
C Print the initial value of the arrays.
|
C
|
PRINT 1000
|
DO 100, IROW = 1, N
|
PRINT 1010, (A(IROW,ICOL), ICOL = 1, IROW),
|
$ (A(ICOL,IROW), ICOL = IROW + 1, N)
|
100 CONTINUE
|
PRINT 1020
|
DO 110, IROW = 1, LDA
|
PRINT 1010, (A(IROW,ICOL), ICOL = 1, N)
|
110 CONTINUE
|
CALL DCOPY (LDA*N, A, 1, AF, 1)
|
PRINT 1030
|
PRINT 1040, B
|
CALL DCOPY (LDB*NRHS, B, 1, X, 1)
|
C
|
C Cholesky factor A.
|
C
|
CALL DPOTRF ('LOWER TRIANGLE OF A STORED', N, AF, LDAF,
|
$ INFO)
|
IF (INFO .NE. 0) THEN
|
PRINT 1050, INFO
|
STOP 1
|
END IF
|
C
|
C Solve Ax=b and print the solution.
|
C
|
CALL DPOTRS ('LOWER TRIANGLE OF A STORED', N, NRHS, AF,
|
$ LDAF, X, LDX, INFO)
|
IF (INFO .NE. 0) THEN
|
PRINT 1060, ABS(INFO)
|
STOP 3
|
END IF
|
PRINT 1070
|
PRINT 1040, X
|
CALL DSYMV ('LOWER TRIANGLE OF A STORED', N, 1.0D0, A, LDA,
|
$ X, 1, 0.0D0, B1, 1)
|
PRINT 1080
|
PRINT 1040, B1
|
C
|
C Refine the solution to Ax=b and print the refined solution.
|
C
|
CALL DPORFS ('LOWER TRIANGLE OF A STORED', N, NRHS, A, LDA,
|
$ AF, LDAF, B, LDB, X, LDX, FERR, BERR, WORK,
|
$ IWORK, INFO)
|
IF (INFO .NE. 0) THEN
|
PRINT 1090, ABS(INFO)
|
STOP 4
|
END IF
|
PRINT 1100
|
PRINT 1040, X
|
CALL DSYMV ('LOWER TRIANGLE OF A STORED', N, 1.0D0, A, LDA,
|
$ X, 1, 0.0D0, B1, 1)
|
PRINT 1110
|
PRINT 1040, B1
|
C
|
1000 FORMAT (1X, 'A in full form:')
|
1010 FORMAT (4(3X, F14.7))
|
1020 FORMAT (/1X, 'A in symmetric form: (* in unused elements)')
|
1030 FORMAT (/1X, 'b:')
|
1040 FORMAT (3X, E25.17)
|
1050 FORMAT (1X, 'Error factoring A, INFO = ', I5)
|
1060 FORMAT (1X, 'Illegal argument to DPOTRS, argument #', I2)
|
1070 FORMAT (/1X, 'Initial solution to Ax=b:')
|
1080 FORMAT (/1X, 'Ax with the initial x:')
|
1090 FORMAT (1X, 'Illegal argument to DPORFS, INFO = ', I2)
|
1100 FORMAT (/1X, 'Refined solution to Ax=b:')
|
1110 FORMAT (/1X, 'Ax with refined x:')
|
C
|
END
|
|