source_dir = getenv('source_root');
addpath([source_dir filesep 'matlab']);

dynare_config;

cd solver-test-functions

testFailed = 0;

skipline()
disp('***  TESTING: nonlinearsolvers.m ***');

tolf = 1e-6;
tolx = 1e-6;
maxit = 50;
factor = 10;

% List of function handles
objfun = { @rosenbrock,
           @powell1,
           @powell2,
           @wood,
           @helicalvalley,
           @watson,
           @chebyquad,
           @brown,
           @discreteboundaryvalue,
           @discreteintegralequation,
           @trigonometric,
           @variablydimensioned,
           @broydentridiagonal,
           @broydenbanded };

% FIXME block_trust_region (mex) or trust_region (matlab) do not work for all n (not sure we can fix that).

%
% Test mex routine
%

t0 = clock;

for i=1:length(objfun)
    switch func2str(objfun{i})
         case 'helicalvalley'
           % FIXME block_trust_region is diverging if x(1)<0.
           x = helicalvalley();
           x(1) = 5;
      case 'chebyquad'
        % Fails with a system of 10 equations. 
        x = objfun{i}(nan(9,1));
      case {'watson', 'brown', 'discreteintegralequation', 'discreteboundaryvalue', 'chebyquad', 'trigonometric', 'variablydimensioned', 'broydenbanded', 'broydentridiagonal'}
        x = objfun{i}(nan(4,1));
      otherwise
        x = objfun{i}();
    end
    try
        [x, errorflag, exitflag] = block_trust_region(objfun{i}, x, tolf, tolx, maxit, factor, true, false);
        if isequal(func2str(objfun{i}), 'powell2')
            if ~errorflag
                testFailed = testFailed+1;
                dprintf('Nonlinear solver is expected to fail on %s function but did not return an error.', func2str(objfun{i}))
            end
        elseif isequal(func2str(objfun{i}), 'trigonometric')
            % FIXME block_trust_region (mex) fails, with exit code equal to 4, but not trust_region (matlab). Would be nice to undertsand the difference. 
            if ~errorflag
                testFailed = testFailed+1;
                dprintf('Nonlinear solver is expected to fail on %s function but did not return an error.', func2str(objfun{i}))
            end
        else
            if errorflag || norm(objfun{i}(x))>tolf
                testFailed = testFailed+1;
                dprintf('Nonlinear solver (mex) failed on %s function (norm(f(x))=%s).', func2str(objfun{i}), num2str(norm(objfun{i}(x))))
            end
        end
    catch
        testFailed = testFailed+1;
        dprintf('Nonlinear solver (mex) failed on %s function.', func2str(objfun{i}))
    end
end

t1 = clock; etime(t1, t0)

%
% Test matlab routine
%

for i=1:length(objfun)
    switch func2str(objfun{i})
      case 'chebyquad'
        % Fails with a system of 10 equations. 
        x = objfun{i}(nan(9,1));
      case {'watson', 'brown', 'discreteintegralequation', 'discreteboundaryvalue', 'trigonometric', 'variablydimensioned', 'broydenbanded', 'broydentridiagonal'}
        x = objfun{i}(nan(10,1));
      otherwise
        x = objfun{i}();
    end
    try
        [x, errorflag, info] = trust_region(objfun{i}, x, 1:length(x), 1:length(x), true, [], tolf, tolx, maxit, factor);
        if isequal(func2str(objfun{i}), 'powell2')
            if ~errorflag
                testFailed = testFailed+1;
                dprintf('Nonlinear solver is expected to fail on %s function but did not return an error.', func2str(objfun{i}))
            end
            if info~=3
                testFailed = testFailed+1;
                dprintf('Nonlinear solver is expected to fail on %s function with info==3 but did not the correct value of info.', func2str(objfun{i}))
            end
        else
            if errorflag
                testFailed = testFailed+1;
                dprintf('Nonlinear solver failed on %s function (info=%s).', func2str(objfun{i}), int2str(info))
            end
        end
    catch
        testFailed = testFailed+1;
        dprintf('Nonlinear solver failed on %s function.', func2str(objfun{i}))
    end
end

t2 = clock;

fprintf('\n*** Elapsed time (in seconds): %.1f\n\n', etime(t2, t0));

quit(testFailed > 0)