dstedc.f.html

来自「famous linear algebra library (LAPACK) p」· HTML 代码 · 共 432 行 · 第 1/3 页

</span><span class="comment">*</span><span class="comment">     .. Parameters ..
</span>      DOUBLE PRECISION   ZERO, ONE, TWO
      PARAMETER          ( ZERO = 0.0D0, ONE = 1.0D0, TWO = 2.0D0 )
<span class="comment">*</span><span class="comment">     ..
</span><span class="comment">*</span><span class="comment">     .. Local Scalars ..
</span>      LOGICAL            LQUERY
      INTEGER            FINISH, I, ICOMPZ, II, J, K, LGN, LIWMIN,
     $                   LWMIN, M, SMLSIZ, START, STOREZ, STRTRW
      DOUBLE PRECISION   EPS, ORGNRM, P, TINY
<span class="comment">*</span><span class="comment">     ..
</span><span class="comment">*</span><span class="comment">     .. External Functions ..
</span>      LOGICAL            <a name="LSAME.137"></a><a href="lsame.f.html#LSAME.1">LSAME</a>
      INTEGER            <a name="ILAENV.138"></a><a href="hfy-index.html#ILAENV">ILAENV</a>
      DOUBLE PRECISION   <a name="DLAMCH.139"></a><a href="dlamch.f.html#DLAMCH.1">DLAMCH</a>, <a name="DLANST.139"></a><a href="dlanst.f.html#DLANST.1">DLANST</a>
      EXTERNAL           <a name="LSAME.140"></a><a href="lsame.f.html#LSAME.1">LSAME</a>, <a name="ILAENV.140"></a><a href="hfy-index.html#ILAENV">ILAENV</a>, <a name="DLAMCH.140"></a><a href="dlamch.f.html#DLAMCH.1">DLAMCH</a>, <a name="DLANST.140"></a><a href="dlanst.f.html#DLANST.1">DLANST</a>
<span class="comment">*</span><span class="comment">     ..
</span><span class="comment">*</span><span class="comment">     .. External Subroutines ..
</span>      EXTERNAL           DGEMM, <a name="DLACPY.143"></a><a href="dlacpy.f.html#DLACPY.1">DLACPY</a>, <a name="DLAED0.143"></a><a href="dlaed0.f.html#DLAED0.1">DLAED0</a>, <a name="DLASCL.143"></a><a href="dlascl.f.html#DLASCL.1">DLASCL</a>, <a name="DLASET.143"></a><a href="dlaset.f.html#DLASET.1">DLASET</a>, <a name="DLASRT.143"></a><a href="dlasrt.f.html#DLASRT.1">DLASRT</a>,
     $                   <a name="DSTEQR.144"></a><a href="dsteqr.f.html#DSTEQR.1">DSTEQR</a>, <a name="DSTERF.144"></a><a href="dsterf.f.html#DSTERF.1">DSTERF</a>, DSWAP, <a name="XERBLA.144"></a><a href="xerbla.f.html#XERBLA.1">XERBLA</a>
<span class="comment">*</span><span class="comment">     ..
</span><span class="comment">*</span><span class="comment">     .. Intrinsic Functions ..
</span>      INTRINSIC          ABS, DBLE, INT, LOG, MAX, MOD, SQRT
<span class="comment">*</span><span class="comment">     ..
</span><span class="comment">*</span><span class="comment">     .. Executable Statements ..
</span><span class="comment">*</span><span class="comment">
</span><span class="comment">*</span><span class="comment">     Test the input parameters.
</span><span class="comment">*</span><span class="comment">
</span>      INFO = 0
      LQUERY = ( LWORK.EQ.-1 .OR. LIWORK.EQ.-1 )
<span class="comment">*</span><span class="comment">
</span>      IF( <a name="LSAME.156"></a><a href="lsame.f.html#LSAME.1">LSAME</a>( COMPZ, <span class="string">'N'</span> ) ) THEN
         ICOMPZ = 0
      ELSE IF( <a name="LSAME.158"></a><a href="lsame.f.html#LSAME.1">LSAME</a>( COMPZ, <span class="string">'V'</span> ) ) THEN
         ICOMPZ = 1
      ELSE IF( <a name="LSAME.160"></a><a href="lsame.f.html#LSAME.1">LSAME</a>( COMPZ, <span class="string">'I'</span> ) ) THEN
         ICOMPZ = 2
      ELSE
         ICOMPZ = -1
      END IF
      IF( ICOMPZ.LT.0 ) THEN
         INFO = -1
      ELSE IF( N.LT.0 ) THEN
         INFO = -2
      ELSE IF( ( LDZ.LT.1 ) .OR.
     $         ( ICOMPZ.GT.0 .AND. LDZ.LT.MAX( 1, N ) ) ) THEN
         INFO = -6
      END IF
