PetscErrorCode FNView_Rational(FN fn,PetscViewer viewer)
{
PetscErrorCode ierr;
PetscBool isascii;
PetscInt i;
char str[50];
PetscFunctionBegin;
ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isascii);CHKERRQ(ierr);
if (isascii) {
if (!fn->nb) {
if (!fn->na) {
ierr = PetscViewerASCIIPrintf(viewer," Constant: 1.0\n");CHKERRQ(ierr);
} else if (fn->na==1) {
ierr = SlepcSNPrintfScalar(str,50,fn->alpha[0],PETSC_FALSE);CHKERRQ(ierr);
ierr = PetscViewerASCIIPrintf(viewer," Constant: %s\n",str);CHKERRQ(ierr);
} else {
ierr = PetscViewerASCIIPrintf(viewer," Polynomial: ");CHKERRQ(ierr);
for (i=0;i<fn->na-1;i++) {
ierr = SlepcSNPrintfScalar(str,50,fn->alpha[i],PETSC_TRUE);CHKERRQ(ierr);
ierr = PetscViewerASCIIPrintf(viewer,"%s*x^%1D",str,fn->na-i-1);CHKERRQ(ierr);
}
ierr = SlepcSNPrintfScalar(str,50,fn->alpha[fn->na-1],PETSC_TRUE);CHKERRQ(ierr);
ierr = PetscViewerASCIIPrintf(viewer,"%s\n",str);CHKERRQ(ierr);
}
} else if (!fn->na) {
ierr = PetscViewerASCIIPrintf(viewer," Inverse polinomial: 1 / (");CHKERRQ(ierr);
for (i=0;i<fn->nb-1;i++) {
ierr = SlepcSNPrintfScalar(str,50,fn->beta[i],PETSC_TRUE);CHKERRQ(ierr);
ierr = PetscViewerASCIIPrintf(viewer,"%s*x^%1D",str,fn->nb-i-1);CHKERRQ(ierr);
}
ierr = SlepcSNPrintfScalar(str,50,fn->beta[fn->nb-1],PETSC_TRUE);CHKERRQ(ierr);
ierr = PetscViewerASCIIPrintf(viewer,"%s)\n",str);CHKERRQ(ierr);
} else {
ierr = PetscViewerASCIIPrintf(viewer," Rational function: (");CHKERRQ(ierr);
for (i=0;i<fn->na-1;i++) {
ierr = SlepcSNPrintfScalar(str,50,fn->alpha[i],PETSC_TRUE);CHKERRQ(ierr);
ierr = PetscViewerASCIIPrintf(viewer,"%s*x^%1D",str,fn->na-i-1);CHKERRQ(ierr);
}
ierr = SlepcSNPrintfScalar(str,50,fn->alpha[fn->na-1],PETSC_TRUE);CHKERRQ(ierr);
ierr = PetscViewerASCIIPrintf(viewer,"%s) / (",str);CHKERRQ(ierr);
for (i=0;i<fn->nb-1;i++) {
ierr = SlepcSNPrintfScalar(str,50,fn->beta[i],PETSC_TRUE);CHKERRQ(ierr);
ierr = PetscViewerASCIIPrintf(viewer,"%s*x^%1D",str,fn->nb-i-1);CHKERRQ(ierr);
}
ierr = SlepcSNPrintfScalar(str,50,fn->beta[fn->nb-1],PETSC_TRUE);CHKERRQ(ierr);
ierr = PetscViewerASCIIPrintf(viewer,"%s)\n",str);CHKERRQ(ierr);
}
}
PetscFunctionReturn(0);
}
int main(int argc,char **argv)
{
Mat M,C,K,A[3]; /* problem matrices */
PEP pep; /* polynomial eigenproblem solver context */
PetscInt m=6,n,II,Istart,Iend,i,j;
PetscScalar z=1.0;
PetscReal h;
char str[50];
PetscErrorCode ierr;
SlepcInitialize(&argc,&argv,(char*)0,help);
ierr = PetscOptionsGetInt(NULL,"-m",&m,NULL);CHKERRQ(ierr);
if (m<2) SETERRQ(PETSC_COMM_SELF,1,"m must be at least 2");
ierr = PetscOptionsGetScalar(NULL,"-z",&z,NULL);CHKERRQ(ierr);
h = 1.0/m;
n = m*(m-1);
ierr = SlepcSNPrintfScalar(str,50,z,PETSC_FALSE);CHKERRQ(ierr);
ierr = PetscPrintf(PETSC_COMM_WORLD,"\nAcoustic wave 2-D, n=%D (m=%D), z=%s\n\n",n,m,str);CHKERRQ(ierr);
/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
Compute the matrices that define the eigensystem, (k^2*M+k*C+K)x=0
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */
/* K has a pattern similar to the 2D Laplacian */
ierr = MatCreate(PETSC_COMM_WORLD,&K);CHKERRQ(ierr);
ierr = MatSetSizes(K,PETSC_DECIDE,PETSC_DECIDE,n,n);CHKERRQ(ierr);
ierr = MatSetFromOptions(K);CHKERRQ(ierr);
ierr = MatSetUp(K);CHKERRQ(ierr);
ierr = MatGetOwnershipRange(K,&Istart,&Iend);CHKERRQ(ierr);
for (II=Istart;II<Iend;II++) {
i = II/m; j = II-i*m;
if (i>0) { ierr = MatSetValue(K,II,II-m,(j==m-1)?-0.5:-1.0,INSERT_VALUES);CHKERRQ(ierr); }
if (i<m-2) { ierr = MatSetValue(K,II,II+m,(j==m-1)?-0.5:-1.0,INSERT_VALUES);CHKERRQ(ierr); }
if (j>0) { ierr = MatSetValue(K,II,II-1,-1.0,INSERT_VALUES);CHKERRQ(ierr); }
if (j<m-1) { ierr = MatSetValue(K,II,II+1,-1.0,INSERT_VALUES);CHKERRQ(ierr); }
ierr = MatSetValue(K,II,II,(j==m-1)?2.0:4.0,INSERT_VALUES);CHKERRQ(ierr);
}
ierr = MatAssemblyBegin(K,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
ierr = MatAssemblyEnd(K,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
/* C is the zero matrix except for a few nonzero elements on the diagonal */
ierr = MatCreate(PETSC_COMM_WORLD,&C);CHKERRQ(ierr);
ierr = MatSetSizes(C,PETSC_DECIDE,PETSC_DECIDE,n,n);CHKERRQ(ierr);
ierr = MatSetFromOptions(C);CHKERRQ(ierr);
ierr = MatSetUp(C);CHKERRQ(ierr);
ierr = MatGetOwnershipRange(C,&Istart,&Iend);CHKERRQ(ierr);
for (i=Istart;i<Iend;i++) {
if (i%m==m-1) {
ierr = MatSetValue(C,i,i,-2*PETSC_PI*h/z,INSERT_VALUES);CHKERRQ(ierr);
}
}
ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
/* M is a diagonal matrix */
ierr = MatCreate(PETSC_COMM_WORLD,&M);CHKERRQ(ierr);
ierr = MatSetSizes(M,PETSC_DECIDE,PETSC_DECIDE,n,n);CHKERRQ(ierr);
ierr = MatSetFromOptions(M);CHKERRQ(ierr);
ierr = MatSetUp(M);CHKERRQ(ierr);
ierr = MatGetOwnershipRange(M,&Istart,&Iend);CHKERRQ(ierr);
for (i=Istart;i<Iend;i++) {
if (i%m==m-1) {
ierr = MatSetValue(M,i,i,2*PETSC_PI*PETSC_PI*h*h,INSERT_VALUES);CHKERRQ(ierr);
} else {
ierr = MatSetValue(M,i,i,4*PETSC_PI*PETSC_PI*h*h,INSERT_VALUES);CHKERRQ(ierr);
}
}
ierr = MatAssemblyBegin(M,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
ierr = MatAssemblyEnd(M,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
Create the eigensolver and solve the problem
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */
ierr = PEPCreate(PETSC_COMM_WORLD,&pep);CHKERRQ(ierr);
A[0] = K; A[1] = C; A[2] = M;
ierr = PEPSetOperators(pep,3,A);CHKERRQ(ierr);
ierr = PEPSetFromOptions(pep);CHKERRQ(ierr);
ierr = PEPSolve(pep);CHKERRQ(ierr);
/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
Display solution and clean up
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */
ierr = PEPPrintSolution(pep,NULL);CHKERRQ(ierr);
ierr = PEPDestroy(&pep);CHKERRQ(ierr);
ierr = MatDestroy(&M);CHKERRQ(ierr);
ierr = MatDestroy(&C);CHKERRQ(ierr);
ierr = MatDestroy(&K);CHKERRQ(ierr);
ierr = SlepcFinalize();CHKERRQ(ierr);
return 0;
}
int main(int argc,char **argv)
{
PetscErrorCode ierr;
FN fn;
PetscInt na,nb;
PetscScalar x,y,yp,p[10],q[10],five=5.0;
char strx[50],str[50];
SlepcInitialize(&argc,&argv,(char*)0,help);
ierr = FNCreate(PETSC_COMM_WORLD,&fn);CHKERRQ(ierr);
/* polynomial p(x) */
na = 5;
p[0] = -3.1; p[1] = 1.1; p[2] = 1.0; p[3] = -2.0; p[4] = 3.5;
ierr = FNSetType(fn,FNRATIONAL);CHKERRQ(ierr);
ierr = FNSetParameters(fn,na,p,0,NULL);CHKERRQ(ierr);
ierr = FNView(fn,NULL);CHKERRQ(ierr);
x = 2.2;
ierr = SlepcSNPrintfScalar(strx,50,x,PETSC_FALSE);CHKERRQ(ierr);
ierr = FNEvaluateFunction(fn,x,&y);CHKERRQ(ierr);
ierr = FNEvaluateDerivative(fn,x,&yp);CHKERRQ(ierr);
ierr = SlepcSNPrintfScalar(str,50,y,PETSC_FALSE);CHKERRQ(ierr);
ierr = PetscPrintf(PETSC_COMM_WORLD," f(%s)=%s\n",strx,str);CHKERRQ(ierr);
ierr = SlepcSNPrintfScalar(str,50,yp,PETSC_FALSE);CHKERRQ(ierr);
ierr = PetscPrintf(PETSC_COMM_WORLD," f'(%s)=%s\n",strx,str);CHKERRQ(ierr);
/* inverse of polynomial 1/q(x) */
nb = 3;
q[0] = -3.1; q[1] = 1.1; q[2] = 1.0;
ierr = FNSetType(fn,FNRATIONAL);CHKERRQ(ierr);
ierr = FNSetParameters(fn,0,NULL,nb,q);CHKERRQ(ierr);
ierr = FNView(fn,NULL);CHKERRQ(ierr);
x = 2.2;
ierr = SlepcSNPrintfScalar(strx,50,x,PETSC_FALSE);CHKERRQ(ierr);
ierr = FNEvaluateFunction(fn,x,&y);CHKERRQ(ierr);
ierr = FNEvaluateDerivative(fn,x,&yp);CHKERRQ(ierr);
ierr = SlepcSNPrintfScalar(str,50,y,PETSC_FALSE);CHKERRQ(ierr);
ierr = PetscPrintf(PETSC_COMM_WORLD," f(%s)=%s\n",strx,str);CHKERRQ(ierr);
ierr = SlepcSNPrintfScalar(str,50,yp,PETSC_FALSE);CHKERRQ(ierr);
ierr = PetscPrintf(PETSC_COMM_WORLD," f'(%s)=%s\n",strx,str);CHKERRQ(ierr);
/* rational p(x)/q(x) */
na = 2; nb = 3;
p[0] = -3.1; p[1] = 1.1;
q[0] = 1.0; q[1] = -2.0; q[2] = 3.5;
ierr = FNSetType(fn,FNRATIONAL);CHKERRQ(ierr);
ierr = FNSetParameters(fn,na,p,nb,q);CHKERRQ(ierr);
ierr = FNView(fn,NULL);CHKERRQ(ierr);
x = 2.2;
ierr = SlepcSNPrintfScalar(strx,50,x,PETSC_FALSE);CHKERRQ(ierr);
ierr = FNEvaluateFunction(fn,x,&y);CHKERRQ(ierr);
ierr = FNEvaluateDerivative(fn,x,&yp);CHKERRQ(ierr);
ierr = SlepcSNPrintfScalar(str,50,y,PETSC_FALSE);CHKERRQ(ierr);
ierr = PetscPrintf(PETSC_COMM_WORLD," f(%s)=%s\n",strx,str);CHKERRQ(ierr);
ierr = SlepcSNPrintfScalar(str,50,yp,PETSC_FALSE);CHKERRQ(ierr);
ierr = PetscPrintf(PETSC_COMM_WORLD," f'(%s)=%s\n",strx,str);CHKERRQ(ierr);
/* constant */
ierr = FNSetType(fn,FNRATIONAL);CHKERRQ(ierr);
ierr = FNSetParameters(fn,1,&five,0,NULL);CHKERRQ(ierr);
ierr = FNView(fn,NULL);CHKERRQ(ierr);
x = 2.2;
ierr = SlepcSNPrintfScalar(strx,50,x,PETSC_FALSE);CHKERRQ(ierr);
ierr = FNEvaluateFunction(fn,x,&y);CHKERRQ(ierr);
ierr = FNEvaluateDerivative(fn,x,&yp);CHKERRQ(ierr);
ierr = SlepcSNPrintfScalar(str,50,y,PETSC_FALSE);CHKERRQ(ierr);
ierr = PetscPrintf(PETSC_COMM_WORLD," f(%s)=%s\n",strx,str);CHKERRQ(ierr);
ierr = SlepcSNPrintfScalar(str,50,yp,PETSC_FALSE);CHKERRQ(ierr);
ierr = PetscPrintf(PETSC_COMM_WORLD," f'(%s)=%s\n",strx,str);CHKERRQ(ierr);
ierr = FNDestroy(&fn);CHKERRQ(ierr);
ierr = SlepcFinalize();
return 0;
}
/*@C
EPSView - Prints the EPS data structure.
Collective on EPS
Input Parameters:
+ eps - the eigenproblem solver context
- viewer - optional visualization context
Options Database Key:
. -eps_view - Calls EPSView() at end of EPSSolve()
Note:
The available visualization contexts include
+ PETSC_VIEWER_STDOUT_SELF - standard output (default)
- PETSC_VIEWER_STDOUT_WORLD - synchronized standard
output where only the first processor opens
the file. All other processors send their
data to the first processor to print.
The user can open an alternative visualization context with
PetscViewerASCIIOpen() - output to a specified file.
Level: beginner
.seealso: STView(), PetscViewerASCIIOpen()
@*/
PetscErrorCode EPSView(EPS eps,PetscViewer viewer)
{
PetscErrorCode ierr;
const char *type,*extr,*bal;
char str[50];
PetscBool isascii,ispower,isexternal,istrivial;
PetscFunctionBegin;
PetscValidHeaderSpecific(eps,EPS_CLASSID,1);
if (!viewer) viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)eps));
PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
PetscCheckSameComm(eps,1,viewer,2);
#if defined(PETSC_USE_COMPLEX)
#define HERM "hermitian"
#else
#define HERM "symmetric"
#endif
ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isascii);
CHKERRQ(ierr);
if (isascii) {
ierr = PetscObjectPrintClassNamePrefixType((PetscObject)eps,viewer);
CHKERRQ(ierr);
if (eps->ops->view) {
ierr = PetscViewerASCIIPushTab(viewer);
CHKERRQ(ierr);
ierr = (*eps->ops->view)(eps,viewer);
CHKERRQ(ierr);
ierr = PetscViewerASCIIPopTab(viewer);
CHKERRQ(ierr);
}
if (eps->problem_type) {
switch (eps->problem_type) {
case EPS_HEP:
type = HERM " eigenvalue problem";
break;
case EPS_GHEP:
type = "generalized " HERM " eigenvalue problem";
break;
case EPS_NHEP:
type = "non-" HERM " eigenvalue problem";
break;
case EPS_GNHEP:
type = "generalized non-" HERM " eigenvalue problem";
break;
case EPS_PGNHEP:
type = "generalized non-" HERM " eigenvalue problem with " HERM " positive definite B";
break;
case EPS_GHIEP:
type = "generalized " HERM "-indefinite eigenvalue problem";
break;
default:
SETERRQ(PetscObjectComm((PetscObject)eps),1,"Wrong value of eps->problem_type");
}
} else type = "not yet set";
ierr = PetscViewerASCIIPrintf(viewer," problem type: %s\n",type);
CHKERRQ(ierr);
if (eps->extraction) {
switch (eps->extraction) {
case EPS_RITZ:
extr = "Rayleigh-Ritz";
break;
case EPS_HARMONIC:
extr = "harmonic Ritz";
break;
case EPS_HARMONIC_RELATIVE:
extr = "relative harmonic Ritz";
break;
case EPS_HARMONIC_RIGHT:
extr = "right harmonic Ritz";
break;
case EPS_HARMONIC_LARGEST:
extr = "largest harmonic Ritz";
break;
case EPS_REFINED:
extr = "refined Ritz";
break;
case EPS_REFINED_HARMONIC:
extr = "refined harmonic Ritz";
break;
default:
SETERRQ(PetscObjectComm((PetscObject)eps),1,"Wrong value of eps->extraction");
}
ierr = PetscViewerASCIIPrintf(viewer," extraction type: %s\n",extr);
CHKERRQ(ierr);
}
if (!eps->ishermitian && eps->balance!=EPS_BALANCE_NONE) {
switch (eps->balance) {
case EPS_BALANCE_ONESIDE:
bal = "one-sided Krylov";
break;
case EPS_BALANCE_TWOSIDE:
bal = "two-sided Krylov";
break;
case EPS_BALANCE_USER:
bal = "user-defined matrix";
break;
default:
SETERRQ(PetscObjectComm((PetscObject)eps),1,"Wrong value of eps->balance");
}
ierr = PetscViewerASCIIPrintf(viewer," balancing enabled: %s",bal);
CHKERRQ(ierr);
if (eps->balance==EPS_BALANCE_ONESIDE || eps->balance==EPS_BALANCE_TWOSIDE) {
ierr = PetscViewerASCIIPrintf(viewer,", with its=%D",eps->balance_its);
CHKERRQ(ierr);
}
if (eps->balance==EPS_BALANCE_TWOSIDE && eps->balance_cutoff!=0.0) {
ierr = PetscViewerASCIIPrintf(viewer," and cutoff=%g",(double)eps->balance_cutoff);
CHKERRQ(ierr);
}
ierr = PetscViewerASCIIPrintf(viewer,"\n");
CHKERRQ(ierr);
}
ierr = PetscViewerASCIIPrintf(viewer," selected portion of the spectrum: ");
CHKERRQ(ierr);
ierr = SlepcSNPrintfScalar(str,50,eps->target,PETSC_FALSE);
CHKERRQ(ierr);
if (!eps->which) {
ierr = PetscViewerASCIIPrintf(viewer,"not yet set\n");
CHKERRQ(ierr);
} else switch (eps->which) {
case EPS_WHICH_USER:
ierr = PetscViewerASCIIPrintf(viewer,"user defined\n");
CHKERRQ(ierr);
break;
case EPS_TARGET_MAGNITUDE:
ierr = PetscViewerASCIIPrintf(viewer,"closest to target: %s (in magnitude)\n",str);
CHKERRQ(ierr);
break;
case EPS_TARGET_REAL:
ierr = PetscViewerASCIIPrintf(viewer,"closest to target: %s (along the real axis)\n",str);
CHKERRQ(ierr);
break;
case EPS_TARGET_IMAGINARY:
ierr = PetscViewerASCIIPrintf(viewer,"closest to target: %s (along the imaginary axis)\n",str);
CHKERRQ(ierr);
break;
case EPS_LARGEST_MAGNITUDE:
ierr = PetscViewerASCIIPrintf(viewer,"largest eigenvalues in magnitude\n");
CHKERRQ(ierr);
break;
case EPS_SMALLEST_MAGNITUDE:
ierr = PetscViewerASCIIPrintf(viewer,"smallest eigenvalues in magnitude\n");
CHKERRQ(ierr);
break;
case EPS_LARGEST_REAL:
ierr = PetscViewerASCIIPrintf(viewer,"largest real parts\n");
CHKERRQ(ierr);
break;
case EPS_SMALLEST_REAL:
ierr = PetscViewerASCIIPrintf(viewer,"smallest real parts\n");
CHKERRQ(ierr);
break;
case EPS_LARGEST_IMAGINARY:
ierr = PetscViewerASCIIPrintf(viewer,"largest imaginary parts\n");
CHKERRQ(ierr);
break;
case EPS_SMALLEST_IMAGINARY:
ierr = PetscViewerASCIIPrintf(viewer,"smallest imaginary parts\n");
CHKERRQ(ierr);
break;
case EPS_ALL:
ierr = PetscViewerASCIIPrintf(viewer,"all eigenvalues in interval [%g,%g]\n",(double)eps->inta,(double)eps->intb);
CHKERRQ(ierr);
break;
default:
SETERRQ(PetscObjectComm((PetscObject)eps),1,"Wrong value of eps->which");
}
if (eps->trueres) {
ierr = PetscViewerASCIIPrintf(viewer," computing true residuals explicitly\n");
CHKERRQ(ierr);
}
if (eps->trackall) {
ierr = PetscViewerASCIIPrintf(viewer," computing all residuals (for tracking convergence)\n");
CHKERRQ(ierr);
}
ierr = PetscViewerASCIIPrintf(viewer," number of eigenvalues (nev): %D\n",eps->nev);
CHKERRQ(ierr);
ierr = PetscViewerASCIIPrintf(viewer," number of column vectors (ncv): %D\n",eps->ncv);
CHKERRQ(ierr);
ierr = PetscViewerASCIIPrintf(viewer," maximum dimension of projected problem (mpd): %D\n",eps->mpd);
CHKERRQ(ierr);
ierr = PetscViewerASCIIPrintf(viewer," maximum number of iterations: %D\n",eps->max_it);
CHKERRQ(ierr);
ierr = PetscViewerASCIIPrintf(viewer," tolerance: %g\n",(double)eps->tol);
CHKERRQ(ierr);
ierr = PetscViewerASCIIPrintf(viewer," convergence test: ");
CHKERRQ(ierr);
switch (eps->conv) {
case EPS_CONV_ABS:
ierr = PetscViewerASCIIPrintf(viewer,"absolute\n");
CHKERRQ(ierr);
break;
case EPS_CONV_EIG:
ierr = PetscViewerASCIIPrintf(viewer,"relative to the eigenvalue\n");
CHKERRQ(ierr);
break;
case EPS_CONV_NORM:
ierr = PetscViewerASCIIPrintf(viewer,"relative to the eigenvalue and matrix norms\n");
CHKERRQ(ierr);
ierr = PetscViewerASCIIPrintf(viewer," computed matrix norms: norm(A)=%g",(double)eps->nrma);
CHKERRQ(ierr);
if (eps->isgeneralized) {
ierr = PetscViewerASCIIPrintf(viewer,", norm(B)=%g",(double)eps->nrmb);
CHKERRQ(ierr);
}
ierr = PetscViewerASCIIPrintf(viewer,"\n");
CHKERRQ(ierr);
break;
case EPS_CONV_USER:
ierr = PetscViewerASCIIPrintf(viewer,"user-defined\n");
CHKERRQ(ierr);
break;
}
if (eps->nini) {
ierr = PetscViewerASCIIPrintf(viewer," dimension of user-provided initial space: %D\n",PetscAbs(eps->nini));
CHKERRQ(ierr);
}
if (eps->nds) {
ierr = PetscViewerASCIIPrintf(viewer," dimension of user-provided deflation space: %D\n",PetscAbs(eps->nds));
CHKERRQ(ierr);
}
} else {
if (eps->ops->view) {
ierr = (*eps->ops->view)(eps,viewer);
CHKERRQ(ierr);
}
}
ierr = PetscObjectTypeCompareAny((PetscObject)eps,&isexternal,EPSARPACK,EPSBLZPACK,EPSTRLAN,EPSBLOPEX,EPSPRIMME,"");
CHKERRQ(ierr);
if (!isexternal) {
ierr = PetscViewerPushFormat(viewer,PETSC_VIEWER_ASCII_INFO);
CHKERRQ(ierr);
if (!eps->V) {
ierr = EPSGetBV(eps,&eps->V);
CHKERRQ(ierr);
}
ierr = BVView(eps->V,viewer);
CHKERRQ(ierr);
if (!eps->rg) {
ierr = EPSGetRG(eps,&eps->rg);
CHKERRQ(ierr);
}
ierr = RGIsTrivial(eps->rg,&istrivial);
CHKERRQ(ierr);
if (!istrivial) {
ierr = RGView(eps->rg,viewer);
CHKERRQ(ierr);
}
ierr = PetscObjectTypeCompare((PetscObject)eps,EPSPOWER,&ispower);
CHKERRQ(ierr);
if (!ispower) {
if (!eps->ds) {
ierr = EPSGetDS(eps,&eps->ds);
CHKERRQ(ierr);
}
ierr = DSView(eps->ds,viewer);
CHKERRQ(ierr);
}
ierr = PetscViewerPopFormat(viewer);
CHKERRQ(ierr);
}
if (!eps->st) {
ierr = EPSGetST(eps,&eps->st);
CHKERRQ(ierr);
}
ierr = STView(eps->st,viewer);
CHKERRQ(ierr);
PetscFunctionReturn(0);
}
int main(int argc,char **argv)
{
Mat M,C,K,A[3]; /* problem matrices */
PEP pep; /* polynomial eigenproblem solver context */
PetscInt n=10,Istart,Iend,i;
PetscScalar z=1.0;
char str[50];
PetscErrorCode ierr;
SlepcInitialize(&argc,&argv,(char*)0,help);
ierr = PetscOptionsGetInt(NULL,"-n",&n,NULL);CHKERRQ(ierr);
ierr = PetscOptionsGetScalar(NULL,"-z",&z,NULL);CHKERRQ(ierr);
ierr = SlepcSNPrintfScalar(str,50,z,PETSC_FALSE);CHKERRQ(ierr);
ierr = PetscPrintf(PETSC_COMM_WORLD,"\nAcoustic wave 1-D, n=%D z=%s\n\n",n,str);CHKERRQ(ierr);
/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
Compute the matrices that define the eigensystem, (k^2*M+k*C+K)x=0
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */
/* K is a tridiagonal */
ierr = MatCreate(PETSC_COMM_WORLD,&K);CHKERRQ(ierr);
ierr = MatSetSizes(K,PETSC_DECIDE,PETSC_DECIDE,n,n);CHKERRQ(ierr);
ierr = MatSetFromOptions(K);CHKERRQ(ierr);
ierr = MatSetUp(K);CHKERRQ(ierr);
ierr = MatGetOwnershipRange(K,&Istart,&Iend);CHKERRQ(ierr);
for (i=Istart;i<Iend;i++) {
if (i>0) {
ierr = MatSetValue(K,i,i-1,-1.0*n,INSERT_VALUES);CHKERRQ(ierr);
}
if (i<n-1) {
ierr = MatSetValue(K,i,i,2.0*n,INSERT_VALUES);CHKERRQ(ierr);
ierr = MatSetValue(K,i,i+1,-1.0*n,INSERT_VALUES);CHKERRQ(ierr);
} else {
ierr = MatSetValue(K,i,i,1.0*n,INSERT_VALUES);CHKERRQ(ierr);
}
}
ierr = MatAssemblyBegin(K,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
ierr = MatAssemblyEnd(K,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
/* C is the zero matrix but one element*/
ierr = MatCreate(PETSC_COMM_WORLD,&C);CHKERRQ(ierr);
ierr = MatSetSizes(C,PETSC_DECIDE,PETSC_DECIDE,n,n);CHKERRQ(ierr);
ierr = MatSetFromOptions(C);CHKERRQ(ierr);
ierr = MatSetUp(C);CHKERRQ(ierr);
ierr = MatGetOwnershipRange(C,&Istart,&Iend);CHKERRQ(ierr);
if (n-1>=Istart && n-1<Iend) {
ierr = MatSetValue(C,n-1,n-1,-2*PETSC_PI/z,INSERT_VALUES);CHKERRQ(ierr);
}
ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
/* M is a diagonal matrix */
ierr = MatCreate(PETSC_COMM_WORLD,&M);CHKERRQ(ierr);
ierr = MatSetSizes(M,PETSC_DECIDE,PETSC_DECIDE,n,n);CHKERRQ(ierr);
ierr = MatSetFromOptions(M);CHKERRQ(ierr);
ierr = MatSetUp(M);CHKERRQ(ierr);
ierr = MatGetOwnershipRange(M,&Istart,&Iend);CHKERRQ(ierr);
for (i=Istart;i<Iend;i++) {
if (i<n-1) {
ierr = MatSetValue(M,i,i,4*PETSC_PI*PETSC_PI/n,INSERT_VALUES);CHKERRQ(ierr);
} else {
ierr = MatSetValue(M,i,i,2*PETSC_PI*PETSC_PI/n,INSERT_VALUES);CHKERRQ(ierr);
}
}
ierr = MatAssemblyBegin(M,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
ierr = MatAssemblyEnd(M,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
Create the eigensolver and solve the problem
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */
ierr = PEPCreate(PETSC_COMM_WORLD,&pep);CHKERRQ(ierr);
A[0] = K; A[1] = C; A[2] = M;
ierr = PEPSetOperators(pep,3,A);CHKERRQ(ierr);
ierr = PEPSetFromOptions(pep);CHKERRQ(ierr);
ierr = PEPSolve(pep);CHKERRQ(ierr);
/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
Display solution and clean up
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */
ierr = PEPPrintSolution(pep,NULL);CHKERRQ(ierr);
ierr = PEPDestroy(&pep);CHKERRQ(ierr);
ierr = MatDestroy(&M);CHKERRQ(ierr);
ierr = MatDestroy(&C);CHKERRQ(ierr);
ierr = MatDestroy(&K);CHKERRQ(ierr);
ierr = SlepcFinalize();CHKERRQ(ierr);
return 0;
}
int main(int argc,char **argv)
{
Mat M,C,K; /* problem matrices */ /* polynomial eigenproblem solver context */
PetscInt n=10,Istart,Iend,i;
PetscScalar z=1.0;
char str[50],file[PETSC_MAX_PATH_LEN];
PetscErrorCode ierr;
PetscViewer view;
SlepcInitialize(&argc,&argv,(char*)0,help);
ierr = PetscOptionsGetInt(NULL,"-n",&n,NULL);CHKERRQ(ierr);
ierr = PetscOptionsGetScalar(NULL,"-z",&z,NULL);CHKERRQ(ierr);
ierr = SlepcSNPrintfScalar(str,50,z,PETSC_FALSE);CHKERRQ(ierr);
ierr = PetscPrintf(PETSC_COMM_WORLD,"\nAcoustic wave 1-D, n=%D z=%s\n\n",n,str);CHKERRQ(ierr);
/* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
Compute the matrices that define the eigensystem, (k^2*M+k*C+K)x=0
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */
/* K is a tridiagonal */
ierr = MatCreate(PETSC_COMM_WORLD,&K);CHKERRQ(ierr);
ierr = MatSetSizes(K,PETSC_DECIDE,PETSC_DECIDE,n,n);CHKERRQ(ierr);
ierr = MatSetFromOptions(K);CHKERRQ(ierr);
ierr = MatSetUp(K);CHKERRQ(ierr);
ierr = MatGetOwnershipRange(K,&Istart,&Iend);CHKERRQ(ierr);
for (i=Istart;i<Iend;i++) {
if (i>0) {
ierr = MatSetValue(K,i,i-1,-1.0*n,INSERT_VALUES);CHKERRQ(ierr);
}
if (i<n-1) {
ierr = MatSetValue(K,i,i,2.0*n,INSERT_VALUES);CHKERRQ(ierr);
ierr = MatSetValue(K,i,i+1,-1.0*n,INSERT_VALUES);CHKERRQ(ierr);
} else {
ierr = MatSetValue(K,i,i,1.0*n,INSERT_VALUES);CHKERRQ(ierr);
}
}
ierr = MatAssemblyBegin(K,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
ierr = MatAssemblyEnd(K,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
/* C is the zero matrix but one element*/
ierr = MatCreate(PETSC_COMM_WORLD,&C);CHKERRQ(ierr);
ierr = MatSetSizes(C,PETSC_DECIDE,PETSC_DECIDE,n,n);CHKERRQ(ierr);
ierr = MatSetFromOptions(C);CHKERRQ(ierr);
ierr = MatSetUp(C);CHKERRQ(ierr);
ierr = MatGetOwnershipRange(C,&Istart,&Iend);CHKERRQ(ierr);
if (n-1>=Istart && n-1<Iend) {
ierr = MatSetValue(C,n-1,n-1,-2*PETSC_PI/z,INSERT_VALUES);CHKERRQ(ierr);
}
ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
/* M is a diagonal matrix */
ierr = MatCreate(PETSC_COMM_WORLD,&M);CHKERRQ(ierr);
ierr = MatSetSizes(M,PETSC_DECIDE,PETSC_DECIDE,n,n);CHKERRQ(ierr);
ierr = MatSetFromOptions(M);CHKERRQ(ierr);
ierr = MatSetUp(M);CHKERRQ(ierr);
ierr = MatGetOwnershipRange(M,&Istart,&Iend);CHKERRQ(ierr);
for (i=Istart;i<Iend;i++) {
if (i<n-1) {
ierr = MatSetValue(M,i,i,4*PETSC_PI*PETSC_PI/n,INSERT_VALUES);CHKERRQ(ierr);
} else {
ierr = MatSetValue(M,i,i,2*PETSC_PI*PETSC_PI/n,INSERT_VALUES);CHKERRQ(ierr);
}
}
ierr = MatAssemblyBegin(M,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
ierr = MatAssemblyEnd(M,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
sprintf(file, "acoustic_wave_1d_m_n%d_z%.1f.petsc", n,z);
ierr = PetscViewerBinaryOpen(PETSC_COMM_WORLD,file,FILE_MODE_WRITE,&view);CHKERRQ(ierr);
ierr = MatView(M,view);CHKERRQ(ierr);
ierr = MatDestroy(&M);CHKERRQ(ierr);
sprintf(file, "acoustic_wave_1d_c_n%d_z%.1f.petsc", n,z);
ierr = PetscViewerBinaryOpen(PETSC_COMM_WORLD,file,FILE_MODE_WRITE,&view);CHKERRQ(ierr);
ierr = MatView(C,view);CHKERRQ(ierr);
ierr = MatDestroy(&C);CHKERRQ(ierr);
sprintf(file, "acoustic_wave_1d_k_n%d_z%.1f.petsc", n,z);
ierr = PetscViewerBinaryOpen(PETSC_COMM_WORLD,file,FILE_MODE_WRITE,&view);CHKERRQ(ierr);
ierr = MatView(K,view);CHKERRQ(ierr);
ierr = MatDestroy(&K);CHKERRQ(ierr);
ierr = SlepcFinalize();CHKERRQ(ierr);
return 0;
}
