zhescal.f

Go to the documentation of this file.

      SUBROUTINE zhescal( UPLO, M, N, IOFFD, ALPHA, A, LDA )
*
*  -- PBLAS auxiliary routine (version 2.0) --
*     University of Tennessee, Knoxville, Oak Ridge National Laboratory,
*     and University of California, Berkeley.
*     April 1, 1998
*
*     .. Scalar Arguments ..
      CHARACTER*1        UPLO
      INTEGER            IOFFD, LDA, M, N
      DOUBLE PRECISION   ALPHA
*     ..
*     .. Array Arguments ..
      COMPLEX*16         A( LDA, * )
*     ..
*
*  Purpose
*  =======
*
*  ZHESCAL  scales  a two-dimensional  array A by the real scalar alpha.
*  The diagonal entries specified by IOFFD of A are supposed to be real.
*
*  Arguments
*  =========
*
*  UPLO    (input) CHARACTER*1
*          On entry,  UPLO  specifies  which trapezoidal part of the ar-
*          ray A is to be scaled as follows:
*             = 'L' or 'l':          the lower trapezoid of A is scaled,
*             = 'U' or 'u':          the upper trapezoid of A is scaled,
*             = 'D' or 'd':       diagonal specified by IOFFD is scaled,
*             Otherwise:                   all of the array A is scaled.
*
*  M       (input) INTEGER
*          On entry,  M  specifies the number of rows of the array A.  M
*          must be at least zero.
*
*  N       (input) INTEGER
*          On entry,  N  specifies the number of columns of the array A.
*          N must be at least zero.
*
*  IOFFD   (input) INTEGER
*          On entry, IOFFD specifies the position of the offdiagonal de-
*          limiting the upper and lower trapezoidal part of A as follows
*          (see the notes below):
*
*             IOFFD = 0  specifies the main diagonal A( i, i ),
*                        with i = 1 ... MIN( M, N ),
*             IOFFD > 0  specifies the subdiagonal   A( i+IOFFD, i ),
*                        with i = 1 ... MIN( M-IOFFD, N ),
*             IOFFD < 0  specifies the superdiagonal A( i, i-IOFFD ),
*                        with i = 1 ... MIN( M, N+IOFFD ).
*
*  ALPHA   (input) DOUBLE PRECISION
*          On entry,  ALPHA  specifies the scalar alpha, i.e., the value
*          by which the diagonal and offdiagonal entries of the array  A
*          as specified by UPLO and IOFFD are scaled.
*
*  A       (input/output) COMPLEX*16 array
*          On entry,  A  is an array of dimension (LDA,N).  Before entry
*          with UPLO = 'U' or 'u', the leading m by n part of the  array
*          A  must contain  the  upper trapezoidal part of the Hermitian
*          matrix to be scaled as specified by  IOFFD, and the  strictly
*          lower trapezoidal part of A is not referenced.  When  UPLO is
*          'L' or 'l', the leading m by n part of the array  A must con-
*          tain the lower trapezoidal part of the Hermitian matrix to be
*          scaled  as  specified by IOFFD, and the strictly upper trape-
*          zoidal part of A is not referenced. On exit, the  entries  of
*          the trapezoid part of A determined by UPLO and IOFFD are sca-
*          led.
*
*  LDA     (input) INTEGER
*          On entry, LDA specifies the leading dimension of the array A.
*          LDA must be at least max( 1, M ).
*
*  Notes
*  =====
*                           N                                    N
*             ----------------------------                  -----------
*            |       d                    |                |           |
*          M |         d        'U'       |                |      'U'  |
*            |  'L'     'D'               |                |d          |
*            |             d              |              M |  d        |
*             ----------------------------                 |   'D'     |
*                                                          |      d    |
*               IOFFD < 0                                  | 'L'    d  |
*                                                          |          d|
*                  N                                       |           |
*             -----------                                   -----------
*            |    d   'U'|
*            |      d    |                                   IOFFD > 0
*          M |       'D' |
*            |          d|                              N
*            |  'L'      |                 ----------------------------
*            |           |                |          'U'               |
*            |           |                |d                           |
*            |           |                | 'D'                        |
*            |           |                |    d                       |
*            |           |                |'L'   d                     |
*             -----------                  ----------------------------
*
*  -- Written on April 1, 1998 by
*     Antoine Petitet, University  of  Tennessee, Knoxville 37996, USA.
*
*  =====================================================================
*
*     .. Parameters ..
      DOUBLE PRECISION   RONE, RZERO
      parameter( rone = 1.0d+0, rzero = 0.0d+0 )
      COMPLEX*16         ZERO
      parameter( zero = ( 0.0d+0, 0.0d+0 ) )
