eq.m

来自「HERE IS A GOOD PSO TOOL BOX」· M 代码 · 共 55 行

function r = eq(lhs,rhs)
% POSYNOMIAL/EQ  Implements '==' for posynomials.
%

sz1 = size(lhs); sz2 = size(rhs);
if( sz1(1) > 1 & sz1(2) > 1)
  error(['Cannot impose pointwise equality with matrices of posynomials.' ...
         char(10) 'The pointwise equality is only allowed between vectors.'])
end
if( sz2(1) > 1 & sz2(2) > 1)
  error(['Cannot impose pointwise equality with matrices of posynomials.' ...
         char(10) 'The pointwise equality is only allowed between vectors.'])
end
if( sz1(1) ~= sz2(1) || sz1(2) ~= sz2(2))
  error(['Cannot make pointwise equality between vectors ' ...
         'with incompatible dimensions.'])
end

% constructing a single equality
if( length(lhs) == 1 & length(rhs) == 1 )
  % check if lhs is a monomial
  if ismonomial(lhs)
    lhs = lhs.monomials{1};
    % now check what is the rhs
    if (isa(rhs,'posynomial'))
      if ismonomial(rhs)
        rhs = rhs.monomials{1};
      else
        error('Not a valid GP equality: right hand side has to be a monomial.')
      end
    end
  else
    error('Not a valid GP equality: left hand side has to be a monomial.')
  end
  % create a GP equality constraint
  r = gpconstraint(lhs,'==',rhs);
  return;
end

% pointwise equality between vectors
if( sz1(1) == 1 & sz2(1) == 1 )
  for k = 1:sz1(2)
    r(1,k) = lhs(1,k) == rhs(1,k);
  end
  return;
end

if( sz1(2) == 1 & sz2(2) == 1 )
  for k = 1:sz1(1)
    r(k,1) = lhs(k,1) == rhs(k,1);
  end
  return;
end