octave-sundials 2.5.0-3+b1 / usr / share / octave / site / m / sundialsTB / kinsol / examples_ser / mkinDiagon_kry.m

This file is indexed.

/usr/share/octave/site/m/sundialsTB/kinsol/examples_ser/mkinDiagon_kry.m is in octave-sundials 2.5.0-3+b1.

This file is owned by root:root, with mode 0o644.

The actual contents of the file can be viewed below. 

  1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
function mkinDiagon_kry
%mkinDiagon_kry - KINSOL example problem (serial, GMRES)
%   Simple diagonal test, using user-supplied preconditioner setup and 
%   solve routines.
%
%   This example does a basic test of the solver by solving the system:
%               f(y) = 0  for
%               f(y) = y(i)^2 - i^2
%
%   No scaling is done.
%   An approximate diagonal preconditioner is used.
%
%   See also: kindiag_sys kindag_pset kindiag_psol

% Radu Serban <radu@llnl.gov>
% Copyright (c) 2005, The Regents of the University of California.
% $Revision: 1.2 $Date: 2007/12/05 21:58:19 $

neq = 128;

strategy = 'None';

fnormtol = 1.0e-5;
scsteptol = 1.0e-4;
maxl   = 10;
maxrs  = 2;
msbset = 5;

data.P = [];

options = KINSetOptions('UserData', data,...
                        'Verbose',true,...
                        'FuncNormTol', fnormtol,...
                        'ScaledStepTol',scsteptol,...
                        'LinearSolver','GMRES',....
                        'KrylovMaxDim', maxl,...
                        'MaxNumRestarts', maxrs,...
                        'MaxNumSetups', msbset,...
                        'PrecSetupFn',@psetfn,...
                        'PrecSolveFn',@psolfn);
KINInit(@sysfn, neq, options);

y0 = 2.0*[1:neq]';
scale = ones(neq,1);

[status, y] = KINSol(y0, strategy, scale, scale);

if status < 0
  fprintf('KINSOL failed. status = %d\n',status);
else
  for i = 1:4:neq
    fprintf('%4d   |  %6.2f  %6.2f  %6.2f  %6.2f\n',...
            i, y(i), y(i+1), y(i+2), y(i+3));
  end
end

stats = KINGetStats;
ls_stats = stats.LSInfo;

stats
ls_stats

KINFree;


% ============================================================

function [fy, flag, new_data] = sysfn(y, data)

neq = length(y);
for i = 1:neq
  fy(i) = y(i)^2 - i^2;
end

new_data = [];  % data was not modified
flag = 0;       % success

% ============================================================

function [flag, new_data] = psetfn(y,yscale,fy,fscale,data)

neq = length(y);

for i = 1:neq
  P(i) = 0.5 / (y(i)+5.0);
end

new_data.P = P;  % updated P in data structure
flag = 0;        % success

% ============================================================

function [x, flag, new_data] = psolfn(y,yscale,fy,fscale,v,data)

P = data.P;

neq = length(y);

for i=1:neq
  x(i) = v(i) * P(i);
end

new_data = []; % data was not modified
flag = 0;      % success

APT Browse - Built by Thomas Orozco.