home *** CD-ROM | disk | FTP | other *** search
/ OS/2 Shareware BBS: 10 Tools / 10-Tools.zip / octa21fb.zip / octave / SCRIPTS.ZIP / scripts / control / zgfslv.m < prev    next >
Encoding:
Text File  |  1999-12-15  |  2.1 KB  |  67 lines
  1. ## Copyright (C) 1996,1998 Auburn University.  All Rights Reserved
  2. ##
  3. ## This file is part of Octave. 
  4. ##
  5. ## Octave is free software; you can redistribute it and/or modify it 
  6. ## under the terms of the GNU General Public License as published by the 
  7. ## Free Software Foundation; either version 2, or (at your option) any 
  8. ## later version. 
  9. ## 
  10. ## Octave is distributed in the hope that it will be useful, but WITHOUT 
  11. ## ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 
  12. ## FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License 
  13. ## for more details.
  14. ## 
  15. ## You should have received a copy of the GNU General Public License 
  16. ## along with Octave; see the file COPYING.  If not, write to the Free 
  17. ## Software Foundation, 59 Temple Place, Suite 330, Boston, MA 02111 USA. 
  18.  
  19. ## -*- texinfo -*-
  20. ## @deftypefn {Function File } {x =} zgfslv(@var{n},@var{m},@var{p},@var{b})
  21. ## solve system of equations for dense zgep problem
  22. ## @end deftypefn
  23.  
  24. function x = zgfslv (n, m, p, b)
  25.   ## Written by A. Scotte Hodel
  26.   ## Converted to Octave by R Bruce Tenison, July 3, 1994
  27.  
  28.   nmp = n+m+p;
  29.   gam1 = (2*n)+m+p;    gam2 = n+p;     gam3 = n+m;
  30.  
  31.   G1 = givens(sqrt(m),-sqrt(p))';
  32.   G2 = givens(m+p,sqrt(n*(m+p)))';
  33.  
  34.   x = b;
  35.  
  36.   ## 1) U1 e^n = sqrt(n)e_1^n
  37.   ## 2) U2 e^m = sqrt(m)e_1^m
  38.   ## 3) U3 e^p = sqrt(p)e_1^p
  39.   xdx1 = 1:n; xdx2 = n+(1:m); xdx3 = n+m+(1:p);
  40.   x(xdx1,1) = zgshsr(x(xdx1,1));
  41.   x(xdx2,1) = zgshsr(x(xdx2,1));
  42.   x(xdx3,1) = zgshsr(x(xdx3,1));
  43.  
  44.   ## 4) Givens rotations to reduce stray non-zero elements
  45.   idx1 = [n+1,n+m+1];     idx2 = [1,n+1];
  46.   x(idx1) = G1'*x(idx1);
  47.   x(idx2) = G2'*x(idx2);
  48.  
  49.   ## 6) Scale x, then back-transform to get x
  50.   en = ones(n,1);  em = ones(m,1);   ep = ones(p,1);
  51.   lam = [gam1*en;gam2*em;gam3*ep]; 
  52.   lam(1) = n+m+p; 
  53.   lam(n+1) = 1;       # dummy value to avoid divide by zero
  54.   lam(n+m+1)=n+m+p;
  55.  
  56.   x = x ./ lam;       x(n+1) = 0;  # minimum norm solution
  57.  
  58.   ## back transform now.
  59.   x(idx2) = G2*x(idx2);
  60.   x(idx1) = G1*x(idx1);
  61.   x(xdx3,1) = zgshsr(x(xdx3,1));
  62.   x(xdx2,1) = zgshsr(x(xdx2,1));
  63.   x(xdx1,1) = zgshsr(x(xdx1,1));
  64.  
  65. endfunction
  66.  
  67.