hide long namesshow long names
hide short namesshow short names
Integer type:  int32  int64  nag_int  show int32  show int32  show int64  show int64  show nag_int  show nag_int

PDF version (NAG web site, 64-bit version, 64-bit version)
Chapter Contents
Chapter Introduction
NAG Toolbox

NAG Toolbox: nag_ode_ivp_rk_zero_simple (d02bj)

Purpose

nag_ode_ivp_rk_zero_simple (d02bj) integrates a system of first-order ordinary differential equations over an interval with suitable initial conditions, using a fixed order Runge–Kutta method, until a user-specified function, if supplied, of the solution is zero, and returns the solution at points specified by you, if desired.

Syntax

[x, y, ifail] = d02bj(x, xend, y, fcn, tol, relabs, output, g, 'n', n)
[x, y, ifail] = nag_ode_ivp_rk_zero_simple(x, xend, y, fcn, tol, relabs, output, g, 'n', n)

Description

nag_ode_ivp_rk_zero_simple (d02bj) advances the solution of a system of ordinary differential equations
yi = fi(x,y1,y2,,yn),  i = 1,2,,n,
yi=fi(x,y1,y2,,yn),  i=1,2,,n,
from x = xx=x to x = xendx=xend using a fixed order Runge–Kutta method. The system is defined by fcn, which evaluates fifi in terms of xx and y = (y1,y2,,yn)y=(y1,y2,,yn). The initial values of y = (y1,y2,,yn)y=(y1,y2,,yn) must be given at x = xx=x.
The solution is returned via the output supplied by you and at points specified by you, if desired: this solution is obtained by C1C1 interpolation on solution values produced by the method. As the integration proceeds a check can be made on the user-specified function g(x,y)g(x,y) to determine an interval where it changes sign. The position of this sign change is then determined accurately by C1C1 interpolation to the solution. It is assumed that g(x,y)g(x,y) is a continuous function of the variables, so that a solution of g(x,y) = 0g(x,y)=0 can be determined by searching for a change in sign in g(x,y)g(x,y). The accuracy of the integration, the interpolation and, indirectly, of the determination of the position where g(x,y) = 0g(x,y)=0, is controlled by the parameters tol and relabs.

References

Shampine L F (1994) Numerical solution of ordinary differential equations Chapman and Hall

Parameters

Compulsory Input Parameters

1:     x – double scalar
The initial value of the independent variable xx.
2:     xend – double scalar
The final value of the independent variable. If xend < xxend<x, integration will proceed in the negative direction.
Constraint: xendxxendx.
3:     y(n) – double array
n, the dimension of the array, must satisfy the constraint n > 0n>0.
The initial values of the solution y1,y2,,yny1,y2,,yn at x = xx=x.
4:     fcn – function handle or string containing name of m-file
fcn must evaluate the functions fifi (i.e., the derivatives yiyi) for given values of its arguments x,y1,,ynx,y1,,yn.
[f] = fcn(x, y)

Input Parameters

1:     x – double scalar
xx, the value of the independent variable.
2:     y(nn) – double array
yiyi, for i = 1,2,,ni=1,2,,n, the value of the variable.

Output Parameters

