Subversion Repositories slepc-dev

/*

   - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -

   SLEPc - Scalable Library for Eigenvalue Problem Computations

   Copyright (c) 2002-2011, Universitat Politecnica de Valencia, Spain

   This file is part of SLEPc.

   SLEPc is free software: you can redistribute it and/or modify it under  the

   terms of version 3 of the GNU Lesser General Public License as published by

   the Free Software Foundation.

   SLEPc  is  distributed in the hope that it will be useful, but WITHOUT  ANY

   WARRANTY;  without even the implied warranty of MERCHANTABILITY or  FITNESS

   FOR  A  PARTICULAR PURPOSE. See the GNU Lesser General Public  License  for

   more details.

   You  should have received a copy of the GNU Lesser General  Public  License

   along with SLEPc. If not, see <http://www.gnu.org/licenses/>.

   - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -

*/

#include <slepc-private/psimpl.h>      /*I "slepcps.h" I*/

#include <slepcblaslapack.h>

PetscErrorCode PSSolve_HEP_QR(PS,PetscScalar*,PetscScalar*);

PetscErrorCode PSSolve_HEP_MRRR(PS,PetscScalar*,PetscScalar*);

typedef PetscErrorCode (*solvefun)(PS,PetscScalar*,PetscScalar*);

static const solvefun solve[] = {

                     PSSolve_HEP_QR,

                     PSSolve_HEP_MRRR

};

static const char *methodname[] = {

                     "Implicit QR method (_steqr)",

                     "Relatively Robust Representations (_stevr)"

};

#undef __FUNCT__  

#define __FUNCT__ "PSAllocate_HEP"

PetscErrorCode PSAllocate_HEP(PS ps,PetscInt ld)

{

  PetscErrorCode ierr;

  PetscFunctionBegin;

  ierr = PSAllocateMat_Private(ps,PS_MAT_A);CHKERRQ(ierr);

  ierr = PSAllocateMat_Private(ps,PS_MAT_Q);CHKERRQ(ierr);

  ierr = PSAllocateMatReal_Private(ps,PS_MAT_T);CHKERRQ(ierr);

  PetscFunctionReturn(0);

}

/*   0       l           k                 n-1

    -----------------------------------------

    |*       ·           ·                  |

    |  *     ·           ·                  |

    |    *   ·           ·                  |

    |      * ·           ·                  |

    |· · · · o           o                  |

    |          o         o                  |

    |            o       o                  |

    |              o     o                  |

    |                o   o                  |

    |                  o o                  |

    |· · · · o o o o o o o x                |

    |                    x x x              |

    |                      x x x            |

    |                        x x x          |

    |                          x x x        |

    |                            x x x      |

    |                              x x x    |

    |                                x x x  |

    |                                  x x x|

    |                                    x x|

    -----------------------------------------

*/

#undef __FUNCT__  

#define __FUNCT__ "PSSwitchFormat_HEP"

PetscErrorCode PSSwitchFormat_HEP(PS ps,PetscBool tocompact)

{

  PetscReal   *T = ps->rmat[PS_MAT_T];

  PetscScalar *A = ps->mat[PS_MAT_A];

  PetscInt    i,n=ps->n,k=ps->k,ld=ps->ld;

  PetscFunctionBegin;

  if (ps->compact==tocompact) PetscFunctionReturn(0);

  if (tocompact) { /* switch from dense (arrow) to compact storage */

    for (i=0;i<k;i++) {

      T[i] = PetscRealPart(A[i+i*ld]);

      T[i+ld] = PetscRealPart(A[k+i*ld]);

    }

    for (i=k;i<n;i++) {

      T[i] = PetscRealPart(A[i+i*ld]);

      T[i+ld] = PetscRealPart(A[i+1+i*ld]);

    }

  } else { /* switch from compact (arrow) to dense storage */

    for (i=0;i<k;i++) {

      A[i+i*ld] = T[i];

      A[k+i*ld] = T[i+ld];

      A[i+k*ld] = T[i+ld];

    }

    A[k+k*ld] = T[k];

    for (i=k+1;i<n;i++) {

      A[i+i*ld] = T[i];

      A[i-1+i*ld] = T[i-1+ld];

      A[i+(i-1)*ld] = T[i-1+ld];

    }

  }

  PetscFunctionReturn(0);

}

