* F07CHF Example Program Text * Mark 21 Release. NAG Copyright 2004. * .. Parameters .. INTEGER NIN, NOUT PARAMETER (NIN=5,NOUT=6) INTEGER NMAX, NRHSMX PARAMETER (NMAX=50,NRHSMX=4) INTEGER LDB, LDX PARAMETER (LDB=NMAX,LDX=NMAX) * .. Local Scalars .. INTEGER I, IFAIL, INFO, J, N, NRHS * .. Local Arrays .. DOUBLE PRECISION B(LDB,NRHSMX), BERR(NRHSMX), D(NMAX), DF(NMAX), + DL(NMAX-1), DLF(NMAX-1), DU(NMAX-1), DU2(NMAX-2), + DUF(NMAX-1), FERR(NRHSMX), WORK(3*NMAX), + X(LDX,NRHSMX) INTEGER IPIV(NMAX), IWORK(NMAX) * .. External Subroutines .. EXTERNAL DCOPY, DGTRFS, DGTTRF, DGTTRS, F06QFF, X04CAF * .. Executable Statements .. WRITE (NOUT,*) 'F07CHF Example Program Results' WRITE (NOUT,*) * Skip heading in data file READ (NIN,*) READ (NIN,*) N, NRHS IF (N.LE.NMAX .AND. NRHS.LE.NRHSMX) THEN * * Read the tridiagonal matrix A from data file * READ (NIN,*) (DU(I),I=1,N-1) READ (NIN,*) (D(I),I=1,N) READ (NIN,*) (DL(I),I=1,N-1) * * Read the right hand matrix B * READ (NIN,*) ((B(I,J),J=1,NRHS),I=1,N) * * Copy A into DUF, DF and DLF, and copy B into X * CALL DCOPY(N-1,DU,1,DUF,1) CALL DCOPY(N,D,1,DF,1) CALL DCOPY(N-1,DL,1,DLF,1) CALL F06QFF('General',N,NRHS,B,LDB,X,LDX) * * Factorize the copy of the tridiagonal matrix A * CALL DGTTRF(N,DLF,DF,DUF,DU2,IPIV,INFO) * IF (INFO.EQ.0) THEN * * Solve the equations AX = B * CALL DGTTRS('No transpose',N,NRHS,DLF,DF,DUF,DU2,IPIV,X,LDX, + INFO) * * Improve the solution and compute error estimates * CALL DGTRFS('No transpose',N,NRHS,DL,D,DU,DLF,DF,DUF,DU2, + IPIV,B,LDB,X,LDX,FERR,BERR,WORK,IWORK,INFO) * * Print the solution and the forward and backward error * estimates * IFAIL = 0 CALL X04CAF('General',' ',N,NRHS,X,LDX,'Solution(s)',IFAIL) * WRITE (NOUT,*) WRITE (NOUT,*) 'Backward errors (machine-dependent)' WRITE (NOUT,99999) (BERR(J),J=1,NRHS) WRITE (NOUT,*) WRITE (NOUT,*) + 'Estimated forward error bounds (machine-dependent)' WRITE (NOUT,99999) (FERR(J),J=1,NRHS) ELSE WRITE (NOUT,99998) 'The (', INFO, ',', INFO, ')', + ' element of the factor U is zero' END IF ELSE WRITE (NOUT,*) 'NMAX and/or NRHSMX too small' END IF STOP * 99999 FORMAT ((3X,1P,7E11.1)) 99998 FORMAT (1X,A,I3,A,I3,A,A) END