1:     f(nn) – double array
The value of fifi, for i = 1,2,,ni=1,2,,n.
5:     tol – double scalar
A positive tolerance for controlling the error in the integration. Hence tol affects the determination of the position where g(x,y) = 0g(x,y)=0, if gg is supplied.
nag_ode_ivp_rk_zero_simple (d02bj) has been designed so that, for most problems, a reduction in tol leads to an approximately proportional reduction in the error in the solution. However, the actual relation between tol and the accuracy achieved cannot be guaranteed. You are strongly recommended to call nag_ode_ivp_rk_zero_simple (d02bj) with more than one value for tol and to compare the results obtained to estimate their accuracy. In the absence of any prior knowledge, you might compare the results obtained by calling nag_ode_ivp_rk_zero_simple (d02bj) with relabs = 'D'relabs='D' and with each of tol = 10.0ptol=10.0-p and tol = 10.0p1tol=10.0-p-1 where pp correct significant digits are required in the solution, yy. The accuracy of the value xx such that g(x,y) = 0g(x,y)=0 is indirectly controlled by varying tol. You should experiment to determine this accuracy.
Constraint: 10.0 × machine precision < tol < 0.0110.0×machine precision<tol<0.01.
6:     relabs – string (length ≥ 1)
The type of error control. At each step in the numerical solution an estimate of the local error, estest, is made. For the current step to be accepted the following condition must be satisfied:
est = max (ei / (τr × max (|yi|,τa))) 1.0
est =max(ei/( τr × max(|yi|,τa) )) 1.0
where τrτr and τaτa are defined by
relabs τrτr τaτa
'M' tol 1.0
'A' εrεr tol / εrtol/εr
'R' tol εaεa
'D' tol εaεa
where εrεr and εaεa are small machine-dependent numbers and eiei is an estimate of the local error at yiyi, computed internally. If the condition is not satisfied, the step size is reduced and the solution is recomputed on the current step. If you wish to measure the error in the computed solution in terms of the number of correct decimal places, then relabs should be set to 'A' on entry, whereas if the error requirement is in terms of the number of correct significant digits, then relabs should be set to 'R'. If you prefer a mixed error test, then relabs should be set to 'M', otherwise if you have no preference, relabs should be set to the default 'D'. Note that in this case 'D' is taken to be 'R'.
Constraint: relabs = 'M'relabs='M', 'A''A', 'R''R' or 'D''D'.
7:     output – function handle or string containing name of m-file
output permits access to intermediate values of the computed solution (for example to print or plot them), at successive user-specified points. It is initially called by nag_ode_ivp_rk_zero_simple (d02bj) with xsol = xxsol=x (the initial value of xx). You must reset xsol to the next point (between the current xsol and xend) where output is to be called, and so on at each call to output. If, after a call to output, the reset point xsol is beyond xend, nag_ode_ivp_rk_zero_simple (d02bj) will integrate to xend with no further calls to output; if a call to output is required at the point xsol = xendxsol=xend, then xsol must be given precisely the value xend.
[xsol] = output(xsol, y)

Input Parameters

1:     xsol – double scalar
The output value of the independent variable xx.
2:     y(nn) – double array
The computed solution at the point xsol.

Output Parameters

1:     xsol – double scalar
You must set xsol to the next value of xx at which output is to be called.
If you do not wish to access intermediate output, the actual parameter output must be the string 'd02bjx'. (nag_ode_ivp_rk_zero_simple_dummy_output (d02bjx) is included in the NAG Toolbox.)
8:     g – function handle or string containing name of m-file
g must evaluate the function g(x,y)g(x,y) for specified values x,yx,y. It specifies the function gg for which the first position xx where g(x,y) = 0g(x,y)=0 is to be found.
[result] = g(x, y)

Input Parameters

1:     x – double scalar
xx, the value of the independent variable.
2:     y(nn) – double array
yiyi, for i = 1,2,,ni=1,2,,n, the value of the variable.

Output Parameters

1:     result – double scalar
The result of the function.
If you do not require the root-finding option, the actual parameter g must be the string 'd02bjw'. (nag_ode_ivp_rk_zero_simple_dummy_g (d02bjw) is included in the NAG Toolbox.)

Optional Input Parameters

1:     n – int64int32nag_int scalar
Default: The dimension of the array y.
nn, the number of equations.
Constraint: n > 0n>0.

Input Parameters Omitted from the MATLAB Interface

w

Output Parameters

1:     x – double scalar
If gg is supplied by you, it contains the point where g(x,y) = 0g(x,y)=0, unless g(x,y)0g(x,y)0 anywhere on the range x to xend, in which case, x will contain xend (and the error indicator ifail = 6ifail=6 is set); if gg is not supplied by you it contains xend. However, if an error has occurred, it contains the value of xx at which the error occurred.
2:     y(n) – double array
The computed values of the solution at the final point x = xx=x.
3:     ifail – int64int32nag_int scalar
ifail = 0ifail=0 unless the function detects an error (see [Error Indicators and Warnings]).

