home *** CD-ROM | disk | FTP | other *** search
/ OS/2 Shareware BBS: 10 Tools / 10-Tools.zip / octa21fb.zip / octave / SCRIPTS.ZIP / scripts / specfun / erfinv.m < prev    next >
Encoding:
Text File  |  1999-10-26  |  1.9 KB  |  73 lines
  1. ## Copyright (C) 1995, 1996  Kurt Hornik
  2. ## 
  3. ## This program is free software; you can redistribute it and/or modify
  4. ## it under the terms of the GNU General Public License as published by
  5. ## the Free Software Foundation; either version 2, or (at your option)
  6. ## any later version.
  7. ## 
  8. ## This program is distributed in the hope that it will be useful, but
  9. ## WITHOUT ANY WARRANTY; without even the implied warranty of
  10. ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11. ## General Public License for more details. 
  12. ## 
  13. ## You should have received a copy of the GNU General Public License
  14. ## along with this file.  If not, write to the Free Software Foundation,
  15. ## 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  16.  
  17. ## -*- texinfo -*-
  18. ## @deftypefn {Mapping Function} {} erfinv (@var{z})
  19. ## Computes the inverse of the error function,
  20. ## @end deftypefn
  21.  
  22. ##  See also: erf, erfc
  23.  
  24. ## Author: KH <Kurt.Hornik@ci.tuwien.ac.at>
  25. ## Created: 27 September 1994
  26. ## Adapted-By: jwe
  27.  
  28. function [y, iterations] = erfinv (x)
  29.   
  30.   if (nargin != 1)
  31.     usage ("erfinv (x)");
  32.   endif
  33.  
  34.   maxit = 100;
  35.   tol = eps;
  36.  
  37.   iterations = 0;
  38.  
  39.   [m, n] = size (x);  
  40.   x = reshape (x, m * n, 1);
  41.   y = zeros (m * n, 1);
  42.  
  43.   i = find ((x < -1) | (x > 1));
  44.   if any (i)
  45.     y(i) = NaN * ones (length (i), 1);
  46.   endif
  47.  
  48.   t = find (x == -1);
  49.   y (t) = (-Inf) * ones (size (t));
  50.  
  51.   t = find (x == 1);
  52.   y (t) = Inf * ones (size (t));
  53.  
  54.   i = find ((x > -1) & (x < 1));
  55.   if any (i)
  56.     s = sqrt (pi) / 2;
  57.     z_old = ones (length (i), 1);
  58.     z_new = sqrt (-log (1 - abs (x(i)))) .* sign (x(i));
  59.     while (any (abs (erf (z_old) - x(i)) > tol * abs (x(i))))
  60.       z_old = z_new;
  61.       z_new = z_old - (erf (z_old) - x(i)) .* exp (z_old.^2) * s;
  62.       if (++iterations > maxit)
  63.     warning ("erfinv: iteration limit exceeded");
  64.     break;
  65.       endif
  66.     endwhile
  67.     y(i) = z_new;
  68.   endif
  69.   
  70.   y = reshape (y, m, n);
  71.     
  72. endfunction
  73.