scilab/modules/arnoldi/tests/unit_tests/znaupd.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 d       = zeros(nev + 1, 1) + 0 * %i;
  17 z       = zeros(nx, nev) + 0* %i;
  18 resid   = zeros(nx, 1) + 0 * %i;
  19 v       = zeros(nx, ncv) + 0 * %i;
  20 workd   = zeros(3 * nx, 1) + 0 * %i;
  21 workev  = zeros(2 * ncv, 1) + 0 * %i;
  22 rwork   = zeros(ncv, 1);
  23 workl   = zeros(3 * ncv * ncv + 5 *ncv, 1) + 0 * %i;
  24 // Build the complex test matrix
  25 A            = diag(10 * ones(nx,1) + %i * 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 sigma = complex(0);
  37 info_znaupd = 0;
  38 // M A I N   L O O P (Reverse communication)
  39 while(ido <> 99)
  40   // Repeatedly call the routine ZNAUPD and take actions indicated by parameter IDO until
  41   // either convergence is indicated or maxitr has been exceeded.
  42   [ido, resid, v, iparam, ipntr, workd, workl, info_znaupd] = znaupd(ido, bmat, nx, which, nev, tol, resid, ncv, v, iparam, ipntr, workd, workl, rwork, info_znaupd);
  43   if(info_znaupd < 0)
  44     printf('\nError with znaupd, info = %d\n', info_znaupd);
  45     printf('Check the documentation of znaupd\n\n');
  46   end
  47   if(ido == -1 | ido == 1)
  48     // Perform matrix vector multiplication
  49     workd(ipntr(2):ipntr(2) + nx - 1) = A * workd(ipntr(1):ipntr(1) + nx - 1);
  50   end
  51 end
  52 // Post-Process using ZNEUPD.
  53 rvec    = 1;
  54 howmany = 'A';
  55 info_zneupd = 0;
  56 [d, z, resid, iparam, ipntr, workd, workl, rwork, info_zneupd] = zneupd(rvec, howmany, _select, d, z, sigma, workev, bmat, nx, which, nev, tol, resid, ncv, v, ...
  57                                                                     iparam, ipntr, workd, workl, rwork, info_zneupd);
  58 assert_checkalmostequal(A * z, z * diag(d(1:3)), sqrt(%eps), 1.e-10);