Error Indicators and Warnings

Errors or warnings detected by the function:
  ifail = 1ifail=1
On entry,tol0.01tol0.01,
ortol is too small
orn0n0,
orrelabs'M'relabs'M', 'A''A', 'R''R' or 'D''D',
orx = xendx=xend.
  ifail = 2ifail=2
With the given value of tol, no further progress can be made across the integration range from the current point x = xx=x. (See Section [Further Comments] for a discussion of this error exit.) The components y(1),y(2),,y(n)y1,y2,,yn contain the computed values of the solution at the current point x = xx=x. If you have supplied gg, then no point at which g(x,y)g(x,y) changes sign has been located up to the point x = xx=x.
  ifail = 3ifail=3
tol is too small for nag_ode_ivp_rk_zero_simple (d02bj) to take an initial step. x and y(1),y(2),,y(n)y1,y2,,yn retain their initial values.
  ifail = 4ifail=4
xsol has not been reset or xsol lies behind x in the direction of integration, after the initial call to output, if the output option was selected.
  ifail = 5ifail=5
A value of xsol returned by the output has not been reset or lies behind the last value of xsol in the direction of integration, if the output option was selected.
  ifail = 6ifail=6
At no point in the range x to xend did the function g(x,y)g(x,y) change sign, if gg was supplied. It is assumed that g(x,y) = 0g(x,y)=0 has no solution.
  ifail = 7ifail=7
A serious error has occurred in an internal call to an interpolation function. Check all (sub)program calls and array dimensions. Seek expert help.

Accuracy

The accuracy of the computation of the solution vector y may be controlled by varying the local error tolerance tol. In general, a decrease in local error tolerance should lead to an increase in accuracy. You are advised to choose relabs = 'D'relabs='D' unless you have a good reason for a different choice.
If the problem is a root-finding one, then the accuracy of the root determined will depend on the properties of g(x,y)g(x,y) and on the values of tol and relabs. You should try to code g without introducing any unnecessary cancellation errors.

Further Comments

If more than one root is required, then to determine the second and later roots nag_ode_ivp_rk_zero_simple (d02bj) may be called again starting a short distance past the previously determined roots. Alternatively you may construct your own root-finding code using nag_roots_contfn_brent_rcomm (c05az), nag_ode_ivp_rkts_onestep (d02pf) and nag_ode_ivp_rkts_interp (d02ps).
If nag_ode_ivp_rk_zero_simple (d02bj) fails with ifail = 3ifail=3, then it can be called again with a larger value of tol if this has not already been tried. If the accuracy requested is really needed and cannot be obtained with this function, the system may be very stiff (see below) or so badly scaled that it cannot be solved to the required accuracy.
If nag_ode_ivp_rk_zero_simple (d02bj) fails with ifail = 2ifail=2, it is probable that it has been called with a value of tol which is so small that a solution cannot be obtained on the range x to xend. This can happen for well-behaved systems and very small values of tol. You should, however, consider whether there is a more fundamental difficulty. For example:
(a) in the region of a singularity (infinite value) of the solution, the function will usually stop with ifail = 2ifail=2, unless overflow occurs first. Numerical integration cannot be continued through a singularity, and analytic treatment should be considered;
(b) for ‘stiff’ equations where the solution contains rapidly decaying components, the function will use very small steps in xx (internally to nag_ode_ivp_rk_zero_simple (d02bj)) to preserve stability. This will exhibit itself by making the computing time excessively long, or occasionally by an exit with ifail = 2ifail=2. Runge–Kutta methods are not efficient in such cases, and you should try nag_ode_ivp_bdf_zero_simple (d02ej).

Example

