*> \brief \b CHETRI_ROOK computes the inverse of HE matrix using the factorization obtained with the bounded Bunch-Kaufman ("rook") diagonal pivoting method.
*
* =========== DOCUMENTATION ===========
*
* Online html documentation available at
* http://www.netlib.org/lapack/explore-html/
*
*> \htmlonly
*> Download CHETRI_ROOK + dependencies
*>
*> [TGZ]
*>
*> [ZIP]
*>
*> [TXT]
*> \endhtmlonly
*
* Definition:
* ===========
*
* SUBROUTINE CHETRI_ROOK( UPLO, N, A, LDA, IPIV, WORK, INFO )
*
* .. Scalar Arguments ..
* CHARACTER UPLO
* INTEGER INFO, LDA, N
* ..
* .. Array Arguments ..
* INTEGER IPIV( * )
* COMPLEX A( LDA, * ), WORK( * )
* ..
*
*
*> \par Purpose:
* =============
*>
*> \verbatim
*>
*> CHETRI_ROOK computes the inverse of a complex Hermitian indefinite matrix
*> A using the factorization A = U*D*U**H or A = L*D*L**H computed by
*> CHETRF_ROOK.
*> \endverbatim
*
* Arguments:
* ==========
*
*> \param[in] UPLO
*> \verbatim
*> UPLO is CHARACTER*1
*> Specifies whether the details of the factorization are stored
*> as an upper or lower triangular matrix.
*> = 'U': Upper triangular, form is A = U*D*U**H;
*> = 'L': Lower triangular, form is A = L*D*L**H.
*> \endverbatim
*>
*> \param[in] N
*> \verbatim
*> N is INTEGER
*> The order of the matrix A. N >= 0.
*> \endverbatim
*>
*> \param[in,out] A
*> \verbatim
*> A is COMPLEX array, dimension (LDA,N)
*> On entry, the block diagonal matrix D and the multipliers
*> used to obtain the factor U or L as computed by CHETRF_ROOK.
*>
*> On exit, if INFO = 0, the (Hermitian) inverse of the original
*> matrix. If UPLO = 'U', the upper triangular part of the
*> inverse is formed and the part of A below the diagonal is not
*> referenced; if UPLO = 'L' the lower triangular part of the
*> inverse is formed and the part of A above the diagonal is
*> not referenced.
*> \endverbatim
*>
*> \param[in] LDA
*> \verbatim
*> LDA is INTEGER
*> The leading dimension of the array A. LDA >= max(1,N).
*> \endverbatim
*>
*> \param[in] IPIV
*> \verbatim
*> IPIV is INTEGER array, dimension (N)
*> Details of the interchanges and the block structure of D
*> as determined by CHETRF_ROOK.
*> \endverbatim
*>
*> \param[out] WORK
*> \verbatim
*> WORK is COMPLEX array, dimension (N)
*> \endverbatim
*>
*> \param[out] INFO
*> \verbatim
*> INFO is INTEGER
*> = 0: successful exit
*> < 0: if INFO = -i, the i-th argument had an illegal value
*> > 0: if INFO = i, D(i,i) = 0; the matrix is singular and its
*> inverse could not be computed.
*> \endverbatim
*
* Authors:
* ========
*
*> \author Univ. of Tennessee
*> \author Univ. of California Berkeley
*> \author Univ. of Colorado Denver
*> \author NAG Ltd.
*
*> \ingroup complexHEcomputational
*
*> \par Contributors:
* ==================
*>
*> \verbatim
*>
*> November 2013, Igor Kozachenko,
*> Computer Science Division,
*> University of California, Berkeley
*>
*> September 2007, Sven Hammarling, Nicholas J. Higham, Craig Lucas,
*> School of Mathematics,
*> University of Manchester
*> \endverbatim
*
* =====================================================================
SUBROUTINE CHETRI_ROOK( UPLO, N, A, LDA, IPIV, WORK, INFO )
*
* -- LAPACK computational routine --
* -- LAPACK is a software package provided by Univ. of Tennessee, --
* -- Univ. of California Berkeley, Univ. of Colorado Denver and NAG Ltd..--
*
* .. Scalar Arguments ..
CHARACTER UPLO
INTEGER INFO, LDA, N
* ..
* .. Array Arguments ..
INTEGER IPIV( * )
COMPLEX A( LDA, * ), WORK( * )
* ..
*
* =====================================================================
*
* .. Parameters ..
REAL ONE
COMPLEX CONE, CZERO
PARAMETER ( ONE = 1.0E+0, CONE = ( 1.0E+0, 0.0E+0 ),
$ CZERO = ( 0.0E+0, 0.0E+0 ) )
* ..
* .. Local Scalars ..
LOGICAL UPPER
INTEGER J, K, KP, KSTEP
REAL AK, AKP1, D, T
COMPLEX AKKP1, TEMP
* ..
* .. External Functions ..
LOGICAL LSAME
COMPLEX CDOTC
EXTERNAL LSAME, CDOTC
* ..
* .. External Subroutines ..
EXTERNAL CCOPY, CHEMV, CSWAP, XERBLA
* ..
* .. Intrinsic Functions ..
INTRINSIC ABS, CONJG, MAX, REAL
* ..
* .. Executable Statements ..
*
* Test the input parameters.
*
INFO = 0
UPPER = LSAME( UPLO, 'U' )
IF( .NOT.UPPER .AND. .NOT.LSAME( UPLO, 'L' ) ) THEN
INFO = -1
ELSE IF( N.LT.0 ) THEN
INFO = -2
ELSE IF( LDA.LT.MAX( 1, N ) ) THEN
INFO = -4
END IF
IF( INFO.NE.0 ) THEN
CALL XERBLA( 'CHETRI_ROOK', -INFO )
RETURN
END IF
*
* Quick return if possible
*
IF( N.EQ.0 )
$ RETURN
*
* Check that the diagonal matrix D is nonsingular.
*
IF( UPPER ) THEN
*
* Upper triangular storage: examine D from bottom to top
*
DO 10 INFO = N, 1, -1
IF( IPIV( INFO ).GT.0 .AND. A( INFO, INFO ).EQ.CZERO )
$ RETURN
10 CONTINUE
ELSE
*
* Lower triangular storage: examine D from top to bottom.
*
DO 20 INFO = 1, N
IF( IPIV( INFO ).GT.0 .AND. A( INFO, INFO ).EQ.CZERO )
$ RETURN
20 CONTINUE
END IF
INFO = 0
*
IF( UPPER ) THEN
*
* Compute inv(A) from the factorization A = U*D*U**H.
*
* K is the main loop index, increasing from 1 to N in steps of
* 1 or 2, depending on the size of the diagonal blocks.
*
K = 1
30 CONTINUE
*
* If K > N, exit from loop.
*
IF( K.GT.N )
$ GO TO 70
*
IF( IPIV( K ).GT.0 ) THEN
*
* 1 x 1 diagonal block
*
* Invert the diagonal block.
*
A( K, K ) = ONE / REAL( A( K, K ) )
*
* Compute column K of the inverse.
*
IF( K.GT.1 ) THEN
CALL CCOPY( K-1, A( 1, K ), 1, WORK, 1 )
CALL CHEMV( UPLO, K-1, -CONE, A, LDA, WORK, 1, CZERO,
$ A( 1, K ), 1 )
A( K, K ) = A( K, K ) - REAL( CDOTC( K-1, WORK, 1, A( 1,
$ K ), 1 ) )
END IF
KSTEP = 1
ELSE
*
* 2 x 2 diagonal block
*
* Invert the diagonal block.
*
T = ABS( A( K, K+1 ) )
AK = REAL( A( K, K ) ) / T
AKP1 = REAL( A( K+1, K+1 ) ) / T
AKKP1 = A( K, K+1 ) / T
D = T*( AK*AKP1-ONE )
A( K, K ) = AKP1 / D
A( K+1, K+1 ) = AK / D
A( K, K+1 ) = -AKKP1 / D
*
* Compute columns K and K+1 of the inverse.
*
IF( K.GT.1 ) THEN
CALL CCOPY( K-1, A( 1, K ), 1, WORK, 1 )
CALL CHEMV( UPLO, K-1, -CONE, A, LDA, WORK, 1, CZERO,
$ A( 1, K ), 1 )
A( K, K ) = A( K, K ) - REAL( CDOTC( K-1, WORK, 1, A( 1,
$ K ), 1 ) )
A( K, K+1 ) = A( K, K+1 ) -
$ CDOTC( K-1, A( 1, K ), 1, A( 1, K+1 ), 1 )
CALL CCOPY( K-1, A( 1, K+1 ), 1, WORK, 1 )
CALL CHEMV( UPLO, K-1, -CONE, A, LDA, WORK, 1, CZERO,
$ A( 1, K+1 ), 1 )
A( K+1, K+1 ) = A( K+1, K+1 ) -
$ REAL( CDOTC( K-1, WORK, 1, A( 1, K+1 ),
$ 1 ) )
END IF
KSTEP = 2
END IF
*
IF( KSTEP.EQ.1 ) THEN
*
* Interchange rows and columns K and IPIV(K) in the leading
* submatrix A(1:k,1:k)
*
KP = IPIV( K )
IF( KP.NE.K ) THEN
*
IF( KP.GT.1 )
$ CALL CSWAP( KP-1, A( 1, K ), 1, A( 1, KP ), 1 )
*
DO 40 J = KP + 1, K - 1
TEMP = CONJG( A( J, K ) )
A( J, K ) = CONJG( A( KP, J ) )
A( KP, J ) = TEMP
40 CONTINUE
*
A( KP, K ) = CONJG( A( KP, K ) )
*
TEMP = A( K, K )
A( K, K ) = A( KP, KP )
A( KP, KP ) = TEMP
END IF
ELSE
*
* Interchange rows and columns K and K+1 with -IPIV(K) and
* -IPIV(K+1) in the leading submatrix A(k+1:n,k+1:n)
*
* (1) Interchange rows and columns K and -IPIV(K)
*
KP = -IPIV( K )
IF( KP.NE.K ) THEN
*
IF( KP.GT.1 )
$ CALL CSWAP( KP-1, A( 1, K ), 1, A( 1, KP ), 1 )
*
DO 50 J = KP + 1, K - 1
TEMP = CONJG( A( J, K ) )
A( J, K ) = CONJG( A( KP, J ) )
A( KP, J ) = TEMP
50 CONTINUE
*
A( KP, K ) = CONJG( A( KP, K ) )
*
TEMP = A( K, K )
A( K, K ) = A( KP, KP )
A( KP, KP ) = TEMP
*
TEMP = A( K, K+1 )
A( K, K+1 ) = A( KP, K+1 )
A( KP, K+1 ) = TEMP
END IF
*
* (2) Interchange rows and columns K+1 and -IPIV(K+1)
*
K = K + 1
KP = -IPIV( K )
IF( KP.NE.K ) THEN
*
IF( KP.GT.1 )
$ CALL CSWAP( KP-1, A( 1, K ), 1, A( 1, KP ), 1 )
*
DO 60 J = KP + 1, K - 1
TEMP = CONJG( A( J, K ) )
A( J, K ) = CONJG( A( KP, J ) )
A( KP, J ) = TEMP
60 CONTINUE
*
A( KP, K ) = CONJG( A( KP, K ) )
*
TEMP = A( K, K )
A( K, K ) = A( KP, KP )
A( KP, KP ) = TEMP
END IF
END IF
*
K = K + 1
GO TO 30
70 CONTINUE
*
ELSE
*
* Compute inv(A) from the factorization A = L*D*L**H.
*
* K is the main loop index, decreasing from N to 1 in steps of
* 1 or 2, depending on the size of the diagonal blocks.
*
K = N
80 CONTINUE
*
* If K < 1, exit from loop.
*
IF( K.LT.1 )
$ GO TO 120
*
IF( IPIV( K ).GT.0 ) THEN
*
* 1 x 1 diagonal block
*
* Invert the diagonal block.
*
A( K, K ) = ONE / REAL( A( K, K ) )
*
* Compute column K of the inverse.
*
IF( K.LT.N ) THEN
CALL CCOPY( N-K, A( K+1, K ), 1, WORK, 1 )
CALL CHEMV( UPLO, N-K, -CONE, A( K+1, K+1 ), LDA, WORK,
$ 1, CZERO, A( K+1, K ), 1 )
A( K, K ) = A( K, K ) - REAL( CDOTC( N-K, WORK, 1,
$ A( K+1, K ), 1 ) )
END IF
KSTEP = 1
ELSE
*
* 2 x 2 diagonal block
*
* Invert the diagonal block.
*
T = ABS( A( K, K-1 ) )
AK = REAL( A( K-1, K-1 ) ) / T
AKP1 = REAL( A( K, K ) ) / T
AKKP1 = A( K, K-1 ) / T
D = T*( AK*AKP1-ONE )
A( K-1, K-1 ) = AKP1 / D
A( K, K ) = AK / D
A( K, K-1 ) = -AKKP1 / D
*
* Compute columns K-1 and K of the inverse.
*
IF( K.LT.N ) THEN
CALL CCOPY( N-K, A( K+1, K ), 1, WORK, 1 )
CALL CHEMV( UPLO, N-K, -CONE, A( K+1, K+1 ), LDA, WORK,
$ 1, CZERO, A( K+1, K ), 1 )
A( K, K ) = A( K, K ) - REAL( CDOTC( N-K, WORK, 1,
$ A( K+1, K ), 1 ) )
A( K, K-1 ) = A( K, K-1 ) -
$ CDOTC( N-K, A( K+1, K ), 1, A( K+1, K-1 ),
$ 1 )
CALL CCOPY( N-K, A( K+1, K-1 ), 1, WORK, 1 )
CALL CHEMV( UPLO, N-K, -CONE, A( K+1, K+1 ), LDA, WORK,
$ 1, CZERO, A( K+1, K-1 ), 1 )
A( K-1, K-1 ) = A( K-1, K-1 ) -
$ REAL( CDOTC( N-K, WORK, 1, A( K+1, K-1 ),
$ 1 ) )
END IF
KSTEP = 2
END IF
*
IF( KSTEP.EQ.1 ) THEN
*
* Interchange rows and columns K and IPIV(K) in the trailing
* submatrix A(k:n,k:n)
*
KP = IPIV( K )
IF( KP.NE.K ) THEN
*
IF( KP.LT.N )
$ CALL CSWAP( N-KP, A( KP+1, K ), 1, A( KP+1, KP ), 1 )
*
DO 90 J = K + 1, KP - 1
TEMP = CONJG( A( J, K ) )
A( J, K ) = CONJG( A( KP, J ) )
A( KP, J ) = TEMP
90 CONTINUE
*
A( KP, K ) = CONJG( A( KP, K ) )
*
TEMP = A( K, K )
A( K, K ) = A( KP, KP )
A( KP, KP ) = TEMP
END IF
ELSE
*
* Interchange rows and columns K and K-1 with -IPIV(K) and
* -IPIV(K-1) in the trailing submatrix A(k-1:n,k-1:n)
*
* (1) Interchange rows and columns K and -IPIV(K)
*
KP = -IPIV( K )
IF( KP.NE.K ) THEN
*
IF( KP.LT.N )
$ CALL CSWAP( N-KP, A( KP+1, K ), 1, A( KP+1, KP ), 1 )
*
DO 100 J = K + 1, KP - 1
TEMP = CONJG( A( J, K ) )
A( J, K ) = CONJG( A( KP, J ) )
A( KP, J ) = TEMP
100 CONTINUE
*
A( KP, K ) = CONJG( A( KP, K ) )
*
TEMP = A( K, K )
A( K, K ) = A( KP, KP )
A( KP, KP ) = TEMP
*
TEMP = A( K, K-1 )
A( K, K-1 ) = A( KP, K-1 )
A( KP, K-1 ) = TEMP
END IF
*
* (2) Interchange rows and columns K-1 and -IPIV(K-1)
*
K = K - 1
KP = -IPIV( K )
IF( KP.NE.K ) THEN
*
IF( KP.LT.N )
$ CALL CSWAP( N-KP, A( KP+1, K ), 1, A( KP+1, KP ), 1 )
*
DO 110 J = K + 1, KP - 1
TEMP = CONJG( A( J, K ) )
A( J, K ) = CONJG( A( KP, J ) )
A( KP, J ) = TEMP
110 CONTINUE
*
A( KP, K ) = CONJG( A( KP, K ) )
*
TEMP = A( K, K )
A( K, K ) = A( KP, KP )
A( KP, KP ) = TEMP
END IF
END IF
*
K = K - 1
GO TO 80
120 CONTINUE
END IF
*
RETURN
*
* End of CHETRI_ROOK
*
END