Line data Source code
1 : /*
2 : - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
3 : SLEPc - Scalable Library for Eigenvalue Problem Computations
4 : Copyright (c) 2002-, Universitat Politecnica de Valencia, Spain
5 :
6 : This file is part of SLEPc.
7 : SLEPc is distributed under a 2-clause BSD license (see LICENSE).
8 : - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
9 : */
10 : /*
11 : This example implements one of the problems found at
12 : NLEVP: A Collection of Nonlinear Eigenvalue Problems,
13 : The University of Manchester.
14 : The details of the collection can be found at:
15 : [1] T. Betcke et al., "NLEVP: A Collection of Nonlinear Eigenvalue
16 : Problems", ACM Trans. Math. Software 39(2), Article 7, 2013.
17 :
18 : The gun problem arises from model of a radio-frequency gun cavity, with
19 : the complex nonlinear function
20 : T(lambda) = K-lambda*M+i*lambda^(1/2)*W1+i*(lambda-108.8774^2)^(1/2)*W2
21 :
22 : Data files can be downloaded from https://slepc.upv.es/datafiles
23 : */
24 :
25 : static char help[] = "Radio-frequency gun cavity.\n\n"
26 : "The command line options are:\n"
27 : "-K <filename1> -M <filename2> -W1 <filename3> -W2 <filename4>, where filename1,..,filename4 are files containing the matrices in PETSc binary form defining the GUN problem.\n\n";
28 :
29 : #include <slepcnep.h>
30 :
31 : #define NMAT 4
32 : #define SIGMA 108.8774
33 :
34 1 : int main(int argc,char **argv)
35 : {
36 1 : Mat A[NMAT]; /* problem matrices */
37 1 : FN f[NMAT]; /* functions to define the nonlinear operator */
38 1 : FN ff[2]; /* auxiliary functions to define the nonlinear operator */
39 1 : NEP nep; /* nonlinear eigensolver context */
40 1 : PetscBool terse,flg;
41 1 : const char* string[NMAT]={"-K","-M","-W1","-W2"};
42 1 : char filename[PETSC_MAX_PATH_LEN];
43 1 : PetscScalar numer[2],sigma;
44 1 : PetscInt i;
45 1 : PetscViewer viewer;
46 :
47 1 : PetscFunctionBeginUser;
48 1 : PetscCall(SlepcInitialize(&argc,&argv,NULL,help));
49 :
50 1 : PetscCall(PetscPrintf(PETSC_COMM_WORLD,"GUN problem\n\n"));
51 : #if !defined(PETSC_USE_COMPLEX)
52 : SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_SUP,"This example requires complex scalars!");
53 : #endif
54 :
55 : /* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
56 : Load the problem matrices
57 : - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */
58 :
59 5 : for (i=0;i<NMAT;i++) {
60 4 : PetscCall(PetscOptionsGetString(NULL,NULL,string[i],filename,sizeof(filename),&flg));
61 4 : PetscCheck(flg,PETSC_COMM_WORLD,PETSC_ERR_USER_INPUT,"Must indicate a filename with the %s option",string[i]);
62 4 : PetscCall(PetscViewerBinaryOpen(PETSC_COMM_WORLD,filename,FILE_MODE_READ,&viewer));
63 4 : PetscCall(MatCreate(PETSC_COMM_WORLD,&A[i]));
64 4 : PetscCall(MatSetFromOptions(A[i]));
65 4 : PetscCall(MatLoad(A[i],viewer));
66 4 : PetscCall(PetscViewerDestroy(&viewer));
67 : }
68 :
69 : /* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
70 : Create the problem functions
71 : - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */
72 :
73 : /* f1=1 */
74 1 : PetscCall(FNCreate(PETSC_COMM_WORLD,&f[0]));
75 1 : PetscCall(FNSetType(f[0],FNRATIONAL));
76 1 : numer[0] = 1.0;
77 1 : PetscCall(FNRationalSetNumerator(f[0],1,numer));
78 :
79 : /* f2=-lambda */
80 1 : PetscCall(FNCreate(PETSC_COMM_WORLD,&f[1]));
81 1 : PetscCall(FNSetType(f[1],FNRATIONAL));
82 1 : numer[0] = -1.0; numer[1] = 0.0;
83 1 : PetscCall(FNRationalSetNumerator(f[1],2,numer));
84 :
85 : /* f3=i*sqrt(lambda) */
86 1 : PetscCall(FNCreate(PETSC_COMM_WORLD,&f[2]));
87 1 : PetscCall(FNSetType(f[2],FNSQRT));
88 1 : PetscCall(FNSetScale(f[2],1.0,PETSC_i));
89 :
90 : /* f4=i*sqrt(lambda-sigma^2) */
91 1 : sigma = SIGMA*SIGMA;
92 1 : PetscCall(FNCreate(PETSC_COMM_WORLD,&ff[0]));
93 1 : PetscCall(FNSetType(ff[0],FNSQRT));
94 1 : PetscCall(FNCreate(PETSC_COMM_WORLD,&ff[1]));
95 1 : PetscCall(FNSetType(ff[1],FNRATIONAL));
96 1 : numer[0] = 1.0; numer[1] = -sigma;
97 1 : PetscCall(FNRationalSetNumerator(ff[1],2,numer));
98 1 : PetscCall(FNCreate(PETSC_COMM_WORLD,&f[3]));
99 1 : PetscCall(FNSetType(f[3],FNCOMBINE));
100 1 : PetscCall(FNCombineSetChildren(f[3],FN_COMBINE_COMPOSE,ff[1],ff[0]));
101 1 : PetscCall(FNSetScale(f[3],1.0,PETSC_i));
102 :
103 : /* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
104 : Create the eigensolver and solve the problem
105 : - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */
106 :
107 1 : PetscCall(NEPCreate(PETSC_COMM_WORLD,&nep));
108 1 : PetscCall(NEPSetSplitOperator(nep,4,A,f,UNKNOWN_NONZERO_PATTERN));
109 1 : PetscCall(NEPSetFromOptions(nep));
110 :
111 1 : PetscCall(NEPSolve(nep));
112 :
113 : /* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
114 : Display solution and clean up
115 : - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */
116 :
117 : /* show detailed info unless -terse option is given by user */
118 1 : PetscCall(PetscOptionsHasName(NULL,NULL,"-terse",&terse));
119 1 : if (terse) PetscCall(NEPErrorView(nep,NEP_ERROR_RELATIVE,NULL));
120 : else {
121 0 : PetscCall(PetscViewerPushFormat(PETSC_VIEWER_STDOUT_WORLD,PETSC_VIEWER_ASCII_INFO_DETAIL));
122 0 : PetscCall(NEPConvergedReasonView(nep,PETSC_VIEWER_STDOUT_WORLD));
123 0 : PetscCall(NEPErrorView(nep,NEP_ERROR_RELATIVE,PETSC_VIEWER_STDOUT_WORLD));
124 0 : PetscCall(PetscViewerPopFormat(PETSC_VIEWER_STDOUT_WORLD));
125 : }
126 1 : PetscCall(NEPDestroy(&nep));
127 5 : for (i=0;i<NMAT;i++) {
128 4 : PetscCall(MatDestroy(&A[i]));
129 4 : PetscCall(FNDestroy(&f[i]));
130 : }
131 3 : for (i=0;i<2;i++) PetscCall(FNDestroy(&ff[i]));
132 1 : PetscCall(SlepcFinalize());
133 : return 0;
134 : }
135 :
136 : /*TEST
137 :
138 : build:
139 : requires: complex
140 :
141 : test:
142 : suffix: 1
143 : args: -K ${DATAFILESPATH}/matrices/complex/gun_K.petsc -M ${DATAFILESPATH}/matrices/complex/gun_M.petsc -W1 ${DATAFILESPATH}/matrices/complex/gun_W1.petsc -W2 ${DATAFILESPATH}/matrices/complex/gun_W2.petsc -nep_type nleigs -rg_type polygon -rg_polygon_vertices 12500-1i,120500-1i,120500+30000i,70000+30000i -nep_target 65000 -nep_nev 24 -terse
144 : requires: double complex datafilespath !defined(PETSC_USE_64BIT_INDICES) !defined(PETSCTEST_VALGRIND)
145 : timeoutfactor: 10
146 :
147 : TEST*/
|