<span class="comment">*</span><span class="comment">
</span>      IF( INFO.EQ.0 ) THEN
<span class="comment">*</span><span class="comment">
</span><span class="comment">*</span><span class="comment">        Compute the workspace requirements
</span><span class="comment">*</span><span class="comment">
</span>         SMLSIZ = <a name="ILAENV.178"></a><a href="hfy-index.html#ILAENV">ILAENV</a>( 9, <span class="string">'<a name="DSTEDC.178"></a><a href="dstedc.f.html#DSTEDC.1">DSTEDC</a>'</span>, <span class="string">' '</span>, 0, 0, 0, 0 )
         IF( N.LE.1 .OR. ICOMPZ.EQ.0 ) THEN
            LIWMIN = 1
            LWMIN = 1
         ELSE IF( N.LE.SMLSIZ ) THEN
            LIWMIN = 1
            LWMIN = 2*( N - 1 )
         ELSE
            LGN = INT( LOG( DBLE( N ) )/LOG( TWO ) )
            IF( 2**LGN.LT.N )
     $         LGN = LGN + 1
            IF( 2**LGN.LT.N )
     $         LGN = LGN + 1
            IF( ICOMPZ.EQ.1 ) THEN
               LWMIN = 1 + 3*N + 2*N*LGN + 3*N**2
               LIWMIN = 6 + 6*N + 5*N*LGN
            ELSE IF( ICOMPZ.EQ.2 ) THEN
               LWMIN = 1 + 4*N + N**2
               LIWMIN = 3 + 5*N
            END IF
         END IF
         WORK( 1 ) = LWMIN
         IWORK( 1 ) = LIWMIN
<span class="comment">*</span><span class="comment">
</span>         IF( LWORK.LT.LWMIN .AND. .NOT. LQUERY ) THEN
            INFO = -8
         ELSE IF( LIWORK.LT.LIWMIN .AND. .NOT. LQUERY ) THEN
            INFO = -10
         END IF
      END IF
<span class="comment">*</span><span class="comment">
</span>      IF( INFO.NE.0 ) THEN
         CALL <a name="XERBLA.210"></a><a href="xerbla.f.html#XERBLA.1">XERBLA</a>( <span class="string">'<a name="DSTEDC.210"></a><a href="dstedc.f.html#DSTEDC.1">DSTEDC</a>'</span>, -INFO )
         RETURN
      ELSE IF (LQUERY) THEN
         RETURN
      END IF
<span class="comment">*</span><span class="comment">
</span><span class="comment">*</span><span class="comment">     Quick return if possible
</span><span class="comment">*</span><span class="comment">
</span>      IF( N.EQ.0 )
     $   RETURN
      IF( N.EQ.1 ) THEN
         IF( ICOMPZ.NE.0 )
     $      Z( 1, 1 ) = ONE
         RETURN
      END IF
<span class="comment">*</span><span class="comment">
</span><span class="comment">*</span><span class="comment">     If the following conditional clause is removed, then the routine
</span><span class="comment">*</span><span class="comment">     will use the Divide and Conquer routine to compute only the
</span><span class="comment">*</span><span class="comment">     eigenvalues, which requires (3N + 3N**2) real workspace and
</span><span class="comment">*</span><span class="comment">     (2 + 5N + 2N lg(N)) integer workspace.
</span><span class="comment">*</span><span class="comment">     Since on many architectures <a name="DSTERF.230"></a><a href="dsterf.f.html#DSTERF.1">DSTERF</a> is much faster than any other
</span><span class="comment">*</span><span class="comment">     algorithm for finding eigenvalues only, it is used here
</span><span class="comment">*</span><span class="comment">     as the default. If the conditional clause is removed, then
</span><span class="comment">*</span><span class="comment">     information on the size of workspace needs to be changed.
</span><span class="comment">*</span><span class="comment">
</span><span class="comment">*</span><span class="comment">     If COMPZ = 'N', use <a name="DSTERF.235"></a><a href="dsterf.f.html#DSTERF.1">DSTERF</a> to compute the eigenvalues.
</span><span class="comment">*</span><span class="comment">
</span>      IF( ICOMPZ.EQ.0 ) THEN
         CALL <a name="DSTERF.238"></a><a href="dsterf.f.html#DSTERF.1">DSTERF</a>( N, D, E, INFO )
         GO TO 50
      END IF
<span class="comment">*</span><span class="comment">
</span><span class="comment">*</span><span class="comment">     If N is smaller than the minimum divide size (SMLSIZ+1), then
</span><span class="comment">*</span><span class="comment">     solve the problem with another solver.
</span><span class="comment">*</span><span class="comment">
</span>      IF( N.LE.SMLSIZ ) THEN
<span class="comment">*</span><span class="comment">
</span>         CALL <a name="DSTEQR.247"></a><a href="dsteqr.f.html#DSTEQR.1">DSTEQR</a>( COMPZ, N, D, E, Z, LDZ, WORK, INFO )
<span class="comment">*</span><span class="comment">
</span>      ELSE
<span class="comment">*</span><span class="comment">
</span><span class="comment">*</span><span class="comment">        If COMPZ = 'V', the Z matrix must be stored elsewhere for later
</span><span class="comment">*</span><span class="comment">        use.
</span><span class="comment">*</span><span class="comment">
</span>         IF( ICOMPZ.EQ.1 ) THEN
            STOREZ = 1 + N*N
         ELSE
            STOREZ = 1
         END IF
<span class="comment">*</span><span class="comment">
</span>         IF( ICOMPZ.EQ.2 ) THEN
            CALL <a name="DLASET.261"></a><a href="dlaset.f.html#DLASET.1">DLASET</a>( <span class="string">'Full'</span>, N, N, ZERO, ONE, Z, LDZ )
         END IF
<span class="comment">*</span><span class="comment">
</span><span class="comment">*</span><span class="comment">        Scale.
</span><span class="comment">*</span><span class="comment">
</span>         ORGNRM = <a name="DLANST.266"></a><a href="dlanst.f.html#DLANST.1">DLANST</a>( <span class="string">'M'</span>, N, D, E )
         IF( ORGNRM.EQ.ZERO )
     $      GO TO 50
<span class="comment">*</span><span class="comment">