This example illustrates the solution of four different problems. In each case the differential system (for a projectile) is
y = tanφ
v = (0.032tanφ)/v(0.02v)/(cosφ)
φ = (0.032)/(v2)
y=tanϕ v= -0.032tanϕv- 0.02v cosϕ ϕ= -0.032v2
over an interval x = 0.0x=0.0 to xend = 10.0xend=10.0 starting with values y = 0.5y=0.5, v = 0.5v=0.5 and φ = π / 5ϕ=π/5. We solve each of the following problems with local error tolerances 1.0e−41.0e−4 and 1.0e−51.0e−5.
(i) To integrate to x = 10.0x=10.0 producing intermediate output at intervals of 2.02.0 until a root is encountered where y = 0.0y=0.0.
(ii) As (i) but with no intermediate output.
(iii) As (i) but with no termination on a root-finding condition.
(iv) As (i) but with no intermediate output and no root-finding termination condition.
function nag_ode_ivp_rk_zero_simple_example
% For communication with save.
global ykeep ncall xkeep;

% Initialize variables and arrays.
x = 0;
xend = 10;
y = [0.5; 0.5; 0.6283185307179586];
tol = 0.0001;
relabs = 'Default';

xOut1 = [];
ncall = 0;
ykeep = zeros(1,length(y));
xkeep = zeros(1,1);

fprintf('nag_ode_ivp_rk_zero_simple example program results\n\n');

fprintf('Case 1: intermediate output, root-finding\n\n');
for j = 4:5
    tol = double(10)^(-j);
    disp(['Calculation with tol = ',num2str(tol)]);
    disp(' X     Y(1)       Y(2)       Y(3)');

    % output outputs intermediate values.
    [xOut, yOut, ifail] = nag_ode_ivp_rk_zero_simple(x, xend, y, @fcn, tol, ...
        relabs, @output, @g);
    if ifail ~= 0
        % Problems in integration.  Print message and exit.
        error( ...
         'Warning: nag_ode_ivp_rk_zero_simple returned with ifail = %1d ', ...
         ifail);
    end

    disp(' ');
    disp(['Root of Y(1) = 0.0 at ',num2str(xOut)]);
    disp('Solution is' );
    fprintf('  %8.4f   %8.4f   %8.4f\n\n', yOut(1), yOut(2), yOut(3));
end

fprintf('Case 2: no intermediate output, root-finding \n\n');
for j = 4:5
    tol = double(10)^(-j);
    disp(['Calculation with tol = ',num2str(tol)]);

    [xOut, yOut, ifail] = nag_ode_ivp_rk_zero_simple(x, xend, y, @fcn, tol, ...
        relabs, 'nag_ode_ivp_rk_zero_simple_dummy_output',@g);
    if ifail ~= 0
        % Problems in integration.  Print message and exit.
        error( ...
         'Warning: nag_ode_ivp_rk_zero_simple returned with ifail = %1d ', ...
         ifail);
    end

    disp(['Root of Y(1) = 0.0 at ',num2str(xOut)]);
    disp('Solution is' );
    fprintf('  %8.4f   %8.4f   %8.4f\n\n', yOut(1), yOut(2), yOut(3));

    % Store the x value for plotting.
    xOut1 = xOut;
end

fprintf('Case 3: intermediate output, no root-finding\n\n');
for j = 4:5
    tol = double(10)^(-j);
    disp(['Calculation with tol = ',num2str(tol)]);
    fprintf(' X     Y(1)       Y(2)       Y(3) \n');

    % save stores intermediate values in xkeep, ykeep, which are
    % plotted later (it also outputs them).
    [xOut, yOut, ifail] = nag_ode_ivp_rk_zero_simple(x, xend, y, @fcn, tol, ...
        relabs, @save,'nag_ode_ivp_rk_zero_simple_dummy_g');
    if ifail ~= 0
        % Problems in integration.  Print message and exit.
        error( ...
         'Warning: nag_ode_ivp_rk_zero_simple returned with ifail = %1d ', ...
         ifail);
    end
    fprintf('\n');
end

