| [w,mdw,me,ma,n,l,prgopt,x,rnorm,mode,iwork,work]=wnnls(w,mdw,me,ma,n,l,prgopt,x,rnorm,mode,iwork,work); |
function [w,mdw,me,ma,n,l,prgopt,x,rnorm,mode,iwork,work]=wnnls(w,mdw,me,ma,n,l,prgopt,x,rnorm,mode,iwork,work);
%***BEGIN PROLOGUE WNNLS
%***PURPOSE Solve a linearly constrained least squares problem with
% equality constraints and nonnegativity constraints on
% selected variables.
%***LIBRARY SLATEC
%***CATEGORY K1A2A
%***TYPE SINGLE PRECISION (WNNLS-S, DWNNLS-D)
%***KEYWORDS CONSTRAINED LEAST SQUARES, CURVE FITTING, DATA FITTING,
% EQUALITY CONSTRAINTS, INEQUALITY CONSTRAINTS,
% NONNEGATIVITY CONSTRAINTS, QUADRATIC PROGRAMMING
%***AUTHOR Hanson, R. J., (SNLA)
% Haskell, K. H., (SNLA)
%***DESCRIPTION
%
% Abstract
%
% This subprogram solves a linearly constrained least squares
% problem. Suppose there are given matrices E and A of
% respective dimensions ME by N and MA by N, and vectors F
% and B of respective lengths ME and MA. This subroutine
% solves the problem
%
% EX = F, (equations to be exactly satisfied)
%
% AX = B, (equations to be approximately satisfied,
% in the least squares sense)
%
% subject to components L+1,...,N nonnegative
%
% Any values ME.GE.0, MA.GE.0 and 0.LE. L .LE.N are permitted.
%
% The problem is reposed as problem WNNLS
%
% (WT*E)X = (WT*F)
% ( A) ( B), (least squares)
% subject to components L+1,...,N nonnegative.
%
% The subprogram chooses the heavy weight (or penalty parameter) WT.
%
% The parameters for WNNLS are
%
% INPUT..
%
% W(*,*),MDW, The array W(*,*) is double subscripted with first
% ME,MA,N,L dimensioning parameter equal to MDW. For this
% discussion let us call M = ME + MA. Then MDW
% must satisfy MDW.GE.M. The condition MDW.LT.M
% is an error.
%
% The array W(*,*) contains the matrices and vectors
%
% (E F)
% (A B)
%
% in rows and columns 1,...,M and 1,...,N+1
% respectively. Columns 1,...,L correspond to
% unconstrained variables X(1),...,X(L). The
% remaining variables are constrained to be
% nonnegative. The condition L.LT.0 or L.GT.N is
% an error.
%
% PRGOPT(*) This real-valued array is the option vector.
% If the user is satisfied with the nominal
% subprogram features set
%
% PRGOPT(1)=1 (or PRGOPT(1)=1.0)
%
% Otherwise PRGOPT(*) is a linked list consisting of
% groups of data of the following form
%
% LINK
% KEY
% DATA SET
%
% The parameters LINK and KEY are each one word.
% The DATA SET can be comprised of several words.
% The number of items depends on the value of KEY.
% The value of LINK points to the first
% entry of the next group of data within
% PRGOPT(*). The exception is when there are
% no more options to change. In that
% case LINK=1 and the values KEY and DATA SET
% are not referenced. The general layout of
% PRGOPT(*) is as follows.
%
% ...PRGOPT(1)=LINK1 (link to first entry of next group)
% . PRGOPT(2)=KEY1 (key to the option change)
% . PRGOPT(3)=DATA VALUE (data value for this change)
% . .
% . .
% . .
% ...PRGOPT(LINK1)=LINK2 (link to the first entry of
% . next group)
% . PRGOPT(LINK1+1)=KEY2 (key to the option change)
% . PRGOPT(LINK1+2)=DATA VALUE
% ... .
% . .
% . .
% ...PRGOPT(LINK)=1 (no more options to change)
%
% Values of LINK that are nonpositive are errors.
% A value of LINK.GT.NLINK=100000 is also an error.
% This helps prevent using invalid but positive
% values of LINK that will probably extend
% beyond the program limits of PRGOPT(*).
% Unrecognized values of KEY are ignored. The
% order of the options is arbitrary and any number
% of options can be changed with the following
% restriction. To prevent cycling in the
% processing of the option array a count of the
% number of options changed is maintained.
% Whenever this count exceeds NOPT=1000 an error
% message is printed and the subprogram returns.
%
% OPTIONS..
%
% KEY=6
% Scale the nonzero columns of the
% entire data matrix
% (E)
% (A)
% to have length one. The DATA SET for
% this option is a single value. It must
% be nonzero if unit length column scaling is
% desired.
%
% KEY=7
% Scale columns of the entire data matrix
% (E)
% (A)
% with a user-provided diagonal matrix.
% The DATA SET for this option consists
% of the N diagonal scaling factors, one for
% each matrix column.
%
% KEY=8
% Change the rank determination tolerance from
% the nominal value of SQRT(SRELPR). This quantity
% can be no smaller than SRELPR, The arithmetic-
% storage precision. The quantity used
% here is internally restricted to be at
% least SRELPR. The DATA SET for this option
% is the new tolerance.
%
% KEY=9
% Change the blow-up parameter from the
% nominal value of SQRT(SRELPR). The reciprocal of
% this parameter is used in rejecting solution
% components as too large when a variable is
% first brought into the active set. Too large
% means that the proposed component times the
% reciprocal of the parameter is not less than
% the ratio of the norms of the right-side
% vector and the data matrix.
% This parameter can be no smaller than SRELPR,
% the arithmetic-storage precision.
%
% For example, suppose we want to provide
% a diagonal matrix to scale the problem
% matrix and change the tolerance used for
% determining linear dependence of dropped col
% vectors. For these options the dimensions of
% PRGOPT(*) must be at least N+6. The FORTRAN
% statements defining these options would
% be as follows.
%
% PRGOPT(1)=N+3 (link to entry N+3 in PRGOPT(*))
% PRGOPT(2)=7 (user-provided scaling key)
%
% CALL SCOPY(N,D,1,PRGOPT(3),1) (copy the N
% scaling factors from a user array called D(*)
% into PRGOPT(3)-PRGOPT(N+2))
%
% PRGOPT(N+3)=N+6 (link to entry N+6 of PRGOPT(*))
% PRGOPT(N+4)=8 (linear dependence tolerance key)
% PRGOPT(N+5)=... (new value of the tolerance)
%
% PRGOPT(N+6)=1 (no more options to change)
%
%
% IWORK(1), The amounts of working storage actually allocated
% IWORK(2) for the working arrays WORK(*) and IWORK(*),
% respectively. These quantities are compared with
% the actual amounts of storage needed for WNNLS( ).
% Insufficient storage allocated for either WORK(*)
% or IWORK(*) is considered an error. This feature
% was included in WNNLS( ) because miscalculating
% the storage formulas for WORK(*) and IWORK(*)
% might very well lead to subtle and hard-to-find
% execution errors.
%
% The length of WORK(*) must be at least
%
% LW = ME+MA+5*N
% This test will not be made if IWORK(1).LE.0.
%
% The length of IWORK(*) must be at least
%
% LIW = ME+MA+N
% This test will not be made if IWORK(2).LE.0.
persistent l1 l2 l3 l4 l5 liw lw xern1 ;
if isempty(l1), l1=0; end;
if isempty(l2), l2=0; end;
if isempty(l3), l3=0; end;
if isempty(l4), l4=0; end;
if isempty(l5), l5=0; end;
if isempty(liw), liw=0; end;
if isempty(lw), lw=0; end;
%
% OUTPUT..
%
% X(*) An array dimensioned at least N, which will
% contain the N components of the solution vector
% on output.
%
% RNORM The residual norm of the solution. The value of
% RNORM contains the residual vector length of the
% equality constraints and least squares equations.
%
% MODE The value of MODE indicates the success or failure
% of the subprogram.
%
% MODE = 0 Subprogram completed successfully.
%
% = 1 Max. number of iterations (equal to
% 3*(N-L)) exceeded. Nearly all problems
% should complete in fewer than this
% number of iterations. An approximate
% solution and its corresponding residual
% vector length are in X(*) and RNORM.
%
% = 2 Usage error occurred. The offending
% condition is noted with the error
% processing subprogram, XERMSG( ).
%
% User-designated
% Working arrays..
%
% WORK(*) A real-valued working array of length at least
% M + 5*N.
%
% IWORK(*) An integer-valued working array of length at least
% M+N.
%
%***REFERENCES K. H. Haskell and R. J. Hanson, An algorithm for
% linear least squares problems with equality and
% nonnegativity constraints, Report SAND77-0552, Sandia
% Laboratories, June 1978.
% K. H. Haskell and R. J. Hanson, Selected algorithms for
% the linearly constrained least squares problem - a
% users guide, Report SAND78-1290, Sandia Laboratories,
% August 1979.
% K. H. Haskell and R. J. Hanson, An algorithm for
% linear least squares problems with equality and
% nonnegativity constraints, Mathematical Programming
% 21 (1981), pp. 98-118.
% R. J. Hanson and K. H. Haskell, Two algorithms for the
% linearly constrained least squares problem, ACM
% Transactions on Mathematical Software, September 1982.
% C. L. Lawson and R. J. Hanson, Solving Least Squares
% Problems, Prentice-Hall, Inc., 1974.
%***ROUTINES CALLED WNLSM, XERMSG
%***REVISION HISTORY (YYMMDD)
% 790701 DATE WRITTEN
% 890206 REVISION DATE from Version 3.2
% 890618 Completely restructured and revised. (WRBRWC)
% 891214 Prologue converted to Version 4.0 format. (BAB)
% 900315 CALLs to XERROR changed to CALLs to XERMSG. (THJ)
% 900510 Convert XERRWV calls to XERMSG calls. (RWC)
% 920501 Reformatted the REFERENCES section. (WRB)
%***end PROLOGUE WNNLS
prgopt_shape=size(prgopt);prgopt=reshape(prgopt,1,[]);
w_shape=size(w);w=reshape([w(:).',zeros(1,ceil(numel(w)./prod([mdw])).*prod([mdw])-numel(w))],mdw,[]);
work_shape=size(work);work=reshape(work,1,[]);
x_shape=size(x);x=reshape(x,1,[]);
iwork_shape=size(iwork);iwork=reshape(iwork,1,[]);
if isempty(xern1), xern1=repmat(' ',1,8); end;
%
%
%***FIRST EXECUTABLE STATEMENT WNNLS
mode = 0;
if( ma+me<=0 || n<=0 )
prgopt_shape=zeros(prgopt_shape);prgopt_shape(:)=prgopt(1:numel(prgopt_shape));prgopt=prgopt_shape;
w_shape=zeros(w_shape);w_shape(:)=w(1:numel(w_shape));w=w_shape;
work_shape=zeros(work_shape);work_shape(:)=work(1:numel(work_shape));work=work_shape;
x_shape=zeros(x_shape);x_shape(:)=x(1:numel(x_shape));x=x_shape;
iwork_shape=zeros(iwork_shape);iwork_shape(:)=iwork(1:numel(iwork_shape));iwork=iwork_shape;
return;
end;
if( iwork(1)>0 )
lw = fix(me + ma + 5.*n);
if( iwork(1)<lw )
xern1=sprintf(['%8i'], lw);
xermsg('SLATEC','WNNLS',['INSUFFICIENT STORAGE ',['ALLOCATED FOR WORK(*), NEED LW = ',xern1]],2,1);
mode = 2;
prgopt_shape=zeros(prgopt_shape);prgopt_shape(:)=prgopt(1:numel(prgopt_shape));prgopt=prgopt_shape;
w_shape=zeros(w_shape);w_shape(:)=w(1:numel(w_shape));w=w_shape;
work_shape=zeros(work_shape);work_shape(:)=work(1:numel(work_shape));work=work_shape;
x_shape=zeros(x_shape);x_shape(:)=x(1:numel(x_shape));x=x_shape;
iwork_shape=zeros(iwork_shape);iwork_shape(:)=iwork(1:numel(iwork_shape));iwork=iwork_shape;
return;
end;
end;
%
if( iwork(2)>0 )
liw = fix(me + ma + n);
if( iwork(2)<liw )
xern1=sprintf(['%8i'], liw);
xermsg('SLATEC','WNNLS',['INSUFFICIENT STORAGE ',['ALLOCATED FOR IWORK(*), NEED LIW = ',xern1]],2,1);
mode = 2;
prgopt_shape=zeros(prgopt_shape);prgopt_shape(:)=prgopt(1:numel(prgopt_shape));prgopt=prgopt_shape;
w_shape=zeros(w_shape);w_shape(:)=w(1:numel(w_shape));w=w_shape;
work_shape=zeros(work_shape);work_shape(:)=work(1:numel(work_shape));work=work_shape;
x_shape=zeros(x_shape);x_shape(:)=x(1:numel(x_shape));x=x_shape;
iwork_shape=zeros(iwork_shape);iwork_shape(:)=iwork(1:numel(iwork_shape));iwork=iwork_shape;
return;
end;
end;
%
if( mdw<me+ma )
xermsg('SLATEC','WNNLS','THE VALUE MDW.LT.ME+MA IS AN ERROR',1,1);
mode = 2;
prgopt_shape=zeros(prgopt_shape);prgopt_shape(:)=prgopt(1:numel(prgopt_shape));prgopt=prgopt_shape;
w_shape=zeros(w_shape);w_shape(:)=w(1:numel(w_shape));w=w_shape;
work_shape=zeros(work_shape);work_shape(:)=work(1:numel(work_shape));work=work_shape;
x_shape=zeros(x_shape);x_shape(:)=x(1:numel(x_shape));x=x_shape;
iwork_shape=zeros(iwork_shape);iwork_shape(:)=iwork(1:numel(iwork_shape));iwork=iwork_shape;
return;
end;
%
if( l<0 || l>n )
xermsg('SLATEC','WNNLS','L.GE.0 .AND. L.LE.N IS REQUIRED',2,1);
mode = 2;
prgopt_shape=zeros(prgopt_shape);prgopt_shape(:)=prgopt(1:numel(prgopt_shape));prgopt=prgopt_shape;
w_shape=zeros(w_shape);w_shape(:)=w(1:numel(w_shape));w=w_shape;
work_shape=zeros(work_shape);work_shape(:)=work(1:numel(work_shape));work=work_shape;
x_shape=zeros(x_shape);x_shape(:)=x(1:numel(x_shape));x=x_shape;
iwork_shape=zeros(iwork_shape);iwork_shape(:)=iwork(1:numel(iwork_shape));iwork=iwork_shape;
return;
end;
%
% THE PURPOSE OF THIS SUBROUTINE IS TO BREAK UP THE ARRAYS
% WORK(*) AND IWORK(*) INTO SEPARATE WORK ARRAYS
% REQUIRED BY THE MAIN SUBROUTINE WNLSM( ).
%
l1 = fix(n + 1);
l2 = fix(l1 + n);
l3 = fix(l2 + me + ma);
l4 = fix(l3 + n);
l5 = fix(l4 + n);
%
[w,mdw,me,ma,n,l,prgopt,x,rnorm,mode,iwork,dumvar12,dumvar13,dumvar14,dumvar15,dumvar16,dumvar17,dumvar18]=wnlsm(w,mdw,me,ma,n,l,prgopt,x,rnorm,mode,iwork,iwork(sub2ind(size(iwork),max(l1,1)):end),work(sub2ind(size(work),max(1,1)):end),work(sub2ind(size(work),max(l1,1)):end),work(sub2ind(size(work),max(l2,1)):end),work(sub2ind(size(work),max(l3,1)):end),work(sub2ind(size(work),max(l4,1)):end),work(sub2ind(size(work),max(l5,1)):end)); dumvar12i=find((iwork(sub2ind(size(iwork),max(l1,1)):end))~=(dumvar12));dumvar13i=find((work(sub2ind(size(work),max(1,1)):end))~=(dumvar13));dumvar14i=find((work(sub2ind(size(work),max(l1,1)):end))~=(dumvar14));dumvar15i=find((work(sub2ind(size(work),max(l2,1)):end))~=(dumvar15));dumvar16i=find((work(sub2ind(size(work),max(l3,1)):end))~=(dumvar16));dumvar17i=find((work(sub2ind(size(work),max(l4,1)):end))~=(dumvar17));dumvar18i=find((work(sub2ind(size(work),max(l5,1)):end))~=(dumvar18)); iwork(l1-1+dumvar12i)=dumvar12(dumvar12i); work(1-1+dumvar13i)=dumvar13(dumvar13i); work(l1-1+dumvar14i)=dumvar14(dumvar14i); work(l2-1+dumvar15i)=dumvar15(dumvar15i); work(l3-1+dumvar16i)=dumvar16(dumvar16i); work(l4-1+dumvar17i)=dumvar17(dumvar17i); work(l5-1+dumvar18i)=dumvar18(dumvar18i);
prgopt_shape=zeros(prgopt_shape);prgopt_shape(:)=prgopt(1:numel(prgopt_shape));prgopt=prgopt_shape;
w_shape=zeros(w_shape);w_shape(:)=w(1:numel(w_shape));w=w_shape;
work_shape=zeros(work_shape);work_shape(:)=work(1:numel(work_shape));work=work_shape;
x_shape=zeros(x_shape);x_shape(:)=x(1:numel(x_shape));x=x_shape;
iwork_shape=zeros(iwork_shape);iwork_shape(:)=iwork(1:numel(iwork_shape));iwork=iwork_shape;
end
%DECK XADD
|
|