sort_internal.m

来自「matlab波形优化算法经常要用到的matlab toolbox工具箱:yalm」· M 代码 · 共 47 行

function [F,vars] = sort_internal(t,X,data);%ii,D,V);

% Hack to figure out all the sorted variables, not just this index.
% Sort is implemented in a slighlt different way (general feature in
% future versions) that allows one element in an operator to modell all
% elements. Reduces the number of calls to the operator code.

ii = data.i;
D = data.D;
V = data.V;

var_start = getvariables(t)-ii+1;
n = length(X);
var_end   = getvariables(t)-ii+1 + n-1;
vars = var_start:var_end;

% Is this a location variable instead of the actuial sort variable. If so,
% shift everything back to get the indicies of the sorted variables.
if data.isthisloc == 1
    vars = vars - n;
end

t   = recover(vars);
loc = recover(vars+n);

[M,m] = derivebounds(X);
X = reshape(X,1,n);

% Standard model
F = set(sum(D,1) == 1) + set(sum(D,2) == 1);
F = F + set(t == sum(V,2));
F = F + set(diff(t) > 0);
for i = 1:n
   di = D(i,:);
   vi = V(i,:);
   F = F + set(-(M-m)'.*(1-di) < X-vi < (M-m)'.*(1-di));
   F = F + set(m'.*di <  vi < M'.*di);   
end

% Cuts
F = F + set(X == sum(V,1));
F = F + set(sum(t) == sum(X));

% Definition of location
F = F + set(loc == D*[(1:n)']) + set(1 <= loc <= n);