#undef __FUNCT__  

#define __FUNCT__ "PSView_HEP"

PetscErrorCode PSView_HEP(PS ps,PetscViewer viewer)

{

  PetscErrorCode    ierr;

  PetscViewerFormat format;

  PetscInt          i,j,r,c;

  PetscReal         value;

  PetscFunctionBegin;

  ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);

  if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) {

    ierr = PetscViewerASCIIPrintf(viewer,"solving the problem with: %s\n",methodname[ps->method]);CHKERRQ(ierr);

    PetscFunctionReturn(0);

  }

  if (ps->compact) {

    ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);

    if (format == PETSC_VIEWER_ASCII_MATLAB) {

      ierr = PetscViewerASCIIPrintf(viewer,"%% Size = %D %D\n",ps->n,ps->n);CHKERRQ(ierr);

      ierr = PetscViewerASCIIPrintf(viewer,"zzz = zeros(%D,3);\n",3*ps->n);CHKERRQ(ierr);

      ierr = PetscViewerASCIIPrintf(viewer,"zzz = [\n");CHKERRQ(ierr);

      for (i=0;i<ps->n;i++) {

        ierr = PetscViewerASCIIPrintf(viewer,"%D %D  %18.16e\n",i+1,i+1,*(ps->rmat[PS_MAT_T]+i));CHKERRQ(ierr);

      }

      for (i=0;i<ps->n-1;i++) {

        r = PetscMax(i+2,ps->k+1);

        c = i+1;

        ierr = PetscViewerASCIIPrintf(viewer,"%D %D  %18.16e\n",r,c,*(ps->rmat[PS_MAT_T]+ps->ld+i));CHKERRQ(ierr);

        ierr = PetscViewerASCIIPrintf(viewer,"%D %D  %18.16e\n",c,r,*(ps->rmat[PS_MAT_T]+ps->ld+i));CHKERRQ(ierr);

      }

      ierr = PetscViewerASCIIPrintf(viewer,"];\n%s = spconvert(zzz);\n",PSMatName[PS_MAT_T]);CHKERRQ(ierr);

    } else {

      for (i=0;i<ps->n;i++) {

        for (j=0;j<ps->n;j++) {

          if (i==j) value = *(ps->rmat[PS_MAT_T]+i);

          else if ((i<ps->k && j==ps->k) || (i==ps->k && j<ps->k)) value = *(ps->rmat[PS_MAT_T]+ps->ld+PetscMin(i,j));

          else if (i==j+1 && i>ps->k) value = *(ps->rmat[PS_MAT_T]+ps->ld+i-1);

          else if (i+1==j && j>ps->k) value = *(ps->rmat[PS_MAT_T]+ps->ld+j-1);

          else value = 0.0;

          ierr = PetscViewerASCIIPrintf(viewer," %18.16e ",value);CHKERRQ(ierr);

        }

        ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);

      }

    }

    ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);

    ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);

  } else {

    ierr = PSViewMat_Private(ps,viewer,PS_MAT_A);CHKERRQ(ierr);

  }

  if (ps->state>PS_STATE_INTERMEDIATE) {

    ierr = PSViewMat_Private(ps,viewer,PS_MAT_Q);CHKERRQ(ierr);

  }

  PetscFunctionReturn(0);

}

#undef __FUNCT__  

#define __FUNCT__ "PSVectors_HEP"

PetscErrorCode PSVectors_HEP(PS ps,PSMatType mat,PetscInt *j,PetscReal *rnorm)