fprintf(['Case 4: no intermediate output, no root-finding ', ...
    '(integrate to xend)\n\n']);
for j = 4:5
    tol = double(10)^(-j);
    disp(['Calculation with tol = ',num2str(tol)]);
    disp(' X     Y(1)       Y(2)       Y(3)');
    fprintf('%2d   %8.4f   %8.4f   %8.4f\n', x, y(1), y(2), y(3));

    [xOut, yOut, ifail] = ...
        nag_ode_ivp_rk_zero_simple(x, xend, y, @fcn, tol, relabs, ...
        'nag_ode_ivp_rk_zero_simple_dummy_output','nag_ode_ivp_rk_zero_simple_dummy_g');
    if ifail ~= 0
        % Problems in integration.  Print message and exit.
        error( ...
         'Warning: nag_ode_ivp_rk_zero_simple returned with ifail = %1d ', ...
         ifail);
    end

    fprintf('%2d   %8.4f   %8.4f   %8.4f\n\n', xOut, yOut(1), ...
        yOut(2), yOut(3));
end
% Plot results.
nres = 0.5*length(xkeep);
xplot = xkeep(nres+1:2*nres);
yplot = ykeep(nres+1:2*nres, :);
fig = figure('Number', 'off');
display_plot(xplot, yplot, xOut1)

function xsolOut = save(xsol, y)
% For communication with main routine.
global ykeep ncall xkeep;

% This version of the intermediate output routine stores the values
% (so they can be plotted in the main routine).
ncall = ncall+1;
ykeep(ncall,:) = y;
xkeep(ncall,:) = xsol;
fprintf('%2d   %8.4f   %8.4f   %8.4f\n', xsol, y(1), y(2), y(3));
xsolOut = xsol + 2;
function xsolOut = output(xsol, y)
% Output intermediate values of solution.
fprintf('%2d   %8.4f   %8.4f   %8.4f\n', xsol, y(1), y(2), y(3));
xsolOut = xsol + 2;
function f = fcn(x,y)
% Evaluate the derivatives.
f = zeros(3,1);
f(1) = tan(y(3));
f(2) = -0.032*tan(y(3))/y(2) - 0.02*y(2)/cos(y(3));
f(3) = -0.032/y(2)^2;
function result = g(x,y)
% Evaluate g(x,y) when root-finding option is selected.
result = y(1);
function display_plot(xplot, yplot, xOut1)
% Formatting for title and axis labels.
titleFmt = {'FontName', 'Helvetica', 'FontWeight', 'Bold', 'FontSize', 14};
labFmt = {'FontName', 'Helvetica', 'FontWeight', 'Bold', 'FontSize', 13};
set(gca, 'FontSize', 13); % for legend, axis tick labels, etc.
% Plot the three curves.
plot(xplot, yplot(:,1), '-+', ...
     xplot, yplot(:,2), '--x', ...
     xplot, yplot(:,3), ':*');
% Mark the height=0 point.
do_stem(xplot, yplot, xOut1);
% Add title.
title('ODE Solution using Runge-Kutta with Root-finding', titleFmt{:});
% Label the axes.
xlabel('x', labFmt{:});
ylabel('Solution', labFmt{:});
% Add legend.
legend('height','velocity','angle','height = 0','Location','Best');
function do_stem(xplot, yplot, xOut1)
% Find the x bin that xOut1 lies in.
for i = 1:length(xplot)
    if xplot(i) > xOut1
        break
    end
end
% Use linear interpolation to find the corresponding y values on the
% two curves.
a1 = xplot(i)-xplot(i-1);
b1 = yplot(i,2)-yplot(i-1,2);
c1 = xOut1-xplot(i-1);
d1 = c1/a1;
e1 = d1*b1;
f1 = e1+yplot(i-1,2);
a2 = xplot(i)-xplot(i-1);
b2 = yplot(i,3)-yplot(i-1,3);
c2 = xOut1-xplot(i-1);
d2 = c2/a2;
e2 = d2*b2;
f2 = e2+yplot(i-1,3);
% Plot the line from the x axis to the two y values.
hold on
stem([xOut1,xOut1],[f1,0],'k:s');
hold on
stem([xOut1,xOut1],[f2,0],'k:s');
 
