PetscErrorCode EPSSetUp_Arnoldi(EPS eps)
{
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = EPSSetDimensions_Default(eps,eps->nev,&eps->ncv,&eps->mpd);CHKERRQ(ierr);
if (eps->ncv>eps->nev+eps->mpd) SETERRQ(PetscObjectComm((PetscObject)eps),1,"The value of ncv must not be larger than nev+mpd");
if (!eps->max_it) eps->max_it = PetscMax(100,2*eps->n/eps->ncv);
if (!eps->which) { ierr = EPSSetWhichEigenpairs_Default(eps);CHKERRQ(ierr); }
if (eps->ishermitian && (eps->which==EPS_LARGEST_IMAGINARY || eps->which==EPS_SMALLEST_IMAGINARY)) SETERRQ(PetscObjectComm((PetscObject)eps),1,"Wrong value of eps->which");
if (!eps->extraction) {
ierr = EPSSetExtraction(eps,EPS_RITZ);CHKERRQ(ierr);
}
if (eps->arbitrary) SETERRQ(PetscObjectComm((PetscObject)eps),PETSC_ERR_SUP,"Arbitrary selection of eigenpairs not supported in this solver");
ierr = EPSAllocateSolution(eps,1);CHKERRQ(ierr);
ierr = EPS_SetInnerProduct(eps);CHKERRQ(ierr);
ierr = DSSetType(eps->ds,DSNHEP);CHKERRQ(ierr);
if (eps->extraction==EPS_REFINED || eps->extraction==EPS_REFINED_HARMONIC) {
ierr = DSSetRefined(eps->ds,PETSC_TRUE);CHKERRQ(ierr);
}
ierr = DSSetExtraRow(eps->ds,PETSC_TRUE);CHKERRQ(ierr);
ierr = DSAllocate(eps->ds,eps->ncv+1);CHKERRQ(ierr);
/* dispatch solve method */
if (eps->isgeneralized && eps->ishermitian && !eps->ispositive) SETERRQ(PetscObjectComm((PetscObject)eps),PETSC_ERR_SUP,"Requested method does not work for indefinite problems");
eps->ops->solve = EPSSolve_Arnoldi;
PetscFunctionReturn(0);
}
PetscErrorCode EPSSetUp_Lanczos(EPS eps)
{
EPS_LANCZOS *lanczos = (EPS_LANCZOS*)eps->data;
BVOrthogRefineType refine;
PetscReal eta;
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = EPSSetDimensions_Default(eps,eps->nev,&eps->ncv,&eps->mpd);CHKERRQ(ierr);
if (eps->ncv>eps->nev+eps->mpd) SETERRQ(PetscObjectComm((PetscObject)eps),1,"The value of ncv must not be larger than nev+mpd");
if (!eps->max_it) eps->max_it = PetscMax(100,2*eps->n/eps->ncv);
if (!eps->which) { ierr = EPSSetWhichEigenpairs_Default(eps);CHKERRQ(ierr); }
switch (eps->which) {
case EPS_LARGEST_IMAGINARY:
case EPS_SMALLEST_IMAGINARY:
case EPS_TARGET_IMAGINARY:
SETERRQ(PetscObjectComm((PetscObject)eps),1,"Wrong value of eps->which");
default: ; /* default case to remove warning */
}
if (!eps->extraction) {
ierr = EPSSetExtraction(eps,EPS_RITZ);CHKERRQ(ierr);
} else if (eps->extraction!=EPS_RITZ) SETERRQ(PetscObjectComm((PetscObject)eps),PETSC_ERR_SUP,"Unsupported extraction type");
if (eps->arbitrary) SETERRQ(PetscObjectComm((PetscObject)eps),PETSC_ERR_SUP,"Arbitrary selection of eigenpairs not supported in this solver");
ierr = EPSAllocateSolution(eps,1);CHKERRQ(ierr);
ierr = EPS_SetInnerProduct(eps);CHKERRQ(ierr);
if (lanczos->reorthog != EPS_LANCZOS_REORTHOG_FULL) {
ierr = BVGetOrthogonalization(eps->V,NULL,&refine,&eta);CHKERRQ(ierr);
ierr = BVSetOrthogonalization(eps->V,BV_ORTHOG_MGS,refine,eta);CHKERRQ(ierr);
ierr = PetscInfo(eps,"Switching to MGS orthogonalization\n");CHKERRQ(ierr);
}
if (lanczos->reorthog == EPS_LANCZOS_REORTHOG_SELECTIVE) {
ierr = BVDuplicate(eps->V,&lanczos->AV);CHKERRQ(ierr);
}
ierr = DSSetType(eps->ds,DSHEP);CHKERRQ(ierr);
ierr = DSSetCompact(eps->ds,PETSC_TRUE);CHKERRQ(ierr);
ierr = DSAllocate(eps->ds,eps->ncv+1);CHKERRQ(ierr);
if (lanczos->reorthog == EPS_LANCZOS_REORTHOG_LOCAL) {
ierr = EPSSetWorkVecs(eps,1);CHKERRQ(ierr);
}
/* dispatch solve method */
if (!eps->ishermitian) SETERRQ(PetscObjectComm((PetscObject)eps),PETSC_ERR_SUP,"Requested method is only available for Hermitian problems");
if (eps->isgeneralized && eps->ishermitian && !eps->ispositive) SETERRQ(PetscObjectComm((PetscObject)eps),PETSC_ERR_SUP,"Requested method does not work for indefinite problems");
eps->ops->solve = EPSSolve_Lanczos;
PetscFunctionReturn(0);
}
PetscErrorCode EPSSetUp_LAPACK(EPS eps)
{
PetscErrorCode ierr,ierra,ierrb;
PetscBool isshift,denseok=PETSC_FALSE;
Mat A,B,OP,Adense,Bdense;
PetscScalar shift,*Ap,*Bp;
PetscInt i,ld,nmat;
KSP ksp;
PC pc;
Vec v;
PetscFunctionBegin;
eps->ncv = eps->n;
if (eps->mpd) { ierr = PetscInfo(eps,"Warning: parameter mpd ignored\n");CHKERRQ(ierr); }
if (!eps->which) { ierr = EPSSetWhichEigenpairs_Default(eps);CHKERRQ(ierr); }
if (eps->balance!=EPS_BALANCE_NONE) { ierr = PetscInfo(eps,"Warning: balancing ignored\n");CHKERRQ(ierr); }
if (eps->extraction) { ierr = PetscInfo(eps,"Warning: extraction type ignored\n");CHKERRQ(ierr); }
ierr = EPSAllocateSolution(eps,0);CHKERRQ(ierr);
/* attempt to get dense representations of A and B separately */
ierr = PetscObjectTypeCompare((PetscObject)eps->st,STSHIFT,&isshift);CHKERRQ(ierr);
if (isshift) {
ierr = STGetNumMatrices(eps->st,&nmat);CHKERRQ(ierr);
ierr = STGetOperators(eps->st,0,&A);CHKERRQ(ierr);
if (nmat>1) { ierr = STGetOperators(eps->st,1,&B);CHKERRQ(ierr); }
PetscPushErrorHandler(PetscIgnoreErrorHandler,NULL);
ierra = SlepcMatConvertSeqDense(A,&Adense);CHKERRQ(ierr);
if (eps->isgeneralized) {
ierrb = SlepcMatConvertSeqDense(B,&Bdense);CHKERRQ(ierr);
} else {
ierrb = 0;
}
PetscPopErrorHandler();
denseok = (ierra == 0 && ierrb == 0)? PETSC_TRUE: PETSC_FALSE;
} else Adense = NULL;
/* setup DS */
if (denseok) {
if (eps->isgeneralized) {
if (eps->ishermitian) {
if (eps->ispositive) {
ierr = DSSetType(eps->ds,DSGHEP);CHKERRQ(ierr);
} else {
ierr = DSSetType(eps->ds,DSGNHEP);CHKERRQ(ierr); /* TODO: should be DSGHIEP */
}
} else {
ierr = DSSetType(eps->ds,DSGNHEP);CHKERRQ(ierr);
}
} else {
if (eps->ishermitian) {
ierr = DSSetType(eps->ds,DSHEP);CHKERRQ(ierr);
} else {
ierr = DSSetType(eps->ds,DSNHEP);CHKERRQ(ierr);
}
}
} else {
ierr = DSSetType(eps->ds,DSNHEP);CHKERRQ(ierr);
}
ierr = DSAllocate(eps->ds,eps->ncv);CHKERRQ(ierr);
ierr = DSGetLeadingDimension(eps->ds,&ld);CHKERRQ(ierr);
ierr = DSSetDimensions(eps->ds,eps->ncv,0,0,0);CHKERRQ(ierr);
if (denseok) {
ierr = STGetShift(eps->st,&shift);CHKERRQ(ierr);
if (shift != 0.0) {
ierr = MatShift(Adense,shift);CHKERRQ(ierr);
}
/* use dummy pc and ksp to avoid problems when B is not positive definite */
ierr = STGetKSP(eps->st,&ksp);CHKERRQ(ierr);
ierr = KSPSetType(ksp,KSPPREONLY);CHKERRQ(ierr);
ierr = KSPGetPC(ksp,&pc);CHKERRQ(ierr);
ierr = PCSetType(pc,PCNONE);CHKERRQ(ierr);
} else {
ierr = PetscInfo(eps,"Using slow explicit operator\n");CHKERRQ(ierr);
ierr = STComputeExplicitOperator(eps->st,&OP);CHKERRQ(ierr);
ierr = MatDestroy(&Adense);CHKERRQ(ierr);
ierr = SlepcMatConvertSeqDense(OP,&Adense);CHKERRQ(ierr);
}
/* fill DS matrices */
ierr = VecCreateSeqWithArray(PETSC_COMM_SELF,1,ld,NULL,&v);CHKERRQ(ierr);
ierr = DSGetArray(eps->ds,DS_MAT_A,&Ap);CHKERRQ(ierr);
for (i=0;i<ld;i++) {
ierr = VecPlaceArray(v,Ap+i*ld);CHKERRQ(ierr);
ierr = MatGetColumnVector(Adense,v,i);CHKERRQ(ierr);
ierr = VecResetArray(v);CHKERRQ(ierr);
}
ierr = DSRestoreArray(eps->ds,DS_MAT_A,&Ap);CHKERRQ(ierr);
if (denseok && eps->isgeneralized) {
ierr = DSGetArray(eps->ds,DS_MAT_B,&Bp);CHKERRQ(ierr);
for (i=0;i<ld;i++) {
ierr = VecPlaceArray(v,Bp+i*ld);CHKERRQ(ierr);
ierr = MatGetColumnVector(Bdense,v,i);CHKERRQ(ierr);
ierr = VecResetArray(v);CHKERRQ(ierr);
}
ierr = DSRestoreArray(eps->ds,DS_MAT_B,&Bp);CHKERRQ(ierr);
}
ierr = VecDestroy(&v);CHKERRQ(ierr);
ierr = MatDestroy(&Adense);CHKERRQ(ierr);
if (!denseok) { ierr = MatDestroy(&OP);CHKERRQ(ierr); }
if (denseok && eps->isgeneralized) { ierr = MatDestroy(&Bdense);CHKERRQ(ierr); }
PetscFunctionReturn(0);
}
PetscErrorCode EPSSetUp_KrylovSchur(EPS eps)
{
PetscErrorCode ierr;
EPS_KRYLOVSCHUR *ctx = (EPS_KRYLOVSCHUR*)eps->data;
enum { EPS_KS_DEFAULT,EPS_KS_SYMM,EPS_KS_SLICE,EPS_KS_INDEF } variant;
PetscFunctionBegin;
/* spectrum slicing requires special treatment of default values */
if (eps->which==EPS_ALL) {
ierr = EPSSetUp_KrylovSchur_Slice(eps);CHKERRQ(ierr);
} else {
ierr = EPSSetDimensions_Default(eps,eps->nev,&eps->ncv,&eps->mpd);CHKERRQ(ierr);
if (eps->ncv>eps->nev+eps->mpd) SETERRQ(PetscObjectComm((PetscObject)eps),1,"The value of ncv must not be larger than nev+mpd");
if (!eps->max_it) eps->max_it = PetscMax(100,2*eps->n/eps->ncv);
if (!eps->which) { ierr = EPSSetWhichEigenpairs_Default(eps);CHKERRQ(ierr); }
}
if (eps->isgeneralized && eps->ishermitian && !eps->ispositive && eps->arbitrary) SETERRQ(PetscObjectComm((PetscObject)eps),PETSC_ERR_SUP,"Arbitrary selection of eigenpairs not implemented for indefinite problems");
if (eps->ishermitian && eps->ispositive && (eps->which==EPS_LARGEST_IMAGINARY || eps->which==EPS_SMALLEST_IMAGINARY)) SETERRQ(PetscObjectComm((PetscObject)eps),1,"Wrong value of eps->which");
if (!eps->extraction) {
ierr = EPSSetExtraction(eps,EPS_RITZ);CHKERRQ(ierr);
} else if (eps->extraction!=EPS_RITZ && eps->extraction!=EPS_HARMONIC)
SETERRQ(PetscObjectComm((PetscObject)eps),PETSC_ERR_SUP,"Unsupported extraction type");
if (!ctx->keep) ctx->keep = 0.5;
ierr = EPSAllocateSolution(eps,1);CHKERRQ(ierr);
ierr = EPS_SetInnerProduct(eps);CHKERRQ(ierr);
if (eps->arbitrary) {
ierr = EPSSetWorkVecs(eps,2);CHKERRQ(ierr);
} else if (eps->ishermitian && !eps->ispositive){
ierr = EPSSetWorkVecs(eps,1);CHKERRQ(ierr);
}
/* dispatch solve method */
if (eps->ishermitian) {
if (eps->which==EPS_ALL) {
if (eps->isgeneralized && !eps->ispositive) SETERRQ(PetscObjectComm((PetscObject)eps),PETSC_ERR_SUP,"Spectrum slicing not implemented for indefinite problems");
else variant = EPS_KS_SLICE;
} else if (eps->isgeneralized && !eps->ispositive) {
variant = EPS_KS_INDEF;
} else {
switch (eps->extraction) {
case EPS_RITZ: variant = EPS_KS_SYMM; break;
case EPS_HARMONIC: variant = EPS_KS_DEFAULT; break;
default: SETERRQ(PetscObjectComm((PetscObject)eps),PETSC_ERR_SUP,"Unsupported extraction type");
}
}
} else {
switch (eps->extraction) {
case EPS_RITZ: variant = EPS_KS_DEFAULT; break;
case EPS_HARMONIC: variant = EPS_KS_DEFAULT; break;
default: SETERRQ(PetscObjectComm((PetscObject)eps),PETSC_ERR_SUP,"Unsupported extraction type");
}
}
switch (variant) {
case EPS_KS_DEFAULT:
eps->ops->solve = EPSSolve_KrylovSchur_Default;
eps->ops->computevectors = EPSComputeVectors_Schur;
ierr = DSSetType(eps->ds,DSNHEP);CHKERRQ(ierr);
ierr = DSAllocate(eps->ds,eps->ncv+1);CHKERRQ(ierr);
break;
case EPS_KS_SYMM:
eps->ops->solve = EPSSolve_KrylovSchur_Symm;
eps->ops->computevectors = EPSComputeVectors_Hermitian;
ierr = DSSetType(eps->ds,DSHEP);CHKERRQ(ierr);
ierr = DSSetCompact(eps->ds,PETSC_TRUE);CHKERRQ(ierr);
ierr = DSSetExtraRow(eps->ds,PETSC_TRUE);CHKERRQ(ierr);
ierr = DSAllocate(eps->ds,eps->ncv+1);CHKERRQ(ierr);
break;
case EPS_KS_SLICE:
eps->ops->solve = EPSSolve_KrylovSchur_Slice;
eps->ops->computevectors = NULL;
ierr = DSSetType(eps->ds,DSHEP);CHKERRQ(ierr);
ierr = DSSetCompact(eps->ds,PETSC_TRUE);CHKERRQ(ierr);
ierr = DSAllocate(eps->ds,ctx->ncv+1);CHKERRQ(ierr);
break;
case EPS_KS_INDEF:
eps->ops->solve = EPSSolve_KrylovSchur_Indefinite;
eps->ops->computevectors = EPSComputeVectors_Indefinite;
ierr = DSSetType(eps->ds,DSGHIEP);CHKERRQ(ierr);
ierr = DSSetCompact(eps->ds,PETSC_TRUE);CHKERRQ(ierr);
ierr = DSAllocate(eps->ds,eps->ncv+1);CHKERRQ(ierr);
break;
default: SETERRQ(PetscObjectComm((PetscObject)eps),1,"Unexpected error");
}
PetscFunctionReturn(0);
}
