home *** CD-ROM | disk | FTP | other *** search
/ OS/2 Shareware BBS: 10 Tools / 10-Tools.zip / octa21eb.zip / octave / SCRIPTS.ZIP / scripts / control / zgfslv.m < prev    next >
Text File  |  1999-03-05  |  2KB  |  65 lines
  1. # Copyright (C) 1996,1998 A. Scottedward Hodel 
  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, 675 Mass Ave, Cambridge, MA 02139, USA. 
  18.  
  19. function x = zgfslv(n,m,p,b)
  20.   # x = zgfslv(n,m,p,b)
  21.   # solve system of equations for dense zgep problem
  22.   
  23.   # Written by A. Scotte Hodel
  24.   # Converted to Octave by R Bruce Tenison, July 3, 1994
  25.  
  26.   nmp = n+m+p;
  27.   gam1 = (2*n)+m+p;    gam2 = n+p;     gam3 = n+m;
  28.  
  29.   G1 = givens(sqrt(m),-sqrt(p))';
  30.   G2 = givens(m+p,sqrt(n*(m+p)))';
  31.  
  32.   x = b;
  33.  
  34.   # 1) U1 e^n = sqrt(n)e_1^n
  35.   # 2) U2 e^m = sqrt(m)e_1^m
  36.   # 3) U3 e^p = sqrt(p)e_1^p
  37.   xdx1 = 1:n; xdx2 = n+(1:m); xdx3 = n+m+(1:p);
  38.   x(xdx1,1) = zgshsr(x(xdx1,1));
  39.   x(xdx2,1) = zgshsr(x(xdx2,1));
  40.   x(xdx3,1) = zgshsr(x(xdx3,1));
  41.  
  42.   # 4) Givens rotations to reduce stray non-zero elements
  43.   idx1 = [n+1,n+m+1];     idx2 = [1,n+1];
  44.   x(idx1) = G1'*x(idx1);
  45.   x(idx2) = G2'*x(idx2);
  46.  
  47.   # 6) Scale x, then back-transform to get x
  48.   en = ones(n,1);  em = ones(m,1);   ep = ones(p,1);
  49.   lam = [gam1*en;gam2*em;gam3*ep]; 
  50.   lam(1) = n+m+p; 
  51.   lam(n+1) = 1;       # dummy value to avoid divide by zero
  52.   lam(n+m+1)=n+m+p;
  53.  
  54.   x = x ./ lam;       x(n+1) = 0;  # minimum norm solution
  55.  
  56.   # back transform now.
  57.   x(idx2) = G2*x(idx2);
  58.   x(idx1) = G1*x(idx1);
  59.   x(xdx3,1) = zgshsr(x(xdx3,1));
  60.   x(xdx2,1) = zgshsr(x(xdx2,1));
  61.   x(xdx1,1) = zgshsr(x(xdx1,1));
  62.  
  63. endfunction
  64.  
  65.