|
PROGRAM TEST
|
IMPLICIT NONE
|
C
|
INTEGER LDA, LDAF, LDB, LDIWRK, LDWORK, LDX
|
INTEGER N, NRHS, NDIAG
|
PARAMETER (N = 4)
|
PARAMETER (NDIAG = 1)
|
PARAMETER (LDA = NDIAG + 1)
|
PARAMETER (LDAF = LDA)
|
PARAMETER (LDB = N)
|
PARAMETER (LDIWRK = N)
|
PARAMETER (LDWORK = 3 * N)
|
PARAMETER (LDX = N)
|
PARAMETER (NRHS = 1)
|
C
|
DOUBLE PRECISION A(LDA,N), AF(LDAF,N), B(LDB,NRHS)
|
DOUBLE PRECISION B1(LDB,NRHS), BERR(NRHS), EPSLON
|
DOUBLE PRECISION FERR(NRHS), WORK(LDWORK), X(LDX,NRHS)
|
INTEGER ICOL, INFO, IROW, IWORK(LDIWRK)
|
C
|
EXTERNAL DCOPY, DSBMV, DPBTRF, DPBTRS, DPBRFS
|
INTRINSIC ABS, DBLE, MAX
|
C
|
C Initialize the array B to store the right hand side
|
C vector b shown.
|
C
|
C 4
|
C b = 4
|
C 4
|
C 4
|
C
|
DATA B / N*4.0D0 /
|
C
|
C Initialize the array A to contain the matrix A shown below
|
C where e=epsilon=small.
|
C
|
C e e*e
|
C A = e*e e e*e
|
C e*e e e*e
|
C e*e e
|
C
|
A(1,1) = SQRT(-1.0D0)
|
EPSLON = ABS((((2.0D0 / 3.0D0) + 8.0D0) - 8.0D0) -
|
$ (2.0D0 / 3.0D0))
|
DO 100, ICOL = 1, N
|
A(NDIAG + 1,ICOL) = EPSLON
|
100 CONTINUE
|
DO 110, ICOL = 2, N
|
A(1,ICOL) = EPSLON * EPSLON
|
110 CONTINUE
|
CALL DCOPY (LDA*N, A, 1, AF, 1)
|
C
|
C Print the A array.
|
C
|
PRINT 1000
|
PRINT 1010, A(2,1), A(1,2), 0.0D0, 0.0D0
|
PRINT 1010, A(1,2), A(2,2), A(1,3), 0.0D0
|
PRINT 1010, 0.0D0, A(1,3), A(2,3), A(1,4)
|
PRINT 1010, 0.0D0, 0.0D0, A(1,4), A(2,4)
|
PRINT 1020
|
PRINT 1010, ((A(IROW,ICOL), ICOL = 1, N), IROW = 1, LDA)
|
C
|
C Make a copy of B and then print B.
|
C
|
CALL DCOPY (N, B, 1, X, 1)
|
PRINT 1030
|
PRINT 1040, B
|
C
|
C Factor A.
|
C
|
CALL DPBTRF ('UPPER TRIANGLE OF A STORED', N, NDIAG,
|
$ 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 DPBTRS ('UPPER TRIANGLE OF A STORED', N, NDIAG, NRHS,
|
$ AF, LDAF, X, LDX, INFO)
|
IF (INFO .NE. 0) THEN
|
PRINT 1060, INFO
|
STOP 2
|
END IF
|
PRINT 1070
|
PRINT 1080, X
|
CALL DSBMV ('UPPER TRIANGLE OF A STORED', N, NDIAG, 1.0D0,
|
$ A, LDA, X, 1, 0.0D0, B1, 1)
|
PRINT 1090
|
PRINT 1040, B1
|
C
|
C Refine the initial solution and print the new solution.
|
C
|
CALL DPBRFS ('UPPER TRIANGLE OF A STORED', N, NDIAG, NRHS,
|
$ A, LDA, AF, LDAF, B, LDB, X, LDX, FERR, BERR,
|
$ WORK, IWORK, INFO)
|
IF (INFO .NE. 0) THEN
|
PRINT 1100, INFO
|
STOP 3
|
END IF
|
PRINT 1110
|
PRINT 1080, X
|
CALL DSBMV ('UPPER TRIANGLE OF A STORED', N, NDIAG, 1.0D0,
|
$ A, LDA, X, 1, 0.0D0, B1, 1)
|
PRINT 1120
|
PRINT 1040, B1
|
C
|
1000 FORMAT (1X, 'A in full form:')
|
1010 FORMAT (4(2X, E15.8))
|
1020 FORMAT (/1X, 'A in banded form: ("NaN" in unused elements)')
|
1030 FORMAT (/1X, 'b:')
|
1040 FORMAT (1X, F19.15)
|
1050 FORMAT (1X, 'Error factoring A in DPBTRF, INFO = ', I5)
|
1060 FORMAT (1X, 'Error solving Ax=b in DPBTRF, INFO = ', I5)
|
1070 FORMAT (/1X, 'Initial solution for Ax=b:')
|
1080 FORMAT (1X, E24.16)
|
1090 FORMAT (/1X, 'Ax with initial x:')
|
1100 FORMAT (1X, 'Error improving solution in DPBRFS, INFO = ',
|
$ I5)
|
1110 FORMAT (/1X, 'Refined solution for Ax=b:')
|
1120 FORMAT (/1X, 'Ax with improved x:')
|
C
|
END
|
|