{

  PetscScalar    *Q = ps->mat[PS_MAT_Q];

  PetscInt       ld = ps->ld;

  PetscErrorCode ierr;

  PetscFunctionBegin;

  if (ps->state<PS_STATE_CONDENSED) SETERRQ(((PetscObject)ps)->comm,PETSC_ERR_ORDER,"Must call PSSolve() first");

  switch (mat) {

    case PS_MAT_X:

    case PS_MAT_Y:

      if (j) {

        ierr = PetscMemcpy(ps->mat[mat]+(*j)*ld,Q+(*j)*ld,ld*sizeof(PetscScalar));CHKERRQ(ierr);

      } else {

        ierr = PetscMemcpy(ps->mat[mat],Q,ld*ld*sizeof(PetscScalar));CHKERRQ(ierr);

      }

      if (rnorm) *rnorm = PetscAbsScalar(Q[ps->n-1+(*j)*ld]);

      break;

    case PS_MAT_U:

    case PS_MAT_VT:

      SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Not implemented yet");

      break;

    default:

      SETERRQ(((PetscObject)ps)->comm,PETSC_ERR_ARG_OUTOFRANGE,"Invalid mat parameter");

  }

  PetscFunctionReturn(0);

}

#undef __FUNCT__  

#define __FUNCT__ "ArrowTridiag"

/*

  ARROWTRIDIAG reduces a symmetric arrowhead matrix of the form

                [ d 0 0 0 e ]

                [ 0 d 0 0 e ]

            A = [ 0 0 d 0 e ]

                [ 0 0 0 d e ]

                [ e e e e d ]

  to tridiagonal form

                [ d e 0 0 0 ]

                [ e d e 0 0 ]

   T = Q'*A*Q = [ 0 e d e 0 ],

                [ 0 0 e d e ]

                [ 0 0 0 e d ]

  where Q is an orthogonal matrix. Rutishauser's algorithm is used to

  perform the reduction, which requires O(n**2) flops. The accumulation

  of the orthogonal factor Q, however, requires O(n**3) flops.

  Arguments

  =========

  N       (input) INTEGER

          The order of the matrix A.  N >= 0.

  D       (input/output) DOUBLE PRECISION array, dimension (N)

          On entry, the diagonal entries of the matrix A to be

          reduced.

          On exit, the diagonal entries of the reduced matrix T.

  E       (input/output) DOUBLE PRECISION array, dimension (N-1)

          On entry, the off-diagonal entries of the matrix A to be

          reduced.

          On exit, the subdiagonal entries of the reduced matrix T.

  Q       (input/output) DOUBLE PRECISION array, dimension (LDQ, N)

          On exit, the orthogonal matrix Q.

  LDQ     (input) INTEGER

          The leading dimension of the array Q.

  Note

  ====

  Based on Fortran code contributed by Daniel Kressner

*/

static PetscErrorCode ArrowTridiag(PetscBLASInt *n,PetscReal *d,PetscReal *e,PetscScalar *Q,PetscBLASInt *ldq)

{

  PetscBLASInt i,j,j2,ld=*ldq,one=1;

  PetscReal    c,s,p,off,temp;

  PetscFunctionBegin;

  if (*n<=2) PetscFunctionReturn(0);

  for (j=0;j<*n-2;j++) {

    /* Eliminate entry e(j) by a rotation in the planes (j,j+1) */

    temp = e[j+1];

    LAPACKlartg_(&temp,&e[j],&c,&s,&e[j+1]);

    s = -s;

    /* Apply rotation to diagonal elements */

    temp   = d[j+1];

    e[j]   = c*s*(temp-d[j]);

    d[j+1] = s*s*d[j] + c*c*temp;

    d[j]   = c*c*d[j] + s*s*temp;

    /* Apply rotation to Q */

    j2 = j+2;

    BLASrot_(&j2,Q+j*ld,&one,Q+(j+1)*ld,&one,&c,&s);

    /* Chase newly introduced off-diagonal entry to the top left corner */

    for (i=j-1;i>=0;i--) {

      off  = -s*e[i];

      e[i] = c*e[i];

      temp = e[i+1];

      LAPACKlartg_(&temp,&off,&c,&s,&e[i+1]);

      s = -s;

      temp = (d[i]-d[i+1])*s - 2.0*c*e[i];

      p = s*temp;

      d[i+1] += p;

      d[i] -= p;

      e[i] = -e[i] - c*temp;

      j2 = j+2;

      BLASrot_(&j2,Q+i*ld,&one,Q+(i+1)*ld,&one,&c,&s);

    }

  }

  PetscFunctionReturn(0);

}

#undef __FUNCT__  

#define __FUNCT__ "PSIntermediate_HEP_Flip"

/*

   Reduce to tridiagonal form by flipping the matrix and using standard

   LAPACK tridiagonal reduction

*/

