Add unitary tests for arnoldi
[scilab.git] / scilab / modules / arnoldi / tests / unit_tests / dnaupd.dia.ref
1 // =============================================================================
2 // Scilab ( http://www.scilab.org/ ) - This file is part of Scilab
3 // Copyright (C) 2012 - SE - Sylvestre Ledru
4 //
5 //  This file is distributed under the same license as the Scilab package.
6 // =============================================================================
7 nx    = 10;
8 nev   = 3;
9 ncv   = 6;
10 bmat  = 'I';
11 which = 'LM';
12 // Local Arrays
13 iparam  = zeros(11, 1);
14 ipntr   = zeros(14, 1);
15 _select = zeros(ncv, 1);
16 dr       = zeros(nev + 1, 1);
17 di      = zeros(nev + 1, 1);
18 z       = zeros(nx, nev + 1);
19 resid   = zeros(nx, 1);
20 v       = zeros(nx, ncv);
21 workd   = zeros(3 * nx, 1);
22 workev  = zeros(3 * ncv, 1);
23 workl   = zeros(3 * ncv * ncv + 6 * ncv, 1);
24 // Build the test matrix
25 A            = diag(10 * ones(nx, 1));
26 A(1:$-1,2:$) = A(1:$-1,2:$) + diag(6 * ones(nx-1,1));
27 A(2:$,1:$-1) = A(2:$,1:$-1) + diag(-6 * ones(nx-1,1));
28 tol    = 0;
29 ido    = 0;
30 ishfts = 1;
31 maxitr = 300;
32 mode1  = 1;
33 iparam(1) = ishfts;
34 iparam(3) = maxitr;
35 iparam(7) = mode1;
36 sigmar = 0; // the real part of the shift
37 sigmai = 0; // the imaginary part of the shift
38 info_dnaupd = 0;
39 // M A I N   L O O P (Reverse communication)
40 while(ido <> 99)
41     // Repeatedly call the routine DNAUPD and take actions indicated by parameter IDO until
42     // either convergence is indicated or maxitr has been exceeded.
43     [ido, resid, v, iparam, ipntr, workd, workl, info_dnaupd] = dnaupd(ido, bmat, nx, which, nev, tol, resid, ncv, v, iparam, ipntr, workd, workl, info_dnaupd);
44     if(info_dnaupd < 0)
45         printf('\nError with dnaupd, info = %d\n',info_dnaupd);
46         printf('Check the documentation of dnaupd\n\n');
47     end
48     if(ido == -1 | ido == 1)
49         // Perform matrix vector multiplication 
50         workd(ipntr(2):ipntr(2) + nx -1) = A * workd(ipntr(1):ipntr(1) + nx - 1);
51     end
52 end
53 // Post-Process using DNEUPD.
54 rvec    = 1;
55 howmany = 'A';
56 info_dneupd = 0;
57 [dr, di, z, resid, v, iparam, ipntr, workd, workl, info_dneupd] = dneupd(rvec, howmany, _select, dr, di, z, sigmar, sigmai, workev, ...
58 bmat, nx, which, nev, tol, resid, ncv, v, ...
59 iparam, ipntr, workd, workl, info_dneupd);
60 d = complex(dr, di);
61 d(nev+1) = [];
62 d = diag(d);
63 c1 = 1:2:nev + 1;
64 c2 = 2:2:nev + 1;
65 if(modulo(nev + 1, 2) == 1)
66     c1($) = [];
67 end
68 z(:,[c1, c2]) = [z(:,c1) + z(:,c2) * %i z(:,c1) - z(:,c2) * %i];
69 z(:,$) = [];
70 assert_checkalmostequal(A * z, z * d, sqrt(%eps), 1.e-10);