Dynamic-Calibration/utils/YALMIP-master/extras/@ncvar/plus.m

167 lines
4.9 KiB
Matlab
Executable File

function y = plus(X,Y)
%PLUS (overloaded)
if isa(X,'sdpvar')
X = ncvar(X);
elseif isa(Y,'sdpvar')
Y = ncvar(Y);
end
X_is_ncvar = isa(X,'ncvar');
Y_is_ncvar = isa(Y,'ncvar');
switch 2*X_is_ncvar+Y_is_ncvar
case 1
if isempty(X)
try
y = full(X - reshape(Y.basis(:,1),Y.dim(1),Y.dim(2)));
catch
error(lasterr);
end
return
end
y = Y;
n_Y = Y.dim(1);
m_Y = Y.dim(2);
[n_X,m_X] = size(X);
x_isscalar = (n_X*m_X==1);
y_isscalar = (n_Y*m_Y==1);
any_scalar = x_isscalar | y_isscalar;
if x_isscalar & y_isscalar
y.basis(1) = y.basis(1)+X;
% Reset info about conic terms
y.conicinfo = [0 0];
return
end
if any_scalar | ([n_Y m_Y]==[n_X m_X])
if y_isscalar
y.basis = repmat(y.basis,n_X*m_X,1);
y.dim(1) = n_X;
y.dim(2) = m_X;
end
y.basis(:,1) = y.basis(:,1)+X(:);
else
error('Matrix dimensions must agree.');
end
% Reset info about conic terms
y.conicinfo = [0 0];
case 2
if isempty(Y)
try
y = full(reshape(X.basis(:,1),X.dim(1),X.dim(2))-Y);
catch
error(lasterr);
end
return
end
y = X;
n_X = X.dim(1);
m_X = X.dim(2);
[n_Y,m_Y] = size(Y);
x_isscalar = (n_X*m_X==1);
y_isscalar = (n_Y*m_Y==1);
any_scalar = x_isscalar | y_isscalar;
% Special special case...
if x_isscalar & y_isscalar
y.basis(1) = y.basis(1)+Y;
% Reset info about conic terms
y.conicinfo = [0 0];
return
end
if any_scalar | ([n_Y m_Y]==[n_X m_X])
if x_isscalar
y.basis = repmat(y.basis,n_Y*m_Y,1);
y.dim(1) = n_Y;
y.dim(2) = m_Y;
end
y.basis(:,1) = y.basis(:,1)+Y(:);
else
error('Matrix dimensions must agree.');
end
% Reset info about conic terms
y.conicinfo = [0 0];
case 3
n_X = X.dim(1);
m_X = X.dim(2);
n_Y = Y.dim(1);
m_Y = Y.dim(2);
x_isscalar = (n_X*m_X==1);
y_isscalar = (n_Y*m_Y==1);
any_scalar = x_isscalar | y_isscalar;
if (~((n_X==n_Y) & (m_X==m_Y))) & ~any_scalar
error('Matrix dimensions must agree.')
end
all_lmi_variables = uniquestripped([X.lmi_variables Y.lmi_variables]);
y = X;
X.basis = [];
y.lmi_variables = all_lmi_variables;
% ismembc faster (buggy?)
in_X = find(ismembc(all_lmi_variables,X.lmi_variables));
in_Y = find(ismembc(all_lmi_variables,Y.lmi_variables));
% in_X = find(ismember(all_lmi_variables,X.lmi_variables));
% in_Y = find(ismember(all_lmi_variables,Y.lmi_variables));
if isequal(X.lmi_variables,Y.lmi_variables) & n_Y==n_X & m_Y==m_X
y.basis = y.basis + Y.basis;
if length(X.lmi_variables)==1
if all(y.basis(:,2)==0)
y = full(y.basis(1));
else
% Reset info about conic terms
y.conicinfo = [0 0];
end
return
end
else
if 1
[ix,jx,sx] = find(y.basis);y.basis = [];
[iy,jy,sy] = find(Y.basis);Y.basis = [];
mapX = [1 1+in_X];
mapY = [1 1+in_Y];
basis_X = sparse(ix,mapX(jx),sx,n_X*m_X,1+length(all_lmi_variables));ix=[];jx=[];sx=[];
basis_Y = sparse(iy,mapY(jy),sy,n_Y*m_Y,1+length(all_lmi_variables));iy=[];jy=[];sy=[];
else
% MATLAB sparse fails on this for huge problems at a certain size
basis_X = spalloc(n_X*m_X,1+length(all_lmi_variables),nnz(X.basis));
basis_Y = spalloc(n_Y*m_Y,1+length(all_lmi_variables),nnz(Y.basis));
basis_X(:,[1 1+in_X])=y.basis;y.basis = [];
basis_Y(:,[1 1+in_Y])=Y.basis;Y.basis = [];
end
% Fix addition of matrix+scalar
if n_X*m_X<n_Y*m_Y
y.dim(1) = n_Y;
y.dim(2) = m_Y;
basis_X = repmat(basis_X,n_Y*m_Y,1);
end
if n_Y*m_Y<n_X*m_X
y.dim(1) = n_X;
y.dim(2) = m_X;
basis_Y = repmat(basis_Y,n_X*m_X,1);
end
% OK, solution is...
y.basis = basis_X;basis_X = [];
y.basis = y.basis+basis_Y;basis_Y = [];
end
% Reset info about conic terms
y.conicinfo = [0 0];
y = clean(y);
otherwise
end