static PetscErrorCode PSIntermediate_HEP_Flip(PS ps)

{

#if defined(SLEPC_MISSING_LAPACK_SYTRD) || defined(SLEPC_MISSING_LAPACK_ORGTR)

  PetscFunctionBegin;

  SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"SYTRD/ORGTR - Lapack routine is unavailable.");

#else

  PetscErrorCode ierr;

  PetscInt       i,j;

  PetscBLASInt   n1,n2,n3,lwork,info,l,n,ld,off;

  PetscScalar    *A,*S,*Q,*work,*tau;

  PetscReal      *d,*e;

  PetscFunctionBegin;

  n  = PetscBLASIntCast(ps->n);

  l  = PetscBLASIntCast(ps->l);

  ld = PetscBLASIntCast(ps->ld);

  n1 = PetscBLASIntCast(ps->k-l+1);  /* size of leading block, excluding locked */

  n2 = PetscBLASIntCast(n-ps->k-1);  /* size of trailing block */

  n3 = n1+n2;

  off = l+l*ld;

  A  = ps->mat[PS_MAT_A];

  Q  = ps->mat[PS_MAT_Q];

  d  = ps->rmat[PS_MAT_T];

  e  = ps->rmat[PS_MAT_T]+ld;

  ierr = PetscMemzero(Q,ld*ld*sizeof(PetscScalar));CHKERRQ(ierr);

  for (i=0;i<n;i++) Q[i+i*ld] = 1.0;

  if (ps->compact) {

    /* reduce to tridiagonal form */

    if (ps->state<PS_STATE_INTERMEDIATE) {

      ierr = PSAllocateMat_Private(ps,PS_MAT_W);CHKERRQ(ierr);

      S = ps->mat[PS_MAT_W];

      ierr = PetscMemzero(S,ld*ld*sizeof(PetscScalar));CHKERRQ(ierr);

      ierr = PSAllocateWork_Private(ps,ld+ld*ld,0,0);CHKERRQ(ierr);

      tau  = ps->work;

      work = ps->work+ld;

      lwork = ld*ld;

      /* Flip matrix S */

      for (i=l;i<n;i++) S[(n3-1-i+l)+(n3-1-i+l)*ld] = d[i];

      for (i=l;i<ps->k;i++) S[(n3-1-i+l)+(n3-1-ps->k+l)*ld] = e[i];

      for (i=ps->k;i<n-1;i++) S[(n3-1-i+l)+(n3-1-i-1+l)*ld] = e[i];

      /* Reduce (2,2)-block of flipped S to tridiagonal form */

      LAPACKsytrd_("L",&n1,S+n2+n2*ld,&ld,d+l,e+l,tau,work,&lwork,&info);

      if (info) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in Lapack xSYTRD %d",info);

      /* Flip back diag and subdiag, put them in d and e */

      for (i=0;i<n1-1;i++) {

        d[l+n1-i-1] = PetscRealPart(S[n2+i+(n2+i)*ld]);

        e[l+n1-i-2] = PetscRealPart(S[n2+i+1+(n2+i)*ld]);

      }

      d[l] = PetscRealPart(S[n3-1+(n3-1)*ld]);

      /* Compute the orthogonal matrix used for tridiagonalization */

      LAPACKorgtr_("L",&n1,S+n2+n2*ld,&ld,tau,work,&lwork,&info);

      if (info) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in Lapack xORGTR %d",info);

      /* Create full-size Q, flipped back to original order */

      for (i=0;i<n1;i++)

        for (j=0;j<n1;j++)

          Q[l+i+(l+j)*ld] = S[n3-i-1+(n3-j-1)*ld];

    }

  } else {

    for (i=0;i<l;i++) { d[i] = PetscRealPart(A[i+i*ld]); e[i] = 0.0; }

    if (ps->state<PS_STATE_INTERMEDIATE) {

      ierr = PSCopyMatrix_Private(ps,PS_MAT_Q,PS_MAT_A);CHKERRQ(ierr);

      ierr = PSAllocateWork_Private(ps,ld+ld*ld,0,0);CHKERRQ(ierr);

      tau  = ps->work;

      work = ps->work+ld;

      lwork = ld*ld;

      LAPACKsytrd_("L",&n3,Q+off,&ld,d+l,e+l,tau,work,&lwork,&info);

      if (info) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in Lapack xSYTRD %d",info);

      LAPACKorgtr_("L",&n3,Q+off,&ld,tau,work,&lwork,&info);

      if (info) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in Lapack xORGTR %d",info);

    } else {

      /* copy tridiagonal to d,e */

      for (i=l;i<n;i++) d[i] = PetscRealPart(A[i+i*ld]);

      for (i=l;i<n-1;i++) e[i] = PetscRealPart(A[(i+1)+i*ld]);

    }

  }

  PetscFunctionReturn(0);