*     ..
*     .. Local Scalars ..
      INTEGER            J, JTMP, MN
*     ..
*     .. External Subroutines ..
      EXTERNAL           zdscal, ztzpad
*     ..
*     .. External Functions ..
      LOGICAL            LSAME
      EXTERNAL           lsame
*     ..
*     .. Intrinsic Functions ..
      INTRINSIC          dble, dcmplx, max, min
*     ..
*     .. Executable Statements ..
*
*     Quick return if possible
*
      IF( m.LE.0 .OR. n.LE.0 )
     $   RETURN
*
*     Start the operations
*
      IF( alpha.EQ.rone ) THEN
*
*        Zeros the imaginary part of the diagonals
*
         IF( lsame( uplo, 'L' ).OR.lsame( uplo, 'U' ).OR.
     $       lsame( uplo, 'D' ) ) THEN
            DO 10 j = max( 0, -ioffd ) + 1, min( m - ioffd, n )
               jtmp = j + ioffd
               a( jtmp, j ) = dcmplx( dble( a( jtmp, j ) ), rzero )
   10       CONTINUE
         END IF
         RETURN
      ELSE IF( alpha.EQ.rzero ) THEN
         CALL ztzpad( uplo, 'n', M, N, IOFFD, ZERO, ZERO, A, LDA )
         RETURN
      END IF
*
      IF( LSAME( UPLO, 'l' ) ) THEN
*
*        Scales the lower triangular part of the array by ALPHA.
*
         MN = MAX( 0, -IOFFD )
         DO 20 J = 1, MIN( MN, N )
            CALL ZDSCAL( M, ALPHA, A( 1, J ), 1 )
   20    CONTINUE
         DO 30 J = MN + 1, MIN( M - IOFFD, N )
            JTMP = J + IOFFD
            A( JTMP, J ) = DCMPLX( ALPHA * DBLE( A( JTMP, J ) ), RZERO )
.GT.            IF( MJTMP )
     $         CALL ZDSCAL( M-JTMP, ALPHA, A( JTMP + 1, J ), 1 )
   30    CONTINUE
*
      ELSE IF( LSAME( UPLO, 'u' ) ) THEN
*
*        Scales the upper triangular part of the array by ALPHA.
*
         MN = MIN( M - IOFFD, N )
         DO 40 J = MAX( 0, -IOFFD ) + 1, MN
            JTMP = J + IOFFD
            CALL ZDSCAL( JTMP - 1, ALPHA, A( 1, J ), 1 )
            A( JTMP, J ) = DCMPLX( ALPHA * DBLE( A( JTMP, J ) ), RZERO )
   40    CONTINUE
         DO 50 J = MAX( 0, MN ) + 1, N
            CALL ZDSCAL( M, ALPHA, A( 1, J ), 1 )
   50    CONTINUE
*
      ELSE IF( LSAME( UPLO, 'd' ) ) THEN
*
*        Scales the diagonal entries by ALPHA.
*
         DO 60 J = MAX( 0, -IOFFD ) + 1, MIN( M - IOFFD, N )
            JTMP = J + IOFFD
            A( JTMP, J ) = DCMPLX( ALPHA * DBLE( A( JTMP, J ) ), RZERO )
   60    CONTINUE
*
      ELSE
*
*        Scales the entire array by ALPHA.
*
         DO 70 J = 1, N
            CALL ZDSCAL( M, ALPHA, A( 1, J ), 1 )
   70    CONTINUE
*
      END IF
*
      RETURN
*
*     End of ZHESCAL
*
      END