nag_ode_ivp_rk_zero_simple example program results

Case 1: intermediate output, root-finding

Calculation with tol = 0.0001
 X     Y(1)       Y(2)       Y(3)
 0     0.5000     0.5000     0.6283
 2     1.5493     0.4055     0.3066
 4     1.7423     0.3743    -0.1289
 6     1.0055     0.4173    -0.5507
 
Root of Y(1) = 0.0 at 7.2882
Solution is
   -0.0000     0.4749    -0.7601

Calculation with tol = 1e-05
 X     Y(1)       Y(2)       Y(3)
 0     0.5000     0.5000     0.6283
 2     1.5493     0.4055     0.3066
 4     1.7423     0.3743    -0.1289
 6     1.0055     0.4173    -0.5507
 
Root of Y(1) = 0.0 at 7.2883
Solution is
    0.0000     0.4749    -0.7601

Case 2: no intermediate output, root-finding 

Calculation with tol = 0.0001
Root of Y(1) = 0.0 at 7.2882
Solution is
   -0.0000     0.4749    -0.7601

Calculation with tol = 1e-05
Root of Y(1) = 0.0 at 7.2883
Solution is
    0.0000     0.4749    -0.7601

Case 3: intermediate output, no root-finding

Calculation with tol = 0.0001
 X     Y(1)       Y(2)       Y(3) 
 0     0.5000     0.5000     0.6283
 2     1.5493     0.4055     0.3066
 4     1.7423     0.3743    -0.1289
 6     1.0055     0.4173    -0.5507
 8    -0.7460     0.5130    -0.8537
10    -3.6283     0.6333    -1.0515

Calculation with tol = 1e-05
 X     Y(1)       Y(2)       Y(3) 
 0     0.5000     0.5000     0.6283
 2     1.5493     0.4055     0.3066
 4     1.7423     0.3743    -0.1289
 6     1.0055     0.4173    -0.5507
 8    -0.7459     0.5130    -0.8537
10    -3.6282     0.6333    -1.0515

Case 4: no intermediate output, no root-finding (integrate to xend)

Calculation with tol = 0.0001
 X     Y(1)       Y(2)       Y(3)
 0     0.5000     0.5000     0.6283
10    -3.6283     0.6333    -1.0515

Calculation with tol = 1e-05
 X     Y(1)       Y(2)       Y(3)
 0     0.5000     0.5000     0.6283
10    -3.6282     0.6333    -1.0515


function d02bj_example
% For communication with save.
global ykeep ncall xkeep;

% Initialize variables and arrays.
x = 0;
xend = 10;
y = [0.5; 0.5; 0.6283185307179586];
tol = 0.0001;
relabs = 'Default';

xOut1 = [];
ncall = 0;
ykeep = zeros(1,length(y));
xkeep = zeros(1,1);

fprintf('d02bj example program results\n\n');

fprintf('Case 1: intermediate output, root-finding\n\n');
for j = 4:5
    tol = double(10)^(-j);
    disp(['Calculation with tol = ',num2str(tol)]);
    disp(' X     Y(1)       Y(2)       Y(3)');

    % output outputs intermediate values.
    [xOut, yOut, ifail] = d02bj(x, xend, y, @fcn, tol, ...
        relabs, @output, @g);
    if ifail ~= 0
        % Problems in integration.  Print message and exit.
        error('Warning: d02bj returned with ifail = %1d',ifail);
    end

    disp(' ');
    disp(['Root of Y(1) = 0.0 at ',num2str(xOut)]);
    disp('Solution is' );
    fprintf('  %8.4f   %8.4f   %8.4f\n\n', yOut(1), yOut(2), yOut(3));
end