#endif

}

#undef __FUNCT__  

#define __FUNCT__ "PSIntermediate_HEP"

/*

   Reduce to tridiagonal form by means of ArrowTridiag.

*/

static PetscErrorCode PSIntermediate_HEP(PS ps)

{

#if defined(SLEPC_MISSING_LAPACK_SYTRD) || defined(SLEPC_MISSING_LAPACK_ORGTR)

  PetscFunctionBegin;

  SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"SYTRD/ORGTR - Lapack routine is unavailable.");

#else

  PetscErrorCode ierr;

  PetscInt       i,j;

  PetscBLASInt   n1,n2,n3,lwork,info,l,n,ld,off;

  PetscScalar    *A,*S,*Q,*work,*tau;

  PetscReal      *d,*e;

  PetscFunctionBegin;

  n  = PetscBLASIntCast(ps->n);

  l  = PetscBLASIntCast(ps->l);

  ld = PetscBLASIntCast(ps->ld);

  n1 = PetscBLASIntCast(ps->k-l+1);  /* size of leading block, excluding locked */

  n2 = PetscBLASIntCast(n-ps->k-1);  /* size of trailing block */

  n3 = n1+n2;

  off = l+l*ld;

  A  = ps->mat[PS_MAT_A];

  Q  = ps->mat[PS_MAT_Q];

  d  = ps->rmat[PS_MAT_T];

  e  = ps->rmat[PS_MAT_T]+ld;

  ierr = PetscMemzero(Q,ld*ld*sizeof(PetscScalar));CHKERRQ(ierr);

  for (i=0;i<n;i++) Q[i+i*ld] = 1.0;

  if (ps->compact) {

    if (ps->state<PS_STATE_INTERMEDIATE) ArrowTridiag(&n1,d+l,e+l,Q+off,&ld);

  } else {

    for (i=0;i<l;i++) { d[i] = PetscRealPart(A[i+i*ld]); e[i] = 0.0; }

    if (ps->state<PS_STATE_INTERMEDIATE) {

      ierr = PSCopyMatrix_Private(ps,PS_MAT_Q,PS_MAT_A);CHKERRQ(ierr);

      ierr = PSAllocateWork_Private(ps,ld+ld*ld,0,0);CHKERRQ(ierr);

      tau  = ps->work;

      work = ps->work+ld;

      lwork = ld*ld;

      LAPACKsytrd_("L",&n3,Q+off,&ld,d+l,e+l,tau,work,&lwork,&info);

      if (info) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in Lapack xSYTRD %d",info);

      LAPACKorgtr_("L",&n3,Q+off,&ld,tau,work,&lwork,&info);

      if (info) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in Lapack xORGTR %d",info);

    } else {

      /* copy tridiagonal to d,e */

      for (i=l;i<n;i++) d[i] = PetscRealPart(A[i+i*ld]);

      for (i=l;i<n-1;i++) e[i] = PetscRealPart(A[(i+1)+i*ld]);

    }

  }

  PetscFunctionReturn(0);

#endif

}

#undef __FUNCT__  

#define __FUNCT__ "PSSolve_HEP_QR"

PetscErrorCode PSSolve_HEP_QR(PS ps,PetscScalar *wr,PetscScalar *wi)

{

#if defined(SLEPC_MISSING_LAPACK_STEQR)

  PetscFunctionBegin;

  SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"STEQR - Lapack routine is unavailable.");

