Return to ztptri.f CVS log

Up to [local] / rpl / lapack / lapack

Cohérence

    1:       SUBROUTINE ZTPTRI( UPLO, DIAG, N, AP, INFO )
    2: *
    3: *  -- LAPACK routine (version 3.2) --
    4: *  -- LAPACK is a software package provided by Univ. of Tennessee,    --
    5: *  -- Univ. of California Berkeley, Univ. of Colorado Denver and NAG Ltd..--
    6: *     November 2006
    7: *
    8: *     .. Scalar Arguments ..
    9:       CHARACTER          DIAG, UPLO
   10:       INTEGER            INFO, N
   11: *     ..
   12: *     .. Array Arguments ..
   13:       COMPLEX*16         AP( * )
   14: *     ..
   15: *
   16: *  Purpose
   17: *  =======
   18: *
   19: *  ZTPTRI computes the inverse of a complex upper or lower triangular
   20: *  matrix A stored in packed format.
   21: *
   22: *  Arguments
   23: *  =========
   24: *
   25: *  UPLO    (input) CHARACTER*1
   26: *          = 'U':  A is upper triangular;
   27: *          = 'L':  A is lower triangular.
   28: *
   29: *  DIAG    (input) CHARACTER*1
   30: *          = 'N':  A is non-unit triangular;
   31: *          = 'U':  A is unit triangular.
   32: *
   33: *  N       (input) INTEGER
   34: *          The order of the matrix A.  N >= 0.
   35: *
   36: *  AP      (input/output) COMPLEX*16 array, dimension (N*(N+1)/2)
   37: *          On entry, the upper or lower triangular matrix A, stored
   38: *          columnwise in a linear array.  The j-th column of A is stored
   39: *          in the array AP as follows:
   40: *          if UPLO = 'U', AP(i + (j-1)*j/2) = A(i,j) for 1<=i<=j;
   41: *          if UPLO = 'L', AP(i + (j-1)*((2*n-j)/2) = A(i,j) for j<=i<=n.
   42: *          See below for further details.
   43: *          On exit, the (triangular) inverse of the original matrix, in
   44: *          the same packed storage format.
   45: *
   46: *  INFO    (output) INTEGER
   47: *          = 0:  successful exit
   48: *          < 0:  if INFO = -i, the i-th argument had an illegal value
   49: *          > 0:  if INFO = i, A(i,i) is exactly zero.  The triangular
   50: *                matrix is singular and its inverse can not be computed.
   51: *
   52: *  Further Details
   53: *  ===============
   54: *
   55: *  A triangular matrix A can be transferred to packed storage using one
   56: *  of the following program segments:
   57: *
   58: *  UPLO = 'U':                      UPLO = 'L':
   59: *
   60: *        JC = 1                           JC = 1
   61: *        DO 2 J = 1, N                    DO 2 J = 1, N
   62: *           DO 1 I = 1, J                    DO 1 I = J, N
   63: *              AP(JC+I-1) = A(I,J)              AP(JC+I-J) = A(I,J)
   64: *      1    CONTINUE                    1    CONTINUE
   65: *           JC = JC + J                      JC = JC + N - J + 1
   66: *      2 CONTINUE                       2 CONTINUE
   67: *
   68: *  =====================================================================
   69: *
   70: *     .. Parameters ..
   71:       COMPLEX*16         ONE, ZERO
   72:       PARAMETER          ( ONE = ( 1.0D+0, 0.0D+0 ),
   73:      $                   ZERO = ( 0.0D+0, 0.0D+0 ) )
   74: *     ..
   75: *     .. Local Scalars ..
   76:       LOGICAL            NOUNIT, UPPER
   77:       INTEGER            J, JC, JCLAST, JJ
   78:       COMPLEX*16         AJJ
   79: *     ..
   80: *     .. External Functions ..
   81:       LOGICAL            LSAME
   82:       EXTERNAL           LSAME
   83: *     ..
   84: *     .. External Subroutines ..
   85:       EXTERNAL           XERBLA, ZSCAL, ZTPMV
   86: *     ..
   87: *     .. Executable Statements ..
   88: *
   89: *     Test the input parameters.
   90: *
   91:       INFO = 0
   92:       UPPER = LSAME( UPLO, 'U' )
   93:       NOUNIT = LSAME( DIAG, 'N' )
   94:       IF( .NOT.UPPER .AND. .NOT.LSAME( UPLO, 'L' ) ) THEN
   95:          INFO = -1
   96:       ELSE IF( .NOT.NOUNIT .AND. .NOT.LSAME( DIAG, 'U' ) ) THEN
   97:          INFO = -2
   98:       ELSE IF( N.LT.0 ) THEN
   99:          INFO = -3
  100:       END IF
  101:       IF( INFO.NE.0 ) THEN
  102:          CALL XERBLA( 'ZTPTRI', -INFO )
  103:          RETURN
  104:       END IF
  105: *
  106: *     Check for singularity if non-unit.
  107: *
  108:       IF( NOUNIT ) THEN
  109:          IF( UPPER ) THEN
  110:             JJ = 0
  111:             DO 10 INFO = 1, N
  112:                JJ = JJ + INFO
  113:                IF( AP( JJ ).EQ.ZERO )
  114:      $            RETURN
  115:    10       CONTINUE
  116:          ELSE
  117:             JJ = 1
  118:             DO 20 INFO = 1, N
  119:                IF( AP( JJ ).EQ.ZERO )
  120:      $            RETURN
  121:                JJ = JJ + N - INFO + 1
  122:    20       CONTINUE
  123:          END IF
  124:          INFO = 0
  125:       END IF
  126: *
  127:       IF( UPPER ) THEN
  128: *
  129: *        Compute inverse of upper triangular matrix.
  130: *
  131:          JC = 1
  132:          DO 30 J = 1, N
  133:             IF( NOUNIT ) THEN
  134:                AP( JC+J-1 ) = ONE / AP( JC+J-1 )
  135:                AJJ = -AP( JC+J-1 )
  136:             ELSE
  137:                AJJ = -ONE
  138:             END IF
  139: *
  140: *           Compute elements 1:j-1 of j-th column.
  141: *
  142:             CALL ZTPMV( 'Upper', 'No transpose', DIAG, J-1, AP,
  143:      $                  AP( JC ), 1 )
  144:             CALL ZSCAL( J-1, AJJ, AP( JC ), 1 )
  145:             JC = JC + J
  146:    30    CONTINUE
  147: *
  148:       ELSE
  149: *
  150: *        Compute inverse of lower triangular matrix.
  151: *
  152:          JC = N*( N+1 ) / 2
  153:          DO 40 J = N, 1, -1
  154:             IF( NOUNIT ) THEN
  155:                AP( JC ) = ONE / AP( JC )
  156:                AJJ = -AP( JC )
  157:             ELSE
  158:                AJJ = -ONE
  159:             END IF
  160:             IF( J.LT.N ) THEN
  161: *
  162: *              Compute elements j+1:n of j-th column.
  163: *
  164:                CALL ZTPMV( 'Lower', 'No transpose', DIAG, N-J,
  165:      $                     AP( JCLAST ), AP( JC+1 ), 1 )
  166:                CALL ZSCAL( N-J, AJJ, AP( JC+1 ), 1 )
  167:             END IF
  168:             JCLAST = JC
  169:             JC = JC - N + J - 2
  170:    40    CONTINUE
  171:       END IF
  172: *
  173:       RETURN
  174: *
  175: *     End of ZTPTRI
  176: *
  177:       END