fprintf('Case 2: no intermediate output, root-finding \n\n');
for j = 4:5
    tol = double(10)^(-j);
    disp(['Calculation with tol = ',num2str(tol)]);

    [xOut, yOut, ifail] = d02bj(x, xend, y, @fcn, tol, ...
        relabs, 'd02bjx',@g);
    if ifail ~= 0
        % Problems in integration.  Print message and exit.
        error('Warning: d02bj returned with ifail = %1d',ifail);
    end

    disp(['Root of Y(1) = 0.0 at ',num2str(xOut)]);
    disp('Solution is' );
    fprintf('  %8.4f   %8.4f   %8.4f\n\n', yOut(1), yOut(2), yOut(3));

    % Store the x value for plotting.
    xOut1 = xOut;
end

fprintf('Case 3: intermediate output, no root-finding\n\n');
for j = 4:5
    tol = double(10)^(-j);
    disp(['Calculation with tol = ',num2str(tol)]);
    fprintf(' X     Y(1)       Y(2)       Y(3) \n');

    % save stores intermediate values in xkeep, ykeep, which are
    % plotted later (it also outputs them).
    [xOut, yOut, ifail] = d02bj(x, xend, y, @fcn, tol, ...
        relabs, @save,'d02bjw');
    if ifail ~= 0
        % Problems in integration.  Print message and exit.
        error('Warning: d02bj returned with ifail = %1d',ifail);
    end
    fprintf('\n');
end

fprintf(['Case 4: no intermediate output, no root-finding ', ...
    '(integrate to xend)\n\n']);
for j = 4:5
    tol = double(10)^(-j);
    disp(['Calculation with tol = ',num2str(tol)]);
    disp(' X     Y(1)       Y(2)       Y(3)');
    fprintf('%2d   %8.4f   %8.4f   %8.4f\n', x, y(1), y(2), y(3));

    [xOut, yOut, ifail] = ...
        d02bj(x, xend, y, @fcn, tol, relabs, 'd02bjx','d02bjw');
    if ifail ~= 0
        % Problems in integration.  Print message and exit.
        error('Warning: d02bj returned with ifail = %1d',ifail);
    end

    fprintf('%2d   %8.4f   %8.4f   %8.4f\n\n', xOut, yOut(1), ...
        yOut(2), yOut(3));
end
% Plot results.
nres = 0.5*length(xkeep);
xplot = xkeep(nres+1:2*nres);
yplot = ykeep(nres+1:2*nres, :);
fig = figure('Number', 'off');
display_plot(xplot, yplot, xOut1)

function xsolOut = save(xsol, y)
% For communication with main routine.
global ykeep ncall xkeep;

% This version of the intermediate output routine stores the values
% (so they can be plotted in the main routine).
ncall = ncall+1;
ykeep(ncall,:) = y;
xkeep(ncall,:) = xsol;
fprintf('%2d   %8.4f   %8.4f   %8.4f\n', xsol, y(1), y(2), y(3));
xsolOut = xsol + 2;
function xsolOut = output(xsol, y)
% Output intermediate values of solution.
fprintf('%2d   %8.4f   %8.4f   %8.4f\n', xsol, y(1), y(2), y(3));
xsolOut = xsol + 2;
function f = fcn(x,y)
% Evaluate the derivatives.
f = zeros(3,1);
f(1) = tan(y(3));
f(2) = -0.032*tan(y(3))/y(2) - 0.02*y(2)/cos(y(3));
f(3) = -0.032/y(2)^2;
function result = g(x,y)
% Evaluate g(x,y) when root-finding option is selected.
result = y(1);
function display_plot(xplot, yplot, xOut1)
% Formatting for title and axis labels.
titleFmt = {'FontName', 'Helvetica', 'FontWeight', 'Bold', 'FontSize', 14};
labFmt = {'FontName', 'Helvetica', 'FontWeight', 'Bold', 'FontSize', 13};
set(gca, 'FontSize', 13); % for legend, axis tick labels, etc.
% Plot the three curves.
plot(xplot, yplot(:,1), '-+', ...
     xplot, yplot(:,2), '--x', ...
     xplot, yplot(:,3), ':*');