#else

  PetscErrorCode ierr;

  PetscInt       i;

  PetscBLASInt   n1,n2,n3,info,l,n,ld,off;

  PetscScalar    *A,*Q;

  PetscReal      *d,*e;

#if defined(PETSC_USE_COMPLEX)

  PetscReal      *ritz;

#endif

  PetscFunctionBegin;

  n  = PetscBLASIntCast(ps->n);

  l  = PetscBLASIntCast(ps->l);

  ld = PetscBLASIntCast(ps->ld);

  n1 = PetscBLASIntCast(ps->k-l+1);  /* size of leading block, excluding locked */

  n2 = PetscBLASIntCast(n-ps->k-1);  /* size of trailing block */

  n3 = n1+n2;

  off = l+l*ld;

  A  = ps->mat[PS_MAT_A];

  Q  = ps->mat[PS_MAT_Q];

  d  = ps->rmat[PS_MAT_T];

  e  = ps->rmat[PS_MAT_T]+ld;

  /* Reduce to tridiagonal form */

  ierr = PSIntermediate_HEP(ps);CHKERRQ(ierr);

  /* Solve the tridiagonal eigenproblem */

  for (i=0;i<l;i++) wr[i] = d[i];

  ierr = PSAllocateWork_Private(ps,0,2*ld,0);CHKERRQ(ierr);

  LAPACKsteqr_("V",&n3,d+l,e+l,Q+off,&ld,ps->rwork,&info);

  if (info) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in Lapack xSTEQR %d",info);

  for (i=l;i<n;i++) wr[i] = d[i];

  if (ps->compact) {

    ierr = PetscMemzero(e,(n-1)*sizeof(PetscReal));CHKERRQ(ierr);

  } else {

    ierr = PetscMemzero(A,ld*ld*sizeof(PetscScalar));CHKERRQ(ierr);

    for (i=0;i<n;i++) A[i+i*ld] = d[i];

  }

  PetscFunctionReturn(0);

#endif

}

#undef __FUNCT__  

#define __FUNCT__ "PSSolve_HEP_MRRR"

PetscErrorCode PSSolve_HEP_MRRR(PS ps,PetscScalar *wr,PetscScalar *wi)

{

#if defined(SLEPC_MISSING_LAPACK_STEVR)

  PetscFunctionBegin;

  SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"STEVR - Lapack routine is unavailable.");

#else

  PetscErrorCode ierr;

  PetscInt       i;

  PetscBLASInt   n1,n2,n3,lwork,liwork,info,l,n,m,ld,off,il,iu,*isuppz;

  PetscScalar    *A,*Q,*W,one=1.0,zero=0.0;

  PetscReal      *d,*e,abstol=0.0,vl,vu;

#if defined(PETSC_USE_COMPLEX)

  PetscReal      *ritz;

#endif

  PetscFunctionBegin;

  n  = PetscBLASIntCast(ps->n);

  l  = PetscBLASIntCast(ps->l);

  ld = PetscBLASIntCast(ps->ld);

  n1 = PetscBLASIntCast(ps->k-l+1);  /* size of leading block, excluding locked */

  n2 = PetscBLASIntCast(n-ps->k-1);  /* size of trailing block */

  n3 = n1+n2;

  off = l+l*ld;

  A  = ps->mat[PS_MAT_A];

  Q  = ps->mat[PS_MAT_Q];

  d  = ps->rmat[PS_MAT_T];

  e  = ps->rmat[PS_MAT_T]+ld;

  /* Reduce to tridiagonal form */

  ierr = PSIntermediate_HEP(ps);CHKERRQ(ierr);

  /* Solve the tridiagonal eigenproblem */

  for (i=0;i<l;i++) wr[i] = d[i];

  if (ps->state<PS_STATE_INTERMEDIATE) {  /* Q contains useful info */

    ierr = PSAllocateMat_Private(ps,PS_MAT_W);CHKERRQ(ierr);

    W = ps->mat[PS_MAT_W];

    ierr = PSCopyMatrix_Private(ps,PS_MAT_W,PS_MAT_Q);CHKERRQ(ierr);

  }

#if defined(PETSC_USE_COMPLEX)

  ierr = PSAllocateMatReal_Private(ps,PS_MAT_Q);CHKERRQ(ierr);

#endif

  lwork = 20*ld;

  liwork = 10*ld;

  ierr = PSAllocateWork_Private(ps,0,lwork+ld,liwork+2*ld);CHKERRQ(ierr);

  isuppz = ps->iwork+liwork;

#if defined(PETSC_USE_COMPLEX)

  ritz = ps->rwork+lwork;

  LAPACKstevr_("V","A",&n3,d+l,e+l,&vl,&vu,&il,&iu,&abstol,&m,ritz+l,ps->rmat[PS_MAT_Q]+off,&ld,isuppz,ps->rwork,&lwork,ps->iwork,&liwork,&info);

  for (i=l;i<n;i++) wr[i] = ritz[i];

#else

  LAPACKstevr_("V","A",&n3,d+l,e+l,&vl,&vu,&il,&iu,&abstol,&m,wr+l,Q+off,&ld,isuppz,ps->rwork,&lwork,ps->iwork,&liwork,&info);

#endif

  if (info) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in Lapack DSTEVR %d",info);

#if defined(PETSC_USE_COMPLEX)

  for (i=l;i<n;i++)

    for (j=l;j<n;j++)

      Q[i+j*ld] = (ps->rmat[PS_MAT_Q])[i+j*ld];

#endif

  if (ps->state<PS_STATE_INTERMEDIATE) {  /* accumulate previous Q */

    BLASgemm_("N","N",&n3,&n3,&n3,&one,W+off,&ld,Q+off,&ld,&zero,A+off,&ld);

    ierr = PSCopyMatrix_Private(ps,PS_MAT_Q,PS_MAT_A);CHKERRQ(ierr);

  }

  for (i=l;i<n;i++) d[i] = wr[i];

  if (ps->compact) {

    ierr = PetscMemzero(e,(n-1)*sizeof(PetscReal));CHKERRQ(ierr);

  } else {

    ierr = PetscMemzero(A,ld*ld*sizeof(PetscScalar));CHKERRQ(ierr);

    for (i=0;i<n;i++) A[i+i*ld] = d[i];

  }

  PetscFunctionReturn(0);

#endif

}

#undef __FUNCT__  

#define __FUNCT__ "PSSort_HEP"

PetscErrorCode PSSort_HEP(PS ps,PetscScalar *wr,PetscScalar *wi,PetscErrorCode (*comp_func)(PetscScalar,PetscScalar,PetscScalar,PetscScalar,PetscInt*,void*),void *comp_ctx)

{

  PetscErrorCode ierr;

  PetscInt       n,l,i,*perm;

  PetscScalar    *A;

  PetscReal      *d;

  PetscFunctionBegin;

  n = ps->n;

  l = ps->l;

  d = ps->rmat[PS_MAT_T];

  ierr = PSAllocateWork_Private(ps,0,0,ps->ld);CHKERRQ(ierr);

  perm = ps->iwork;

  ierr = PSSortEigenvaluesReal_Private(ps,l,n,d,perm,comp_func,comp_ctx);CHKERRQ(ierr);

  for (i=l;i<n;i++) wr[i] = d[perm[i]];

  ierr = PSPermuteColumns_Private(ps,l,n,PS_MAT_Q,perm);CHKERRQ(ierr);

  if (ps->compact) {

    for (i=l;i<n;i++) d[i] = PetscRealPart(wr[i]);

  } else {

    A  = ps->mat[PS_MAT_A];

    for (i=l;i<n;i++) A[i+i*ps->ld] = wr[i];

  }

  PetscFunctionReturn(0);

}

#undef __FUNCT__  

#define __FUNCT__ "PSCond_HEP"

PetscErrorCode PSCond_HEP(PS ps,PetscReal *cond)

{

#if defined(PETSC_MISSING_LAPACK_GETRF) || defined(SLEPC_MISSING_LAPACK_GETRI) || defined(SLEPC_MISSING_LAPACK_LANGE) || defined(SLEPC_MISSING_LAPACK_LANHS)

  PetscFunctionBegin;

  SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"GETRF/GETRI/LANGE/LANHS - Lapack routines are unavailable.");

#else

  PetscErrorCode ierr;

  PetscScalar    *work;

  PetscReal      *rwork;