% Mark the height=0 point.
do_stem(xplot, yplot, xOut1);
% Add title.
title('ODE Solution using Runge-Kutta with Root-finding', titleFmt{:});
% Label the axes.
xlabel('x', labFmt{:});
ylabel('Solution', labFmt{:});
% Add legend.
legend('height','velocity','angle','height = 0','Location','Best');
function do_stem(xplot, yplot, xOut1)
% Find the x bin that xOut1 lies in.
for i = 1:length(xplot)
    if xplot(i) > xOut1
        break
    end
end
% Use linear interpolation to find the corresponding y values on the
% two curves.
a1 = xplot(i)-xplot(i-1);
b1 = yplot(i,2)-yplot(i-1,2);
c1 = xOut1-xplot(i-1);
d1 = c1/a1;
e1 = d1*b1;
f1 = e1+yplot(i-1,2);
a2 = xplot(i)-xplot(i-1);
b2 = yplot(i,3)-yplot(i-1,3);
c2 = xOut1-xplot(i-1);
d2 = c2/a2;
e2 = d2*b2;
f2 = e2+yplot(i-1,3);
% Plot the line from the x axis to the two y values.
hold on
stem([xOut1,xOut1],[f1,0],'k:s');
hold on
stem([xOut1,xOut1],[f2,0],'k:s');
 
d02bj example program results

Case 1: intermediate output, root-finding

Calculation with tol = 0.0001
 X     Y(1)       Y(2)       Y(3)
 0     0.5000     0.5000     0.6283
 2     1.5493     0.4055     0.3066
 4     1.7423     0.3743    -0.1289
 6     1.0055     0.4173    -0.5507
 
Root of Y(1) = 0.0 at 7.2882
Solution is
   -0.0000     0.4749    -0.7601

Calculation with tol = 1e-05
 X     Y(1)       Y(2)       Y(3)
 0     0.5000     0.5000     0.6283
 2     1.5493     0.4055     0.3066
 4     1.7423     0.3743    -0.1289
 6     1.0055     0.4173    -0.5507
 
Root of Y(1) = 0.0 at 7.2883
Solution is
    0.0000     0.4749    -0.7601

Case 2: no intermediate output, root-finding 

Calculation with tol = 0.0001
Root of Y(1) = 0.0 at 7.2882
Solution is
   -0.0000     0.4749    -0.7601

Calculation with tol = 1e-05
Root of Y(1) = 0.0 at 7.2883
Solution is
    0.0000     0.4749    -0.7601

Case 3: intermediate output, no root-finding

Calculation with tol = 0.0001
 X     Y(1)       Y(2)       Y(3) 
 0     0.5000     0.5000     0.6283
 2     1.5493     0.4055     0.3066
 4     1.7423     0.3743    -0.1289
 6     1.0055     0.4173    -0.5507
 8    -0.7460     0.5130    -0.8537
10    -3.6283     0.6333    -1.0515

Calculation with tol = 1e-05
 X     Y(1)       Y(2)       Y(3) 
 0     0.5000     0.5000     0.6283
 2     1.5493     0.4055     0.3066
 4     1.7423     0.3743    -0.1289
 6     1.0055     0.4173    -0.5507
 8    -0.7459     0.5130    -0.8537
10    -3.6282     0.6333    -1.0515

Case 4: no intermediate output, no root-finding (integrate to xend)

Calculation with tol = 0.0001
 X     Y(1)       Y(2)       Y(3)
 0     0.5000     0.5000     0.6283
10    -3.6283     0.6333    -1.0515

Calculation with tol = 1e-05
 X     Y(1)       Y(2)       Y(3)
 0     0.5000     0.5000     0.6283
10    -3.6282     0.6333    -1.0515



PDF version (NAG web site, 64-bit version, 64-bit version)
Chapter Contents
Chapter Introduction
NAG Toolbox

© The Numerical Algorithms Group Ltd, Oxford, UK. 2009–2013